1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
|
/*-
* Copyright (c) 2014-present MongoDB, Inc.
* Copyright (c) 2008-2014 WiredTiger, Inc.
* All rights reserved.
*
* See the file LICENSE for redistribution information.
*/
#include "util.h"
int
util_cerr(WT_CURSOR *cursor, const char *op, int ret)
{
return (util_err(cursor->session, ret, "%s: cursor.%s", cursor->uri, op));
}
/*
* util_err --
* Report an error.
*/
int
util_err(WT_SESSION *session, int e, const char *fmt, ...)
{
va_list ap;
(void)fprintf(stderr, "%s: ", progname);
if (fmt != NULL) {
va_start(ap, fmt);
(void)vfprintf(stderr, fmt, ap);
va_end(ap);
if (e != 0)
(void)fprintf(stderr, ": ");
}
if (e != 0)
(void)fprintf(
stderr, "%s", session == NULL ? wiredtiger_strerror(e) : session->strerror(session, e));
(void)fprintf(stderr, "\n");
return (1);
}
/*
* util_read_line --
* Read a line from stdin into a ULINE.
*/
int
util_read_line(WT_SESSION *session, ULINE *l, bool eof_expected, bool *eofp)
{
static uint64_t line = 0;
size_t len;
int ch;
++line;
*eofp = false;
if (l->memsize == 0) {
if ((l->mem = realloc(l->mem, l->memsize + 1024)) == NULL)
return (util_err(session, errno, NULL));
l->memsize = 1024;
}
for (len = 0;; ++len) {
if ((ch = getchar()) == EOF) {
if (len == 0) {
if (eof_expected) {
*eofp = true;
return (0);
}
return (util_err(session, 0, "line %" PRIu64 ": unexpected end-of-file", line));
}
return (util_err(session, 0, "line %" PRIu64 ": no newline terminator", line));
}
if (ch == '\n')
break;
/*
* We nul-terminate the string so it's easier to convert the line into a record number, that
* means we always need one extra byte at the end.
*/
if (len >= l->memsize - 1) {
if ((l->mem = realloc(l->mem, l->memsize + 1024)) == NULL)
return (util_err(session, errno, NULL));
l->memsize += 1024;
}
((uint8_t *)l->mem)[len] = (uint8_t)ch;
}
((uint8_t *)l->mem)[len] = '\0'; /* nul-terminate */
return (0);
}
/*
* util_str2num --
* Convert a string to a number.
*/
int
util_str2num(WT_SESSION *session, const char *p, bool endnul, uint64_t *vp)
{
uint64_t v;
char *endptr;
/*
* strtouq takes lots of things like hex values, signs and so on and so forth -- none of them
* are OK with us. Check the string starts with digit, that turns off the special processing.
*/
if (!__wt_isdigit((u_char)p[0]))
goto format;
errno = 0;
v = __wt_strtouq(p, &endptr, 0);
if (v == ULLONG_MAX && errno == ERANGE)
return (util_err(session, ERANGE, "%s: invalid number", p));
/*
* In most cases we expect the number to be a string and end with a nul byte (and we want to
* confirm that because it's a user-entered command-line argument), but we allow the caller to
* configure that test off.
*/
if (endnul && endptr[0] != '\0')
format:
return (util_err(session, EINVAL, "%s: invalid number", p));
*vp = v;
return (0);
}
/*
* util_flush --
* Flush the file successfully, or drop it.
*/
int
util_flush(WT_SESSION *session, const char *uri)
{
WT_DECL_RET;
size_t len;
char *buf;
len = strlen(uri) + 100;
if ((buf = malloc(len)) == NULL)
return (util_err(session, errno, NULL));
if ((ret = __wt_snprintf(buf, len, "target=(\"%s\")", uri)) != 0) {
free(buf);
return (util_err(session, ret, NULL));
}
ret = session->checkpoint(session, buf);
free(buf);
if (ret == 0)
return (0);
(void)util_err(session, ret, "%s: session.checkpoint", uri);
if ((ret = session->drop(session, uri, NULL)) != 0)
(void)util_err(session, ret, "%s: session.drop", uri);
return (1);
}
/*
* util_usage --
* Display a usage statement.
*/
void
util_usage(const char *usage, const char *tag, const char *list[])
{
const char **p;
if (usage != NULL)
fprintf(stderr, "usage: %s %s %s\n", progname, usage_prefix, usage);
if (tag != NULL)
fprintf(stderr, "%s\n", tag);
if (list != NULL)
for (p = list; *p != NULL; p += 2)
fprintf(stderr, " %s%s%s\n", p[0], strlen(p[0]) > 2 ? "\n " : " ", p[1]);
}
|
