tools/source_rpmmd.c

   1 #include <sys/types.h>
   2 #include <limits.h>
   3 #include <fcntl.h>
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include <string.h>
   7 #include <expat.h>
   8
   9 #include "pool.h"
  10 #include "source_rpmmd.h"
  11
  12
  13 enum state {
  14   STATE_START,
  15   STATE_METADATA,
  16   STATE_PACKAGE,
  17   STATE_NAME,
  18   STATE_ARCH,
  19   STATE_VERSION,
  20   STATE_FORMAT,
  21   STATE_PROVIDES,
  22   STATE_PROVIDESENTRY,
  23   STATE_REQUIRES,
  24   STATE_REQUIRESENTRY,
  25   STATE_OBSOLETES,
  26   STATE_OBSOLETESENTRY,
  27   STATE_CONFLICTS,
  28   STATE_CONFLICTSENTRY,
  29   STATE_RECOMMENDS,
  30   STATE_RECOMMENDSENTRY,
  31   STATE_SUPPLEMENTS,
  32   STATE_SUPPLEMENTSENTRY,
  33   STATE_SUGGESTS,
  34   STATE_SUGGESTSENTRY,
  35   STATE_ENHANCES,
  36   STATE_ENHANCESENTRY,
  37   STATE_FRESHENS,
  38   STATE_FRESHENSENTRY,
  39   STATE_FILE,
  40   NUMSTATES
  41 };
  42
  43
  44 struct stateswitch {
  45   enum state from;
  46   char *ename;
  47   enum state to;
  48   int docontent;
  49 };
  50
  51 static struct stateswitch stateswitches[] = {
  52   { STATE_START,       "metadata",        STATE_METADATA, 0 },
  53   { STATE_METADATA,    "package",         STATE_PACKAGE, 0 },
  54   { STATE_PACKAGE,     "name",            STATE_NAME, 1 },
  55   { STATE_PACKAGE,     "arch",            STATE_ARCH, 1 },
  56   { STATE_PACKAGE,     "version",         STATE_VERSION, 0 },
  57   { STATE_PACKAGE,     "format",          STATE_FORMAT, 0 },
  58   { STATE_FORMAT,      "rpm:provides",    STATE_PROVIDES, 0 },
  59   { STATE_FORMAT,      "rpm:requires",    STATE_REQUIRES, 0 },
  60   { STATE_FORMAT,      "rpm:obsoletes",   STATE_OBSOLETES , 0 },
  61   { STATE_FORMAT,      "rpm:conflicts",   STATE_CONFLICTS , 0 },
  62   { STATE_FORMAT,      "rpm:recommends" , STATE_RECOMMENDS , 0 },
  63   { STATE_FORMAT,      "rpm:supplements", STATE_SUPPLEMENTS, 0 },
  64   { STATE_FORMAT,      "rpm:suggests",    STATE_SUGGESTS, 0 },
  65   { STATE_FORMAT,      "rpm:enhances",    STATE_ENHANCES, 0 },
  66   { STATE_FORMAT,      "rpm:freshens",    STATE_FRESHENS, 0 },
  67   { STATE_FORMAT,      "file",            STATE_FILE, 1 },
  68   { STATE_PROVIDES,    "rpm:entry",       STATE_PROVIDESENTRY, 0 },
  69   { STATE_REQUIRES,    "rpm:entry",       STATE_REQUIRESENTRY, 0 },
  70   { STATE_OBSOLETES,   "rpm:entry",       STATE_OBSOLETESENTRY, 0 },
  71   { STATE_CONFLICTS,   "rpm:entry",       STATE_CONFLICTSENTRY, 0 },
  72   { STATE_RECOMMENDS,  "rpm:entry",       STATE_RECOMMENDSENTRY, 0 },
  73   { STATE_SUPPLEMENTS, "rpm:entry",       STATE_SUPPLEMENTSENTRY, 0 },
  74   { STATE_SUGGESTS,    "rpm:entry",       STATE_SUGGESTSENTRY, 0 },
  75   { STATE_ENHANCES,    "rpm:entry",       STATE_ENHANCESENTRY, 0 },
  76   { STATE_FRESHENS,    "rpm:entry",       STATE_FRESHENSENTRY, 0 },
  77   { NUMSTATES}
  78 };
  79
  80 struct deps {
  81   unsigned int provides;
  82   unsigned int requires;
  83   unsigned int obsoletes;
  84   unsigned int conflicts;
  85   unsigned int recommends;
  86   unsigned int supplements;
  87   unsigned int enhances;
  88   unsigned int suggests;
  89   unsigned int freshens;
  90 };
  91
  92 struct parsedata {
  93   int depth;
  94   enum state state;
  95   int statedepth;
  96   char *content;
  97   int lcontent;
  98   int acontent;
  99   int docontent;
 100   int numpacks;
 101   int pack;
 102   Pool *pool;
 103   Source *source;
 104   Solvable *start;
 105   struct deps *deps;
 106   struct stateswitch *swtab[NUMSTATES];
 107   enum state sbtab[NUMSTATES];
 108 };
 109
 110 static Id
 111 makeevr_atts(Pool *pool, struct parsedata *pd, const char **atts)
 112 {
 113   const char *e, *v, *r, *v2;
 114   char *c;
 115   int l;
 116
 117   e = v = r = 0;
 118   for (; *atts; atts += 2)
 119     {
 120       if (!strcmp(*atts, "epoch"))
 121         e = atts[1];
 122       else if (!strcmp(*atts, "ver"))
 123         v = atts[1];
 124       else if (!strcmp(*atts, "rel"))
 125         r = atts[1];
 126     }
 127   if (e && !strcmp(e, "0"))
 128     e = 0;
 129   if (v && !e)
 130     {
 131       for (v2 = v; *v2 >= '0' && *v2 <= '9'; v2++)
 132         ;
 133       if (v2 > v && *v2 == ':')
 134         e = "0";
 135     }
 136   l = 1;
 137   if (e)
 138     l += strlen(e) + 1;
 139   if (v)
 140     l += strlen(v);
 141   if (r)
 142     l += strlen(r) + 1;
 143   if (l > pd->acontent)
 144     {
 145       pd->content = realloc(pd->content, l + 256);
 146       pd->acontent = l + 256;
 147     }
 148   c = pd->content;
 149   if (e)
 150     {
 151       strcpy(c, e);
 152       c += strlen(c);
 153       *c++ = ':';
 154     }
 155   if (v)
 156     {
 157       strcpy(c, v);
 158       c += strlen(c);
 159     }
 160   if (r)
 161     {
 162       *c++ = '-';
 163       strcpy(c, r);
 164       c += strlen(c);
 165     }
 166   *c = 0;
 167   if (!*pd->content)
 168     return 0;
 169 #if 0
 170   fprintf(stderr, "evr: %s\n", pd->content);
 171 #endif
 172   return str2id(pool, pd->content, 1);
 173 }
 174
 175 static char *flagtab[] = {
 176   "GT",
 177   "EQ",
 178   "GE",
 179   "LT",
 180   "NE",
 181   "LE"
 182 };
 183
 184 static unsigned int
 185 adddep(Pool *pool, struct parsedata *pd, unsigned int olddeps, const char **atts, int isreq)
 186 {
 187   Id id, name;
 188   const char *n, *f, *k;
 189   const char **a;
 190
 191   n = f = k = 0;
 192   for (a = atts; *a; a += 2)
 193     {
 194       if (!strcmp(*a, "name"))
 195         n = a[1];
 196       else if (!strcmp(*a, "flags"))
 197         f = a[1];
 198       else if (!strcmp(*a, "kind"))
 199         k = a[1];
 200       else if (isreq && !strcmp(*a, "pre") && a[1][0] == '1')
 201         isreq = 2;
 202     }
 203   if (!n)
 204     return olddeps;
 205   if (k && !strcmp(k, "package"))
 206     k = 0;
 207   if (k)
 208     {
 209       int l = strlen(k) + 1 + strlen(n) + 1;
 210       if (l > pd->acontent)
 211         {
 212           pd->content = realloc(pd->content, l + 256);
 213           pd->acontent = l + 256;
 214         }
 215       sprintf(pd->content, "%s:%s", k, n);
 216       name = str2id(pool, pd->content, 1);
 217     }
 218   else
 219     name = str2id(pool, (char *)n, 1);
 220   if (f)
 221     {
 222       Id evr = makeevr_atts(pool, pd, atts);
 223       int flags;
 224       for (flags = 0; flags < 6; flags++)
 225         if (!strcmp(f, flagtab[flags]))
 226           break;
 227       flags = flags < 6 ? flags + 1 : 0;
 228       id = rel2id(pool, name, evr, flags, 1);
 229     }
 230   else
 231     id = name;
 232 #if 0
 233   fprintf(stderr, "new dep %s%s%s\n", id2str(pool, d), id2rel(pool, d), id2evr(pool, d));
 234 #endif
 235   return source_addid_dep(pd->source, olddeps, id, isreq);
 236 }
 237
 238
 239 static void XMLCALL
 240 startElement(void *userData, const char *name, const char **atts)
 241 {
 242   struct parsedata *pd = userData;
 243   Pool *pool = pd->pool;
 244   Solvable *s = pd->start ? pd->start + pd->pack : 0;
 245   struct stateswitch *sw;
 246
 247   if (pd->depth != pd->statedepth)
 248     {
 249       pd->depth++;
 250       return;
 251     }
 252   pd->depth++;
 253   for (sw = pd->swtab[pd->state]; sw->from == pd->state; sw++)
 254     if (!strcmp(sw->ename, name))
 255       break;
 256   if (sw->from != pd->state)
 257     {
 258 #if 0
 259       fprintf(stderr, "into unknown: %s\n", name);
 260 #endif
 261       return;
 262     }
 263   pd->state = sw->to;
 264   pd->docontent = sw->docontent;
 265   pd->statedepth = pd->depth;
 266   pd->lcontent = 0;
 267   *pd->content = 0;
 268   switch(pd->state)
 269     {
 270     case STATE_METADATA:
 271       for (; *atts; atts += 2)
 272         {
 273           if (!strcmp(*atts, "packages"))
 274             {
 275               pd->numpacks = atoi(atts[1]);
 276 #if 0
 277               fprintf(stderr, "numpacks: %d\n", pd->numpacks);
 278 #endif
 279               pool->solvables = realloc(pool->solvables, (pool->nsolvables + pd->numpacks) * sizeof(Solvable));
 280               pd->start = pool->solvables + pd->source->start;
 281               memset(pd->start, 0, pd->numpacks * sizeof(Solvable));
 282               pd->deps = calloc(pd->numpacks, sizeof(struct deps));
 283             }
 284         }
 285       break;
 286     case STATE_PACKAGE:
 287       if (pd->pack >= pd->numpacks)
 288         {
 289           fprintf(stderr, "repomd lied about the package number\n");
 290           exit(1);
 291         }
 292 #if 0
 293       fprintf(stderr, "package #%d\n", pd->pack);
 294 #endif
 295       break;
 296     case STATE_VERSION:
 297       s->evr = makeevr_atts(pool, pd, atts);
 298       break;
 299     case STATE_PROVIDES:
 300       pd->deps[pd->pack].provides = 0;
 301       break;
 302     case STATE_PROVIDESENTRY:
 303       pd->deps[pd->pack].provides = adddep(pool, pd, pd->deps[pd->pack].provides, atts, 0);
 304       break;
 305     case STATE_REQUIRES:
 306       pd->deps[pd->pack].requires = 0;
 307       break;
 308     case STATE_REQUIRESENTRY:
 309       pd->deps[pd->pack].requires = adddep(pool, pd, pd->deps[pd->pack].requires, atts, 1);
 310       break;
 311     case STATE_OBSOLETES:
 312       pd->deps[pd->pack].obsoletes = 0;
 313       break;
 314     case STATE_OBSOLETESENTRY:
 315       pd->deps[pd->pack].obsoletes = adddep(pool, pd, pd->deps[pd->pack].obsoletes, atts, 0);
 316       break;
 317     case STATE_CONFLICTS:
 318       pd->deps[pd->pack].conflicts = 0;
 319       break;
 320     case STATE_CONFLICTSENTRY:
 321       pd->deps[pd->pack].conflicts = adddep(pool, pd, pd->deps[pd->pack].conflicts, atts, 0);
 322       break;
 323     case STATE_RECOMMENDS:
 324       pd->deps[pd->pack].recommends = 0;
 325       break;
 326     case STATE_RECOMMENDSENTRY:
 327       pd->deps[pd->pack].recommends = adddep(pool, pd, pd->deps[pd->pack].recommends, atts, 0);
 328       break;
 329     case STATE_SUPPLEMENTS:
 330       pd->deps[pd->pack].supplements= 0;
 331       break;
 332     case STATE_SUPPLEMENTSENTRY:
 333       pd->deps[pd->pack].supplements = adddep(pool, pd, pd->deps[pd->pack].supplements, atts, 0);
 334       break;
 335     case STATE_SUGGESTS:
 336       pd->deps[pd->pack].suggests = 0;
 337       break;
 338     case STATE_SUGGESTSENTRY:
 339       pd->deps[pd->pack].suggests = adddep(pool, pd, pd->deps[pd->pack].suggests, atts, 0);
 340       break;
 341     case STATE_ENHANCES:
 342       pd->deps[pd->pack].enhances = 0;
 343       break;
 344     case STATE_ENHANCESENTRY:
 345       pd->deps[pd->pack].enhances = adddep(pool, pd, pd->deps[pd->pack].enhances, atts, 0);
 346       break;
 347     case STATE_FRESHENS:
 348       pd->deps[pd->pack].freshens = 0;
 349       break;
 350     case STATE_FRESHENSENTRY:
 351       pd->deps[pd->pack].freshens = adddep(pool, pd, pd->deps[pd->pack].freshens, atts, 0);
 352       break;
 353     default:
 354       break;
 355     }
 356 }
 357
 358 static void XMLCALL
 359 endElement(void *userData, const char *name)
 360 {
 361   struct parsedata *pd = userData;
 362   Pool *pool = pd->pool;
 363   Solvable *s = pd->start ? pd->start + pd->pack : 0;
 364   Id id;
 365
 366   if (pd->depth != pd->statedepth)
 367     {
 368       pd->depth--;
 369       // printf("back from unknown %d %d %d\n", pd->state, pd->depth, pd->statedepth);
 370       return;
 371     }
 372   pd->depth--;
 373   pd->statedepth--;
 374   switch (pd->state)
 375     {
 376     case STATE_PACKAGE:
 377       s->source = pd->source;
 378       if (!s->arch)
 379         s->arch = ARCH_NOARCH;
 380       if (s->arch != ARCH_SRC && s->arch != ARCH_NOSRC)
 381         pd->deps[pd->pack].provides = source_addid_dep(pd->source, pd->deps[pd->pack].provides, rel2id(pool, s->name, s->evr, REL_EQ, 1), 0);
 382       pd->deps[pd->pack].supplements = source_fix_legacy(pd->source, pd->deps[pd->pack].provides, pd->deps[pd->pack].supplements);
 383       pd->pack++;
 384       break;
 385     case STATE_NAME:
 386       s->name = str2id(pool, pd->content, 1);
 387       break;
 388     case STATE_ARCH:
 389       s->arch = str2id(pool, pd->content, 1);
 390       break;
 391     case STATE_FILE:
 392       id = str2id(pool, pd->content, 1);
 393       pd->deps[pd->pack].provides = source_addid(pd->source, pd->deps[pd->pack].provides, id);
 394       break;
 395     default:
 396       break;
 397     }
 398   pd->state = pd->sbtab[pd->state];
 399   pd->docontent = 0;
 400   // printf("back from known %d %d %d\n", pd->state, pd->depth, pd->statedepth);
 401 }
 402
 403 static void XMLCALL
 404 characterData(void *userData, const XML_Char *s, int len)
 405 {
 406   struct parsedata *pd = userData;
 407   int l;
 408   char *c;
 409
 410   if (!pd->docontent)
 411     return;
 412   l = pd->lcontent + len + 1;
 413   if (l > pd->acontent)
 414     {
 415       pd->content = realloc(pd->content, l + 256);
 416       pd->acontent = l + 256;
 417     }
 418   c = pd->content + pd->lcontent;
 419   pd->lcontent += len;
 420   while (len-- > 0)
 421     *c++ = *s++;
 422   *c = 0;
 423 }
 424
 425
 426 #define BUFF_SIZE 8192
 427
 428 Source *
 429 pool_addsource_rpmmd(Pool *pool, FILE *fp)
 430 {
 431   struct parsedata pd;
 432   char buf[BUFF_SIZE];
 433   int i, l;
 434   Source *source;
 435   Solvable *s;
 436   struct deps *deps;
 437   struct stateswitch *sw;
 438
 439   source = pool_addsource_empty(pool);
 440   memset(&pd, 0, sizeof(pd));
 441   for (i = 0, sw = stateswitches; sw->from != NUMSTATES; i++, sw++)
 442     {
 443       if (!pd.swtab[sw->from])
 444         pd.swtab[sw->from] = sw;
 445       pd.sbtab[sw->to] = sw->from;
 446     }
 447   pd.pool = pool;
 448   pd.source = source;
 449   pd.content = malloc(256);
 450   pd.acontent = 256;
 451   pd.lcontent = 0;
 452   XML_Parser parser = XML_ParserCreate(NULL);
 453   XML_SetUserData(parser, &pd);
 454   XML_SetElementHandler(parser, startElement, endElement);
 455   XML_SetCharacterDataHandler(parser, characterData);
 456   for (;;)
 457     {
 458       l = fread(buf, 1, sizeof(buf), fp);
 459       if (XML_Parse(parser, buf, l, l == 0) == XML_STATUS_ERROR)
 460         {
 461           fprintf(stderr, "%s at line %u\n", XML_ErrorString(XML_GetErrorCode(parser)), (unsigned int)XML_GetCurrentLineNumber(parser));
 462           exit(1);
 463         }
 464       if (l == 0)
 465         break;
 466     }
 467   XML_ParserFree(parser);
 468
 469   pool->nsolvables += pd.pack;
 470   source->nsolvables = pd.pack;
 471
 472   deps = pd.deps;
 473   s = pool->solvables + source->start;
 474   for (i = 0; i < pd.pack; i++, s++)
 475     {
 476       s->provides = deps[i].provides;
 477       s->requires = deps[i].requires;
 478       s->conflicts = deps[i].conflicts;
 479       s->obsoletes = deps[i].obsoletes;
 480       s->recommends = deps[i].recommends;
 481       s->supplements = deps[i].supplements;
 482       s->suggests = deps[i].suggests;
 483       s->enhances = deps[i].enhances;
 484       s->freshens = deps[i].freshens;
 485     }
 486   free(deps);
 487   free(pd.content);
 488   return source;
 489 }