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