popt/poptparse.c

   1 /** \ingroup popt
   2  * \file popt/poptparse.c
   3  */
   4
   5 /* (C) 1998-2000 Red Hat, Inc. -- Licensing details are in the COPYING
   6    file accompanying popt source distributions, available from
   7    ftp://ftp.rpm.org/pub/rpm/dist. */
   8
   9 #include "system.h"
  10
  11 #define POPT_ARGV_ARRAY_GROW_DELTA 5
  12
  13 int poptDupArgv(int argc, const char **argv,
  14                 int * argcPtr, const char *** argvPtr)
  15 {
  16     size_t nb = (argc + 1) * sizeof(*argv);
  17     const char ** argv2;
  18     char * dst;
  19     int i;
  20
  21     if (argc <= 0 || argv == NULL)      /* XXX can't happen */
  22         return POPT_ERROR_NOARG;
  23     for (i = 0; i < argc; i++) {
  24         if (argv[i] == NULL)
  25             return POPT_ERROR_NOARG;
  26         nb += strlen(argv[i]) + 1;
  27     }
  28
  29     dst = malloc(nb);
  30     if (dst == NULL)                    /* XXX can't happen */
  31         return POPT_ERROR_MALLOC;
  32     argv2 = (void *) dst;
  33     dst += (argc + 1) * sizeof(*argv);
  34
  35     /*@-branchstate@*/
  36     for (i = 0; i < argc; i++) {
  37         argv2[i] = dst;
  38         dst += strlen(strcpy(dst, argv[i])) + 1;
  39     }
  40     /*@=branchstate@*/
  41     argv2[argc] = NULL;
  42
  43     if (argvPtr) {
  44         *argvPtr = argv2;
  45     } else {
  46         free(argv2);
  47         argv2 = NULL;
  48     }
  49     if (argcPtr)
  50         *argcPtr = argc;
  51     return 0;
  52 }
  53
  54 int poptParseArgvString(const char * s, int * argcPtr, const char *** argvPtr)
  55 {
  56     const char * src;
  57     char quote = '\0';
  58     int argvAlloced = POPT_ARGV_ARRAY_GROW_DELTA;
  59     const char ** argv = malloc(sizeof(*argv) * argvAlloced);
  60     int argc = 0;
  61     int buflen = strlen(s) + 1;
  62     char * buf = memset(alloca(buflen), 0, buflen);
  63     int rc = POPT_ERROR_MALLOC;
  64
  65     if (argv == NULL) return rc;
  66     argv[argc] = buf;
  67
  68     for (src = s; *src != '\0'; src++) {
  69         if (quote == *src) {
  70             quote = '\0';
  71         } else if (quote != '\0') {
  72             if (*src == '\\') {
  73                 src++;
  74                 if (!*src) {
  75                     rc = POPT_ERROR_BADQUOTE;
  76                     goto exit;
  77                 }
  78                 if (*src != quote) *buf++ = '\\';
  79             }
  80             *buf++ = *src;
  81         } else if (isspace(*src)) {
  82             if (*argv[argc] != '\0') {
  83                 buf++, argc++;
  84                 if (argc == argvAlloced) {
  85                     argvAlloced += POPT_ARGV_ARRAY_GROW_DELTA;
  86                     argv = realloc(argv, sizeof(*argv) * argvAlloced);
  87                     if (argv == NULL) goto exit;
  88                 }
  89                 argv[argc] = buf;
  90             }
  91         } else switch (*src) {
  92           case '"':
  93           case '\'':
  94             quote = *src;
  95             /*@switchbreak@*/ break;
  96           case '\\':
  97             src++;
  98             if (!*src) {
  99                 rc = POPT_ERROR_BADQUOTE;
 100                 goto exit;
 101             }
 102             /*@fallthrough@*/
 103           default:
 104             *buf++ = *src;
 105             /*@switchbreak@*/ break;
 106         }
 107     }
 108
 109     if (strlen(argv[argc])) {
 110         argc++, buf++;
 111     }
 112
 113     rc = poptDupArgv(argc, argv, argcPtr, argvPtr);
 114
 115 exit:
 116     if (argv) free(argv);
 117     return rc;
 118 }