- allow repositories that don't consist of a single block of solvables
[platform/upstream/libsolv.git] / 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 "util.h"
18 #include "repo_rpmmd.h"
19
20
21 enum state {
22   STATE_START,
23   STATE_METADATA,
24   STATE_PACKAGE,
25   STATE_NAME,
26   STATE_ARCH,
27   STATE_VERSION,
28   STATE_FORMAT,
29   STATE_VENDOR,
30   STATE_PROVIDES,
31   STATE_PROVIDESENTRY,
32   STATE_REQUIRES,
33   STATE_REQUIRESENTRY,
34   STATE_OBSOLETES,
35   STATE_OBSOLETESENTRY,
36   STATE_CONFLICTS,
37   STATE_CONFLICTSENTRY,
38   STATE_RECOMMENDS,
39   STATE_RECOMMENDSENTRY,
40   STATE_SUPPLEMENTS,
41   STATE_SUPPLEMENTSENTRY,
42   STATE_SUGGESTS,
43   STATE_SUGGESTSENTRY,
44   STATE_ENHANCES,
45   STATE_ENHANCESENTRY,
46   STATE_FRESHENS,
47   STATE_FRESHENSENTRY,
48   STATE_FILE,
49   NUMSTATES
50 };
51
52
53 struct stateswitch {
54   enum state from;
55   char *ename;
56   enum state to;
57   int docontent;
58 };
59
60 static struct stateswitch stateswitches[] = {
61   { STATE_START,       "metadata",        STATE_METADATA, 0 },
62   { STATE_METADATA,    "package",         STATE_PACKAGE, 0 },
63   { STATE_PACKAGE,     "name",            STATE_NAME, 1 },
64   { STATE_PACKAGE,     "arch",            STATE_ARCH, 1 },
65   { STATE_PACKAGE,     "version",         STATE_VERSION, 0 },
66   { STATE_PACKAGE,     "format",          STATE_FORMAT, 0 },
67   { STATE_FORMAT,      "rpm:vendor",      STATE_VENDOR, 1 },
68   { STATE_FORMAT,      "rpm:provides",    STATE_PROVIDES, 0 },
69   { STATE_FORMAT,      "rpm:requires",    STATE_REQUIRES, 0 },
70   { STATE_FORMAT,      "rpm:obsoletes",   STATE_OBSOLETES , 0 },
71   { STATE_FORMAT,      "rpm:conflicts",   STATE_CONFLICTS , 0 },
72   { STATE_FORMAT,      "rpm:recommends" , STATE_RECOMMENDS , 0 },
73   { STATE_FORMAT,      "rpm:supplements", STATE_SUPPLEMENTS, 0 },
74   { STATE_FORMAT,      "rpm:suggests",    STATE_SUGGESTS, 0 },
75   { STATE_FORMAT,      "rpm:enhances",    STATE_ENHANCES, 0 },
76   { STATE_FORMAT,      "rpm:freshens",    STATE_FRESHENS, 0 },
77   { STATE_FORMAT,      "file",            STATE_FILE, 1 },
78   { STATE_PROVIDES,    "rpm:entry",       STATE_PROVIDESENTRY, 0 },
79   { STATE_REQUIRES,    "rpm:entry",       STATE_REQUIRESENTRY, 0 },
80   { STATE_OBSOLETES,   "rpm:entry",       STATE_OBSOLETESENTRY, 0 },
81   { STATE_CONFLICTS,   "rpm:entry",       STATE_CONFLICTSENTRY, 0 },
82   { STATE_RECOMMENDS,  "rpm:entry",       STATE_RECOMMENDSENTRY, 0 },
83   { STATE_SUPPLEMENTS, "rpm:entry",       STATE_SUPPLEMENTSENTRY, 0 },
84   { STATE_SUGGESTS,    "rpm:entry",       STATE_SUGGESTSENTRY, 0 },
85   { STATE_ENHANCES,    "rpm:entry",       STATE_ENHANCESENTRY, 0 },
86   { STATE_FRESHENS,    "rpm:entry",       STATE_FRESHENSENTRY, 0 },
87   { NUMSTATES}
88 };
89
90 struct parsedata {
91   int depth;
92   enum state state;
93   int statedepth;
94   char *content;
95   int lcontent;
96   int acontent;
97   int docontent;
98   int numpacks;
99   int pack;
100   Pool *pool;
101   Repo *repo;
102   Solvable *start;
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 = xrealloc(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;
185   const char *n, *f, *k;
186   const char **a;
187
188   n = f = k = 0;
189   for (a = atts; *a; a += 2)
190     {
191       if (!strcmp(*a, "name"))
192         n = a[1];
193       else if (!strcmp(*a, "flags"))
194         f = a[1];
195       else if (!strcmp(*a, "kind"))
196         k = a[1];
197       else if (isreq && !strcmp(*a, "pre") && a[1][0] == '1')
198         isreq = 2;
199     }
200   if (!n)
201     return olddeps;
202   if (k && !strcmp(k, "package"))
203     k = 0;
204   if (k)
205     {
206       int l = strlen(k) + 1 + strlen(n) + 1;
207       if (l > pd->acontent)
208         {
209           pd->content = xrealloc(pd->content, l + 256);
210           pd->acontent = l + 256;
211         }
212       sprintf(pd->content, "%s:%s", k, n); 
213       name = str2id(pool, pd->content, 1); 
214     }
215   else
216     name = str2id(pool, (char *)n, 1);
217   if (f)
218     {
219       Id evr = makeevr_atts(pool, pd, atts);
220       int flags;
221       for (flags = 0; flags < 6; flags++)
222         if (!strcmp(f, flagtab[flags]))
223           break;
224       flags = flags < 6 ? flags + 1 : 0;
225       id = rel2id(pool, name, evr, flags, 1);
226     }
227   else
228     id = name;
229 #if 0
230   fprintf(stderr, "new dep %s%s%s\n", id2str(pool, d), id2rel(pool, d), id2evr(pool, d));
231 #endif
232   return repo_addid_dep(pd->repo, olddeps, id, isreq);
233 }
234
235
236 static void XMLCALL
237 startElement(void *userData, const char *name, const char **atts)
238 {
239   struct parsedata *pd = userData;
240   Pool *pool = pd->pool;
241   Solvable *s = pd->start ? pd->start + pd->pack : 0;
242   struct stateswitch *sw;
243
244   if (pd->depth != pd->statedepth)
245     {
246       pd->depth++;
247       return;
248     }
249   pd->depth++;
250   for (sw = pd->swtab[pd->state]; sw->from == pd->state; sw++)
251     if (!strcmp(sw->ename, name))
252       break;
253   if (sw->from != pd->state)
254     {
255 #if 0
256       fprintf(stderr, "into unknown: %s\n", name);
257 #endif
258       return;
259     }
260   pd->state = sw->to;
261   pd->docontent = sw->docontent;
262   pd->statedepth = pd->depth;
263   pd->lcontent = 0;
264   *pd->content = 0;
265   switch(pd->state)
266     {
267     case STATE_METADATA:
268       for (; *atts; atts += 2)
269         {
270           if (!strcmp(*atts, "packages"))
271             {
272               pd->numpacks = atoi(atts[1]);
273 #if 0
274               fprintf(stderr, "numpacks: %d\n", pd->numpacks);
275 #endif
276               pool->solvables = xrealloc2(pool->solvables, (pool->nsolvables + pd->numpacks), sizeof(Solvable));
277               pd->start = pool->solvables + pool->nsolvables;
278               memset(pd->start, 0, pd->numpacks * sizeof(Solvable));
279             }
280         }
281       break;
282     case STATE_PACKAGE:
283       if (pd->pack >= pd->numpacks)
284         {
285           fprintf(stderr, "repomd lied about the package number\n");
286           exit(1);
287         }
288 #if 0
289       fprintf(stderr, "package #%d\n", pd->pack);
290 #endif
291       break;
292     case STATE_VERSION:
293       s->evr = makeevr_atts(pool, pd, atts);
294       break;
295     case STATE_PROVIDES:
296       s->provides = 0;
297       break;
298     case STATE_PROVIDESENTRY:
299       s->provides = adddep(pool, pd, s->provides, atts, 0);
300       break;
301     case STATE_REQUIRES:
302       s->requires = 0;
303       break;
304     case STATE_REQUIRESENTRY:
305       s->requires = adddep(pool, pd, s->requires, atts, 1);
306       break;
307     case STATE_OBSOLETES:
308       s->obsoletes = 0;
309       break;
310     case STATE_OBSOLETESENTRY:
311       s->obsoletes = adddep(pool, pd, s->obsoletes, atts, 0);
312       break;
313     case STATE_CONFLICTS:
314       s->conflicts = 0;
315       break;
316     case STATE_CONFLICTSENTRY:
317       s->conflicts = adddep(pool, pd, s->conflicts, atts, 0);
318       break;
319     case STATE_RECOMMENDS:
320       s->recommends = 0;
321       break;
322     case STATE_RECOMMENDSENTRY:
323       s->recommends = adddep(pool, pd, s->recommends, atts, 0);
324       break;
325     case STATE_SUPPLEMENTS:
326       s->supplements= 0;
327       break;
328     case STATE_SUPPLEMENTSENTRY:
329       s->supplements = adddep(pool, pd, s->supplements, atts, 0);
330       break;
331     case STATE_SUGGESTS:
332       s->suggests = 0;
333       break;
334     case STATE_SUGGESTSENTRY:
335       s->suggests = adddep(pool, pd, s->suggests, atts, 0);
336       break;
337     case STATE_ENHANCES:
338       s->enhances = 0;
339       break;
340     case STATE_ENHANCESENTRY:
341       s->enhances = adddep(pool, pd, s->enhances, atts, 0);
342       break;
343     case STATE_FRESHENS:
344       s->freshens = 0;
345       break;
346     case STATE_FRESHENSENTRY:
347       s->freshens = adddep(pool, pd, s->freshens, atts, 0);
348       break;
349     default:
350       break;
351     }
352 }
353
354 static void XMLCALL
355 endElement(void *userData, const char *name)
356 {
357   struct parsedata *pd = userData;
358   Pool *pool = pd->pool;
359   Solvable *s = pd->start ? pd->start + pd->pack : 0;
360   Id id;
361
362   if (pd->depth != pd->statedepth)
363     {
364       pd->depth--;
365       // printf("back from unknown %d %d %d\n", pd->state, pd->depth, pd->statedepth);
366       return;
367     }
368   pd->depth--;
369   pd->statedepth--;
370   switch (pd->state)
371     {
372     case STATE_PACKAGE:
373       s->repo = pd->repo;
374       if (!s->arch)
375         s->arch = ARCH_NOARCH;
376       if (s->arch != ARCH_SRC && s->arch != ARCH_NOSRC)
377         s->provides = repo_addid_dep(pd->repo, s->provides, rel2id(pool, s->name, s->evr, REL_EQ, 1), 0);
378       s->supplements = repo_fix_legacy(pd->repo, s->provides, s->supplements);
379       pd->pack++;
380       break;
381     case STATE_NAME:
382       s->name = str2id(pool, pd->content, 1);
383       break;
384     case STATE_ARCH:
385       s->arch = str2id(pool, pd->content, 1);
386       break;
387     case STATE_VENDOR:
388       s->vendor = str2id(pool, pd->content, 1);
389       break;
390     case STATE_FILE:
391       id = str2id(pool, pd->content, 1);
392       s->provides = repo_addid(pd->repo, s->provides, id);
393       break;
394     default:
395       break;
396     }
397   pd->state = pd->sbtab[pd->state];
398   pd->docontent = 0;
399   // printf("back from known %d %d %d\n", pd->state, pd->depth, pd->statedepth);
400 }
401
402 static void XMLCALL
403 characterData(void *userData, const XML_Char *s, int len)
404 {
405   struct parsedata *pd = userData;
406   int l;
407   char *c;
408
409   if (!pd->docontent)
410     return;
411   l = pd->lcontent + len + 1;
412   if (l > pd->acontent)
413     {
414       pd->content = xrealloc(pd->content, l + 256);
415       pd->acontent = l + 256;
416     }
417   c = pd->content + pd->lcontent;
418   pd->lcontent += len;
419   while (len-- > 0)
420     *c++ = *s++;
421   *c = 0;
422 }
423
424
425 #define BUFF_SIZE 8192
426
427 void
428 repo_add_rpmmd(Repo *repo, FILE *fp)
429 {
430   Pool *pool = repo->pool;
431   struct parsedata pd;
432   char buf[BUFF_SIZE];
433   int i, l;
434   struct stateswitch *sw;
435
436   if (!repo->start || repo->start == repo->end)
437     repo->start = pool->nsolvables;
438   repo->end = pool->nsolvables;
439
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.repo = repo;
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   repo->nsolvables += pd.pack;
471   repo->end += pd.pack;
472
473   free(pd.content);
474 }