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 #ifdef ENABLE_RPMDB
  28
  29 #include <rpm/rpmio.h>
  30 #include <rpm/rpmpgp.h>
  31 #ifndef RPM5
  32 #include <rpm/header.h>
  33 #endif
  34 #include <rpm/rpmdb.h>
  35
  36 #ifndef DB_CREATE
  37 # if defined(SUSE) || defined(HAVE_RPM_DB_H)
  38 #  include <rpm/db.h>
  39 # else
  40 #  include <db.h>
  41 # endif
  42 #endif
  43
  44 #endif
  45
  46 #include "pool.h"
  47 #include "repo.h"
  48 #include "hash.h"
  49 #include "util.h"
  50 #include "queue.h"
  51 #include "chksum.h"
  52 #include "repo_rpmdb.h"
  53 #include "repo_solv.h"
  54 #ifdef ENABLE_COMPLEX_DEPS
  55 #include "pool_parserpmrichdep.h"
  56 #endif
  57
  58 /* 3: added triggers */
  59 /* 4: fixed triggers */
  60 /* 5: fixed checksum copying */
  61 /* 6: add SOLVABLE_PREREQ_IGNOREINST support */
  62 /* 7: fix bug in ignoreinst logic */
  63 #define RPMDB_COOKIE_VERSION 7
  64
  65 #define TAG_NAME                1000
  66 #define TAG_VERSION             1001
  67 #define TAG_RELEASE             1002
  68 #define TAG_EPOCH               1003
  69 #define TAG_SUMMARY             1004
  70 #define TAG_DESCRIPTION         1005
  71 #define TAG_BUILDTIME           1006
  72 #define TAG_BUILDHOST           1007
  73 #define TAG_INSTALLTIME         1008
  74 #define TAG_SIZE                1009
  75 #define TAG_DISTRIBUTION        1010
  76 #define TAG_VENDOR              1011
  77 #define TAG_LICENSE             1014
  78 #define TAG_PACKAGER            1015
  79 #define TAG_GROUP               1016
  80 #define TAG_URL                 1020
  81 #define TAG_ARCH                1022
  82 #define TAG_FILESIZES           1028
  83 #define TAG_FILEMODES           1030
  84 #define TAG_FILEMD5S            1035
  85 #define TAG_FILELINKTOS         1036
  86 #define TAG_FILEFLAGS           1037
  87 #define TAG_SOURCERPM           1044
  88 #define TAG_PROVIDENAME         1047
  89 #define TAG_REQUIREFLAGS        1048
  90 #define TAG_REQUIRENAME         1049
  91 #define TAG_REQUIREVERSION      1050
  92 #define TAG_NOSOURCE            1051
  93 #define TAG_NOPATCH             1052
  94 #define TAG_CONFLICTFLAGS       1053
  95 #define TAG_CONFLICTNAME        1054
  96 #define TAG_CONFLICTVERSION     1055
  97 #define TAG_TRIGGERNAME         1066
  98 #define TAG_TRIGGERVERSION      1067
  99 #define TAG_TRIGGERFLAGS        1068
 100 #define TAG_CHANGELOGTIME       1080
 101 #define TAG_CHANGELOGNAME       1081
 102 #define TAG_CHANGELOGTEXT       1082
 103 #define TAG_OBSOLETENAME        1090
 104 #define TAG_FILEDEVICES         1095
 105 #define TAG_FILEINODES          1096
 106 #define TAG_SOURCEPACKAGE       1106
 107 #define TAG_PROVIDEFLAGS        1112
 108 #define TAG_PROVIDEVERSION      1113
 109 #define TAG_OBSOLETEFLAGS       1114
 110 #define TAG_OBSOLETEVERSION     1115
 111 #define TAG_DIRINDEXES          1116
 112 #define TAG_BASENAMES           1117
 113 #define TAG_DIRNAMES            1118
 114 #define TAG_PAYLOADFORMAT       1124
 115 #define TAG_PATCHESNAME         1133
 116 #define TAG_FILECOLORS          1140
 117 #define TAG_OLDSUGGESTSNAME     1156
 118 #define TAG_OLDSUGGESTSVERSION  1157
 119 #define TAG_OLDSUGGESTSFLAGS    1158
 120 #define TAG_OLDENHANCESNAME     1159
 121 #define TAG_OLDENHANCESVERSION  1160
 122 #define TAG_OLDENHANCESFLAGS    1161
 123
 124 /* rpm5 tags */
 125 #define TAG_DISTEPOCH           1218
 126
 127 /* rpm4 tags */
 128 #define TAG_LONGFILESIZES       5008
 129 #define TAG_LONGSIZE            5009
 130 #define TAG_RECOMMENDNAME       5046
 131 #define TAG_RECOMMENDVERSION    5047
 132 #define TAG_RECOMMENDFLAGS      5048
 133 #define TAG_SUGGESTNAME         5049
 134 #define TAG_SUGGESTVERSION      5050
 135 #define TAG_SUGGESTFLAGS        5051
 136 #define TAG_SUPPLEMENTNAME      5052
 137 #define TAG_SUPPLEMENTVERSION   5053
 138 #define TAG_SUPPLEMENTFLAGS     5054
 139 #define TAG_ENHANCENAME         5055
 140 #define TAG_ENHANCEVERSION      5056
 141 #define TAG_ENHANCEFLAGS        5057
 142
 143 /* signature tags */
 144 #define TAG_SIGBASE             256
 145 #define TAG_SIGMD5              (TAG_SIGBASE + 5)
 146 #define TAG_SHA1HEADER          (TAG_SIGBASE + 13)
 147
 148 #define SIGTAG_SIZE             1000
 149 #define SIGTAG_PGP              1002    /* RSA signature */
 150 #define SIGTAG_MD5              1004    /* header+payload md5 checksum */
 151 #define SIGTAG_GPG              1005    /* DSA signature */
 152
 153 #define DEP_LESS                (1 << 1)
 154 #define DEP_GREATER             (1 << 2)
 155 #define DEP_EQUAL               (1 << 3)
 156 #define DEP_STRONG              (1 << 27)
 157 #define DEP_PRE_IN              ((1 << 6) | (1 << 9) | (1 << 10))
 158 #define DEP_PRE_UN              ((1 << 6) | (1 << 11) | (1 << 12))
 159
 160 #define FILEFLAG_GHOST          (1 <<  6)
 161
 162
 163 #ifdef RPM5
 164 # define RPM_INDEX_SIZE 4       /* just the rpmdbid */
 165 #else
 166 # define RPM_INDEX_SIZE 8       /* rpmdbid + array index */
 167 #endif
 168
 169 /* some limits to guard against corrupt rpms */
 170 #define MAX_SIG_CNT             0x100000
 171 #define MAX_SIG_DSIZE           0x100000
 172
 173 #define MAX_HDR_CNT             0x100000
 174 #define MAX_HDR_DSIZE           0x2000000
 175
 176 typedef struct rpmhead {
 177   int cnt;
 178   unsigned int dcnt;
 179   unsigned char *dp;
 180   int forcebinary;              /* sigh, see rh#478907 */
 181   unsigned char data[1];
 182 } RpmHead;
 183
 184
 185 static inline unsigned char *
 186 headfindtag(RpmHead *h, int tag)
 187 {
 188   unsigned int i;
 189   unsigned char *d, taga[4];
 190   d = h->dp - 16;
 191   taga[0] = tag >> 24;
 192   taga[1] = tag >> 16;
 193   taga[2] = tag >> 8;
 194   taga[3] = tag;
 195   for (i = 0; i < h->cnt; i++, d -= 16)
 196     if (d[3] == taga[3] && d[2] == taga[2] && d[1] == taga[1] && d[0] == taga[0])
 197       return d;
 198   return 0;
 199 }
 200
 201 static int
 202 headexists(RpmHead *h, int tag)
 203 {
 204   return headfindtag(h, tag) ? 1 : 0;
 205 }
 206
 207 static unsigned int *
 208 headint32array(RpmHead *h, int tag, int *cnt)
 209 {
 210   unsigned int i, o, *r;
 211   unsigned char *d = headfindtag(h, tag);
 212
 213   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 4)
 214     return 0;
 215   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
 216   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
 217   if (o > h->dcnt || i > h->dcnt || o + 4 * i > h->dcnt)
 218     return 0;
 219   d = h->dp + o;
 220   r = solv_calloc(i ? i : 1, sizeof(unsigned int));
 221   if (cnt)
 222     *cnt = i;
 223   for (o = 0; o < i; o++, d += 4)
 224     r[o] = d[0] << 24 | d[1] << 16 | d[2] << 8 | d[3];
 225   return r;
 226 }
 227
 228 /* returns the first entry of an integer array */
 229 static unsigned int
 230 headint32(RpmHead *h, int tag)
 231 {
 232   unsigned int i, o;
 233   unsigned char *d = headfindtag(h, tag);
 234
 235   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 4)
 236     return 0;
 237   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
 238   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
 239   if (i == 0 || o > h->dcnt || i > h->dcnt || o + 4 * i > h->dcnt)
 240     return 0;
 241   d = h->dp + o;
 242   return d[0] << 24 | d[1] << 16 | d[2] << 8 | d[3];
 243 }
 244
 245 static unsigned long long *
 246 headint64array(RpmHead *h, int tag, int *cnt)
 247 {
 248   unsigned int i, o;
 249   unsigned long long *r;
 250   unsigned char *d = headfindtag(h, tag);
 251
 252   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 5)
 253     return 0;
 254   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
 255   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
 256   if (o > h->dcnt || i > h->dcnt || o + 8 * i > h->dcnt)
 257     return 0;
 258   d = h->dp + o;
 259   r = solv_calloc(i ? i : 1, sizeof(unsigned long long));
 260   if (cnt)
 261     *cnt = i;
 262   for (o = 0; o < i; o++, d += 8)
 263     {
 264       unsigned int x = d[0] << 24 | d[1] << 16 | d[2] << 8 | d[3];
 265       r[o] = (unsigned long long)x << 32 | (d[4] << 24 | d[5] << 16 | d[6] << 8 | d[7]);
 266     }
 267   return r;
 268 }
 269
 270 /* returns the first entry of an 64bit integer array */
 271 static unsigned long long
 272 headint64(RpmHead *h, int tag)
 273 {
 274   unsigned int i, o;
 275   unsigned char *d = headfindtag(h, tag);
 276   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 5)
 277     return 0;
 278   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
 279   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
 280   if (i == 0 || o > h->dcnt || i > h->dcnt || o + 8 * i > h->dcnt)
 281     return 0;
 282   d = h->dp + o;
 283   i = d[0] << 24 | d[1] << 16 | d[2] << 8 | d[3];
 284   return (unsigned long long)i << 32 | (d[4] << 24 | d[5] << 16 | d[6] << 8 | d[7]);
 285 }
 286
 287 static unsigned int *
 288 headint16array(RpmHead *h, int tag, int *cnt)
 289 {
 290   unsigned int i, o, *r;
 291   unsigned char *d = headfindtag(h, tag);
 292
 293   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 3)
 294     return 0;
 295   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
 296   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
 297   if (o > h->dcnt || i > h->dcnt || o + 2 * i > h->dcnt)
 298     return 0;
 299   d = h->dp + o;
 300   r = solv_calloc(i ? i : 1, sizeof(unsigned int));
 301   if (cnt)
 302     *cnt = i;
 303   for (o = 0; o < i; o++, d += 2)
 304     r[o] = d[0] << 8 | d[1];
 305   return r;
 306 }
 307
 308 static char *
 309 headstring(RpmHead *h, int tag)
 310 {
 311   unsigned int o;
 312   unsigned char *d = headfindtag(h, tag);
 313   /* 6: STRING, 9: I18NSTRING */
 314   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || (d[7] != 6 && d[7] != 9))
 315     return 0;
 316   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
 317   if (o >= h->dcnt)
 318     return 0;
 319   return (char *)h->dp + o;
 320 }
 321
 322 static char **
 323 headstringarray(RpmHead *h, int tag, int *cnt)
 324 {
 325   unsigned int i, o;
 326   unsigned char *d = headfindtag(h, tag);
 327   char **r;
 328
 329   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 8)
 330     return 0;
 331   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
 332   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
 333   if (o > h->dcnt || i > h->dcnt)
 334     return 0;
 335   r = solv_calloc(i ? i : 1, sizeof(char *));
 336   if (cnt)
 337     *cnt = i;
 338   d = h->dp + o;
 339   for (o = 0; o < i; o++)
 340     {
 341       r[o] = (char *)d;
 342       if (o + 1 < i)
 343         d += strlen((char *)d) + 1;
 344       if (d >= h->dp + h->dcnt)
 345         {
 346           solv_free(r);
 347           return 0;
 348         }
 349     }
 350   return r;
 351 }
 352
 353 static unsigned char *
 354 headbinary(RpmHead *h, int tag, unsigned int *sizep)
 355 {
 356   unsigned int i, o;
 357   unsigned char *d = headfindtag(h, tag);
 358   if (!d || d[4] != 0 || d[5] != 0 || d[6] != 0 || d[7] != 7)
 359     return 0;
 360   o = d[8] << 24 | d[9] << 16 | d[10] << 8 | d[11];
 361   i = d[12] << 24 | d[13] << 16 | d[14] << 8 | d[15];
 362   if (o > h->dcnt || i > h->dcnt || o + i > h->dcnt)
 363     return 0;
 364   if (sizep)
 365     *sizep = i;
 366   return h->dp + o;
 367 }
 368
 369 static char *headtoevr(RpmHead *h)
 370 {
 371   unsigned int epoch;
 372   char *version, *v;
 373   char *release;
 374   char *evr;
 375   char *distepoch;
 376
 377   version  = headstring(h, TAG_VERSION);
 378   release  = headstring(h, TAG_RELEASE);
 379   epoch = headint32(h, TAG_EPOCH);
 380   if (!version || !release)
 381     {
 382       fprintf(stderr, "headtoevr: bad rpm header\n");
 383       return 0;
 384     }
 385   for (v = version; *v >= '0' && *v <= '9'; v++)
 386     ;
 387   if (epoch || (v != version && *v == ':'))
 388     {
 389       char epochbuf[11];        /* 32bit decimal will fit in */
 390       sprintf(epochbuf, "%u", epoch);
 391       evr = solv_malloc(strlen(epochbuf) + 1 + strlen(version) + 1 + strlen(release) + 1);
 392       sprintf(evr, "%s:%s-%s", epochbuf, version, release);
 393     }
 394   else
 395     {
 396       evr = solv_malloc(strlen(version) + 1 + strlen(release) + 1);
 397       sprintf(evr, "%s-%s", version, release);
 398     }
 399   distepoch = headstring(h, TAG_DISTEPOCH);
 400   if (distepoch && *distepoch)
 401     {
 402       int l = strlen(evr);
 403       evr = solv_realloc(evr, l + strlen(distepoch) + 2);
 404       evr[l++] = ':';
 405       strcpy(evr + l, distepoch);
 406     }
 407   return evr;
 408 }
 409
 410
 411 static void
 412 setutf8string(Repodata *repodata, Id handle, Id tag, const char *str)
 413 {
 414   if (str[solv_validutf8(str)])
 415     {
 416       char *ustr = solv_latin1toutf8(str);      /* not utf8, assume latin1 */
 417       repodata_set_str(repodata, handle, tag, ustr);
 418       solv_free(ustr);
 419     }
 420   else
 421     repodata_set_str(repodata, handle, tag, str);
 422 }
 423
 424 static int
 425 ignq_sortcmp(const void *va, const void *vb, void *dp)
 426 {
 427   int r = *(Id *)va - *(Id *)vb;
 428   if (!r)
 429     r = ((Id *)va)[1] - ((Id *)vb)[1];
 430   return r;
 431 }
 432
 433 /*
 434  * strong: 0: ignore strongness
 435  *         1: filter to strong
 436  *         2: filter to weak
 437  */
 438 static unsigned int
 439 makedeps(Pool *pool, Repo *repo, RpmHead *rpmhead, int tagn, int tagv, int tagf, int flags, Queue *ignq)
 440 {
 441   char **n, **v;
 442   unsigned int *f;
 443   int i, cc, nc, vc, fc;
 444   int haspre, premask, has_ign;
 445   unsigned int olddeps;
 446   Id *ida;
 447   int strong = 0;
 448
 449   n = headstringarray(rpmhead, tagn, &nc);
 450   if (!n)
 451     {
 452       switch (tagn)
 453         {
 454         case TAG_SUGGESTNAME:
 455           tagn = TAG_OLDSUGGESTSNAME;
 456           tagv = TAG_OLDSUGGESTSVERSION;
 457           tagf = TAG_OLDSUGGESTSFLAGS;
 458           strong = -1;
 459           break;
 460         case TAG_ENHANCENAME:
 461           tagn = TAG_OLDENHANCESNAME;
 462           tagv = TAG_OLDENHANCESVERSION;
 463           tagf = TAG_OLDENHANCESFLAGS;
 464           strong = -1;
 465           break;
 466         case TAG_RECOMMENDNAME:
 467           tagn = TAG_OLDSUGGESTSNAME;
 468           tagv = TAG_OLDSUGGESTSVERSION;
 469           tagf = TAG_OLDSUGGESTSFLAGS;
 470           strong = 1;
 471           break;
 472         case TAG_SUPPLEMENTNAME:
 473           tagn = TAG_OLDENHANCESNAME;
 474           tagv = TAG_OLDENHANCESVERSION;
 475           tagf = TAG_OLDENHANCESFLAGS;
 476           strong = 1;
 477           break;
 478         default:
 479           return 0;
 480         }
 481       n = headstringarray(rpmhead, tagn, &nc);
 482     }
 483   if (!n || !nc)
 484     return 0;
 485   vc = fc = 0;
 486   v = headstringarray(rpmhead, tagv, &vc);
 487   f = headint32array(rpmhead, tagf, &fc);
 488   if (!v || !f || nc != vc || nc != fc)
 489     {
 490       char *pkgname = rpm_query(rpmhead, 0);
 491       pool_error(pool, 0, "bad dependency entries for %s: %d %d %d", pkgname ? pkgname : "<NULL>", nc, vc, fc);
 492       solv_free(pkgname);
 493       solv_free(n);
 494       solv_free(v);
 495       solv_free(f);
 496       return 0;
 497     }
 498
 499   cc = nc;
 500   haspre = 0;   /* add no prereq marker */
 501   premask = tagn == TAG_REQUIRENAME ? DEP_PRE_IN | DEP_PRE_UN : 0;
 502   if ((flags & RPM_ADD_NO_RPMLIBREQS) || strong)
 503     {
 504       /* we do filtering */
 505       cc = 0;
 506       for (i = 0; i < nc; i++)
 507         {
 508           if (strong && (f[i] & DEP_STRONG) != (strong < 0 ? 0 : DEP_STRONG))
 509             continue;
 510           if ((flags & RPM_ADD_NO_RPMLIBREQS) != 0)
 511             if (!strncmp(n[i], "rpmlib(", 7))
 512               continue;
 513           if ((f[i] & premask) != 0)
 514             haspre = 1;
 515           cc++;
 516         }
 517     }
 518   else if (premask)
 519     {
 520       /* no filtering, just look for the first prereq */
 521       for (i = 0; i < nc; i++)
 522         if ((f[i] & premask) != 0)
 523           {
 524             haspre = 1;
 525             break;
 526           }
 527     }
 528   if (cc == 0)
 529     {
 530       solv_free(n);
 531       solv_free(v);
 532       solv_free(f);
 533       return 0;
 534     }
 535   cc += haspre;         /* add slot for the prereq marker */
 536   olddeps = repo_reserve_ids(repo, 0, cc);
 537   ida = repo->idarraydata + olddeps;
 538
 539   has_ign = 0;
 540   for (i = 0; ; i++)
 541     {
 542       Id id;
 543       if (i == nc)
 544         {
 545           if (haspre != 1)
 546             break;
 547           haspre = 2;   /* pass two: prereqs */
 548           i = 0;
 549           *ida++ = SOLVABLE_PREREQMARKER;
 550         }
 551       if (strong && (f[i] & DEP_STRONG) != (strong < 0 ? 0 : DEP_STRONG))
 552         continue;
 553       if (haspre)
 554         {
 555           if (haspre == 1 && (f[i] & premask) != 0)
 556             continue;
 557           if (haspre == 2 && (f[i] & premask) == 0)
 558             continue;
 559         }
 560       if ((flags & RPM_ADD_NO_RPMLIBREQS) != 0)
 561         if (!strncmp(n[i], "rpmlib(", 7))
 562           continue;
 563 #ifdef ENABLE_COMPLEX_DEPS
 564       if ((f[i] & (DEP_LESS|DEP_EQUAL|DEP_GREATER)) == 0 && n[i][0] == '(')
 565         {
 566           id = pool_parserpmrichdep(pool, n[i]);
 567           if (id)
 568             *ida++ = id;
 569           else
 570             cc--;
 571           continue;
 572         }
 573 #endif
 574       id = pool_str2id(pool, n[i], 1);
 575       if (f[i] & (DEP_LESS|DEP_GREATER|DEP_EQUAL))
 576         {
 577           Id evr;
 578           int fl = 0;
 579           if ((f[i] & DEP_LESS) != 0)
 580             fl |= REL_LT;
 581           if ((f[i] & DEP_EQUAL) != 0)
 582             fl |= REL_EQ;
 583           if ((f[i] & DEP_GREATER) != 0)
 584             fl |= REL_GT;
 585           if (v[i][0] == '0' && v[i][1] == ':' && v[i][2])
 586             evr = pool_str2id(pool, v[i] + 2, 1);
 587           else
 588             evr = pool_str2id(pool, v[i], 1);
 589           id = pool_rel2id(pool, id, evr, fl, 1);
 590         }
 591       *ida++ = id;
 592       if (haspre == 2 && ignq)
 593         {
 594           int is_ign = (f[i] & DEP_PRE_IN) != 0 && (f[i] & DEP_PRE_UN) == 0 ? 1 : 0;
 595           has_ign |= is_ign;
 596           queue_push2(ignq, id, is_ign);
 597         }
 598     }
 599   *ida++ = 0;
 600   repo->idarraysize += cc + 1;
 601   solv_free(n);
 602   solv_free(v);
 603   solv_free(f);
 604   if (ignq && ignq->count)
 605     {
 606       int j = 0;
 607       if (has_ign && ignq->count == 2)
 608         j = 1;
 609       else if (has_ign)
 610         {
 611           Id id, lastid = 0;
 612
 613           solv_sort(ignq->elements, ignq->count / 2, sizeof(Id) * 2, ignq_sortcmp, 0);
 614           for (i = j = 0; i < ignq->count; i += 2)
 615             {
 616               id = ignq->elements[i];
 617               if (id != lastid && ignq->elements[i + 1] > 0)
 618                 ignq->elements[j++] = id;
 619               lastid = id;
 620             }
 621         }
 622       queue_truncate(ignq, j);
 623     }
 624   return olddeps;
 625 }
 626
 627
 628 static void
 629 adddudata(Repodata *data, Id handle, RpmHead *rpmhead, char **dn, unsigned int *di, int fc, int dc)
 630 {
 631   Id did;
 632   int i, fszc;
 633   unsigned int *fkb, *fn, *fsz, *fm, *fino;
 634   unsigned long long *fsz64;
 635   unsigned int inotest[256], inotestok;
 636
 637   if (!fc)
 638     return;
 639   if ((fsz64 = headint64array(rpmhead, TAG_LONGFILESIZES, &fszc)) != 0)
 640     {
 641       /* convert to kbyte */
 642       fsz = solv_malloc2(fszc, sizeof(*fsz));
 643       for (i = 0; i < fszc; i++)
 644         fsz[i] = fsz64[i] ? fsz64[i] / 1024 + 1 : 0;
 645       solv_free(fsz64);
 646     }
 647   else if ((fsz = headint32array(rpmhead, TAG_FILESIZES, &fszc)) != 0)
 648     {
 649       /* convert to kbyte */
 650       for (i = 0; i < fszc; i++)
 651         if (fsz[i])
 652           fsz[i] = fsz[i] / 1024 + 1;
 653     }
 654   else
 655     return;
 656   if (fc != fszc)
 657     {
 658       solv_free(fsz);
 659       return;
 660     }
 661
 662   /* stupid rpm records sizes of directories, so we have to check the mode */
 663   fm = headint16array(rpmhead, TAG_FILEMODES, &fszc);
 664   if (!fm || fc != fszc)
 665     {
 666       solv_free(fsz);
 667       solv_free(fm);
 668       return;
 669     }
 670   fino = headint32array(rpmhead, TAG_FILEINODES, &fszc);
 671   if (!fino || fc != fszc)
 672     {
 673       solv_free(fsz);
 674       solv_free(fm);
 675       solv_free(fino);
 676       return;
 677     }
 678
 679   /* kill hardlinked entries */
 680   inotestok = 0;
 681   if (fc < sizeof(inotest))
 682     {
 683       /* quick test just hashing the inode numbers */
 684       memset(inotest, 0, sizeof(inotest));
 685       for (i = 0; i < fc; i++)
 686         {
 687           int off, bit;
 688           if (fsz[i] == 0 || !S_ISREG(fm[i]))
 689             continue;   /* does not matter */
 690           off = (fino[i] >> 5) & (sizeof(inotest)/sizeof(*inotest) - 1);
 691           bit = 1 << (fino[i] & 31);
 692           if ((inotest[off] & bit) != 0)
 693             break;
 694           inotest[off] |= bit;
 695         }
 696       if (i == fc)
 697         inotestok = 1;  /* no conflict found */
 698     }
 699   if (!inotestok)
 700     {
 701       /* hardlinked files are possible, check ino/dev pairs */
 702       unsigned int *fdev = headint32array(rpmhead, TAG_FILEDEVICES, &fszc);
 703       unsigned int *fx, j;
 704       unsigned int mask, hash, hh;
 705       if (!fdev || fc != fszc)
 706         {
 707           solv_free(fsz);
 708           solv_free(fm);
 709           solv_free(fdev);
 710           solv_free(fino);
 711           return;
 712         }
 713       mask = fc;
 714       while ((mask & (mask - 1)) != 0)
 715         mask = mask & (mask - 1);
 716       mask <<= 2;
 717       if (mask > sizeof(inotest)/sizeof(*inotest))
 718         fx = solv_calloc(mask, sizeof(unsigned int));
 719       else
 720         {
 721           fx = inotest;
 722           memset(fx, 0, mask * sizeof(unsigned int));
 723         }
 724       mask--;
 725       for (i = 0; i < fc; i++)
 726         {
 727           if (fsz[i] == 0 || !S_ISREG(fm[i]))
 728             continue;
 729           hash = (fino[i] + fdev[i] * 31) & mask;
 730           hh = 7;
 731           while ((j = fx[hash]) != 0)
 732             {
 733               if (fino[j - 1] == fino[i] && fdev[j - 1] == fdev[i])
 734                 {
 735                   fsz[i] = 0;   /* kill entry */
 736                   break;
 737                 }
 738               hash = (hash + hh++) & mask;
 739             }
 740           if (!j)
 741             fx[hash] = i + 1;
 742         }
 743       if (fx != inotest)
 744         solv_free(fx);
 745       solv_free(fdev);
 746     }
 747   solv_free(fino);
 748
 749   /* sum up inode count and kbytes for each directory */
 750   fn = solv_calloc(dc, sizeof(unsigned int));
 751   fkb = solv_calloc(dc, sizeof(unsigned int));
 752   for (i = 0; i < fc; i++)
 753     {
 754       if (di[i] >= dc)
 755         continue;       /* corrupt entry */
 756       fn[di[i]]++;
 757       if (fsz[i] == 0 || !S_ISREG(fm[i]))
 758         continue;
 759       fkb[di[i]] += fsz[i];
 760     }
 761   solv_free(fsz);
 762   solv_free(fm);
 763   /* commit */
 764   for (i = 0; i < dc; i++)
 765     {
 766       if (!fn[i])
 767         continue;
 768       if (!*dn[i])
 769         {
 770           Solvable *s = data->repo->pool->solvables + handle;
 771           if (s->arch == ARCH_SRC || s->arch == ARCH_NOSRC)
 772             did = repodata_str2dir(data, "/usr/src", 1);
 773           else
 774             continue;   /* work around rpm bug */
 775         }
 776       else
 777         did = repodata_str2dir(data, dn[i], 1);
 778       repodata_add_dirnumnum(data, handle, SOLVABLE_DISKUSAGE, did, fkb[i], fn[i]);
 779     }
 780   solv_free(fn);
 781   solv_free(fkb);
 782 }
 783
 784 static int
 785 is_filtered(const char *dir)
 786 {
 787   if (!dir)
 788     return 1;
 789   /* the dirs always have a trailing / in rpm */
 790   if (strstr(dir, "bin/"))
 791     return 0;
 792   if (!strncmp(dir, "/etc/", 5))
 793     return 0;
 794   if (!strcmp(dir, "/usr/lib/"))
 795     return 2;
 796   return 1;
 797 }
 798
 799 static void
 800 addfilelist(Repodata *data, Id handle, RpmHead *rpmhead, int flags)
 801 {
 802   char **bn;
 803   char **dn;
 804   unsigned int *di;
 805   int bnc, dnc, dic;
 806   int i;
 807   Id lastdid = 0;
 808   unsigned int lastdii = -1;
 809   int lastfiltered = 0;
 810
 811   if (!data)
 812     return;
 813   bn = headstringarray(rpmhead, TAG_BASENAMES, &bnc);
 814   if (!bn)
 815     return;
 816   dn = headstringarray(rpmhead, TAG_DIRNAMES, &dnc);
 817   if (!dn)
 818     {
 819       solv_free(bn);
 820       return;
 821     }
 822   di = headint32array(rpmhead, TAG_DIRINDEXES, &dic);
 823   if (!di)
 824     {
 825       solv_free(bn);
 826       solv_free(dn);
 827       return;
 828     }
 829   if (bnc != dic)
 830     {
 831       pool_error(data->repo->pool, 0, "bad filelist");
 832       return;
 833     }
 834
 835   adddudata(data, handle, rpmhead, dn, di, bnc, dnc);
 836
 837   for (i = 0; i < bnc; i++)
 838     {
 839       Id did;
 840       char *b = bn[i];
 841
 842       if (lastdid && di[i] == lastdii)
 843         did = lastdid;
 844       else
 845         {
 846           if (di[i] >= dnc)
 847             continue;   /* corrupt entry */
 848           lastdii = di[i];
 849           if ((flags & RPM_ADD_FILTERED_FILELIST) != 0)
 850             {
 851               lastfiltered = is_filtered(dn[di[i]]);
 852               if (lastfiltered == 1)
 853                 continue;
 854             }
 855           did = repodata_str2dir(data, dn[lastdii], 1);
 856           if (!did)
 857             did = repodata_str2dir(data, "/", 1);
 858           lastdid = did;
 859         }
 860       if (b && *b == '/')       /* work around rpm bug */
 861         b++;
 862       if (lastfiltered)
 863         {
 864           if (lastfiltered != 2 || strcmp(b, "sendmail"))
 865             continue;
 866         }
 867       repodata_add_dirstr(data, handle, SOLVABLE_FILELIST, did, b);
 868     }
 869   solv_free(bn);
 870   solv_free(dn);
 871   solv_free(di);
 872 }
 873
 874 static void
 875 addchangelog(Repodata *data, Id handle, RpmHead *rpmhead)
 876 {
 877   char **cn;
 878   char **cx;
 879   unsigned int *ct;
 880   int i, cnc, cxc, ctc;
 881   Queue hq;
 882
 883   ct = headint32array(rpmhead, TAG_CHANGELOGTIME, &ctc);
 884   cx = headstringarray(rpmhead, TAG_CHANGELOGTEXT, &cxc);
 885   cn = headstringarray(rpmhead, TAG_CHANGELOGNAME, &cnc);
 886   if (!ct || !cx || !cn || !ctc || ctc != cxc || ctc != cnc)
 887     {
 888       solv_free(ct);
 889       solv_free(cx);
 890       solv_free(cn);
 891       return;
 892     }
 893   queue_init(&hq);
 894   for (i = 0; i < ctc; i++)
 895     {
 896       Id h = repodata_new_handle(data);
 897       if (ct[i])
 898         repodata_set_num(data, h, SOLVABLE_CHANGELOG_TIME, ct[i]);
 899       if (cn[i])
 900         setutf8string(data, h, SOLVABLE_CHANGELOG_AUTHOR, cn[i]);
 901       if (cx[i])
 902         setutf8string(data, h, SOLVABLE_CHANGELOG_TEXT, cx[i]);
 903       queue_push(&hq, h);
 904     }
 905   for (i = 0; i < hq.count; i++)
 906     repodata_add_flexarray(data, handle, SOLVABLE_CHANGELOG, hq.elements[i]);
 907   queue_free(&hq);
 908   solv_free(ct);
 909   solv_free(cx);
 910   solv_free(cn);
 911 }
 912
 913 static void
 914 set_description_author(Repodata *data, Id handle, char *str)
 915 {
 916   char *aut, *p;
 917   for (aut = str; (aut = strchr(aut, '\n')) != 0; aut++)
 918     if (!strncmp(aut, "\nAuthors:\n--------\n", 19))
 919       break;
 920   if (aut)
 921     {
 922       /* oh my, found SUSE special author section */
 923       int l = aut - str;
 924       str = solv_strdup(str);
 925       aut = str + l;
 926       str[l] = 0;
 927       while (l > 0 && str[l - 1] == '\n')
 928         str[--l] = 0;
 929       if (l)
 930         setutf8string(data, handle, SOLVABLE_DESCRIPTION, str);
 931       p = aut + 19;
 932       aut = str;        /* copy over */
 933       while (*p == ' ' || *p == '\n')
 934         p++;
 935       while (*p)
 936         {
 937           if (*p == '\n')
 938             {
 939               *aut++ = *p++;
 940               while (*p == ' ')
 941                 p++;
 942               continue;
 943             }
 944           *aut++ = *p++;
 945         }
 946       while (aut != str && aut[-1] == '\n')
 947         aut--;
 948       *aut = 0;
 949       if (*str)
 950         setutf8string(data, handle, SOLVABLE_AUTHORS, str);
 951       free(str);
 952     }
 953   else if (*str)
 954     setutf8string(data, handle, SOLVABLE_DESCRIPTION, str);
 955 }
 956
 957 static int
 958 rpmhead2solv(Pool *pool, Repo *repo, Repodata *data, Solvable *s, RpmHead *rpmhead, int flags)
 959 {
 960   char *name;
 961   char *evr;
 962   char *sourcerpm;
 963   Queue ignq;
 964   Id ignqbuf[64];
 965
 966   name = headstring(rpmhead, TAG_NAME);
 967   if (!name)
 968     {
 969       pool_error(pool, 0, "package has no name");
 970       return 0;
 971     }
 972   if (!strcmp(name, "gpg-pubkey"))
 973     return 0;
 974   s->name = pool_str2id(pool, name, 1);
 975   sourcerpm = headstring(rpmhead, TAG_SOURCERPM);
 976   if (sourcerpm || (rpmhead->forcebinary && !headexists(rpmhead, TAG_SOURCEPACKAGE)))
 977     s->arch = pool_str2id(pool, headstring(rpmhead, TAG_ARCH), 1);
 978   else
 979     {
 980       if (headexists(rpmhead, TAG_NOSOURCE) || headexists(rpmhead, TAG_NOPATCH))
 981         s->arch = ARCH_NOSRC;
 982       else
 983         s->arch = ARCH_SRC;
 984     }
 985   if (!s->arch)
 986     s->arch = ARCH_NOARCH;
 987   evr = headtoevr(rpmhead);
 988   s->evr = pool_str2id(pool, evr, 1);
 989   s->vendor = pool_str2id(pool, headstring(rpmhead, TAG_VENDOR), 1);
 990
 991   queue_init_buffer(&ignq, ignqbuf, sizeof(ignqbuf)/sizeof(*ignqbuf));
 992
 993   s->provides = makedeps(pool, repo, rpmhead, TAG_PROVIDENAME, TAG_PROVIDEVERSION, TAG_PROVIDEFLAGS, 0, 0);
 994   if (s->arch != ARCH_SRC && s->arch != ARCH_NOSRC)
 995     s->provides = repo_addid_dep(repo, s->provides, pool_rel2id(pool, s->name, s->evr, REL_EQ, 1), 0);
 996   s->requires = makedeps(pool, repo, rpmhead, TAG_REQUIRENAME, TAG_REQUIREVERSION, TAG_REQUIREFLAGS, flags, &ignq);
 997   s->conflicts = makedeps(pool, repo, rpmhead, TAG_CONFLICTNAME, TAG_CONFLICTVERSION, TAG_CONFLICTFLAGS, 0, 0);
 998   s->obsoletes = makedeps(pool, repo, rpmhead, TAG_OBSOLETENAME, TAG_OBSOLETEVERSION, TAG_OBSOLETEFLAGS, 0, 0);
 999
1000   s->recommends = makedeps(pool, repo, rpmhead, TAG_RECOMMENDNAME, TAG_RECOMMENDVERSION, TAG_RECOMMENDFLAGS, 0, 0);
1001   s->suggests = makedeps(pool, repo, rpmhead, TAG_SUGGESTNAME, TAG_SUGGESTVERSION, TAG_SUGGESTFLAGS, 0, 0);
1002   s->supplements = makedeps(pool, repo, rpmhead, TAG_SUPPLEMENTNAME, TAG_SUPPLEMENTVERSION, TAG_SUPPLEMENTFLAGS, 0, 0);
1003   s->enhances  = makedeps(pool, repo, rpmhead, TAG_ENHANCENAME, TAG_ENHANCEVERSION, TAG_ENHANCEFLAGS, 0, 0);
1004
1005   repo_rewrite_suse_deps(s, 0);
1006
1007   if (data && ignq.count)
1008     repodata_set_idarray(data, s - pool->solvables, SOLVABLE_PREREQ_IGNOREINST, &ignq);
1009   queue_free(&ignq);
1010
1011   if (data)
1012     {
1013       Id handle;
1014       char *str;
1015       unsigned int u32;
1016       unsigned long long u64;
1017
1018       handle = s - pool->solvables;
1019       str = headstring(rpmhead, TAG_SUMMARY);
1020       if (str)
1021         setutf8string(data, handle, SOLVABLE_SUMMARY, str);
1022       str = headstring(rpmhead, TAG_DESCRIPTION);
1023       if (str)
1024         set_description_author(data, handle, str);
1025       str = headstring(rpmhead, TAG_GROUP);
1026       if (str)
1027         repodata_set_poolstr(data, handle, SOLVABLE_GROUP, str);
1028       str = headstring(rpmhead, TAG_LICENSE);
1029       if (str)
1030         repodata_set_poolstr(data, handle, SOLVABLE_LICENSE, str);
1031       str = headstring(rpmhead, TAG_URL);
1032       if (str)
1033         repodata_set_str(data, handle, SOLVABLE_URL, str);
1034       str = headstring(rpmhead, TAG_DISTRIBUTION);
1035       if (str)
1036         repodata_set_poolstr(data, handle, SOLVABLE_DISTRIBUTION, str);
1037       str = headstring(rpmhead, TAG_PACKAGER);
1038       if (str)
1039         repodata_set_poolstr(data, handle, SOLVABLE_PACKAGER, str);
1040       if ((flags & RPM_ADD_WITH_PKGID) != 0)
1041         {
1042           unsigned char *chksum;
1043           unsigned int chksumsize;
1044           chksum = headbinary(rpmhead, TAG_SIGMD5, &chksumsize);
1045           if (chksum && chksumsize == 16)
1046             repodata_set_bin_checksum(data, handle, SOLVABLE_PKGID, REPOKEY_TYPE_MD5, chksum);
1047         }
1048       if ((flags & RPM_ADD_WITH_HDRID) != 0)
1049         {
1050           str = headstring(rpmhead, TAG_SHA1HEADER);
1051           if (str && strlen(str) == 40)
1052             repodata_set_checksum(data, handle, SOLVABLE_HDRID, REPOKEY_TYPE_SHA1, str);
1053           else if (str && strlen(str) == 64)
1054             repodata_set_checksum(data, handle, SOLVABLE_HDRID, REPOKEY_TYPE_SHA256, str);
1055         }
1056       u32 = headint32(rpmhead, TAG_BUILDTIME);
1057       if (u32)
1058         repodata_set_num(data, handle, SOLVABLE_BUILDTIME, u32);
1059       u32 = headint32(rpmhead, TAG_INSTALLTIME);
1060       if (u32)
1061         repodata_set_num(data, handle, SOLVABLE_INSTALLTIME, u32);
1062       u64 = headint64(rpmhead, TAG_LONGSIZE);
1063       if (u64)
1064         repodata_set_num(data, handle, SOLVABLE_INSTALLSIZE, u64);
1065       else
1066         {
1067           u32 = headint32(rpmhead, TAG_SIZE);
1068           if (u32)
1069             repodata_set_num(data, handle, SOLVABLE_INSTALLSIZE, u32);
1070         }
1071       if (sourcerpm)
1072         repodata_set_sourcepkg(data, handle, sourcerpm);
1073       if ((flags & RPM_ADD_TRIGGERS) != 0)
1074         {
1075           unsigned int ida = makedeps(pool, repo, rpmhead, TAG_TRIGGERNAME, TAG_TRIGGERVERSION, TAG_TRIGGERFLAGS, 0, 0);
1076           Id id, lastid = 0;
1077           for (lastid = 0; (id = repo->idarraydata[ida]) != 0; ida++, lastid = id)
1078             if (id != lastid)
1079               repodata_add_idarray(data, handle, SOLVABLE_TRIGGERS, id);
1080         }
1081       if ((flags & RPM_ADD_NO_FILELIST) == 0)
1082         addfilelist(data, handle, rpmhead, flags);
1083       if ((flags & RPM_ADD_WITH_CHANGELOG) != 0)
1084         addchangelog(data, handle, rpmhead);
1085     }
1086   solv_free(evr);
1087   return 1;
1088 }
1089
1090 static inline unsigned int
1091 getu32(const unsigned char *dp)
1092 {
1093   return dp[0] << 24 | dp[1] << 16 | dp[2] << 8 | dp[3];
1094 }
1095
1096
1097 /******************************************************************/
1098 /*  Rpm Database stuff
1099  */
1100
1101 struct rpmdbstate {
1102   Pool *pool;
1103   char *rootdir;
1104
1105   RpmHead *rpmhead;     /* header storage space */
1106   int rpmheadsize;
1107
1108 #ifdef ENABLE_RPMDB
1109   int dbopened;
1110   DB_ENV *dbenv;        /* database environment */
1111   DB *db;               /* packages database */
1112   int byteswapped;      /* endianess of packages database */
1113   int is_ostree;        /* read-only db that lives in /usr/share/rpm */
1114 #endif
1115 };
1116
1117 #ifdef ENABLE_RPMDB
1118
1119 struct rpmdbentry {
1120   Id rpmdbid;
1121   Id nameoff;
1122 };
1123
1124 #define ENTRIES_BLOCK 255
1125 #define NAMEDATA_BLOCK 1023
1126
1127
1128 static inline Id
1129 db2rpmdbid(unsigned char *db, int byteswapped)
1130 {
1131 #ifdef RPM5
1132   return db[0] << 24 | db[1] << 16 | db[2] << 8 | db[3];
1133 #else
1134 # if defined(WORDS_BIGENDIAN)
1135   if (!byteswapped)
1136 # else
1137   if (byteswapped)
1138 # endif
1139     return db[0] << 24 | db[1] << 16 | db[2] << 8 | db[3];
1140   else
1141     return db[3] << 24 | db[2] << 16 | db[1] << 8 | db[0];
1142 #endif
1143 }
1144
1145 static inline void
1146 rpmdbid2db(unsigned char *db, Id id, int byteswapped)
1147 {
1148 #ifdef RPM5
1149   db[0] = id >> 24, db[1] = id >> 16, db[2] = id >> 8, db[3] = id;
1150 #else
1151 # if defined(WORDS_BIGENDIAN)
1152   if (!byteswapped)
1153 # else
1154   if (byteswapped)
1155 # endif
1156     db[0] = id >> 24, db[1] = id >> 16, db[2] = id >> 8, db[3] = id;
1157   else
1158     db[3] = id >> 24, db[2] = id >> 16, db[1] = id >> 8, db[0] = id;
1159 #endif
1160 }
1161
1162 #ifdef FEDORA
1163 int
1164 serialize_dbenv_ops(struct rpmdbstate *state)
1165 {
1166   char lpath[PATH_MAX];
1167   mode_t oldmask;
1168   int fd;
1169   struct flock fl;
1170
1171   snprintf(lpath, PATH_MAX, "%s/var/lib/rpm/.dbenv.lock", state->rootdir ? state->rootdir : "");
1172   oldmask = umask(022);
1173   fd = open(lpath, (O_RDWR|O_CREAT), 0644);
1174   umask(oldmask);
1175   if (fd < 0)
1176     return -1;
1177   memset(&fl, 0, sizeof(fl));
1178   fl.l_type = F_WRLCK;
1179   fl.l_whence = SEEK_SET;
1180   for (;;)
1181     {
1182       if (fcntl(fd, F_SETLKW, &fl) != -1)
1183         return fd;
1184       if (errno != EINTR)
1185         break;
1186     }
1187   close(fd);
1188   return -1;
1189 }
1190 #endif
1191
1192 /* should look in /usr/lib/rpm/macros instead, but we want speed... */
1193 static int
1194 opendbenv(struct rpmdbstate *state)
1195 {
1196   const char *rootdir = state->rootdir;
1197   char dbpath[PATH_MAX];
1198   DB_ENV *dbenv = 0;
1199   int r;
1200
1201   if (db_env_create(&dbenv, 0))
1202     return pool_error(state->pool, 0, "db_env_create: %s", strerror(errno));
1203 #if defined(FEDORA) && (DB_VERSION_MAJOR >= 5 || (DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 5))
1204   dbenv->set_thread_count(dbenv, 8);
1205 #endif
1206   snprintf(dbpath, PATH_MAX, "%s/var/lib/rpm", rootdir ? rootdir : "");
1207   if (access(dbpath, W_OK) == -1)
1208     {
1209       snprintf(dbpath, PATH_MAX, "%s/usr/share/rpm/Packages", rootdir ? rootdir : "");
1210       if (access(dbpath, R_OK) == 0)
1211         state->is_ostree = 1;
1212       snprintf(dbpath, PATH_MAX, "%s%s", rootdir ? rootdir : "", state->is_ostree ? "/usr/share/rpm" : "/var/lib/rpm");
1213       r = dbenv->open(dbenv, dbpath, DB_CREATE|DB_PRIVATE|DB_INIT_MPOOL, 0);
1214     }
1215   else
1216     {
1217 #ifdef FEDORA
1218       int serialize_fd = serialize_dbenv_ops(state);
1219       r = dbenv->open(dbenv, dbpath, DB_CREATE|DB_INIT_CDB|DB_INIT_MPOOL, 0644);
1220       if (serialize_fd >= 0)
1221         close(serialize_fd);
1222 #else
1223       r = dbenv->open(dbenv, dbpath, DB_CREATE|DB_PRIVATE|DB_INIT_MPOOL, 0);
1224 #endif
1225     }
1226   if (r)
1227     {
1228       pool_error(state->pool, 0, "dbenv->open: %s", strerror(errno));
1229       dbenv->close(dbenv, 0);
1230       return 0;
1231     }
1232   state->dbenv = dbenv;
1233   return 1;
1234 }
1235
1236 static void
1237 closedbenv(struct rpmdbstate *state)
1238 {
1239 #ifdef FEDORA
1240   uint32_t eflags = 0;
1241 #endif
1242
1243   if (!state->dbenv)
1244     return;
1245 #ifdef FEDORA
1246   (void)state->dbenv->get_open_flags(state->dbenv, &eflags);
1247   if (!(eflags & DB_PRIVATE))
1248     {
1249       int serialize_fd = serialize_dbenv_ops(state);
1250       state->dbenv->close(state->dbenv, 0);
1251       if (serialize_fd >= 0)
1252         close(serialize_fd);
1253     }
1254   else
1255     state->dbenv->close(state->dbenv, 0);
1256 #else
1257   state->dbenv->close(state->dbenv, 0);
1258 #endif
1259   state->dbenv = 0;
1260 }
1261
1262 #endif
1263
1264 static void
1265 freestate(struct rpmdbstate *state)
1266 {
1267   /* close down */
1268   if (!state)
1269     return;
1270 #ifdef ENABLE_RPMDB
1271   if (state->db)
1272     state->db->close(state->db, 0);
1273   if (state->dbenv)
1274     closedbenv(state);
1275 #endif
1276   if (state->rootdir)
1277     solv_free(state->rootdir);
1278   solv_free(state->rpmhead);
1279 }
1280
1281 void *
1282 rpm_state_create(Pool *pool, const char *rootdir)
1283 {
1284   struct rpmdbstate *state;
1285   state = solv_calloc(1, sizeof(*state));
1286   state->pool = pool;
1287   if (rootdir)
1288     state->rootdir = solv_strdup(rootdir);
1289   return state;
1290 }
1291
1292 void *
1293 rpm_state_free(void *state)
1294 {
1295   freestate(state);
1296   return solv_free(state);
1297 }
1298
1299
1300 #ifdef ENABLE_RPMDB
1301
1302 static int
1303 openpkgdb(struct rpmdbstate *state)
1304 {
1305   if (state->dbopened)
1306     return state->dbopened > 0 ? 1 : 0;
1307   state->dbopened = -1;
1308   if (!state->dbenv && !opendbenv(state))
1309     return 0;
1310   if (db_create(&state->db, state->dbenv, 0))
1311     {
1312       pool_error(state->pool, 0, "db_create: %s", strerror(errno));
1313       state->db = 0;
1314       closedbenv(state);
1315       return 0;
1316     }
1317   if (state->db->open(state->db, 0, "Packages", 0, DB_UNKNOWN, DB_RDONLY, 0664))
1318     {
1319       pool_error(state->pool, 0, "db->open Packages: %s", strerror(errno));
1320       state->db->close(state->db, 0);
1321       state->db = 0;
1322       closedbenv(state);
1323       return 0;
1324     }
1325   if (state->db->get_byteswapped(state->db, &state->byteswapped))
1326     {
1327       pool_error(state->pool, 0, "db->get_byteswapped: %s", strerror(errno));
1328       state->db->close(state->db, 0);
1329       state->db = 0;
1330       closedbenv(state);
1331       return 0;
1332     }
1333   state->dbopened = 1;
1334   return 1;
1335 }
1336
1337 /* get the rpmdbids of all installed packages from the Name index database.
1338  * This is much faster then querying the big Packages database */
1339 static struct rpmdbentry *
1340 getinstalledrpmdbids(struct rpmdbstate *state, const char *index, const char *match, int *nentriesp, char **namedatap)
1341 {
1342   DB_ENV *dbenv = 0;
1343   DB *db = 0;
1344   DBC *dbc = 0;
1345   int byteswapped;
1346   DBT dbkey;
1347   DBT dbdata;
1348   unsigned char *dp;
1349   int dl;
1350   Id nameoff;
1351
1352   char *namedata = 0;
1353   int namedatal = 0;
1354   struct rpmdbentry *entries = 0;
1355   int nentries = 0;
1356
1357   *nentriesp = 0;
1358   if (namedatap)
1359     *namedatap = 0;
1360
1361   if (!state->dbenv && !opendbenv(state))
1362     return 0;
1363   dbenv = state->dbenv;
1364   if (db_create(&db, dbenv, 0))
1365     {
1366       pool_error(state->pool, 0, "db_create: %s", strerror(errno));
1367       return 0;
1368     }
1369   if (db->open(db, 0, index, 0, DB_UNKNOWN, DB_RDONLY, 0664))
1370     {
1371       pool_error(state->pool, 0, "db->open %s: %s", index, strerror(errno));
1372       db->close(db, 0);
1373       return 0;
1374     }
1375   if (db->get_byteswapped(db, &byteswapped))
1376     {
1377       pool_error(state->pool, 0, "db->get_byteswapped: %s", strerror(errno));
1378       db->close(db, 0);
1379       return 0;
1380     }
1381   if (db->cursor(db, NULL, &dbc, 0))
1382     {
1383       pool_error(state->pool, 0, "db->cursor: %s", strerror(errno));
1384       db->close(db, 0);
1385       return 0;
1386     }
1387   memset(&dbkey, 0, sizeof(dbkey));
1388   memset(&dbdata, 0, sizeof(dbdata));
1389   if (match)
1390     {
1391       dbkey.data = (void *)match;
1392       dbkey.size = strlen(match);
1393     }
1394   while (dbc->c_get(dbc, &dbkey, &dbdata, match ? DB_SET : DB_NEXT) == 0)
1395     {
1396       if (!match && dbkey.size == 10 && !memcmp(dbkey.data, "gpg-pubkey", 10))
1397         continue;
1398       dl = dbdata.size;
1399       dp = dbdata.data;
1400       nameoff = namedatal;
1401       if (namedatap)
1402         {
1403           namedata = solv_extend(namedata, namedatal, dbkey.size + 1, 1, NAMEDATA_BLOCK);
1404           memcpy(namedata + namedatal, dbkey.data, dbkey.size);
1405           namedata[namedatal + dbkey.size] = 0;
1406           namedatal += dbkey.size + 1;
1407         }
1408       while(dl >= RPM_INDEX_SIZE)
1409         {
1410           entries = solv_extend(entries, nentries, 1, sizeof(*entries), ENTRIES_BLOCK);
1411           entries[nentries].rpmdbid = db2rpmdbid(dp, byteswapped);
1412           entries[nentries].nameoff = nameoff;
1413           nentries++;
1414           dp += RPM_INDEX_SIZE;
1415           dl -= RPM_INDEX_SIZE;
1416         }
1417       if (match)
1418         break;
1419     }
1420   dbc->c_close(dbc);
1421   db->close(db, 0);
1422   /* make sure that enteries is != 0 if there was no error */
1423   if (!entries)
1424     entries = solv_extend(entries, 1, 1, sizeof(*entries), ENTRIES_BLOCK);
1425   *nentriesp = nentries;
1426   if (namedatap)
1427     *namedatap = namedata;
1428   return entries;
1429 }
1430
1431 /* common code, return dbid on success, -1 on error */
1432 static int
1433 getrpm_dbdata(struct rpmdbstate *state, DBT *dbdata, int dbid)
1434 {
1435   unsigned int dsize, cnt, l;
1436   RpmHead *rpmhead;
1437
1438   if (dbdata->size < 8)
1439     return pool_error(state->pool, -1, "corrupt rpm database (size)");
1440   cnt = getu32((const unsigned char *)dbdata->data);
1441   dsize = getu32((const unsigned char *)dbdata->data + 4);
1442   if (cnt >= MAX_HDR_CNT || dsize >= MAX_HDR_DSIZE)
1443     return pool_error(state->pool, -1, "corrupt rpm database (cnt/dcnt)");
1444   l = cnt * 16 + dsize;
1445   if (8 + l > dbdata->size)
1446     return pool_error(state->pool, -1, "corrupt rpm database (data size)");
1447   if (l + 1 > state->rpmheadsize)
1448     {
1449       state->rpmheadsize = l + 128;
1450       state->rpmhead = solv_realloc(state->rpmhead, sizeof(*rpmhead) + state->rpmheadsize);
1451     }
1452   rpmhead = state->rpmhead;
1453   rpmhead->forcebinary = 1;
1454   rpmhead->cnt = cnt;
1455   rpmhead->dcnt = dsize;
1456   memcpy(rpmhead->data, (unsigned char *)dbdata->data + 8, l);
1457   rpmhead->data[l] = 0;
1458   rpmhead->dp = rpmhead->data + cnt * 16;
1459   return dbid;
1460 }
1461
1462 /* retrive header by rpmdbid, returns 0 if not found, -1 on error */
1463 static int
1464 getrpm_dbid(struct rpmdbstate *state, Id dbid)
1465 {
1466   unsigned char buf[4];
1467   DBT dbkey;
1468   DBT dbdata;
1469
1470   if (dbid <= 0)
1471     return pool_error(state->pool, -1, "illegal rpmdbid %d", dbid);
1472   if (state->dbopened != 1 && !openpkgdb(state))
1473     return -1;
1474   rpmdbid2db(buf, dbid, state->byteswapped);
1475   memset(&dbkey, 0, sizeof(dbkey));
1476   memset(&dbdata, 0, sizeof(dbdata));
1477   dbkey.data = buf;
1478   dbkey.size = 4;
1479   dbdata.data = 0;
1480   dbdata.size = 0;
1481   if (state->db->get(state->db, NULL, &dbkey, &dbdata, 0))
1482     return 0;
1483   return getrpm_dbdata(state, &dbdata, dbid);
1484 }
1485
1486 /* retrive header by berkeleydb cursor, returns 0 on EOF, -1 on error */
1487 static Id
1488 getrpm_cursor(struct rpmdbstate *state, DBC *dbc)
1489 {
1490   DBT dbkey;
1491   DBT dbdata;
1492   Id dbid;
1493
1494   memset(&dbkey, 0, sizeof(dbkey));
1495   memset(&dbdata, 0, sizeof(dbdata));
1496   while (dbc->c_get(dbc, &dbkey, &dbdata, DB_NEXT) == 0)
1497     {
1498       if (dbkey.size != 4)
1499         return pool_error(state->pool, -1, "corrupt Packages database (key size)");
1500       dbid = db2rpmdbid(dbkey.data, state->byteswapped);
1501       if (dbid)         /* ignore join key */
1502         return getrpm_dbdata(state, &dbdata, dbid);
1503     }
1504   return 0;
1505 }
1506
1507 static int
1508 count_headers(struct rpmdbstate *state)
1509 {
1510   Pool *pool = state->pool;
1511   char dbpath[PATH_MAX];
1512   struct stat statbuf;
1513   DB *db = 0;
1514   DBC *dbc = 0;
1515   int count = 0;
1516   DBT dbkey;
1517   DBT dbdata;
1518
1519   snprintf(dbpath, PATH_MAX, "%s%s/Name", state->rootdir ? state->rootdir : "", state->is_ostree ? "/usr/share/rpm" : "/var/lib/rpm");
1520   if (stat(dbpath, &statbuf))
1521     return 0;
1522   memset(&dbkey, 0, sizeof(dbkey));
1523   memset(&dbdata, 0, sizeof(dbdata));
1524   if (db_create(&db, state->dbenv, 0))
1525     {
1526       pool_error(pool, 0, "db_create: %s", strerror(errno));
1527       return 0;
1528     }
1529   if (db->open(db, 0, "Name", 0, DB_UNKNOWN, DB_RDONLY, 0664))
1530     {
1531       pool_error(pool, 0, "db->open Name: %s", strerror(errno));
1532       db->close(db, 0);
1533       return 0;
1534     }
1535   if (db->cursor(db, NULL, &dbc, 0))
1536     {
1537       db->close(db, 0);
1538       pool_error(pool, 0, "db->cursor: %s", strerror(errno));
1539       return 0;
1540     }
1541   while (dbc->c_get(dbc, &dbkey, &dbdata, DB_NEXT) == 0)
1542     count += dbdata.size / RPM_INDEX_SIZE;
1543   dbc->c_close(dbc);
1544   db->close(db, 0);
1545   return count;
1546 }
1547
1548 /******************************************************************/
1549
1550 static Offset
1551 copydeps(Pool *pool, Repo *repo, Offset fromoff, Repo *fromrepo)
1552 {
1553   int cc;
1554   Id *ida, *from;
1555   Offset ido;
1556
1557   if (!fromoff)
1558     return 0;
1559   from = fromrepo->idarraydata + fromoff;
1560   for (ida = from, cc = 0; *ida; ida++, cc++)
1561     ;
1562   if (cc == 0)
1563     return 0;
1564   ido = repo_reserve_ids(repo, 0, cc);
1565   ida = repo->idarraydata + ido;
1566   memcpy(ida, from, (cc + 1) * sizeof(Id));
1567   repo->idarraysize += cc + 1;
1568   return ido;
1569 }
1570
1571 #define COPYDIR_DIRCACHE_SIZE 512
1572
1573 static Id copydir_complex(Pool *pool, Repodata *data, Repodata *fromdata, Id did, Id *cache);
1574
1575 static inline Id
1576 copydir(Pool *pool, Repodata *data, Repodata *fromdata, Id did, Id *cache)
1577 {
1578   if (cache && cache[did & 255] == did)
1579     return cache[(did & 255) + 256];
1580   return copydir_complex(pool, data, fromdata, did, cache);
1581 }
1582
1583 static Id
1584 copydir_complex(Pool *pool, Repodata *data, Repodata *fromdata, Id did, Id *cache)
1585 {
1586   Id parent = dirpool_parent(&fromdata->dirpool, did);
1587   Id compid = dirpool_compid(&fromdata->dirpool, did);
1588   if (parent)
1589     parent = copydir(pool, data, fromdata, parent, cache);
1590   if (data->localpool || fromdata->localpool)
1591     compid = repodata_translate_id(data, fromdata, compid, 1);
1592   compid = dirpool_add_dir(&data->dirpool, parent, compid, 1);
1593   if (cache)
1594     {
1595       cache[did & 255] = did;
1596       cache[(did & 255) + 256] = compid;
1597     }
1598   return compid;
1599 }
1600
1601 struct solvable_copy_cbdata {
1602   Repodata *data;
1603   Id handle;
1604   Id subhandle;
1605   Id *dircache;
1606 };
1607
1608 static int
1609 solvable_copy_cb(void *vcbdata, Solvable *r, Repodata *fromdata, Repokey *key, KeyValue *kv)
1610 {
1611   struct solvable_copy_cbdata *cbdata = vcbdata;
1612   Id id, keyname;
1613   Repodata *data = cbdata->data;
1614   Id handle = cbdata->handle;
1615   Pool *pool = data->repo->pool;
1616
1617   keyname = key->name;
1618   switch(key->type)
1619     {
1620     case REPOKEY_TYPE_ID:
1621     case REPOKEY_TYPE_CONSTANTID:
1622     case REPOKEY_TYPE_IDARRAY:  /* used for triggers */
1623       id = kv->id;
1624       if (data->localpool || fromdata->localpool)
1625         id = repodata_translate_id(data, fromdata, id, 1);
1626       if (key->type == REPOKEY_TYPE_ID)
1627         repodata_set_id(data, handle, keyname, id);
1628       else if (key->type == REPOKEY_TYPE_CONSTANTID)
1629         repodata_set_constantid(data, handle, keyname, id);
1630       else
1631         repodata_add_idarray(data, handle, keyname, id);
1632       break;
1633     case REPOKEY_TYPE_STR:
1634       repodata_set_str(data, handle, keyname, kv->str);
1635       break;
1636     case REPOKEY_TYPE_VOID:
1637       repodata_set_void(data, handle, keyname);
1638       break;
1639     case REPOKEY_TYPE_NUM:
1640       repodata_set_num(data, handle, keyname, SOLV_KV_NUM64(kv));
1641       break;
1642     case REPOKEY_TYPE_CONSTANT:
1643       repodata_set_constant(data, handle, keyname, kv->num);
1644       break;
1645     case REPOKEY_TYPE_DIRNUMNUMARRAY:
1646       id = kv->id;
1647       id = copydir(pool, data, fromdata, id, cbdata->dircache);
1648       repodata_add_dirnumnum(data, handle, keyname, id, kv->num, kv->num2);
1649       break;
1650     case REPOKEY_TYPE_DIRSTRARRAY:
1651       id = kv->id;
1652       id = copydir(pool, data, fromdata, id, cbdata->dircache);
1653       repodata_add_dirstr(data, handle, keyname, id, kv->str);
1654       break;
1655     case REPOKEY_TYPE_FLEXARRAY:
1656       if (kv->eof == 2)
1657         {
1658           assert(cbdata->subhandle);
1659           cbdata->handle = cbdata->subhandle;
1660           cbdata->subhandle = 0;
1661           break;
1662         }
1663       if (!kv->entry)
1664         {
1665           assert(!cbdata->subhandle);
1666           cbdata->subhandle = cbdata->handle;
1667         }
1668       cbdata->handle = repodata_new_handle(data);
1669       repodata_add_flexarray(data, cbdata->subhandle, keyname, cbdata->handle);
1670       break;
1671     default:
1672       if (solv_chksum_len(key->type))
1673         {
1674           repodata_set_bin_checksum(data, handle, keyname, key->type, (const unsigned char *)kv->str);
1675           break;
1676         }
1677       break;
1678     }
1679   return 0;
1680 }
1681
1682 static void
1683 solvable_copy(Solvable *s, Solvable *r, Repodata *data, Id *dircache)
1684 {
1685   int p, i;
1686   Repo *repo = s->repo;
1687   Pool *pool = repo->pool;
1688   Repo *fromrepo = r->repo;
1689   struct solvable_copy_cbdata cbdata;
1690
1691   /* copy solvable data */
1692   s->name = r->name;
1693   s->evr = r->evr;
1694   s->arch = r->arch;
1695   s->vendor = r->vendor;
1696   s->provides = copydeps(pool, repo, r->provides, fromrepo);
1697   s->requires = copydeps(pool, repo, r->requires, fromrepo);
1698   s->conflicts = copydeps(pool, repo, r->conflicts, fromrepo);
1699   s->obsoletes = copydeps(pool, repo, r->obsoletes, fromrepo);
1700   s->recommends = copydeps(pool, repo, r->recommends, fromrepo);
1701   s->suggests = copydeps(pool, repo, r->suggests, fromrepo);
1702   s->supplements = copydeps(pool, repo, r->supplements, fromrepo);
1703   s->enhances  = copydeps(pool, repo, r->enhances, fromrepo);
1704
1705   /* copy all attributes */
1706   if (!data)
1707     return;
1708   cbdata.data = data;
1709   cbdata.handle = s - pool->solvables;
1710   cbdata.subhandle = 0;
1711   cbdata.dircache = dircache;
1712   p = r - fromrepo->pool->solvables;
1713 #if 0
1714   repo_search(fromrepo, p, 0, 0, SEARCH_NO_STORAGE_SOLVABLE | SEARCH_SUB | SEARCH_ARRAYSENTINEL, solvable_copy_cb, &cbdata);
1715 #else
1716   FOR_REPODATAS(fromrepo, i, data)
1717     {
1718       if (p >= data->start && p < data->end)
1719         repodata_search(data, p, 0, SEARCH_SUB | SEARCH_ARRAYSENTINEL, solvable_copy_cb, &cbdata);
1720       cbdata.dircache = 0;      /* only for first repodata */
1721     }
1722 #endif
1723 }
1724
1725 /* used to sort entries by package name that got returned in some database order */
1726 static int
1727 rpmids_sort_cmp(const void *va, const void *vb, void *dp)
1728 {
1729   struct rpmdbentry const *a = va, *b = vb;
1730   char *namedata = dp;
1731   int r;
1732   r = strcmp(namedata + a->nameoff, namedata + b->nameoff);
1733   if (r)
1734     return r;
1735   return a->rpmdbid - b->rpmdbid;
1736 }
1737
1738 static int
1739 pkgids_sort_cmp(const void *va, const void *vb, void *dp)
1740 {
1741   Repo *repo = dp;
1742   Pool *pool = repo->pool;
1743   Solvable *a = pool->solvables + *(Id *)va;
1744   Solvable *b = pool->solvables + *(Id *)vb;
1745   Id *rpmdbid;
1746
1747   if (a->name != b->name)
1748     return strcmp(pool_id2str(pool, a->name), pool_id2str(pool, b->name));
1749   rpmdbid = repo->rpmdbid;
1750   return rpmdbid[(a - pool->solvables) - repo->start] - rpmdbid[(b - pool->solvables) - repo->start];
1751 }
1752
1753 static void
1754 swap_solvables(Repo *repo, Repodata *data, Id pa, Id pb)
1755 {
1756   Pool *pool = repo->pool;
1757   Solvable tmp;
1758
1759   tmp = pool->solvables[pa];
1760   pool->solvables[pa] = pool->solvables[pb];
1761   pool->solvables[pb] = tmp;
1762   if (repo->rpmdbid)
1763     {
1764       Id tmpid = repo->rpmdbid[pa - repo->start];
1765       repo->rpmdbid[pa - repo->start] = repo->rpmdbid[pb - repo->start];
1766       repo->rpmdbid[pb - repo->start] = tmpid;
1767     }
1768   /* only works if nothing is already internalized! */
1769   if (data)
1770     repodata_swap_attrs(data, pa, pb);
1771 }
1772
1773 static void
1774 mkrpmdbcookie(struct stat *st, unsigned char *cookie, int flags)
1775 {
1776   int f = 0;
1777   memset(cookie, 0, 32);
1778   cookie[3] = RPMDB_COOKIE_VERSION;
1779   memcpy(cookie + 16, &st->st_ino, sizeof(st->st_ino));
1780   memcpy(cookie + 24, &st->st_dev, sizeof(st->st_dev));
1781   if ((flags & RPM_ADD_WITH_PKGID) != 0)
1782     f |= 1;
1783   if ((flags & RPM_ADD_WITH_HDRID) != 0)
1784     f |= 2;
1785   if ((flags & RPM_ADD_WITH_CHANGELOG) != 0)
1786     f |= 4;
1787   if ((flags & RPM_ADD_NO_FILELIST) == 0)
1788     f |= 8;
1789   if ((flags & RPM_ADD_NO_RPMLIBREQS) != 0)
1790     cookie[1] = 1;
1791   cookie[0] = f;
1792 }
1793
1794 /*
1795  * read rpm db as repo
1796  *
1797  */
1798
1799 int
1800 repo_add_rpmdb(Repo *repo, Repo *ref, int flags)
1801 {
1802   Pool *pool = repo->pool;
1803   char dbpath[PATH_MAX];
1804   struct stat packagesstat;
1805   unsigned char newcookie[32];
1806   const unsigned char *oldcookie = 0;
1807   Id oldcookietype = 0;
1808   Repodata *data;
1809   int count = 0, done = 0;
1810   struct rpmdbstate state;
1811   int i;
1812   Solvable *s;
1813   unsigned int now;
1814
1815   now = solv_timems(0);
1816   memset(&state, 0, sizeof(state));
1817   state.pool = pool;
1818   if (flags & REPO_USE_ROOTDIR)
1819     state.rootdir = solv_strdup(pool_get_rootdir(pool));
1820
1821   data = repo_add_repodata(repo, flags);
1822
1823   if (ref && !(ref->nsolvables && ref->rpmdbid && ref->pool == repo->pool))
1824     {
1825       if ((flags & RPMDB_EMPTY_REFREPO) != 0)
1826         repo_empty(ref, 1);
1827       ref = 0;
1828     }
1829
1830   if (!opendbenv(&state))
1831     {
1832       solv_free(state.rootdir);
1833       return -1;
1834     }
1835
1836   /* XXX: should get ro lock of Packages database! */
1837   snprintf(dbpath, PATH_MAX, "%s%s/Packages", state.rootdir ? state.rootdir : "", state.is_ostree ? "/usr/share/rpm" : "/var/lib/rpm");
1838   if (stat(dbpath, &packagesstat))
1839     {
1840       pool_error(pool, -1, "%s: %s", dbpath, strerror(errno));
1841       freestate(&state);
1842       return -1;
1843     }
1844   mkrpmdbcookie(&packagesstat, newcookie, flags);
1845   repodata_set_bin_checksum(data, SOLVID_META, REPOSITORY_RPMDBCOOKIE, REPOKEY_TYPE_SHA256, newcookie);
1846
1847   if (ref)
1848     oldcookie = repo_lookup_bin_checksum(ref, SOLVID_META, REPOSITORY_RPMDBCOOKIE, &oldcookietype);
1849   if (!ref || !oldcookie || oldcookietype != REPOKEY_TYPE_SHA256 || memcmp(oldcookie, newcookie, 32) != 0)
1850     {
1851       int solvstart = 0, solvend = 0;
1852       Id dbid;
1853       DBC *dbc = 0;
1854
1855       if (ref && (flags & RPMDB_EMPTY_REFREPO) != 0)
1856         repo_empty(ref, 1);     /* get it out of the way */
1857       if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1858         count = count_headers(&state);
1859       if (!openpkgdb(&state))
1860         {
1861           freestate(&state);
1862           return -1;
1863         }
1864       if (state.db->cursor(state.db, NULL, &dbc, 0))
1865         {
1866           freestate(&state);
1867           return pool_error(pool, -1, "db->cursor failed");
1868         }
1869       i = 0;
1870       s = 0;
1871       while ((dbid = getrpm_cursor(&state, dbc)) != 0)
1872         {
1873           if (dbid == -1)
1874             {
1875               dbc->c_close(dbc);
1876               freestate(&state);
1877               return -1;
1878             }
1879           if (!s)
1880             {
1881               s = pool_id2solvable(pool, repo_add_solvable(repo));
1882               if (!solvstart)
1883                 solvstart = s - pool->solvables;
1884               solvend = s - pool->solvables + 1;
1885             }
1886           if (!repo->rpmdbid)
1887             repo->rpmdbid = repo_sidedata_create(repo, sizeof(Id));
1888           repo->rpmdbid[(s - pool->solvables) - repo->start] = dbid;
1889           if (rpmhead2solv(pool, repo, data, s, state.rpmhead, flags | RPM_ADD_TRIGGERS))
1890             {
1891               i++;
1892               s = 0;
1893             }
1894           else
1895             {
1896               /* We can reuse this solvable, but make sure it's still
1897                  associated with this repo.  */
1898               memset(s, 0, sizeof(*s));
1899               s->repo = repo;
1900             }
1901           if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1902             {
1903               if (done < count)
1904                 done++;
1905               if (done < count && (done - 1) * 100 / count != done * 100 / count)
1906                 pool_debug(pool, SOLV_ERROR, "%%%% %d\n", done * 100 / count);
1907             }
1908         }
1909       dbc->c_close(dbc);
1910       if (s)
1911         {
1912           /* oops, could not reuse. free it instead */
1913           repo_free_solvable(repo, s - pool->solvables, 1);
1914           solvend--;
1915           s = 0;
1916         }
1917       /* now sort all solvables in the new solvstart..solvend block */
1918       if (solvend - solvstart > 1)
1919         {
1920           Id *pkgids = solv_malloc2(solvend - solvstart, sizeof(Id));
1921           for (i = solvstart; i < solvend; i++)
1922             pkgids[i - solvstart] = i;
1923           solv_sort(pkgids, solvend - solvstart, sizeof(Id), pkgids_sort_cmp, repo);
1924           /* adapt order */
1925           for (i = solvstart; i < solvend; i++)
1926             {
1927               int j = pkgids[i - solvstart];
1928               while (j < i)
1929                 j = pkgids[i - solvstart] = pkgids[j - solvstart];
1930               if (j != i)
1931                 swap_solvables(repo, data, i, j);
1932             }
1933           solv_free(pkgids);
1934         }
1935     }
1936   else
1937     {
1938       Id dircache[COPYDIR_DIRCACHE_SIZE];               /* see copydir */
1939       struct rpmdbentry *entries = 0, *rp;
1940       int nentries = 0;
1941       char *namedata = 0;
1942       unsigned int refmask, h;
1943       Id id, *refhash;
1944       int res;
1945
1946       memset(dircache, 0, sizeof(dircache));
1947
1948       /* get ids of installed rpms */
1949       entries = getinstalledrpmdbids(&state, "Name", 0, &nentries, &namedata);
1950       if (!entries)
1951         {
1952           freestate(&state);
1953           return -1;
1954         }
1955
1956       /* sort by name */
1957       if (nentries > 1)
1958         solv_sort(entries, nentries, sizeof(*entries), rpmids_sort_cmp, namedata);
1959
1960       /* create hash from dbid to ref */
1961       refmask = mkmask(ref->nsolvables);
1962       refhash = solv_calloc(refmask + 1, sizeof(Id));
1963       for (i = 0; i < ref->end - ref->start; i++)
1964         {
1965           if (!ref->rpmdbid[i])
1966             continue;
1967           h = ref->rpmdbid[i] & refmask;
1968           while (refhash[h])
1969             h = (h + 317) & refmask;
1970           refhash[h] = i + 1;   /* make it non-zero */
1971         }
1972
1973       /* count the misses, they will cost us time */
1974       if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1975         {
1976           for (i = 0, rp = entries; i < nentries; i++, rp++)
1977             {
1978               if (refhash)
1979                 {
1980                   Id dbid = rp->rpmdbid;
1981                   h = dbid & refmask;
1982                   while ((id = refhash[h]))
1983                     {
1984                       if (ref->rpmdbid[id - 1] == dbid)
1985                         break;
1986                       h = (h + 317) & refmask;
1987                     }
1988                   if (id)
1989                     continue;
1990                 }
1991               count++;
1992             }
1993         }
1994
1995       if (ref && (flags & RPMDB_EMPTY_REFREPO) != 0)
1996         s = pool_id2solvable(pool, repo_add_solvable_block_before(repo, nentries, ref));
1997       else
1998         s = pool_id2solvable(pool, repo_add_solvable_block(repo, nentries));
1999       if (!repo->rpmdbid)
2000         repo->rpmdbid = repo_sidedata_create(repo, sizeof(Id));
2001
2002       for (i = 0, rp = entries; i < nentries; i++, rp++, s++)
2003         {
2004           Id dbid = rp->rpmdbid;
2005           repo->rpmdbid[(s - pool->solvables) - repo->start] = rp->rpmdbid;
2006           if (refhash)
2007             {
2008               h = dbid & refmask;
2009               while ((id = refhash[h]))
2010                 {
2011                   if (ref->rpmdbid[id - 1] == dbid)
2012                     break;
2013                   h = (h + 317) & refmask;
2014                 }
2015               if (id)
2016                 {
2017                   Solvable *r = ref->pool->solvables + ref->start + (id - 1);
2018                   if (r->repo == ref)
2019                     {
2020                       solvable_copy(s, r, data, dircache);
2021                       continue;
2022                     }
2023                 }
2024             }
2025           res = getrpm_dbid(&state, dbid);
2026           if (res <= 0)
2027             {
2028               if (!res)
2029                 pool_error(pool, -1, "inconsistent rpm database, key %d not found. run 'rpm --rebuilddb' to fix.", dbid);
2030               freestate(&state);
2031               solv_free(entries);
2032               solv_free(namedata);
2033               solv_free(refhash);
2034               return -1;
2035             }
2036           rpmhead2solv(pool, repo, data, s, state.rpmhead, flags | RPM_ADD_TRIGGERS);
2037           if ((flags & RPMDB_REPORT_PROGRESS) != 0)
2038             {
2039               if (done < count)
2040                 done++;
2041               if (done < count && (done - 1) * 100 / count != done * 100 / count)
2042                 pool_debug(pool, SOLV_ERROR, "%%%% %d\n", done * 100 / count);
2043             }
2044         }
2045
2046       solv_free(entries);
2047       solv_free(namedata);
2048       solv_free(refhash);
2049       if (ref && (flags & RPMDB_EMPTY_REFREPO) != 0)
2050         repo_empty(ref, 1);
2051     }
2052
2053   freestate(&state);
2054   if (!(flags & REPO_NO_INTERNALIZE))
2055     repodata_internalize(data);
2056   if ((flags & RPMDB_REPORT_PROGRESS) != 0)
2057     pool_debug(pool, SOLV_ERROR, "%%%% 100\n");
2058   POOL_DEBUG(SOLV_DEBUG_STATS, "repo_add_rpmdb took %d ms\n", solv_timems(now));
2059   POOL_DEBUG(SOLV_DEBUG_STATS, "repo size: %d solvables\n", repo->nsolvables);
2060   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)));
2061   return 0;
2062 }
2063
2064 int
2065 repo_add_rpmdb_reffp(Repo *repo, FILE *fp, int flags)
2066 {
2067   int res;
2068   Repo *ref = 0;
2069
2070   if (!fp)
2071     return repo_add_rpmdb(repo, 0, flags);
2072   ref = repo_create(repo->pool, "add_rpmdb_reffp");
2073   if (repo_add_solv(ref, fp, 0) != 0)
2074     {
2075       repo_free(ref, 1);
2076       ref = 0;
2077     }
2078   if (ref && ref->start == ref->end)
2079     {
2080       repo_free(ref, 1);
2081       ref = 0;
2082     }
2083   if (ref)
2084     repo_disable_paging(ref);
2085   res = repo_add_rpmdb(repo, ref, flags | RPMDB_EMPTY_REFREPO);
2086   if (ref)
2087     repo_free(ref, 1);
2088   return res;
2089 }
2090
2091 #endif
2092
2093 Id
2094 repo_add_rpm(Repo *repo, const char *rpm, int flags)
2095 {
2096   unsigned int sigdsize, sigcnt, l;
2097   Pool *pool = repo->pool;
2098   Solvable *s;
2099   RpmHead *rpmhead = 0;
2100   int rpmheadsize = 0;
2101   char *payloadformat;
2102   FILE *fp;
2103   unsigned char lead[4096];
2104   int headerstart, headerend;
2105   struct stat stb;
2106   Repodata *data;
2107   unsigned char pkgid[16];
2108   unsigned char leadsigid[16];
2109   unsigned char hdrid[32];
2110   int pkgidtype, leadsigidtype, hdridtype;
2111   Id chksumtype = 0;
2112   Chksum *chksumh = 0;
2113   Chksum *leadsigchksumh = 0;
2114   int forcebinary = 0;
2115
2116   data = repo_add_repodata(repo, flags);
2117
2118   if ((flags & RPM_ADD_WITH_SHA256SUM) != 0)
2119     chksumtype = REPOKEY_TYPE_SHA256;
2120   else if ((flags & RPM_ADD_WITH_SHA1SUM) != 0)
2121     chksumtype = REPOKEY_TYPE_SHA1;
2122
2123   if ((fp = fopen(flags & REPO_USE_ROOTDIR ? pool_prepend_rootdir_tmp(pool, rpm) : rpm, "r")) == 0)
2124     {
2125       pool_error(pool, -1, "%s: %s", rpm, strerror(errno));
2126       return 0;
2127     }
2128   if (fstat(fileno(fp), &stb))
2129     {
2130       pool_error(pool, -1, "fstat: %s", strerror(errno));
2131       fclose(fp);
2132       return 0;
2133     }
2134   if (chksumtype)
2135     chksumh = solv_chksum_create(chksumtype);
2136   if ((flags & RPM_ADD_WITH_LEADSIGID) != 0)
2137     leadsigchksumh = solv_chksum_create(REPOKEY_TYPE_MD5);
2138   if (fread(lead, 96 + 16, 1, fp) != 1 || getu32(lead) != 0xedabeedb)
2139     {
2140       pool_error(pool, -1, "%s: not a rpm", rpm);
2141       fclose(fp);
2142       return 0;
2143     }
2144   forcebinary = lead[6] != 0 || lead[7] != 1;
2145   if (chksumh)
2146     solv_chksum_add(chksumh, lead, 96 + 16);
2147   if (leadsigchksumh)
2148     solv_chksum_add(leadsigchksumh, lead, 96 + 16);
2149   if (lead[78] != 0 || lead[79] != 5)
2150     {
2151       pool_error(pool, -1, "%s: not a rpm v5 header", rpm);
2152       fclose(fp);
2153       return 0;
2154     }
2155   if (getu32(lead + 96) != 0x8eade801)
2156     {
2157       pool_error(pool, -1, "%s: bad signature header", rpm);
2158       fclose(fp);
2159       return 0;
2160     }
2161   sigcnt = getu32(lead + 96 + 8);
2162   sigdsize = getu32(lead + 96 + 12);
2163   if (sigcnt >= MAX_SIG_CNT || sigdsize >= MAX_SIG_DSIZE)
2164     {
2165       pool_error(pool, -1, "%s: bad signature header", rpm);
2166       fclose(fp);
2167       return 0;
2168     }
2169   sigdsize += sigcnt * 16;
2170   sigdsize = (sigdsize + 7) & ~7;
2171   headerstart = 96 + 16 + sigdsize;
2172   pkgidtype = leadsigidtype = hdridtype = 0;
2173   if ((flags & (RPM_ADD_WITH_PKGID | RPM_ADD_WITH_HDRID)) != 0)
2174     {
2175       /* extract pkgid or hdrid from the signature header */
2176       if (sigdsize + 1 > rpmheadsize)
2177         {
2178           rpmheadsize = sigdsize + 128;
2179           rpmhead = solv_realloc(rpmhead, sizeof(*rpmhead) + rpmheadsize);
2180         }
2181       if (fread(rpmhead->data, sigdsize, 1, fp) != 1)
2182         {
2183           pool_error(pool, -1, "%s: unexpected EOF", rpm);
2184           fclose(fp);
2185           return 0;
2186         }
2187       rpmhead->data[sigdsize] = 0;
2188       if (chksumh)
2189         solv_chksum_add(chksumh, rpmhead->data, sigdsize);
2190       if (leadsigchksumh)
2191         solv_chksum_add(leadsigchksumh, rpmhead->data, sigdsize);
2192       rpmhead->forcebinary = 0;
2193       rpmhead->cnt = sigcnt;
2194       rpmhead->dcnt = sigdsize - sigcnt * 16;
2195       rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
2196       if ((flags & RPM_ADD_WITH_PKGID) != 0)
2197         {
2198           unsigned char *chksum;
2199           unsigned int chksumsize;
2200           chksum = headbinary(rpmhead, SIGTAG_MD5, &chksumsize);
2201           if (chksum && chksumsize == 16)
2202             {
2203               pkgidtype = REPOKEY_TYPE_MD5;
2204               memcpy(pkgid, chksum, 16);
2205             }
2206         }
2207       if ((flags & RPM_ADD_WITH_HDRID) != 0)
2208         {
2209           const char *str = headstring(rpmhead, TAG_SHA1HEADER);
2210           if (str && strlen(str) == 40)
2211             {
2212               if (solv_hex2bin(&str, hdrid, 20) == 20)
2213                 hdridtype = REPOKEY_TYPE_SHA1;
2214             }
2215           else if (str && strlen(str) == 64)
2216             {
2217               if (solv_hex2bin(&str, hdrid, 32) == 32)
2218                 hdridtype = REPOKEY_TYPE_SHA256;
2219             }
2220         }
2221     }
2222   else
2223     {
2224       /* just skip the signature header */
2225       while (sigdsize)
2226         {
2227           l = sigdsize > 4096 ? 4096 : sigdsize;
2228           if (fread(lead, l, 1, fp) != 1)
2229             {
2230               pool_error(pool, -1, "%s: unexpected EOF", rpm);
2231               fclose(fp);
2232               return 0;
2233             }
2234           if (chksumh)
2235             solv_chksum_add(chksumh, lead, l);
2236           if (leadsigchksumh)
2237             solv_chksum_add(leadsigchksumh, lead, l);
2238           sigdsize -= l;
2239         }
2240     }
2241   if (leadsigchksumh)
2242     {
2243       leadsigchksumh = solv_chksum_free(leadsigchksumh, leadsigid);
2244       leadsigidtype = REPOKEY_TYPE_MD5;
2245     }
2246   if (fread(lead, 16, 1, fp) != 1)
2247     {
2248       pool_error(pool, -1, "%s: unexpected EOF", rpm);
2249       fclose(fp);
2250       return 0;
2251     }
2252   if (chksumh)
2253     solv_chksum_add(chksumh, lead, 16);
2254   if (getu32(lead) != 0x8eade801)
2255     {
2256       pool_error(pool, -1, "%s: bad header", rpm);
2257       fclose(fp);
2258       return 0;
2259     }
2260   sigcnt = getu32(lead + 8);
2261   sigdsize = getu32(lead + 12);
2262   if (sigcnt >= MAX_HDR_CNT || sigdsize >= MAX_HDR_DSIZE)
2263     {
2264       pool_error(pool, -1, "%s: bad header", rpm);
2265       fclose(fp);
2266       return 0;
2267     }
2268   l = sigdsize + sigcnt * 16;
2269   headerend = headerstart + 16 + l;
2270   if (l + 1 > rpmheadsize)
2271     {
2272       rpmheadsize = l + 128;
2273       rpmhead = solv_realloc(rpmhead, sizeof(*rpmhead) + rpmheadsize);
2274     }
2275   if (fread(rpmhead->data, l, 1, fp) != 1)
2276     {
2277       pool_error(pool, -1, "%s: unexpected EOF", rpm);
2278       fclose(fp);
2279       return 0;
2280     }
2281   rpmhead->data[l] = 0;
2282   if (chksumh)
2283     solv_chksum_add(chksumh, rpmhead->data, l);
2284   rpmhead->forcebinary = forcebinary;
2285   rpmhead->cnt = sigcnt;
2286   rpmhead->dcnt = sigdsize;
2287   rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
2288   if (headexists(rpmhead, TAG_PATCHESNAME))
2289     {
2290       /* this is a patch rpm, ignore */
2291       pool_error(pool, -1, "%s: is patch rpm", rpm);
2292       fclose(fp);
2293       solv_chksum_free(chksumh, 0);
2294       solv_free(rpmhead);
2295       return 0;
2296     }
2297   payloadformat = headstring(rpmhead, TAG_PAYLOADFORMAT);
2298   if (payloadformat && !strcmp(payloadformat, "drpm"))
2299     {
2300       /* this is a delta rpm */
2301       pool_error(pool, -1, "%s: is delta rpm", rpm);
2302       fclose(fp);
2303       solv_chksum_free(chksumh, 0);
2304       solv_free(rpmhead);
2305       return 0;
2306     }
2307   if (chksumh)
2308     while ((l = fread(lead, 1, sizeof(lead), fp)) > 0)
2309       solv_chksum_add(chksumh, lead, l);
2310   fclose(fp);
2311   s = pool_id2solvable(pool, repo_add_solvable(repo));
2312   if (!rpmhead2solv(pool, repo, data, s, rpmhead, flags & ~(RPM_ADD_WITH_HDRID | RPM_ADD_WITH_PKGID)))
2313     {
2314       repo_free_solvable(repo, s - pool->solvables, 1);
2315       solv_chksum_free(chksumh, 0);
2316       solv_free(rpmhead);
2317       return 0;
2318     }
2319   if (!(flags & REPO_NO_LOCATION))
2320     repodata_set_location(data, s - pool->solvables, 0, 0, rpm);
2321   if (S_ISREG(stb.st_mode))
2322     repodata_set_num(data, s - pool->solvables, SOLVABLE_DOWNLOADSIZE, (unsigned long long)stb.st_size);
2323   repodata_set_num(data, s - pool->solvables, SOLVABLE_HEADEREND, headerend);
2324   if (pkgidtype)
2325     repodata_set_bin_checksum(data, s - pool->solvables, SOLVABLE_PKGID, pkgidtype, pkgid);
2326   if (hdridtype)
2327     repodata_set_bin_checksum(data, s - pool->solvables, SOLVABLE_HDRID, hdridtype, hdrid);
2328   if (leadsigidtype)
2329     repodata_set_bin_checksum(data, s - pool->solvables, SOLVABLE_LEADSIGID, leadsigidtype, leadsigid);
2330   if (chksumh)
2331     {
2332       repodata_set_bin_checksum(data, s - pool->solvables, SOLVABLE_CHECKSUM, chksumtype, solv_chksum_get(chksumh, 0));
2333       chksumh = solv_chksum_free(chksumh, 0);
2334     }
2335   solv_free(rpmhead);
2336   if (!(flags & REPO_NO_INTERNALIZE))
2337     repodata_internalize(data);
2338   return s - pool->solvables;
2339 }
2340
2341 Id
2342 repo_add_rpm_handle(Repo *repo, void *rpmhandle, int flags)
2343 {
2344   Pool *pool = repo->pool;
2345   Repodata *data;
2346   RpmHead *rpmhead = rpmhandle;
2347   Solvable *s;
2348   char *payloadformat;
2349
2350   data = repo_add_repodata(repo, flags);
2351   if (headexists(rpmhead, TAG_PATCHESNAME))
2352     {
2353       pool_error(pool, -1, "is a patch rpm");
2354       return 0;
2355     }
2356   payloadformat = headstring(rpmhead, TAG_PAYLOADFORMAT);
2357   if (payloadformat && !strcmp(payloadformat, "drpm"))
2358     {
2359       /* this is a delta rpm */
2360       pool_error(pool, -1, "is a delta rpm");
2361       return 0;
2362     }
2363   s = pool_id2solvable(pool, repo_add_solvable(repo));
2364   if (!rpmhead2solv(pool, repo, data, s, rpmhead, flags))
2365     {
2366       repo_free_solvable(repo, s - pool->solvables, 1);
2367       return 0;
2368     }
2369   if (!(flags & REPO_NO_INTERNALIZE))
2370     repodata_internalize(data);
2371   return s - pool->solvables;
2372 }
2373
2374 static inline void
2375 linkhash(const char *lt, char *hash)
2376 {
2377   unsigned int r = 0;
2378   const unsigned char *str = (const unsigned char *)lt;
2379   int l, c;
2380
2381   l = strlen(lt);
2382   while ((c = *str++) != 0)
2383     r += (r << 3) + c;
2384   sprintf(hash, "%08x%08x%08x%08x", r, l, 0, 0);
2385 }
2386
2387 void
2388 rpm_iterate_filelist(void *rpmhandle, int flags, void (*cb)(void *, const char *, struct filelistinfo *), void *cbdata)
2389 {
2390   RpmHead *rpmhead = rpmhandle;
2391   char **bn;
2392   char **dn;
2393   char **md = 0;
2394   char **lt = 0;
2395   unsigned int *di, diidx;
2396   unsigned int *co = 0;
2397   unsigned int *ff = 0;
2398   unsigned int lastdir;
2399   int lastdirl;
2400   unsigned int *fm;
2401   int cnt, dcnt, cnt2;
2402   int i, l1, l;
2403   char *space = 0;
2404   int spacen = 0;
2405   char md5[33];
2406   struct filelistinfo info;
2407
2408   dn = headstringarray(rpmhead, TAG_DIRNAMES, &dcnt);
2409   if (!dn)
2410     return;
2411   if ((flags & RPM_ITERATE_FILELIST_ONLYDIRS) != 0)
2412     {
2413       for (i = 0; i < dcnt; i++)
2414         (*cb)(cbdata, dn[i], 0);
2415       solv_free(dn);
2416       return;
2417     }
2418   bn = headstringarray(rpmhead, TAG_BASENAMES, &cnt);
2419   if (!bn)
2420     {
2421       solv_free(dn);
2422       return;
2423     }
2424   di = headint32array(rpmhead, TAG_DIRINDEXES, &cnt2);
2425   if (!di || cnt != cnt2)
2426     {
2427       solv_free(di);
2428       solv_free(bn);
2429       solv_free(dn);
2430       return;
2431     }
2432   fm = headint16array(rpmhead, TAG_FILEMODES, &cnt2);
2433   if (!fm || cnt != cnt2)
2434     {
2435       solv_free(fm);
2436       solv_free(di);
2437       solv_free(bn);
2438       solv_free(dn);
2439       return;
2440     }
2441   if ((flags & RPM_ITERATE_FILELIST_WITHMD5) != 0)
2442     {
2443       md = headstringarray(rpmhead, TAG_FILEMD5S, &cnt2);
2444       if (!md || cnt != cnt2)
2445         {
2446           solv_free(md);
2447           solv_free(fm);
2448           solv_free(di);
2449           solv_free(bn);
2450           solv_free(dn);
2451           return;
2452         }
2453     }
2454   if ((flags & RPM_ITERATE_FILELIST_WITHCOL) != 0)
2455     {
2456       co = headint32array(rpmhead, TAG_FILECOLORS, &cnt2);
2457       if (co && cnt != cnt2)
2458         {
2459           solv_free(co);
2460           solv_free(md);
2461           solv_free(fm);
2462           solv_free(di);
2463           solv_free(bn);
2464           solv_free(dn);
2465           return;
2466         }
2467     }
2468   if ((flags & RPM_ITERATE_FILELIST_NOGHOSTS) != 0)
2469     {
2470       ff = headint32array(rpmhead, TAG_FILEFLAGS, &cnt2);
2471       if (!ff || cnt != cnt2)
2472         {
2473           solv_free(ff);
2474           solv_free(co);
2475           solv_free(md);
2476           solv_free(fm);
2477           solv_free(di);
2478           solv_free(bn);
2479           solv_free(dn);
2480           return;
2481         }
2482     }
2483   lastdir = dcnt;
2484   lastdirl = 0;
2485   memset(&info, 0, sizeof(info));
2486   for (i = 0; i < cnt; i++)
2487     {
2488       if (ff && (ff[i] & FILEFLAG_GHOST) != 0)
2489         continue;
2490       diidx = di[i];
2491       if (diidx >= dcnt)
2492         continue;
2493       l1 = lastdir == diidx ? lastdirl : strlen(dn[diidx]);
2494       l = l1 + strlen(bn[i]) + 1;
2495       if (l > spacen)
2496         {
2497           spacen = l + 16;
2498           space = solv_realloc(space, spacen);
2499         }
2500       if (lastdir != diidx)
2501         {
2502           strcpy(space, dn[diidx]);
2503           lastdir = diidx;
2504           lastdirl = l1;
2505         }
2506       strcpy(space + l1, bn[i]);
2507       info.diridx = diidx;
2508       info.dirlen = l1;
2509       if (fm)
2510         info.mode = fm[i];
2511       if (md)
2512         {
2513           info.digest = md[i];
2514           if (fm && S_ISLNK(fm[i]))
2515             {
2516               info.digest = 0;
2517               if (!lt)
2518                 {
2519                   lt = headstringarray(rpmhead, TAG_FILELINKTOS, &cnt2);
2520                   if (cnt != cnt2)
2521                     lt = solv_free(lt);
2522                 }
2523               if (lt)
2524                 {
2525                   linkhash(lt[i], md5);
2526                   info.digest = md5;
2527                 }
2528             }
2529           if (!info.digest)
2530             {
2531               sprintf(md5, "%08x%08x%08x%08x", (fm[i] >> 12) & 65535, 0, 0, 0);
2532               info.digest = md5;
2533             }
2534         }
2535       info.color = co ? co[i] : 0;
2536       (*cb)(cbdata, space, &info);
2537     }
2538   solv_free(space);
2539   solv_free(lt);
2540   solv_free(md);
2541   solv_free(fm);
2542   solv_free(di);
2543   solv_free(bn);
2544   solv_free(dn);
2545   solv_free(co);
2546   solv_free(ff);
2547 }
2548
2549 char *
2550 rpm_query(void *rpmhandle, Id what)
2551 {
2552   const char *name, *arch, *sourcerpm;
2553   char *evr, *r;
2554   int l;
2555
2556   RpmHead *rpmhead = rpmhandle;
2557   r = 0;
2558   switch (what)
2559     {
2560     case 0:
2561       name = headstring(rpmhead, TAG_NAME);
2562       if (!name)
2563         name = "";
2564       sourcerpm = headstring(rpmhead, TAG_SOURCERPM);
2565       if (sourcerpm || (rpmhead->forcebinary && !headexists(rpmhead, TAG_SOURCEPACKAGE)))
2566         arch = headstring(rpmhead, TAG_ARCH);
2567       else
2568         {
2569           if (headexists(rpmhead, TAG_NOSOURCE) || headexists(rpmhead, TAG_NOPATCH))
2570             arch = "nosrc";
2571           else
2572             arch = "src";
2573         }
2574       if (!arch)
2575         arch = "noarch";
2576       evr = headtoevr(rpmhead);
2577       l = strlen(name) + 1 + strlen(evr ? evr : "") + 1 + strlen(arch) + 1;
2578       r = solv_malloc(l);
2579       sprintf(r, "%s-%s.%s", name, evr ? evr : "", arch);
2580       solv_free(evr);
2581       break;
2582     case SOLVABLE_NAME:
2583       name = headstring(rpmhead, TAG_NAME);
2584       r = solv_strdup(name);
2585       break;
2586     case SOLVABLE_SUMMARY:
2587       name = headstring(rpmhead, TAG_SUMMARY);
2588       r = solv_strdup(name);
2589       break;
2590     case SOLVABLE_DESCRIPTION:
2591       name = headstring(rpmhead, TAG_DESCRIPTION);
2592       r = solv_strdup(name);
2593       break;
2594     case SOLVABLE_EVR:
2595       r = headtoevr(rpmhead);
2596       break;
2597     }
2598   return r;
2599 }
2600
2601 unsigned long long
2602 rpm_query_num(void *rpmhandle, Id what, unsigned long long notfound)
2603 {
2604   RpmHead *rpmhead = rpmhandle;
2605   unsigned int u32;
2606
2607   switch (what)
2608     {
2609     case SOLVABLE_INSTALLTIME:
2610       u32 = headint32(rpmhead, TAG_INSTALLTIME);
2611       return u32 ? u32 : notfound;
2612     }
2613   return notfound;
2614 }
2615
2616 #ifdef ENABLE_RPMDB
2617
2618 int
2619 rpm_installedrpmdbids(void *rpmstate, const char *index, const char *match, Queue *rpmdbidq)
2620 {
2621   struct rpmdbentry *entries;
2622   int nentries, i;
2623
2624   entries = getinstalledrpmdbids(rpmstate, index ? index : "Name", match, &nentries, 0);
2625   if (rpmdbidq)
2626     {
2627       queue_empty(rpmdbidq);
2628       for (i = 0; i < nentries; i++)
2629         queue_push(rpmdbidq, entries[i].rpmdbid);
2630     }
2631   solv_free(entries);
2632   return nentries;
2633 }
2634
2635 void *
2636 rpm_byrpmdbid(void *rpmstate, Id rpmdbid)
2637 {
2638   struct rpmdbstate *state = rpmstate;
2639   int r;
2640
2641   r = getrpm_dbid(state, rpmdbid);
2642   if (!r)
2643     pool_error(state->pool, 0, "header #%d not in database", rpmdbid);
2644   return r <= 0 ? 0 : state->rpmhead;
2645 }
2646
2647 #endif
2648
2649 void *
2650 rpm_byfp(void *rpmstate, FILE *fp, const char *name)
2651 {
2652   struct rpmdbstate *state = rpmstate;
2653   /* int headerstart, headerend; */
2654   RpmHead *rpmhead;
2655   unsigned int sigdsize, sigcnt, l;
2656   unsigned char lead[4096];
2657   int forcebinary = 0;
2658
2659   if (fread(lead, 96 + 16, 1, fp) != 1 || getu32(lead) != 0xedabeedb)
2660     {
2661       pool_error(state->pool, 0, "%s: not a rpm", name);
2662       return 0;
2663     }
2664   forcebinary = lead[6] != 0 || lead[7] != 1;
2665   if (lead[78] != 0 || lead[79] != 5)
2666     {
2667       pool_error(state->pool, 0, "%s: not a V5 header", name);
2668       return 0;
2669     }
2670   if (getu32(lead + 96) != 0x8eade801)
2671     {
2672       pool_error(state->pool, 0, "%s: bad signature header", name);
2673       return 0;
2674     }
2675   sigcnt = getu32(lead + 96 + 8);
2676   sigdsize = getu32(lead + 96 + 12);
2677   if (sigcnt >= MAX_SIG_CNT || sigdsize >= MAX_SIG_DSIZE)
2678     {
2679       pool_error(state->pool, 0, "%s: bad signature header", name);
2680       return 0;
2681     }
2682   sigdsize += sigcnt * 16;
2683   sigdsize = (sigdsize + 7) & ~7;
2684   /* headerstart = 96 + 16 + sigdsize; */
2685   while (sigdsize)
2686     {
2687       l = sigdsize > 4096 ? 4096 : sigdsize;
2688       if (fread(lead, l, 1, fp) != 1)
2689         {
2690           pool_error(state->pool, 0, "%s: unexpected EOF", name);
2691           return 0;
2692         }
2693       sigdsize -= l;
2694     }
2695   if (fread(lead, 16, 1, fp) != 1)
2696     {
2697       pool_error(state->pool, 0, "%s: unexpected EOF", name);
2698       return 0;
2699     }
2700   if (getu32(lead) != 0x8eade801)
2701     {
2702       pool_error(state->pool, 0, "%s: bad header", name);
2703       return 0;
2704     }
2705   sigcnt = getu32(lead + 8);
2706   sigdsize = getu32(lead + 12);
2707   if (sigcnt >= MAX_HDR_CNT || sigdsize >= MAX_HDR_DSIZE)
2708     {
2709       pool_error(state->pool, 0, "%s: bad header", name);
2710       return 0;
2711     }
2712   l = sigdsize + sigcnt * 16;
2713   /* headerend = headerstart + 16 + l; */
2714   if (l + 1 > state->rpmheadsize)
2715     {
2716       state->rpmheadsize = l + 128;
2717       state->rpmhead = solv_realloc(state->rpmhead, sizeof(*state->rpmhead) + state->rpmheadsize);
2718     }
2719   rpmhead = state->rpmhead;
2720   if (fread(rpmhead->data, l, 1, fp) != 1)
2721     {
2722       pool_error(state->pool, 0, "%s: unexpected EOF", name);
2723       return 0;
2724     }
2725   rpmhead->data[l] = 0;
2726   rpmhead->forcebinary = forcebinary;
2727   rpmhead->cnt = sigcnt;
2728   rpmhead->dcnt = sigdsize;
2729   rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
2730   return rpmhead;
2731 }
2732
2733 #ifdef ENABLE_RPMDB_BYRPMHEADER
2734
2735 void *
2736 rpm_byrpmh(void *rpmstate, Header h)
2737 {
2738   struct rpmdbstate *state = rpmstate;
2739   const unsigned char *uh;
2740   unsigned int sigdsize, sigcnt, l;
2741   RpmHead *rpmhead;
2742
2743 #ifndef RPM5
2744   uh = headerUnload(h);
2745 #else
2746   uh = headerUnload(h, NULL);
2747 #endif
2748   if (!uh)
2749     return 0;
2750   sigcnt = getu32(uh);
2751   sigdsize = getu32(uh + 4);
2752   if (sigcnt >= MAX_HDR_CNT || sigdsize >= MAX_HDR_DSIZE)
2753     return 0;
2754   l = sigdsize + sigcnt * 16;
2755   if (l + 1 > state->rpmheadsize)
2756     {
2757       state->rpmheadsize = l + 128;
2758       state->rpmhead = solv_realloc(state->rpmhead, sizeof(*state->rpmhead) + state->rpmheadsize);
2759     }
2760   rpmhead = state->rpmhead;
2761   memcpy(rpmhead->data, uh + 8, l);
2762   rpmhead->data[l] = 0;
2763   free((void *)uh);
2764   rpmhead->forcebinary = 0;
2765   rpmhead->cnt = sigcnt;
2766   rpmhead->dcnt = sigdsize;
2767   rpmhead->dp = rpmhead->data + sigcnt * 16;
2768   return rpmhead;
2769 }
2770
2771 #endif
2772