tools/repo_rpmmd.c

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