1 /* Test argv iterator
2 Copyright (C) 2008-2012 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
21 #include "argv-iter.h"
22
23 #include <stdio.h>
24 #include <string.h>
25
26 #include "macros.h"
27
28 #define ARRAY_CARDINALITY(Array) (sizeof (Array) / sizeof *(Array))
29 #define STREQ(a, b) (strcmp (a, b) == 0)
30
31 static FILE *
write_nul_delimited_argv(char ** argv)32 write_nul_delimited_argv (char **argv)
33 {
34 FILE *fp = tmpfile ();
35 ASSERT (fp);
36 while (*argv)
37 {
38 size_t len = strlen (*argv) + 1;
39 ASSERT (fwrite (*argv, len, 1, fp) == 1);
40 argv++;
41 }
42 ASSERT (fflush (fp) == 0);
43 rewind (fp);
44 return fp;
45 }
46
47 int
main(void)48 main (void)
49 {
50 /* set_program_name (argv[0]); placate overzealous "syntax-check" test. */
51 static char one[] = "1";
52 static char two[] = "2";
53 static char three[] = "3";
54 static char *av[][4] = {
55 {NULL},
56 {one, NULL},
57 {one, two, NULL},
58 {one, two, three, NULL}
59 };
60
61 int use_stream;
62 for (use_stream = 0; use_stream < 2; use_stream++)
63 {
64 size_t i;
65 for (i = 0; i < ARRAY_CARDINALITY (av); i++)
66 {
67 FILE *fp;
68 struct argv_iterator *ai;
69 size_t n_found = 0;
70 if (use_stream)
71 {
72 /* Generate an identical list to be read via FP. */
73 ASSERT ((fp = write_nul_delimited_argv (av[i])) != NULL);
74 ai = argv_iter_init_stream (fp);
75 }
76 else
77 {
78 fp = NULL;
79 ai = argv_iter_init_argv (av[i]);
80 }
81 ASSERT (ai);
82
83 while (1)
84 {
85 enum argv_iter_err ai_err;
86 char *s = argv_iter (ai, &ai_err);
87 ASSERT ((i == n_found) == (ai_err == AI_ERR_EOF));
88 ASSERT ((s == NULL) ^ (ai_err == AI_ERR_OK));
89 ASSERT (ai_err == AI_ERR_OK || ai_err == AI_ERR_EOF);
90 if (ai_err == AI_ERR_OK)
91 ++n_found;
92 if (ai_err == AI_ERR_EOF)
93 break;
94 /* In stream mode, the strings are equal, but
95 in argv mode the actual pointers are equal. */
96 ASSERT (use_stream
97 ? STREQ (s, av[i][n_found - 1])
98 : s == av[i][n_found - 1]);
99 }
100 ASSERT (argv_iter_n_args (ai) == i);
101 argv_iter_free (ai);
102 if (fp)
103 ASSERT (fclose (fp) == 0);
104 }
105 }
106
107 return 0;
108 }
109