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 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     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 /* 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_INT64_TYPE:
 452         {   uint64_t * it = (uint64_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 = htonll(*it);
 457             }
 458             t = (unsigned char *) it;
 459         }   break;
 460         case RPM_INT32_TYPE:
 461         {   int32_t * it = (int32_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 = htonl(*it);
 466             }
 467             t = (unsigned char *) it;
 468         }   break;
 469         case RPM_INT16_TYPE:
 470         {   int16_t * it = (int16_t *) t;
 471             for (; ie.info.count > 0; ie.info.count--, it += 1) {
 472                 if (dataEnd && ((unsigned char *)it) >= dataEnd)
 473                     return -1;
 474                 *it = htons(*it);
 475             }
 476             t = (unsigned char *) it;
 477         }   break;
 478         default:
 479             t += ie.length;
 480             break;
 481         }
 482
 483         dl += ie.length;
 484         tl += tdel;
 485         ieprev = ie;    /* structure assignment */
 486
 487     }
 488     tdel = (tprev ? (t - tprev) : 0);
 489     tl += tdel;
 490
 491     /* XXX
 492      * There are two hacks here:
 493      *  1) tl is 16b (i.e. REGION_TAG_COUNT) short while doing headerReload().
 494      *  2) the 8/98 rpm bug with inserting i18n tags needs to use tl, not dl.
 495      */
 496     if (tl+REGION_TAG_COUNT == dl)
 497         tl += REGION_TAG_COUNT;
 498
 499     return dl;
 500 }
 501
 502 /** \ingroup header
 503  * doHeaderUnload.
 504  * @param h             header
 505  * @retval *lengthPtr   no. bytes in unloaded header blob
 506  * @return              unloaded header blob (NULL on error)
 507  */
 508 static void * doHeaderUnload(Header h,
 509                 size_t * lengthPtr)
 510 {
 511     int32_t * ei = NULL;
 512     entryInfo pe;
 513     char * dataStart;
 514     char * te;
 515     unsigned pad;
 516     unsigned len;
 517     int32_t il = 0;
 518     int32_t dl = 0;
 519     indexEntry entry;
 520     rpmTagType type;
 521     int i;
 522     int drlen, ndribbles;
 523     int driplen, ndrips;
 524     int legacy = 0;
 525
 526     /* Sort entries by (offset,tag). */
 527     headerUnsort(h);
 528
 529     /* Compute (il,dl) for all tags, including those deleted in region. */
 530     pad = 0;
 531     drlen = ndribbles = driplen = ndrips = 0;
 532     for (i = 0, entry = h->index; i < h->indexUsed; i++, entry++) {
 533         if (ENTRY_IS_REGION(entry)) {
 534             int32_t rdl = -entry->info.offset;  /* negative offset */
 535             int32_t ril = rdl/sizeof(*pe);
 536             int rid = entry->info.offset;
 537
 538             il += ril;
 539             dl += entry->rdlen + entry->info.count;
 540             /* XXX Legacy regions do not include the region tag and data. */
 541             if (i == 0 && (h->flags & HEADERFLAG_LEGACY))
 542                 il += 1;
 543
 544             /* Skip rest of entries in region, but account for dribbles. */
 545             for (; i < h->indexUsed && entry->info.offset <= rid+1; i++, entry++) {
 546                 if (entry->info.offset <= rid)
 547                     continue;
 548
 549                 /* Alignment */
 550                 type = entry->info.type;
 551                 if (typeSizes[type] > 1) {
 552                     unsigned diff;
 553                     diff = typeSizes[type] - (dl % typeSizes[type]);
 554                     if (diff != typeSizes[type]) {
 555                         drlen += diff;
 556                         pad += diff;
 557                         dl += diff;
 558                     }
 559                 }
 560
 561                 ndribbles++;
 562                 il++;
 563                 drlen += entry->length;
 564                 dl += entry->length;
 565             }
 566             i--;
 567             entry--;
 568             continue;
 569         }
 570
 571         /* Ignore deleted drips. */
 572         if (entry->data == NULL || entry->length <= 0)
 573             continue;
 574
 575         /* Alignment */
 576         type = entry->info.type;
 577         if (typeSizes[type] > 1) {
 578             unsigned diff;
 579             diff = typeSizes[type] - (dl % typeSizes[type]);
 580             if (diff != typeSizes[type]) {
 581                 driplen += diff;
 582                 pad += diff;
 583                 dl += diff;
 584             } else
 585                 diff = 0;
 586         }
 587
 588         ndrips++;
 589         il++;
 590         driplen += entry->length;
 591         dl += entry->length;
 592     }
 593
 594     /* Sanity checks on header intro. */
 595     if (hdrchkTags(il) || hdrchkData(dl))
 596         goto errxit;
 597
 598     len = sizeof(il) + sizeof(dl) + (il * sizeof(*pe)) + dl;
 599
 600     ei = xmalloc(len);
 601     ei[0] = htonl(il);
 602     ei[1] = htonl(dl);
 603
 604     pe = (entryInfo) &ei[2];
 605     dataStart = te = (char *) (pe + il);
 606
 607     pad = 0;
 608     for (i = 0, entry = h->index; i < h->indexUsed; i++, entry++) {
 609         const char * src;
 610         unsigned char *t;
 611         int count;
 612         int rdlen;
 613
 614         if (entry->data == NULL || entry->length <= 0)
 615             continue;
 616
 617         t = (unsigned char*)te;
 618         pe->tag = htonl(entry->info.tag);
 619         pe->type = htonl(entry->info.type);
 620         pe->count = htonl(entry->info.count);
 621
 622         if (ENTRY_IS_REGION(entry)) {
 623             int32_t rdl = -entry->info.offset;  /* negative offset */
 624             int32_t ril = rdl/sizeof(*pe) + ndribbles;
 625             int rid = entry->info.offset;
 626
 627             src = (char *)entry->data;
 628             rdlen = entry->rdlen;
 629
 630             /* XXX Legacy regions do not include the region tag and data. */
 631             if (i == 0 && (h->flags & HEADERFLAG_LEGACY)) {
 632                 int32_t stei[4];
 633
 634                 legacy = 1;
 635                 memcpy(pe+1, src, rdl);
 636                 memcpy(te, src + rdl, rdlen);
 637                 te += rdlen;
 638
 639                 pe->offset = htonl(te - dataStart);
 640                 stei[0] = pe->tag;
 641                 stei[1] = pe->type;
 642                 stei[2] = htonl(-rdl-entry->info.count);
 643                 stei[3] = pe->count;
 644                 memcpy(te, stei, entry->info.count);
 645                 te += entry->info.count;
 646                 ril++;
 647                 rdlen += entry->info.count;
 648
 649                 count = regionSwab(NULL, ril, 0, pe, t, NULL, 0);
 650                 if (count != rdlen)
 651                     goto errxit;
 652
 653             } else {
 654
 655                 memcpy(pe+1, src + sizeof(*pe), ((ril-1) * sizeof(*pe)));
 656                 memcpy(te, src + (ril * sizeof(*pe)), rdlen+entry->info.count+drlen);
 657                 te += rdlen;
 658                 {
 659                     entryInfo se = (entryInfo)src;
 660                     int off = ntohl(se->offset);
 661                     pe->offset = (off) ? htonl(te - dataStart) : htonl(off);
 662                 }
 663                 te += entry->info.count + drlen;
 664
 665                 count = regionSwab(NULL, ril, 0, pe, t, NULL, 0);
 666                 if (count != (rdlen + entry->info.count + drlen))
 667                     goto errxit;
 668             }
 669
 670             /* Skip rest of entries in region. */
 671             while (i < h->indexUsed && entry->info.offset <= rid+1) {
 672                 i++;
 673                 entry++;
 674             }
 675             i--;
 676             entry--;
 677             pe += ril;
 678             continue;
 679         }
 680
 681         /* Ignore deleted drips. */
 682         if (entry->data == NULL || entry->length <= 0)
 683             continue;
 684
 685         /* Alignment */
 686         type = entry->info.type;
 687         if (typeSizes[type] > 1) {
 688             unsigned diff;
 689             diff = typeSizes[type] - ((te - dataStart) % typeSizes[type]);
 690             if (diff != typeSizes[type]) {
 691                 memset(te, 0, diff);
 692                 te += diff;
 693                 pad += diff;
 694             }
 695         }
 696
 697         pe->offset = htonl(te - dataStart);
 698
 699         /* copy data w/ endian conversions */
 700         switch (entry->info.type) {
 701         case RPM_INT64_TYPE:
 702             count = entry->info.count;
 703             src = entry->data;
 704             while (count--) {
 705                 *((uint64_t *)te) = htonll(*((uint64_t *)src));
 706                 te += sizeof(uint64_t);
 707                 src += sizeof(uint64_t);
 708             }
 709             break;
 710
 711         case RPM_INT32_TYPE:
 712             count = entry->info.count;
 713             src = entry->data;
 714             while (count--) {
 715                 *((int32_t *)te) = htonl(*((int32_t *)src));
 716                 te += sizeof(int32_t);
 717                 src += sizeof(int32_t);
 718             }
 719             break;
 720
 721         case RPM_INT16_TYPE:
 722             count = entry->info.count;
 723             src = entry->data;
 724             while (count--) {
 725                 *((int16_t *)te) = htons(*((int16_t *)src));
 726                 te += sizeof(int16_t);
 727                 src += sizeof(int16_t);
 728             }
 729             break;
 730
 731         default:
 732             memcpy(te, entry->data, entry->length);
 733             te += entry->length;
 734             break;
 735         }
 736         pe++;
 737     }
 738
 739     /* Insure that there are no memcpy underruns/overruns. */
 740     if (((char *)pe) != dataStart)
 741         goto errxit;
 742     if ((((char *)ei)+len) != te)
 743         goto errxit;
 744
 745     if (lengthPtr)
 746         *lengthPtr = len;
 747
 748     h->flags &= ~HEADERFLAG_SORTED;
 749     headerSort(h);
 750
 751     return (void *) ei;
 752
 753 errxit:
 754     ei = _free(ei);
 755     return (void *) ei;
 756 }
 757
 758 void * headerUnload(Header h)
 759 {
 760     size_t length;
 761     void * uh = doHeaderUnload(h, &length);
 762     return uh;
 763 }
 764
 765 /**
 766  * Find matching (tag,type) entry in header.
 767  * @param h             header
 768  * @param tag           entry tag
 769  * @param type          entry type
 770  * @return              header entry
 771  */
 772 static
 773 indexEntry findEntry(Header h, rpmTag tag, rpmTagType type)
 774 {
 775     indexEntry entry, entry2, last;
 776     struct indexEntry_s key;
 777
 778     if (h == NULL) return NULL;
 779     if (!(h->flags & HEADERFLAG_SORTED)) headerSort(h);
 780
 781     key.info.tag = tag;
 782
 783     entry2 = entry =
 784         bsearch(&key, h->index, h->indexUsed, sizeof(*h->index), indexCmp);
 785     if (entry == NULL)
 786         return NULL;
 787
 788     if (type == RPM_NULL_TYPE)
 789         return entry;
 790
 791     /* look backwards */
 792     while (entry->info.tag == tag && entry->info.type != type &&
 793            entry > h->index) entry--;
 794
 795     if (entry->info.tag == tag && entry->info.type == type)
 796         return entry;
 797
 798     last = h->index + h->indexUsed;
 799     /* FIX: entry2 = entry. Code looks bogus as well. */
 800     while (entry2->info.tag == tag && entry2->info.type != type &&
 801            entry2 < last) entry2++;
 802
 803     if (entry->info.tag == tag && entry->info.type == type)
 804         return entry;
 805
 806     return NULL;
 807 }
 808
 809 int headerDel(Header h, rpmTag tag)
 810 {
 811     indexEntry last = h->index + h->indexUsed;
 812     indexEntry entry, first;
 813     int ne;
 814
 815     entry = findEntry(h, tag, RPM_NULL_TYPE);
 816     if (!entry) return 1;
 817
 818     /* Make sure entry points to the first occurence of this tag. */
 819     while (entry > h->index && (entry - 1)->info.tag == tag)
 820         entry--;
 821
 822     /* Free data for tags being removed. */
 823     for (first = entry; first < last; first++) {
 824         rpm_data_t data;
 825         if (first->info.tag != tag)
 826             break;
 827         data = first->data;
 828         first->data = NULL;
 829         first->length = 0;
 830         if (ENTRY_IN_REGION(first))
 831             continue;
 832         data = _free(data);
 833     }
 834
 835     ne = (first - entry);
 836     if (ne > 0) {
 837         h->indexUsed -= ne;
 838         ne = last - first;
 839         if (ne > 0)
 840             memmove(entry, first, (ne * sizeof(*entry)));
 841     }
 842
 843     return 0;
 844 }
 845
 846 int headerRemoveEntry(Header h, rpmTag tag)
 847 {
 848     return headerDel(h, tag);
 849 }
 850
 851 Header headerLoad(void * uh)
 852 {
 853     int32_t * ei = (int32_t *) uh;
 854     int32_t il = ntohl(ei[0]);          /* index length */
 855     int32_t dl = ntohl(ei[1]);          /* data length */
 856     size_t pvlen = sizeof(il) + sizeof(dl) +
 857                (il * sizeof(struct entryInfo_s)) + dl;
 858     void * pv = uh;
 859     Header h = NULL;
 860     entryInfo pe;
 861     unsigned char * dataStart;
 862     unsigned char * dataEnd;
 863     indexEntry entry;
 864     int rdlen;
 865     int i;
 866
 867     /* Sanity checks on header intro. */
 868     if (hdrchkTags(il) || hdrchkData(dl))
 869         goto errxit;
 870
 871     ei = (int32_t *) pv;
 872     pe = (entryInfo) &ei[2];
 873     dataStart = (unsigned char *) (pe + il);
 874     dataEnd = dataStart + dl;
 875
 876     h = xcalloc(1, sizeof(*h));
 877     h->blob = uh;
 878     h->indexAlloced = il + 1;
 879     h->indexUsed = il;
 880     h->index = xcalloc(h->indexAlloced, sizeof(*h->index));
 881     h->flags |= HEADERFLAG_SORTED;
 882     h->nrefs = 0;
 883     h = headerLink(h);
 884
 885     entry = h->index;
 886     i = 0;
 887     if (!(htonl(pe->tag) < HEADER_I18NTABLE)) {
 888         h->flags |= HEADERFLAG_LEGACY;
 889         entry->info.type = REGION_TAG_TYPE;
 890         entry->info.tag = HEADER_IMAGE;
 891         entry->info.count = REGION_TAG_COUNT;
 892         entry->info.offset = ((unsigned char *)pe - dataStart); /* negative offset */
 893
 894         entry->data = pe;
 895         entry->length = pvlen - sizeof(il) - sizeof(dl);
 896         rdlen = regionSwab(entry+1, il, 0, pe, dataStart, dataEnd, entry->info.offset);
 897 #if 0   /* XXX don't check, the 8/98 i18n bug fails here. */
 898         if (rdlen != dl)
 899             goto errxit;
 900 #endif
 901         entry->rdlen = rdlen;
 902         entry++;
 903         h->indexUsed++;
 904     } else {
 905         int32_t rdl;
 906         int32_t ril;
 907
 908         h->flags &= ~HEADERFLAG_LEGACY;
 909
 910         entry->info.type = htonl(pe->type);
 911         entry->info.count = htonl(pe->count);
 912
 913         if (hdrchkType(entry->info.type))
 914             goto errxit;
 915         if (hdrchkTags(entry->info.count))
 916             goto errxit;
 917
 918         {   int off = ntohl(pe->offset);
 919
 920             if (hdrchkData(off))
 921                 goto errxit;
 922             if (off) {
 923                 size_t nb = REGION_TAG_COUNT;
 924                 int32_t stei[nb];
 925                 /* XXX Hmm, why the copy? */
 926                 memcpy(&stei, dataStart + off, nb);
 927                 rdl = -ntohl(stei[2]);  /* negative offset */
 928                 ril = rdl/sizeof(*pe);
 929                 if (hdrchkTags(ril) || hdrchkData(rdl))
 930                     goto errxit;
 931                 entry->info.tag = htonl(pe->tag);
 932             } else {
 933                 ril = il;
 934                 rdl = (ril * sizeof(struct entryInfo_s));
 935                 entry->info.tag = HEADER_IMAGE;
 936             }
 937         }
 938         entry->info.offset = -rdl;      /* negative offset */
 939
 940         entry->data = pe;
 941         entry->length = pvlen - sizeof(il) - sizeof(dl);
 942         rdlen = regionSwab(entry+1, ril-1, 0, pe+1, dataStart, dataEnd, entry->info.offset);
 943         if (rdlen < 0)
 944             goto errxit;
 945         entry->rdlen = rdlen;
 946
 947         if (ril < h->indexUsed) {
 948             indexEntry newEntry = entry + ril;
 949             int ne = (h->indexUsed - ril);
 950             int rid = entry->info.offset+1;
 951             int rc;
 952
 953             /* Load dribble entries from region. */
 954             rc = regionSwab(newEntry, ne, 0, pe+ril, dataStart, dataEnd, rid);
 955             if (rc < 0)
 956                 goto errxit;
 957             rdlen += rc;
 958
 959           { indexEntry firstEntry = newEntry;
 960             int save = h->indexUsed;
 961             int j;
 962
 963             /* Dribble entries replace duplicate region entries. */
 964             h->indexUsed -= ne;
 965             for (j = 0; j < ne; j++, newEntry++) {
 966                 (void) headerDel(h, newEntry->info.tag);
 967                 if (newEntry->info.tag == HEADER_BASENAMES)
 968                     (void) headerDel(h, HEADER_OLDFILENAMES);
 969             }
 970
 971             /* If any duplicate entries were replaced, move new entries down. */
 972             if (h->indexUsed < (save - ne)) {
 973                 memmove(h->index + h->indexUsed, firstEntry,
 974                         (ne * sizeof(*entry)));
 975             }
 976             h->indexUsed += ne;
 977           }
 978         }
 979     }
 980
 981     h->flags &= ~HEADERFLAG_SORTED;
 982     headerSort(h);
 983
 984     return h;
 985
 986 errxit:
 987     if (h) {
 988         h->index = _free(h->index);
 989         h = _free(h);
 990     }
 991     return h;
 992 }
 993
 994 Header headerReload(Header h, rpmTag tag)
 995 {
 996     Header nh;
 997     size_t length;
 998     void * uh = doHeaderUnload(h, &length);
 999
1000     h = headerFree(h);
1001     if (uh == NULL)
1002         return NULL;
1003     nh = headerLoad(uh);
1004     if (nh == NULL) {
1005         uh = _free(uh);
1006         return NULL;
1007     }
1008     if (nh->flags & HEADERFLAG_ALLOCATED)
1009         uh = _free(uh);
1010     nh->flags |= HEADERFLAG_ALLOCATED;
1011     if (ENTRY_IS_REGION(nh->index)) {
1012         if (tag == HEADER_SIGNATURES || tag == HEADER_IMMUTABLE)
1013             nh->index[0].info.tag = tag;
1014     }
1015     return nh;
1016 }
1017
1018 Header headerCopyLoad(const void * uh)
1019 {
1020     int32_t * ei = (int32_t *) uh;
1021     int32_t il = ntohl(ei[0]);          /* index length */
1022     int32_t dl = ntohl(ei[1]);          /* data length */
1023     size_t pvlen = sizeof(il) + sizeof(dl) +
1024                         (il * sizeof(struct entryInfo_s)) + dl;
1025     void * nuh = NULL;
1026     Header h = NULL;
1027
1028     /* Sanity checks on header intro. */
1029     if (!(hdrchkTags(il) || hdrchkData(dl)) && pvlen < headerMaxbytes) {
1030         nuh = memcpy(xmalloc(pvlen), uh, pvlen);
1031         if ((h = headerLoad(nuh)) != NULL)
1032             h->flags |= HEADERFLAG_ALLOCATED;
1033     }
1034     if (h == NULL)
1035         nuh = _free(nuh);
1036     return h;
1037 }
1038
1039 /** \ingroup header
1040  * Read (and load) header from file handle.
1041  * @param fd            file handle
1042  * @param magicp        read (and verify) 8 bytes of (magic, 0)?
1043  * @return              header (or NULL on error)
1044  */
1045 Header headerRead(FD_t fd, enum hMagic magicp)
1046 {
1047     int32_t block[4];
1048     int32_t reserved;
1049     int32_t * ei = NULL;
1050     int32_t il;
1051     int32_t dl;
1052     int32_t magic;
1053     Header h = NULL;
1054     size_t len;
1055     int i;
1056
1057     memset(block, 0, sizeof(block));
1058     i = 2;
1059     if (magicp == HEADER_MAGIC_YES)
1060         i += 2;
1061
1062     /* FIX: cast? */
1063     if (timedRead(fd, (char *)block, i*sizeof(*block)) != (i * sizeof(*block)))
1064         goto exit;
1065
1066     i = 0;
1067
1068     if (magicp == HEADER_MAGIC_YES) {
1069         magic = block[i++];
1070         if (memcmp(&magic, header_magic, sizeof(magic)))
1071             goto exit;
1072         reserved = block[i++];
1073     }
1074
1075     il = ntohl(block[i]);       i++;
1076     dl = ntohl(block[i]);       i++;
1077
1078     len = sizeof(il) + sizeof(dl) + (il * sizeof(struct entryInfo_s)) + dl;
1079
1080     /* Sanity checks on header intro. */
1081     if (hdrchkTags(il) || hdrchkData(dl) || len > headerMaxbytes)
1082         goto exit;
1083
1084     ei = xmalloc(len);
1085     ei[0] = htonl(il);
1086     ei[1] = htonl(dl);
1087     len -= sizeof(il) + sizeof(dl);
1088
1089     /* FIX: cast? */
1090     if (timedRead(fd, (char *)&ei[2], len) != len)
1091         goto exit;
1092
1093     h = headerLoad(ei);
1094
1095 exit:
1096     if (h) {
1097         if (h->flags & HEADERFLAG_ALLOCATED)
1098             ei = _free(ei);
1099         h->flags |= HEADERFLAG_ALLOCATED;
1100     } else if (ei)
1101         ei = _free(ei);
1102         /* FIX: timedRead macro obscures annotation */
1103     return h;
1104 }
1105
1106 int headerWrite(FD_t fd, Header h, enum hMagic magicp)
1107 {
1108     ssize_t nb;
1109     size_t length;
1110     void * uh;
1111
1112     if (h == NULL)
1113         return 1;
1114     uh = doHeaderUnload(h, &length);
1115     if (uh == NULL)
1116         return 1;
1117     switch (magicp) {
1118     case HEADER_MAGIC_YES:
1119         nb = Fwrite(header_magic, sizeof(char), sizeof(header_magic), fd);
1120         if (nb != sizeof(header_magic))
1121             goto exit;
1122         break;
1123     case HEADER_MAGIC_NO:
1124         break;
1125     }
1126
1127     nb = Fwrite(uh, sizeof(char), length, fd);
1128
1129 exit:
1130     uh = _free(uh);
1131     return (nb == length ? 0 : 1);
1132 }
1133
1134 int headerIsEntry(Header h, rpmTag tag)
1135 {
1136                 /* FIX: h modified by sort. */
1137     return (findEntry(h, tag, RPM_NULL_TYPE) ? 1 : 0);
1138
1139 }
1140
1141 /** \ingroup header
1142  * Retrieve data from header entry.
1143  * Relevant flags (others are ignored), if neither is set allocation
1144  * behavior depends on data type(!)
1145  *     HEADERGET_MINMEM: return pointers to header memory
1146  *     HEADERGET_ALLOC: always return malloced memory, overrides MINMEM
1147  *
1148  * @todo Permit retrieval of regions other than HEADER_IMUTABLE.
1149  * @param entry         header entry
1150  * @param td            tag data container
1151  * @param minMem        string pointers refer to header memory?
1152  * @param flags         flags to control memory allocation
1153  * @return              1 on success, otherwise error.
1154  */
1155 static int copyTdEntry(const indexEntry entry, rpmtd td, headerGetFlags flags)
1156 {
1157     rpm_count_t count = entry->info.count;
1158     int rc = 1;         /* XXX 1 on success. */
1159     /* ALLOC overrides MINMEM */
1160     int allocMem = flags & HEADERGET_ALLOC;
1161     int minMem = allocMem ? 0 : flags & HEADERGET_MINMEM;
1162
1163     assert(td != NULL);
1164     switch (entry->info.type) {
1165     case RPM_BIN_TYPE:
1166         /*
1167          * XXX This only works for
1168          * XXX  "sealed" HEADER_IMMUTABLE/HEADER_SIGNATURES/HEADER_IMAGE.
1169          * XXX This will *not* work for unsealed legacy HEADER_IMAGE (i.e.
1170          * XXX a legacy header freshly read, but not yet unloaded to the rpmdb).
1171          */
1172         if (ENTRY_IS_REGION(entry)) {
1173             int32_t * ei = ((int32_t *)entry->data) - 2;
1174             entryInfo pe = (entryInfo) (ei + 2);
1175             unsigned char * dataStart = (unsigned char *) (pe + ntohl(ei[0]));
1176             int32_t rdl = -entry->info.offset;  /* negative offset */
1177             int32_t ril = rdl/sizeof(*pe);
1178
1179             rdl = entry->rdlen;
1180             count = 2 * sizeof(*ei) + (ril * sizeof(*pe)) + rdl;
1181             if (entry->info.tag == HEADER_IMAGE) {
1182                 ril -= 1;
1183                 pe += 1;
1184             } else {
1185                 count += REGION_TAG_COUNT;
1186                 rdl += REGION_TAG_COUNT;
1187             }
1188
1189             td->data = xmalloc(count);
1190             ei = (int32_t *) td->data;
1191             ei[0] = htonl(ril);
1192             ei[1] = htonl(rdl);
1193
1194             pe = (entryInfo) memcpy(ei + 2, pe, (ril * sizeof(*pe)));
1195
1196             dataStart = (unsigned char *) memcpy(pe + ril, dataStart, rdl);
1197
1198             rc = regionSwab(NULL, ril, 0, pe, dataStart, dataStart + rdl, 0);
1199             /* XXX 1 on success. */
1200             rc = (rc < 0) ? 0 : 1;
1201         } else {
1202             count = entry->length;
1203             td->data = (!minMem
1204                 ? memcpy(xmalloc(count), entry->data, count)
1205                 : entry->data);
1206         }
1207         break;
1208     case RPM_STRING_TYPE:
1209         if (count == 1) {
1210             td->data = allocMem ? xstrdup(entry->data) : entry->data;
1211             break;
1212         }
1213     case RPM_STRING_ARRAY_TYPE:
1214     case RPM_I18NSTRING_TYPE:
1215     {   const char ** ptrEntry;
1216         int tableSize = count * sizeof(char *);
1217         char * t;
1218         int i;
1219
1220         if (minMem) {
1221             td->data = xmalloc(tableSize);
1222             ptrEntry = (const char **) td->data;
1223             t = entry->data;
1224         } else {
1225             t = xmalloc(tableSize + entry->length);
1226             td->data = (void *)t;
1227             ptrEntry = (const char **) td->data;
1228             t += tableSize;
1229             memcpy(t, entry->data, entry->length);
1230         }
1231         for (i = 0; i < count; i++) {
1232             *ptrEntry++ = t;
1233             t = strchr(t, 0);
1234             t++;
1235         }
1236     }   break;
1237     case RPM_CHAR_TYPE:
1238     case RPM_INT8_TYPE:
1239     case RPM_INT16_TYPE:
1240     case RPM_INT32_TYPE:
1241     case RPM_INT64_TYPE:
1242         if (allocMem) {
1243             td->data = xmalloc(entry->length);
1244             memcpy(td->data, entry->data, entry->length);
1245         } else {
1246             td->data = entry->data;
1247         }
1248         break;
1249     default:
1250         /* WTH? Don't mess with unknown data types... */
1251         rc = 0;
1252         td->data = NULL;
1253         break;
1254     }
1255     td->type = entry->info.type;
1256     td->count = count;
1257
1258     td->flags = RPMTD_IMMUTABLE;
1259     if (td->data && entry->data != td->data) {
1260         td->flags |= RPMTD_ALLOCED;
1261     }
1262
1263     return rc;
1264 }
1265
1266 /** \ingroup header
1267  * Retrieve data from header entry.
1268  * Just a compat wrapper around copyTdEntry() until everything's converted
1269  * to new interface.
1270  * @param entry         header entry
1271  * @retval type         address of type (or NULL)
1272  * @retval p            address of data (or NULL)
1273  * @retval c            address of count (or NULL)
1274  * @param minMem        string pointers refer to header memory?
1275  * @return              1 on success, otherwise error.
1276  */
1277 static int copyEntry(const indexEntry entry,
1278                 rpmTagType * type,
1279                 rpm_data_t * p,
1280                 rpm_count_t * c,
1281                 int minMem)
1282 {
1283     struct rpmtd_s td;
1284     int rc;
1285
1286     rpmtdReset(&td);
1287     rc = copyTdEntry(entry, &td, minMem ? HEADERGET_MINMEM : HEADERGET_DEFAULT);
1288     TDWRAP();
1289     return rc;
1290 }
1291
1292 /**
1293  * Does locale match entry in header i18n table?
1294  *
1295  * \verbatim
1296  * The range [l,le) contains the next locale to match:
1297  *    ll[_CC][.EEEEE][@dddd]
1298  * where
1299  *    ll        ISO language code (in lowercase).
1300  *    CC        (optional) ISO coutnry code (in uppercase).
1301  *    EEEEE     (optional) encoding (not really standardized).
1302  *    dddd      (optional) dialect.
1303  * \endverbatim
1304  *
1305  * @param td            header i18n table data, NUL terminated
1306  * @param l             start of locale to match
1307  * @param le            end of locale to match
1308  * @return              1 on good match, 2 on weak match, 0 on no match
1309  */
1310 static int headerMatchLocale(const char *td, const char *l, const char *le)
1311 {
1312     const char *fe;
1313
1314     /* First try a complete match. */
1315     if (strlen(td) == (le-l) && !strncmp(td, l, (le - l)))
1316         return 1;
1317
1318     /* Next, try stripping optional dialect and matching.  */
1319     for (fe = l; fe < le && *fe != '@'; fe++)
1320         {};
1321     if (fe < le && !strncmp(td, l, (fe - l)))
1322         return 1;
1323
1324     /* Next, try stripping optional codeset and matching.  */
1325     for (fe = l; fe < le && *fe != '.'; fe++)
1326         {};
1327     if (fe < le && !strncmp(td, l, (fe - l)))
1328         return 1;
1329
1330     /* Finally, try stripping optional country code and matching. */
1331     for (fe = l; fe < le && *fe != '_'; fe++)
1332         {};
1333     if (fe < le && !strncmp(td, l, (fe - l)))
1334         return 2;
1335
1336     return 0;
1337 }
1338
1339 /**
1340  * Return i18n string from header that matches locale.
1341  * @param h             header
1342  * @param entry         i18n string data
1343  * @retval td           tag data container
1344  * @param flags         flags to control allocation
1345  * @return              1 always
1346  */
1347 static int copyI18NEntry(Header h, indexEntry entry, rpmtd td,
1348                                                 headerGetFlags flags)
1349 {
1350     const char *lang, *l, *le;
1351     indexEntry table;
1352
1353     td->type = RPM_STRING_TYPE;
1354     td->count = 1;
1355     /* if no match, just return the first string */
1356     td->data = entry->data;
1357
1358     /* XXX Drepper sez' this is the order. */
1359     if ((lang = getenv("LANGUAGE")) == NULL &&
1360         (lang = getenv("LC_ALL")) == NULL &&
1361         (lang = getenv("LC_MESSAGES")) == NULL &&
1362         (lang = getenv("LANG")) == NULL)
1363             goto exit;
1364
1365     if ((table = findEntry(h, HEADER_I18NTABLE, RPM_STRING_ARRAY_TYPE)) == NULL)
1366         goto exit;
1367
1368     for (l = lang; *l != '\0'; l = le) {
1369         const char *t;
1370         char *ed, *ed_weak = NULL;
1371         int langNum;
1372
1373         while (*l && *l == ':')                 /* skip leading colons */
1374             l++;
1375         if (*l == '\0')
1376             break;
1377         for (le = l; *le && *le != ':'; le++)   /* find end of this locale */
1378             {};
1379
1380         /* For each entry in the header ... */
1381         for (langNum = 0, t = table->data, ed = entry->data;
1382              langNum < entry->info.count;
1383              langNum++, t += strlen(t) + 1, ed += strlen(ed) + 1) {
1384
1385             int match = headerMatchLocale(t, l, le);
1386             if (match == 1) {
1387                 td->data = ed;
1388                 goto exit;
1389             } else if (match == 2) {
1390                 ed_weak = ed;
1391             }
1392         }
1393         if (ed_weak) {
1394             td->data = ed_weak;
1395             goto exit;
1396         }
1397     }
1398
1399 exit:
1400     if (flags & HEADERGET_ALLOC) {
1401         td->data = xstrdup(td->data);
1402     }
1403
1404     return 1;
1405 }
1406
1407 /**
1408  * Retrieve tag data from header.
1409  * @param h             header
1410  * @param tag           tag to retrieve
1411  * @retval td           tag data container
1412  * @param flags         flags to control retrieval
1413  * @return              1 on success, 0 on not found
1414  */
1415 static int intGetTdEntry(Header h, rpmTag tag, rpmtd td, headerGetFlags flags)
1416 {
1417     indexEntry entry;
1418     int rc;
1419
1420     assert(td != NULL);
1421     /* ensure clean state */
1422     rpmtdReset(td);
1423     td->tag = tag;
1424     /* First find the tag */
1425     /* FIX: h modified by sort. */
1426     entry = findEntry(h, tag, RPM_NULL_TYPE);
1427     if (entry == NULL) {
1428         /* Td is zeroed above, just return... */
1429         return 0;
1430     }
1431
1432     if (flags & HEADERGET_RAW) {
1433         rc = copyTdEntry(entry, td, flags);
1434     } else {
1435         switch (entry->info.type) {
1436         case RPM_I18NSTRING_TYPE:
1437             rc = copyI18NEntry(h, entry, td, flags);
1438             break;
1439         default:
1440             rc = copyTdEntry(entry, td, flags);
1441             break;
1442         }
1443     }
1444
1445     /* XXX 1 on success */
1446     return ((rc == 1) ? 1 : 0);
1447 }
1448
1449 void * headerFreeTag(Header h, rpm_data_t data, rpmTagType type)
1450 {
1451     if (data) {
1452         if (type == RPM_FORCEFREE_TYPE ||
1453             type == RPM_STRING_ARRAY_TYPE ||
1454             type == RPM_I18NSTRING_TYPE ||
1455             type == RPM_BIN_TYPE)
1456                 data = _free(data);
1457     }
1458     return NULL;
1459 }
1460
1461 /*
1462  * XXX temporary kludgery until tag extensions have been converted to
1463  * take rpmtd as argument
1464  */
1465 static int intGetTagExt(Header h, rpmTag tag, rpmtd td, headerTagTagFunction tagfunc)
1466 {
1467     int rc;
1468     rpmtdReset(td);
1469     rc = tagfunc(h, td);
1470     td->tag = tag;
1471     return rc;
1472 }
1473
1474 int headerGet(Header h, rpmTag tag, rpmtd td, headerGetFlags flags)
1475 {
1476     int rc;
1477     headerTagTagFunction tagfunc = NULL;
1478
1479     assert(td != NULL);
1480
1481     if (flags & HEADERGET_EXT) {
1482         tagfunc = rpmHeaderTagFunc(tag);
1483     }
1484
1485     if (tagfunc) {
1486         rc = intGetTagExt(h, tag, td, tagfunc);
1487     } else {
1488         rc = intGetTdEntry(h, tag, td, flags);
1489     }
1490
1491     assert(tag == td->tag);
1492     return rc;
1493 }
1494
1495 static int headerGetWrap(Header h, rpmTag tag,
1496                 rpmTagType * type,
1497                 rpm_data_t * p,
1498                 rpm_count_t * c,
1499                 headerGetFlags flags)
1500 {
1501     struct rpmtd_s td;
1502     int rc;
1503
1504     rc = headerGet(h, tag, &td, flags);
1505     TDWRAP();
1506     return rc;
1507 }
1508
1509 int headerGetEntry(Header h, rpmTag tag,
1510                         rpmTagType * type,
1511                         rpm_data_t * p,
1512                         rpm_count_t * c)
1513 {
1514     return headerGetWrap(h, tag, type, p, c, HEADERGET_DEFAULT);
1515 }
1516
1517 int headerGetEntryMinMemory(Header h, rpmTag tag,
1518                         rpmTagType * type,
1519                         rpm_data_t * p,
1520                         rpm_count_t * c)
1521 {
1522     return headerGetWrap(h, tag, type, (rpm_data_t) p, c, HEADERGET_MINMEM);
1523 }
1524
1525 int headerGetRawEntry(Header h, rpmTag tag, rpmTagType * type, rpm_data_t * p,
1526                 rpm_count_t * c)
1527 {
1528     indexEntry entry;
1529     int rc;
1530
1531     if (p == NULL) return headerIsEntry(h, tag);
1532
1533     /* First find the tag */
1534                 /* FIX: h modified by sort. */
1535     entry = findEntry(h, tag, RPM_NULL_TYPE);
1536     if (!entry) {
1537         if (p) *p = NULL;
1538         if (c) *c = 0;
1539         return 0;
1540     }
1541
1542     rc = copyEntry(entry, type, p, c, 0);
1543
1544     /* XXX 1 on success */
1545     return ((rc == 1) ? 1 : 0);
1546 }
1547
1548 /**
1549  */
1550 static void copyData(rpmTagType type, rpm_data_t dstPtr,
1551                 rpm_constdata_t srcPtr, rpm_count_t cnt, int dataLength)
1552 {
1553     switch (type) {
1554     case RPM_STRING_ARRAY_TYPE:
1555     case RPM_I18NSTRING_TYPE:
1556     {   const char ** av = (const char **) srcPtr;
1557         char * t = dstPtr;
1558
1559         while (cnt-- > 0 && dataLength > 0) {
1560             const char * s;
1561             if ((s = *av++) == NULL)
1562                 continue;
1563             do {
1564                 *t++ = *s++;
1565             } while (s[-1] && --dataLength > 0);
1566         }
1567     }   break;
1568
1569     default:
1570         memmove(dstPtr, srcPtr, dataLength);
1571         break;
1572     }
1573 }
1574
1575 /**
1576  * Return (malloc'ed) copy of entry data.
1577  * @param type          entry data type
1578  * @param p             entry data
1579  * @param c             entry item count
1580  * @retval lengthPtr    no. bytes in returned data
1581  * @return              (malloc'ed) copy of entry data, NULL on error
1582  */
1583 static void *
1584 grabData(rpmTagType type, rpm_constdata_t p, rpm_count_t c, int * lengthPtr)
1585 {
1586     rpm_data_t data = NULL;
1587     int length;
1588
1589     length = dataLength(type, p, c, 0, NULL);
1590     if (length > 0) {
1591         data = xmalloc(length);
1592         copyData(type, data, p, c, length);
1593     }
1594
1595     if (lengthPtr)
1596         *lengthPtr = length;
1597     return data;
1598 }
1599
1600 int headerAddEntry(Header h, rpmTag tag, rpmTagType type,
1601                         rpm_constdata_t p, rpm_count_t c)
1602 {
1603     indexEntry entry;
1604     rpm_data_t data;
1605     int length;
1606
1607     /* Count must always be >= 1 for headerAddEntry. */
1608     if (c <= 0)
1609         return 0;
1610
1611     if (hdrchkType(type))
1612         return 0;
1613     if (hdrchkData(c))
1614         return 0;
1615
1616     length = 0;
1617     data = grabData(type, p, c, &length);
1618     if (data == NULL || length <= 0)
1619         return 0;
1620
1621     /* Allocate more index space if necessary */
1622     if (h->indexUsed == h->indexAlloced) {
1623         h->indexAlloced += INDEX_MALLOC_SIZE;
1624         h->index = xrealloc(h->index, h->indexAlloced * sizeof(*h->index));
1625     }
1626
1627     /* Fill in the index */
1628     entry = h->index + h->indexUsed;
1629     entry->info.tag = tag;
1630     entry->info.type = type;
1631     entry->info.count = c;
1632     entry->info.offset = 0;
1633     entry->data = data;
1634     entry->length = length;
1635
1636     if (h->indexUsed > 0 && tag < h->index[h->indexUsed-1].info.tag)
1637         h->flags &= ~HEADERFLAG_SORTED;
1638     h->indexUsed++;
1639
1640     return 1;
1641 }
1642
1643 int headerAppendEntry(Header h, rpmTag tag, rpmTagType type,
1644                 rpm_constdata_t p, rpm_count_t c)
1645 {
1646     indexEntry entry;
1647     int length;
1648
1649     if (type == RPM_STRING_TYPE || type == RPM_I18NSTRING_TYPE) {
1650         /* we can't do this */
1651         return 0;
1652     }
1653
1654     /* Find the tag entry in the header. */
1655     entry = findEntry(h, tag, type);
1656     if (!entry)
1657         return 0;
1658
1659     length = dataLength(type, p, c, 0, NULL);
1660     if (length < 0)
1661         return 0;
1662
1663     if (ENTRY_IN_REGION(entry)) {
1664         char * t = xmalloc(entry->length + length);
1665         memcpy(t, entry->data, entry->length);
1666         entry->data = t;
1667         entry->info.offset = 0;
1668     } else
1669         entry->data = xrealloc(entry->data, entry->length + length);
1670
1671     copyData(type, ((char *) entry->data) + entry->length, p, c, length);
1672
1673     entry->length += length;
1674
1675     entry->info.count += c;
1676
1677     return 1;
1678 }
1679
1680 int headerAddOrAppendEntry(Header h, rpmTag tag, rpmTagType type,
1681                 rpm_constdata_t p, rpm_count_t c)
1682 {
1683     return (findEntry(h, tag, type)
1684         ? headerAppendEntry(h, tag, type, p, c)
1685         : headerAddEntry(h, tag, type, p, c));
1686 }
1687
1688 int headerPut(Header h, rpmtd td, headerPutFlags flags)
1689 {
1690     int rc;
1691
1692     assert(td != NULL);
1693     if (flags & HEADERPUT_APPEND) {
1694         rc = headerAddOrAppendEntry(h, td->tag, td->type, td->data, td->count);
1695     } else {
1696         rc = headerAddEntry(h, td->tag, td->type, td->data, td->count);
1697     }
1698     return rc;
1699 }
1700
1701 int headerAddI18NString(Header h, rpmTag tag, const char * string,
1702                 const char * lang)
1703 {
1704     indexEntry table, entry;
1705     const char ** strArray;
1706     int length;
1707     int ghosts;
1708     rpm_count_t i, langNum;
1709     char * buf;
1710
1711     table = findEntry(h, HEADER_I18NTABLE, RPM_STRING_ARRAY_TYPE);
1712     entry = findEntry(h, tag, RPM_I18NSTRING_TYPE);
1713
1714     if (!table && entry)
1715         return 0;               /* this shouldn't ever happen!! */
1716
1717     if (!table && !entry) {
1718         const char * charArray[2];
1719         rpm_count_t count = 0;
1720         if (!lang || (lang[0] == 'C' && lang[1] == '\0')) {
1721             charArray[count++] = "C";
1722         } else {
1723             charArray[count++] = "C";
1724             charArray[count++] = lang;
1725         }
1726         if (!headerAddEntry(h, HEADER_I18NTABLE, RPM_STRING_ARRAY_TYPE,
1727                         &charArray, count))
1728             return 0;
1729         table = findEntry(h, HEADER_I18NTABLE, RPM_STRING_ARRAY_TYPE);
1730     }
1731
1732     if (!table)
1733         return 0;
1734     if (!lang) lang = "C";
1735
1736     {   const char * l = table->data;
1737         for (langNum = 0; langNum < table->info.count; langNum++) {
1738             if (!strcmp(l, lang)) break;
1739             l += strlen(l) + 1;
1740         }
1741     }
1742
1743     if (langNum >= table->info.count) {
1744         length = strlen(lang) + 1;
1745         if (ENTRY_IN_REGION(table)) {
1746             char * t = xmalloc(table->length + length);
1747             memcpy(t, table->data, table->length);
1748             table->data = t;
1749             table->info.offset = 0;
1750         } else
1751             table->data = xrealloc(table->data, table->length + length);
1752         memmove(((char *)table->data) + table->length, lang, length);
1753         table->length += length;
1754         table->info.count++;
1755     }
1756
1757     if (!entry) {
1758         int rc;
1759         strArray = xmalloc(sizeof(*strArray) * (langNum + 1));
1760         for (i = 0; i < langNum; i++)
1761             strArray[i] = "";
1762         strArray[langNum] = string;
1763         rc = headerAddEntry(h, tag, RPM_I18NSTRING_TYPE, strArray, langNum + 1);
1764         free(strArray);
1765         return rc;
1766     } else if (langNum >= entry->info.count) {
1767         ghosts = langNum - entry->info.count;
1768
1769         length = strlen(string) + 1 + ghosts;
1770         if (ENTRY_IN_REGION(entry)) {
1771             char * t = xmalloc(entry->length + length);
1772             memcpy(t, entry->data, entry->length);
1773             entry->data = t;
1774             entry->info.offset = 0;
1775         } else
1776             entry->data = xrealloc(entry->data, entry->length + length);
1777
1778         memset(((char *)entry->data) + entry->length, '\0', ghosts);
1779         memmove(((char *)entry->data) + entry->length + ghosts, string, strlen(string)+1);
1780
1781         entry->length += length;
1782         entry->info.count = langNum + 1;
1783     } else {
1784         char *b, *be, *e, *ee, *t;
1785         size_t bn, sn, en;
1786
1787         /* Set beginning/end pointers to previous data */
1788         b = be = e = ee = entry->data;
1789         for (i = 0; i < table->info.count; i++) {
1790             if (i == langNum)
1791                 be = ee;
1792             ee += strlen(ee) + 1;
1793             if (i == langNum)
1794                 e  = ee;
1795         }
1796
1797         /* Get storage for new buffer */
1798         bn = (be-b);
1799         sn = strlen(string) + 1;
1800         en = (ee-e);
1801         length = bn + sn + en;
1802         t = buf = xmalloc(length);
1803
1804         /* Copy values into new storage */
1805         memcpy(t, b, bn);
1806         t += bn;
1807         memcpy(t, string, sn);
1808         t += sn;
1809         memcpy(t, e, en);
1810         t += en;
1811
1812         /* Replace i18N string array */
1813         entry->length -= strlen(be) + 1;
1814         entry->length += sn;
1815
1816         if (ENTRY_IN_REGION(entry)) {
1817             entry->info.offset = 0;
1818         } else
1819             entry->data = _free(entry->data);
1820         entry->data = buf;
1821     }
1822
1823     return 0;
1824 }
1825
1826 int headerModifyEntry(Header h, rpmTag tag, rpmTagType type,
1827                         rpm_constdata_t p, rpm_count_t c)
1828 {
1829     indexEntry entry;
1830     rpm_data_t oldData;
1831     rpm_data_t data;
1832     int length;
1833
1834     /* First find the tag */
1835     entry = findEntry(h, tag, type);
1836     if (!entry)
1837         return 0;
1838
1839     length = 0;
1840     data = grabData(type, p, c, &length);
1841     if (data == NULL || length <= 0)
1842         return 0;
1843
1844     /* make sure entry points to the first occurence of this tag */
1845     while (entry > h->index && (entry - 1)->info.tag == tag)
1846         entry--;
1847
1848     /* free after we've grabbed the new data in case the two are intertwined;
1849        that's a bad idea but at least we won't break */
1850     oldData = entry->data;
1851
1852     entry->info.count = c;
1853     entry->info.type = type;
1854     entry->data = data;
1855     entry->length = length;
1856
1857     if (ENTRY_IN_REGION(entry)) {
1858         entry->info.offset = 0;
1859     } else
1860         oldData = _free(oldData);
1861
1862     return 1;
1863 }
1864
1865 /**
1866  * Header tag iterator data structure.
1867  */
1868 struct headerIterator_s {
1869     Header h;           /*!< Header being iterated. */
1870     int next_index;     /*!< Next tag index. */
1871 };
1872
1873 HeaderIterator headerFreeIterator(HeaderIterator hi)
1874 {
1875     if (hi != NULL) {
1876         hi->h = headerFree(hi->h);
1877         hi = _free(hi);
1878     }
1879     return hi;
1880 }
1881
1882 HeaderIterator headerInitIterator(Header h)
1883 {
1884     HeaderIterator hi = xmalloc(sizeof(*hi));
1885
1886     headerSort(h);
1887
1888     hi->h = headerLink(h);
1889     hi->next_index = 0;
1890     return hi;
1891 }
1892
1893 int headerNext(HeaderIterator hi, rpmtd td)
1894 {
1895     Header h = hi->h;
1896     int slot = hi->next_index;
1897     indexEntry entry = NULL;
1898     int rc;
1899
1900     assert(td != NULL);
1901     rpmtdReset(td);
1902
1903     for (slot = hi->next_index; slot < h->indexUsed; slot++) {
1904         entry = h->index + slot;
1905         if (!ENTRY_IS_REGION(entry))
1906             break;
1907     }
1908     hi->next_index = slot;
1909     if (entry == NULL || slot >= h->indexUsed)
1910         return 0;
1911
1912         /* LCL: no clue */
1913     hi->next_index++;
1914
1915     td->tag = entry->info.tag;
1916
1917     rc = copyTdEntry(entry, td, HEADERGET_DEFAULT);
1918
1919     /* XXX 1 on success */
1920     return ((rc == 1) ? 1 : 0);
1921 }
1922
1923 int headerNextIterator(HeaderIterator hi,
1924                 rpmTag * tag,
1925                 rpmTagType * type,
1926                 rpm_data_t * p,
1927                 rpm_count_t * c)
1928 {
1929     struct rpmtd_s td;
1930     int rc;
1931
1932     rc = headerNext(hi, &td);
1933     if (tag)
1934         *tag = td.tag;
1935     TDWRAP();
1936     return rc;
1937 }
1938
1939 /** \ingroup header
1940  * Duplicate a header.
1941  * @param h             header
1942  * @return              new header instance
1943  */
1944 Header headerCopy(Header h)
1945 {
1946     Header nh = headerNew();
1947     HeaderIterator hi;
1948     struct rpmtd_s td;
1949
1950     hi = headerInitIterator(h);
1951     while (headerNext(hi, &td)) {
1952         if (rpmtdCount(&td) > 0) {
1953             (void) headerPut(nh, &td, HEADERPUT_DEFAULT);
1954         }
1955         rpmtdFreeData(&td);
1956     }
1957     hi = headerFreeIterator(hi);
1958
1959     return headerReload(nh, HEADER_IMAGE);
1960 }
1961
1962 void headerCopyTags(Header headerFrom, Header headerTo,
1963                     const rpmTag * tagstocopy)
1964 {
1965     const rpmTag * p;
1966     struct rpmtd_s td;
1967
1968     if (headerFrom == headerTo)
1969         return;
1970
1971     for (p = tagstocopy; *p != 0; p++) {
1972         if (headerIsEntry(headerTo, *p))
1973             continue;
1974         if (!headerGet(headerFrom, *p, &td, HEADERGET_MINMEM))
1975             continue;
1976         (void) headerPut(headerTo, &td, HEADERPUT_DEFAULT);
1977         rpmtdFreeData(&td);
1978     }
1979 }
1980
1981 void * headerFreeData(rpm_data_t data, rpmTagType type)
1982 {
1983     if (data) {
1984         if (type == RPM_FORCEFREE_TYPE ||
1985             type == RPM_STRING_ARRAY_TYPE ||
1986             type == RPM_I18NSTRING_TYPE ||
1987             type == RPM_BIN_TYPE)
1988                 free(data); /* XXX _free() */
1989     }
1990     return NULL;
1991 }