ext/repo_rpmdb.c

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