rpmio/argv.c

   1 /** \ingroup rpmio
   2  * \file rpmio/argv.c
   3  */
   4
   5 #include "system.h"
   6
   7 #include <stdlib.h>
   8 #include <rpm/argv.h>
   9 #include <rpm/rpmstring.h>
  10
  11 #include "debug.h"
  12
  13 void argvPrint(const char * msg, ARGV_const_t argv, FILE * fp)
  14 {
  15     ARGV_const_t av;
  16
  17     if (fp == NULL) fp = stderr;
  18
  19     if (msg)
  20         fprintf(fp, "===================================== %s\n", msg);
  21
  22     if (argv)
  23     for (av = argv; *av; av++)
  24         fprintf(fp, "%s\n", *av);
  25
  26 }
  27
  28 ARGV_t argvNew(void)
  29 {
  30     ARGV_t argv = xcalloc(1, sizeof(*argv));
  31     return argv;
  32 }
  33
  34 ARGI_t argiFree(ARGI_t argi)
  35 {
  36     if (argi) {
  37         argi->nvals = 0;
  38         argi->vals = _free(argi->vals);
  39     }
  40     argi = _free(argi);
  41     return NULL;
  42 }
  43
  44 ARGV_t argvFree(ARGV_t argv)
  45 {
  46     ARGV_t av;
  47
  48     if (argv)
  49     for (av = argv; *av; av++)
  50         *av = _free(*av);
  51     argv = _free(argv);
  52     return NULL;
  53 }
  54
  55 int argiCount(ARGI_const_t argi)
  56 {
  57     int nvals = 0;
  58     if (argi)
  59         nvals = argi->nvals;
  60     return nvals;
  61 }
  62
  63 ARGint_t argiData(ARGI_const_t argi)
  64 {
  65     ARGint_t vals = NULL;
  66     if (argi && argi->nvals > 0)
  67         vals = argi->vals;
  68     return vals;
  69 }
  70
  71 int argvCount(ARGV_const_t argv)
  72 {
  73     int argc = 0;
  74     if (argv)
  75     while (argv[argc] != NULL)
  76         argc++;
  77     return argc;
  78 }
  79
  80 ARGV_t argvData(ARGV_t argv)
  81 {
  82     return argv;
  83 }
  84
  85 int argvCmp(const void * a, const void * b)
  86 {
  87     const char *astr = *(ARGV_t)a;
  88     const char *bstr = *(ARGV_t)b;
  89     return strcmp(astr, bstr);
  90 }
  91
  92 int argvSort(ARGV_t argv, int (*compar)(const void *, const void *))
  93 {
  94     if (compar == NULL)
  95         compar = argvCmp;
  96     qsort(argv, argvCount(argv), sizeof(*argv), compar);
  97     return 0;
  98 }
  99
 100 ARGV_t argvSearch(ARGV_const_t argv, const char *val,
 101                 int (*compar)(const void *, const void *))
 102 {
 103     if (argv == NULL)
 104         return NULL;
 105     if (compar == NULL)
 106         compar = argvCmp;
 107     return bsearch(&val, argv, argvCount(argv), sizeof(*argv), compar);
 108 }
 109
 110 int argiAdd(ARGI_t * argip, int ix, int val)
 111 {
 112     ARGI_t argi;
 113
 114     if (argip == NULL)
 115         return -1;
 116     if (*argip == NULL)
 117         *argip = xcalloc(1, sizeof(**argip));
 118     argi = *argip;
 119     if (ix < 0)
 120         ix = argi->nvals;
 121     if (ix >= argi->nvals) {
 122         argi->vals = xrealloc(argi->vals, (ix + 1) * sizeof(*argi->vals));
 123         memset(argi->vals + argi->nvals, 0,
 124                 (ix - argi->nvals) * sizeof(*argi->vals));
 125         argi->nvals = ix + 1;
 126     }
 127     argi->vals[ix] = val;
 128     return 0;
 129 }
 130
 131 int argvAdd(ARGV_t * argvp, const char *val)
 132 {
 133     ARGV_t argv;
 134     int argc;
 135
 136     if (argvp == NULL)
 137         return -1;
 138     argc = argvCount(*argvp);
 139     *argvp = xrealloc(*argvp, (argc + 1 + 1) * sizeof(**argvp));
 140     argv = *argvp;
 141     argv[argc++] = xstrdup(val);
 142     argv[argc  ] = NULL;
 143     return 0;
 144 }
 145
 146 int argvAddNum(ARGV_t *argvp, int val)
 147 {
 148     char *valstr = NULL;
 149     int rc;
 150     rasprintf(&valstr, "%d", val);
 151     rc = argvAdd(argvp, valstr);
 152     free(valstr);
 153     return rc;
 154 }
 155
 156 int argvAppend(ARGV_t * argvp, ARGV_const_t av)
 157 {
 158     ARGV_t argv = *argvp;
 159     int argc = argvCount(argv);
 160     int ac = argvCount(av);
 161     int i;
 162
 163     argv = xrealloc(argv, (argc + ac + 1) * sizeof(*argv));
 164     for (i = 0; i < ac; i++)
 165         argv[argc + i] = xstrdup(av[i]);
 166     argv[argc + ac] = NULL;
 167     *argvp = argv;
 168     return 0;
 169 }
 170
 171 ARGV_t argvSplitString(const char * str, const char * seps, argvFlags flags)
 172 {
 173     char *dest = NULL;
 174     ARGV_t argv;
 175     int argc = 1;
 176     const char * s;
 177     char * t;
 178     int c;
 179
 180     if (str == NULL || seps == NULL)
 181         return NULL;
 182
 183     dest = xmalloc(strlen(str) + 1);
 184     for (argc = 1, s = str, t = dest; (c = *s); s++, t++) {
 185         if (strchr(seps, c)) {
 186             argc++;
 187             c = '\0';
 188         }
 189         *t = c;
 190     }
 191     *t = '\0';
 192
 193     argv = xmalloc( (argc + 1) * sizeof(*argv));
 194
 195     for (c = 0, s = dest; s < t; s+= strlen(s) + 1) {
 196         if (*s == '\0' && (flags & ARGV_SKIPEMPTY))
 197             continue;
 198         argv[c] = xstrdup(s);
 199         c++;
 200     }
 201     argv[c] = NULL;
 202     free(dest);
 203     return argv;
 204 }
 205
 206 /* Backwards compatibility */
 207 int argvSplit(ARGV_t * argvp, const char * str, const char * seps)
 208 {
 209     if (argvp) {
 210         *argvp = argvSplitString(str, seps, ARGV_SKIPEMPTY);
 211     }
 212     return 0;
 213 }
 214
 215 char *argvJoin(ARGV_const_t argv, const char *sep)
 216 {
 217     char *dest = NULL;
 218     char * const *arg;
 219
 220     for (arg = argv; arg && *arg; arg++) {
 221         rstrscat(&dest, *arg, *(arg+1) ? sep : "", NULL);
 222     }
 223     return dest;
 224 }
 225