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