|
11708
|
1 /* Test argv iterator |
|
|
2 Copyright (C) 2008-2009 Free Software Foundation, Inc. |
|
|
3 |
|
|
4 This program is free software: you can redistribute it and/or modify |
|
|
5 it under the terms of the GNU General Public License as published by |
|
|
6 the Free Software Foundation; either version 3 of the License, or |
|
|
7 (at your option) any later version. |
|
|
8 |
|
|
9 This program is distributed in the hope that it will be useful, |
|
|
10 but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
|
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
|
12 GNU General Public License for more details. |
|
|
13 |
|
|
14 You should have received a copy of the GNU General Public License |
|
|
15 along with this program. If not, see <http://www.gnu.org/licenses/>. */ |
|
|
16 |
|
|
17 /* Written by Jim Meyering. */ |
|
|
18 |
|
|
19 #include <config.h> |
|
|
20 #include <stdlib.h> |
|
|
21 #include <stdio.h> |
|
|
22 #include <string.h> |
|
|
23 |
|
|
24 #define ARRAY_CARDINALITY(Array) (sizeof (Array) / sizeof *(Array)) |
|
|
25 #define STREQ(s1, s2) ((strcmp (s1, s2) == 0)) |
|
|
26 #define ASSERT(expr) \ |
|
|
27 do \ |
|
|
28 { \ |
|
|
29 if (!(expr)) \ |
|
|
30 { \ |
|
|
31 fprintf (stderr, "%s:%d: assertion failed\n", __FILE__, __LINE__); \ |
|
|
32 fflush (stderr); \ |
|
|
33 abort (); \ |
|
|
34 } \ |
|
|
35 } \ |
|
|
36 while (0) |
|
|
37 |
|
|
38 #include "argv-iter.h" |
|
|
39 |
|
|
40 static FILE * |
|
|
41 write_nul_delimited_argv (char **argv) |
|
|
42 { |
|
|
43 FILE *fp = tmpfile (); |
|
|
44 ASSERT (fp); |
|
|
45 while (*argv) |
|
|
46 { |
|
|
47 size_t len = strlen (*argv) + 1; |
|
|
48 ASSERT (fwrite (*argv, len, 1, fp) == 1); |
|
|
49 argv++; |
|
|
50 } |
|
|
51 ASSERT (fflush (fp) == 0); |
|
|
52 rewind (fp); |
|
|
53 return fp; |
|
|
54 } |
|
|
55 |
|
|
56 int |
|
|
57 main () |
|
|
58 { |
|
|
59 /* set_program_name (argv[0]); placate overzealous "syntax-check" test. */ |
|
|
60 static char *av[][4] = { |
|
|
61 {NULL}, |
|
|
62 {"1", NULL}, |
|
|
63 {"1", "2", NULL}, |
|
|
64 {"1", "2", "3", NULL} |
|
|
65 }; |
|
|
66 |
|
|
67 int use_stream; |
|
|
68 for (use_stream = 0; use_stream < 2; use_stream++) |
|
|
69 { |
|
|
70 size_t i; |
|
|
71 for (i = 0; i < ARRAY_CARDINALITY (av); i++) |
|
|
72 { |
|
|
73 FILE *fp; |
|
|
74 struct argv_iterator *ai; |
|
|
75 size_t n_found = 0; |
|
|
76 if (use_stream) |
|
|
77 { |
|
|
78 /* Generate an identical list to be read via FP. */ |
|
|
79 ASSERT ((fp = write_nul_delimited_argv (av[i])) != NULL); |
|
|
80 ai = argv_iter_init_stream (fp); |
|
|
81 } |
|
|
82 else |
|
|
83 { |
|
|
84 fp = NULL; |
|
|
85 ai = argv_iter_init_argv (av[i]); |
|
|
86 } |
|
|
87 ASSERT (ai); |
|
|
88 |
|
|
89 while (1) |
|
|
90 { |
|
|
91 enum argv_iter_err ai_err; |
|
|
92 char *s = argv_iter (ai, &ai_err); |
|
|
93 ASSERT ((i == n_found) == (ai_err == AI_ERR_EOF)); |
|
|
94 ASSERT ((s == NULL) ^ (ai_err == AI_ERR_OK)); |
|
|
95 ASSERT (ai_err == AI_ERR_OK || ai_err == AI_ERR_EOF); |
|
|
96 if (ai_err == AI_ERR_OK) |
|
|
97 ++n_found; |
|
|
98 if (ai_err == AI_ERR_EOF) |
|
|
99 break; |
|
|
100 /* In stream mode, the strings are equal, but |
|
|
101 in argv mode the actual pointers are equal. */ |
|
|
102 ASSERT (use_stream |
|
|
103 ? STREQ (s, av[i][n_found - 1]) |
|
|
104 : s == av[i][n_found - 1]); |
|
|
105 } |
|
|
106 ASSERT (argv_iter_n_args (ai) == i); |
|
|
107 argv_iter_free (ai); |
|
|
108 if (fp) |
|
|
109 ASSERT (fclose (fp) == 0); |
|
|
110 } |
|
|
111 } |
|
|
112 |
|
|
113 return 0; |
|
|
114 } |
