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     default:
1604       if (solv_chksum_len(key->type))
1605         {
1606           repodata_set_bin_checksum(data, handle, keyname, key->type, (const unsigned char *)kv->str);
1607           break;
1608         }
1609       break;
1610     }
1611   return 0;
1612 }
1613
1614 static void
1615 solvable_copy(Solvable *s, Solvable *r, Repodata *data, Id *dircache)
1616 {
1617   int p, i;
1618   Repo *repo = s->repo;
1619   Pool *pool = repo->pool;
1620   Repo *fromrepo = r->repo;
1621   struct solvable_copy_cbdata cbdata;
1622
1623   /* copy solvable data */
1624   s->name = r->name;
1625   s->evr = r->evr;
1626   s->arch = r->arch;
1627   s->vendor = r->vendor;
1628   s->provides = copydeps(pool, repo, r->provides, fromrepo);
1629   s->requires = copydeps(pool, repo, r->requires, fromrepo);
1630   s->conflicts = copydeps(pool, repo, r->conflicts, fromrepo);
1631   s->obsoletes = copydeps(pool, repo, r->obsoletes, fromrepo);
1632   s->recommends = copydeps(pool, repo, r->recommends, fromrepo);
1633   s->suggests = copydeps(pool, repo, r->suggests, fromrepo);
1634   s->supplements = copydeps(pool, repo, r->supplements, fromrepo);
1635   s->enhances  = copydeps(pool, repo, r->enhances, fromrepo);
1636
1637   /* copy all attributes */
1638   if (!data)
1639     return;
1640   cbdata.data = data;
1641   cbdata.handle = s - pool->solvables;
1642   cbdata.subhandle = 0;
1643   cbdata.dircache = dircache;
1644   p = r - fromrepo->pool->solvables;
1645 #if 0
1646   repo_search(fromrepo, p, 0, 0, SEARCH_NO_STORAGE_SOLVABLE | SEARCH_SUB | SEARCH_ARRAYSENTINEL, solvable_copy_cb, &cbdata);
1647 #else
1648   FOR_REPODATAS(fromrepo, i, data)
1649     {
1650       if (p >= data->start && p < data->end)
1651         repodata_search(data, p, 0, SEARCH_SUB | SEARCH_ARRAYSENTINEL, solvable_copy_cb, &cbdata);
1652       cbdata.dircache = 0;      /* only for first repodata */
1653     }
1654 #endif
1655 }
1656
1657 /* used to sort entries by package name that got returned in some database order */
1658 static int
1659 rpmids_sort_cmp(const void *va, const void *vb, void *dp)
1660 {
1661   struct rpmdbentry const *a = va, *b = vb;
1662   char *namedata = dp;
1663   int r;
1664   r = strcmp(namedata + a->nameoff, namedata + b->nameoff);
1665   if (r)
1666     return r;
1667   return a->rpmdbid - b->rpmdbid;
1668 }
1669
1670 static int
1671 pkgids_sort_cmp(const void *va, const void *vb, void *dp)
1672 {
1673   Repo *repo = dp;
1674   Pool *pool = repo->pool;
1675   Solvable *a = pool->solvables + *(Id *)va;
1676   Solvable *b = pool->solvables + *(Id *)vb;
1677   Id *rpmdbid;
1678
1679   if (a->name != b->name)
1680     return strcmp(pool_id2str(pool, a->name), pool_id2str(pool, b->name));
1681   rpmdbid = repo->rpmdbid;
1682   return rpmdbid[(a - pool->solvables) - repo->start] - rpmdbid[(b - pool->solvables) - repo->start];
1683 }
1684
1685 static void
1686 swap_solvables(Repo *repo, Repodata *data, Id pa, Id pb)
1687 {
1688   Pool *pool = repo->pool;
1689   Solvable tmp;
1690
1691   tmp = pool->solvables[pa];
1692   pool->solvables[pa] = pool->solvables[pb];
1693   pool->solvables[pb] = tmp;
1694   if (repo->rpmdbid)
1695     {
1696       Id tmpid = repo->rpmdbid[pa - repo->start];
1697       repo->rpmdbid[pa - repo->start] = repo->rpmdbid[pb - repo->start];
1698       repo->rpmdbid[pb - repo->start] = tmpid;
1699     }
1700   /* only works if nothing is already internalized! */
1701   if (data)
1702     repodata_swap_attrs(data, pa, pb);
1703 }
1704
1705 static void
1706 mkrpmdbcookie(struct stat *st, unsigned char *cookie, int flags)
1707 {
1708   int f = 0;
1709   memset(cookie, 0, 32);
1710   cookie[3] = RPMDB_COOKIE_VERSION;
1711   memcpy(cookie + 16, &st->st_ino, sizeof(st->st_ino));
1712   memcpy(cookie + 24, &st->st_dev, sizeof(st->st_dev));
1713   if ((flags & RPM_ADD_WITH_PKGID) != 0)
1714     f |= 1;
1715   if ((flags & RPM_ADD_WITH_HDRID) != 0)
1716     f |= 2;
1717   if ((flags & RPM_ADD_WITH_CHANGELOG) != 0)
1718     f |= 4;
1719   if ((flags & RPM_ADD_NO_FILELIST) == 0)
1720     f |= 8;
1721   if ((flags & RPM_ADD_NO_RPMLIBREQS) != 0)
1722     cookie[1] = 1;
1723   cookie[0] = f;
1724 }
1725
1726 /*
1727  * read rpm db as repo
1728  *
1729  */
1730
1731 int
1732 repo_add_rpmdb(Repo *repo, Repo *ref, int flags)
1733 {
1734   Pool *pool = repo->pool;
1735   char dbpath[PATH_MAX];
1736   struct stat packagesstat;
1737   unsigned char newcookie[32];
1738   const unsigned char *oldcookie = 0;
1739   Id oldcookietype = 0;
1740   Repodata *data;
1741   int count = 0, done = 0;
1742   struct rpmdbstate state;
1743   int i;
1744   Solvable *s;
1745   unsigned int now;
1746
1747   now = solv_timems(0);
1748   memset(&state, 0, sizeof(state));
1749   state.pool = pool;
1750   if (flags & REPO_USE_ROOTDIR)
1751     state.rootdir = solv_strdup(pool_get_rootdir(pool));
1752
1753   data = repo_add_repodata(repo, flags);
1754
1755   if (ref && !(ref->nsolvables && ref->rpmdbid && ref->pool == repo->pool))
1756     {
1757       if ((flags & RPMDB_EMPTY_REFREPO) != 0)
1758         repo_empty(ref, 1);
1759       ref = 0;
1760     }
1761
1762   if (!opendbenv(&state))
1763     {
1764       solv_free(state.rootdir);
1765       return -1;
1766     }
1767
1768   /* XXX: should get ro lock of Packages database! */
1769   snprintf(dbpath, PATH_MAX, "%s/var/lib/rpm/Packages", state.rootdir ? state.rootdir : "");
1770   if (stat(dbpath, &packagesstat))
1771     {
1772       pool_error(pool, -1, "%s: %s", dbpath, strerror(errno));
1773       freestate(&state);
1774       return -1;
1775     }
1776   mkrpmdbcookie(&packagesstat, newcookie, flags);
1777   repodata_set_bin_checksum(data, SOLVID_META, REPOSITORY_RPMDBCOOKIE, REPOKEY_TYPE_SHA256, newcookie);
1778
1779   if (ref)
1780     oldcookie = repo_lookup_bin_checksum(ref, SOLVID_META, REPOSITORY_RPMDBCOOKIE, &oldcookietype);
1781   if (!ref || !oldcookie || oldcookietype != REPOKEY_TYPE_SHA256 || memcmp(oldcookie, newcookie, 32) != 0)
1782     {
1783       int solvstart = 0, solvend = 0;
1784       Id dbid;
1785       DBC *dbc = 0;
1786
1787       if (ref && (flags & RPMDB_EMPTY_REFREPO) != 0)
1788         repo_empty(ref, 1);     /* get it out of the way */
1789       if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1790         count = count_headers(&state);
1791       if (!openpkgdb(&state))
1792         {
1793           freestate(&state);
1794           return -1;
1795         }
1796       if (state.db->cursor(state.db, NULL, &dbc, 0))
1797         {
1798           freestate(&state);
1799           return pool_error(pool, -1, "db->cursor failed");
1800         }
1801       i = 0;
1802       s = 0;
1803       while ((dbid = getrpmcursor(&state, dbc)) != 0)
1804         {
1805           if (dbid == -1)
1806             {
1807               dbc->c_close(dbc);
1808               freestate(&state);
1809               return -1;
1810             }
1811           if (!s)
1812             {
1813               s = pool_id2solvable(pool, repo_add_solvable(repo));
1814               if (!solvstart)
1815                 solvstart = s - pool->solvables;
1816               solvend = s - pool->solvables + 1;
1817             }
1818           if (!repo->rpmdbid)
1819             repo->rpmdbid = repo_sidedata_create(repo, sizeof(Id));
1820           repo->rpmdbid[(s - pool->solvables) - repo->start] = dbid;
1821           if (rpm2solv(pool, repo, data, s, state.rpmhead, flags | RPM_ADD_TRIGGERS))
1822             {
1823               i++;
1824               s = 0;
1825             }
1826           else
1827             {
1828               /* We can reuse this solvable, but make sure it's still
1829                  associated with this repo.  */
1830               memset(s, 0, sizeof(*s));
1831               s->repo = repo;
1832             }
1833           if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1834             {
1835               if (done < count)
1836                 done++;
1837               if (done < count && (done - 1) * 100 / count != done * 100 / count)
1838                 pool_debug(pool, SOLV_ERROR, "%%%% %d\n", done * 100 / count);
1839             }
1840         }
1841       dbc->c_close(dbc);
1842       if (s)
1843         {
1844           /* oops, could not reuse. free it instead */
1845           repo_free_solvable(repo, s - pool->solvables, 1);
1846           solvend--;
1847           s = 0;
1848         }
1849       /* now sort all solvables in the new solvstart..solvend block */
1850       if (solvend - solvstart > 1)
1851         {
1852           Id *pkgids = solv_malloc2(solvend - solvstart, sizeof(Id));
1853           for (i = solvstart; i < solvend; i++)
1854             pkgids[i - solvstart] = i;
1855           solv_sort(pkgids, solvend - solvstart, sizeof(Id), pkgids_sort_cmp, repo);
1856           /* adapt order */
1857           for (i = solvstart; i < solvend; i++)
1858             {
1859               int j = pkgids[i - solvstart];
1860               while (j < i)
1861                 j = pkgids[i - solvstart] = pkgids[j - solvstart];
1862               if (j != i)
1863                 swap_solvables(repo, data, i, j);
1864             }
1865           solv_free(pkgids);
1866         }
1867     }
1868   else
1869     {
1870       Id dircache[COPYDIR_DIRCACHE_SIZE];               /* see copydir */
1871       struct rpmdbentry *entries = 0, *rp;
1872       int nentries = 0;
1873       char *namedata = 0;
1874       unsigned int refmask, h;
1875       Id id, *refhash;
1876       int res;
1877
1878       memset(dircache, 0, sizeof(dircache));
1879
1880       /* get ids of installed rpms */
1881       entries = getinstalledrpmdbids(&state, "Name", 0, &nentries, &namedata);
1882       if (!entries)
1883         {
1884           freestate(&state);
1885           return -1;
1886         }
1887
1888       /* sort by name */
1889       if (nentries > 1)
1890         solv_sort(entries, nentries, sizeof(*entries), rpmids_sort_cmp, namedata);
1891
1892       /* create hash from dbid to ref */
1893       refmask = mkmask(ref->nsolvables);
1894       refhash = solv_calloc(refmask + 1, sizeof(Id));
1895       for (i = 0; i < ref->end - ref->start; i++)
1896         {
1897           if (!ref->rpmdbid[i])
1898             continue;
1899           h = ref->rpmdbid[i] & refmask;
1900           while (refhash[h])
1901             h = (h + 317) & refmask;
1902           refhash[h] = i + 1;   /* make it non-zero */
1903         }
1904
1905       /* count the misses, they will cost us time */
1906       if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1907         {
1908           for (i = 0, rp = entries; i < nentries; i++, rp++)
1909             {
1910               if (refhash)
1911                 {
1912                   Id dbid = rp->rpmdbid;
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                     continue;
1922                 }
1923               count++;
1924             }
1925         }
1926
1927       if (ref && (flags & RPMDB_EMPTY_REFREPO) != 0)
1928         s = pool_id2solvable(pool, repo_add_solvable_block_before(repo, nentries, ref));
1929       else
1930         s = pool_id2solvable(pool, repo_add_solvable_block(repo, nentries));
1931       if (!repo->rpmdbid)
1932         repo->rpmdbid = repo_sidedata_create(repo, sizeof(Id));
1933
1934       for (i = 0, rp = entries; i < nentries; i++, rp++, s++)
1935         {
1936           Id dbid = rp->rpmdbid;
1937           repo->rpmdbid[(s - pool->solvables) - repo->start] = rp->rpmdbid;
1938           if (refhash)
1939             {
1940               h = dbid & refmask;
1941               while ((id = refhash[h]))
1942                 {
1943                   if (ref->rpmdbid[id - 1] == dbid)
1944                     break;
1945                   h = (h + 317) & refmask;
1946                 }
1947               if (id)
1948                 {
1949                   Solvable *r = ref->pool->solvables + ref->start + (id - 1);
1950                   if (r->repo == ref)
1951                     {
1952                       solvable_copy(s, r, data, dircache);
1953                       continue;
1954                     }
1955                 }
1956             }
1957           res = getrpmdbid(&state, dbid);
1958           if (res <= 0)
1959             {
1960               if (!res)
1961                 pool_error(pool, -1, "inconsistent rpm database, key %d not found. run 'rpm --rebuilddb' to fix.", dbid);
1962               freestate(&state);
1963               solv_free(entries);
1964               solv_free(namedata);
1965               solv_free(refhash);
1966               return -1;
1967             }
1968           rpm2solv(pool, repo, data, s, state.rpmhead, flags | RPM_ADD_TRIGGERS);
1969           if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1970             {
1971               if (done < count)
1972                 done++;
1973               if (done < count && (done - 1) * 100 / count != done * 100 / count)
1974                 pool_debug(pool, SOLV_ERROR, "%%%% %d\n", done * 100 / count);
1975             }
1976         }
1977
1978       solv_free(entries);
1979       solv_free(namedata);
1980       solv_free(refhash);
1981       if (ref && (flags & RPMDB_EMPTY_REFREPO) != 0)
1982         repo_empty(ref, 1);
1983     }
1984
1985   freestate(&state);
1986   if (!(flags & REPO_NO_INTERNALIZE))
1987     repodata_internalize(data);
1988   if ((flags & RPMDB_REPORT_PROGRESS) != 0)
1989     pool_debug(pool, SOLV_ERROR, "%%%% 100\n");
1990   POOL_DEBUG(SOLV_DEBUG_STATS, "repo_add_rpmdb took %d ms\n", solv_timems(now));
1991   POOL_DEBUG(SOLV_DEBUG_STATS, "repo size: %d solvables\n", repo->nsolvables);
1992   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)));
1993   return 0;
1994 }
1995
1996 int
1997 repo_add_rpmdb_reffp(Repo *repo, FILE *fp, int flags)
1998 {
1999   int res;
2000   Repo *ref = 0;
2001
2002   if (!fp)
2003     return repo_add_rpmdb(repo, 0, flags);
2004   ref = repo_create(repo->pool, "add_rpmdb_reffp");
2005   if (repo_add_solv(ref, fp, 0) != 0)
2006     {
2007       repo_free(ref, 1);
2008       ref = 0;
2009     }
2010   if (ref && ref->start == ref->end)
2011     {
2012       repo_free(ref, 1);
2013       ref = 0;
2014     }
2015   if (ref)
2016     repo_disable_paging(ref);
2017   res = repo_add_rpmdb(repo, ref, flags | RPMDB_EMPTY_REFREPO);
2018   if (ref)
2019     repo_free(ref, 1);
2020   return res;
2021 }
2022
2023 static inline unsigned int
2024 getu32(const unsigned char *dp)
2025 {
2026   return dp[0] << 24 | dp[1] << 16 | dp[2] << 8 | dp[3];
2027 }
2028
2029
2030 Id
2031 repo_add_rpm(Repo *repo, const char *rpm, int flags)
2032 {
2033   unsigned int sigdsize, sigcnt, l;
2034   Pool *pool = repo->pool;
2035   Solvable *s;
2036   RpmHead *rpmhead = 0;
2037   int rpmheadsize = 0;
2038   char *payloadformat;
2039   FILE *fp;
2040   unsigned char lead[4096];
2041   int headerstart, headerend;
2042   struct stat stb;
2043   Repodata *data;
2044   unsigned char pkgid[16];
2045   unsigned char leadsigid[16];
2046   unsigned char hdrid[32];
2047   int pkgidtype, leadsigidtype, hdridtype;
2048   Id chksumtype = 0;
2049   Chksum *chksumh = 0;
2050   Chksum *leadsigchksumh = 0;
2051   int forcebinary = 0;
2052
2053   data = repo_add_repodata(repo, flags);
2054
2055   if ((flags & RPM_ADD_WITH_SHA256SUM) != 0)
2056     chksumtype = REPOKEY_TYPE_SHA256;
2057   else if ((flags & RPM_ADD_WITH_SHA1SUM) != 0)
2058     chksumtype = REPOKEY_TYPE_SHA1;
2059
2060   if ((fp = fopen(flags & REPO_USE_ROOTDIR ? pool_prepend_rootdir_tmp(pool, rpm) : rpm, "r")) == 0)
2061     {
2062       pool_error(pool, -1, "%s: %s", rpm, strerror(errno));
2063       return 0;
2064     }
2065   if (fstat(fileno(fp), &stb))
2066     {
2067       pool_error(pool, -1, "fstat: %s", strerror(errno));
2068       fclose(fp);
2069       return 0;
2070     }
2071   if (chksumtype)
2072     chksumh = solv_chksum_create(chksumtype);
2073   if ((flags & RPM_ADD_WITH_LEADSIGID) != 0)
2074     leadsigchksumh = solv_chksum_create(REPOKEY_TYPE_MD5);
2075   if (fread(lead, 96 + 16, 1, fp) != 1 || getu32(lead) != 0xedabeedb)
2076     {
2077       pool_error(pool, -1, "%s: not a rpm", rpm);
2078       fclose(fp);
2079       return 0;
2080     }
2081   forcebinary = lead[6] != 0 || lead[7] != 1;
2082   if (chksumh)
2083     solv_chksum_add(chksumh, lead, 96 + 16);
2084   if (leadsigchksumh)
2085     solv_chksum_add(leadsigchksumh, lead, 96 + 16);
2086   if (lead[78] != 0 || lead[79] != 5)
2087     {
2088       pool_error(pool, -1, "%s: not a rpm v5 header", rpm);
2089       fclose(fp);
2090       return 0;
2091     }
2092   if (getu32(lead + 96) != 0x8eade801)
2093     {
2094       pool_error(pool, -1, "%s: bad signature header", rpm);
2095       fclose(fp);
2096       return 0;
2097     }
2098   sigcnt = getu32(lead + 96 + 8);
2099   sigdsize = getu32(lead + 96 + 12);
2100   if (sigcnt >= 0x100000 || sigdsize >= 0x100000)
2101     {
2102       pool_error(pool, -1, "%s: bad signature header", rpm);
2103       fclose(fp);
2104       return 0;
2105     }
2106   sigdsize += sigcnt * 16;
2107   sigdsize = (sigdsize + 7) & ~7;
2108   headerstart = 96 + 16 + sigdsize;
2109   pkgidtype = leadsigidtype = hdridtype = 0;
2110   if ((flags & (RPM_ADD_WITH_PKGID | RPM_ADD_WITH_HDRID)) != 0)
2111     {
2112       /* extract pkgid or hdrid from the signature header */
2113       if (sigdsize > rpmheadsize)
2114         {
2115           rpmheadsize = sigdsize + 128;
2116           rpmhead = solv_realloc(rpmhead, sizeof(*rpmhead) + rpmheadsize);
2117         }
2118       if (fread(rpmhead->data, sigdsize, 1, fp) != 1)
2119         {
2120           pool_error(pool, -1, "%s: unexpected EOF", rpm);
2121           fclose(fp);
2122           return 0;
2123         }
2124       if (chksumh)
2125         solv_chksum_add(chksumh, rpmhead->data, sigdsize);
2126       if (leadsigchksumh)
2127         solv_chksum_add(leadsigchksumh, rpmhead->data, sigdsize);
2128       rpmhead->forcebinary = 0;
2129       rpmhead->cnt = sigcnt;
2130       rpmhead->dcnt = sigdsize - sigcnt * 16;
2131       rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
2132       if ((flags & RPM_ADD_WITH_PKGID) != 0)
2133         {
2134           unsigned char *chksum;
2135           unsigned int chksumsize;
2136           chksum = headbinary(rpmhead, SIGTAG_MD5, &chksumsize);
2137           if (chksum && chksumsize == 16)
2138             {
2139               pkgidtype = REPOKEY_TYPE_MD5;
2140               memcpy(pkgid, chksum, 16);
2141             }
2142         }
2143       if ((flags & RPM_ADD_WITH_HDRID) != 0)
2144         {
2145           const char *str = headstring(rpmhead, TAG_SHA1HEADER);
2146           if (str && strlen(str) == 40)
2147             {
2148               if (solv_hex2bin(&str, hdrid, 20) == 20)
2149                 hdridtype = REPOKEY_TYPE_SHA1;
2150             }
2151           else if (str && strlen(str) == 64)
2152             {
2153               if (solv_hex2bin(&str, hdrid, 32) == 32)
2154                 hdridtype = REPOKEY_TYPE_SHA256;
2155             }
2156         }
2157     }
2158   else
2159     {
2160       /* just skip the signature header */
2161       while (sigdsize)
2162         {
2163           l = sigdsize > 4096 ? 4096 : sigdsize;
2164           if (fread(lead, l, 1, fp) != 1)
2165             {
2166               pool_error(pool, -1, "%s: unexpected EOF", rpm);
2167               fclose(fp);
2168               return 0;
2169             }
2170           if (chksumh)
2171             solv_chksum_add(chksumh, lead, l);
2172           if (leadsigchksumh)
2173             solv_chksum_add(leadsigchksumh, lead, l);
2174           sigdsize -= l;
2175         }
2176     }
2177   if (leadsigchksumh)
2178     {
2179       leadsigchksumh = solv_chksum_free(leadsigchksumh, leadsigid);
2180       leadsigidtype = REPOKEY_TYPE_MD5;
2181     }
2182   if (fread(lead, 16, 1, fp) != 1)
2183     {
2184       pool_error(pool, -1, "%s: unexpected EOF", rpm);
2185       fclose(fp);
2186       return 0;
2187     }
2188   if (chksumh)
2189     solv_chksum_add(chksumh, lead, 16);
2190   if (getu32(lead) != 0x8eade801)
2191     {
2192       pool_error(pool, -1, "%s: bad header", rpm);
2193       fclose(fp);
2194       return 0;
2195     }
2196   sigcnt = getu32(lead + 8);
2197   sigdsize = getu32(lead + 12);
2198   if (sigcnt >= 0x100000 || sigdsize >= 0x2000000)
2199     {
2200       pool_error(pool, -1, "%s: bad header", rpm);
2201       fclose(fp);
2202       return 0;
2203     }
2204   l = sigdsize + sigcnt * 16;
2205   headerend = headerstart + 16 + l;
2206   if (l > rpmheadsize)
2207     {
2208       rpmheadsize = l + 128;
2209       rpmhead = solv_realloc(rpmhead, sizeof(*rpmhead) + rpmheadsize);
2210     }
2211   if (fread(rpmhead->data, l, 1, fp) != 1)
2212     {
2213       pool_error(pool, -1, "%s: unexpected EOF", rpm);
2214       fclose(fp);
2215       return 0;
2216     }
2217   if (chksumh)
2218     solv_chksum_add(chksumh, rpmhead->data, l);
2219   rpmhead->forcebinary = forcebinary;
2220   rpmhead->cnt = sigcnt;
2221   rpmhead->dcnt = sigdsize;
2222   rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
2223   if (headexists(rpmhead, TAG_PATCHESNAME))
2224     {
2225       /* this is a patch rpm, ignore */
2226       pool_error(pool, -1, "%s: is patch rpm", rpm);
2227       fclose(fp);
2228       solv_chksum_free(chksumh, 0);
2229       solv_free(rpmhead);
2230       return 0;
2231     }
2232   payloadformat = headstring(rpmhead, TAG_PAYLOADFORMAT);
2233   if (payloadformat && !strcmp(payloadformat, "drpm"))
2234     {
2235       /* this is a delta rpm */
2236       pool_error(pool, -1, "%s: is delta rpm", rpm);
2237       fclose(fp);
2238       solv_chksum_free(chksumh, 0);
2239       solv_free(rpmhead);
2240       return 0;
2241     }
2242   if (chksumh)
2243     while ((l = fread(lead, 1, sizeof(lead), fp)) > 0)
2244       solv_chksum_add(chksumh, lead, l);
2245   fclose(fp);
2246   s = pool_id2solvable(pool, repo_add_solvable(repo));
2247   if (!rpm2solv(pool, repo, data, s, rpmhead, flags & ~(RPM_ADD_WITH_HDRID | RPM_ADD_WITH_PKGID)))
2248     {
2249       repo_free_solvable(repo, s - pool->solvables, 1);
2250       solv_chksum_free(chksumh, 0);
2251       solv_free(rpmhead);
2252       return 0;
2253     }
2254   if (!(flags & REPO_NO_LOCATION))
2255     repodata_set_location(data, s - pool->solvables, 0, 0, rpm);
2256   if (S_ISREG(stb.st_mode))
2257     repodata_set_num(data, s - pool->solvables, SOLVABLE_DOWNLOADSIZE, (unsigned long long)stb.st_size);
2258   repodata_set_num(data, s - pool->solvables, SOLVABLE_HEADEREND, headerend);
2259   if (pkgidtype)
2260     repodata_set_bin_checksum(data, s - pool->solvables, SOLVABLE_PKGID, pkgidtype, pkgid);
2261   if (hdridtype)
2262     repodata_set_bin_checksum(data, s - pool->solvables, SOLVABLE_HDRID, hdridtype, hdrid);
2263   if (leadsigidtype)
2264     repodata_set_bin_checksum(data, s - pool->solvables, SOLVABLE_LEADSIGID, leadsigidtype, leadsigid);
2265   if (chksumh)
2266     {
2267       repodata_set_bin_checksum(data, s - pool->solvables, SOLVABLE_CHECKSUM, chksumtype, solv_chksum_get(chksumh, 0));
2268       chksumh = solv_chksum_free(chksumh, 0);
2269     }
2270   solv_free(rpmhead);
2271   if (!(flags & REPO_NO_INTERNALIZE))
2272     repodata_internalize(data);
2273   return s - pool->solvables;
2274 }
2275
2276 Id
2277 repo_add_rpm_handle(Repo *repo, void *rpmhandle, int flags)
2278 {
2279   Pool *pool = repo->pool;
2280   Repodata *data;
2281   RpmHead *rpmhead = rpmhandle;
2282   Solvable *s;
2283   char *payloadformat;
2284
2285   data = repo_add_repodata(repo, flags);
2286   if (headexists(rpmhead, TAG_PATCHESNAME))
2287     {
2288       pool_error(pool, -1, "is a patch rpm");
2289       return 0;
2290     }
2291   payloadformat = headstring(rpmhead, TAG_PAYLOADFORMAT);
2292   if (payloadformat && !strcmp(payloadformat, "drpm"))
2293     {
2294       /* this is a delta rpm */
2295       pool_error(pool, -1, "is a delta rpm");
2296       return 0;
2297     }
2298   s = pool_id2solvable(pool, repo_add_solvable(repo));
2299   if (!rpm2solv(pool, repo, data, s, rpmhead, flags))
2300     {
2301       repo_free_solvable(repo, s - pool->solvables, 1);
2302       return 0;
2303     }
2304   if (!(flags & REPO_NO_INTERNALIZE))
2305     repodata_internalize(data);
2306   return s - pool->solvables;
2307 }
2308
2309 static inline void
2310 linkhash(const char *lt, char *hash)
2311 {
2312   unsigned int r = 0;
2313   const unsigned char *str = (const unsigned char *)lt;
2314   int l, c;
2315
2316   l = strlen(lt);
2317   while ((c = *str++) != 0)
2318     r += (r << 3) + c;
2319   sprintf(hash, "%08x%08x%08x%08x", r, l, 0, 0);
2320 }
2321
2322 void
2323 rpm_iterate_filelist(void *rpmhandle, int flags, void (*cb)(void *, const char *, struct filelistinfo *), void *cbdata)
2324 {
2325   RpmHead *rpmhead = rpmhandle;
2326   char **bn;
2327   char **dn;
2328   char **md = 0;
2329   char **lt = 0;
2330   unsigned int *di, diidx;
2331   unsigned int *co = 0;
2332   unsigned int *ff = 0;
2333   unsigned int lastdir;
2334   int lastdirl;
2335   unsigned int *fm;
2336   int cnt, dcnt, cnt2;
2337   int i, l1, l;
2338   char *space = 0;
2339   int spacen = 0;
2340   char md5[33];
2341   struct filelistinfo info;
2342
2343   dn = headstringarray(rpmhead, TAG_DIRNAMES, &dcnt);
2344   if (!dn)
2345     return;
2346   if ((flags & RPM_ITERATE_FILELIST_ONLYDIRS) != 0)
2347     {
2348       for (i = 0; i < dcnt; i++)
2349         (*cb)(cbdata, dn[i], 0);
2350       solv_free(dn);
2351       return;
2352     }
2353   bn = headstringarray(rpmhead, TAG_BASENAMES, &cnt);
2354   if (!bn)
2355     {
2356       solv_free(dn);
2357       return;
2358     }
2359   di = headint32array(rpmhead, TAG_DIRINDEXES, &cnt2);
2360   if (!di || cnt != cnt2)
2361     {
2362       solv_free(di);
2363       solv_free(bn);
2364       solv_free(dn);
2365       return;
2366     }
2367   fm = headint16array(rpmhead, TAG_FILEMODES, &cnt2);
2368   if (!fm || cnt != cnt2)
2369     {
2370       solv_free(fm);
2371       solv_free(di);
2372       solv_free(bn);
2373       solv_free(dn);
2374       return;
2375     }
2376   if ((flags & RPM_ITERATE_FILELIST_WITHMD5) != 0)
2377     {
2378       md = headstringarray(rpmhead, TAG_FILEMD5S, &cnt2);
2379       if (!md || cnt != cnt2)
2380         {
2381           solv_free(md);
2382           solv_free(fm);
2383           solv_free(di);
2384           solv_free(bn);
2385           solv_free(dn);
2386           return;
2387         }
2388     }
2389   if ((flags & RPM_ITERATE_FILELIST_WITHCOL) != 0)
2390     {
2391       co = headint32array(rpmhead, TAG_FILECOLORS, &cnt2);
2392       if (!co || cnt != cnt2)
2393         {
2394           solv_free(co);
2395           solv_free(md);
2396           solv_free(fm);
2397           solv_free(di);
2398           solv_free(bn);
2399           solv_free(dn);
2400           return;
2401         }
2402     }
2403   if ((flags & RPM_ITERATE_FILELIST_NOGHOSTS) != 0)
2404     {
2405       ff = headint32array(rpmhead, TAG_FILEFLAGS, &cnt2);
2406       if (!ff || cnt != cnt2)
2407         {
2408           solv_free(ff);
2409           solv_free(co);
2410           solv_free(md);
2411           solv_free(fm);
2412           solv_free(di);
2413           solv_free(bn);
2414           solv_free(dn);
2415           return;
2416         }
2417     }
2418   lastdir = dcnt;
2419   lastdirl = 0;
2420   memset(&info, 0, sizeof(info));
2421   for (i = 0; i < cnt; i++)
2422     {
2423       if (ff && (ff[i] & FILEFLAG_GHOST) != 0)
2424         continue;
2425       diidx = di[i];
2426       if (diidx >= dcnt)
2427         continue;
2428       l1 = lastdir == diidx ? lastdirl : strlen(dn[diidx]);
2429       l = l1 + strlen(bn[i]) + 1;
2430       if (l > spacen)
2431         {
2432           spacen = l + 16;
2433           space = solv_realloc(space, spacen);
2434         }
2435       if (lastdir != diidx)
2436         {
2437           strcpy(space, dn[diidx]);
2438           lastdir = diidx;
2439           lastdirl = l1;
2440         }
2441       strcpy(space + l1, bn[i]);
2442       info.diridx = diidx;
2443       info.dirlen = l1;
2444       if (fm)
2445         info.mode = fm[i];
2446       if (md)
2447         {
2448           info.digest = md[i];
2449           if (fm && S_ISLNK(fm[i]))
2450             {
2451               info.digest = 0;
2452               if (!lt)
2453                 {
2454                   lt = headstringarray(rpmhead, TAG_FILELINKTOS, &cnt2);
2455                   if (cnt != cnt2)
2456                     lt = solv_free(lt);
2457                 }
2458               if (lt)
2459                 {
2460                   linkhash(lt[i], md5);
2461                   info.digest = md5;
2462                 }
2463             }
2464           if (!info.digest)
2465             {
2466               sprintf(md5, "%08x%08x%08x%08x", (fm[i] >> 12) & 65535, 0, 0, 0);
2467               info.digest = md5;
2468             }
2469         }
2470       if (co)
2471         info.color = co[i];
2472       (*cb)(cbdata, space, &info);
2473     }
2474   solv_free(space);
2475   solv_free(lt);
2476   solv_free(md);
2477   solv_free(fm);
2478   solv_free(di);
2479   solv_free(bn);
2480   solv_free(dn);
2481   solv_free(co);
2482   solv_free(ff);
2483 }
2484
2485 char *
2486 rpm_query(void *rpmhandle, Id what)
2487 {
2488   const char *name, *arch, *sourcerpm;
2489   char *evr, *r;
2490   int l;
2491
2492   RpmHead *rpmhead = rpmhandle;
2493   r = 0;
2494   switch (what)
2495     {
2496     case 0:
2497       name = headstring(rpmhead, TAG_NAME);
2498       if (!name)
2499         name = "";
2500       sourcerpm = headstring(rpmhead, TAG_SOURCERPM);
2501       if (sourcerpm || (rpmhead->forcebinary && !headexists(rpmhead, TAG_SOURCEPACKAGE)))
2502         arch = headstring(rpmhead, TAG_ARCH);
2503       else
2504         {
2505           if (headexists(rpmhead, TAG_NOSOURCE) || headexists(rpmhead, TAG_NOPATCH))
2506             arch = "nosrc";
2507           else
2508             arch = "src";
2509         }
2510       if (!arch)
2511         arch = "noarch";
2512       evr = headtoevr(rpmhead);
2513       l = strlen(name) + 1 + strlen(evr ? evr : "") + 1 + strlen(arch) + 1;
2514       r = solv_malloc(l);
2515       sprintf(r, "%s-%s.%s", name, evr ? evr : "", arch);
2516       solv_free(evr);
2517       break;
2518     case SOLVABLE_NAME:
2519       name = headstring(rpmhead, TAG_NAME);
2520       r = solv_strdup(name);
2521       break;
2522     case SOLVABLE_SUMMARY:
2523       name = headstring(rpmhead, TAG_SUMMARY);
2524       r = solv_strdup(name);
2525       break;
2526     case SOLVABLE_DESCRIPTION:
2527       name = headstring(rpmhead, TAG_DESCRIPTION);
2528       r = solv_strdup(name);
2529       break;
2530     case SOLVABLE_EVR:
2531       r = headtoevr(rpmhead);
2532       break;
2533     }
2534   return r;
2535 }
2536
2537 unsigned long long
2538 rpm_query_num(void *rpmhandle, Id what, unsigned long long notfound)
2539 {
2540   RpmHead *rpmhead = rpmhandle;
2541   unsigned int u32;
2542
2543   switch (what)
2544     {
2545     case SOLVABLE_INSTALLTIME:
2546       u32 = headint32(rpmhead, TAG_INSTALLTIME);
2547       return u32 ? u32 : notfound;
2548     }
2549   return notfound;
2550 }
2551
2552 int
2553 rpm_installedrpmdbids(void *rpmstate, const char *index, const char *match, Queue *rpmdbidq)
2554 {
2555   struct rpmdbentry *entries;
2556   int nentries, i;
2557
2558   entries = getinstalledrpmdbids(rpmstate, index ? index : "Name", match, &nentries, 0);
2559   if (rpmdbidq)
2560     {
2561       queue_empty(rpmdbidq);
2562       for (i = 0; i < nentries; i++)
2563         queue_push(rpmdbidq, entries[i].rpmdbid);
2564     }
2565   solv_free(entries);
2566   return nentries;
2567 }
2568
2569 void *
2570 rpm_byrpmdbid(void *rpmstate, Id rpmdbid)
2571 {
2572   struct rpmdbstate *state = rpmstate;
2573   int r;
2574
2575   r = getrpmdbid(state, rpmdbid);
2576   if (!r)
2577     pool_error(state->pool, 0, "header #%d not in database", rpmdbid);
2578   return r <= 0 ? 0 : state->rpmhead;
2579 }
2580
2581 void *
2582 rpm_byfp(void *rpmstate, FILE *fp, const char *name)
2583 {
2584   struct rpmdbstate *state = rpmstate;
2585   /* int headerstart, headerend; */
2586   RpmHead *rpmhead;
2587   unsigned int sigdsize, sigcnt, l;
2588   unsigned char lead[4096];
2589   int forcebinary = 0;
2590
2591   if (fread(lead, 96 + 16, 1, fp) != 1 || getu32(lead) != 0xedabeedb)
2592     {
2593       pool_error(state->pool, 0, "%s: not a rpm", name);
2594       return 0;
2595     }
2596   forcebinary = lead[6] != 0 || lead[7] != 1;
2597   if (lead[78] != 0 || lead[79] != 5)
2598     {
2599       pool_error(state->pool, 0, "%s: not a V5 header", name);
2600       return 0;
2601     }
2602   if (getu32(lead + 96) != 0x8eade801)
2603     {
2604       pool_error(state->pool, 0, "%s: bad signature header", name);
2605       return 0;
2606     }
2607   sigcnt = getu32(lead + 96 + 8);
2608   sigdsize = getu32(lead + 96 + 12);
2609   if (sigcnt >= 0x100000 || sigdsize >= 0x100000)
2610     {
2611       pool_error(state->pool, 0, "%s: bad signature header", name);
2612       return 0;
2613     }
2614   sigdsize += sigcnt * 16;
2615   sigdsize = (sigdsize + 7) & ~7;
2616   /* headerstart = 96 + 16 + sigdsize; */
2617   while (sigdsize)
2618     {
2619       l = sigdsize > 4096 ? 4096 : sigdsize;
2620       if (fread(lead, l, 1, fp) != 1)
2621         {
2622           pool_error(state->pool, 0, "%s: unexpected EOF", name);
2623           return 0;
2624         }
2625       sigdsize -= l;
2626     }
2627   if (fread(lead, 16, 1, fp) != 1)
2628     {
2629       pool_error(state->pool, 0, "%s: unexpected EOF", name);
2630       return 0;
2631     }
2632   if (getu32(lead) != 0x8eade801)
2633     {
2634       pool_error(state->pool, 0, "%s: bad header", name);
2635       return 0;
2636     }
2637   sigcnt = getu32(lead + 8);
2638   sigdsize = getu32(lead + 12);
2639   if (sigcnt >= 0x100000 || sigdsize >= 0x2000000)
2640     {
2641       pool_error(state->pool, 0, "%s: bad header", name);
2642       return 0;
2643     }
2644   l = sigdsize + sigcnt * 16;
2645   /* headerend = headerstart + 16 + l; */
2646   if (l > state->rpmheadsize)
2647     {
2648       state->rpmheadsize = l + 128;
2649       state->rpmhead = solv_realloc(state->rpmhead, sizeof(*state->rpmhead) + state->rpmheadsize);
2650     }
2651   rpmhead = state->rpmhead;
2652   if (fread(rpmhead->data, l, 1, fp) != 1)
2653     {
2654       pool_error(state->pool, 0, "%s: unexpected EOF", name);
2655       return 0;
2656     }
2657   rpmhead->forcebinary = forcebinary;
2658   rpmhead->cnt = sigcnt;
2659   rpmhead->dcnt = sigdsize;
2660   rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
2661   return rpmhead;
2662 }
2663
2664 #ifdef ENABLE_RPMDB_BYRPMHEADER
2665
2666 void *
2667 rpm_byrpmh(void *rpmstate, Header h)
2668 {
2669   struct rpmdbstate *state = rpmstate;
2670   const unsigned char *uh;
2671   unsigned int sigdsize, sigcnt, l;
2672   RpmHead *rpmhead;
2673
2674 #ifndef RPM5
2675   uh = headerUnload(h);
2676 #else
2677   uh = headerUnload(h, NULL);
2678 #endif
2679   if (!uh)
2680     return 0;
2681   sigcnt = getu32(uh);
2682   sigdsize = getu32(uh + 4);
2683   l = sigdsize + sigcnt * 16;
2684   if (l > state->rpmheadsize)
2685     {
2686       state->rpmheadsize = l + 128;
2687       state->rpmhead = solv_realloc(state->rpmhead, sizeof(*state->rpmhead) + state->rpmheadsize);
2688     }
2689   rpmhead = state->rpmhead;
2690   memcpy(rpmhead->data, uh + 8, l - 8);
2691   free((void *)uh);
2692   rpmhead->forcebinary = 0;
2693   rpmhead->cnt = sigcnt;
2694   rpmhead->dcnt = sigdsize;
2695   rpmhead->dp = rpmhead->data + rpmhead->cnt * 16;
2696   return rpmhead;
2697 }
2698
2699 #endif
2700