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