tests/test-argv-iter.c

   1 /* Test argv iterator
   2    Copyright (C) 2008-2017 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 *
  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
  48 main (void)
  49 {
  50   static char one[] = "1";
  51   static char two[] = "2";
  52   static char three[] = "3";
  53   static char *av[][4] = {
  54     {NULL},
  55     {one, NULL},
  56     {one, two, NULL},
  57     {one, two, three, NULL}
  58   };
  59
  60   int use_stream;
  61   for (use_stream = 0; use_stream < 2; use_stream++)
  62     {
  63       size_t i;
  64       for (i = 0; i < ARRAY_CARDINALITY (av); i++)
  65         {
  66           FILE *fp;
  67           struct argv_iterator *ai;
  68           size_t n_found = 0;
  69           if (use_stream)
  70             {
  71               /* Generate an identical list to be read via FP.  */
  72               ASSERT ((fp = write_nul_delimited_argv (av[i])) != NULL);
  73               ai = argv_iter_init_stream (fp);
  74             }
  75           else
  76             {
  77               fp = NULL;
  78               ai = argv_iter_init_argv (av[i]);
  79             }
  80           ASSERT (ai);
  81
  82           while (1)
  83             {
  84               enum argv_iter_err ai_err;
  85               char *s = argv_iter (ai, &ai_err);
  86               ASSERT ((i == n_found) == (ai_err == AI_ERR_EOF));
  87               ASSERT ((s == NULL) ^ (ai_err == AI_ERR_OK));
  88               ASSERT (ai_err == AI_ERR_OK || ai_err == AI_ERR_EOF);
  89               if (ai_err == AI_ERR_OK)
  90                 ++n_found;
  91               if (ai_err == AI_ERR_EOF)
  92                 break;
  93               /* In stream mode, the strings are equal, but
  94                  in argv mode the actual pointers are equal.  */
  95               ASSERT (use_stream
  96                       ? STREQ (s, av[i][n_found - 1])
  97                       : s == av[i][n_found - 1]);
  98             }
  99           ASSERT (argv_iter_n_args (ai) == i);
 100           argv_iter_free (ai);
 101           if (fp)
 102             ASSERT (fclose (fp) == 0);
 103         }
 104     }
 105
 106   return 0;
 107 }