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