5 /* RPM - Copyright (C) 1995-2002 Red Hat Software */
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. */
13 #include <rpm/rpmtypes.h>
14 #include <rpm/rpmstring.h>
15 #include "lib/header_internal.h"
23 const unsigned char rpm_header_magic[8] = {
24 0x8e, 0xad, 0xe8, 0x01, 0x00, 0x00, 0x00, 0x00
28 * Alignment needed for header data types.
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 */
50 * Size of header data types.
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 */
72 * Maximum no. of bytes permitted in a header.
74 static const size_t headerMaxbytes = (32*1024*1024);
77 * HEADER_EXT_TAG format function prototype.
78 * This is allowed to fail, which indicates the tag doesn't exist.
81 * @retval td tag data container
82 * @param flags modifier flags
83 * @return 0 on success
85 typedef int (*headerTagTagFunction) (Header h, rpmtd td, headerGetFlags hgflags);
87 extern void *rpmHeaderTagFunc(rpmTag tag);
89 Header headerLink(Header h)
91 if (h == NULL) return NULL;
95 fprintf(stderr, "--> h %p ++ %d at %s:%u\n", h, h->nrefs, __FILE__, __LINE__);
100 Header headerUnlink(Header h)
102 if (h == NULL) return NULL;
104 fprintf(stderr, "--> h %p -- %d at %s:%u\n", h, h->nrefs, __FILE__, __LINE__);
109 Header headerFree(Header h)
111 (void) headerUnlink(h);
113 if (h == NULL || h->nrefs > 0)
114 return NULL; /* XXX return previous header? */
117 indexEntry entry = h->index;
119 for (i = 0; i < h->indexUsed; i++, entry++) {
120 if ((h->flags & HEADERFLAG_ALLOCATED) && ENTRY_IS_REGION(entry)) {
121 if (entry->length > 0) {
122 int32_t * ei = entry->data;
123 if ((ei - 2) == h->blob) h->blob = _free(h->blob);
126 } else if (!ENTRY_IN_REGION(entry)) {
127 entry->data = _free(entry->data);
131 h->index = _free(h->index);
138 Header headerNew(void)
140 Header h = xcalloc(1, sizeof(*h));
143 h->indexAlloced = INDEX_MALLOC_SIZE;
146 h->flags |= HEADERFLAG_SORTED;
148 h->index = (h->indexAlloced
149 ? xcalloc(h->indexAlloced, sizeof(*h->index))
153 return headerLink(h);
156 int headerVerifyInfo(int il, int dl, const void * pev, void * iv, int negate)
158 entryInfo pe = (entryInfo) pev;
162 for (i = 0; i < il; i++) {
163 info->tag = ntohl(pe[i].tag);
164 info->type = ntohl(pe[i].type);
165 info->offset = ntohl(pe[i].offset);
167 info->offset = -info->offset;
168 info->count = ntohl(pe[i].count);
170 if (hdrchkType(info->type))
172 if (hdrchkAlign(info->type, info->offset))
174 if (!negate && hdrchkRange(dl, info->offset))
176 if (hdrchkData(info->count))
185 static int indexCmp(const void * avp, const void * bvp)
187 indexEntry ap = (indexEntry) avp, bp = (indexEntry) bvp;
188 return (ap->info.tag - bp->info.tag);
191 void headerSort(Header h)
193 if (!(h->flags & HEADERFLAG_SORTED)) {
194 qsort(h->index, h->indexUsed, sizeof(*h->index), indexCmp);
195 h->flags |= HEADERFLAG_SORTED;
201 static int offsetCmp(const void * avp, const void * bvp)
203 indexEntry ap = (indexEntry) avp, bp = (indexEntry) bvp;
204 int rc = (ap->info.offset - bp->info.offset);
207 /* Within a region, entries sort by address. Added drips sort by tag. */
208 if (ap->info.offset < 0)
209 rc = (((char *)ap->data) - ((char *)bp->data));
211 rc = (ap->info.tag - bp->info.tag);
217 * Restore tags in header to original ordering.
220 void headerUnsort(Header h)
222 qsort(h->index, h->indexUsed, sizeof(*h->index), offsetCmp);
225 unsigned headerSizeof(Header h, enum hMagic magicp)
228 unsigned int size = 0;
237 case HEADER_MAGIC_YES:
238 size += sizeof(rpm_header_magic);
240 case HEADER_MAGIC_NO:
244 size += 2 * sizeof(int32_t); /* count of index entries */
246 for (i = 0, entry = h->index; i < h->indexUsed; i++, entry++) {
249 /* Regions go in as is ... */
250 if (ENTRY_IS_REGION(entry)) {
251 size += entry->length;
252 /* XXX Legacy regions do not include the region tag and data. */
253 if (i == 0 && (h->flags & HEADERFLAG_LEGACY))
254 size += sizeof(struct entryInfo_s) + entry->info.count;
258 /* ... and region elements are skipped. */
259 if (entry->info.offset < 0)
263 type = entry->info.type;
264 if (typeSizes[type] > 1) {
265 unsigned diff = typeSizes[type] - (size % typeSizes[type]);
266 if (diff != typeSizes[type]) {
271 size += sizeof(struct entryInfo_s) + entry->length;
278 * Return length of entry data.
279 * @param type entry data type
280 * @param p entry data
281 * @param count entry item count
282 * @param onDisk data is concatenated strings (with NUL's))?
283 * @param pend pointer to end of data (or NULL)
284 * @return no. bytes in data, -1 on failure
286 static int dataLength(rpmTagType type, rpm_constdata_t p, rpm_count_t count,
287 int onDisk, rpm_constdata_t pend)
289 const unsigned char * s = p;
290 const unsigned char * se = pend;
294 case RPM_STRING_TYPE:
302 length++; /* count nul terminator too. */
305 case RPM_STRING_ARRAY_TYPE:
306 case RPM_I18NSTRING_TYPE:
307 /* These are like RPM_STRING_TYPE, except they're *always* an array */
308 /* Compute sum of length of all strings, including nul terminators */
312 length++; /* count nul terminator too */
320 const char ** av = (const char **)p;
322 /* add one for null termination */
323 length += strlen(*av++) + 1;
329 if (typeSizes[type] == -1)
331 length = typeSizes[(type & 0xf)] * count;
332 if (length < 0 || (se && (s + length) > se))
341 * Swap int32_t and int16_t arrays within header region.
343 * If a header region tag is in the set to be swabbed, as the data for a
344 * a header region is located after all other tag data.
346 * @param entry header entry
347 * @param il no. of entries
348 * @param dl start no. bytes of data
349 * @param pe header physical entry pointer (swapped)
350 * @param dataStart header data start
351 * @param dataEnd header data end
352 * @param regionid region offset
353 * @return no. bytes of data in region, -1 on error
355 static int regionSwab(indexEntry entry, int il, int dl,
357 unsigned char * dataStart,
358 const unsigned char * dataEnd,
361 for (; il > 0; il--, pe++) {
362 struct indexEntry_s ie;
365 ie.info.tag = ntohl(pe->tag);
366 ie.info.type = ntohl(pe->type);
367 ie.info.count = ntohl(pe->count);
368 ie.info.offset = ntohl(pe->offset);
370 if (hdrchkType(ie.info.type))
372 if (hdrchkData(ie.info.count))
374 if (hdrchkData(ie.info.offset))
376 if (hdrchkAlign(ie.info.type, ie.info.offset))
379 ie.data = dataStart + ie.info.offset;
380 if (dataEnd && (unsigned char *)ie.data >= dataEnd)
383 ie.length = dataLength(ie.info.type, ie.data, ie.info.count, 1, dataEnd);
384 if (ie.length < 0 || hdrchkData(ie.length))
390 ie.info.offset = regionid;
391 *entry = ie; /* structure assignment */
397 if (typeSizes[type] > 1) {
398 unsigned diff = typeSizes[type] - (dl % typeSizes[type]);
399 if (diff != typeSizes[type]) {
404 /* Perform endian conversions */
405 switch (ntohl(pe->type)) {
407 { uint64_t * it = ie.data;
408 for (; ie.info.count > 0; ie.info.count--, it += 1) {
409 if (dataEnd && ((unsigned char *)it) >= dataEnd)
415 { int32_t * it = ie.data;
416 for (; ie.info.count > 0; ie.info.count--, it += 1) {
417 if (dataEnd && ((unsigned char *)it) >= dataEnd)
423 { int16_t * it = ie.data;
424 for (; ie.info.count > 0; ie.info.count--, it += 1) {
425 if (dataEnd && ((unsigned char *)it) >= dataEnd)
441 * @retval *lengthPtr no. bytes in unloaded header blob
442 * @return unloaded header blob (NULL on error)
444 static void * doHeaderUnload(Header h,
457 int drlen, ndribbles;
459 if (h == NULL) return NULL;
461 /* Sort entries by (offset,tag). */
464 /* Compute (il,dl) for all tags, including those deleted in region. */
465 drlen = ndribbles = 0;
466 for (i = 0, entry = h->index; i < h->indexUsed; i++, entry++) {
467 if (ENTRY_IS_REGION(entry)) {
468 int32_t rdl = -entry->info.offset; /* negative offset */
469 int32_t ril = rdl/sizeof(*pe);
470 int rid = entry->info.offset;
473 dl += entry->rdlen + entry->info.count;
474 /* XXX Legacy regions do not include the region tag and data. */
475 if (i == 0 && (h->flags & HEADERFLAG_LEGACY))
478 /* Skip rest of entries in region, but account for dribbles. */
479 for (; i < h->indexUsed && entry->info.offset <= rid+1; i++, entry++) {
480 if (entry->info.offset <= rid)
484 type = entry->info.type;
485 if (typeSizes[type] > 1) {
486 unsigned diff = typeSizes[type] - (dl % typeSizes[type]);
487 if (diff != typeSizes[type]) {
495 drlen += entry->length;
503 /* Ignore deleted drips. */
504 if (entry->data == NULL || entry->length <= 0)
508 type = entry->info.type;
509 if (typeSizes[type] > 1) {
510 unsigned diff = typeSizes[type] - (dl % typeSizes[type]);
511 if (diff != typeSizes[type]) {
520 /* Sanity checks on header intro. */
521 if (hdrchkTags(il) || hdrchkData(dl))
524 len = sizeof(il) + sizeof(dl) + (il * sizeof(*pe)) + dl;
530 pe = (entryInfo) &ei[2];
531 dataStart = te = (char *) (pe + il);
533 for (i = 0, entry = h->index; i < h->indexUsed; i++, entry++) {
539 if (entry->data == NULL || entry->length <= 0)
542 t = (unsigned char*)te;
543 pe->tag = htonl(entry->info.tag);
544 pe->type = htonl(entry->info.type);
545 pe->count = htonl(entry->info.count);
547 if (ENTRY_IS_REGION(entry)) {
548 int32_t rdl = -entry->info.offset; /* negative offset */
549 int32_t ril = rdl/sizeof(*pe) + ndribbles;
550 int rid = entry->info.offset;
552 src = (char *)entry->data;
553 rdlen = entry->rdlen;
555 /* XXX Legacy regions do not include the region tag and data. */
556 if (i == 0 && (h->flags & HEADERFLAG_LEGACY)) {
559 memcpy(pe+1, src, rdl);
560 memcpy(te, src + rdl, rdlen);
563 pe->offset = htonl(te - dataStart);
566 stei[2] = htonl(-rdl-entry->info.count);
568 memcpy(te, stei, entry->info.count);
569 te += entry->info.count;
571 rdlen += entry->info.count;
573 count = regionSwab(NULL, ril, 0, pe, t, NULL, 0);
579 memcpy(pe+1, src + sizeof(*pe), ((ril-1) * sizeof(*pe)));
580 memcpy(te, src + (ril * sizeof(*pe)), rdlen+entry->info.count+drlen);
583 entryInfo se = (entryInfo)src;
584 int off = ntohl(se->offset);
585 pe->offset = (off) ? htonl(te - dataStart) : htonl(off);
587 te += entry->info.count + drlen;
589 count = regionSwab(NULL, ril, 0, pe, t, NULL, 0);
590 if (count != (rdlen + entry->info.count + drlen))
594 /* Skip rest of entries in region. */
595 while (i < h->indexUsed && entry->info.offset <= rid+1) {
605 /* Ignore deleted drips. */
606 if (entry->data == NULL || entry->length <= 0)
610 type = entry->info.type;
611 if (typeSizes[type] > 1) {
613 diff = typeSizes[type] - ((te - dataStart) % typeSizes[type]);
614 if (diff != typeSizes[type]) {
620 pe->offset = htonl(te - dataStart);
622 /* copy data w/ endian conversions */
623 switch (entry->info.type) {
625 count = entry->info.count;
628 *((uint64_t *)te) = htonll(*((uint64_t *)src));
629 te += sizeof(uint64_t);
630 src += sizeof(uint64_t);
635 count = entry->info.count;
638 *((int32_t *)te) = htonl(*((int32_t *)src));
639 te += sizeof(int32_t);
640 src += sizeof(int32_t);
645 count = entry->info.count;
648 *((int16_t *)te) = htons(*((int16_t *)src));
649 te += sizeof(int16_t);
650 src += sizeof(int16_t);
655 memcpy(te, entry->data, entry->length);
662 /* Insure that there are no memcpy underruns/overruns. */
663 if (((char *)pe) != dataStart)
665 if ((((char *)ei)+len) != te)
671 h->flags &= ~HEADERFLAG_SORTED;
681 void * headerUnload(Header h)
684 void * uh = doHeaderUnload(h, &length);
689 * Find matching (tag,type) entry in header.
691 * @param tag entry tag
692 * @param type entry type
693 * @return header entry
696 indexEntry findEntry(Header h, rpmTag tag, rpmTagType type)
699 struct indexEntry_s key;
701 if (h == NULL) return NULL;
702 if (!(h->flags & HEADERFLAG_SORTED)) headerSort(h);
706 entry = bsearch(&key, h->index, h->indexUsed, sizeof(*h->index), indexCmp);
710 if (type == RPM_NULL_TYPE)
714 while (entry->info.tag == tag && entry->info.type != type &&
715 entry > h->index) entry--;
717 if (entry->info.tag == tag && entry->info.type == type)
723 int headerDel(Header h, rpmTag tag)
725 indexEntry last = h->index + h->indexUsed;
726 indexEntry entry, first;
729 entry = findEntry(h, tag, RPM_NULL_TYPE);
730 if (!entry) return 1;
732 /* Make sure entry points to the first occurence of this tag. */
733 while (entry > h->index && (entry - 1)->info.tag == tag)
736 /* Free data for tags being removed. */
737 for (first = entry; first < last; first++) {
739 if (first->info.tag != tag)
744 if (ENTRY_IN_REGION(first))
749 ne = (first - entry);
754 memmove(entry, first, (ne * sizeof(*entry)));
760 Header headerLoad(void * uh)
762 int32_t * ei = (int32_t *) uh;
763 int32_t il = ntohl(ei[0]); /* index length */
764 int32_t dl = ntohl(ei[1]); /* data length */
765 size_t pvlen = sizeof(il) + sizeof(dl) +
766 (il * sizeof(struct entryInfo_s)) + dl;
770 unsigned char * dataStart;
771 unsigned char * dataEnd;
775 /* Sanity checks on header intro. */
776 if (hdrchkTags(il) || hdrchkData(dl))
780 pe = (entryInfo) &ei[2];
781 dataStart = (unsigned char *) (pe + il);
782 dataEnd = dataStart + dl;
784 h = xcalloc(1, sizeof(*h));
786 h->indexAlloced = il + 1;
789 h->index = xcalloc(h->indexAlloced, sizeof(*h->index));
790 h->flags |= HEADERFLAG_SORTED;
795 if (!(htonl(pe->tag) < HEADER_I18NTABLE)) {
796 h->flags |= HEADERFLAG_LEGACY;
797 entry->info.type = REGION_TAG_TYPE;
798 entry->info.tag = HEADER_IMAGE;
799 entry->info.count = REGION_TAG_COUNT;
800 entry->info.offset = ((unsigned char *)pe - dataStart); /* negative offset */
803 entry->length = pvlen - sizeof(il) - sizeof(dl);
804 rdlen = regionSwab(entry+1, il, 0, pe, dataStart, dataEnd, entry->info.offset);
807 entry->rdlen = rdlen;
813 h->flags &= ~HEADERFLAG_LEGACY;
815 entry->info.type = htonl(pe->type);
816 entry->info.count = htonl(pe->count);
818 if (hdrchkType(entry->info.type))
820 if (hdrchkTags(entry->info.count))
823 { int off = ntohl(pe->offset);
828 size_t nb = REGION_TAG_COUNT;
830 /* XXX Hmm, why the copy? */
831 memcpy(&stei, dataStart + off, nb);
832 rdl = -ntohl(stei[2]); /* negative offset */
833 ril = rdl/sizeof(*pe);
834 if (hdrchkTags(ril) || hdrchkData(rdl))
836 entry->info.tag = htonl(pe->tag);
839 rdl = (ril * sizeof(struct entryInfo_s));
840 entry->info.tag = HEADER_IMAGE;
843 entry->info.offset = -rdl; /* negative offset */
846 entry->length = pvlen - sizeof(il) - sizeof(dl);
847 rdlen = regionSwab(entry+1, ril-1, 0, pe+1, dataStart, dataEnd, entry->info.offset);
850 entry->rdlen = rdlen;
852 if (ril < h->indexUsed) {
853 indexEntry newEntry = entry + ril;
854 int ne = (h->indexUsed - ril);
855 int rid = entry->info.offset+1;
858 /* Load dribble entries from region. */
859 rc = regionSwab(newEntry, ne, 0, pe+ril, dataStart, dataEnd, rid);
864 { indexEntry firstEntry = newEntry;
865 int save = h->indexUsed;
868 /* Dribble entries replace duplicate region entries. */
870 for (j = 0; j < ne; j++, newEntry++) {
871 (void) headerDel(h, newEntry->info.tag);
872 if (newEntry->info.tag == RPMTAG_BASENAMES)
873 (void) headerDel(h, RPMTAG_OLDFILENAMES);
876 /* If any duplicate entries were replaced, move new entries down. */
877 if (h->indexUsed < (save - ne)) {
878 memmove(h->index + h->indexUsed, firstEntry,
879 (ne * sizeof(*entry)));
886 h->flags &= ~HEADERFLAG_SORTED;
888 h->flags |= HEADERFLAG_ALLOCATED;
894 h->index = _free(h->index);
900 Header headerReload(Header h, rpmTag tag)
904 void * uh = doHeaderUnload(h, &length);
914 if (ENTRY_IS_REGION(nh->index)) {
915 if (tag == HEADER_SIGNATURES || tag == HEADER_IMMUTABLE)
916 nh->index[0].info.tag = tag;
921 Header headerCopyLoad(const void * uh)
923 int32_t * ei = (int32_t *) uh;
924 int32_t il = ntohl(ei[0]); /* index length */
925 int32_t dl = ntohl(ei[1]); /* data length */
926 size_t pvlen = sizeof(il) + sizeof(dl) +
927 (il * sizeof(struct entryInfo_s)) + dl;
931 /* Sanity checks on header intro. */
932 if (!(hdrchkTags(il) || hdrchkData(dl)) && pvlen < headerMaxbytes) {
933 nuh = memcpy(xmalloc(pvlen), uh, pvlen);
934 if ((h = headerLoad(nuh)) == NULL)
941 * Read (and load) header from file handle.
942 * @param fd file handle
943 * @param magicp read (and verify) 8 bytes of (magic, 0)?
944 * @return header (or NULL on error)
946 Header headerRead(FD_t fd, enum hMagic magicp)
958 memset(block, 0, sizeof(block));
960 if (magicp == HEADER_MAGIC_YES)
964 if (timedRead(fd, (char *)block, i*sizeof(*block)) != (i * sizeof(*block)))
969 if (magicp == HEADER_MAGIC_YES) {
971 if (memcmp(&magic, rpm_header_magic, sizeof(magic)))
973 reserved = block[i++];
976 il = ntohl(block[i]); i++;
977 dl = ntohl(block[i]); i++;
979 len = sizeof(il) + sizeof(dl) + (il * sizeof(struct entryInfo_s)) + dl;
981 /* Sanity checks on header intro. */
982 if (hdrchkTags(il) || hdrchkData(dl) || len > headerMaxbytes)
988 len -= sizeof(il) + sizeof(dl);
991 if (timedRead(fd, (char *)&ei[2], len) != len)
997 if (h == NULL && ei != NULL) {
1003 int headerWrite(FD_t fd, Header h, enum hMagic magicp)
1009 uh = doHeaderUnload(h, &length);
1013 case HEADER_MAGIC_YES:
1014 nb = Fwrite(rpm_header_magic, sizeof(uint8_t), sizeof(rpm_header_magic), fd);
1015 if (nb != sizeof(rpm_header_magic))
1018 case HEADER_MAGIC_NO:
1022 nb = Fwrite(uh, sizeof(char), length, fd);
1026 return (nb == length ? 0 : 1);
1029 int headerIsEntry(Header h, rpmTag tag)
1031 /* FIX: h modified by sort. */
1032 return (findEntry(h, tag, RPM_NULL_TYPE) ? 1 : 0);
1037 * Retrieve data from header entry.
1038 * Relevant flags (others are ignored), if neither is set allocation
1039 * behavior depends on data type(!)
1040 * HEADERGET_MINMEM: return pointers to header memory
1041 * HEADERGET_ALLOC: always return malloced memory, overrides MINMEM
1043 * @todo Permit retrieval of regions other than HEADER_IMUTABLE.
1044 * @param entry header entry
1045 * @param td tag data container
1046 * @param minMem string pointers refer to header memory?
1047 * @param flags flags to control memory allocation
1048 * @return 1 on success, otherwise error.
1050 static int copyTdEntry(const indexEntry entry, rpmtd td, headerGetFlags flags)
1052 rpm_count_t count = entry->info.count;
1053 int rc = 1; /* XXX 1 on success. */
1054 /* ALLOC overrides MINMEM */
1055 int allocMem = flags & HEADERGET_ALLOC;
1056 int minMem = allocMem ? 0 : flags & HEADERGET_MINMEM;
1057 int argvArray = (flags & HEADERGET_ARGV) ? 1 : 0;
1060 td->flags = RPMTD_IMMUTABLE;
1061 switch (entry->info.type) {
1064 * XXX This only works for
1065 * XXX "sealed" HEADER_IMMUTABLE/HEADER_SIGNATURES/HEADER_IMAGE.
1066 * XXX This will *not* work for unsealed legacy HEADER_IMAGE (i.e.
1067 * XXX a legacy header freshly read, but not yet unloaded to the rpmdb).
1069 if (ENTRY_IS_REGION(entry)) {
1070 int32_t * ei = ((int32_t *)entry->data) - 2;
1071 entryInfo pe = (entryInfo) (ei + 2);
1072 unsigned char * dataStart = (unsigned char *) (pe + ntohl(ei[0]));
1073 int32_t rdl = -entry->info.offset; /* negative offset */
1074 int32_t ril = rdl/sizeof(*pe);
1077 count = 2 * sizeof(*ei) + (ril * sizeof(*pe)) + rdl;
1078 if (entry->info.tag == HEADER_IMAGE) {
1082 count += REGION_TAG_COUNT;
1083 rdl += REGION_TAG_COUNT;
1086 td->data = xmalloc(count);
1087 ei = (int32_t *) td->data;
1091 pe = (entryInfo) memcpy(ei + 2, pe, (ril * sizeof(*pe)));
1093 dataStart = (unsigned char *) memcpy(pe + ril, dataStart, rdl);
1095 rc = regionSwab(NULL, ril, 0, pe, dataStart, dataStart + rdl, 0);
1096 /* don't return data on failure */
1098 td->data = _free(td->data);
1100 /* XXX 1 on success. */
1101 rc = (rc < 0) ? 0 : 1;
1103 count = entry->length;
1105 ? memcpy(xmalloc(count), entry->data, count)
1109 case RPM_STRING_TYPE:
1111 td->data = allocMem ? xstrdup(entry->data) : entry->data;
1114 case RPM_STRING_ARRAY_TYPE:
1115 case RPM_I18NSTRING_TYPE:
1116 { const char ** ptrEntry;
1117 int tableSize = (count + argvArray) * sizeof(char *);
1122 td->data = xmalloc(tableSize);
1123 ptrEntry = (const char **) td->data;
1126 t = xmalloc(tableSize + entry->length);
1127 td->data = (void *)t;
1128 ptrEntry = (const char **) td->data;
1130 memcpy(t, entry->data, entry->length);
1132 for (i = 0; i < count; i++) {
1139 td->flags |= RPMTD_ARGV;
1144 case RPM_INT16_TYPE:
1145 case RPM_INT32_TYPE:
1146 case RPM_INT64_TYPE:
1148 td->data = xmalloc(entry->length);
1149 memcpy(td->data, entry->data, entry->length);
1151 td->data = entry->data;
1155 /* WTH? Don't mess with unknown data types... */
1160 td->type = entry->info.type;
1163 if (td->data && entry->data != td->data) {
1164 td->flags |= RPMTD_ALLOCED;
1171 * Does locale match entry in header i18n table?
1174 * The range [l,le) contains the next locale to match:
1175 * ll[_CC][.EEEEE][@dddd]
1177 * ll ISO language code (in lowercase).
1178 * CC (optional) ISO coutnry code (in uppercase).
1179 * EEEEE (optional) encoding (not really standardized).
1180 * dddd (optional) dialect.
1183 * @param td header i18n table data, NUL terminated
1184 * @param l start of locale to match
1185 * @param le end of locale to match
1186 * @return 1 on good match, 2 on weak match, 0 on no match
1188 static int headerMatchLocale(const char *td, const char *l, const char *le)
1192 /* First try a complete match. */
1193 if (strlen(td) == (le-l) && rstreqn(td, l, (le - l)))
1196 /* Next, try stripping optional dialect and matching. */
1197 for (fe = l; fe < le && *fe != '@'; fe++)
1199 if (fe < le && rstreqn(td, l, (fe - l)))
1202 /* Next, try stripping optional codeset and matching. */
1203 for (fe = l; fe < le && *fe != '.'; fe++)
1205 if (fe < le && rstreqn(td, l, (fe - l)))
1208 /* Finally, try stripping optional country code and matching. */
1209 for (fe = l; fe < le && *fe != '_'; fe++)
1211 if (fe < le && rstreqn(td, l, (fe - l)))
1218 * Return i18n string from header that matches locale.
1220 * @param entry i18n string data
1221 * @retval td tag data container
1222 * @param flags flags to control allocation
1225 static int copyI18NEntry(Header h, indexEntry entry, rpmtd td,
1226 headerGetFlags flags)
1228 const char *lang, *l, *le;
1231 td->type = RPM_STRING_TYPE;
1233 /* if no match, just return the first string */
1234 td->data = entry->data;
1236 /* XXX Drepper sez' this is the order. */
1237 if ((lang = getenv("LANGUAGE")) == NULL &&
1238 (lang = getenv("LC_ALL")) == NULL &&
1239 (lang = getenv("LC_MESSAGES")) == NULL &&
1240 (lang = getenv("LANG")) == NULL)
1243 if ((table = findEntry(h, HEADER_I18NTABLE, RPM_STRING_ARRAY_TYPE)) == NULL)
1246 for (l = lang; *l != '\0'; l = le) {
1248 char *ed, *ed_weak = NULL;
1251 while (*l && *l == ':') /* skip leading colons */
1255 for (le = l; *le && *le != ':'; le++) /* find end of this locale */
1258 /* For each entry in the header ... */
1259 for (langNum = 0, t = table->data, ed = entry->data;
1260 langNum < entry->info.count;
1261 langNum++, t += strlen(t) + 1, ed += strlen(ed) + 1) {
1263 int match = headerMatchLocale(t, l, le);
1267 } else if (match == 2) {
1278 if (flags & HEADERGET_ALLOC) {
1279 td->data = xstrdup(td->data);
1280 td->flags |= RPMTD_ALLOCED;
1287 * Retrieve tag data from header.
1289 * @retval td tag data container
1290 * @param flags flags to control retrieval
1291 * @return 1 on success, 0 on not found
1293 static int intGetTdEntry(Header h, rpmtd td, headerGetFlags flags)
1298 /* First find the tag */
1299 /* FIX: h modified by sort. */
1300 entry = findEntry(h, td->tag, RPM_NULL_TYPE);
1301 if (entry == NULL) {
1302 /* Td is zeroed above, just return... */
1306 if (flags & HEADERGET_RAW) {
1307 rc = copyTdEntry(entry, td, flags);
1309 switch (entry->info.type) {
1310 case RPM_I18NSTRING_TYPE:
1311 rc = copyI18NEntry(h, entry, td, flags);
1314 rc = copyTdEntry(entry, td, flags);
1319 /* XXX 1 on success */
1320 return ((rc == 1) ? 1 : 0);
1323 int headerGet(Header h, rpmTag tag, rpmtd td, headerGetFlags flags)
1326 headerTagTagFunction tagfunc = intGetTdEntry;
1328 if (td == NULL) return 0;
1333 if (flags & HEADERGET_EXT) {
1334 headerTagTagFunction extfunc = rpmHeaderTagFunc(tag);
1335 if (extfunc) tagfunc = extfunc;
1337 rc = tagfunc(h, td, flags);
1339 assert(tag == td->tag);
1345 static void copyData(rpmTagType type, rpm_data_t dstPtr,
1346 rpm_constdata_t srcPtr, rpm_count_t cnt, int dataLength)
1349 case RPM_STRING_ARRAY_TYPE:
1350 case RPM_I18NSTRING_TYPE:
1351 { const char ** av = (const char **) srcPtr;
1354 while (cnt-- > 0 && dataLength > 0) {
1356 if ((s = *av++) == NULL)
1360 } while (s[-1] && --dataLength > 0);
1365 memmove(dstPtr, srcPtr, dataLength);
1371 * Return (malloc'ed) copy of entry data.
1372 * @param type entry data type
1373 * @param p entry data
1374 * @param c entry item count
1375 * @retval lengthPtr no. bytes in returned data
1376 * @return (malloc'ed) copy of entry data, NULL on error
1379 grabData(rpmTagType type, rpm_constdata_t p, rpm_count_t c, int * lengthPtr)
1381 rpm_data_t data = NULL;
1384 length = dataLength(type, p, c, 0, NULL);
1386 data = xmalloc(length);
1387 copyData(type, data, p, c, length);
1391 *lengthPtr = length;
1395 static int intAddEntry(Header h, rpmtd td)
1401 /* Count must always be >= 1 for headerAddEntry. */
1405 if (hdrchkType(td->type))
1407 if (hdrchkData(td->count))
1411 data = grabData(td->type, td->data, td->count, &length);
1412 if (data == NULL || length <= 0)
1415 /* Allocate more index space if necessary */
1416 if (h->indexUsed == h->indexAlloced) {
1417 h->indexAlloced += INDEX_MALLOC_SIZE;
1418 h->index = xrealloc(h->index, h->indexAlloced * sizeof(*h->index));
1421 /* Fill in the index */
1422 entry = h->index + h->indexUsed;
1423 entry->info.tag = td->tag;
1424 entry->info.type = td->type;
1425 entry->info.count = td->count;
1426 entry->info.offset = 0;
1428 entry->length = length;
1430 if (h->indexUsed > 0 && td->tag < h->index[h->indexUsed-1].info.tag)
1431 h->flags &= ~HEADERFLAG_SORTED;
1437 static int intAppendEntry(Header h, rpmtd td)
1442 if (td->type == RPM_STRING_TYPE || td->type == RPM_I18NSTRING_TYPE) {
1443 /* we can't do this */
1447 /* Find the tag entry in the header. */
1448 entry = findEntry(h, td->tag, td->type);
1452 length = dataLength(td->type, td->data, td->count, 0, NULL);
1456 if (ENTRY_IN_REGION(entry)) {
1457 char * t = xmalloc(entry->length + length);
1458 memcpy(t, entry->data, entry->length);
1460 entry->info.offset = 0;
1462 entry->data = xrealloc(entry->data, entry->length + length);
1464 copyData(td->type, ((char *) entry->data) + entry->length,
1465 td->data, td->count, length);
1467 entry->length += length;
1469 entry->info.count += td->count;
1474 int headerPut(Header h, rpmtd td, headerPutFlags flags)
1479 if (flags & HEADERPUT_APPEND) {
1480 rc = findEntry(h, td->tag, td->type) ?
1481 intAppendEntry(h, td) :
1484 rc = intAddEntry(h, td);
1490 * Sanity check data types against tag table before putting. Assume
1491 * append on all array-types.
1493 static int headerPutType(Header h, rpmTag tag, rpmTagType reqtype,
1494 rpm_constdata_t data, rpm_count_t size)
1497 rpmTagType type = rpmTagGetType(tag);
1498 headerPutFlags flags = HEADERPUT_APPEND;
1501 /* Basic sanity checks: type must match and there must be data to put */
1502 if ((type & RPM_MASK_TYPE) != reqtype
1503 || size < 1 || data == NULL || h == NULL) {
1508 * Non-array types can't be appended to. Binary types use size
1509 * for data length, for other non-array types size must be 1.
1511 if ((type & RPM_MASK_RETURN_TYPE) != RPM_ARRAY_RETURN_TYPE) {
1512 flags = HEADERPUT_DEFAULT;
1513 if ((type & RPM_MASK_TYPE) != RPM_BIN_TYPE && size != 1) {
1521 td.type = type & RPM_MASK_TYPE;
1522 td.data = (void *) data;
1525 valid = headerPut(h, &td, flags);
1531 int headerPutString(Header h, rpmTag tag, const char *val)
1533 rpmTagType type = rpmTagGetType(tag) & RPM_MASK_TYPE;
1534 const void *sptr = NULL;
1536 /* string arrays expect char **, arrange that */
1537 if (type == RPM_STRING_ARRAY_TYPE || type == RPM_I18NSTRING_TYPE) {
1539 } else if (type == RPM_STRING_TYPE) {
1545 return headerPutType(h, tag, type, sptr, 1);
1548 int headerPutStringArray(Header h, rpmTag tag, const char **array, rpm_count_t size)
1550 return headerPutType(h, tag, RPM_STRING_ARRAY_TYPE, array, size);
1553 int headerPutChar(Header h, rpmTag tag, char *val, rpm_count_t size)
1555 return headerPutType(h, tag, RPM_CHAR_TYPE, val, size);
1558 int headerPutUint8(Header h, rpmTag tag, uint8_t *val, rpm_count_t size)
1560 return headerPutType(h, tag, RPM_INT8_TYPE, val, size);
1563 int headerPutUint16(Header h, rpmTag tag, uint16_t *val, rpm_count_t size)
1565 return headerPutType(h, tag, RPM_INT16_TYPE, val, size);
1568 int headerPutUint32(Header h, rpmTag tag, uint32_t *val, rpm_count_t size)
1570 return headerPutType(h, tag, RPM_INT32_TYPE, val, size);
1573 int headerPutUint64(Header h, rpmTag tag, uint64_t *val, rpm_count_t size)
1575 return headerPutType(h, tag, RPM_INT64_TYPE, val, size);
1578 int headerPutBin(Header h, rpmTag tag, uint8_t *val, rpm_count_t size)
1580 return headerPutType(h, tag, RPM_BIN_TYPE, val, size);
1583 int headerAddI18NString(Header h, rpmTag tag, const char * string,
1586 indexEntry table, entry;
1587 const char ** strArray;
1590 rpm_count_t i, langNum;
1593 table = findEntry(h, HEADER_I18NTABLE, RPM_STRING_ARRAY_TYPE);
1594 entry = findEntry(h, tag, RPM_I18NSTRING_TYPE);
1596 if (!table && entry)
1597 return 0; /* this shouldn't ever happen!! */
1599 if (!table && !entry) {
1600 const char * charArray[2];
1601 rpm_count_t count = 0;
1603 if (!lang || (lang[0] == 'C' && lang[1] == '\0')) {
1604 charArray[count++] = "C";
1606 charArray[count++] = "C";
1607 charArray[count++] = lang;
1611 td.tag = HEADER_I18NTABLE;
1612 td.type = RPM_STRING_ARRAY_TYPE;
1613 td.data = (void *) charArray;
1615 if (!headerPut(h, &td, HEADERPUT_DEFAULT))
1617 table = findEntry(h, HEADER_I18NTABLE, RPM_STRING_ARRAY_TYPE);
1622 if (!lang) lang = "C";
1624 { const char * l = table->data;
1625 for (langNum = 0; langNum < table->info.count; langNum++) {
1626 if (rstreq(l, lang)) break;
1631 if (langNum >= table->info.count) {
1632 length = strlen(lang) + 1;
1633 if (ENTRY_IN_REGION(table)) {
1634 char * t = xmalloc(table->length + length);
1635 memcpy(t, table->data, table->length);
1637 table->info.offset = 0;
1639 table->data = xrealloc(table->data, table->length + length);
1640 memmove(((char *)table->data) + table->length, lang, length);
1641 table->length += length;
1642 table->info.count++;
1648 strArray = xmalloc(sizeof(*strArray) * (langNum + 1));
1649 for (i = 0; i < langNum; i++)
1651 strArray[langNum] = string;
1655 td.type = RPM_I18NSTRING_TYPE;
1657 td.count = langNum + 1;
1658 rc = headerPut(h, &td, HEADERPUT_DEFAULT);
1661 } else if (langNum >= entry->info.count) {
1662 ghosts = langNum - entry->info.count;
1664 length = strlen(string) + 1 + ghosts;
1665 if (ENTRY_IN_REGION(entry)) {
1666 char * t = xmalloc(entry->length + length);
1667 memcpy(t, entry->data, entry->length);
1669 entry->info.offset = 0;
1671 entry->data = xrealloc(entry->data, entry->length + length);
1673 memset(((char *)entry->data) + entry->length, '\0', ghosts);
1674 memmove(((char *)entry->data) + entry->length + ghosts, string, strlen(string)+1);
1676 entry->length += length;
1677 entry->info.count = langNum + 1;
1679 char *b, *be, *e, *ee, *t;
1682 /* Set beginning/end pointers to previous data */
1683 b = be = e = ee = entry->data;
1684 for (i = 0; i < table->info.count; i++) {
1687 ee += strlen(ee) + 1;
1692 /* Get storage for new buffer */
1694 sn = strlen(string) + 1;
1696 length = bn + sn + en;
1697 t = buf = xmalloc(length);
1699 /* Copy values into new storage */
1702 memcpy(t, string, sn);
1707 /* Replace i18N string array */
1708 entry->length -= strlen(be) + 1;
1709 entry->length += sn;
1711 if (ENTRY_IN_REGION(entry)) {
1712 entry->info.offset = 0;
1714 entry->data = _free(entry->data);
1721 int headerMod(Header h, rpmtd td)
1728 /* First find the tag */
1729 entry = findEntry(h, td->tag, td->type);
1734 data = grabData(td->type, td->data, td->count, &length);
1735 if (data == NULL || length <= 0)
1738 /* make sure entry points to the first occurence of this tag */
1739 while (entry > h->index && (entry - 1)->info.tag == td->tag)
1742 /* free after we've grabbed the new data in case the two are intertwined;
1743 that's a bad idea but at least we won't break */
1744 oldData = entry->data;
1746 entry->info.count = td->count;
1747 entry->info.type = td->type;
1749 entry->length = length;
1751 if (ENTRY_IN_REGION(entry)) {
1752 entry->info.offset = 0;
1754 oldData = _free(oldData);
1760 * Header tag iterator data structure.
1762 struct headerIterator_s {
1763 Header h; /*!< Header being iterated. */
1764 int next_index; /*!< Next tag index. */
1767 HeaderIterator headerFreeIterator(HeaderIterator hi)
1770 hi->h = headerFree(hi->h);
1776 HeaderIterator headerInitIterator(Header h)
1778 HeaderIterator hi = xmalloc(sizeof(*hi));
1782 hi->h = headerLink(h);
1787 int headerNext(HeaderIterator hi, rpmtd td)
1791 indexEntry entry = NULL;
1797 for (slot = hi->next_index; slot < h->indexUsed; slot++) {
1798 entry = h->index + slot;
1799 if (!ENTRY_IS_REGION(entry))
1802 hi->next_index = slot;
1803 if (entry == NULL || slot >= h->indexUsed)
1808 td->tag = entry->info.tag;
1810 rc = copyTdEntry(entry, td, HEADERGET_DEFAULT);
1812 /* XXX 1 on success */
1813 return ((rc == 1) ? 1 : 0);
1817 * Duplicate a header.
1819 * @return new header instance
1821 Header headerCopy(Header h)
1823 Header nh = headerNew();
1827 hi = headerInitIterator(h);
1828 while (headerNext(hi, &td)) {
1829 if (rpmtdCount(&td) > 0) {
1830 (void) headerPut(nh, &td, HEADERPUT_DEFAULT);
1834 hi = headerFreeIterator(hi);
1836 return headerReload(nh, HEADER_IMAGE);
1839 void headerCopyTags(Header headerFrom, Header headerTo,
1840 const rpmTag * tagstocopy)
1845 if (headerFrom == headerTo)
1848 for (p = tagstocopy; *p != 0; p++) {
1849 if (headerIsEntry(headerTo, *p))
1851 if (!headerGet(headerFrom, *p, &td, HEADERGET_MINMEM))
1853 (void) headerPut(headerTo, &td, HEADERPUT_DEFAULT);
1858 unsigned int headerGetInstance(Header h)
1860 return h ? h->instance : 0;
1863 void headerSetInstance(Header h, unsigned int instance)
1865 h->instance = instance;
1868 char * headerGetAsString(Header h, rpmTag tag)
1873 if (headerGet(h, tag, &td, HEADERGET_EXT)) {
1874 if (rpmtdCount(&td) == 1) {
1875 res = rpmtdFormat(&td, RPMTD_FORMAT_STRING, NULL);
1882 const char * headerGetString(Header h, rpmTag tag)
1884 const char *res = NULL;
1887 if (headerGet(h, tag, &td, HEADERGET_MINMEM)) {
1888 if (rpmtdCount(&td) == 1) {
1889 res = rpmtdGetString(&td);
1896 uint64_t headerGetNumber(Header h, rpmTag tag)
1901 if (headerGet(h, tag, &td, HEADERGET_EXT)) {
1902 if (rpmtdCount(&td) == 1) {
1903 res = rpmtdGetNumber(&td);