util: Fix bad constant in ovs_scan() implementation on 64-bit.
[sliver-openvswitch.git] / lib / util.c
1 /*
2  * Copyright (c) 2008, 2009, 2010, 2011, 2012, 2013 Nicira, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at:
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #include <config.h>
18 #include "util.h"
19 #include <ctype.h>
20 #include <errno.h>
21 #include <limits.h>
22 #include <pthread.h>
23 #include <stdarg.h>
24 #include <stdint.h>
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <sys/stat.h>
29 #include <unistd.h>
30 #include "bitmap.h"
31 #include "byte-order.h"
32 #include "coverage.h"
33 #include "ovs-thread.h"
34 #include "vlog.h"
35 #ifdef HAVE_PTHREAD_SET_NAME_NP
36 #include <pthread_np.h>
37 #endif
38
39 VLOG_DEFINE_THIS_MODULE(util);
40
41 COVERAGE_DEFINE(util_xalloc);
42
43 /* argv[0] without directory names. */
44 const char *program_name;
45
46 /* Name for the currently running thread or process, for log messages, process
47  * listings, and debuggers. */
48 DEFINE_PER_THREAD_MALLOCED_DATA(char *, subprogram_name);
49
50 /* --version option output. */
51 static char *program_version;
52
53 /* Buffer used by ovs_strerror(). */
54 DEFINE_STATIC_PER_THREAD_DATA(struct { char s[128]; },
55                               strerror_buffer,
56                               { "" });
57
58 void
59 ovs_assert_failure(const char *where, const char *function,
60                    const char *condition)
61 {
62     /* Prevent an infinite loop (or stack overflow) in case VLOG_ABORT happens
63      * to trigger an assertion failure of its own. */
64     static int reentry = 0;
65
66     switch (reentry++) {
67     case 0:
68         VLOG_ABORT("%s: assertion %s failed in %s()",
69                    where, condition, function);
70         NOT_REACHED();
71
72     case 1:
73         fprintf(stderr, "%s: assertion %s failed in %s()",
74                 where, condition, function);
75         abort();
76
77     default:
78         abort();
79     }
80 }
81
82 void
83 out_of_memory(void)
84 {
85     ovs_abort(0, "virtual memory exhausted");
86 }
87
88 void *
89 xcalloc(size_t count, size_t size)
90 {
91     void *p = count && size ? calloc(count, size) : malloc(1);
92     COVERAGE_INC(util_xalloc);
93     if (p == NULL) {
94         out_of_memory();
95     }
96     return p;
97 }
98
99 void *
100 xzalloc(size_t size)
101 {
102     return xcalloc(1, size);
103 }
104
105 void *
106 xmalloc(size_t size)
107 {
108     void *p = malloc(size ? size : 1);
109     COVERAGE_INC(util_xalloc);
110     if (p == NULL) {
111         out_of_memory();
112     }
113     return p;
114 }
115
116 void *
117 xrealloc(void *p, size_t size)
118 {
119     p = realloc(p, size ? size : 1);
120     COVERAGE_INC(util_xalloc);
121     if (p == NULL) {
122         out_of_memory();
123     }
124     return p;
125 }
126
127 void *
128 xmemdup(const void *p_, size_t size)
129 {
130     void *p = xmalloc(size);
131     memcpy(p, p_, size);
132     return p;
133 }
134
135 char *
136 xmemdup0(const char *p_, size_t length)
137 {
138     char *p = xmalloc(length + 1);
139     memcpy(p, p_, length);
140     p[length] = '\0';
141     return p;
142 }
143
144 char *
145 xstrdup(const char *s)
146 {
147     return xmemdup0(s, strlen(s));
148 }
149
150 char *
151 xvasprintf(const char *format, va_list args)
152 {
153     va_list args2;
154     size_t needed;
155     char *s;
156
157     va_copy(args2, args);
158     needed = vsnprintf(NULL, 0, format, args);
159
160     s = xmalloc(needed + 1);
161
162     vsnprintf(s, needed + 1, format, args2);
163     va_end(args2);
164
165     return s;
166 }
167
168 void *
169 x2nrealloc(void *p, size_t *n, size_t s)
170 {
171     *n = *n == 0 ? 1 : 2 * *n;
172     return xrealloc(p, *n * s);
173 }
174
175 char *
176 xasprintf(const char *format, ...)
177 {
178     va_list args;
179     char *s;
180
181     va_start(args, format);
182     s = xvasprintf(format, args);
183     va_end(args);
184
185     return s;
186 }
187
188 /* Similar to strlcpy() from OpenBSD, but it never reads more than 'size - 1'
189  * bytes from 'src' and doesn't return anything. */
190 void
191 ovs_strlcpy(char *dst, const char *src, size_t size)
192 {
193     if (size > 0) {
194         size_t len = strnlen(src, size - 1);
195         memcpy(dst, src, len);
196         dst[len] = '\0';
197     }
198 }
199
200 /* Copies 'src' to 'dst'.  Reads no more than 'size - 1' bytes from 'src'.
201  * Always null-terminates 'dst' (if 'size' is nonzero), and writes a zero byte
202  * to every otherwise unused byte in 'dst'.
203  *
204  * Except for performance, the following call:
205  *     ovs_strzcpy(dst, src, size);
206  * is equivalent to these two calls:
207  *     memset(dst, '\0', size);
208  *     ovs_strlcpy(dst, src, size);
209  *
210  * (Thus, ovs_strzcpy() is similar to strncpy() without some of the pitfalls.)
211  */
212 void
213 ovs_strzcpy(char *dst, const char *src, size_t size)
214 {
215     if (size > 0) {
216         size_t len = strnlen(src, size - 1);
217         memcpy(dst, src, len);
218         memset(dst + len, '\0', size - len);
219     }
220 }
221
222 /* Prints 'format' on stderr, formatting it like printf() does.  If 'err_no' is
223  * nonzero, then it is formatted with ovs_retval_to_string() and appended to
224  * the message inside parentheses.  Then, terminates with abort().
225  *
226  * This function is preferred to ovs_fatal() in a situation where it would make
227  * sense for a monitoring process to restart the daemon.
228  *
229  * 'format' should not end with a new-line, because this function will add one
230  * itself. */
231 void
232 ovs_abort(int err_no, const char *format, ...)
233 {
234     va_list args;
235
236     va_start(args, format);
237     ovs_abort_valist(err_no, format, args);
238 }
239
240 /* Same as ovs_abort() except that the arguments are supplied as a va_list. */
241 void
242 ovs_abort_valist(int err_no, const char *format, va_list args)
243 {
244     ovs_error_valist(err_no, format, args);
245     abort();
246 }
247
248 /* Prints 'format' on stderr, formatting it like printf() does.  If 'err_no' is
249  * nonzero, then it is formatted with ovs_retval_to_string() and appended to
250  * the message inside parentheses.  Then, terminates with EXIT_FAILURE.
251  *
252  * 'format' should not end with a new-line, because this function will add one
253  * itself. */
254 void
255 ovs_fatal(int err_no, const char *format, ...)
256 {
257     va_list args;
258
259     va_start(args, format);
260     ovs_fatal_valist(err_no, format, args);
261 }
262
263 /* Same as ovs_fatal() except that the arguments are supplied as a va_list. */
264 void
265 ovs_fatal_valist(int err_no, const char *format, va_list args)
266 {
267     ovs_error_valist(err_no, format, args);
268     exit(EXIT_FAILURE);
269 }
270
271 /* Prints 'format' on stderr, formatting it like printf() does.  If 'err_no' is
272  * nonzero, then it is formatted with ovs_retval_to_string() and appended to
273  * the message inside parentheses.
274  *
275  * 'format' should not end with a new-line, because this function will add one
276  * itself. */
277 void
278 ovs_error(int err_no, const char *format, ...)
279 {
280     va_list args;
281
282     va_start(args, format);
283     ovs_error_valist(err_no, format, args);
284     va_end(args);
285 }
286
287 /* Same as ovs_error() except that the arguments are supplied as a va_list. */
288 void
289 ovs_error_valist(int err_no, const char *format, va_list args)
290 {
291     const char *subprogram_name = get_subprogram_name();
292     int save_errno = errno;
293
294     if (subprogram_name[0]) {
295         fprintf(stderr, "%s(%s): ", program_name, subprogram_name);
296     } else {
297         fprintf(stderr, "%s: ", program_name);
298     }
299
300     vfprintf(stderr, format, args);
301     if (err_no != 0) {
302         fprintf(stderr, " (%s)", ovs_retval_to_string(err_no));
303     }
304     putc('\n', stderr);
305
306     errno = save_errno;
307 }
308
309 /* Many OVS functions return an int which is one of:
310  * - 0: no error yet
311  * - >0: errno value
312  * - EOF: end of file (not necessarily an error; depends on the function called)
313  *
314  * Returns the appropriate human-readable string. The caller must copy the
315  * string if it wants to hold onto it, as the storage may be overwritten on
316  * subsequent function calls.
317  */
318 const char *
319 ovs_retval_to_string(int retval)
320 {
321     return (!retval ? ""
322             : retval == EOF ? "End of file"
323             : ovs_strerror(retval));
324 }
325
326 const char *
327 ovs_strerror(int error)
328 {
329     enum { BUFSIZE = sizeof strerror_buffer_get()->s };
330     int save_errno;
331     char *buffer;
332     char *s;
333
334     save_errno = errno;
335     buffer = strerror_buffer_get()->s;
336
337 #if STRERROR_R_CHAR_P
338     /* GNU style strerror_r() might return an immutable static string, or it
339      * might write and return 'buffer', but in either case we can pass the
340      * returned string directly to the caller. */
341     s = strerror_r(error, buffer, BUFSIZE);
342 #else  /* strerror_r() returns an int. */
343     s = buffer;
344     if (strerror_r(error, buffer, BUFSIZE)) {
345         /* strerror_r() is only allowed to fail on ERANGE (because the buffer
346          * is too short).  We don't check the actual failure reason because
347          * POSIX requires strerror_r() to return the error but old glibc
348          * (before 2.13) returns -1 and sets errno. */
349         snprintf(buffer, BUFSIZE, "Unknown error %d", error);
350     }
351 #endif
352
353     errno = save_errno;
354
355     return s;
356 }
357
358 /* Sets global "program_name" and "program_version" variables.  Should
359  * be called at the beginning of main() with "argv[0]" as the argument
360  * to 'argv0'.
361  *
362  * 'version' should contain the version of the caller's program.  If 'version'
363  * is the same as the VERSION #define, the caller is assumed to be part of Open
364  * vSwitch.  Otherwise, it is assumed to be an external program linking against
365  * the Open vSwitch libraries.
366  *
367  * The 'date' and 'time' arguments should likely be called with
368  * "__DATE__" and "__TIME__" to use the time the binary was built.
369  * Alternatively, the "set_program_name" macro may be called to do this
370  * automatically.
371  */
372 void
373 set_program_name__(const char *argv0, const char *version, const char *date,
374                    const char *time)
375 {
376     const char *slash = strrchr(argv0, '/');
377
378     assert_single_threaded();
379
380     program_name = slash ? slash + 1 : argv0;
381
382     free(program_version);
383
384     if (!strcmp(version, VERSION)) {
385         program_version = xasprintf("%s (Open vSwitch) "VERSION"\n"
386                                     "Compiled %s %s\n",
387                                     program_name, date, time);
388     } else {
389         program_version = xasprintf("%s %s\n"
390                                     "Open vSwitch Library "VERSION"\n"
391                                     "Compiled %s %s\n",
392                                     program_name, version, date, time);
393     }
394 }
395
396 /* Returns the name of the currently running thread or process. */
397 const char *
398 get_subprogram_name(void)
399 {
400     const char *name = subprogram_name_get();
401     return name ? name : "";
402 }
403
404 /* Sets the formatted value of 'format' as the name of the currently running
405  * thread or process.  (This appears in log messages and may also be visible in
406  * system process listings and debuggers.) */
407 void
408 set_subprogram_name(const char *format, ...)
409 {
410     char *pname;
411
412     if (format) {
413         va_list args;
414
415         va_start(args, format);
416         pname = xvasprintf(format, args);
417         va_end(args);
418     } else {
419         pname = xstrdup(program_name);
420     }
421
422     free(subprogram_name_set(pname));
423
424 #if HAVE_GLIBC_PTHREAD_SETNAME_NP
425     pthread_setname_np(pthread_self(), pname);
426 #elif HAVE_NETBSD_PTHREAD_SETNAME_NP
427     pthread_setname_np(pthread_self(), "%s", pname);
428 #elif HAVE_PTHREAD_SET_NAME_NP
429     pthread_set_name_np(pthread_self(), pname);
430 #endif
431 }
432
433 /* Returns a pointer to a string describing the program version.  The
434  * caller must not modify or free the returned string.
435  */
436 const char *
437 get_program_version(void)
438 {
439     return program_version;
440 }
441
442 /* Print the version information for the program.  */
443 void
444 ovs_print_version(uint8_t min_ofp, uint8_t max_ofp)
445 {
446     printf("%s", program_version);
447     if (min_ofp || max_ofp) {
448         printf("OpenFlow versions %#x:%#x\n", min_ofp, max_ofp);
449     }
450 }
451
452 /* Writes the 'size' bytes in 'buf' to 'stream' as hex bytes arranged 16 per
453  * line.  Numeric offsets are also included, starting at 'ofs' for the first
454  * byte in 'buf'.  If 'ascii' is true then the corresponding ASCII characters
455  * are also rendered alongside. */
456 void
457 ovs_hex_dump(FILE *stream, const void *buf_, size_t size,
458              uintptr_t ofs, bool ascii)
459 {
460   const uint8_t *buf = buf_;
461   const size_t per_line = 16; /* Maximum bytes per line. */
462
463   while (size > 0)
464     {
465       size_t start, end, n;
466       size_t i;
467
468       /* Number of bytes on this line. */
469       start = ofs % per_line;
470       end = per_line;
471       if (end - start > size)
472         end = start + size;
473       n = end - start;
474
475       /* Print line. */
476       fprintf(stream, "%08jx  ", (uintmax_t) ROUND_DOWN(ofs, per_line));
477       for (i = 0; i < start; i++)
478         fprintf(stream, "   ");
479       for (; i < end; i++)
480         fprintf(stream, "%02hhx%c",
481                 buf[i - start], i == per_line / 2 - 1? '-' : ' ');
482       if (ascii)
483         {
484           for (; i < per_line; i++)
485             fprintf(stream, "   ");
486           fprintf(stream, "|");
487           for (i = 0; i < start; i++)
488             fprintf(stream, " ");
489           for (; i < end; i++) {
490               int c = buf[i - start];
491               putc(c >= 32 && c < 127 ? c : '.', stream);
492           }
493           for (; i < per_line; i++)
494             fprintf(stream, " ");
495           fprintf(stream, "|");
496         }
497       fprintf(stream, "\n");
498
499       ofs += n;
500       buf += n;
501       size -= n;
502     }
503 }
504
505 bool
506 str_to_int(const char *s, int base, int *i)
507 {
508     long long ll;
509     bool ok = str_to_llong(s, base, &ll);
510     *i = ll;
511     return ok;
512 }
513
514 bool
515 str_to_long(const char *s, int base, long *li)
516 {
517     long long ll;
518     bool ok = str_to_llong(s, base, &ll);
519     *li = ll;
520     return ok;
521 }
522
523 bool
524 str_to_llong(const char *s, int base, long long *x)
525 {
526     int save_errno = errno;
527     char *tail;
528     errno = 0;
529     *x = strtoll(s, &tail, base);
530     if (errno == EINVAL || errno == ERANGE || tail == s || *tail != '\0') {
531         errno = save_errno;
532         *x = 0;
533         return false;
534     } else {
535         errno = save_errno;
536         return true;
537     }
538 }
539
540 bool
541 str_to_uint(const char *s, int base, unsigned int *u)
542 {
543     return str_to_int(s, base, (int *) u);
544 }
545
546 bool
547 str_to_ulong(const char *s, int base, unsigned long *ul)
548 {
549     return str_to_long(s, base, (long *) ul);
550 }
551
552 bool
553 str_to_ullong(const char *s, int base, unsigned long long *ull)
554 {
555     return str_to_llong(s, base, (long long *) ull);
556 }
557
558 /* Converts floating-point string 's' into a double.  If successful, stores
559  * the double in '*d' and returns true; on failure, stores 0 in '*d' and
560  * returns false.
561  *
562  * Underflow (e.g. "1e-9999") is not considered an error, but overflow
563  * (e.g. "1e9999)" is. */
564 bool
565 str_to_double(const char *s, double *d)
566 {
567     int save_errno = errno;
568     char *tail;
569     errno = 0;
570     *d = strtod(s, &tail);
571     if (errno == EINVAL || (errno == ERANGE && *d != 0)
572         || tail == s || *tail != '\0') {
573         errno = save_errno;
574         *d = 0;
575         return false;
576     } else {
577         errno = save_errno;
578         return true;
579     }
580 }
581
582 /* Returns the value of 'c' as a hexadecimal digit. */
583 int
584 hexit_value(int c)
585 {
586     switch (c) {
587     case '0': case '1': case '2': case '3': case '4':
588     case '5': case '6': case '7': case '8': case '9':
589         return c - '0';
590
591     case 'a': case 'A':
592         return 0xa;
593
594     case 'b': case 'B':
595         return 0xb;
596
597     case 'c': case 'C':
598         return 0xc;
599
600     case 'd': case 'D':
601         return 0xd;
602
603     case 'e': case 'E':
604         return 0xe;
605
606     case 'f': case 'F':
607         return 0xf;
608
609     default:
610         return -1;
611     }
612 }
613
614 /* Returns the integer value of the 'n' hexadecimal digits starting at 's', or
615  * UINT_MAX if one of those "digits" is not really a hex digit.  If 'ok' is
616  * nonnull, '*ok' is set to true if the conversion succeeds or to false if a
617  * non-hex digit is detected. */
618 unsigned int
619 hexits_value(const char *s, size_t n, bool *ok)
620 {
621     unsigned int value;
622     size_t i;
623
624     value = 0;
625     for (i = 0; i < n; i++) {
626         int hexit = hexit_value(s[i]);
627         if (hexit < 0) {
628             if (ok) {
629                 *ok = false;
630             }
631             return UINT_MAX;
632         }
633         value = (value << 4) + hexit;
634     }
635     if (ok) {
636         *ok = true;
637     }
638     return value;
639 }
640
641 /* Returns the current working directory as a malloc()'d string, or a null
642  * pointer if the current working directory cannot be determined. */
643 char *
644 get_cwd(void)
645 {
646     long int path_max;
647     size_t size;
648
649     /* Get maximum path length or at least a reasonable estimate. */
650     path_max = pathconf(".", _PC_PATH_MAX);
651     size = (path_max < 0 ? 1024
652             : path_max > 10240 ? 10240
653             : path_max);
654
655     /* Get current working directory. */
656     for (;;) {
657         char *buf = xmalloc(size);
658         if (getcwd(buf, size)) {
659             return xrealloc(buf, strlen(buf) + 1);
660         } else {
661             int error = errno;
662             free(buf);
663             if (error != ERANGE) {
664                 VLOG_WARN("getcwd failed (%s)", ovs_strerror(error));
665                 return NULL;
666             }
667             size *= 2;
668         }
669     }
670 }
671
672 static char *
673 all_slashes_name(const char *s)
674 {
675     return xstrdup(s[0] == '/' && s[1] == '/' && s[2] != '/' ? "//"
676                    : s[0] == '/' ? "/"
677                    : ".");
678 }
679
680 /* Returns the directory name portion of 'file_name' as a malloc()'d string,
681  * similar to the POSIX dirname() function but thread-safe. */
682 char *
683 dir_name(const char *file_name)
684 {
685     size_t len = strlen(file_name);
686     while (len > 0 && file_name[len - 1] == '/') {
687         len--;
688     }
689     while (len > 0 && file_name[len - 1] != '/') {
690         len--;
691     }
692     while (len > 0 && file_name[len - 1] == '/') {
693         len--;
694     }
695     return len ? xmemdup0(file_name, len) : all_slashes_name(file_name);
696 }
697
698 /* Returns the file name portion of 'file_name' as a malloc()'d string,
699  * similar to the POSIX basename() function but thread-safe. */
700 char *
701 base_name(const char *file_name)
702 {
703     size_t end, start;
704
705     end = strlen(file_name);
706     while (end > 0 && file_name[end - 1] == '/') {
707         end--;
708     }
709
710     if (!end) {
711         return all_slashes_name(file_name);
712     }
713
714     start = end;
715     while (start > 0 && file_name[start - 1] != '/') {
716         start--;
717     }
718
719     return xmemdup0(file_name + start, end - start);
720 }
721
722 /* If 'file_name' starts with '/', returns a copy of 'file_name'.  Otherwise,
723  * returns an absolute path to 'file_name' considering it relative to 'dir',
724  * which itself must be absolute.  'dir' may be null or the empty string, in
725  * which case the current working directory is used.
726  *
727  * Returns a null pointer if 'dir' is null and getcwd() fails. */
728 char *
729 abs_file_name(const char *dir, const char *file_name)
730 {
731     if (file_name[0] == '/') {
732         return xstrdup(file_name);
733     } else if (dir && dir[0]) {
734         char *separator = dir[strlen(dir) - 1] == '/' ? "" : "/";
735         return xasprintf("%s%s%s", dir, separator, file_name);
736     } else {
737         char *cwd = get_cwd();
738         if (cwd) {
739             char *abs_name = xasprintf("%s/%s", cwd, file_name);
740             free(cwd);
741             return abs_name;
742         } else {
743             return NULL;
744         }
745     }
746 }
747
748 /* Like readlink(), but returns the link name as a null-terminated string in
749  * allocated memory that the caller must eventually free (with free()).
750  * Returns NULL on error, in which case errno is set appropriately. */
751 char *
752 xreadlink(const char *filename)
753 {
754     size_t size;
755
756     for (size = 64; ; size *= 2) {
757         char *buf = xmalloc(size);
758         ssize_t retval = readlink(filename, buf, size);
759         int error = errno;
760
761         if (retval >= 0 && retval < size) {
762             buf[retval] = '\0';
763             return buf;
764         }
765
766         free(buf);
767         if (retval < 0) {
768             errno = error;
769             return NULL;
770         }
771     }
772 }
773
774 /* Returns a version of 'filename' with symlinks in the final component
775  * dereferenced.  This differs from realpath() in that:
776  *
777  *     - 'filename' need not exist.
778  *
779  *     - If 'filename' does exist as a symlink, its referent need not exist.
780  *
781  *     - Only symlinks in the final component of 'filename' are dereferenced.
782  *
783  * The caller must eventually free the returned string (with free()). */
784 char *
785 follow_symlinks(const char *filename)
786 {
787     struct stat s;
788     char *fn;
789     int i;
790
791     fn = xstrdup(filename);
792     for (i = 0; i < 10; i++) {
793         char *linkname;
794         char *next_fn;
795
796         if (lstat(fn, &s) != 0 || !S_ISLNK(s.st_mode)) {
797             return fn;
798         }
799
800         linkname = xreadlink(fn);
801         if (!linkname) {
802             VLOG_WARN("%s: readlink failed (%s)",
803                       filename, ovs_strerror(errno));
804             return fn;
805         }
806
807         if (linkname[0] == '/') {
808             /* Target of symlink is absolute so use it raw. */
809             next_fn = linkname;
810         } else {
811             /* Target of symlink is relative so add to 'fn''s directory. */
812             char *dir = dir_name(fn);
813
814             if (!strcmp(dir, ".")) {
815                 next_fn = linkname;
816             } else {
817                 char *separator = dir[strlen(dir) - 1] == '/' ? "" : "/";
818                 next_fn = xasprintf("%s%s%s", dir, separator, linkname);
819                 free(linkname);
820             }
821
822             free(dir);
823         }
824
825         free(fn);
826         fn = next_fn;
827     }
828
829     VLOG_WARN("%s: too many levels of symlinks", filename);
830     free(fn);
831     return xstrdup(filename);
832 }
833
834 /* Pass a value to this function if it is marked with
835  * __attribute__((warn_unused_result)) and you genuinely want to ignore
836  * its return value.  (Note that every scalar type can be implicitly
837  * converted to bool.) */
838 void ignore(bool x OVS_UNUSED) { }
839
840 /* Returns an appropriate delimiter for inserting just before the 0-based item
841  * 'index' in a list that has 'total' items in it. */
842 const char *
843 english_list_delimiter(size_t index, size_t total)
844 {
845     return (index == 0 ? ""
846             : index < total - 1 ? ", "
847             : total > 2 ? ", and "
848             : " and ");
849 }
850
851 /* Given a 32 bit word 'n', calculates floor(log_2('n')).  This is equivalent
852  * to finding the bit position of the most significant one bit in 'n'.  It is
853  * an error to call this function with 'n' == 0. */
854 int
855 log_2_floor(uint32_t n)
856 {
857     ovs_assert(n);
858
859 #if !defined(UINT_MAX) || !defined(UINT32_MAX)
860 #error "Someone screwed up the #includes."
861 #elif __GNUC__ >= 4 && UINT_MAX == UINT32_MAX
862     return 31 - __builtin_clz(n);
863 #else
864     {
865         int log = 0;
866
867 #define BIN_SEARCH_STEP(BITS)                   \
868         if (n >= (1 << BITS)) {                 \
869             log += BITS;                        \
870             n >>= BITS;                         \
871         }
872         BIN_SEARCH_STEP(16);
873         BIN_SEARCH_STEP(8);
874         BIN_SEARCH_STEP(4);
875         BIN_SEARCH_STEP(2);
876         BIN_SEARCH_STEP(1);
877 #undef BIN_SEARCH_STEP
878         return log;
879     }
880 #endif
881 }
882
883 /* Given a 32 bit word 'n', calculates ceil(log_2('n')).  It is an error to
884  * call this function with 'n' == 0. */
885 int
886 log_2_ceil(uint32_t n)
887 {
888     return log_2_floor(n) + !is_pow2(n);
889 }
890
891 /* Returns the number of trailing 0-bits in 'n'.  Undefined if 'n' == 0. */
892 #if !defined(UINT_MAX) || !defined(UINT32_MAX)
893 #error "Someone screwed up the #includes."
894 #elif __GNUC__ >= 4 && UINT_MAX == UINT32_MAX
895 /* Defined inline in util.h. */
896 #else
897 static int
898 raw_ctz(uint32_t n)
899 {
900     unsigned int k;
901     int count = 31;
902
903 #define CTZ_STEP(X)                             \
904     k = n << (X);                               \
905     if (k) {                                    \
906         count -= X;                             \
907         n = k;                                  \
908     }
909     CTZ_STEP(16);
910     CTZ_STEP(8);
911     CTZ_STEP(4);
912     CTZ_STEP(2);
913     CTZ_STEP(1);
914 #undef CTZ_STEP
915
916     return count;
917 }
918 #endif
919
920 /* Returns the number of 1-bits in 'x', between 0 and 32 inclusive. */
921 unsigned int
922 popcount(uint32_t x)
923 {
924     /* In my testing, this implementation is over twice as fast as any other
925      * portable implementation that I tried, including GCC 4.4
926      * __builtin_popcount(), although nonportable asm("popcnt") was over 50%
927      * faster. */
928 #define INIT1(X)                                \
929     ((((X) & (1 << 0)) != 0) +                  \
930      (((X) & (1 << 1)) != 0) +                  \
931      (((X) & (1 << 2)) != 0) +                  \
932      (((X) & (1 << 3)) != 0) +                  \
933      (((X) & (1 << 4)) != 0) +                  \
934      (((X) & (1 << 5)) != 0) +                  \
935      (((X) & (1 << 6)) != 0) +                  \
936      (((X) & (1 << 7)) != 0))
937 #define INIT2(X)   INIT1(X),  INIT1((X) +  1)
938 #define INIT4(X)   INIT2(X),  INIT2((X) +  2)
939 #define INIT8(X)   INIT4(X),  INIT4((X) +  4)
940 #define INIT16(X)  INIT8(X),  INIT8((X) +  8)
941 #define INIT32(X) INIT16(X), INIT16((X) + 16)
942 #define INIT64(X) INIT32(X), INIT32((X) + 32)
943
944     static const uint8_t popcount8[256] = {
945         INIT64(0), INIT64(64), INIT64(128), INIT64(192)
946     };
947
948     return (popcount8[x & 0xff] +
949             popcount8[(x >> 8) & 0xff] +
950             popcount8[(x >> 16) & 0xff] +
951             popcount8[x >> 24]);
952 }
953
954 /* Returns true if the 'n' bytes starting at 'p' are zeros. */
955 bool
956 is_all_zeros(const uint8_t *p, size_t n)
957 {
958     size_t i;
959
960     for (i = 0; i < n; i++) {
961         if (p[i] != 0x00) {
962             return false;
963         }
964     }
965     return true;
966 }
967
968 /* Returns true if the 'n' bytes starting at 'p' are 0xff. */
969 bool
970 is_all_ones(const uint8_t *p, size_t n)
971 {
972     size_t i;
973
974     for (i = 0; i < n; i++) {
975         if (p[i] != 0xff) {
976             return false;
977         }
978     }
979     return true;
980 }
981
982 /* Copies 'n_bits' bits starting from bit 'src_ofs' in 'src' to the 'n_bits'
983  * starting from bit 'dst_ofs' in 'dst'.  'src' is 'src_len' bytes long and
984  * 'dst' is 'dst_len' bytes long.
985  *
986  * If you consider all of 'src' to be a single unsigned integer in network byte
987  * order, then bit N is the bit with value 2**N.  That is, bit 0 is the bit
988  * with value 1 in src[src_len - 1], bit 1 is the bit with value 2, bit 2 is
989  * the bit with value 4, ..., bit 8 is the bit with value 1 in src[src_len -
990  * 2], and so on.  Similarly for 'dst'.
991  *
992  * Required invariants:
993  *   src_ofs + n_bits <= src_len * 8
994  *   dst_ofs + n_bits <= dst_len * 8
995  *   'src' and 'dst' must not overlap.
996  */
997 void
998 bitwise_copy(const void *src_, unsigned int src_len, unsigned int src_ofs,
999              void *dst_, unsigned int dst_len, unsigned int dst_ofs,
1000              unsigned int n_bits)
1001 {
1002     const uint8_t *src = src_;
1003     uint8_t *dst = dst_;
1004
1005     src += src_len - (src_ofs / 8 + 1);
1006     src_ofs %= 8;
1007
1008     dst += dst_len - (dst_ofs / 8 + 1);
1009     dst_ofs %= 8;
1010
1011     if (src_ofs == 0 && dst_ofs == 0) {
1012         unsigned int n_bytes = n_bits / 8;
1013         if (n_bytes) {
1014             dst -= n_bytes - 1;
1015             src -= n_bytes - 1;
1016             memcpy(dst, src, n_bytes);
1017
1018             n_bits %= 8;
1019             src--;
1020             dst--;
1021         }
1022         if (n_bits) {
1023             uint8_t mask = (1 << n_bits) - 1;
1024             *dst = (*dst & ~mask) | (*src & mask);
1025         }
1026     } else {
1027         while (n_bits > 0) {
1028             unsigned int max_copy = 8 - MAX(src_ofs, dst_ofs);
1029             unsigned int chunk = MIN(n_bits, max_copy);
1030             uint8_t mask = ((1 << chunk) - 1) << dst_ofs;
1031
1032             *dst &= ~mask;
1033             *dst |= ((*src >> src_ofs) << dst_ofs) & mask;
1034
1035             src_ofs += chunk;
1036             if (src_ofs == 8) {
1037                 src--;
1038                 src_ofs = 0;
1039             }
1040             dst_ofs += chunk;
1041             if (dst_ofs == 8) {
1042                 dst--;
1043                 dst_ofs = 0;
1044             }
1045             n_bits -= chunk;
1046         }
1047     }
1048 }
1049
1050 /* Zeros the 'n_bits' bits starting from bit 'dst_ofs' in 'dst'.  'dst' is
1051  * 'dst_len' bytes long.
1052  *
1053  * If you consider all of 'dst' to be a single unsigned integer in network byte
1054  * order, then bit N is the bit with value 2**N.  That is, bit 0 is the bit
1055  * with value 1 in dst[dst_len - 1], bit 1 is the bit with value 2, bit 2 is
1056  * the bit with value 4, ..., bit 8 is the bit with value 1 in dst[dst_len -
1057  * 2], and so on.
1058  *
1059  * Required invariant:
1060  *   dst_ofs + n_bits <= dst_len * 8
1061  */
1062 void
1063 bitwise_zero(void *dst_, unsigned int dst_len, unsigned dst_ofs,
1064              unsigned int n_bits)
1065 {
1066     uint8_t *dst = dst_;
1067
1068     if (!n_bits) {
1069         return;
1070     }
1071
1072     dst += dst_len - (dst_ofs / 8 + 1);
1073     dst_ofs %= 8;
1074
1075     if (dst_ofs) {
1076         unsigned int chunk = MIN(n_bits, 8 - dst_ofs);
1077
1078         *dst &= ~(((1 << chunk) - 1) << dst_ofs);
1079
1080         n_bits -= chunk;
1081         if (!n_bits) {
1082             return;
1083         }
1084
1085         dst--;
1086     }
1087
1088     while (n_bits >= 8) {
1089         *dst-- = 0;
1090         n_bits -= 8;
1091     }
1092
1093     if (n_bits) {
1094         *dst &= ~((1 << n_bits) - 1);
1095     }
1096 }
1097
1098 /* Sets to 1 all of the 'n_bits' bits starting from bit 'dst_ofs' in 'dst'.
1099  * 'dst' is 'dst_len' bytes long.
1100  *
1101  * If you consider all of 'dst' to be a single unsigned integer in network byte
1102  * order, then bit N is the bit with value 2**N.  That is, bit 0 is the bit
1103  * with value 1 in dst[dst_len - 1], bit 1 is the bit with value 2, bit 2 is
1104  * the bit with value 4, ..., bit 8 is the bit with value 1 in dst[dst_len -
1105  * 2], and so on.
1106  *
1107  * Required invariant:
1108  *   dst_ofs + n_bits <= dst_len * 8
1109  */
1110 void
1111 bitwise_one(void *dst_, unsigned int dst_len, unsigned dst_ofs,
1112             unsigned int n_bits)
1113 {
1114     uint8_t *dst = dst_;
1115
1116     if (!n_bits) {
1117         return;
1118     }
1119
1120     dst += dst_len - (dst_ofs / 8 + 1);
1121     dst_ofs %= 8;
1122
1123     if (dst_ofs) {
1124         unsigned int chunk = MIN(n_bits, 8 - dst_ofs);
1125
1126         *dst |= ((1 << chunk) - 1) << dst_ofs;
1127
1128         n_bits -= chunk;
1129         if (!n_bits) {
1130             return;
1131         }
1132
1133         dst--;
1134     }
1135
1136     while (n_bits >= 8) {
1137         *dst-- = 0xff;
1138         n_bits -= 8;
1139     }
1140
1141     if (n_bits) {
1142         *dst |= (1 << n_bits) - 1;
1143     }
1144 }
1145
1146 /* Scans the 'n_bits' bits starting from bit 'dst_ofs' in 'dst' for 1-bits.
1147  * Returns false if any 1-bits are found, otherwise true.  'dst' is 'dst_len'
1148  * bytes long.
1149  *
1150  * If you consider all of 'dst' to be a single unsigned integer in network byte
1151  * order, then bit N is the bit with value 2**N.  That is, bit 0 is the bit
1152  * with value 1 in dst[dst_len - 1], bit 1 is the bit with value 2, bit 2 is
1153  * the bit with value 4, ..., bit 8 is the bit with value 1 in dst[dst_len -
1154  * 2], and so on.
1155  *
1156  * Required invariant:
1157  *   dst_ofs + n_bits <= dst_len * 8
1158  */
1159 bool
1160 bitwise_is_all_zeros(const void *p_, unsigned int len, unsigned int ofs,
1161                      unsigned int n_bits)
1162 {
1163     const uint8_t *p = p_;
1164
1165     if (!n_bits) {
1166         return true;
1167     }
1168
1169     p += len - (ofs / 8 + 1);
1170     ofs %= 8;
1171
1172     if (ofs) {
1173         unsigned int chunk = MIN(n_bits, 8 - ofs);
1174
1175         if (*p & (((1 << chunk) - 1) << ofs)) {
1176             return false;
1177         }
1178
1179         n_bits -= chunk;
1180         if (!n_bits) {
1181             return true;
1182         }
1183
1184         p--;
1185     }
1186
1187     while (n_bits >= 8) {
1188         if (*p) {
1189             return false;
1190         }
1191         n_bits -= 8;
1192         p--;
1193     }
1194
1195     if (n_bits && *p & ((1 << n_bits) - 1)) {
1196         return false;
1197     }
1198
1199     return true;
1200 }
1201
1202 /* Copies the 'n_bits' low-order bits of 'value' into the 'n_bits' bits
1203  * starting at bit 'dst_ofs' in 'dst', which is 'dst_len' bytes long.
1204  *
1205  * If you consider all of 'dst' to be a single unsigned integer in network byte
1206  * order, then bit N is the bit with value 2**N.  That is, bit 0 is the bit
1207  * with value 1 in dst[dst_len - 1], bit 1 is the bit with value 2, bit 2 is
1208  * the bit with value 4, ..., bit 8 is the bit with value 1 in dst[dst_len -
1209  * 2], and so on.
1210  *
1211  * Required invariants:
1212  *   dst_ofs + n_bits <= dst_len * 8
1213  *   n_bits <= 64
1214  */
1215 void
1216 bitwise_put(uint64_t value,
1217             void *dst, unsigned int dst_len, unsigned int dst_ofs,
1218             unsigned int n_bits)
1219 {
1220     ovs_be64 n_value = htonll(value);
1221     bitwise_copy(&n_value, sizeof n_value, 0,
1222                  dst, dst_len, dst_ofs,
1223                  n_bits);
1224 }
1225
1226 /* Returns the value of the 'n_bits' bits starting at bit 'src_ofs' in 'src',
1227  * which is 'src_len' bytes long.
1228  *
1229  * If you consider all of 'src' to be a single unsigned integer in network byte
1230  * order, then bit N is the bit with value 2**N.  That is, bit 0 is the bit
1231  * with value 1 in src[src_len - 1], bit 1 is the bit with value 2, bit 2 is
1232  * the bit with value 4, ..., bit 8 is the bit with value 1 in src[src_len -
1233  * 2], and so on.
1234  *
1235  * Required invariants:
1236  *   src_ofs + n_bits <= src_len * 8
1237  *   n_bits <= 64
1238  */
1239 uint64_t
1240 bitwise_get(const void *src, unsigned int src_len,
1241             unsigned int src_ofs, unsigned int n_bits)
1242 {
1243     ovs_be64 value = htonll(0);
1244
1245     bitwise_copy(src, src_len, src_ofs,
1246                  &value, sizeof value, 0,
1247                  n_bits);
1248     return ntohll(value);
1249 }
1250 \f
1251 /* ovs_scan */
1252
1253 struct scan_spec {
1254     unsigned int width;
1255     enum {
1256         SCAN_DISCARD,
1257         SCAN_CHAR,
1258         SCAN_SHORT,
1259         SCAN_INT,
1260         SCAN_LONG,
1261         SCAN_LLONG,
1262         SCAN_INTMAX_T,
1263         SCAN_PTRDIFF_T,
1264         SCAN_SIZE_T
1265     } type;
1266 };
1267
1268 static const char *
1269 skip_spaces(const char *s)
1270 {
1271     while (isspace((unsigned char) *s)) {
1272         s++;
1273     }
1274     return s;
1275 }
1276
1277 static const char *
1278 scan_int(const char *s, const struct scan_spec *spec, int base, va_list *args)
1279 {
1280     const char *start = s;
1281     uintmax_t value;
1282     bool negative;
1283     int n_digits;
1284
1285     negative = *s == '-';
1286     s += *s == '-' || *s == '+';
1287
1288     if ((!base || base == 16) && *s == '0' && (s[1] == 'x' || s[1] == 'X')) {
1289         base = 16;
1290         s += 2;
1291     } else if (!base) {
1292         base = *s == '0' ? 8 : 10;
1293     }
1294
1295     if (s - start >= spec->width) {
1296         return NULL;
1297     }
1298
1299     value = 0;
1300     n_digits = 0;
1301     while (s - start < spec->width) {
1302         int digit = hexit_value(*s);
1303
1304         if (digit < 0 || digit >= base) {
1305             break;
1306         }
1307         value = value * base + digit;
1308         n_digits++;
1309         s++;
1310     }
1311     if (!n_digits) {
1312         return NULL;
1313     }
1314
1315     if (negative) {
1316         value = -value;
1317     }
1318
1319     switch (spec->type) {
1320     case SCAN_DISCARD:
1321         break;
1322     case SCAN_CHAR:
1323         *va_arg(*args, char *) = value;
1324         break;
1325     case SCAN_SHORT:
1326         *va_arg(*args, short int *) = value;
1327         break;
1328     case SCAN_INT:
1329         *va_arg(*args, int *) = value;
1330         break;
1331     case SCAN_LONG:
1332         *va_arg(*args, long int *) = value;
1333         break;
1334     case SCAN_LLONG:
1335         *va_arg(*args, long long int *) = value;
1336         break;
1337     case SCAN_INTMAX_T:
1338         *va_arg(*args, intmax_t *) = value;
1339         break;
1340     case SCAN_PTRDIFF_T:
1341         *va_arg(*args, ptrdiff_t *) = value;
1342         break;
1343     case SCAN_SIZE_T:
1344         *va_arg(*args, size_t *) = value;
1345         break;
1346     }
1347     return s;
1348 }
1349
1350 static const char *
1351 skip_digits(const char *s)
1352 {
1353     while (*s >= '0' && *s <= '9') {
1354         s++;
1355     }
1356     return s;
1357 }
1358
1359 static const char *
1360 scan_float(const char *s, const struct scan_spec *spec, va_list *args)
1361 {
1362     const char *start = s;
1363     long double value;
1364     char *tail;
1365     char *copy;
1366     bool ok;
1367
1368     s += *s == '+' || *s == '-';
1369     s = skip_digits(s);
1370     if (*s == '.') {
1371         s = skip_digits(s + 1);
1372     }
1373     if (*s == 'e' || *s == 'E') {
1374         s++;
1375         s += *s == '+' || *s == '-';
1376         s = skip_digits(s);
1377     }
1378
1379     if (s - start > spec->width) {
1380         s = start + spec->width;
1381     }
1382
1383     copy = xmemdup0(start, s - start);
1384     value = strtold(copy, &tail);
1385     ok = *tail == '\0';
1386     free(copy);
1387     if (!ok) {
1388         return NULL;
1389     }
1390
1391     switch (spec->type) {
1392     case SCAN_DISCARD:
1393         break;
1394     case SCAN_INT:
1395         *va_arg(*args, float *) = value;
1396         break;
1397     case SCAN_LONG:
1398         *va_arg(*args, double *) = value;
1399         break;
1400     case SCAN_LLONG:
1401         *va_arg(*args, long double *) = value;
1402         break;
1403
1404     case SCAN_CHAR:
1405     case SCAN_SHORT:
1406     case SCAN_INTMAX_T:
1407     case SCAN_PTRDIFF_T:
1408     case SCAN_SIZE_T:
1409         NOT_REACHED();
1410     }
1411     return s;
1412 }
1413
1414 static void
1415 scan_output_string(const struct scan_spec *spec,
1416                    const char *s, size_t n,
1417                    va_list *args)
1418 {
1419     if (spec->type != SCAN_DISCARD) {
1420         char *out = va_arg(*args, char *);
1421         memcpy(out, s, n);
1422         out[n] = '\0';
1423     }
1424 }
1425
1426 static const char *
1427 scan_string(const char *s, const struct scan_spec *spec, va_list *args)
1428 {
1429     size_t n;
1430
1431     for (n = 0; n < spec->width; n++) {
1432         if (!s[n] || isspace((unsigned char) s[n])) {
1433             break;
1434         }
1435     }
1436     if (!n) {
1437         return NULL;
1438     }
1439
1440     scan_output_string(spec, s, n, args);
1441     return s + n;
1442 }
1443
1444 static const char *
1445 parse_scanset(const char *p_, unsigned long *set, bool *complemented)
1446 {
1447     const uint8_t *p = (const uint8_t *) p_;
1448
1449     *complemented = *p == '^';
1450     p += *complemented;
1451
1452     if (*p == ']') {
1453         bitmap_set1(set, ']');
1454         p++;
1455     }
1456
1457     while (*p && *p != ']') {
1458         if (p[1] == '-' && p[2] != ']' && p[2] > *p) {
1459             bitmap_set_multiple(set, *p, p[2] - *p + 1, true);
1460             p += 3;
1461         } else {
1462             bitmap_set1(set, *p++);
1463         }
1464     }
1465     if (*p == ']') {
1466         p++;
1467     }
1468     return (const char *) p;
1469 }
1470
1471 static const char *
1472 scan_set(const char *s, const struct scan_spec *spec, const char **pp,
1473          va_list *args)
1474 {
1475     unsigned long set[BITMAP_N_LONGS(UCHAR_MAX + 1)];
1476     bool complemented;
1477     unsigned int n;
1478
1479     /* Parse the scan set. */
1480     memset(set, 0, sizeof set);
1481     *pp = parse_scanset(*pp, set, &complemented);
1482
1483     /* Parse the data. */
1484     n = 0;
1485     while (s[n]
1486            && bitmap_is_set(set, (unsigned char) s[n]) == !complemented
1487            && n < spec->width) {
1488         n++;
1489     }
1490     if (!n) {
1491         return NULL;
1492     }
1493     scan_output_string(spec, s, n, args);
1494     return s + n;
1495 }
1496
1497 static const char *
1498 scan_chars(const char *s, const struct scan_spec *spec, va_list *args)
1499 {
1500     unsigned int n = spec->width == UINT_MAX ? 1 : spec->width;
1501
1502     if (strlen(s) < n) {
1503         return NULL;
1504     }
1505     if (spec->type != SCAN_DISCARD) {
1506         memcpy(va_arg(*args, char *), s, n);
1507     }
1508     return s + n;
1509 }
1510
1511 /* This is an implementation of the standard sscanf() function, with the
1512  * following exceptions:
1513  *
1514  *   - It returns true if the entire template was successfully scanned and
1515  *     converted, false if any conversion failed.
1516  *
1517  *   - The standard doesn't define sscanf() behavior when an out-of-range value
1518  *     is scanned, e.g. if a "%"PRIi8 conversion scans "-1" or "0x1ff".  Some
1519  *     implementations consider this an error and stop scanning.  This
1520  *     implementation never considers an out-of-range value an error; instead,
1521  *     it stores the least-significant bits of the converted value in the
1522  *     destination, e.g. the value 255 for both examples earlier.
1523  *
1524  *   - Only single-byte characters are supported, that is, the 'l' modifier
1525  *     on %s, %[, and %c is not supported.  The GNU extension 'a' modifier is
1526  *     also not supported.
1527  *
1528  *   - %p is not supported.
1529  */
1530 bool
1531 ovs_scan(const char *s, const char *template, ...)
1532 {
1533     const char *const start = s;
1534     bool ok = false;
1535     const char *p;
1536     va_list args;
1537
1538     va_start(args, template);
1539     p = template;
1540     while (*p != '\0') {
1541         struct scan_spec spec;
1542         unsigned char c = *p++;
1543         bool discard;
1544
1545         if (isspace(c)) {
1546             s = skip_spaces(s);
1547             continue;
1548         } else if (c != '%') {
1549             if (*s != c) {
1550                 goto exit;
1551             }
1552             s++;
1553             continue;
1554         } else if (*p == '%') {
1555             if (*s++ != '%') {
1556                 goto exit;
1557             }
1558             p++;
1559             continue;
1560         }
1561
1562         /* Parse '*' flag. */
1563         discard = *p == '*';
1564         p += discard;
1565
1566         /* Parse field width. */
1567         spec.width = 0;
1568         while (*p >= '0' && *p <= '9') {
1569             spec.width = spec.width * 10 + (*p++ - '0');
1570         }
1571         if (spec.width == 0) {
1572             spec.width = UINT_MAX;
1573         }
1574
1575         /* Parse type modifier. */
1576         switch (*p) {
1577         case 'h':
1578             if (p[1] == 'h') {
1579                 spec.type = SCAN_CHAR;
1580                 p += 2;
1581             } else {
1582                 spec.type = SCAN_SHORT;
1583                 p++;
1584             }
1585             break;
1586
1587         case 'j':
1588             spec.type = SCAN_INTMAX_T;
1589             p++;
1590             break;
1591
1592         case 'l':
1593             if (p[1] == 'l') {
1594                 spec.type = SCAN_LLONG;
1595                 p += 2;
1596             } else {
1597                 spec.type = SCAN_LONG;
1598                 p++;
1599             }
1600             break;
1601
1602         case 'L':
1603         case 'q':
1604             spec.type = SCAN_LLONG;
1605             p++;
1606             break;
1607
1608         case 't':
1609             spec.type = SCAN_PTRDIFF_T;
1610             p++;
1611             break;
1612
1613         case 'z':
1614             spec.type = SCAN_SIZE_T;
1615             p++;
1616             break;
1617
1618         default:
1619             spec.type = SCAN_INT;
1620             break;
1621         }
1622
1623         if (discard) {
1624             spec.type = SCAN_DISCARD;
1625         }
1626
1627         c = *p++;
1628         if (c != 'c' && c != 'n' && c != '[') {
1629             s = skip_spaces(s);
1630         }
1631         switch (c) {
1632         case 'd':
1633             s = scan_int(s, &spec, 10, &args);
1634             break;
1635
1636         case 'i':
1637             s = scan_int(s, &spec, 0, &args);
1638             break;
1639
1640         case 'o':
1641             s = scan_int(s, &spec, 8, &args);
1642             break;
1643
1644         case 'u':
1645             s = scan_int(s, &spec, 10, &args);
1646             break;
1647
1648         case 'x':
1649         case 'X':
1650             s = scan_int(s, &spec, 16, &args);
1651             break;
1652
1653         case 'e':
1654         case 'f':
1655         case 'g':
1656         case 'E':
1657         case 'G':
1658             s = scan_float(s, &spec, &args);
1659             break;
1660
1661         case 's':
1662             s = scan_string(s, &spec, &args);
1663             break;
1664
1665         case '[':
1666             s = scan_set(s, &spec, &p, &args);
1667             break;
1668
1669         case 'c':
1670             s = scan_chars(s, &spec, &args);
1671             break;
1672
1673         case 'n':
1674             if (spec.type != SCAN_DISCARD) {
1675                 *va_arg(args, int *) = s - start;
1676             }
1677             break;
1678         }
1679
1680         if (!s) {
1681             goto exit;
1682         }
1683     }
1684     ok = true;
1685
1686 exit:
1687     va_end(args);
1688     return ok;
1689 }
1690