- repo_rpmdb: read in RPMTAG_SHA1HEADER
[platform/upstream/libsolv.git] / ext / repo_rpmdb.c
1 /*
2  * Copyright (c) 2007-2012, Novell Inc.
3  *
4  * This program is licensed under the BSD license, read LICENSE.BSD
5  * for further information
6  */
7
8 /*
9  * repo_rpmdb
10  *
11  * convert rpm db to repo
12  *
13  */
14
15 #include <sys/types.h>
16 #include <sys/stat.h>
17 #include <limits.h>
18 #include <fcntl.h>
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <string.h>
22 #include <unistd.h>
23 #include <assert.h>
24 #include <stdint.h>
25 #include <errno.h>
26
27 #include <rpm/rpmio.h>
28 #include <rpm/rpmpgp.h>
29 #ifndef RPM5
30 #include <rpm/header.h>
31 #endif
32 #include <rpm/rpmdb.h>
33
34 #ifndef DB_CREATE
35 # ifdef FEDORA
36 #  include <db.h>
37 # else
38 #  include <rpm/db.h>
39 # endif
40 #endif
41
42 #include "pool.h"
43 #include "repo.h"
44 #include "hash.h"
45 #include "util.h"
46 #include "queue.h"
47 #include "chksum.h"
48 #include "repo_rpmdb.h"
49
50 /* 3: added triggers */
51 /* 4: fixed triggers */
52 #define RPMDB_COOKIE_VERSION 4
53
54 #define TAG_SIGBASE             256
55 #define TAG_SHA1HEADER          TAG_SIGBASE+13
56 #define TAG_NAME                1000
57 #define TAG_VERSION             1001
58 #define TAG_RELEASE             1002
59 #define TAG_EPOCH               1003
60 #define TAG_SUMMARY             1004
61 #define TAG_DESCRIPTION         1005
62 #define TAG_BUILDTIME           1006
63 #define TAG_BUILDHOST           1007
64 #define TAG_INSTALLTIME         1008
65 #define TAG_SIZE                1009
66 #define TAG_DISTRIBUTION        1010
67 #define TAG_VENDOR              1011
68 #define TAG_LICENSE             1014
69 #define TAG_PACKAGER            1015
70 #define TAG_GROUP               1016
71 #define TAG_URL                 1020
72 #define TAG_ARCH                1022
73 #define TAG_FILESIZES           1028
74 #define TAG_FILEMODES           1030
75 #define TAG_FILEMD5S            1035
76 #define TAG_FILELINKTOS         1036
77 #define TAG_FILEFLAGS           1037
78 #define TAG_SOURCERPM           1044
79 #define TAG_PROVIDENAME         1047
80 #define TAG_REQUIREFLAGS        1048
81 #define TAG_REQUIRENAME         1049
82 #define TAG_REQUIREVERSION      1050
83 #define TAG_NOSOURCE            1051
84 #define TAG_NOPATCH             1052
85 #define TAG_CONFLICTFLAGS       1053
86 #define TAG_CONFLICTNAME        1054
87 #define TAG_CONFLICTVERSION     1055
88 #define TAG_TRIGGERNAME         1066
89 #define TAG_TRIGGERVERSION      1067
90 #define TAG_TRIGGERFLAGS        1068
91 #define TAG_OBSOLETENAME        1090
92 #define TAG_FILEDEVICES         1095
93 #define TAG_FILEINODES          1096
94 #define TAG_PROVIDEFLAGS        1112
95 #define TAG_PROVIDEVERSION      1113
96 #define TAG_OBSOLETEFLAGS       1114
97 #define TAG_OBSOLETEVERSION     1115
98 #define TAG_DIRINDEXES          1116
99 #define TAG_BASENAMES           1117
100 #define TAG_DIRNAMES            1118
101 #define TAG_PAYLOADFORMAT       1124
102 #define TAG_PATCHESNAME         1133
103 #define TAG_FILECOLORS          1140
104 #define TAG_SUGGESTSNAME        1156
105 #define TAG_SUGGESTSVERSION     1157
106 #define TAG_SUGGESTSFLAGS       1158
107 #define TAG_ENHANCESNAME        1159
108 #define TAG_ENHANCESVERSION     1160
109 #define TAG_ENHANCESFLAGS       1161
110
111 /* rpm5 tags */
112 #define TAG_DISTEPOCH           1218
113
114 /* rpm4 tags */
115 #define TAG_LONGFILESIZES       5008
116 #define TAG_LONGSIZE            5009
117
118 #define SIGTAG_SIZE             1000
119 #define SIGTAG_PGP              1002    /* RSA signature */
120 #define SIGTAG_MD5              1004    /* header+payload md5 checksum */
121 #define SIGTAG_GPG              1005    /* DSA signature */
122
123 #define DEP_LESS                (1 << 1)
124 #define DEP_GREATER             (1 << 2)
125 #define DEP_EQUAL               (1 << 3)
126 #define DEP_STRONG              (1 << 27)
127 #define DEP_PRE                 ((1 << 6) | (1 << 9) | (1 << 10) | (1 << 11) | (1 << 12))
128
129 #define FILEFLAG_GHOST          (1 <<  6)
130
131
132 #ifdef RPM5
133 # define RPM_INDEX_SIZE 4
134 #else
135 # define RPM_INDEX_SIZE 8
136 #endif
137
138 struct rpmid {
139   unsigned int dbid;
140   char *name;
141 };
142
143 typedef struct rpmhead {
144   int cnt;
145   int dcnt;
146   unsigned char *dp;
147   unsigned char data[1];
148 } RpmHead;
149
150
151 static inline unsigned char *
152 headfindtag(RpmHead *h, int tag)
153 {
154   unsigned int i;
155   unsigned char *d, taga[4];
156   d = h->dp - 16;
157   taga[0] = tag >> 24;
158   taga[1] = tag >> 16;
159   taga[2] = tag >> 8;
160   taga[3] = tag;
161   for (i = 0; i < h->cnt; i++, d -= 16)
162     if (d[3] == taga[3] && d[2] == taga[2] && d[1] == taga[1] && d[0] == taga[0])
163       return d;
164   return 0;
165 }
166
167 static int
168 headexists(RpmHead *h, int tag)
169 {
170   return headfindtag(h, tag) ? 1 : 0;
171 }
172
173 static unsigned int *
174 headint32array(RpmHead *h, int tag, int *cnt)
175 {
176   unsigned int i, o, *r;
177   unsigned char *d = headfindtag(h, tag);
178
179   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 4)
180     return 0;
181   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
182   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
183   if (o + 4 * i > h->dcnt)
184     return 0;
185   d = h->dp + o;
186   r = solv_calloc(i ? i : 1, sizeof(unsigned int));
187   if (cnt)
188     *cnt = i;
189   for (o = 0; o < i; o++, d += 4)
190     r[o] = d[0] << 24 | d[1] << 16 | d[2] << 8 | d[3];
191   return r;
192 }
193
194 /* returns the first entry of an integer array */
195 static unsigned int
196 headint32(RpmHead *h, int tag)
197 {
198   unsigned int i, o;
199   unsigned char *d = headfindtag(h, tag);
200
201   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 4)
202     return 0;
203   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
204   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
205   if (i == 0 || o + 4 * i > h->dcnt)
206     return 0;
207   d = h->dp + o;
208   return d[0] << 24 | d[1] << 16 | d[2] << 8 | d[3];
209 }
210
211 /* returns the first entry of an 64bit integer array */
212 static unsigned long long
213 headint64(RpmHead *h, int tag)
214 {
215   unsigned int i, o;
216   unsigned char *d = headfindtag(h, tag);
217   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 5)
218     return 0;
219   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
220   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
221   if (i == 0 || o + 8 * i > h->dcnt)
222     return 0;
223   d = h->dp + o;
224   i = d[0] << 24 | d[1] << 16 | d[2] << 8 | d[3];
225   return (unsigned long long)i << 32 | (d[4] << 24 | d[5] << 16 | d[6] << 8 | d[7]);
226 }
227
228 static unsigned int *
229 headint16array(RpmHead *h, int tag, int *cnt)
230 {
231   unsigned int i, o, *r;
232   unsigned char *d = headfindtag(h, tag);
233
234   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 3)
235     return 0;
236   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
237   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
238   if (o + 4 * i > h->dcnt)
239     return 0;
240   d = h->dp + o;
241   r = solv_calloc(i ? i : 1, sizeof(unsigned int));
242   if (cnt)
243     *cnt = i;
244   for (o = 0; o < i; o++, d += 2)
245     r[o] = d[0] << 8 | d[1];
246   return r;
247 }
248
249 static char *
250 headstring(RpmHead *h, int tag)
251 {
252   unsigned int o;
253   unsigned char *d = headfindtag(h, tag);
254   /* 6: STRING, 9: I18NSTRING */
255   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || (d[7] != 6 && d[7] != 9))
256     return 0;
257   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
258   if (o >= h->dcnt)
259     return 0;
260   return (char *)h->dp + o;
261 }
262
263 static char **
264 headstringarray(RpmHead *h, int tag, int *cnt)
265 {
266   unsigned int i, o;
267   unsigned char *d = headfindtag(h, tag);
268   char **r;
269
270   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 8)
271     return 0;
272   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
273   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
274   r = solv_calloc(i ? i : 1, sizeof(char *));
275   if (cnt)
276     *cnt = i;
277   d = h->dp + o;
278   for (o = 0; o < i; o++)
279     {
280       r[o] = (char *)d;
281       if (o + 1 < i)
282         d += strlen((char *)d) + 1;
283       if (d >= h->dp + h->dcnt)
284         {
285           solv_free(r);
286           return 0;
287         }
288     }
289   return r;
290 }
291
292 static unsigned char *
293 headbinary(RpmHead *h, int tag, unsigned int *sizep)
294 {
295   unsigned int i, o;
296   unsigned char *d = headfindtag(h, tag);
297   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 7)
298     return 0;
299   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
300   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
301   if (o > h->dcnt || o + i < o || o + i > h->dcnt)
302     return 0;
303   if (sizep)
304     *sizep = i;
305   return h->dp + o;
306 }
307
308 static char *headtoevr(RpmHead *h)
309 {
310   unsigned int epoch;
311   char *version, *v;
312   char *release;
313   char *evr;
314   char *distepoch;
315
316   version  = headstring(h, TAG_VERSION);
317   release  = headstring(h, TAG_RELEASE);
318   epoch = headint32(h, TAG_EPOCH);
319   if (!version || !release)
320     {
321       fprintf(stderr, "headtoevr: bad rpm header\n");
322       return 0;
323     }
324   for (v = version; *v >= '0' && *v <= '9'; v++)
325     ;
326   if (epoch || (v != version && *v == ':'))
327     {
328       char epochbuf[11];        /* 32bit decimal will fit in */
329       sprintf(epochbuf, "%u", epoch);
330       evr = solv_malloc(strlen(epochbuf) + 1 + strlen(version) + 1 + strlen(release) + 1);
331       sprintf(evr, "%s:%s-%s", epochbuf, version, release);
332     }
333   else
334     {
335       evr = solv_malloc(strlen(version) + 1 + strlen(release) + 1);
336       sprintf(evr, "%s-%s", version, release);
337     }
338   distepoch = headstring(h, TAG_DISTEPOCH);
339   if (distepoch && *distepoch)
340     {
341       int l = strlen(evr);
342       evr = solv_realloc(evr, l + strlen(distepoch) + 2);
343       evr[l++] = ':';
344       strcpy(evr + l, distepoch);
345     }
346   return evr;
347 }
348
349
350 static void
351 setutf8string(Repodata *repodata, Id handle, Id tag, const char *str)
352 {
353   const unsigned char *cp;
354   int state = 0;
355   int c;
356   unsigned char *buf = 0, *bp;
357
358   /* check if it's already utf8, code taken from screen ;-) */
359   cp = (const unsigned char *)str;
360   while ((c = *cp++) != 0)
361     {
362       if (state)
363         {
364           if ((c & 0xc0) != 0x80)
365             break; /* encoding error */
366           c = (c & 0x3f) | (state << 6);
367           if (!(state & 0x40000000))
368             {
369               /* check for overlong sequences */
370               if ((c & 0x820823e0) == 0x80000000)
371                 c = 0xfdffffff;
372               else if ((c & 0x020821f0) == 0x02000000)
373                 c = 0xfff7ffff;
374               else if ((c & 0x000820f8) == 0x00080000)
375                 c = 0xffffd000;
376               else if ((c & 0x0000207c) == 0x00002000)
377                 c = 0xffffff70;
378             }
379         }
380       else
381         {
382           /* new sequence */
383           if (c >= 0xfe)
384             break;
385           else if (c >= 0xfc)
386             c = (c & 0x01) | 0xbffffffc;    /* 5 bytes to follow */
387           else if (c >= 0xf8)
388             c = (c & 0x03) | 0xbfffff00;    /* 4 */
389           else if (c >= 0xf0)
390             c = (c & 0x07) | 0xbfffc000;    /* 3 */
391           else if (c >= 0xe0)
392             c = (c & 0x0f) | 0xbff00000;    /* 2 */
393           else if (c >= 0xc2)
394             c = (c & 0x1f) | 0xfc000000;    /* 1 */
395           else if (c >= 0x80)
396             break;
397         }
398       state = (c & 0x80000000) ? c : 0;
399     }
400   if (c)
401     {
402       /* not utf8, assume latin1 */
403       buf = solv_malloc(2 * strlen(str) + 1);
404       cp = (const unsigned char *)str;
405       str = (char *)buf;
406       bp = buf;
407       while ((c = *cp++) != 0)
408         {
409           if (c >= 0xc0)
410             {
411               *bp++ = 0xc3;
412               c ^= 0x80;
413             }
414           else if (c >= 0x80)
415             *bp++ = 0xc2;
416           *bp++ = c;
417         }
418       *bp++ = 0;
419     }
420   repodata_set_str(repodata, handle, tag, str);
421   if (buf)
422     solv_free(buf);
423 }
424
425
426 #define MAKEDEPS_FILTER_WEAK    (1 << 0)
427 #define MAKEDEPS_FILTER_STRONG  (1 << 1)
428 #define MAKEDEPS_NO_RPMLIB      (1 << 2)
429
430 /*
431  * strong: 0: ignore strongness
432  *         1: filter to strong
433  *         2: filter to weak
434  */
435 static unsigned int
436 makedeps(Pool *pool, Repo *repo, RpmHead *rpmhead, int tagn, int tagv, int tagf, int flags)
437 {
438   char **n, **v;
439   unsigned int *f;
440   int i, cc, nc, vc, fc;
441   int haspre;
442   unsigned int olddeps;
443   Id *ida;
444   int strong;
445
446   strong = flags & (MAKEDEPS_FILTER_STRONG|MAKEDEPS_FILTER_WEAK);
447   n = headstringarray(rpmhead, tagn, &nc);
448   if (!n)
449     return 0;
450   v = headstringarray(rpmhead, tagv, &vc);
451   if (!v)
452     {
453       solv_free(n);
454       return 0;
455     }
456   f = headint32array(rpmhead, tagf, &fc);
457   if (!f)
458     {
459       solv_free(n);
460       free(v);
461       return 0;
462     }
463   if (nc != vc || nc != fc)
464     {
465       fprintf(stderr, "bad dependency entries\n");
466       return 0;
467     }
468
469   cc = nc;
470   haspre = 0;   /* add no prereq marker */
471   if (flags)
472     {
473       /* we do filtering */
474       cc = 0;
475       for (i = 0; i < nc; i++)
476         {
477           if (strong && (f[i] & DEP_STRONG) != (strong == MAKEDEPS_FILTER_WEAK ? 0 : DEP_STRONG))
478             continue;
479           if ((flags & MAKEDEPS_NO_RPMLIB) != 0)
480             if (!strncmp(n[i], "rpmlib(", 7))
481               continue;
482           if ((f[i] & DEP_PRE) != 0)
483             haspre = 1;
484           cc++;
485         }
486     }
487   else if (tagn == TAG_REQUIRENAME)
488     {
489       /* no filtering, just look for the first prereq */
490       for (i = 0; i < nc; i++)
491         if ((f[i] & DEP_PRE) != 0)
492           {
493             haspre = 1;
494             break;
495           }
496     }
497   if (cc == 0)
498     {
499       solv_free(n);
500       solv_free(v);
501       solv_free(f);
502       return 0;
503     }
504   cc += haspre;
505   olddeps = repo_reserve_ids(repo, 0, cc);
506   ida = repo->idarraydata + olddeps;
507   for (i = 0; ; i++)
508     {
509       if (i == nc)
510         {
511           if (haspre != 1)
512             break;
513           haspre = 2;   /* pass two: prereqs */
514           i = 0;
515           *ida++ = SOLVABLE_PREREQMARKER;
516         }
517       if (strong && (f[i] & DEP_STRONG) != (strong == MAKEDEPS_FILTER_WEAK ? 0 : DEP_STRONG))
518         continue;
519       if (haspre == 1 && (f[i] & DEP_PRE) != 0)
520         continue;
521       if (haspre == 2 && (f[i] & DEP_PRE) == 0)
522         continue;
523       if ((flags & MAKEDEPS_NO_RPMLIB) != 0)
524         if (!strncmp(n[i], "rpmlib(", 7))
525           continue;
526       if (f[i] & (DEP_LESS|DEP_GREATER|DEP_EQUAL))
527         {
528           Id name, evr;
529           int flags = 0;
530           if ((f[i] & DEP_LESS) != 0)
531             flags |= 4;
532           if ((f[i] & DEP_EQUAL) != 0)
533             flags |= 2;
534           if ((f[i] & DEP_GREATER) != 0)
535             flags |= 1;
536           name = pool_str2id(pool, n[i], 1);
537           if (v[i][0] == '0' && v[i][1] == ':' && v[i][2])
538             evr = pool_str2id(pool, v[i] + 2, 1);
539           else
540             evr = pool_str2id(pool, v[i], 1);
541           *ida++ = pool_rel2id(pool, name, evr, flags, 1);
542         }
543       else
544         *ida++ = pool_str2id(pool, n[i], 1);
545     }
546   *ida++ = 0;
547   repo->idarraysize += cc + 1;
548   solv_free(n);
549   solv_free(v);
550   solv_free(f);
551   return olddeps;
552 }
553
554
555 #ifdef USE_FILEFILTER
556
557 #define FILEFILTER_EXACT    0
558 #define FILEFILTER_STARTS   1
559 #define FILEFILTER_CONTAINS 2
560
561 struct filefilter {
562   int dirmatch;
563   char *dir;
564   char *base;
565 };
566
567 static struct filefilter filefilters[] = {
568   { FILEFILTER_CONTAINS, "/bin/", 0},
569   { FILEFILTER_CONTAINS, "/sbin/", 0},
570   { FILEFILTER_CONTAINS, "/lib/", 0},
571   { FILEFILTER_CONTAINS, "/lib64/", 0},
572   { FILEFILTER_CONTAINS, "/etc/", 0},
573   { FILEFILTER_STARTS, "/usr/games/", 0},
574   { FILEFILTER_EXACT, "/usr/share/dict/", "words"},
575   { FILEFILTER_STARTS, "/usr/share/", "magic.mime"},
576   { FILEFILTER_STARTS, "/opt/gnome/games/", 0},
577 };
578
579 #endif
580
581 static void
582 adddudata(Pool *pool, Repo *repo, Repodata *data, Solvable *s, RpmHead *rpmhead, char **dn, unsigned int *di, int fc, int dc)
583 {
584   Id handle, did;
585   int i, fszc;
586   unsigned int *fkb, *fn, *fsz, *fm, *fino;
587   unsigned int inotest[256], inotestok;
588
589   if (!fc)
590     return;
591   /* XXX: use TAG_LONGFILESIZES if available */
592   fsz = headint32array(rpmhead, TAG_FILESIZES, &fszc);
593   if (!fsz || fc != fszc)
594     {
595       solv_free(fsz);
596       return;
597     }
598   /* stupid rpm records sizes of directories, so we have to check the mode */
599   fm = headint16array(rpmhead, TAG_FILEMODES, &fszc);
600   if (!fm || fc != fszc)
601     {
602       solv_free(fsz);
603       solv_free(fm);
604       return;
605     }
606   fino = headint32array(rpmhead, TAG_FILEINODES, &fszc);
607   if (!fino || fc != fszc)
608     {
609       solv_free(fsz);
610       solv_free(fm);
611       solv_free(fino);
612       return;
613     }
614   inotestok = 0;
615   if (fc < sizeof(inotest))
616     {
617       memset(inotest, 0, sizeof(inotest));
618       for (i = 0; i < fc; i++)
619         {
620           int off, bit;
621           if (fsz[i] == 0 || !S_ISREG(fm[i]))
622             continue;
623           off = (fino[i] >> 5) & (sizeof(inotest)/sizeof(*inotest) - 1);
624           bit = 1 << (fino[i] & 31);
625           if ((inotest[off] & bit) != 0)
626             break;
627           inotest[off] |= bit;
628         }
629       if (i == fc)
630         inotestok = 1;
631     }
632   if (!inotestok)
633     {
634       unsigned int *fdev = headint32array(rpmhead, TAG_FILEDEVICES, &fszc);
635       unsigned int *fx, j;
636       unsigned int mask, hash, hh;
637       if (!fdev || fc != fszc)
638         {
639           solv_free(fsz);
640           solv_free(fm);
641           solv_free(fdev);
642           solv_free(fino);
643           return;
644         }
645       mask = fc;
646       while ((mask & (mask - 1)) != 0)
647         mask = mask & (mask - 1);
648       mask <<= 2;
649       if (mask > sizeof(inotest)/sizeof(*inotest))
650         fx = solv_calloc(mask, sizeof(unsigned int));
651       else
652         {
653           fx = inotest;
654           memset(fx, 0, mask * sizeof(unsigned int));
655         }
656       mask--;
657       for (i = 0; i < fc; i++)
658         {
659           if (fsz[i] == 0 || !S_ISREG(fm[i]))
660             continue;
661           hash = (fino[i] + fdev[i] * 31) & mask;
662           hh = 7;
663           while ((j = fx[hash]) != 0)
664             {
665               if (fino[j - 1] == fino[i] && fdev[j - 1] == fdev[i])
666                 {
667                   fsz[i] = 0;   /* kill entry */
668                   break;
669                 }
670               hash = (hash + hh++) & mask;
671             }
672           if (!j)
673             fx[hash] = i + 1;
674         }
675       if (fx != inotest)
676         solv_free(fx);
677       solv_free(fdev);
678     }
679   solv_free(fino);
680   fn = solv_calloc(dc, sizeof(unsigned int));
681   fkb = solv_calloc(dc, sizeof(unsigned int));
682   for (i = 0; i < fc; i++)
683     {
684       if (di[i] >= dc)
685         continue;
686       fn[di[i]]++;
687       if (fsz[i] == 0 || !S_ISREG(fm[i]))
688         continue;
689       fkb[di[i]] += fsz[i] / 1024 + 1;
690     }
691   solv_free(fsz);
692   solv_free(fm);
693   /* commit */
694   handle = s - pool->solvables;
695   for (i = 0; i < dc; i++)
696     {
697       if (!fn[i])
698         continue;
699       if (!*dn[i])
700         {
701           if (s->arch == ARCH_SRC || s->arch == ARCH_NOSRC)
702             did = repodata_str2dir(data, "/usr/src", 1);
703           else
704             continue;   /* work around rpm bug */
705         }
706       else
707         did = repodata_str2dir(data, dn[i], 1);
708       repodata_add_dirnumnum(data, handle, SOLVABLE_DISKUSAGE, did, fkb[i], fn[i]);
709     }
710   solv_free(fn);
711   solv_free(fkb);
712 }
713
714 /* assumes last processed array is provides! */
715 static unsigned int
716 addfileprovides(Pool *pool, Repo *repo, Repodata *data, Solvable *s, RpmHead *rpmhead, unsigned int olddeps)
717 {
718   char **bn;
719   char **dn;
720   unsigned int *di;
721   int bnc, dnc, dic;
722   int i;
723 #ifdef USE_FILEFILTER
724   int j;
725   struct filefilter *ff;
726 #endif
727 #if 0
728   char *fn = 0;
729   int fna = 0;
730 #endif
731
732   if (!data)
733     return olddeps;
734   bn = headstringarray(rpmhead, TAG_BASENAMES, &bnc);
735   if (!bn)
736     return olddeps;
737   dn = headstringarray(rpmhead, TAG_DIRNAMES, &dnc);
738   if (!dn)
739     {
740       solv_free(bn);
741       return olddeps;
742     }
743   di = headint32array(rpmhead, TAG_DIRINDEXES, &dic);
744   if (!di)
745     {
746       solv_free(bn);
747       solv_free(dn);
748       return olddeps;
749     }
750   if (bnc != dic)
751     {
752       fprintf(stderr, "bad filelist\n");
753       return olddeps;
754     }
755
756   if (data)
757     adddudata(pool, repo, data, s, rpmhead, dn, di, bnc, dnc);
758
759   for (i = 0; i < bnc; i++)
760     {
761 #ifdef USE_FILEFILTER
762       ff = filefilters;
763       for (j = 0; j < sizeof(filefilters)/sizeof(*filefilters); j++, ff++)
764         {
765           if (ff->dir)
766             {
767               switch (ff->dirmatch)
768                 {
769                 case FILEFILTER_STARTS:
770                   if (strncmp(dn[di[i]], ff->dir, strlen(ff->dir)))
771                     continue;
772                   break;
773                 case FILEFILTER_CONTAINS:
774                   if (!strstr(dn[di[i]], ff->dir))
775                     continue;
776                   break;
777                 case FILEFILTER_EXACT:
778                 default:
779                   if (strcmp(dn[di[i]], ff->dir))
780                     continue;
781                   break;
782                 }
783             }
784           if (ff->base)
785             {
786               if (strcmp(bn[i], ff->base))
787                 continue;
788             }
789           break;
790         }
791       if (j == sizeof(filefilters)/sizeof(*filefilters))
792         continue;
793 #endif
794 #if 0
795       j = strlen(bn[i]) + strlen(dn[di[i]]) + 1;
796       if (j > fna)
797         {
798           fna = j + 256;
799           fn = solv_realloc(fn, fna);
800         }
801       strcpy(fn, dn[di[i]]);
802       strcat(fn, bn[i]);
803       olddeps = repo_addid_dep(repo, olddeps, pool_str2id(pool, fn, 1), SOLVABLE_FILEMARKER);
804 #endif
805       if (data)
806         {
807           Id handle, did;
808           char *b = bn[i];
809
810           handle = s - pool->solvables;
811           did = repodata_str2dir(data, dn[di[i]], 1);
812           if (!did)
813             {
814               did = repodata_str2dir(data, "/", 1);
815               if (b && b[0] == '/')
816                 b++;    /* work around rpm bug */
817             }
818           repodata_add_dirstr(data, handle, SOLVABLE_FILELIST, did, b);
819         }
820     }
821 #if 0
822   if (fn)
823     solv_free(fn);
824 #endif
825   solv_free(bn);
826   solv_free(dn);
827   solv_free(di);
828   return olddeps;
829 }
830
831 static void
832 addsourcerpm(Pool *pool, Repodata *data, Id handle, char *sourcerpm, char *name, char *evr)
833 {
834   const char *p, *sevr, *sarch;
835
836   p = strrchr(sourcerpm, '.');
837   if (!p || strcmp(p, ".rpm") != 0)
838     return;
839   p--;
840   while (p > sourcerpm && *p != '.')
841     p--;
842   if (*p != '.' || p == sourcerpm)
843     return;
844   sarch = p-- + 1;
845   while (p > sourcerpm && *p != '-')
846     p--;
847   if (*p != '-' || p == sourcerpm)
848     return;
849   p--;
850   while (p > sourcerpm && *p != '-')
851     p--;
852   if (*p != '-' || p == sourcerpm)
853     return;
854   sevr = p + 1;
855   if (!strcmp(sarch, "src.rpm"))
856     repodata_set_constantid(data, handle, SOLVABLE_SOURCEARCH, ARCH_SRC);
857   else if (!strcmp(sarch, "nosrc.rpm"))
858     repodata_set_constantid(data, handle, SOLVABLE_SOURCEARCH, ARCH_NOSRC);
859   else
860     repodata_set_constantid(data, handle, SOLVABLE_SOURCEARCH, pool_strn2id(pool, sarch, strlen(sarch) - 4, 1));
861   if (evr && !strncmp(sevr, evr, sarch - sevr - 1) && evr[sarch - sevr - 1] == 0)
862     repodata_set_void(data, handle, SOLVABLE_SOURCEEVR);
863   else
864     repodata_set_id(data, handle, SOLVABLE_SOURCEEVR, pool_strn2id(pool, sevr, sarch - sevr - 1, 1));
865   if (name && !strncmp(sourcerpm, name, sevr - sourcerpm - 1) && name[sevr - sourcerpm - 1] == 0)
866     repodata_set_void(data, handle, SOLVABLE_SOURCENAME);
867   else
868     repodata_set_id(data, handle, SOLVABLE_SOURCENAME, pool_strn2id(pool, sourcerpm, sevr - sourcerpm - 1, 1));
869 }
870
871 static int
872 rpm2solv(Pool *pool, Repo *repo, Repodata *data, Solvable *s, RpmHead *rpmhead, int flags)
873 {
874   char *name;
875   char *evr;
876   char *sourcerpm;
877
878   name = headstring(rpmhead, TAG_NAME);
879   if (!strcmp(name, "gpg-pubkey"))
880     return 0;
881   s->name = pool_str2id(pool, name, 1);
882   if (!s->name)
883     {
884       fprintf(stderr, "package has no name\n");
885       return 0;
886     }
887   sourcerpm = headstring(rpmhead, TAG_SOURCERPM);
888   if (sourcerpm)
889     s->arch = pool_str2id(pool, headstring(rpmhead, TAG_ARCH), 1);
890   else
891     {
892       if (headexists(rpmhead, TAG_NOSOURCE) || headexists(rpmhead, TAG_NOPATCH))
893         s->arch = ARCH_NOSRC;
894       else
895         s->arch = ARCH_SRC;
896     }
897   if (!s->arch)
898     s->arch = ARCH_NOARCH;
899   evr = headtoevr(rpmhead);
900   s->evr = pool_str2id(pool, evr, 1);
901   s->vendor = pool_str2id(pool, headstring(rpmhead, TAG_VENDOR), 1);
902
903   s->provides = makedeps(pool, repo, rpmhead, TAG_PROVIDENAME, TAG_PROVIDEVERSION, TAG_PROVIDEFLAGS, 0);
904   if ((flags & RPM_ADD_NO_FILELIST) == 0)
905     s->provides = addfileprovides(pool, repo, data, s, rpmhead, s->provides);
906   if (s->arch != ARCH_SRC && s->arch != ARCH_NOSRC)
907     s->provides = repo_addid_dep(repo, s->provides, pool_rel2id(pool, s->name, s->evr, REL_EQ, 1), 0);
908   s->requires = makedeps(pool, repo, rpmhead, TAG_REQUIRENAME, TAG_REQUIREVERSION, TAG_REQUIREFLAGS, (flags & RPM_ADD_NO_RPMLIBREQS) ? MAKEDEPS_NO_RPMLIB : 0);
909   s->conflicts = makedeps(pool, repo, rpmhead, TAG_CONFLICTNAME, TAG_CONFLICTVERSION, TAG_CONFLICTFLAGS, 0);
910   s->obsoletes = makedeps(pool, repo, rpmhead, TAG_OBSOLETENAME, TAG_OBSOLETEVERSION, TAG_OBSOLETEFLAGS, 0);
911
912   s->recommends = makedeps(pool, repo, rpmhead, TAG_SUGGESTSNAME, TAG_SUGGESTSVERSION, TAG_SUGGESTSFLAGS, MAKEDEPS_FILTER_STRONG);
913   s->suggests = makedeps(pool, repo, rpmhead, TAG_SUGGESTSNAME, TAG_SUGGESTSVERSION, TAG_SUGGESTSFLAGS, MAKEDEPS_FILTER_WEAK);
914   s->supplements = makedeps(pool, repo, rpmhead, TAG_ENHANCESNAME, TAG_ENHANCESVERSION, TAG_ENHANCESFLAGS, MAKEDEPS_FILTER_STRONG);
915   s->enhances  = makedeps(pool, repo, rpmhead, TAG_ENHANCESNAME, TAG_ENHANCESVERSION, TAG_ENHANCESFLAGS, MAKEDEPS_FILTER_WEAK);
916   s->supplements = repo_fix_supplements(repo, s->provides, s->supplements, 0);
917   s->conflicts = repo_fix_conflicts(repo, s->conflicts);
918
919   if (data)
920     {
921       Id handle;
922       char *str;
923       unsigned int u32;
924       unsigned long long u64;
925
926       handle = s - pool->solvables;
927       str = headstring(rpmhead, TAG_SUMMARY);
928       if (str)
929         setutf8string(data, handle, SOLVABLE_SUMMARY, str);
930       str = headstring(rpmhead, TAG_DESCRIPTION);
931       if (str)
932         {
933           char *aut, *p;
934           for (aut = str; (aut = strchr(aut, '\n')) != 0; aut++)
935             if (!strncmp(aut, "\nAuthors:\n--------\n", 19))
936               break;
937           if (aut)
938             {
939               /* oh my, found SUSE special author section */
940               int l = aut - str;
941               str = solv_strdup(str);
942               aut = str + l;
943               str[l] = 0;
944               while (l > 0 && str[l - 1] == '\n')
945                 str[--l] = 0;
946               if (l)
947                 setutf8string(data, handle, SOLVABLE_DESCRIPTION, str);
948               p = aut + 19;
949               aut = str;        /* copy over */
950               while (*p == ' ' || *p == '\n')
951                 p++;
952               while (*p)
953                 {
954                   if (*p == '\n')
955                     {
956                       *aut++ = *p++;
957                       while (*p == ' ')
958                         p++;
959                       continue;
960                     }
961                   *aut++ = *p++;
962                 }
963               while (aut != str && aut[-1] == '\n')
964                 aut--;
965               *aut = 0;
966               if (*str)
967                 setutf8string(data, handle, SOLVABLE_AUTHORS, str);
968               free(str);
969             }
970           else if (*str)
971             setutf8string(data, handle, SOLVABLE_DESCRIPTION, str);
972         }
973       str = headstring(rpmhead, TAG_GROUP);
974       if (str)
975         repodata_set_poolstr(data, handle, SOLVABLE_GROUP, str);
976       str = headstring(rpmhead, TAG_LICENSE);
977       if (str)
978         repodata_set_poolstr(data, handle, SOLVABLE_LICENSE, str);
979       str = headstring(rpmhead, TAG_URL);
980       if (str)
981         repodata_set_str(data, handle, SOLVABLE_URL, str);
982       str = headstring(rpmhead, TAG_DISTRIBUTION);
983       if (str)
984         repodata_set_poolstr(data, handle, SOLVABLE_DISTRIBUTION, str);
985       str = headstring(rpmhead, TAG_PACKAGER);
986       if (str)
987         repodata_set_poolstr(data, handle, SOLVABLE_PACKAGER, str);
988       if ((flags & RPM_ADD_WITH_HDRID) != 0)
989         {
990           str = headstring(rpmhead, TAG_SHA1HEADER);
991           if (str)
992             repodata_set_poolstr(data, handle, SOLVABLE_HDRID, str);
993         }
994       u32 = headint32(rpmhead, TAG_BUILDTIME);
995       if (u32)
996         repodata_set_num(data, handle, SOLVABLE_BUILDTIME, u32);
997       u32 = headint32(rpmhead, TAG_INSTALLTIME);
998       if (u32)
999         repodata_set_num(data, handle, SOLVABLE_INSTALLTIME, u32);
1000       u64 = headint64(rpmhead, TAG_LONGSIZE);
1001       if (u64)
1002         repodata_set_num(data, handle, SOLVABLE_INSTALLSIZE, u64);
1003       else
1004         {
1005           u32 = headint32(rpmhead, TAG_SIZE);
1006           if (u32)
1007             repodata_set_num(data, handle, SOLVABLE_INSTALLSIZE, u32);
1008         }
1009       if (sourcerpm)
1010         addsourcerpm(pool, data, handle, sourcerpm, name, evr);
1011       if ((flags & RPM_ADD_TRIGGERS) != 0)
1012         {
1013           Id id, lastid;
1014           unsigned int ida = makedeps(pool, repo, rpmhead, TAG_TRIGGERNAME, TAG_TRIGGERVERSION, TAG_TRIGGERFLAGS, 0);
1015
1016           lastid = 0;
1017           for (; (id = repo->idarraydata[ida]) != 0; ida++)
1018             {
1019               /* we currently do not support rel ids in incore data, so
1020                * strip off versioning information */
1021               while (ISRELDEP(id))
1022                 {
1023                   Reldep *rd = GETRELDEP(pool, id);
1024                   id = rd->name;
1025                 }
1026               if (id == lastid)
1027                 continue;
1028               repodata_add_idarray(data, handle, SOLVABLE_TRIGGERS, id);
1029               lastid = id;
1030             }
1031         }
1032     }
1033   solv_free(evr);
1034   return 1;
1035 }
1036
1037 static Id
1038 copyreldep(Pool *pool, Pool *frompool, Id id)
1039 {
1040   Reldep *rd = GETRELDEP(frompool, id);
1041   Id name = rd->name, evr = rd->evr;
1042   if (ISRELDEP(name))
1043     name = copyreldep(pool, frompool, name);
1044   else
1045     name = pool_str2id(pool, pool_id2str(frompool, name), 1);
1046   if (ISRELDEP(evr))
1047     evr = copyreldep(pool, frompool, evr);
1048   else
1049     evr = pool_str2id(pool, pool_id2str(frompool, evr), 1);
1050   return pool_rel2id(pool, name, evr, rd->flags, 1);
1051 }
1052
1053 static Offset
1054 copydeps(Pool *pool, Repo *repo, Offset fromoff, Repo *fromrepo)
1055 {
1056   int cc;
1057   Id id, *ida, *from;
1058   Offset ido;
1059   Pool *frompool = fromrepo->pool;
1060
1061   if (!fromoff)
1062     return 0;
1063   from = fromrepo->idarraydata + fromoff;
1064   for (ida = from, cc = 0; *ida; ida++, cc++)
1065     ;
1066   if (cc == 0)
1067     return 0;
1068   ido = repo_reserve_ids(repo, 0, cc);
1069   ida = repo->idarraydata + ido;
1070   if (frompool && pool != frompool)
1071     {
1072       while (*from)
1073         {
1074           id = *from++;
1075           if (ISRELDEP(id))
1076             id = copyreldep(pool, frompool, id);
1077           else
1078             id = pool_str2id(pool, pool_id2str(frompool, id), 1);
1079           *ida++ = id;
1080         }
1081       *ida = 0;
1082     }
1083   else
1084     memcpy(ida, from, (cc + 1) * sizeof(Id));
1085   repo->idarraysize += cc + 1;
1086   return ido;
1087 }
1088
1089 #define COPYDIR_DIRCACHE_SIZE 512
1090
1091 static Id copydir_complex(Pool *pool, Repodata *data, Stringpool *fromspool, Repodata *fromdata, Id did, Id *cache);
1092
1093 static inline Id
1094 copydir(Pool *pool, Repodata *data, Stringpool *fromspool, Repodata *fromdata, Id did, Id *cache)
1095 {
1096   if (cache && cache[did & 255] == did)
1097     return cache[(did & 255) + 256];
1098   return copydir_complex(pool, data, fromspool, fromdata, did, cache);
1099 }
1100
1101 static Id
1102 copydir_complex(Pool *pool, Repodata *data, Stringpool *fromspool, Repodata *fromdata, Id did, Id *cache)
1103 {
1104   Id parent = dirpool_parent(&fromdata->dirpool, did);
1105   Id compid = dirpool_compid(&fromdata->dirpool, did);
1106   if (parent)
1107     parent = copydir(pool, data, fromspool, fromdata, parent, cache);
1108   if (fromspool != &pool->ss)
1109     compid = pool_str2id(pool, stringpool_id2str(fromspool, compid), 1);
1110   compid = dirpool_add_dir(&data->dirpool, parent, compid, 1);
1111   if (cache)
1112     {
1113       cache[did & 255] = did;
1114       cache[(did & 255) + 256] = compid;
1115     }
1116   return compid;
1117 }
1118
1119 struct solvable_copy_cbdata {
1120   Repodata *data;
1121   Id handle;
1122   Id *dircache;
1123 };
1124
1125 static int
1126 solvable_copy_cb(void *vcbdata, Solvable *r, Repodata *fromdata, Repokey *key, KeyValue *kv)
1127 {
1128   struct solvable_copy_cbdata *cbdata = vcbdata;
1129   Id id, keyname;
1130   Repodata *data = cbdata->data;
1131   Id handle = cbdata->handle;
1132   Pool *pool = data->repo->pool, *frompool = fromdata->repo->pool;
1133   Stringpool *fromspool = fromdata->localpool ? &fromdata->spool : &frompool->ss;
1134
1135   keyname = key->name;
1136   if (keyname >= ID_NUM_INTERNAL && pool != frompool)
1137     keyname = pool_str2id(pool, pool_id2str(frompool, keyname), 1);
1138   switch(key->type)
1139     {
1140     case REPOKEY_TYPE_ID:
1141     case REPOKEY_TYPE_CONSTANTID:
1142     case REPOKEY_TYPE_IDARRAY:  /* used for triggers */
1143       id = kv->id;
1144       assert(!data->localpool); /* implement me! */
1145       if (pool != frompool || fromdata->localpool)
1146         {
1147           if (ISRELDEP(id))
1148             id = copyreldep(pool, frompool, id);
1149           else
1150             id = pool_str2id(pool, stringpool_id2str(fromspool, id), 1);
1151         }
1152       if (key->type == REPOKEY_TYPE_ID)
1153         repodata_set_id(data, handle, keyname, id);
1154       else if (key->type == REPOKEY_TYPE_CONSTANTID)
1155         repodata_set_constantid(data, handle, keyname, id);
1156       else
1157         repodata_add_idarray(data, handle, keyname, id);
1158       break;
1159     case REPOKEY_TYPE_STR:
1160       repodata_set_str(data, handle, keyname, kv->str);
1161       break;
1162     case REPOKEY_TYPE_VOID:
1163       repodata_set_void(data, handle, keyname);
1164       break;
1165     case REPOKEY_TYPE_NUM:
1166       repodata_set_num(data, handle, keyname, SOLV_KV_NUM64(kv));
1167       break;
1168     case REPOKEY_TYPE_CONSTANT:
1169       repodata_set_constant(data, handle, keyname, kv->num);
1170       break;
1171     case REPOKEY_TYPE_DIRNUMNUMARRAY:
1172       id = kv->id;
1173       assert(!data->localpool); /* implement me! */
1174       id = copydir(pool, data, fromspool, fromdata, id, cbdata->dircache);
1175       repodata_add_dirnumnum(data, handle, keyname, id, kv->num, kv->num2);
1176       break;
1177     case REPOKEY_TYPE_DIRSTRARRAY:
1178       id = kv->id;
1179       assert(!data->localpool); /* implement me! */
1180       id = copydir(pool, data, fromspool, fromdata, id, cbdata->dircache);
1181       repodata_add_dirstr(data, handle, keyname, id, kv->str);
1182       break;
1183     default:
1184       break;
1185     }
1186   return 0;
1187 }
1188
1189 static void
1190 solvable_copy(Solvable *s, Solvable *r, Repodata *data, Id *dircache)
1191 {
1192   Repo *repo = s->repo;
1193   Repo *fromrepo = r->repo;
1194   Pool *pool = repo->pool;
1195   struct solvable_copy_cbdata cbdata;
1196
1197   /* copy solvable data */
1198   if (pool == fromrepo->pool)
1199     {
1200       s->name = r->name;
1201       s->evr = r->evr;
1202       s->arch = r->arch;
1203       s->vendor = r->vendor;
1204     }
1205   else
1206     {
1207       if (r->name)
1208         s->name = pool_str2id(pool, pool_id2str(fromrepo->pool, r->name), 1);
1209       if (r->evr)
1210         s->evr = pool_str2id(pool, pool_id2str(fromrepo->pool, r->evr), 1);
1211       if (r->arch)
1212         s->arch = pool_str2id(pool, pool_id2str(fromrepo->pool, r->arch), 1);
1213       if (r->vendor)
1214         s->vendor = pool_str2id(pool, pool_id2str(fromrepo->pool, r->vendor), 1);
1215     }
1216   s->provides = copydeps(pool, repo, r->provides, fromrepo);
1217   s->requires = copydeps(pool, repo, r->requires, fromrepo);
1218   s->conflicts = copydeps(pool, repo, r->conflicts, fromrepo);
1219   s->obsoletes = copydeps(pool, repo, r->obsoletes, fromrepo);
1220   s->recommends = copydeps(pool, repo, r->recommends, fromrepo);
1221   s->suggests = copydeps(pool, repo, r->suggests, fromrepo);
1222   s->supplements = copydeps(pool, repo, r->supplements, fromrepo);
1223   s->enhances  = copydeps(pool, repo, r->enhances, fromrepo);
1224
1225   /* copy all attributes */
1226   if (!data)
1227     return;
1228   cbdata.data = data;
1229   cbdata.handle = s - pool->solvables;
1230   cbdata.dircache = dircache;
1231   repo_search(fromrepo, (r - fromrepo->pool->solvables), 0, 0, SEARCH_NO_STORAGE_SOLVABLE, solvable_copy_cb, &cbdata);
1232 }
1233
1234 /* used to sort entries returned in some database order */
1235 static int
1236 rpmids_sort_cmp(const void *va, const void *vb, void *dp)
1237 {
1238   struct rpmid const *a = va, *b = vb;
1239   int r;
1240   r = strcmp(a->name, b->name);
1241   if (r)
1242     return r;
1243   return a->dbid - b->dbid;
1244 }
1245
1246 static int
1247 pkgids_sort_cmp(const void *va, const void *vb, void *dp)
1248 {
1249   Repo *repo = dp;
1250   Pool *pool = repo->pool;
1251   Solvable *a = pool->solvables + *(Id *)va;
1252   Solvable *b = pool->solvables + *(Id *)vb;
1253   Id *rpmdbid;
1254
1255   if (a->name != b->name)
1256     return strcmp(pool_id2str(pool, a->name), pool_id2str(pool, b->name));
1257   rpmdbid = repo->rpmdbid;
1258   return rpmdbid[(a - pool->solvables) - repo->start] - rpmdbid[(b - pool->solvables) - repo->start];
1259 }
1260
1261 static void
1262 swap_solvables(Repo *repo, Repodata *data, Id pa, Id pb)
1263 {
1264   Pool *pool = repo->pool;
1265   Solvable tmp;
1266
1267   tmp = pool->solvables[pa];
1268   pool->solvables[pa] = pool->solvables[pb];
1269   pool->solvables[pb] = tmp;
1270   if (repo->rpmdbid)
1271     {
1272       Id tmpid = repo->rpmdbid[pa - repo->start];
1273       repo->rpmdbid[pa - repo->start] = repo->rpmdbid[pb - repo->start];
1274       repo->rpmdbid[pb - repo->start] = tmpid;
1275     }
1276   /* only works if nothing is already internalized! */
1277   if (data)
1278     repodata_swap_attrs(data, pa, pb);
1279 }
1280
1281
1282 static inline Id db2rpmdbid(unsigned char *db, int byteswapped)
1283 {
1284 #ifdef RPM5
1285   return db[0] << 24 | db[1] << 16 | db[2] << 8 | db[3];
1286 #else
1287 # if defined(WORDS_BIGENDIAN)
1288   if (!byteswapped)
1289 # else
1290   if (byteswapped)
1291 # endif
1292     return db[0] << 24 | db[1] << 16 | db[2] << 8 | db[3];
1293   else
1294     return db[3] << 24 | db[2] << 16 | db[1] << 8 | db[0];
1295 #endif
1296 }
1297
1298 static inline void rpmdbid2db(unsigned char *db, Id id, int byteswapped)
1299 {
1300 #ifdef RPM5
1301   db[0] = id >> 24, db[1] = id >> 16, db[2] = id >> 8, db[3] = id;
1302 #else
1303 # if defined(WORDS_BIGENDIAN)
1304   if (!byteswapped)
1305 # else
1306   if (byteswapped)
1307 # endif
1308     db[0] = id >> 24, db[1] = id >> 16, db[2] = id >> 8, db[3] = id;
1309   else
1310     db[3] = id >> 24, db[2] = id >> 16, db[1] = id >> 8, db[0] = id;
1311 #endif
1312 }
1313
1314 static void
1315 mkrpmdbcookie(struct stat *st, unsigned char *cookie)
1316 {
1317   memset(cookie, 0, 32);
1318   cookie[3] = RPMDB_COOKIE_VERSION;
1319   memcpy(cookie + 16, &st->st_ino, sizeof(st->st_ino));
1320   memcpy(cookie + 24, &st->st_dev, sizeof(st->st_dev));
1321 }
1322
1323 /* should look in /usr/lib/rpm/macros instead, but we want speed... */
1324 static DB_ENV *
1325 opendbenv(const char *rootdir)
1326 {
1327   char dbpath[PATH_MAX];
1328   DB_ENV *dbenv = 0;
1329   int r;
1330
1331   if (db_env_create(&dbenv, 0))
1332     {
1333       perror("db_env_create");
1334       return 0;
1335     }
1336 #if defined(FEDORA) && (DB_VERSION_MAJOR >= 5 || (DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 5))
1337   dbenv->set_thread_count(dbenv, 8);
1338 #endif
1339   snprintf(dbpath, PATH_MAX, "%s/var/lib/rpm", rootdir ? rootdir : "");
1340   if (access(dbpath, W_OK) == -1)
1341     {
1342       r = dbenv->open(dbenv, dbpath, DB_CREATE|DB_PRIVATE|DB_INIT_MPOOL, 0);
1343     }
1344   else
1345     {
1346 #ifdef FEDORA
1347       r = dbenv->open(dbenv, dbpath, DB_CREATE|DB_INIT_CDB|DB_INIT_MPOOL, 0644);
1348 #else
1349       r = dbenv->open(dbenv, dbpath, DB_CREATE|DB_PRIVATE|DB_INIT_MPOOL, 0);
1350 #endif
1351     }
1352   if (r)
1353     {
1354       perror("dbenv open");
1355       dbenv->close(dbenv, 0);
1356       return 0;
1357     }
1358   return dbenv;
1359 }
1360
1361
1362 static int
1363 count_headers(const char *rootdir, DB_ENV *dbenv)
1364 {
1365   char dbpath[PATH_MAX];
1366   struct stat statbuf;
1367   DB *db = 0;
1368   DBC *dbc = 0;
1369   int count = 0;
1370   DBT dbkey;
1371   DBT dbdata;
1372
1373   snprintf(dbpath, PATH_MAX, "%s/var/lib/rpm/Name", rootdir);
1374   if (stat(dbpath, &statbuf))
1375     return 0;
1376   memset(&dbkey, 0, sizeof(dbkey));
1377   memset(&dbdata, 0, sizeof(dbdata));
1378   if (db_create(&db, dbenv, 0))
1379     {
1380       perror("db_create");
1381       return 0;
1382     }
1383   if (db->open(db, 0, "Name", 0, DB_UNKNOWN, DB_RDONLY, 0664))
1384     {
1385       perror("db->open Name index");
1386       return 0;
1387     }
1388   if (db->cursor(db, NULL, &dbc, 0))
1389     {
1390       perror("db->cursor");
1391       return 0;
1392     }
1393   while (dbc->c_get(dbc, &dbkey, &dbdata, DB_NEXT) == 0)
1394     count += dbdata.size / RPM_INDEX_SIZE;
1395   dbc->c_close(dbc);
1396   db->close(db, 0);
1397   return count;
1398 }
1399
1400 /*
1401  * read rpm db as repo
1402  *
1403  */
1404
1405 int
1406 repo_add_rpmdb(Repo *repo, Repo *ref, const char *rootdir, int flags)
1407 {
1408   Pool *pool = repo->pool;
1409   unsigned char buf[16];
1410   DB *db = 0;
1411   DBC *dbc = 0;
1412   int byteswapped;
1413   unsigned int dbid;
1414   unsigned char *dp;
1415   int dl, nrpmids;
1416   struct rpmid *rpmids, *rp;
1417   int i;
1418   int rpmheadsize;
1419   RpmHead *rpmhead;
1420   Solvable *s;
1421   Id id, *refhash;
1422   unsigned int refmask, h;
1423   char dbpath[PATH_MAX];
1424   DB_ENV *dbenv = 0;
1425   DBT dbkey;
1426   DBT dbdata;
1427   struct stat packagesstat;
1428   unsigned char newcookie[32];
1429   const unsigned char *oldcookie = 0;
1430   Id oldcookietype = 0;
1431   Repodata *data;
1432   int count = 0, done = 0;
1433   unsigned int now;
1434
1435   now = solv_timems(0);
1436   memset(&dbkey, 0, sizeof(dbkey));
1437   memset(&dbdata, 0, sizeof(dbdata));
1438
1439   if (!rootdir)
1440     rootdir = "";
1441
1442   data = repo_add_repodata(repo, flags);
1443
1444   if (ref && !(ref->nsolvables && ref->rpmdbid))
1445     ref = 0;
1446
1447   if (!(dbenv = opendbenv(rootdir)))
1448     {
1449       return pool_error(pool, -1, "repo_add_rpmdb: opendbenv failed");
1450     }
1451
1452   /* XXX: should get ro lock of Packages database! */
1453   snprintf(dbpath, PATH_MAX, "%s/var/lib/rpm/Packages", rootdir);
1454   if (stat(dbpath, &packagesstat))
1455     {
1456       return pool_error(pool, -1, "repo_add_rpmdb: %s: %s", dbpath, strerror(errno));
1457     }
1458   mkrpmdbcookie(&packagesstat, newcookie);
1459   repodata_set_bin_checksum(data, SOLVID_META, REPOSITORY_RPMDBCOOKIE, REPOKEY_TYPE_SHA256, newcookie);
1460
1461   if (ref)
1462     oldcookie = repo_lookup_bin_checksum(ref, SOLVID_META, REPOSITORY_RPMDBCOOKIE, &oldcookietype);
1463   if (!ref || !oldcookie || oldcookietype != REPOKEY_TYPE_SHA256 || memcmp(oldcookie, newcookie, 32) != 0)
1464     {
1465       Id *pkgids;
1466       int solvstart = 0, solvend = 0;
1467
1468       if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1469         count = count_headers(rootdir, dbenv);
1470       if (db_create(&db, dbenv, 0))
1471         {
1472           pool_error(pool, -1, "repo_add_rpmdb: db_create: %s", strerror(errno));
1473           dbenv->close(dbenv, 0);
1474           return -1;
1475         }
1476       if (db->open(db, 0, "Packages", 0, DB_UNKNOWN, DB_RDONLY, 0664))
1477         {
1478           pool_error(pool, -1, "repo_add_rpmdb: db->open Packages index failed: %s", strerror(errno));
1479           db->close(db, 0);
1480           dbenv->close(dbenv, 0);
1481           return -1;
1482         }
1483       if (db->get_byteswapped(db, &byteswapped))
1484         {
1485           db->close(db, 0);
1486           dbenv->close(dbenv, 0);
1487           return pool_error(pool, -1, "repo_add_rpmdb: db->get_byteswapped failed");
1488         }
1489       if (db->cursor(db, NULL, &dbc, 0))
1490         {
1491           db->close(db, 0);
1492           dbenv->close(dbenv, 0);
1493           return pool_error(pool, -1, "repo_add_rpmdb: db->cursor failed");
1494         }
1495       rpmheadsize = 0;
1496       rpmhead = 0;
1497       i = 0;
1498       s = 0;
1499       while (dbc->c_get(dbc, &dbkey, &dbdata, DB_NEXT) == 0)
1500         {
1501           if (!s)
1502             {
1503               s = pool_id2solvable(pool, repo_add_solvable(repo));
1504               if (!solvstart)
1505                 solvstart = s - pool->solvables;
1506               solvend = s - pool->solvables + 1;
1507             }
1508           if (!repo->rpmdbid)
1509             repo->rpmdbid = repo_sidedata_create(repo, sizeof(Id));
1510           if (dbkey.size != 4)
1511             {
1512               dbc->c_close(dbc);
1513               db->close(db, 0);
1514               dbenv->close(dbenv, 0);
1515               return pool_error(pool, -1, "corrupt Packages database (key size)");
1516             }
1517           dbid = db2rpmdbid(dbkey.data, byteswapped);
1518           if (dbid == 0)                /* the join key */
1519             continue;
1520           if (dbdata.size < 8)
1521             {
1522               dbc->c_close(dbc);
1523               db->close(db, 0);
1524               dbenv->close(dbenv, 0);
1525               return pool_error(pool, -1, "corrupt rpm database (size %u)\n", dbdata.size);
1526             }
1527           if (dbdata.size > rpmheadsize)
1528             {
1529               rpmheadsize = dbdata.size + 128;
1530               rpmhead = solv_realloc(rpmhead, sizeof(*rpmhead) + rpmheadsize);
1531             }
1532           memcpy(buf, dbdata.data, 8);
1533           rpmhead->cnt = buf[0] << 24  | buf[1] << 16  | buf[2] << 8 | buf[3];
1534           rpmhead->dcnt = buf[4] << 24  | buf[5] << 16  | buf[6] << 8 | buf[7];
1535           if (8 + rpmhead->cnt * 16 + rpmhead->dcnt > dbdata.size)
1536             {
1537               dbc->c_close(dbc);
1538               db->close(db, 0);
1539               dbenv->close(dbenv, 0);
1540               return pool_error(pool, -1, "corrupt rpm database (data size)\n");
1541             }
1542           memcpy(rpmhead->data, (unsigned char *)dbdata.data + 8, rpmhead->cnt * 16 + rpmhead->dcnt);
1543           rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
1544           repo->rpmdbid[(s - pool->solvables) - repo->start] = dbid;
1545           if (rpm2solv(pool, repo, data, s, rpmhead, flags | RPM_ADD_TRIGGERS))
1546             {
1547               i++;
1548               s = 0;
1549             }
1550           else
1551             {
1552               /* We can reuse this solvable, but make sure it's still
1553                  associated with this repo.  */
1554               memset(s, 0, sizeof(*s));
1555               s->repo = repo;
1556             }
1557           if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1558             {
1559               if (done < count)
1560                 done++;
1561               if (done < count && (done - 1) * 100 / count != done * 100 / count)
1562                 pool_debug(pool, SOLV_ERROR, "%%%% %d\n", done * 100 / count);
1563             }
1564         }
1565       if (s)
1566         {
1567           /* oops, could not reuse. free it instead */
1568           repo_free_solvable(repo, s - pool->solvables, 1);
1569           solvend--;
1570           s = 0;
1571         }
1572       dbc->c_close(dbc);
1573       db->close(db, 0);
1574       db = 0;
1575       /* now sort all solvables in the new solvstart..solvend block */
1576       if (solvend - solvstart > 1)
1577         {
1578           pkgids = solv_malloc2(solvend - solvstart, sizeof(Id));
1579           for (i = solvstart; i < solvend; i++)
1580             pkgids[i - solvstart] = i;
1581           solv_sort(pkgids, solvend - solvstart, sizeof(Id), pkgids_sort_cmp, repo);
1582           /* adapt order */
1583           for (i = solvstart; i < solvend; i++)
1584             {
1585               int j = pkgids[i - solvstart];
1586               while (j < i)
1587                 j = pkgids[i - solvstart] = pkgids[j - solvstart];
1588               if (j != i)
1589                 swap_solvables(repo, data, i, j);
1590             }
1591           solv_free(pkgids);
1592         }
1593     }
1594   else
1595     {
1596       Id dircache[COPYDIR_DIRCACHE_SIZE];               /* see copydir */
1597
1598       memset(dircache, 0, sizeof(dircache));
1599       if (db_create(&db, dbenv, 0))
1600         {
1601           pool_error(pool, -1, "repo_add_rpmdb: db_create: %s", strerror(errno));
1602           dbenv->close(dbenv, 0);
1603           return -1;
1604         }
1605       if (db->open(db, 0, "Name", 0, DB_UNKNOWN, DB_RDONLY, 0664))
1606         {
1607           pool_error(pool, -1, "repo_add_rpmdb: db->open Name index failed: %s", strerror(errno));
1608           db->close(db, 0);
1609           dbenv->close(dbenv, 0);
1610           return -1;
1611         }
1612       if (db->get_byteswapped(db, &byteswapped))
1613         {
1614           db->close(db, 0);
1615           dbenv->close(dbenv, 0);
1616           return pool_error(pool, -1, "repo_add_rpmdb: db->get_byteswapped failed");
1617         }
1618       if (db->cursor(db, NULL, &dbc, 0))
1619         {
1620           db->close(db, 0);
1621           dbenv->close(dbenv, 0);
1622           return pool_error(pool, -1, "repo_add_rpmdb: db->cursor failed");
1623         }
1624       nrpmids = 0;
1625       rpmids = 0;
1626       while (dbc->c_get(dbc, &dbkey, &dbdata, DB_NEXT) == 0)
1627         {
1628           if (dbkey.size == 10 && !memcmp(dbkey.data, "gpg-pubkey", 10))
1629             continue;
1630           dl = dbdata.size;
1631           dp = dbdata.data;
1632           while(dl >= RPM_INDEX_SIZE)
1633             {
1634               rpmids = solv_extend(rpmids, nrpmids, 1, sizeof(*rpmids), 255);
1635               rpmids[nrpmids].dbid = db2rpmdbid(dp, byteswapped);
1636               rpmids[nrpmids].name = solv_malloc((int)dbkey.size + 1);
1637               memcpy(rpmids[nrpmids].name, dbkey.data, (int)dbkey.size);
1638               rpmids[nrpmids].name[(int)dbkey.size] = 0;
1639               nrpmids++;
1640               dp += RPM_INDEX_SIZE;
1641               dl -= RPM_INDEX_SIZE;
1642             }
1643         }
1644       dbc->c_close(dbc);
1645       db->close(db, 0);
1646       db = 0;
1647
1648       /* sort rpmids */
1649       solv_sort(rpmids, nrpmids, sizeof(*rpmids), rpmids_sort_cmp, 0);
1650
1651       rpmheadsize = 0;
1652       rpmhead = 0;
1653
1654       /* create hash from dbid to ref */
1655       refmask = mkmask(ref->nsolvables);
1656       refhash = solv_calloc(refmask + 1, sizeof(Id));
1657       for (i = 0; i < ref->end - ref->start; i++)
1658         {
1659           if (!ref->rpmdbid[i])
1660             continue;
1661           h = ref->rpmdbid[i] & refmask;
1662           while (refhash[h])
1663             h = (h + 317) & refmask;
1664           refhash[h] = i + 1;   /* make it non-zero */
1665         }
1666
1667       /* count the misses, they will cost us time */
1668       if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1669         {
1670           for (i = 0, rp = rpmids; i < nrpmids; i++, rp++)
1671             {
1672               dbid = rp->dbid;
1673               if (refhash)
1674                 {
1675                   h = dbid & refmask;
1676                   while ((id = refhash[h]))
1677                     {
1678                       if (ref->rpmdbid[id - 1] == dbid)
1679                         break;
1680                       h = (h + 317) & refmask;
1681                     }
1682                   if (id)
1683                     continue;
1684                 }
1685               count++;
1686             }
1687         }
1688
1689       s = pool_id2solvable(pool, repo_add_solvable_block(repo, nrpmids));
1690       if (!repo->rpmdbid)
1691         repo->rpmdbid = repo_sidedata_create(repo, sizeof(Id));
1692
1693       for (i = 0, rp = rpmids; i < nrpmids; i++, rp++, s++)
1694         {
1695           dbid = rp->dbid;
1696           repo->rpmdbid[(s - pool->solvables) - repo->start] = rp->dbid;
1697           if (refhash)
1698             {
1699               h = dbid & refmask;
1700               while ((id = refhash[h]))
1701                 {
1702                   if (ref->rpmdbid[id - 1] == dbid)
1703                     break;
1704                   h = (h + 317) & refmask;
1705                 }
1706               if (id)
1707                 {
1708                   Solvable *r = ref->pool->solvables + ref->start + (id - 1);
1709                   if (r->repo == ref)
1710                     {
1711                       solvable_copy(s, r, data, dircache);
1712                       continue;
1713                     }
1714                 }
1715             }
1716           if (!db)
1717             {
1718               if (db_create(&db, dbenv, 0))
1719                 {
1720                   pool_error(pool, -1, "repo_add_rpmdb: db_create: %s", strerror(errno));
1721                   dbenv->close(dbenv, 0);
1722                   return -1;
1723                 }
1724               if (db->open(db, 0, "Packages", 0, DB_UNKNOWN, DB_RDONLY, 0664))
1725                 {
1726                   pool_error(pool, -1, "repo_add_rpmdb: db->open Packages index failed: %s", strerror(errno));
1727                   db->close(db, 0);
1728                   dbenv->close(dbenv, 0);
1729                   return -1;
1730                 }
1731               if (db->get_byteswapped(db, &byteswapped))
1732                 {
1733                   db->close(db, 0);
1734                   dbenv->close(dbenv, 0);
1735                   return pool_error(pool, -1, "repo_add_rpmdb: db->get_byteswapped failed");
1736                 }
1737             }
1738           rpmdbid2db(buf, rp->dbid, byteswapped);
1739           dbkey.data = buf;
1740           dbkey.size = 4;
1741           dbdata.data = 0;
1742           dbdata.size = 0;
1743           if (db->get(db, NULL, &dbkey, &dbdata, 0))
1744             {
1745               db->close(db, 0);
1746               dbenv->close(dbenv, 0);
1747               return pool_error(pool, -1, "inconsistent rpm database, key %d not found. run 'rpm --rebuilddb' to fix.", dbid);
1748             }
1749           if (dbdata.size < 8)
1750             {
1751               db->close(db, 0);
1752               dbenv->close(dbenv, 0);
1753               return pool_error(pool, -1, "corrupt Packages database (size)");
1754             }
1755           if (dbdata.size > rpmheadsize)
1756             {
1757               rpmheadsize = dbdata.size + 128;
1758               rpmhead = solv_realloc(rpmhead, sizeof(*rpmhead) + rpmheadsize);
1759             }
1760           memcpy(buf, dbdata.data, 8);
1761           rpmhead->cnt = buf[0] << 24  | buf[1] << 16  | buf[2] << 8 | buf[3];
1762           rpmhead->dcnt = buf[4] << 24  | buf[5] << 16  | buf[6] << 8 | buf[7];
1763           if (8 + rpmhead->cnt * 16 + rpmhead->dcnt > dbdata.size)
1764             {
1765               db->close(db, 0);
1766               dbenv->close(dbenv, 0);
1767               return pool_error(pool, -1, "corrupt Packages database (data size)");
1768             }
1769           memcpy(rpmhead->data, (unsigned char *)dbdata.data + 8, rpmhead->cnt * 16 + rpmhead->dcnt);
1770           rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
1771
1772           rpm2solv(pool, repo, data, s, rpmhead, flags | RPM_ADD_TRIGGERS);
1773           if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1774             {
1775               if (done < count)
1776                 done++;
1777               if (done < count && (done - 1) * 100 / count != done * 100 / count)
1778                 pool_debug(pool, SOLV_ERROR, "%%%% %d\n", done * 100 / count);
1779             }
1780         }
1781
1782       if (refhash)
1783         solv_free(refhash);
1784       if (rpmids)
1785         {
1786           for (i = 0; i < nrpmids; i++)
1787             solv_free(rpmids[i].name);
1788           solv_free(rpmids);
1789         }
1790     }
1791   if (db)
1792     db->close(db, 0);
1793   dbenv->close(dbenv, 0);
1794   if (rpmhead)
1795     solv_free(rpmhead);
1796   if (!(flags & REPO_NO_INTERNALIZE))
1797     repodata_internalize(data);
1798   if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1799     pool_debug(pool, SOLV_ERROR, "%%%% 100\n");
1800   POOL_DEBUG(SOLV_DEBUG_STATS, "repo_add_rpmdb took %d ms\n", solv_timems(now));
1801   POOL_DEBUG(SOLV_DEBUG_STATS, "repo size: %d solvables\n", repo->nsolvables);
1802   POOL_DEBUG(SOLV_DEBUG_STATS, "repo memory used: %d K incore, %d K idarray\n", repodata_memused(data)/1024, repo->idarraysize / (int)(1024/sizeof(Id)));
1803   return 0;
1804 }
1805
1806
1807 static inline unsigned int
1808 getu32(const unsigned char *dp)
1809 {
1810   return dp[0] << 24 | dp[1] << 16 | dp[2] << 8 | dp[3];
1811 }
1812
1813
1814 Id
1815 repo_add_rpm(Repo *repo, const char *rpm, int flags)
1816 {
1817   int sigdsize, sigcnt, l;
1818   Pool *pool = repo->pool;
1819   Solvable *s;
1820   RpmHead *rpmhead = 0;
1821   int rpmheadsize = 0;
1822   char *payloadformat;
1823   FILE *fp;
1824   unsigned char lead[4096];
1825   int headerstart, headerend;
1826   struct stat stb;
1827   Repodata *data;
1828   unsigned char pkgid[16];
1829   int gotpkgid;
1830   Id chksumtype = 0;
1831   void *chksumh = 0;
1832
1833   data = repo_add_repodata(repo, flags);
1834
1835   if ((flags & RPM_ADD_WITH_SHA256SUM) != 0)
1836     chksumtype = REPOKEY_TYPE_SHA256;
1837   else if ((flags & RPM_ADD_WITH_SHA1SUM) != 0)
1838     chksumtype = REPOKEY_TYPE_SHA1;
1839
1840   if ((fp = fopen(rpm, "r")) == 0)
1841     {
1842       pool_error(pool, -1, "%s: %s", rpm, strerror(errno));
1843       return 0;
1844     }
1845   if (fstat(fileno(fp), &stb))
1846     {
1847       pool_error(pool, -1, "fstat: %s", strerror(errno));
1848       fclose(fp);
1849       return 0;
1850     }
1851   if (chksumtype)
1852     chksumh = solv_chksum_create(chksumtype);
1853   if (fread(lead, 96 + 16, 1, fp) != 1 || getu32(lead) != 0xedabeedb)
1854     {
1855       pool_error(pool, -1, "%s: not a rpm", rpm);
1856       fclose(fp);
1857       return 0;
1858     }
1859   if (chksumh)
1860     solv_chksum_add(chksumh, lead, 96 + 16);
1861   if (lead[78] != 0 || lead[79] != 5)
1862     {
1863       pool_error(pool, -1, "%s: not a rpm v5 header", rpm);
1864       fclose(fp);
1865       return 0;
1866     }
1867   if (getu32(lead + 96) != 0x8eade801)
1868     {
1869       pool_error(pool, -1, "%s: bad signature header", rpm);
1870       fclose(fp);
1871       return 0;
1872     }
1873   sigcnt = getu32(lead + 96 + 8);
1874   sigdsize = getu32(lead + 96 + 12);
1875   if (sigcnt >= 0x100000 || sigdsize >= 0x100000)
1876     {
1877       pool_error(pool, -1, "%s: bad signature header", rpm);
1878       fclose(fp);
1879       return 0;
1880     }
1881   sigdsize += sigcnt * 16;
1882   sigdsize = (sigdsize + 7) & ~7;
1883   headerstart = 96 + 16 + sigdsize;
1884   gotpkgid = 0;
1885   if ((flags & RPM_ADD_WITH_PKGID) != 0)
1886     {
1887       unsigned char *chksum;
1888       unsigned int chksumsize;
1889       /* extract pkgid from the signature header */
1890       if (sigdsize > rpmheadsize)
1891         {
1892           rpmheadsize = sigdsize + 128;
1893           rpmhead = solv_realloc(rpmhead, sizeof(*rpmhead) + rpmheadsize);
1894         }
1895       if (fread(rpmhead->data, sigdsize, 1, fp) != 1)
1896         {
1897           pool_error(pool, -1, "%s: unexpected EOF", rpm);
1898           fclose(fp);
1899           return 0;
1900         }
1901       if (chksumh)
1902         solv_chksum_add(chksumh, rpmhead->data, sigdsize);
1903       rpmhead->cnt = sigcnt;
1904       rpmhead->dcnt = sigdsize - sigcnt * 16;
1905       rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
1906       chksum = headbinary(rpmhead, SIGTAG_MD5, &chksumsize);
1907       if (chksum && chksumsize == 16)
1908         {
1909           gotpkgid = 1;
1910           memcpy(pkgid, chksum, 16);
1911         }
1912     }
1913   else
1914     {
1915       /* just skip the signature header */
1916       while (sigdsize)
1917         {
1918           l = sigdsize > 4096 ? 4096 : sigdsize;
1919           if (fread(lead, l, 1, fp) != 1)
1920             {
1921               pool_error(pool, -1, "%s: unexpected EOF", rpm);
1922               fclose(fp);
1923               return 0;
1924             }
1925           if (chksumh)
1926             solv_chksum_add(chksumh, lead, l);
1927           sigdsize -= l;
1928         }
1929     }
1930   if (fread(lead, 16, 1, fp) != 1)
1931     {
1932       pool_error(pool, -1, "%s: unexpected EOF", rpm);
1933       fclose(fp);
1934       return 0;
1935     }
1936   if (chksumh)
1937     solv_chksum_add(chksumh, lead, 16);
1938   if (getu32(lead) != 0x8eade801)
1939     {
1940       pool_error(pool, -1, "%s: bad header", rpm);
1941       fclose(fp);
1942       return 0;
1943     }
1944   sigcnt = getu32(lead + 8);
1945   sigdsize = getu32(lead + 12);
1946   if (sigcnt >= 0x100000 || sigdsize >= 0x800000)
1947     {
1948       pool_error(pool, -1, "%s: bad header", rpm);
1949       fclose(fp);
1950       return 0;
1951     }
1952   l = sigdsize + sigcnt * 16;
1953   headerend = headerstart + 16 + l;
1954   if (l > rpmheadsize)
1955     {
1956       rpmheadsize = l + 128;
1957       rpmhead = solv_realloc(rpmhead, sizeof(*rpmhead) + rpmheadsize);
1958     }
1959   if (fread(rpmhead->data, l, 1, fp) != 1)
1960     {
1961       pool_error(pool, -1, "%s: unexpected EOF", rpm);
1962       fclose(fp);
1963       return 0;
1964     }
1965   if (chksumh)
1966     solv_chksum_add(chksumh, rpmhead->data, l);
1967   rpmhead->cnt = sigcnt;
1968   rpmhead->dcnt = sigdsize;
1969   rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
1970   if (headexists(rpmhead, TAG_PATCHESNAME))
1971     {
1972       /* this is a patch rpm, ignore */
1973       pool_error(pool, -1, "%s: is patch rpm", rpm);
1974       fclose(fp);
1975       return 0;
1976     }
1977   payloadformat = headstring(rpmhead, TAG_PAYLOADFORMAT);
1978   if (payloadformat && !strcmp(payloadformat, "drpm"))
1979     {
1980       /* this is a delta rpm */
1981       pool_error(pool, -1, "%s: is delta rpm", rpm);
1982       fclose(fp);
1983       return 0;
1984     }
1985   if (chksumh)
1986     while ((l = fread(lead, 1, sizeof(lead), fp)) > 0)
1987       solv_chksum_add(chksumh, lead, l);
1988   fclose(fp);
1989   s = pool_id2solvable(pool, repo_add_solvable(repo));
1990   if (!rpm2solv(pool, repo, data, s, rpmhead, flags))
1991     {
1992       repo_free_solvable(repo, s - pool->solvables, 1);
1993       return 0;
1994     }
1995   repodata_set_location(data, s - pool->solvables, 0, 0, rpm);
1996   if (S_ISREG(stb.st_mode))
1997     repodata_set_num(data, s - pool->solvables, SOLVABLE_DOWNLOADSIZE, (unsigned long long)stb.st_size);
1998   repodata_set_num(data, s - pool->solvables, SOLVABLE_HEADEREND, headerend);
1999   if (gotpkgid)
2000     repodata_set_bin_checksum(data, s - pool->solvables, SOLVABLE_PKGID, REPOKEY_TYPE_MD5, pkgid);
2001   if (chksumh)
2002     {
2003       repodata_set_bin_checksum(data, s - pool->solvables, SOLVABLE_CHECKSUM, chksumtype, solv_chksum_get(chksumh, 0));
2004       chksumh = solv_chksum_free(chksumh, 0);
2005     }
2006   if (rpmhead)
2007     solv_free(rpmhead);
2008   if (!(flags & REPO_NO_INTERNALIZE))
2009     repodata_internalize(data);
2010   return s - pool->solvables;
2011 }
2012
2013 static inline void
2014 linkhash(const char *lt, char *hash)
2015 {
2016   unsigned int r = 0;
2017   const unsigned char *str = (const unsigned char *)lt;
2018   int l, c;
2019
2020   l = strlen(lt);
2021   while ((c = *str++) != 0)
2022     r += (r << 3) + c;
2023   sprintf(hash, "%08x", r);
2024   sprintf(hash + 8, "%08x", l);
2025   sprintf(hash + 16, "%08x", 0);
2026   sprintf(hash + 24, "%08x", 0);
2027 }
2028
2029 void
2030 rpm_iterate_filelist(void *rpmhandle, int flags, void (*cb)(void *, const char *, int, const char *), void *cbdata)
2031 {
2032   RpmHead *rpmhead = rpmhandle;
2033   char **bn;
2034   char **dn;
2035   char **md = 0;
2036   char **lt = 0;
2037   unsigned int *di, diidx;
2038   unsigned int *co = 0;
2039   unsigned int *ff = 0;
2040   unsigned int lastdir;
2041   int lastdirl;
2042   unsigned int *fm;
2043   int cnt, dcnt, cnt2;
2044   int i, l1, l;
2045   char *space = 0;
2046   int spacen = 0;
2047   char md5[33], *md5p = 0;
2048
2049   dn = headstringarray(rpmhead, TAG_DIRNAMES, &dcnt);
2050   if (!dn)
2051     return;
2052   if ((flags & RPM_ITERATE_FILELIST_ONLYDIRS) != 0)
2053     {
2054       for (i = 0; i < dcnt; i++)
2055         (*cb)(cbdata, dn[i], 0, (char *)0);
2056       solv_free(dn);
2057       return;
2058     }
2059   bn = headstringarray(rpmhead, TAG_BASENAMES, &cnt);
2060   if (!bn)
2061     {
2062       solv_free(dn);
2063       return;
2064     }
2065   di = headint32array(rpmhead, TAG_DIRINDEXES, &cnt2);
2066   if (!di || cnt != cnt2)
2067     {
2068       solv_free(di);
2069       solv_free(bn);
2070       solv_free(dn);
2071       return;
2072     }
2073   fm = headint16array(rpmhead, TAG_FILEMODES, &cnt2);
2074   if (!fm || cnt != cnt2)
2075     {
2076       solv_free(fm);
2077       solv_free(di);
2078       solv_free(bn);
2079       solv_free(dn);
2080       return;
2081     }
2082   if ((flags & RPM_ITERATE_FILELIST_WITHMD5) != 0)
2083     {
2084       md = headstringarray(rpmhead, TAG_FILEMD5S, &cnt2);
2085       if (!md || cnt != cnt2)
2086         {
2087           solv_free(md);
2088           solv_free(fm);
2089           solv_free(di);
2090           solv_free(bn);
2091           solv_free(dn);
2092           return;
2093         }
2094     }
2095   if ((flags & RPM_ITERATE_FILELIST_WITHCOL) != 0)
2096     {
2097       co = headint32array(rpmhead, TAG_FILECOLORS, &cnt2);
2098       if (!co || cnt != cnt2)
2099         {
2100           solv_free(co);
2101           solv_free(md);
2102           solv_free(fm);
2103           solv_free(di);
2104           solv_free(bn);
2105           solv_free(dn);
2106           return;
2107         }
2108     }
2109   if ((flags & RPM_ITERATE_FILELIST_NOGHOSTS) != 0)
2110     {
2111       ff = headint32array(rpmhead, TAG_FILEFLAGS, &cnt2);
2112       if (!ff || cnt != cnt2)
2113         {
2114           solv_free(ff);
2115           solv_free(co);
2116           solv_free(md);
2117           solv_free(fm);
2118           solv_free(di);
2119           solv_free(bn);
2120           solv_free(dn);
2121           return;
2122         }
2123     }
2124   lastdir = dcnt;
2125   lastdirl = 0;
2126   for (i = 0; i < cnt; i++)
2127     {
2128       if (ff && (ff[i] & FILEFLAG_GHOST) != 0)
2129         continue;
2130       diidx = di[i];
2131       if (diidx >= dcnt)
2132         continue;
2133       l1 = lastdir == diidx ? lastdirl : strlen(dn[diidx]);
2134       if (l1 == 0)
2135         continue;
2136       l = l1 + strlen(bn[i]) + 1;
2137       if (l > spacen)
2138         {
2139           spacen = l + 16;
2140           space = solv_realloc(space, spacen);
2141         }
2142       if (lastdir != diidx)
2143         {
2144           strcpy(space, dn[diidx]);
2145           lastdir = diidx;
2146           lastdirl = l1;
2147         }
2148       strcpy(space + l1, bn[i]);
2149       if (md)
2150         {
2151           md5p = md[i];
2152           if (S_ISLNK(fm[i]))
2153             {
2154               md5p = 0;
2155               if (!lt)
2156                 {
2157                   lt = headstringarray(rpmhead, TAG_FILELINKTOS, &cnt2);
2158                   if (cnt != cnt2)
2159                     lt = solv_free(lt);
2160                 }
2161               if (lt)
2162                 {
2163                   linkhash(lt[i], md5);
2164                   md5p = md5;
2165                 }
2166             }
2167           if (!md5p)
2168             {
2169               sprintf(md5, "%08x%08x%08x%08x", (fm[i] >> 12) & 65535, 0, 0, 0);
2170               md5p = md5;
2171             }
2172         }
2173       (*cb)(cbdata, space, co ? (fm[i] | co[i] << 24) : fm[i], md5p);
2174     }
2175   solv_free(space);
2176   solv_free(lt);
2177   solv_free(md);
2178   solv_free(fm);
2179   solv_free(di);
2180   solv_free(bn);
2181   solv_free(dn);
2182   solv_free(co);
2183   solv_free(ff);
2184 }
2185
2186 char *
2187 rpm_query(void *rpmhandle, Id what)
2188 {
2189   const char *name, *arch, *sourcerpm;
2190   char *evr, *r;
2191   int l;
2192
2193   RpmHead *rpmhead = rpmhandle;
2194   r = 0;
2195   switch (what)
2196     {
2197     case 0:
2198       name = headstring(rpmhead, TAG_NAME);
2199       if (!name)
2200         name = "";
2201       sourcerpm = headstring(rpmhead, TAG_SOURCERPM);
2202       if (sourcerpm)
2203         arch = headstring(rpmhead, TAG_ARCH);
2204       else
2205         {
2206           if (headexists(rpmhead, TAG_NOSOURCE) || headexists(rpmhead, TAG_NOPATCH))
2207             arch = "nosrc";
2208           else
2209             arch = "src";
2210         }
2211       if (!arch)
2212         arch = "noarch";
2213       evr = headtoevr(rpmhead);
2214       if (!evr)
2215         break;
2216       l = strlen(name) + 1 + strlen(evr) + 1 + strlen(arch) + 1;
2217       r = solv_malloc(l);
2218       sprintf(r, "%s-%s.%s", name, evr, arch);
2219       free(evr);
2220       break;
2221     case SOLVABLE_NAME:
2222       name = headstring(rpmhead, TAG_NAME);
2223       r = solv_strdup(name);
2224       break;
2225     case SOLVABLE_EVR:
2226       r = headtoevr(rpmhead);
2227       break;
2228     }
2229   return r;
2230 }
2231
2232
2233 struct rpm_by_state {
2234   RpmHead *rpmhead;
2235   int rpmheadsize;
2236
2237   int dbopened;
2238   DB_ENV *dbenv;
2239   DB *db;
2240   int byteswapped;
2241 };
2242
2243 struct rpmdbentry {
2244   Id rpmdbid;
2245   Id nameoff;
2246 };
2247
2248 #define ENTRIES_BLOCK 255
2249 #define NAMEDATA_BLOCK 1023
2250
2251 static struct rpmdbentry *
2252 getinstalledrpmdbids(struct rpm_by_state *state, const char *index, const char *match, int *nentriesp, char **namedatap)
2253 {
2254   DB_ENV *dbenv = 0;
2255   DB *db = 0;
2256   DBC *dbc = 0;
2257   int byteswapped;
2258   DBT dbkey;
2259   DBT dbdata;
2260   unsigned char *dp;
2261   int dl;
2262
2263   char *namedata = 0;
2264   int namedatal = 0;
2265   struct rpmdbentry *entries = 0;
2266   int nentries = 0;
2267
2268   *nentriesp = 0;
2269   *namedatap = 0;
2270
2271   dbenv = state->dbenv;
2272   if (db_create(&db, dbenv, 0))
2273     {
2274       perror("db_create");
2275       return 0;
2276     }
2277   if (db->open(db, 0, index, 0, DB_UNKNOWN, DB_RDONLY, 0664))
2278     {
2279       perror("db->open index");
2280       db->close(db, 0);
2281       return 0;
2282     }
2283   if (db->get_byteswapped(db, &byteswapped))
2284     {
2285       perror("db->get_byteswapped");
2286       db->close(db, 0);
2287       return 0;
2288     }
2289   if (db->cursor(db, NULL, &dbc, 0))
2290     {
2291       perror("db->cursor");
2292       db->close(db, 0);
2293       return 0;
2294     }
2295   memset(&dbkey, 0, sizeof(dbkey));
2296   memset(&dbdata, 0, sizeof(dbdata));
2297   if (match)
2298     {
2299       dbkey.data = (void *)match;
2300       dbkey.size = strlen(match);
2301     }
2302   while (dbc->c_get(dbc, &dbkey, &dbdata, match ? DB_SET : DB_NEXT) == 0)
2303     {
2304       if (!match && dbkey.size == 10 && !memcmp(dbkey.data, "gpg-pubkey", 10))
2305         continue;
2306       dl = dbdata.size;
2307       dp = dbdata.data;
2308       while(dl >= RPM_INDEX_SIZE)
2309         {
2310           entries = solv_extend(entries, nentries, 1, sizeof(*entries), ENTRIES_BLOCK);
2311           entries[nentries].rpmdbid = db2rpmdbid(dp, byteswapped);
2312           entries[nentries].nameoff = namedatal;
2313           nentries++;
2314           namedata = solv_extend(namedata, namedatal, dbkey.size + 1, 1, NAMEDATA_BLOCK);
2315           memcpy(namedata + namedatal, dbkey.data, dbkey.size);
2316           namedata[namedatal + dbkey.size] = 0;
2317           namedatal += dbkey.size + 1;
2318           dp += RPM_INDEX_SIZE;
2319           dl -= RPM_INDEX_SIZE;
2320         }
2321       if (match)
2322         break;
2323     }
2324   dbc->c_close(dbc);
2325   db->close(db, 0);
2326   *nentriesp = nentries;
2327   *namedatap = namedata;
2328   return entries;
2329 }
2330
2331 static void
2332 freestate(struct rpm_by_state *state)
2333 {
2334   /* close down */
2335   if (!state)
2336     return;
2337   if (state->db)
2338     state->db->close(state->db, 0);
2339   if (state->dbenv)
2340     state->dbenv->close(state->dbenv, 0);
2341   solv_free(state->rpmhead);
2342 }
2343
2344 int
2345 rpm_installedrpmdbids(const char *rootdir, const char *index, const char *match, Queue *rpmdbidq)
2346 {
2347   struct rpm_by_state state;
2348   struct rpmdbentry *entries;
2349   int nentries, i;
2350   char *namedata;
2351
2352   if (!index)
2353     index = "Name";
2354   if (rpmdbidq)
2355     queue_empty(rpmdbidq);
2356   memset(&state, 0, sizeof(state));
2357   if (!(state.dbenv = opendbenv(rootdir)))
2358     return 0;
2359   entries = getinstalledrpmdbids(&state, index, match, &nentries, &namedata);
2360   if (rpmdbidq)
2361     for (i = 0; i < nentries; i++)
2362       queue_push(rpmdbidq, entries[i].rpmdbid);
2363   solv_free(entries);
2364   solv_free(namedata);
2365   freestate(&state);
2366   return nentries;
2367 }
2368
2369 void *
2370 rpm_byrpmdbid(Id rpmdbid, const char *rootdir, void **statep)
2371 {
2372   struct rpm_by_state *state = *statep;
2373   unsigned char buf[16];
2374   DBT dbkey;
2375   DBT dbdata;
2376   RpmHead *rpmhead;
2377
2378   if (!rpmdbid)
2379     {
2380       /* close down */
2381       freestate(state);
2382       solv_free(state);
2383       *statep = (void *)0;
2384       return 0;
2385     }
2386
2387   if (!state)
2388     {
2389       state = solv_calloc(1, sizeof(*state));
2390       *statep = state;
2391     }
2392   if (!state->dbopened)
2393     {
2394       state->dbopened = 1;
2395       if (!state->dbenv && !(state->dbenv = opendbenv(rootdir)))
2396         return 0;
2397       if (db_create(&state->db, state->dbenv, 0))
2398         {
2399           perror("db_create");
2400           state->db = 0;
2401           state->dbenv->close(state->dbenv, 0);
2402           state->dbenv = 0;
2403           return 0;
2404         }
2405       if (state->db->open(state->db, 0, "Packages", 0, DB_UNKNOWN, DB_RDONLY, 0664))
2406         {
2407           perror("db->open var/lib/rpm/Packages");
2408           state->db->close(state->db, 0);
2409           state->db = 0;
2410           state->dbenv->close(state->dbenv, 0);
2411           state->dbenv = 0;
2412           return 0;
2413         }
2414       if (state->db->get_byteswapped(state->db, &state->byteswapped))
2415         {
2416           perror("db->get_byteswapped");
2417           state->db->close(state->db, 0);
2418           state->db = 0;
2419           state->dbenv->close(state->dbenv, 0);
2420           state->dbenv = 0;
2421           return 0;
2422         }
2423     }
2424   rpmdbid2db(buf, rpmdbid, state->byteswapped);
2425   memset(&dbkey, 0, sizeof(dbkey));
2426   memset(&dbdata, 0, sizeof(dbdata));
2427   dbkey.data = buf;
2428   dbkey.size = 4;
2429   dbdata.data = 0;
2430   dbdata.size = 0;
2431   if (state->db->get(state->db, NULL, &dbkey, &dbdata, 0))
2432     {
2433       perror("db->get");
2434       return 0;
2435     }
2436   if (dbdata.size < 8)
2437     {
2438       fprintf(stderr, "corrupt rpm database (size)\n");
2439       return 0;
2440     }
2441   if (dbdata.size > state->rpmheadsize)
2442     {
2443       state->rpmheadsize = dbdata.size + 128;
2444       state->rpmhead = solv_realloc(state->rpmhead, sizeof(*rpmhead) + state->rpmheadsize);
2445     }
2446   rpmhead = state->rpmhead;
2447   memcpy(buf, dbdata.data, 8);
2448   rpmhead->cnt = buf[0] << 24  | buf[1] << 16  | buf[2] << 8 | buf[3];
2449   rpmhead->dcnt = buf[4] << 24  | buf[5] << 16  | buf[6] << 8 | buf[7];
2450   if (8 + rpmhead->cnt * 16 + rpmhead->dcnt > dbdata.size)
2451     {
2452       fprintf(stderr, "corrupt rpm database (data size)\n");
2453       return 0;
2454     }
2455   memcpy(rpmhead->data, (unsigned char *)dbdata.data + 8, rpmhead->cnt * 16 + rpmhead->dcnt);
2456   rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
2457   return rpmhead;
2458 }
2459
2460 void *
2461 rpm_byfp(FILE *fp, const char *name, void **statep)
2462 {
2463   struct rpm_by_state *state = *statep;
2464   /* int headerstart, headerend; */
2465   RpmHead *rpmhead;
2466   int sigdsize, sigcnt, l;
2467   unsigned char lead[4096];
2468
2469   if (!fp)
2470     return rpm_byrpmdbid(0, 0, statep);
2471   if (!state)
2472     {
2473       state = solv_calloc(1, sizeof(*state));
2474       *statep = state;
2475     }
2476   if (fread(lead, 96 + 16, 1, fp) != 1 || getu32(lead) != 0xedabeedb)
2477     {
2478       fprintf(stderr, "%s: not a rpm\n", name);
2479       return 0;
2480     }
2481   if (lead[78] != 0 || lead[79] != 5)
2482     {
2483       fprintf(stderr, "%s: not a V5 header\n", name);
2484       return 0;
2485     }
2486   if (getu32(lead + 96) != 0x8eade801)
2487     {
2488       fprintf(stderr, "%s: bad signature header\n", name);
2489       return 0;
2490     }
2491   sigcnt = getu32(lead + 96 + 8);
2492   sigdsize = getu32(lead + 96 + 12);
2493   if (sigcnt >= 0x100000 || sigdsize >= 0x100000)
2494     {
2495       fprintf(stderr, "%s: bad signature header\n", name);
2496       return 0;
2497     }
2498   sigdsize += sigcnt * 16;
2499   sigdsize = (sigdsize + 7) & ~7;
2500   /* headerstart = 96 + 16 + sigdsize; */
2501   while (sigdsize)
2502     {
2503       l = sigdsize > 4096 ? 4096 : sigdsize;
2504       if (fread(lead, l, 1, fp) != 1)
2505         {
2506           fprintf(stderr, "%s: unexpected EOF\n", name);
2507           return 0;
2508         }
2509       sigdsize -= l;
2510     }
2511   if (fread(lead, 16, 1, fp) != 1)
2512     {
2513       fprintf(stderr, "%s: unexpected EOF\n", name);
2514       return 0;
2515     }
2516   if (getu32(lead) != 0x8eade801)
2517     {
2518       fprintf(stderr, "%s: bad header\n", name);
2519       fclose(fp);
2520       return 0;
2521     }
2522   sigcnt = getu32(lead + 8);
2523   sigdsize = getu32(lead + 12);
2524   if (sigcnt >= 0x100000 || sigdsize >= 0x800000)
2525     {
2526       fprintf(stderr, "%s: bad header\n", name);
2527       fclose(fp);
2528       return 0;
2529     }
2530   l = sigdsize + sigcnt * 16;
2531   /* headerend = headerstart + 16 + l; */
2532   if (l > state->rpmheadsize)
2533     {
2534       state->rpmheadsize = l + 128;
2535       state->rpmhead = solv_realloc(state->rpmhead, sizeof(*state->rpmhead) + state->rpmheadsize);
2536     }
2537   rpmhead = state->rpmhead;
2538   if (fread(rpmhead->data, l, 1, fp) != 1)
2539     {
2540       fprintf(stderr, "%s: unexpected EOF\n", name);
2541       fclose(fp);
2542       return 0;
2543     }
2544   rpmhead->cnt = sigcnt;
2545   rpmhead->dcnt = sigdsize;
2546   rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
2547   return rpmhead;
2548 }
2549
2550 void *
2551 rpm_byrpmh(Header h, void **statep)
2552 {
2553   struct rpm_by_state *state = *statep;
2554   const unsigned char *uh;
2555   int sigdsize, sigcnt, l;
2556   RpmHead *rpmhead;
2557
2558 #ifndef RPM5
2559   uh = headerUnload(h);
2560 #else
2561   uh = headerUnload(h, NULL);
2562 #endif
2563   if (!uh)
2564     return 0;
2565   sigcnt = getu32(uh);
2566   sigdsize = getu32(uh + 4);
2567   l = sigdsize + sigcnt * 16;
2568   if (!state)
2569     {
2570       state = solv_calloc(1, sizeof(*state));
2571       *statep = state;
2572     }
2573   if (l > state->rpmheadsize)
2574     {
2575       state->rpmheadsize = l + 128;
2576       state->rpmhead = solv_realloc(state->rpmhead, sizeof(*state->rpmhead) + state->rpmheadsize);
2577     }
2578   rpmhead = state->rpmhead;
2579   memcpy(rpmhead->data, uh + 8, l - 8);
2580   free((void *)uh);
2581   rpmhead->cnt = sigcnt;
2582   rpmhead->dcnt = sigdsize;
2583   rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
2584   return rpmhead;
2585 }
2586
2587
2588 #ifdef ENABLE_RPMDB_PUBKEY
2589
2590 static char *
2591 r64dec1(char *p, unsigned int *vp, int *eofp)
2592 {
2593   int i, x;
2594   unsigned int v = 0;
2595
2596   for (i = 0; i < 4; )
2597     {
2598       x = *p++;
2599       if (!x)
2600         return 0;
2601       if (x >= 'A' && x <= 'Z')
2602         x -= 'A';
2603       else if (x >= 'a' && x <= 'z')
2604         x -= 'a' - 26;
2605       else if (x >= '0' && x <= '9')
2606         x -= '0' - 52;
2607       else if (x == '+')
2608         x = 62;
2609       else if (x == '/')
2610         x = 63;
2611       else if (x == '=')
2612         {
2613           x = 0;
2614           if (i == 0)
2615             {
2616               *eofp = 3;
2617               *vp = 0;
2618               return p - 1;
2619             }
2620           *eofp += 1;
2621         }
2622       else
2623         continue;
2624       v = v << 6 | x;
2625       i++;
2626     }
2627   *vp = v;
2628   return p;
2629 }
2630
2631 static unsigned int
2632 crc24(unsigned char *p, int len)
2633 {
2634   unsigned int crc = 0xb704ceL;
2635   int i;
2636
2637   while (len--)
2638     {
2639       crc ^= (*p++) << 16;
2640       for (i = 0; i < 8; i++)
2641         if ((crc <<= 1) & 0x1000000)
2642           crc ^= 0x1864cfbL;
2643     }
2644   return crc & 0xffffffL;
2645 }
2646
2647 static unsigned char *
2648 unarmor(char *pubkey, int *pktlp)
2649 {
2650   char *p;
2651   int l, eof;
2652   unsigned char *buf, *bp;
2653   unsigned int v;
2654
2655   *pktlp = 0;
2656   while (strncmp(pubkey, "-----BEGIN PGP PUBLIC KEY BLOCK-----", 36) != 0)
2657     {
2658       pubkey = strchr(pubkey, '\n');
2659       if (!pubkey)
2660         return 0;
2661       pubkey++;
2662     }
2663   pubkey = strchr(pubkey, '\n');
2664   if (!pubkey++)
2665     return 0;
2666   /* skip header lines */
2667   for (;;)
2668     {
2669       while (*pubkey == ' ' || *pubkey == '\t')
2670         pubkey++;
2671       if (*pubkey == '\n')
2672         break;
2673       pubkey = strchr(pubkey, '\n');
2674       if (!pubkey++)
2675         return 0;
2676     }
2677   pubkey++;
2678   p = strchr(pubkey, '=');
2679   if (!p)
2680     return 0;
2681   l = p - pubkey;
2682   bp = buf = solv_malloc(l * 3 / 4 + 4);
2683   eof = 0;
2684   while (!eof)
2685     {
2686       pubkey = r64dec1(pubkey, &v, &eof);
2687       if (!pubkey)
2688         {
2689           solv_free(buf);
2690           return 0;
2691         }
2692       *bp++ = v >> 16;
2693       *bp++ = v >> 8;
2694       *bp++ = v;
2695     }
2696   while (*pubkey == ' ' || *pubkey == '\t' || *pubkey == '\n' || *pubkey == '\r')
2697     pubkey++;
2698   bp -= eof;
2699   if (*pubkey != '=' || (pubkey = r64dec1(pubkey + 1, &v, &eof)) == 0)
2700     {
2701       solv_free(buf);
2702       return 0;
2703     }
2704   if (v != crc24(buf, bp - buf))
2705     {
2706       solv_free(buf);
2707       return 0;
2708     }
2709   while (*pubkey == ' ' || *pubkey == '\t' || *pubkey == '\n' || *pubkey == '\r')
2710     pubkey++;
2711   if (strncmp(pubkey, "-----END PGP PUBLIC KEY BLOCK-----", 34) != 0)
2712     {
2713       solv_free(buf);
2714       return 0;
2715     }
2716   *pktlp = bp - buf;
2717   return buf;
2718 }
2719
2720 static void
2721 parsekeydata(Solvable *s, Repodata *data, unsigned char *p, int pl)
2722 {
2723   int x, tag, l;
2724   unsigned char keyid[8];
2725   unsigned int kcr = 0, maxex = 0;
2726   unsigned char *pubkey = 0;
2727   unsigned char *userid = 0;
2728 #if 0
2729   int pubkeyl = 0;
2730   int useridl = 0;
2731 #endif
2732
2733   for (; pl; p += l, pl -= l)
2734     {
2735       x = *p++;
2736       pl--;
2737       if (!(x & 128) || pl <= 0)
2738         return;
2739       if ((x & 64) == 0)
2740         {
2741           /* old format */
2742           tag = (x & 0x3c) >> 2;
2743           x &= 3;
2744           if (x == 3)
2745             return;
2746           l = 1 << x;
2747           if (pl < l)
2748             return;
2749           x = 0;
2750           while (l--)
2751             {
2752               x = x << 8 | *p++;
2753               pl--;
2754             }
2755           l = x;
2756         }
2757       else
2758         {
2759           tag = (x & 0x3f);
2760           x = *p++;
2761           pl--;
2762           if (x < 192)
2763             l = x;
2764           else if (x >= 192 && x < 224)
2765             {
2766               if (pl <= 0)
2767                 return;
2768               l = ((x - 192) << 8) + *p++ + 192;
2769               pl--;
2770             }
2771           else if (x == 255)
2772             {
2773               if (pl <= 4)
2774                 return;
2775               l = p[0] << 24 | p[1] << 16 | p[2] << 8 | p[3];
2776               p += 4;
2777               pl -= 4;
2778             }
2779           else
2780             return;
2781         }
2782       if (pl < l)
2783         return;
2784       if (tag == 6)
2785         {
2786           pubkey = solv_realloc(pubkey, l);
2787           if (l)
2788             memcpy(pubkey, p, l);
2789 #if 0
2790           pubkeyl = l;
2791 #endif
2792           kcr = 0;
2793           if (p[0] == 3)
2794             {
2795               unsigned int ex;
2796               void *h;
2797               kcr = p[1] << 24 | p[2] << 16 | p[3] << 8 | p[4];
2798               ex = 0;
2799               if (p[5] || p[6])
2800                 {
2801                   ex = kcr + 24*3600 * (p[5] << 8 | p[6]);
2802                   if (ex > maxex)
2803                     maxex = ex;
2804                 }
2805               memset(keyid, 0, 8);
2806               if (p[7] == 1)    /* RSA */
2807                 {
2808                   int i, ql;
2809                   unsigned char fp[16];
2810                   char fpx[32 + 1];
2811                   unsigned char *q;
2812
2813                   ql = ((p[8] << 8 | p[9]) + 7) / 8;
2814                   memcpy(keyid, p + 10 + ql - 8, 8);
2815                   h = solv_chksum_create(REPOKEY_TYPE_MD5);
2816                   solv_chksum_add(h, p + 10, ql);
2817                   q = p + 10 + ql;
2818                   ql = ((q[0] << 8 | q[1]) + 7) / 8;
2819                   solv_chksum_add(h, q + 2, ql);
2820                   solv_chksum_free(h, fp);
2821                   for (i = 0; i < 16; i++)
2822                     sprintf(fpx + i * 2, "%02x", fp[i]);
2823                   setutf8string(data, s - s->repo->pool->solvables, PUBKEY_FINGERPRINT, fpx);
2824                 }
2825             }
2826           else if (p[0] == 4)
2827             {
2828               int i;
2829               void *h;
2830               unsigned char hdr[3];
2831               unsigned char fp[20];
2832               char fpx[40 + 1];
2833
2834               kcr = p[1] << 24 | p[2] << 16 | p[3] << 8 | p[4];
2835               hdr[0] = 0x99;
2836               hdr[1] = l >> 8;
2837               hdr[2] = l;
2838               h = solv_chksum_create(REPOKEY_TYPE_SHA1);
2839               solv_chksum_add(h, hdr, 3);
2840               solv_chksum_add(h, p, l);
2841               solv_chksum_free(h, fp);
2842               for (i = 0; i < 20; i++)
2843                 sprintf(fpx + i * 2, "%02x", fp[i]);
2844               setutf8string(data, s - s->repo->pool->solvables, PUBKEY_FINGERPRINT, fpx);
2845               memcpy(keyid, fp + 12, 8);
2846             }
2847         }
2848       if (tag == 2)
2849         {
2850           if (p[0] == 3 && p[1] == 5)
2851             {
2852 #if 0
2853               Id htype = 0;
2854 #endif
2855               /* printf("V3 signature packet\n"); */
2856               if (l < 17)
2857                 continue;
2858               if (p[2] != 0x10 && p[2] != 0x11 && p[2] != 0x12 && p[2] != 0x13 && p[2] != 0x1f)
2859                 continue;
2860               if (!memcmp(keyid, p + 6, 8))
2861                 {
2862                   /* printf("SELF SIG\n"); */
2863                 }
2864               else
2865                 {
2866                   /* printf("OTHER SIG\n"); */
2867                 }
2868 #if 0
2869               if (p[16] == 1)
2870                 htype = REPOKEY_TYPE_MD5;
2871               else if (p[16] == 2)
2872                 htype = REPOKEY_TYPE_SHA1;
2873               else if (p[16] == 8)
2874                 htype = REPOKEY_TYPE_SHA256;
2875               if (htype)
2876                 {
2877                   void *h = solv_chksum_create(htype);
2878                   unsigned char b[3], *cs;
2879
2880                   b[0] = 0x99;
2881                   b[1] = pubkeyl >> 8;
2882                   b[2] = pubkeyl;
2883                   solv_chksum_add(h, b, 3);
2884                   solv_chksum_add(h, pubkey, pubkeyl);
2885                   if (p[2] >= 0x10 && p[2] <= 0x13)
2886                     solv_chksum_add(h, userid, useridl);
2887                   solv_chksum_add(h, p + 2, 5);
2888                   cs = solv_chksum_get(h, 0);
2889                   solv_chksum_free(h, 0);
2890                 }
2891 #endif
2892             }
2893           if (p[0] == 4)
2894             {
2895               int j, ql, haveissuer;
2896               unsigned char *q;
2897               unsigned int ex = 0;
2898 #if 0
2899               unsigned int scr = 0;
2900 #endif
2901               unsigned char issuer[8];
2902
2903               /* printf("V4 signature packet\n"); */
2904               if (l < 6)
2905                 continue;
2906               if (p[1] != 0x10 && p[1] != 0x11 && p[1] != 0x12 && p[1] != 0x13 && p[1] != 0x1f)
2907                 continue;
2908               haveissuer = 0;
2909               ex = 0;
2910               q = p + 4;
2911               for (j = 0; q && j < 2; j++)
2912                 {
2913                   if (q + 2 > p + l)
2914                     {
2915                       q = 0;
2916                       break;
2917                     }
2918                   ql = q[0] << 8 | q[1];
2919                   q += 2;
2920                   if (q + ql > p + l)
2921                     {
2922                       q = 0;
2923                       break;
2924                     }
2925                   while (ql)
2926                     {
2927                       int sl;
2928                       x = *q++;
2929                       ql--;
2930                       if (x < 192)
2931                         sl = x;
2932                       else if (x == 255)
2933                         {
2934                           if (ql < 4)
2935                             {
2936                               q = 0;
2937                               break;
2938                             }
2939                           sl = q[0] << 24 | q[1] << 16 | q[2] << 8 | q[3];
2940                           q += 4;
2941                           ql -= 4;
2942                         }
2943                       else
2944                         {
2945                           if (ql < 1)
2946                             {
2947                               q = 0;
2948                               break;
2949                             }
2950                           sl = ((x - 192) << 8) + *q++ + 192;
2951                           ql--;
2952                         }
2953                       if (ql < sl)
2954                         {
2955                           q = 0;
2956                           break;
2957                         }
2958                       x = q[0] & 127;
2959                       /* printf("%d SIGSUB %d %d\n", j, x, sl); */
2960                       if (x == 16 && sl == 9 && !haveissuer)
2961                         {
2962                           memcpy(issuer, q + 1, 8);
2963                           haveissuer = 1;
2964                         }
2965 #if 0
2966                       if (x == 2 && j == 0)
2967                         scr = q[1] << 24 | q[2] << 16 | q[3] << 8 | q[4];
2968 #endif
2969                       if (x == 9 && j == 0)
2970                         ex = q[1] << 24 | q[2] << 16 | q[3] << 8 | q[4];
2971                       q += sl;
2972                       ql -= sl;
2973                     }
2974                 }
2975               if (ex)
2976                 ex += kcr;
2977               if (haveissuer)
2978                 {
2979 #if 0
2980                   Id htype = 0;
2981                   if (p[3] == 1)
2982                     htype = REPOKEY_TYPE_MD5;
2983                   else if (p[3] == 2)
2984                     htype = REPOKEY_TYPE_SHA1;
2985                   else if (p[3] == 8)
2986                     htype = REPOKEY_TYPE_SHA256;
2987                   if (htype && pubkeyl)
2988                     {
2989                       void *h = solv_chksum_create(htype);
2990                       unsigned char b[6], *cs;
2991                       unsigned int hl;
2992
2993                       b[0] = 0x99;
2994                       b[1] = pubkeyl >> 8;
2995                       b[2] = pubkeyl;
2996                       solv_chksum_add(h, b, 3);
2997                       solv_chksum_add(h, pubkey, pubkeyl);
2998                       if (p[1] >= 0x10 && p[1] <= 0x13)
2999                         {
3000                           b[0] = 0xb4;
3001                           b[1] = useridl >> 24;
3002                           b[2] = useridl >> 16;
3003                           b[3] = useridl >> 8;
3004                           b[4] = useridl;
3005                           solv_chksum_add(h, b, 5);
3006                           solv_chksum_add(h, userid, useridl);
3007                         }
3008                       hl = 6 + (p[4] << 8 | p[5]);
3009                       solv_chksum_add(h, p, hl);
3010                       b[0] = 4;
3011                       b[1] = 0xff;
3012                       b[2] = hl >> 24;
3013                       b[3] = hl >> 16;
3014                       b[4] = hl >> 8;
3015                       b[5] = hl;
3016                       solv_chksum_add(h, b, 6);
3017                       cs = solv_chksum_get(h, 0);
3018                       solv_chksum_free(h, 0);
3019                     }
3020 #endif
3021                   if (!memcmp(keyid, issuer, 8))
3022                     {
3023                       /* printf("SELF SIG cr %d ex %d\n", cr, ex); */
3024                       if (ex > maxex)
3025                         maxex = ex;
3026                     }
3027                   else
3028                     {
3029                       /* printf("OTHER SIG cr %d ex %d\n", cr, ex); */
3030                     }
3031                 }
3032             }
3033         }
3034       if (tag == 13)
3035         {
3036           userid = solv_realloc(userid, l);
3037           if (l)
3038             memcpy(userid, p, l);
3039 #if 0
3040           useridl = l;
3041 #endif
3042         }
3043     }
3044   if (maxex)
3045     repodata_set_num(data, s - s->repo->pool->solvables, PUBKEY_EXPIRES, maxex);
3046   solv_free(pubkey);
3047   solv_free(userid);
3048 }
3049
3050 /* this is private to rpm, but rpm lacks an interface to retrieve
3051  * the values. Sigh. */
3052 struct pgpDigParams_s {
3053     const char * userid;
3054     const unsigned char * hash;
3055     const char * params[4];
3056     unsigned char tag;
3057     unsigned char version;               /*!< version number. */
3058     unsigned char time[4];               /*!< time that the key was created. */
3059     unsigned char pubkey_algo;           /*!< public key algorithm. */
3060     unsigned char hash_algo;
3061     unsigned char sigtype;
3062     unsigned char hashlen;
3063     unsigned char signhash16[2];
3064     unsigned char signid[8];
3065     unsigned char saved;
3066 };
3067
3068 struct pgpDig_s {
3069     struct pgpDigParams_s signature;
3070     struct pgpDigParams_s pubkey;
3071 };
3072
3073 static int
3074 pubkey2solvable(Solvable *s, Repodata *data, char *pubkey)
3075 {
3076   Pool *pool = s->repo->pool;
3077   unsigned char *pkts;
3078   unsigned int btime;
3079   int pktsl, i;
3080   pgpDig dig = 0;
3081   char keyid[16 + 1];
3082   char evrbuf[8 + 1 + 8 + 1];
3083
3084   pkts = unarmor(pubkey, &pktsl);
3085   if (!pkts)
3086     return 0;
3087   setutf8string(data, s - s->repo->pool->solvables, SOLVABLE_DESCRIPTION, pubkey);
3088   parsekeydata(s, data, pkts, pktsl);
3089   /* only rpm knows how to do the release calculation, we don't dare
3090    * to recreate all the bugs */
3091 #ifndef RPM5
3092   dig = pgpNewDig();
3093 #else
3094   dig = pgpDigNew(RPMVSF_DEFAULT, 0);
3095 #endif
3096   (void) pgpPrtPkts(pkts, pktsl, dig, 0);
3097   btime = dig->pubkey.time[0] << 24 | dig->pubkey.time[1] << 16 | dig->pubkey.time[2] << 8 | dig->pubkey.signid[3];
3098   sprintf(evrbuf, "%02x%02x%02x%02x-%02x%02x%02x%02x", dig->pubkey.signid[4], dig->pubkey.signid[5], dig->pubkey.signid[6], dig->pubkey.signid[7], dig->pubkey.time[0], dig->pubkey.time[1], dig->pubkey.time[2], dig->pubkey.time[3]);
3099   repodata_set_num(data, s - s->repo->pool->solvables, SOLVABLE_BUILDTIME, btime);
3100
3101   s->name = pool_str2id(pool, "gpg-pubkey", 1);
3102   s->evr = pool_str2id(pool, evrbuf, 1);
3103   s->arch = 1;
3104   for (i = 0; i < 8; i++)
3105     sprintf(keyid + 2 * i, "%02x", dig->pubkey.signid[i]);
3106   repodata_set_str(data, s - s->repo->pool->solvables, PUBKEY_KEYID, keyid);
3107   if (dig->pubkey.userid)
3108     setutf8string(data, s - s->repo->pool->solvables, SOLVABLE_SUMMARY, dig->pubkey.userid);
3109 #ifndef RPM5
3110   (void)pgpFreeDig(dig);
3111 #else
3112   (void)pgpDigFree(dig);
3113 #endif
3114   solv_free((void *)pkts);
3115   return 1;
3116 }
3117
3118 int
3119 repo_add_rpmdb_pubkeys(Repo *repo, const char *rootdir, int flags)
3120 {
3121   Pool *pool = repo->pool;
3122   struct rpm_by_state state;
3123   struct rpmdbentry *entries;
3124   int nentries, i;
3125   char *namedata, *str;
3126   unsigned int u32;
3127   Repodata *data;
3128   Solvable *s;
3129
3130   data = repo_add_repodata(repo, flags);
3131
3132   memset(&state, 0, sizeof(state));
3133   if (!(state.dbenv = opendbenv(rootdir)))
3134     return 0;
3135   entries = getinstalledrpmdbids(&state, "Name", "gpg-pubkey", &nentries, &namedata);
3136   for (i = 0 ; i < nentries; i++)
3137     {
3138       void *statep = &state;
3139       RpmHead *rpmhead = rpm_byrpmdbid(entries[i].rpmdbid, rootdir, &statep);
3140       if (!rpmhead)
3141         continue;
3142       str = headstring(rpmhead, TAG_DESCRIPTION);
3143       if (!str)
3144         continue;
3145       s = pool_id2solvable(pool, repo_add_solvable(repo));
3146       pubkey2solvable(s, data, str);
3147       u32 = headint32(rpmhead, TAG_INSTALLTIME);
3148       if (u32)
3149         repodata_set_num(data, s - pool->solvables, SOLVABLE_INSTALLTIME, u32);
3150       if (!repo->rpmdbid)
3151         repo->rpmdbid = repo_sidedata_create(repo, sizeof(Id));
3152       repo->rpmdbid[s - pool->solvables - repo->start] = entries[i].rpmdbid;
3153     }
3154   solv_free(entries);
3155   solv_free(namedata);
3156   freestate(&state);
3157   if (!(flags & REPO_NO_INTERNALIZE))
3158     repodata_internalize(data);
3159   return 0;
3160 }
3161
3162 Id
3163 repo_add_pubkey(Repo *repo, const char *key, int flags)
3164 {
3165   Pool *pool = repo->pool;
3166   Repodata *data;
3167   Solvable *s;
3168   char *buf;
3169   int bufl, l, ll;
3170   FILE *fp;
3171
3172   data = repo_add_repodata(repo, flags);
3173   buf = 0;
3174   bufl = 0;
3175   if ((fp = fopen(key, "r")) == 0)
3176     {
3177       pool_error(pool, -1, "%s: %s", key, strerror(errno));
3178       return 0;
3179     }
3180   for (l = 0; ;)
3181     {
3182       if (bufl - l < 4096)
3183         {
3184           bufl += 4096;
3185           buf = solv_realloc(buf, bufl);
3186         }
3187       ll = fread(buf, 1, bufl - l, fp);
3188       if (ll < 0)
3189         {
3190           fclose(fp);
3191           pool_error(pool, -1, "%s: %s", key, strerror(errno));
3192           return 0;
3193         }
3194       if (ll == 0)
3195         break;
3196       l += ll;
3197     }
3198   buf[l] = 0;
3199   fclose(fp);
3200   s = pool_id2solvable(pool, repo_add_solvable(repo));
3201   if (!pubkey2solvable(s, data, buf))
3202     {
3203       repo_free_solvable(repo, s - pool->solvables, 1);
3204       solv_free(buf);
3205       return 0;
3206     }
3207   solv_free(buf);
3208   if (!(flags & REPO_NO_INTERNALIZE))
3209     repodata_internalize(data);
3210   return s - pool->solvables;
3211 }
3212
3213 #endif /* ENABLE_RPMDB_PUBKEY */