]>
Commit | Line | Data |
---|---|---|
18607dcb FB |
1 | /* |
2 | * Simple C functions to supplement the C library | |
5fafdf24 | 3 | * |
18607dcb FB |
4 | * Copyright (c) 2006 Fabrice Bellard |
5 | * | |
6 | * Permission is hereby granted, free of charge, to any person obtaining a copy | |
7 | * of this software and associated documentation files (the "Software"), to deal | |
8 | * in the Software without restriction, including without limitation the rights | |
9 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
10 | * copies of the Software, and to permit persons to whom the Software is | |
11 | * furnished to do so, subject to the following conditions: | |
12 | * | |
13 | * The above copyright notice and this permission notice shall be included in | |
14 | * all copies or substantial portions of the Software. | |
15 | * | |
16 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
17 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
18 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL | |
19 | * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
20 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
21 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | |
22 | * THE SOFTWARE. | |
23 | */ | |
faf07963 | 24 | #include "qemu-common.h" |
8d371d4b | 25 | #include "host-utils.h" |
9f9b17a4 | 26 | #include <math.h> |
18607dcb FB |
27 | |
28 | void pstrcpy(char *buf, int buf_size, const char *str) | |
29 | { | |
30 | int c; | |
31 | char *q = buf; | |
32 | ||
33 | if (buf_size <= 0) | |
34 | return; | |
35 | ||
36 | for(;;) { | |
37 | c = *str++; | |
38 | if (c == 0 || q >= buf + buf_size - 1) | |
39 | break; | |
40 | *q++ = c; | |
41 | } | |
42 | *q = '\0'; | |
43 | } | |
44 | ||
45 | /* strcat and truncate. */ | |
46 | char *pstrcat(char *buf, int buf_size, const char *s) | |
47 | { | |
48 | int len; | |
49 | len = strlen(buf); | |
5fafdf24 | 50 | if (len < buf_size) |
18607dcb FB |
51 | pstrcpy(buf + len, buf_size - len, s); |
52 | return buf; | |
53 | } | |
54 | ||
55 | int strstart(const char *str, const char *val, const char **ptr) | |
56 | { | |
57 | const char *p, *q; | |
58 | p = str; | |
59 | q = val; | |
60 | while (*q != '\0') { | |
61 | if (*p != *q) | |
62 | return 0; | |
63 | p++; | |
64 | q++; | |
65 | } | |
66 | if (ptr) | |
67 | *ptr = p; | |
68 | return 1; | |
69 | } | |
70 | ||
71 | int stristart(const char *str, const char *val, const char **ptr) | |
72 | { | |
73 | const char *p, *q; | |
74 | p = str; | |
75 | q = val; | |
76 | while (*q != '\0') { | |
cd390083 | 77 | if (qemu_toupper(*p) != qemu_toupper(*q)) |
18607dcb FB |
78 | return 0; |
79 | p++; | |
80 | q++; | |
81 | } | |
82 | if (ptr) | |
83 | *ptr = p; | |
84 | return 1; | |
85 | } | |
3c6b2088 | 86 | |
d43277c5 BS |
87 | /* XXX: use host strnlen if available ? */ |
88 | int qemu_strnlen(const char *s, int max_len) | |
89 | { | |
90 | int i; | |
91 | ||
92 | for(i = 0; i < max_len; i++) { | |
93 | if (s[i] == '\0') { | |
94 | break; | |
95 | } | |
96 | } | |
97 | return i; | |
98 | } | |
99 | ||
3c6b2088 FB |
100 | time_t mktimegm(struct tm *tm) |
101 | { | |
102 | time_t t; | |
103 | int y = tm->tm_year + 1900, m = tm->tm_mon + 1, d = tm->tm_mday; | |
104 | if (m < 3) { | |
105 | m += 12; | |
106 | y--; | |
107 | } | |
108 | t = 86400 * (d + (153 * m - 457) / 5 + 365 * y + y / 4 - y / 100 + | |
109 | y / 400 - 719469); | |
110 | t += 3600 * tm->tm_hour + 60 * tm->tm_min + tm->tm_sec; | |
111 | return t; | |
112 | } | |
b39ade83 | 113 | |
ad46db9a | 114 | int qemu_fls(int i) |
b39ade83 | 115 | { |
8d371d4b | 116 | return 32 - clz32(i); |
b39ade83 | 117 | } |
44e3ee8a | 118 | |
6f1953c4 CH |
119 | /* |
120 | * Make sure data goes on disk, but if possible do not bother to | |
121 | * write out the inode just for timestamp updates. | |
122 | * | |
123 | * Unfortunately even in 2009 many operating systems do not support | |
124 | * fdatasync and have to fall back to fsync. | |
125 | */ | |
126 | int qemu_fdatasync(int fd) | |
127 | { | |
5f6b9e8f | 128 | #ifdef CONFIG_FDATASYNC |
6f1953c4 CH |
129 | return fdatasync(fd); |
130 | #else | |
131 | return fsync(fd); | |
132 | #endif | |
133 | } | |
134 | ||
44e3ee8a AL |
135 | /* io vectors */ |
136 | ||
137 | void qemu_iovec_init(QEMUIOVector *qiov, int alloc_hint) | |
138 | { | |
7267c094 | 139 | qiov->iov = g_malloc(alloc_hint * sizeof(struct iovec)); |
44e3ee8a AL |
140 | qiov->niov = 0; |
141 | qiov->nalloc = alloc_hint; | |
249aa745 | 142 | qiov->size = 0; |
44e3ee8a AL |
143 | } |
144 | ||
522584a5 AL |
145 | void qemu_iovec_init_external(QEMUIOVector *qiov, struct iovec *iov, int niov) |
146 | { | |
147 | int i; | |
148 | ||
149 | qiov->iov = iov; | |
150 | qiov->niov = niov; | |
151 | qiov->nalloc = -1; | |
152 | qiov->size = 0; | |
153 | for (i = 0; i < niov; i++) | |
154 | qiov->size += iov[i].iov_len; | |
155 | } | |
156 | ||
44e3ee8a AL |
157 | void qemu_iovec_add(QEMUIOVector *qiov, void *base, size_t len) |
158 | { | |
522584a5 AL |
159 | assert(qiov->nalloc != -1); |
160 | ||
44e3ee8a AL |
161 | if (qiov->niov == qiov->nalloc) { |
162 | qiov->nalloc = 2 * qiov->nalloc + 1; | |
7267c094 | 163 | qiov->iov = g_realloc(qiov->iov, qiov->nalloc * sizeof(struct iovec)); |
44e3ee8a AL |
164 | } |
165 | qiov->iov[qiov->niov].iov_base = base; | |
166 | qiov->iov[qiov->niov].iov_len = len; | |
249aa745 | 167 | qiov->size += len; |
44e3ee8a AL |
168 | ++qiov->niov; |
169 | } | |
170 | ||
40b4f539 | 171 | /* |
b8a83a4f KW |
172 | * Copies iovecs from src to the end of dst. It starts copying after skipping |
173 | * the given number of bytes in src and copies until src is completely copied | |
174 | * or the total size of the copied iovec reaches size.The size of the last | |
175 | * copied iovec is changed in order to fit the specified total size if it isn't | |
176 | * a perfect fit already. | |
40b4f539 | 177 | */ |
b8a83a4f KW |
178 | void qemu_iovec_copy(QEMUIOVector *dst, QEMUIOVector *src, uint64_t skip, |
179 | size_t size) | |
40b4f539 KW |
180 | { |
181 | int i; | |
182 | size_t done; | |
b8a83a4f KW |
183 | void *iov_base; |
184 | uint64_t iov_len; | |
40b4f539 KW |
185 | |
186 | assert(dst->nalloc != -1); | |
187 | ||
188 | done = 0; | |
189 | for (i = 0; (i < src->niov) && (done != size); i++) { | |
b8a83a4f KW |
190 | if (skip >= src->iov[i].iov_len) { |
191 | /* Skip the whole iov */ | |
192 | skip -= src->iov[i].iov_len; | |
193 | continue; | |
194 | } else { | |
195 | /* Skip only part (or nothing) of the iov */ | |
196 | iov_base = (uint8_t*) src->iov[i].iov_base + skip; | |
197 | iov_len = src->iov[i].iov_len - skip; | |
198 | skip = 0; | |
199 | } | |
200 | ||
201 | if (done + iov_len > size) { | |
202 | qemu_iovec_add(dst, iov_base, size - done); | |
40b4f539 KW |
203 | break; |
204 | } else { | |
b8a83a4f | 205 | qemu_iovec_add(dst, iov_base, iov_len); |
40b4f539 | 206 | } |
b8a83a4f | 207 | done += iov_len; |
40b4f539 KW |
208 | } |
209 | } | |
210 | ||
b8a83a4f KW |
211 | void qemu_iovec_concat(QEMUIOVector *dst, QEMUIOVector *src, size_t size) |
212 | { | |
213 | qemu_iovec_copy(dst, src, 0, size); | |
214 | } | |
215 | ||
44e3ee8a AL |
216 | void qemu_iovec_destroy(QEMUIOVector *qiov) |
217 | { | |
522584a5 AL |
218 | assert(qiov->nalloc != -1); |
219 | ||
7267c094 | 220 | g_free(qiov->iov); |
44e3ee8a AL |
221 | } |
222 | ||
be959463 AL |
223 | void qemu_iovec_reset(QEMUIOVector *qiov) |
224 | { | |
522584a5 AL |
225 | assert(qiov->nalloc != -1); |
226 | ||
be959463 AL |
227 | qiov->niov = 0; |
228 | qiov->size = 0; | |
229 | } | |
230 | ||
44e3ee8a AL |
231 | void qemu_iovec_to_buffer(QEMUIOVector *qiov, void *buf) |
232 | { | |
233 | uint8_t *p = (uint8_t *)buf; | |
234 | int i; | |
235 | ||
236 | for (i = 0; i < qiov->niov; ++i) { | |
237 | memcpy(p, qiov->iov[i].iov_base, qiov->iov[i].iov_len); | |
238 | p += qiov->iov[i].iov_len; | |
239 | } | |
240 | } | |
241 | ||
249aa745 | 242 | void qemu_iovec_from_buffer(QEMUIOVector *qiov, const void *buf, size_t count) |
44e3ee8a AL |
243 | { |
244 | const uint8_t *p = (const uint8_t *)buf; | |
249aa745 | 245 | size_t copy; |
44e3ee8a AL |
246 | int i; |
247 | ||
249aa745 AL |
248 | for (i = 0; i < qiov->niov && count; ++i) { |
249 | copy = count; | |
250 | if (copy > qiov->iov[i].iov_len) | |
251 | copy = qiov->iov[i].iov_len; | |
252 | memcpy(qiov->iov[i].iov_base, p, copy); | |
253 | p += copy; | |
254 | count -= copy; | |
44e3ee8a AL |
255 | } |
256 | } | |
db1a4972 | 257 | |
b8a83a4f KW |
258 | void qemu_iovec_memset(QEMUIOVector *qiov, int c, size_t count) |
259 | { | |
260 | size_t n; | |
261 | int i; | |
262 | ||
263 | for (i = 0; i < qiov->niov && count; ++i) { | |
264 | n = MIN(count, qiov->iov[i].iov_len); | |
265 | memset(qiov->iov[i].iov_base, c, n); | |
266 | count -= n; | |
267 | } | |
268 | } | |
269 | ||
e0d9c6f9 CT |
270 | void qemu_iovec_memset_skip(QEMUIOVector *qiov, int c, size_t count, |
271 | size_t skip) | |
272 | { | |
273 | int i; | |
274 | size_t done; | |
275 | void *iov_base; | |
276 | uint64_t iov_len; | |
277 | ||
278 | done = 0; | |
279 | for (i = 0; (i < qiov->niov) && (done != count); i++) { | |
280 | if (skip >= qiov->iov[i].iov_len) { | |
281 | /* Skip the whole iov */ | |
282 | skip -= qiov->iov[i].iov_len; | |
283 | continue; | |
284 | } else { | |
285 | /* Skip only part (or nothing) of the iov */ | |
286 | iov_base = (uint8_t*) qiov->iov[i].iov_base + skip; | |
287 | iov_len = qiov->iov[i].iov_len - skip; | |
288 | skip = 0; | |
289 | } | |
290 | ||
291 | if (done + iov_len > count) { | |
292 | memset(iov_base, c, count - done); | |
293 | break; | |
294 | } else { | |
295 | memset(iov_base, c, iov_len); | |
296 | } | |
297 | done += iov_len; | |
298 | } | |
299 | } | |
300 | ||
db1a4972 PB |
301 | #ifndef _WIN32 |
302 | /* Sets a specific flag */ | |
303 | int fcntl_setfl(int fd, int flag) | |
304 | { | |
305 | int flags; | |
306 | ||
307 | flags = fcntl(fd, F_GETFL); | |
308 | if (flags == -1) | |
309 | return -errno; | |
310 | ||
311 | if (fcntl(fd, F_SETFL, flags | flag) == -1) | |
312 | return -errno; | |
313 | ||
314 | return 0; | |
315 | } | |
316 | #endif | |
317 | ||
9f9b17a4 JS |
318 | /* |
319 | * Convert string to bytes, allowing either B/b for bytes, K/k for KB, | |
320 | * M/m for MB, G/g for GB or T/t for TB. Default without any postfix | |
321 | * is MB. End pointer will be returned in *end, if not NULL. A valid | |
322 | * value must be terminated by whitespace, ',' or '\0'. Return -1 on | |
323 | * error. | |
324 | */ | |
a732e1ba JR |
325 | int64_t strtosz_suffix_unit(const char *nptr, char **end, |
326 | const char default_suffix, int64_t unit) | |
9f9b17a4 | 327 | { |
70b4f4bb | 328 | int64_t retval = -1; |
f3bd362a JS |
329 | char *endptr; |
330 | unsigned char c, d; | |
9f9b17a4 JS |
331 | int mul_required = 0; |
332 | double val, mul, integral, fraction; | |
333 | ||
334 | errno = 0; | |
335 | val = strtod(nptr, &endptr); | |
336 | if (isnan(val) || endptr == nptr || errno != 0) { | |
337 | goto fail; | |
338 | } | |
7eb05349 JS |
339 | fraction = modf(val, &integral); |
340 | if (fraction != 0) { | |
9f9b17a4 JS |
341 | mul_required = 1; |
342 | } | |
343 | /* | |
344 | * Any whitespace character is fine for terminating the number, | |
345 | * in addition we accept ',' to handle strings where the size is | |
346 | * part of a multi token argument. | |
347 | */ | |
348 | c = *endptr; | |
d8427002 | 349 | d = c; |
f3bd362a | 350 | if (qemu_isspace(c) || c == '\0' || c == ',') { |
9f9b17a4 | 351 | c = 0; |
d8427002 JS |
352 | if (default_suffix) { |
353 | d = default_suffix; | |
354 | } else { | |
355 | d = c; | |
356 | } | |
9f9b17a4 | 357 | } |
a2afc2c1 | 358 | switch (qemu_toupper(d)) { |
2be22ca5 | 359 | case STRTOSZ_DEFSUFFIX_B: |
9f9b17a4 JS |
360 | mul = 1; |
361 | if (mul_required) { | |
362 | goto fail; | |
363 | } | |
364 | break; | |
2be22ca5 | 365 | case STRTOSZ_DEFSUFFIX_KB: |
a732e1ba | 366 | mul = unit; |
9f9b17a4 JS |
367 | break; |
368 | case 0: | |
369 | if (mul_required) { | |
370 | goto fail; | |
371 | } | |
2be22ca5 | 372 | case STRTOSZ_DEFSUFFIX_MB: |
a732e1ba | 373 | mul = unit * unit; |
9f9b17a4 | 374 | break; |
2be22ca5 | 375 | case STRTOSZ_DEFSUFFIX_GB: |
a732e1ba | 376 | mul = unit * unit * unit; |
9f9b17a4 | 377 | break; |
2be22ca5 | 378 | case STRTOSZ_DEFSUFFIX_TB: |
a732e1ba | 379 | mul = unit * unit * unit * unit; |
9f9b17a4 JS |
380 | break; |
381 | default: | |
382 | goto fail; | |
383 | } | |
384 | /* | |
385 | * If not terminated by whitespace, ',', or \0, increment endptr | |
386 | * to point to next character, then check that we are terminated | |
387 | * by an appropriate separating character, ie. whitespace, ',', or | |
388 | * \0. If not, we are seeing trailing garbage, thus fail. | |
389 | */ | |
390 | if (c != 0) { | |
391 | endptr++; | |
f3bd362a | 392 | if (!qemu_isspace(*endptr) && *endptr != ',' && *endptr != 0) { |
9f9b17a4 JS |
393 | goto fail; |
394 | } | |
395 | } | |
70b4f4bb | 396 | if ((val * mul >= INT64_MAX) || val < 0) { |
9f9b17a4 JS |
397 | goto fail; |
398 | } | |
399 | retval = val * mul; | |
400 | ||
401 | fail: | |
402 | if (end) { | |
403 | *end = endptr; | |
404 | } | |
405 | ||
406 | return retval; | |
407 | } | |
d8427002 | 408 | |
a732e1ba JR |
409 | int64_t strtosz_suffix(const char *nptr, char **end, const char default_suffix) |
410 | { | |
fdc9c41a | 411 | return strtosz_suffix_unit(nptr, end, default_suffix, 1024); |
a732e1ba JR |
412 | } |
413 | ||
70b4f4bb | 414 | int64_t strtosz(const char *nptr, char **end) |
d8427002 JS |
415 | { |
416 | return strtosz_suffix(nptr, end, STRTOSZ_DEFSUFFIX_MB); | |
417 | } |