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