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