lib/header.c

   1 /** \ingroup header
   2  * \file lib/header.c
   3  */
   4
   5 /* RPM - Copyright (C) 1995-2002 Red Hat Software */
   6
   7 /* Data written to file descriptors is in network byte order.    */
   8 /* Data read from file descriptors is expected to be in          */
   9 /* network byte order and is converted on the fly to host order. */
  10
  11 #include "system.h"
  12
  13 #include <rpm/rpmtag.h>
  14 #include <rpm/rpmstring.h>
  15 #include "lib/header_internal.h"
  16
  17 #include "debug.h"
  18
  19 int _hdr_debug = 0;
  20
  21 /** \ingroup header
  22  */
  23 static unsigned char const header_magic[8] = {
  24         0x8e, 0xad, 0xe8, 0x01, 0x00, 0x00, 0x00, 0x00
  25 };
  26
  27 /** \ingroup header
  28  * Alignment needed for header data types.
  29  */
  30 static const int typeAlign[16] =  {
  31     1,  /*!< RPM_NULL_TYPE */
  32     1,  /*!< RPM_CHAR_TYPE */
  33     1,  /*!< RPM_INT8_TYPE */
  34     2,  /*!< RPM_INT16_TYPE */
  35     4,  /*!< RPM_INT32_TYPE */
  36     8,  /*!< RPM_INT64_TYPE */
  37     1,  /*!< RPM_STRING_TYPE */
  38     1,  /*!< RPM_BIN_TYPE */
  39     1,  /*!< RPM_STRING_ARRAY_TYPE */
  40     1,  /*!< RPM_I18NSTRING_TYPE */
  41     0,
  42     0,
  43     0,
  44     0,
  45     0,
  46     0
  47 };
  48
  49 /** \ingroup header
  50  * Size of header data types.
  51  */
  52 static const int typeSizes[16] =  {
  53     0,  /*!< RPM_NULL_TYPE */
  54     1,  /*!< RPM_CHAR_TYPE */
  55     1,  /*!< RPM_INT8_TYPE */
  56     2,  /*!< RPM_INT16_TYPE */
  57     4,  /*!< RPM_INT32_TYPE */
  58     -1, /*!< RPM_INT64_TYPE */
  59     -1, /*!< RPM_STRING_TYPE */
  60     1,  /*!< RPM_BIN_TYPE */
  61     -1, /*!< RPM_STRING_ARRAY_TYPE */
  62     -1, /*!< RPM_I18NSTRING_TYPE */
  63     0,
  64     0,
  65     0,
  66     0,
  67     0,
  68     0
  69 };
  70
  71 /* dumb macro to avoid 50 copies of this code while converting... */
  72 #define TDWRAP() \
  73     if (type) \
  74         *type = td.type; \
  75     if (p) \
  76         *p = td.data; \
  77     else \
  78         rpmtdFreeData(&td); \
  79     if (c) \
  80         *c = td.count
  81
  82 /** \ingroup header
  83  * Maximum no. of bytes permitted in a header.
  84  */
  85 static const size_t headerMaxbytes = (32*1024*1024);
  86
  87 /** \ingroup header
  88  * HEADER_EXT_TAG format function prototype.
  89  * This is allowed to fail, which indicates the tag doesn't exist.
  90  *
  91  * @param h             header
  92  * @retval td           tag data container
  93  * @return              0 on success
  94  */
  95 typedef int (*headerTagTagFunction) (Header h, rpmtd td);
  96
  97 extern void *rpmHeaderTagFunc(rpmTag tag);
  98
  99 Header headerLink(Header h)
 100 {
 101     if (h == NULL) return NULL;
 102
 103     h->nrefs++;
 104 if (_hdr_debug)
 105 fprintf(stderr, "--> h  %p ++ %d at %s:%u\n", h, h->nrefs, __FILE__, __LINE__);
 106
 107     return h;
 108 }
 109
 110 Header headerUnlink(Header h)
 111 {
 112     if (h == NULL) return NULL;
 113 if (_hdr_debug)
 114 fprintf(stderr, "--> h  %p -- %d at %s:%u\n", h, h->nrefs, __FILE__, __LINE__);
 115     h->nrefs--;
 116     return NULL;
 117 }
 118
 119 Header headerFree(Header h)
 120 {
 121     (void) headerUnlink(h);
 122
 123     if (h == NULL || h->nrefs > 0)
 124         return NULL;    /* XXX return previous header? */
 125
 126     if (h->index) {
 127         indexEntry entry = h->index;
 128         int i;
 129         for (i = 0; i < h->indexUsed; i++, entry++) {
 130             if ((h->flags & HEADERFLAG_ALLOCATED) && ENTRY_IS_REGION(entry)) {
 131                 if (entry->length > 0) {
 132                     int32_t * ei = entry->data;
 133                     if ((ei - 2) == h->blob) h->blob = _free(h->blob);
 134                     entry->data = NULL;
 135                 }
 136             } else if (!ENTRY_IN_REGION(entry)) {
 137                 entry->data = _free(entry->data);
 138             }
 139             entry->data = NULL;
 140         }
 141         h->index = _free(h->index);
 142     }
 143
 144     h = _free(h);
 145     return h;
 146 }
 147
 148 Header headerNew(void)
 149 {
 150     Header h = xcalloc(1, sizeof(*h));
 151
 152     h->blob = NULL;
 153     h->indexAlloced = INDEX_MALLOC_SIZE;
 154     h->indexUsed = 0;
 155     h->flags |= HEADERFLAG_SORTED;
 156
 157     h->index = (h->indexAlloced
 158         ? xcalloc(h->indexAlloced, sizeof(*h->index))
 159         : NULL);
 160
 161     h->nrefs = 0;
 162     return headerLink(h);
 163 }
 164
 165 int headerVerifyInfo(int il, int dl, const void * pev, void * iv, int negate)
 166 {
 167     entryInfo pe = (entryInfo) pev;
 168     entryInfo info = iv;
 169     int i;
 170
 171     for (i = 0; i < il; i++) {
 172         info->tag = ntohl(pe[i].tag);
 173         info->type = ntohl(pe[i].type);
 174         info->offset = ntohl(pe[i].offset);
 175         if (negate)
 176             info->offset = -info->offset;
 177         info->count = ntohl(pe[i].count);
 178
 179         if (hdrchkType(info->type))
 180             return i;
 181         if (hdrchkAlign(info->type, info->offset))
 182             return i;
 183         if (!negate && hdrchkRange(dl, info->offset))
 184             return i;
 185         if (hdrchkData(info->count))
 186             return i;
 187
 188     }
 189     return -1;
 190 }
 191
 192 /**
 193  */
 194 static int indexCmp(const void * avp, const void * bvp)
 195 {
 196     indexEntry ap = (indexEntry) avp, bp = (indexEntry) bvp;
 197     return (ap->info.tag - bp->info.tag);
 198 }
 199
 200 void headerSort(Header h)
 201 {
 202     if (!(h->flags & HEADERFLAG_SORTED)) {
 203         qsort(h->index, h->indexUsed, sizeof(*h->index), indexCmp);
 204         h->flags |= HEADERFLAG_SORTED;
 205     }
 206 }
 207
 208 /**
 209  */
 210 static int offsetCmp(const void * avp, const void * bvp)
 211 {
 212     indexEntry ap = (indexEntry) avp, bp = (indexEntry) bvp;
 213     int rc = (ap->info.offset - bp->info.offset);
 214
 215     if (rc == 0) {
 216         /* Within a region, entries sort by address. Added drips sort by tag. */
 217         if (ap->info.offset < 0)
 218             rc = (((char *)ap->data) - ((char *)bp->data));
 219         else
 220             rc = (ap->info.tag - bp->info.tag);
 221     }
 222     return rc;
 223 }
 224
 225 /** \ingroup header
 226  * Restore tags in header to original ordering.
 227  * @param h             header
 228  */
 229 void headerUnsort(Header h)
 230 {
 231     qsort(h->index, h->indexUsed, sizeof(*h->index), offsetCmp);
 232 }
 233
 234 unsigned headerSizeof(Header h, enum hMagic magicp)
 235 {
 236     indexEntry entry;
 237     unsigned int size = 0;
 238     unsigned int pad = 0;
 239     int i;
 240
 241     if (h == NULL)
 242         return size;
 243
 244     headerSort(h);
 245
 246     switch (magicp) {
 247     case HEADER_MAGIC_YES:
 248         size += sizeof(header_magic);
 249         break;
 250     case HEADER_MAGIC_NO:
 251         break;
 252     }
 253
 254     size += 2 * sizeof(int32_t);        /* count of index entries */
 255
 256     for (i = 0, entry = h->index; i < h->indexUsed; i++, entry++) {
 257         unsigned diff;
 258         rpmTagType type;
 259
 260         /* Regions go in as is ... */
 261         if (ENTRY_IS_REGION(entry)) {
 262             size += entry->length;
 263             /* XXX Legacy regions do not include the region tag and data. */
 264             if (i == 0 && (h->flags & HEADERFLAG_LEGACY))
 265                 size += sizeof(struct entryInfo_s) + entry->info.count;
 266             continue;
 267         }
 268
 269         /* ... and region elements are skipped. */
 270         if (entry->info.offset < 0)
 271             continue;
 272
 273         /* Alignment */
 274         type = entry->info.type;
 275         if (typeSizes[type] > 1) {
 276             diff = typeSizes[type] - (size % typeSizes[type]);
 277             if (diff != typeSizes[type]) {
 278                 size += diff;
 279                 pad += diff;
 280             }
 281         }
 282
 283         size += sizeof(struct entryInfo_s) + entry->length;
 284     }
 285
 286     return size;
 287 }
 288
 289 /**
 290  * Return length of entry data.
 291  * @param type          entry data type
 292  * @param p             entry data
 293  * @param count         entry item count
 294  * @param onDisk        data is concatenated strings (with NUL's))?
 295  * @param pend          pointer to end of data (or NULL)
 296  * @return              no. bytes in data, -1 on failure
 297  */
 298 static int dataLength(rpmTagType type, rpm_constdata_t p, rpm_count_t count,
 299                          int onDisk, rpm_constdata_t pend)
 300 {
 301     const unsigned char * s = p;
 302     const unsigned char * se = pend;
 303     int length = 0;
 304
 305     switch (type) {
 306     case RPM_STRING_TYPE:
 307         if (count != 1)
 308             return -1;
 309         while (*s++) {
 310             if (se && s > se)
 311                 return -1;
 312             length++;
 313         }
 314         length++;       /* count nul terminator too. */
 315         break;
 316
 317     case RPM_STRING_ARRAY_TYPE:
 318     case RPM_I18NSTRING_TYPE:
 319         /* These are like RPM_STRING_TYPE, except they're *always* an array */
 320         /* Compute sum of length of all strings, including nul terminators */
 321
 322         if (onDisk) {
 323             while (count--) {
 324                 length++;       /* count nul terminator too */
 325                while (*s++) {
 326                     if (se && s > se)
 327                         return -1;
 328                     length++;
 329                 }
 330             }
 331         } else {
 332             const char ** av = (const char **)p;
 333             while (count--) {
 334                 /* add one for null termination */
 335                 length += strlen(*av++) + 1;
 336             }
 337         }
 338         break;
 339
 340     default:
 341         if (typeSizes[type] == -1)
 342             return -1;
 343         length = typeSizes[(type & 0xf)] * count;
 344         if (length < 0 || (se && (s + length) > se))
 345             return -1;
 346         break;
 347     }
 348
 349     return length;
 350 }
 351
 352 /** \ingroup header
 353  * Swap int32_t and int16_t arrays within header region.
 354  *
 355  * This code is way more twisty than I would like.
 356  *
 357  * A bug with RPM_I18NSTRING_TYPE in rpm-2.5.x (fixed in August 1998)
 358  * causes the offset and length of elements in a header region to disagree
 359  * regarding the total length of the region data.
 360  *
 361  * The "fix" is to compute the size using both offset and length and
 362  * return the larger of the two numbers as the size of the region.
 363  * Kinda like computing left and right Riemann sums of the data elements
 364  * to determine the size of a data structure, go figger :-).
 365  *
 366  * There's one other twist if a header region tag is in the set to be swabbed,
 367  * as the data for a header region is located after all other tag data.
 368  *
 369  * @param entry         header entry
 370  * @param il            no. of entries
 371  * @param dl            start no. bytes of data
 372  * @param pe            header physical entry pointer (swapped)
 373  * @param dataStart     header data start
 374  * @param dataEnd       header data end
 375  * @param regionid      region offset
 376  * @return              no. bytes of data in region, -1 on error
 377  */
 378 static int regionSwab(indexEntry entry, int il, int dl,
 379                 entryInfo pe,
 380                 unsigned char * dataStart,
 381                 const unsigned char * dataEnd,
 382                 int regionid)
 383 {
 384     unsigned char * tprev = NULL;
 385     unsigned char * t = NULL;
 386     int tdel = 0;
 387     int tl = dl;
 388     struct indexEntry_s ieprev;
 389
 390     memset(&ieprev, 0, sizeof(ieprev));
 391     for (; il > 0; il--, pe++) {
 392         struct indexEntry_s ie;
 393         rpmTagType type;
 394
 395         ie.info.tag = ntohl(pe->tag);
 396         ie.info.type = ntohl(pe->type);
 397         ie.info.count = ntohl(pe->count);
 398         ie.info.offset = ntohl(pe->offset);
 399
 400         if (hdrchkType(ie.info.type))
 401             return -1;
 402         if (hdrchkData(ie.info.count))
 403             return -1;
 404         if (hdrchkData(ie.info.offset))
 405             return -1;
 406         if (hdrchkAlign(ie.info.type, ie.info.offset))
 407             return -1;
 408
 409         ie.data = t = dataStart + ie.info.offset;
 410         if (dataEnd && t >= dataEnd)
 411             return -1;
 412
 413         ie.length = dataLength(ie.info.type, ie.data, ie.info.count, 1, dataEnd);
 414         if (ie.length < 0 || hdrchkData(ie.length))
 415             return -1;
 416
 417         ie.rdlen = 0;
 418
 419         if (entry) {
 420             ie.info.offset = regionid;
 421             *entry = ie;        /* structure assignment */
 422             entry++;
 423         }
 424
 425         /* Alignment */
 426         type = ie.info.type;
 427         if (typeSizes[type] > 1) {
 428             unsigned diff;
 429             diff = typeSizes[type] - (dl % typeSizes[type]);
 430             if (diff != typeSizes[type]) {
 431                 dl += diff;
 432                 if (ieprev.info.type == RPM_I18NSTRING_TYPE)
 433                     ieprev.length += diff;
 434             }
 435         }
 436         tdel = (tprev ? (t - tprev) : 0);
 437         if (ieprev.info.type == RPM_I18NSTRING_TYPE)
 438             tdel = ieprev.length;
 439
 440         if (ie.info.tag >= HEADER_I18NTABLE) {
 441             tprev = t;
 442         } else {
 443             tprev = dataStart;
 444             /* XXX HEADER_IMAGE tags don't include region sub-tag. */
 445             if (ie.info.tag == HEADER_IMAGE)
 446                 tprev -= REGION_TAG_COUNT;
 447         }
 448
 449         /* Perform endian conversions */
 450         switch (ntohl(pe->type)) {
 451         case RPM_INT32_TYPE:
 452         {   int32_t * it = (int32_t *)t;
 453             for (; ie.info.count > 0; ie.info.count--, it += 1) {
 454                 if (dataEnd && ((unsigned char *)it) >= dataEnd)
 455                     return -1;
 456                 *it = htonl(*it);
 457             }
 458             t = (unsigned char *) it;
 459         }   break;
 460         case RPM_INT16_TYPE:
 461         {   int16_t * it = (int16_t *) t;
 462             for (; ie.info.count > 0; ie.info.count--, it += 1) {
 463                 if (dataEnd && ((unsigned char *)it) >= dataEnd)
 464                     return -1;
 465                 *it = htons(*it);
 466             }
 467             t = (unsigned char *) it;
 468         }   break;
 469         default:
 470             t += ie.length;
 471             break;
 472         }
 473
 474         dl += ie.length;
 475         tl += tdel;
 476         ieprev = ie;    /* structure assignment */
 477
 478     }
 479     tdel = (tprev ? (t - tprev) : 0);
 480     tl += tdel;
 481
 482     /* XXX
 483      * There are two hacks here:
 484      *  1) tl is 16b (i.e. REGION_TAG_COUNT) short while doing headerReload().
 485      *  2) the 8/98 rpm bug with inserting i18n tags needs to use tl, not dl.
 486      */
 487     if (tl+REGION_TAG_COUNT == dl)
 488         tl += REGION_TAG_COUNT;
 489
 490     return dl;
 491 }
 492
 493 /** \ingroup header
 494  * doHeaderUnload.
 495  * @param h             header
 496  * @retval *lengthPtr   no. bytes in unloaded header blob
 497  * @return              unloaded header blob (NULL on error)
 498  */
 499 static void * doHeaderUnload(Header h,
 500                 size_t * lengthPtr)
 501 {
 502     int32_t * ei = NULL;
 503     entryInfo pe;
 504     char * dataStart;
 505     char * te;
 506     unsigned pad;
 507     unsigned len;
 508     int32_t il = 0;
 509     int32_t dl = 0;
 510     indexEntry entry;
 511     rpmTagType type;
 512     int i;
 513     int drlen, ndribbles;
 514     int driplen, ndrips;
 515     int legacy = 0;
 516
 517     /* Sort entries by (offset,tag). */
 518     headerUnsort(h);
 519
 520     /* Compute (il,dl) for all tags, including those deleted in region. */
 521     pad = 0;
 522     drlen = ndribbles = driplen = ndrips = 0;
 523     for (i = 0, entry = h->index; i < h->indexUsed; i++, entry++) {
 524         if (ENTRY_IS_REGION(entry)) {
 525             int32_t rdl = -entry->info.offset;  /* negative offset */
 526             int32_t ril = rdl/sizeof(*pe);
 527             int rid = entry->info.offset;
 528
 529             il += ril;
 530             dl += entry->rdlen + entry->info.count;
 531             /* XXX Legacy regions do not include the region tag and data. */
 532             if (i == 0 && (h->flags & HEADERFLAG_LEGACY))
 533                 il += 1;
 534
 535             /* Skip rest of entries in region, but account for dribbles. */
 536             for (; i < h->indexUsed && entry->info.offset <= rid+1; i++, entry++) {
 537                 if (entry->info.offset <= rid)
 538                     continue;
 539
 540                 /* Alignment */
 541                 type = entry->info.type;
 542                 if (typeSizes[type] > 1) {
 543                     unsigned diff;
 544                     diff = typeSizes[type] - (dl % typeSizes[type]);
 545                     if (diff != typeSizes[type]) {
 546                         drlen += diff;
 547                         pad += diff;
 548                         dl += diff;
 549                     }
 550                 }
 551
 552                 ndribbles++;
 553                 il++;
 554                 drlen += entry->length;
 555                 dl += entry->length;
 556             }
 557             i--;
 558             entry--;
 559             continue;
 560         }
 561
 562         /* Ignore deleted drips. */
 563         if (entry->data == NULL || entry->length <= 0)
 564             continue;
 565
 566         /* Alignment */
 567         type = entry->info.type;
 568         if (typeSizes[type] > 1) {
 569             unsigned diff;
 570             diff = typeSizes[type] - (dl % typeSizes[type]);
 571             if (diff != typeSizes[type]) {
 572                 driplen += diff;
 573                 pad += diff;
 574                 dl += diff;
 575             } else
 576                 diff = 0;
 577         }
 578
 579         ndrips++;
 580         il++;
 581         driplen += entry->length;
 582         dl += entry->length;
 583     }
 584
 585     /* Sanity checks on header intro. */
 586     if (hdrchkTags(il) || hdrchkData(dl))
 587         goto errxit;
 588
 589     len = sizeof(il) + sizeof(dl) + (il * sizeof(*pe)) + dl;
 590
 591     ei = xmalloc(len);
 592     ei[0] = htonl(il);
 593     ei[1] = htonl(dl);
 594
 595     pe = (entryInfo) &ei[2];
 596     dataStart = te = (char *) (pe + il);
 597
 598     pad = 0;
 599     for (i = 0, entry = h->index; i < h->indexUsed; i++, entry++) {
 600         const char * src;
 601         unsigned char *t;
 602         int count;
 603         int rdlen;
 604
 605         if (entry->data == NULL || entry->length <= 0)
 606             continue;
 607
 608         t = (unsigned char*)te;
 609         pe->tag = htonl(entry->info.tag);
 610         pe->type = htonl(entry->info.type);
 611         pe->count = htonl(entry->info.count);
 612
 613         if (ENTRY_IS_REGION(entry)) {
 614             int32_t rdl = -entry->info.offset;  /* negative offset */
 615             int32_t ril = rdl/sizeof(*pe) + ndribbles;
 616             int rid = entry->info.offset;
 617
 618             src = (char *)entry->data;
 619             rdlen = entry->rdlen;
 620
 621             /* XXX Legacy regions do not include the region tag and data. */
 622             if (i == 0 && (h->flags & HEADERFLAG_LEGACY)) {
 623                 int32_t stei[4];
 624
 625                 legacy = 1;
 626                 memcpy(pe+1, src, rdl);
 627                 memcpy(te, src + rdl, rdlen);
 628                 te += rdlen;
 629
 630                 pe->offset = htonl(te - dataStart);
 631                 stei[0] = pe->tag;
 632                 stei[1] = pe->type;
 633                 stei[2] = htonl(-rdl-entry->info.count);
 634                 stei[3] = pe->count;
 635                 memcpy(te, stei, entry->info.count);
 636                 te += entry->info.count;
 637                 ril++;
 638                 rdlen += entry->info.count;
 639
 640                 count = regionSwab(NULL, ril, 0, pe, t, NULL, 0);
 641                 if (count != rdlen)
 642                     goto errxit;
 643
 644             } else {
 645
 646                 memcpy(pe+1, src + sizeof(*pe), ((ril-1) * sizeof(*pe)));
 647                 memcpy(te, src + (ril * sizeof(*pe)), rdlen+entry->info.count+drlen);
 648                 te += rdlen;
 649                 {
 650                     entryInfo se = (entryInfo)src;
 651                     int off = ntohl(se->offset);
 652                     pe->offset = (off) ? htonl(te - dataStart) : htonl(off);
 653                 }
 654                 te += entry->info.count + drlen;
 655
 656                 count = regionSwab(NULL, ril, 0, pe, t, NULL, 0);
 657                 if (count != (rdlen + entry->info.count + drlen))
 658                     goto errxit;
 659             }
 660
 661             /* Skip rest of entries in region. */
 662             while (i < h->indexUsed && entry->info.offset <= rid+1) {
 663                 i++;
 664                 entry++;
 665             }
 666             i--;
 667             entry--;
 668             pe += ril;
 669             continue;
 670         }
 671
 672         /* Ignore deleted drips. */
 673         if (entry->data == NULL || entry->length <= 0)
 674             continue;
 675
 676         /* Alignment */
 677         type = entry->info.type;
 678         if (typeSizes[type] > 1) {
 679             unsigned diff;
 680             diff = typeSizes[type] - ((te - dataStart) % typeSizes[type]);
 681             if (diff != typeSizes[type]) {
 682                 memset(te, 0, diff);
 683                 te += diff;
 684                 pad += diff;
 685             }
 686         }
 687
 688         pe->offset = htonl(te - dataStart);
 689
 690         /* copy data w/ endian conversions */
 691         switch (entry->info.type) {
 692         case RPM_INT32_TYPE:
 693             count = entry->info.count;
 694             src = entry->data;
 695             while (count--) {
 696                 *((int32_t *)te) = htonl(*((int32_t *)src));
 697                 te += sizeof(int32_t);
 698                 src += sizeof(int32_t);
 699             }
 700             break;
 701
 702         case RPM_INT16_TYPE:
 703             count = entry->info.count;
 704             src = entry->data;
 705             while (count--) {
 706                 *((int16_t *)te) = htons(*((int16_t *)src));
 707                 te += sizeof(int16_t);
 708                 src += sizeof(int16_t);
 709             }
 710             break;
 711
 712         default:
 713             memcpy(te, entry->data, entry->length);
 714             te += entry->length;
 715             break;
 716         }
 717         pe++;
 718     }
 719
 720     /* Insure that there are no memcpy underruns/overruns. */
 721     if (((char *)pe) != dataStart)
 722         goto errxit;
 723     if ((((char *)ei)+len) != te)
 724         goto errxit;
 725
 726     if (lengthPtr)
 727         *lengthPtr = len;
 728
 729     h->flags &= ~HEADERFLAG_SORTED;
 730     headerSort(h);
 731
 732     return (void *) ei;
 733
 734 errxit:
 735     ei = _free(ei);
 736     return (void *) ei;
 737 }
 738
 739 void * headerUnload(Header h)
 740 {
 741     size_t length;
 742     void * uh = doHeaderUnload(h, &length);
 743     return uh;
 744 }
 745
 746 /**
 747  * Find matching (tag,type) entry in header.
 748  * @param h             header
 749  * @param tag           entry tag
 750  * @param type          entry type
 751  * @return              header entry
 752  */
 753 static
 754 indexEntry findEntry(Header h, rpmTag tag, rpmTagType type)
 755 {
 756     indexEntry entry, entry2, last;
 757     struct indexEntry_s key;
 758
 759     if (h == NULL) return NULL;
 760     if (!(h->flags & HEADERFLAG_SORTED)) headerSort(h);
 761
 762     key.info.tag = tag;
 763
 764     entry2 = entry =
 765         bsearch(&key, h->index, h->indexUsed, sizeof(*h->index), indexCmp);
 766     if (entry == NULL)
 767         return NULL;
 768
 769     if (type == RPM_NULL_TYPE)
 770         return entry;
 771
 772     /* look backwards */
 773     while (entry->info.tag == tag && entry->info.type != type &&
 774            entry > h->index) entry--;
 775
 776     if (entry->info.tag == tag && entry->info.type == type)
 777         return entry;
 778
 779     last = h->index + h->indexUsed;
 780     /* FIX: entry2 = entry. Code looks bogus as well. */
 781     while (entry2->info.tag == tag && entry2->info.type != type &&
 782            entry2 < last) entry2++;
 783
 784     if (entry->info.tag == tag && entry->info.type == type)
 785         return entry;
 786
 787     return NULL;
 788 }
 789
 790 int headerRemoveEntry(Header h, rpmTag tag)
 791 {
 792     indexEntry last = h->index + h->indexUsed;
 793     indexEntry entry, first;
 794     int ne;
 795
 796     entry = findEntry(h, tag, RPM_NULL_TYPE);
 797     if (!entry) return 1;
 798
 799     /* Make sure entry points to the first occurence of this tag. */
 800     while (entry > h->index && (entry - 1)->info.tag == tag)
 801         entry--;
 802
 803     /* Free data for tags being removed. */
 804     for (first = entry; first < last; first++) {
 805         rpm_data_t data;
 806         if (first->info.tag != tag)
 807             break;
 808         data = first->data;
 809         first->data = NULL;
 810         first->length = 0;
 811         if (ENTRY_IN_REGION(first))
 812             continue;
 813         data = _free(data);
 814     }
 815
 816     ne = (first - entry);
 817     if (ne > 0) {
 818         h->indexUsed -= ne;
 819         ne = last - first;
 820         if (ne > 0)
 821             memmove(entry, first, (ne * sizeof(*entry)));
 822     }
 823
 824     return 0;
 825 }
 826
 827 Header headerLoad(void * uh)
 828 {
 829     int32_t * ei = (int32_t *) uh;
 830     int32_t il = ntohl(ei[0]);          /* index length */
 831     int32_t dl = ntohl(ei[1]);          /* data length */
 832     size_t pvlen = sizeof(il) + sizeof(dl) +
 833                (il * sizeof(struct entryInfo_s)) + dl;
 834     void * pv = uh;
 835     Header h = NULL;
 836     entryInfo pe;
 837     unsigned char * dataStart;
 838     unsigned char * dataEnd;
 839     indexEntry entry;
 840     int rdlen;
 841     int i;
 842
 843     /* Sanity checks on header intro. */
 844     if (hdrchkTags(il) || hdrchkData(dl))
 845         goto errxit;
 846
 847     ei = (int32_t *) pv;
 848     pe = (entryInfo) &ei[2];
 849     dataStart = (unsigned char *) (pe + il);
 850     dataEnd = dataStart + dl;
 851
 852     h = xcalloc(1, sizeof(*h));
 853     h->blob = uh;
 854     h->indexAlloced = il + 1;
 855     h->indexUsed = il;
 856     h->index = xcalloc(h->indexAlloced, sizeof(*h->index));
 857     h->flags |= HEADERFLAG_SORTED;
 858     h->nrefs = 0;
 859     h = headerLink(h);
 860
 861     entry = h->index;
 862     i = 0;
 863     if (!(htonl(pe->tag) < HEADER_I18NTABLE)) {
 864         h->flags |= HEADERFLAG_LEGACY;
 865         entry->info.type = REGION_TAG_TYPE;
 866         entry->info.tag = HEADER_IMAGE;
 867         entry->info.count = REGION_TAG_COUNT;
 868         entry->info.offset = ((unsigned char *)pe - dataStart); /* negative offset */
 869
 870         entry->data = pe;
 871         entry->length = pvlen - sizeof(il) - sizeof(dl);
 872         rdlen = regionSwab(entry+1, il, 0, pe, dataStart, dataEnd, entry->info.offset);
 873 #if 0   /* XXX don't check, the 8/98 i18n bug fails here. */
 874         if (rdlen != dl)
 875             goto errxit;
 876 #endif
 877         entry->rdlen = rdlen;
 878         entry++;
 879         h->indexUsed++;
 880     } else {
 881         int32_t rdl;
 882         int32_t ril;
 883
 884         h->flags &= ~HEADERFLAG_LEGACY;
 885
 886         entry->info.type = htonl(pe->type);
 887         entry->info.count = htonl(pe->count);
 888
 889         if (hdrchkType(entry->info.type))
 890             goto errxit;
 891         if (hdrchkTags(entry->info.count))
 892             goto errxit;
 893
 894         {   int off = ntohl(pe->offset);
 895
 896             if (hdrchkData(off))
 897                 goto errxit;
 898             if (off) {
 899                 size_t nb = REGION_TAG_COUNT;
 900                 int32_t stei[nb];
 901                 /* XXX Hmm, why the copy? */
 902                 memcpy(&stei, dataStart + off, nb);
 903                 rdl = -ntohl(stei[2]);  /* negative offset */
 904                 ril = rdl/sizeof(*pe);
 905                 if (hdrchkTags(ril) || hdrchkData(rdl))
 906                     goto errxit;
 907                 entry->info.tag = htonl(pe->tag);
 908             } else {
 909                 ril = il;
 910                 rdl = (ril * sizeof(struct entryInfo_s));
 911                 entry->info.tag = HEADER_IMAGE;
 912             }
 913         }
 914         entry->info.offset = -rdl;      /* negative offset */
 915
 916         entry->data = pe;
 917         entry->length = pvlen - sizeof(il) - sizeof(dl);
 918         rdlen = regionSwab(entry+1, ril-1, 0, pe+1, dataStart, dataEnd, entry->info.offset);
 919         if (rdlen < 0)
 920             goto errxit;
 921         entry->rdlen = rdlen;
 922
 923         if (ril < h->indexUsed) {
 924             indexEntry newEntry = entry + ril;
 925             int ne = (h->indexUsed - ril);
 926             int rid = entry->info.offset+1;
 927             int rc;
 928
 929             /* Load dribble entries from region. */
 930             rc = regionSwab(newEntry, ne, 0, pe+ril, dataStart, dataEnd, rid);
 931             if (rc < 0)
 932                 goto errxit;
 933             rdlen += rc;
 934
 935           { indexEntry firstEntry = newEntry;
 936             int save = h->indexUsed;
 937             int j;
 938
 939             /* Dribble entries replace duplicate region entries. */
 940             h->indexUsed -= ne;
 941             for (j = 0; j < ne; j++, newEntry++) {
 942                 (void) headerRemoveEntry(h, newEntry->info.tag);
 943                 if (newEntry->info.tag == HEADER_BASENAMES)
 944                     (void) headerRemoveEntry(h, HEADER_OLDFILENAMES);
 945             }
 946
 947             /* If any duplicate entries were replaced, move new entries down. */
 948             if (h->indexUsed < (save - ne)) {
 949                 memmove(h->index + h->indexUsed, firstEntry,
 950                         (ne * sizeof(*entry)));
 951             }
 952             h->indexUsed += ne;
 953           }
 954         }
 955     }
 956
 957     h->flags &= ~HEADERFLAG_SORTED;
 958     headerSort(h);
 959
 960     return h;
 961
 962 errxit:
 963     if (h) {
 964         h->index = _free(h->index);
 965         h = _free(h);
 966     }
 967     return h;
 968 }
 969
 970 Header headerReload(Header h, rpmTag tag)
 971 {
 972     Header nh;
 973     size_t length;
 974     void * uh = doHeaderUnload(h, &length);
 975
 976     h = headerFree(h);
 977     if (uh == NULL)
 978         return NULL;
 979     nh = headerLoad(uh);
 980     if (nh == NULL) {
 981         uh = _free(uh);
 982         return NULL;
 983     }
 984     if (nh->flags & HEADERFLAG_ALLOCATED)
 985         uh = _free(uh);
 986     nh->flags |= HEADERFLAG_ALLOCATED;
 987     if (ENTRY_IS_REGION(nh->index)) {
 988         if (tag == HEADER_SIGNATURES || tag == HEADER_IMMUTABLE)
 989             nh->index[0].info.tag = tag;
 990     }
 991     return nh;
 992 }
 993
 994 Header headerCopyLoad(const void * uh)
 995 {
 996     int32_t * ei = (int32_t *) uh;
 997     int32_t il = ntohl(ei[0]);          /* index length */
 998     int32_t dl = ntohl(ei[1]);          /* data length */
 999     size_t pvlen = sizeof(il) + sizeof(dl) +
1000                         (il * sizeof(struct entryInfo_s)) + dl;
1001     void * nuh = NULL;
1002     Header h = NULL;
1003
1004     /* Sanity checks on header intro. */
1005     if (!(hdrchkTags(il) || hdrchkData(dl)) && pvlen < headerMaxbytes) {
1006         nuh = memcpy(xmalloc(pvlen), uh, pvlen);
1007         if ((h = headerLoad(nuh)) != NULL)
1008             h->flags |= HEADERFLAG_ALLOCATED;
1009     }
1010     if (h == NULL)
1011         nuh = _free(nuh);
1012     return h;
1013 }
1014
1015 /** \ingroup header
1016  * Read (and load) header from file handle.
1017  * @param fd            file handle
1018  * @param magicp        read (and verify) 8 bytes of (magic, 0)?
1019  * @return              header (or NULL on error)
1020  */
1021 Header headerRead(FD_t fd, enum hMagic magicp)
1022 {
1023     int32_t block[4];
1024     int32_t reserved;
1025     int32_t * ei = NULL;
1026     int32_t il;
1027     int32_t dl;
1028     int32_t magic;
1029     Header h = NULL;
1030     size_t len;
1031     int i;
1032
1033     memset(block, 0, sizeof(block));
1034     i = 2;
1035     if (magicp == HEADER_MAGIC_YES)
1036         i += 2;
1037
1038     /* FIX: cast? */
1039     if (timedRead(fd, (char *)block, i*sizeof(*block)) != (i * sizeof(*block)))
1040         goto exit;
1041
1042     i = 0;
1043
1044     if (magicp == HEADER_MAGIC_YES) {
1045         magic = block[i++];
1046         if (memcmp(&magic, header_magic, sizeof(magic)))
1047             goto exit;
1048         reserved = block[i++];
1049     }
1050
1051     il = ntohl(block[i]);       i++;
1052     dl = ntohl(block[i]);       i++;
1053
1054     len = sizeof(il) + sizeof(dl) + (il * sizeof(struct entryInfo_s)) + dl;
1055
1056     /* Sanity checks on header intro. */
1057     if (hdrchkTags(il) || hdrchkData(dl) || len > headerMaxbytes)
1058         goto exit;
1059
1060     ei = xmalloc(len);
1061     ei[0] = htonl(il);
1062     ei[1] = htonl(dl);
1063     len -= sizeof(il) + sizeof(dl);
1064
1065     /* FIX: cast? */
1066     if (timedRead(fd, (char *)&ei[2], len) != len)
1067         goto exit;
1068
1069     h = headerLoad(ei);
1070
1071 exit:
1072     if (h) {
1073         if (h->flags & HEADERFLAG_ALLOCATED)
1074             ei = _free(ei);
1075         h->flags |= HEADERFLAG_ALLOCATED;
1076     } else if (ei)
1077         ei = _free(ei);
1078         /* FIX: timedRead macro obscures annotation */
1079     return h;
1080 }
1081
1082 int headerWrite(FD_t fd, Header h, enum hMagic magicp)
1083 {
1084     ssize_t nb;
1085     size_t length;
1086     void * uh;
1087
1088     if (h == NULL)
1089         return 1;
1090     uh = doHeaderUnload(h, &length);
1091     if (uh == NULL)
1092         return 1;
1093     switch (magicp) {
1094     case HEADER_MAGIC_YES:
1095         nb = Fwrite(header_magic, sizeof(char), sizeof(header_magic), fd);
1096         if (nb != sizeof(header_magic))
1097             goto exit;
1098         break;
1099     case HEADER_MAGIC_NO:
1100         break;
1101     }
1102
1103     nb = Fwrite(uh, sizeof(char), length, fd);
1104
1105 exit:
1106     uh = _free(uh);
1107     return (nb == length ? 0 : 1);
1108 }
1109
1110 int headerIsEntry(Header h, rpmTag tag)
1111 {
1112                 /* FIX: h modified by sort. */
1113     return (findEntry(h, tag, RPM_NULL_TYPE) ? 1 : 0);
1114
1115 }
1116
1117 /** \ingroup header
1118  * Retrieve data from header entry.
1119  * @todo Permit retrieval of regions other than HEADER_IMUTABLE.
1120  * @param entry         header entry
1121  * @param td            tag data container
1122  * @param minMem        string pointers refer to header memory?
1123  * @return              1 on success, otherwise error.
1124  */
1125 static int copyTdEntry(const indexEntry entry, rpmtd td, int minMem)
1126 {
1127     rpm_count_t count = entry->info.count;
1128     int rc = 1;         /* XXX 1 on success. */
1129
1130     assert(td != NULL);
1131     switch (entry->info.type) {
1132     case RPM_BIN_TYPE:
1133         /*
1134          * XXX This only works for
1135          * XXX  "sealed" HEADER_IMMUTABLE/HEADER_SIGNATURES/HEADER_IMAGE.
1136          * XXX This will *not* work for unsealed legacy HEADER_IMAGE (i.e.
1137          * XXX a legacy header freshly read, but not yet unloaded to the rpmdb).
1138          */
1139         if (ENTRY_IS_REGION(entry)) {
1140             int32_t * ei = ((int32_t *)entry->data) - 2;
1141             entryInfo pe = (entryInfo) (ei + 2);
1142             unsigned char * dataStart = (unsigned char *) (pe + ntohl(ei[0]));
1143             int32_t rdl = -entry->info.offset;  /* negative offset */
1144             int32_t ril = rdl/sizeof(*pe);
1145
1146             rdl = entry->rdlen;
1147             count = 2 * sizeof(*ei) + (ril * sizeof(*pe)) + rdl;
1148             if (entry->info.tag == HEADER_IMAGE) {
1149                 ril -= 1;
1150                 pe += 1;
1151             } else {
1152                 count += REGION_TAG_COUNT;
1153                 rdl += REGION_TAG_COUNT;
1154             }
1155
1156             td->data = xmalloc(count);
1157             ei = (int32_t *) td->data;
1158             ei[0] = htonl(ril);
1159             ei[1] = htonl(rdl);
1160
1161             pe = (entryInfo) memcpy(ei + 2, pe, (ril * sizeof(*pe)));
1162
1163             dataStart = (unsigned char *) memcpy(pe + ril, dataStart, rdl);
1164
1165             rc = regionSwab(NULL, ril, 0, pe, dataStart, dataStart + rdl, 0);
1166             /* XXX 1 on success. */
1167             rc = (rc < 0) ? 0 : 1;
1168         } else {
1169             count = entry->length;
1170             td->data = (!minMem
1171                 ? memcpy(xmalloc(count), entry->data, count)
1172                 : entry->data);
1173         }
1174         break;
1175     case RPM_STRING_TYPE:
1176         if (count == 1) {
1177             td->data = entry->data;
1178             break;
1179         }
1180     case RPM_STRING_ARRAY_TYPE:
1181     case RPM_I18NSTRING_TYPE:
1182     {   const char ** ptrEntry;
1183         int tableSize = count * sizeof(char *);
1184         char * t;
1185         int i;
1186
1187         if (minMem) {
1188             td->data = xmalloc(tableSize);
1189             ptrEntry = (const char **) td->data;
1190             t = entry->data;
1191         } else {
1192             t = xmalloc(tableSize + entry->length);
1193             td->data = (void *)t;
1194             ptrEntry = (const char **) td->data;
1195             t += tableSize;
1196             memcpy(t, entry->data, entry->length);
1197         }
1198         for (i = 0; i < count; i++) {
1199             *ptrEntry++ = t;
1200             t = strchr(t, 0);
1201             t++;
1202         }
1203     }   break;
1204
1205     default:
1206         td->data = entry->data;
1207         break;
1208     }
1209     td->type = entry->info.type;
1210     td->count = count;
1211     td->freeData = (entry->data == td->data) ? 0 : 1;
1212
1213     return rc;
1214 }
1215
1216 /** \ingroup header
1217  * Retrieve data from header entry.
1218  * Just a compat wrapper around copyTdEntry() until everything's converted
1219  * to new interface.
1220  * @param entry         header entry
1221  * @retval type         address of type (or NULL)
1222  * @retval p            address of data (or NULL)
1223  * @retval c            address of count (or NULL)
1224  * @param minMem        string pointers refer to header memory?
1225  * @return              1 on success, otherwise error.
1226  */
1227 static int copyEntry(const indexEntry entry,
1228                 rpmTagType * type,
1229                 rpm_data_t * p,
1230                 rpm_count_t * c,
1231                 int minMem)
1232 {
1233     struct rpmtd_s td;
1234     int rc;
1235
1236     rpmtdReset(&td);
1237     rc = copyTdEntry(entry, &td, minMem);
1238     TDWRAP();
1239     return rc;
1240 }
1241
1242 /**
1243  * Does locale match entry in header i18n table?
1244  *
1245  * \verbatim
1246  * The range [l,le) contains the next locale to match:
1247  *    ll[_CC][.EEEEE][@dddd]
1248  * where
1249  *    ll        ISO language code (in lowercase).
1250  *    CC        (optional) ISO coutnry code (in uppercase).
1251  *    EEEEE     (optional) encoding (not really standardized).
1252  *    dddd      (optional) dialect.
1253  * \endverbatim
1254  *
1255  * @param td            header i18n table data, NUL terminated
1256  * @param l             start of locale to match
1257  * @param le            end of locale to match
1258  * @return              1 on good match, 2 on weak match, 0 on no match
1259  */
1260 static int headerMatchLocale(const char *td, const char *l, const char *le)
1261 {
1262     const char *fe;
1263
1264     /* First try a complete match. */
1265     if (strlen(td) == (le-l) && !strncmp(td, l, (le - l)))
1266         return 1;
1267
1268     /* Next, try stripping optional dialect and matching.  */
1269     for (fe = l; fe < le && *fe != '@'; fe++)
1270         {};
1271     if (fe < le && !strncmp(td, l, (fe - l)))
1272         return 1;
1273
1274     /* Next, try stripping optional codeset and matching.  */
1275     for (fe = l; fe < le && *fe != '.'; fe++)
1276         {};
1277     if (fe < le && !strncmp(td, l, (fe - l)))
1278         return 1;
1279
1280     /* Finally, try stripping optional country code and matching. */
1281     for (fe = l; fe < le && *fe != '_'; fe++)
1282         {};
1283     if (fe < le && !strncmp(td, l, (fe - l)))
1284         return 2;
1285
1286     return 0;
1287 }
1288
1289 /**
1290  * Return i18n string from header that matches locale.
1291  * @param h             header
1292  * @param entry         i18n string data
1293  * @retval td           tag data container
1294  * @return              1 always
1295  */
1296 static int copyI18NEntry(Header h, indexEntry entry, rpmtd td)
1297 {
1298     const char *lang, *l, *le;
1299     indexEntry table;
1300
1301     td->type = RPM_STRING_TYPE;
1302     td->count = 1;
1303     /* if no match, just return the first string */
1304     td->data = entry->data;
1305
1306     /* XXX Drepper sez' this is the order. */
1307     if ((lang = getenv("LANGUAGE")) == NULL &&
1308         (lang = getenv("LC_ALL")) == NULL &&
1309         (lang = getenv("LC_MESSAGES")) == NULL &&
1310         (lang = getenv("LANG")) == NULL)
1311             goto exit;
1312
1313     if ((table = findEntry(h, HEADER_I18NTABLE, RPM_STRING_ARRAY_TYPE)) == NULL)
1314         goto exit;
1315
1316     for (l = lang; *l != '\0'; l = le) {
1317         const char *t;
1318         char *ed, *ed_weak = NULL;
1319         int langNum;
1320
1321         while (*l && *l == ':')                 /* skip leading colons */
1322             l++;
1323         if (*l == '\0')
1324             break;
1325         for (le = l; *le && *le != ':'; le++)   /* find end of this locale */
1326             {};
1327
1328         /* For each entry in the header ... */
1329         for (langNum = 0, t = table->data, ed = entry->data;
1330              langNum < entry->info.count;
1331              langNum++, t += strlen(t) + 1, ed += strlen(ed) + 1) {
1332
1333             int match = headerMatchLocale(t, l, le);
1334             if (match == 1) {
1335                 td->data = ed;
1336                 goto exit;
1337             } else if (match == 2) {
1338                 ed_weak = ed;
1339             }
1340         }
1341         if (ed_weak) {
1342             td->data = ed_weak;
1343             goto exit;
1344         }
1345     }
1346
1347 exit:
1348     return 1;
1349 }
1350
1351 /**
1352  * Retrieve tag data from header.
1353  * @param h             header
1354  * @param tag           tag to retrieve
1355  * @retval td           tag data container
1356  * @param flags         flags to control retrieval
1357  * @return              1 on success, 0 on not found
1358  */
1359 static int intGetTdEntry(Header h, rpmTag tag, rpmtd td, headerGetFlags flags)
1360 {
1361     indexEntry entry;
1362     int rc;
1363
1364     assert(td != NULL);
1365     /* ensure clean state */
1366     rpmtdReset(td);
1367     td->tag = tag;
1368     /* First find the tag */
1369     /* FIX: h modified by sort. */
1370     entry = findEntry(h, tag, RPM_NULL_TYPE);
1371     if (entry == NULL) {
1372         /* Td is zeroed above, just return... */
1373         return 0;
1374     }
1375
1376     if (flags & HEADERGET_RAW) {
1377         rc = copyTdEntry(entry, td, (flags & HEADERGET_MINMEM));
1378     } else {
1379         switch (entry->info.type) {
1380         case RPM_I18NSTRING_TYPE:
1381             rc = copyI18NEntry(h, entry, td);
1382             break;
1383         default:
1384             rc = copyTdEntry(entry, td, (flags & HEADERGET_MINMEM));
1385             break;
1386         }
1387     }
1388
1389     /* XXX 1 on success */
1390     return ((rc == 1) ? 1 : 0);
1391 }
1392
1393 void * headerFreeTag(Header h, rpm_data_t data, rpmTagType type)
1394 {
1395     if (data) {
1396         if (type == RPM_FORCEFREE_TYPE ||
1397             type == RPM_STRING_ARRAY_TYPE ||
1398             type == RPM_I18NSTRING_TYPE ||
1399             type == RPM_BIN_TYPE)
1400                 data = _free(data);
1401     }
1402     return NULL;
1403 }
1404
1405 /*
1406  * XXX temporary kludgery until tag extensions have been converted to
1407  * take rpmtd as argument
1408  */
1409 static int intGetTagExt(Header h, rpmTag tag, rpmtd td, headerTagTagFunction tagfunc)
1410 {
1411     int rc;
1412     rpmtdReset(td);
1413     rc = tagfunc(h, td);
1414     td->tag = tag;
1415     /* XXX extension returns are reversed, ugh */
1416     return rc ? 0 : 1;
1417 }
1418
1419 int headerGet(Header h, rpmTag tag, rpmtd td, headerGetFlags flags)
1420 {
1421     int rc;
1422     headerTagTagFunction tagfunc = NULL;
1423
1424     assert(td != NULL);
1425
1426     if (flags & HEADERGET_EXT) {
1427         tagfunc = rpmHeaderTagFunc(tag);
1428     }
1429
1430     if (tagfunc) {
1431         rc = intGetTagExt(h, tag, td, tagfunc);
1432     } else {
1433         rc = intGetTdEntry(h, tag, td, flags);
1434     }
1435
1436     assert(tag == td->tag);
1437     return rc;
1438 }
1439
1440 static int headerGetWrap(Header h, rpmTag tag,
1441                 rpmTagType * type,
1442                 rpm_data_t * p,
1443                 rpm_count_t * c,
1444                 headerGetFlags flags)
1445 {
1446     struct rpmtd_s td;
1447     int rc;
1448
1449     rc = headerGet(h, tag, &td, flags);
1450     TDWRAP();
1451     return rc;
1452 }
1453
1454 int headerGetEntry(Header h, rpmTag tag,
1455                         rpmTagType * type,
1456                         rpm_data_t * p,
1457                         rpm_count_t * c)
1458 {
1459     return headerGetWrap(h, tag, type, p, c, HEADERGET_DEFAULT);
1460 }
1461
1462 int headerGetEntryMinMemory(Header h, rpmTag tag,
1463                         rpmTagType * type,
1464                         rpm_data_t * p,
1465                         rpm_count_t * c)
1466 {
1467     return headerGetWrap(h, tag, type, (rpm_data_t) p, c, HEADERGET_MINMEM);
1468 }
1469
1470 int headerGetRawEntry(Header h, rpmTag tag, rpmTagType * type, rpm_data_t * p,
1471                 rpm_count_t * c)
1472 {
1473     indexEntry entry;
1474     int rc;
1475
1476     if (p == NULL) return headerIsEntry(h, tag);
1477
1478     /* First find the tag */
1479                 /* FIX: h modified by sort. */
1480     entry = findEntry(h, tag, RPM_NULL_TYPE);
1481     if (!entry) {
1482         if (p) *p = NULL;
1483         if (c) *c = 0;
1484         return 0;
1485     }
1486
1487     rc = copyEntry(entry, type, p, c, 0);
1488
1489     /* XXX 1 on success */
1490     return ((rc == 1) ? 1 : 0);
1491 }
1492
1493 /**
1494  */
1495 static void copyData(rpmTagType type, rpm_data_t dstPtr,
1496                 rpm_constdata_t srcPtr, rpm_count_t cnt, int dataLength)
1497 {
1498     switch (type) {
1499     case RPM_STRING_ARRAY_TYPE:
1500     case RPM_I18NSTRING_TYPE:
1501     {   const char ** av = (const char **) srcPtr;
1502         char * t = dstPtr;
1503
1504         while (cnt-- > 0 && dataLength > 0) {
1505             const char * s;
1506             if ((s = *av++) == NULL)
1507                 continue;
1508             do {
1509                 *t++ = *s++;
1510             } while (s[-1] && --dataLength > 0);
1511         }
1512     }   break;
1513
1514     default:
1515         memmove(dstPtr, srcPtr, dataLength);
1516         break;
1517     }
1518 }
1519
1520 /**
1521  * Return (malloc'ed) copy of entry data.
1522  * @param type          entry data type
1523  * @param p             entry data
1524  * @param c             entry item count
1525  * @retval lengthPtr    no. bytes in returned data
1526  * @return              (malloc'ed) copy of entry data, NULL on error
1527  */
1528 static void *
1529 grabData(rpmTagType type, rpm_constdata_t p, rpm_count_t c, int * lengthPtr)
1530 {
1531     rpm_data_t data = NULL;
1532     int length;
1533
1534     length = dataLength(type, p, c, 0, NULL);
1535     if (length > 0) {
1536         data = xmalloc(length);
1537         copyData(type, data, p, c, length);
1538     }
1539
1540     if (lengthPtr)
1541         *lengthPtr = length;
1542     return data;
1543 }
1544
1545 int headerAddEntry(Header h, rpmTag tag, rpmTagType type,
1546                         rpm_constdata_t p, rpm_count_t c)
1547 {
1548     indexEntry entry;
1549     rpm_data_t data;
1550     int length;
1551
1552     /* Count must always be >= 1 for headerAddEntry. */
1553     if (c <= 0)
1554         return 0;
1555
1556     if (hdrchkType(type))
1557         return 0;
1558     if (hdrchkData(c))
1559         return 0;
1560
1561     length = 0;
1562     data = grabData(type, p, c, &length);
1563     if (data == NULL || length <= 0)
1564         return 0;
1565
1566     /* Allocate more index space if necessary */
1567     if (h->indexUsed == h->indexAlloced) {
1568         h->indexAlloced += INDEX_MALLOC_SIZE;
1569         h->index = xrealloc(h->index, h->indexAlloced * sizeof(*h->index));
1570     }
1571
1572     /* Fill in the index */
1573     entry = h->index + h->indexUsed;
1574     entry->info.tag = tag;
1575     entry->info.type = type;
1576     entry->info.count = c;
1577     entry->info.offset = 0;
1578     entry->data = data;
1579     entry->length = length;
1580
1581     if (h->indexUsed > 0 && tag < h->index[h->indexUsed-1].info.tag)
1582         h->flags &= ~HEADERFLAG_SORTED;
1583     h->indexUsed++;
1584
1585     return 1;
1586 }
1587
1588 int headerAppendEntry(Header h, rpmTag tag, rpmTagType type,
1589                 rpm_constdata_t p, rpm_count_t c)
1590 {
1591     indexEntry entry;
1592     int length;
1593
1594     if (type == RPM_STRING_TYPE || type == RPM_I18NSTRING_TYPE) {
1595         /* we can't do this */
1596         return 0;
1597     }
1598
1599     /* Find the tag entry in the header. */
1600     entry = findEntry(h, tag, type);
1601     if (!entry)
1602         return 0;
1603
1604     length = dataLength(type, p, c, 0, NULL);
1605     if (length < 0)
1606         return 0;
1607
1608     if (ENTRY_IN_REGION(entry)) {
1609         char * t = xmalloc(entry->length + length);
1610         memcpy(t, entry->data, entry->length);
1611         entry->data = t;
1612         entry->info.offset = 0;
1613     } else
1614         entry->data = xrealloc(entry->data, entry->length + length);
1615
1616     copyData(type, ((char *) entry->data) + entry->length, p, c, length);
1617
1618     entry->length += length;
1619
1620     entry->info.count += c;
1621
1622     return 1;
1623 }
1624
1625 int headerAddOrAppendEntry(Header h, rpmTag tag, rpmTagType type,
1626                 rpm_constdata_t p, rpm_count_t c)
1627 {
1628     return (findEntry(h, tag, type)
1629         ? headerAppendEntry(h, tag, type, p, c)
1630         : headerAddEntry(h, tag, type, p, c));
1631 }
1632
1633 int headerPut(Header h, rpmtd td, headerPutFlags flags)
1634 {
1635     int rc;
1636
1637     assert(td != NULL);
1638     if (flags & HEADERPUT_APPEND) {
1639         rc = headerAddOrAppendEntry(h, td->tag, td->type, td->data, td->count);
1640     } else {
1641         rc = headerAddEntry(h, td->tag, td->type, td->data, td->count);
1642     }
1643     return rc;
1644 }
1645
1646 int headerAddI18NString(Header h, rpmTag tag, const char * string,
1647                 const char * lang)
1648 {
1649     indexEntry table, entry;
1650     const char ** strArray;
1651     int length;
1652     int ghosts;
1653     rpm_count_t i, langNum;
1654     char * buf;
1655
1656     table = findEntry(h, HEADER_I18NTABLE, RPM_STRING_ARRAY_TYPE);
1657     entry = findEntry(h, tag, RPM_I18NSTRING_TYPE);
1658
1659     if (!table && entry)
1660         return 0;               /* this shouldn't ever happen!! */
1661
1662     if (!table && !entry) {
1663         const char * charArray[2];
1664         rpm_count_t count = 0;
1665         if (!lang || (lang[0] == 'C' && lang[1] == '\0')) {
1666             charArray[count++] = "C";
1667         } else {
1668             charArray[count++] = "C";
1669             charArray[count++] = lang;
1670         }
1671         if (!headerAddEntry(h, HEADER_I18NTABLE, RPM_STRING_ARRAY_TYPE,
1672                         &charArray, count))
1673             return 0;
1674         table = findEntry(h, HEADER_I18NTABLE, RPM_STRING_ARRAY_TYPE);
1675     }
1676
1677     if (!table)
1678         return 0;
1679     if (!lang) lang = "C";
1680
1681     {   const char * l = table->data;
1682         for (langNum = 0; langNum < table->info.count; langNum++) {
1683             if (!strcmp(l, lang)) break;
1684             l += strlen(l) + 1;
1685         }
1686     }
1687
1688     if (langNum >= table->info.count) {
1689         length = strlen(lang) + 1;
1690         if (ENTRY_IN_REGION(table)) {
1691             char * t = xmalloc(table->length + length);
1692             memcpy(t, table->data, table->length);
1693             table->data = t;
1694             table->info.offset = 0;
1695         } else
1696             table->data = xrealloc(table->data, table->length + length);
1697         memmove(((char *)table->data) + table->length, lang, length);
1698         table->length += length;
1699         table->info.count++;
1700     }
1701
1702     if (!entry) {
1703         int rc;
1704         strArray = xmalloc(sizeof(*strArray) * (langNum + 1));
1705         for (i = 0; i < langNum; i++)
1706             strArray[i] = "";
1707         strArray[langNum] = string;
1708         rc = headerAddEntry(h, tag, RPM_I18NSTRING_TYPE, strArray, langNum + 1);
1709         free(strArray);
1710         return rc;
1711     } else if (langNum >= entry->info.count) {
1712         ghosts = langNum - entry->info.count;
1713
1714         length = strlen(string) + 1 + ghosts;
1715         if (ENTRY_IN_REGION(entry)) {
1716             char * t = xmalloc(entry->length + length);
1717             memcpy(t, entry->data, entry->length);
1718             entry->data = t;
1719             entry->info.offset = 0;
1720         } else
1721             entry->data = xrealloc(entry->data, entry->length + length);
1722
1723         memset(((char *)entry->data) + entry->length, '\0', ghosts);
1724         memmove(((char *)entry->data) + entry->length + ghosts, string, strlen(string)+1);
1725
1726         entry->length += length;
1727         entry->info.count = langNum + 1;
1728     } else {
1729         char *b, *be, *e, *ee, *t;
1730         size_t bn, sn, en;
1731
1732         /* Set beginning/end pointers to previous data */
1733         b = be = e = ee = entry->data;
1734         for (i = 0; i < table->info.count; i++) {
1735             if (i == langNum)
1736                 be = ee;
1737             ee += strlen(ee) + 1;
1738             if (i == langNum)
1739                 e  = ee;
1740         }
1741
1742         /* Get storage for new buffer */
1743         bn = (be-b);
1744         sn = strlen(string) + 1;
1745         en = (ee-e);
1746         length = bn + sn + en;
1747         t = buf = xmalloc(length);
1748
1749         /* Copy values into new storage */
1750         memcpy(t, b, bn);
1751         t += bn;
1752         memcpy(t, string, sn);
1753         t += sn;
1754         memcpy(t, e, en);
1755         t += en;
1756
1757         /* Replace i18N string array */
1758         entry->length -= strlen(be) + 1;
1759         entry->length += sn;
1760
1761         if (ENTRY_IN_REGION(entry)) {
1762             entry->info.offset = 0;
1763         } else
1764             entry->data = _free(entry->data);
1765         entry->data = buf;
1766     }
1767
1768     return 0;
1769 }
1770
1771 int headerModifyEntry(Header h, rpmTag tag, rpmTagType type,
1772                         rpm_constdata_t p, rpm_count_t c)
1773 {
1774     indexEntry entry;
1775     rpm_data_t oldData;
1776     rpm_data_t data;
1777     int length;
1778
1779     /* First find the tag */
1780     entry = findEntry(h, tag, type);
1781     if (!entry)
1782         return 0;
1783
1784     length = 0;
1785     data = grabData(type, p, c, &length);
1786     if (data == NULL || length <= 0)
1787         return 0;
1788
1789     /* make sure entry points to the first occurence of this tag */
1790     while (entry > h->index && (entry - 1)->info.tag == tag)
1791         entry--;
1792
1793     /* free after we've grabbed the new data in case the two are intertwined;
1794        that's a bad idea but at least we won't break */
1795     oldData = entry->data;
1796
1797     entry->info.count = c;
1798     entry->info.type = type;
1799     entry->data = data;
1800     entry->length = length;
1801
1802     if (ENTRY_IN_REGION(entry)) {
1803         entry->info.offset = 0;
1804     } else
1805         oldData = _free(oldData);
1806
1807     return 1;
1808 }
1809
1810 /**
1811  * Header tag iterator data structure.
1812  */
1813 struct headerIterator_s {
1814     Header h;           /*!< Header being iterated. */
1815     int next_index;     /*!< Next tag index. */
1816 };
1817
1818 HeaderIterator headerFreeIterator(HeaderIterator hi)
1819 {
1820     if (hi != NULL) {
1821         hi->h = headerFree(hi->h);
1822         hi = _free(hi);
1823     }
1824     return hi;
1825 }
1826
1827 HeaderIterator headerInitIterator(Header h)
1828 {
1829     HeaderIterator hi = xmalloc(sizeof(*hi));
1830
1831     headerSort(h);
1832
1833     hi->h = headerLink(h);
1834     hi->next_index = 0;
1835     return hi;
1836 }
1837
1838 int headerNext(HeaderIterator hi, rpmtd td)
1839 {
1840     Header h = hi->h;
1841     int slot = hi->next_index;
1842     indexEntry entry = NULL;
1843     int rc;
1844
1845     assert(td != NULL);
1846     rpmtdReset(td);
1847
1848     for (slot = hi->next_index; slot < h->indexUsed; slot++) {
1849         entry = h->index + slot;
1850         if (!ENTRY_IS_REGION(entry))
1851             break;
1852     }
1853     hi->next_index = slot;
1854     if (entry == NULL || slot >= h->indexUsed)
1855         return 0;
1856
1857         /* LCL: no clue */
1858     hi->next_index++;
1859
1860     td->tag = entry->info.tag;
1861
1862     rc = copyTdEntry(entry, td, 0);
1863
1864     /* XXX 1 on success */
1865     return ((rc == 1) ? 1 : 0);
1866 }
1867
1868 int headerNextIterator(HeaderIterator hi,
1869                 rpmTag * tag,
1870                 rpmTagType * type,
1871                 rpm_data_t * p,
1872                 rpm_count_t * c)
1873 {
1874     struct rpmtd_s td;
1875     int rc;
1876
1877     rc = headerNext(hi, &td);
1878     if (tag)
1879         *tag = td.tag;
1880     TDWRAP();
1881     return rc;
1882 }
1883
1884 /** \ingroup header
1885  * Duplicate a header.
1886  * @param h             header
1887  * @return              new header instance
1888  */
1889 Header headerCopy(Header h)
1890 {
1891     Header nh = headerNew();
1892     HeaderIterator hi;
1893     struct rpmtd_s td;
1894
1895     hi = headerInitIterator(h);
1896     while (headerNext(hi, &td)) {
1897         if (rpmtdCount(&td) > 0) {
1898             (void) headerPut(nh, &td, HEADERPUT_DEFAULT);
1899         }
1900         rpmtdFreeData(&td);
1901     }
1902     hi = headerFreeIterator(hi);
1903
1904     return headerReload(nh, HEADER_IMAGE);
1905 }
1906
1907 void headerCopyTags(Header headerFrom, Header headerTo,
1908                     const rpmTag * tagstocopy)
1909 {
1910     const rpmTag * p;
1911     struct rpmtd_s td;
1912
1913     if (headerFrom == headerTo)
1914         return;
1915
1916     for (p = tagstocopy; *p != 0; p++) {
1917         if (headerIsEntry(headerTo, *p))
1918             continue;
1919         if (!headerGet(headerFrom, *p, &td, HEADERGET_MINMEM))
1920             continue;
1921         (void) headerPut(headerTo, &td, HEADERPUT_DEFAULT);
1922         rpmtdFreeData(&td);
1923     }
1924 }
1925
1926 void * headerFreeData(rpm_data_t data, rpmTagType type)
1927 {
1928     if (data) {
1929         if (type == RPM_FORCEFREE_TYPE ||
1930             type == RPM_STRING_ARRAY_TYPE ||
1931             type == RPM_I18NSTRING_TYPE ||
1932             type == RPM_BIN_TYPE)
1933                 free(data); /* XXX _free() */
1934     }
1935     return NULL;
1936 }