ext/repo_rpmdb.c

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