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