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