ext/repo_rpmdb.c

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