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