16c95142eSMartin Matuska /*-
26c95142eSMartin Matuska * Copyright (c) 2009 Michihiro NAKAJIMA
36c95142eSMartin Matuska * All rights reserved.
46c95142eSMartin Matuska *
56c95142eSMartin Matuska * Redistribution and use in source and binary forms, with or without
66c95142eSMartin Matuska * modification, are permitted provided that the following conditions
76c95142eSMartin Matuska * are met:
86c95142eSMartin Matuska * 1. Redistributions of source code must retain the above copyright
96c95142eSMartin Matuska * notice, this list of conditions and the following disclaimer.
106c95142eSMartin Matuska * 2. Redistributions in binary form must reproduce the above copyright
116c95142eSMartin Matuska * notice, this list of conditions and the following disclaimer in the
126c95142eSMartin Matuska * documentation and/or other materials provided with the distribution.
136c95142eSMartin Matuska *
146c95142eSMartin Matuska * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
156c95142eSMartin Matuska * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
166c95142eSMartin Matuska * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
176c95142eSMartin Matuska * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
186c95142eSMartin Matuska * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
196c95142eSMartin Matuska * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
206c95142eSMartin Matuska * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
216c95142eSMartin Matuska * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
226c95142eSMartin Matuska * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
236c95142eSMartin Matuska * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
246c95142eSMartin Matuska */
256c95142eSMartin Matuska
266c95142eSMartin Matuska #include "archive_platform.h"
276c95142eSMartin Matuska
286c95142eSMartin Matuska #ifdef HAVE_ERRNO_H
296c95142eSMartin Matuska #include <errno.h>
306c95142eSMartin Matuska #endif
316c95142eSMartin Matuska #ifdef HAVE_STDLIB_H
326c95142eSMartin Matuska #include <stdlib.h>
336c95142eSMartin Matuska #endif
346c95142eSMartin Matuska
356c95142eSMartin Matuska #include "archive.h"
366c95142eSMartin Matuska #include "archive_endian.h"
376c95142eSMartin Matuska #include "archive_private.h"
386c95142eSMartin Matuska #include "archive_read_private.h"
396c95142eSMartin Matuska
406c95142eSMartin Matuska struct rpm {
416c95142eSMartin Matuska int64_t total_in;
426c95142eSMartin Matuska size_t hpos;
436c95142eSMartin Matuska size_t hlen;
446c95142eSMartin Matuska unsigned char header[16];
456c95142eSMartin Matuska enum {
466c95142eSMartin Matuska ST_LEAD, /* Skipping 'Lead' section. */
476c95142eSMartin Matuska ST_HEADER, /* Reading 'Header' section;
486c95142eSMartin Matuska * first 16 bytes. */
496c95142eSMartin Matuska ST_HEADER_DATA, /* Skipping 'Header' section. */
506c95142eSMartin Matuska ST_PADDING, /* Skipping padding data after the
516c95142eSMartin Matuska * 'Header' section. */
526c95142eSMartin Matuska ST_ARCHIVE /* Reading 'Archive' section. */
536c95142eSMartin Matuska } state;
546c95142eSMartin Matuska int first_header;
556c95142eSMartin Matuska };
566c95142eSMartin Matuska #define RPM_LEAD_SIZE 96 /* Size of 'Lead' section. */
576c95142eSMartin Matuska
586c95142eSMartin Matuska static int rpm_bidder_bid(struct archive_read_filter_bidder *,
596c95142eSMartin Matuska struct archive_read_filter *);
606c95142eSMartin Matuska static int rpm_bidder_init(struct archive_read_filter *);
616c95142eSMartin Matuska
626c95142eSMartin Matuska static ssize_t rpm_filter_read(struct archive_read_filter *,
636c95142eSMartin Matuska const void **);
646c95142eSMartin Matuska static int rpm_filter_close(struct archive_read_filter *);
656c95142eSMartin Matuska
666c95142eSMartin Matuska #if ARCHIVE_VERSION_NUMBER < 4000000
676c95142eSMartin Matuska /* Deprecated; remove in libarchive 4.0 */
686c95142eSMartin Matuska int
archive_read_support_compression_rpm(struct archive * a)696c95142eSMartin Matuska archive_read_support_compression_rpm(struct archive *a)
706c95142eSMartin Matuska {
716c95142eSMartin Matuska return archive_read_support_filter_rpm(a);
726c95142eSMartin Matuska }
736c95142eSMartin Matuska #endif
746c95142eSMartin Matuska
75833a452eSMartin Matuska static const struct archive_read_filter_bidder_vtable
76833a452eSMartin Matuska rpm_bidder_vtable = {
77833a452eSMartin Matuska .bid = rpm_bidder_bid,
78833a452eSMartin Matuska .init = rpm_bidder_init,
79833a452eSMartin Matuska };
80833a452eSMartin Matuska
816c95142eSMartin Matuska int
archive_read_support_filter_rpm(struct archive * _a)826c95142eSMartin Matuska archive_read_support_filter_rpm(struct archive *_a)
836c95142eSMartin Matuska {
846c95142eSMartin Matuska struct archive_read *a = (struct archive_read *)_a;
856c95142eSMartin Matuska
86833a452eSMartin Matuska return __archive_read_register_bidder(a, NULL, "rpm",
87833a452eSMartin Matuska &rpm_bidder_vtable);
886c95142eSMartin Matuska }
896c95142eSMartin Matuska
906c95142eSMartin Matuska static int
rpm_bidder_bid(struct archive_read_filter_bidder * self,struct archive_read_filter * filter)916c95142eSMartin Matuska rpm_bidder_bid(struct archive_read_filter_bidder *self,
926c95142eSMartin Matuska struct archive_read_filter *filter)
936c95142eSMartin Matuska {
946c95142eSMartin Matuska const unsigned char *b;
956c95142eSMartin Matuska ssize_t avail;
966c95142eSMartin Matuska int bits_checked;
976c95142eSMartin Matuska
986c95142eSMartin Matuska (void)self; /* UNUSED */
996c95142eSMartin Matuska
1006c95142eSMartin Matuska b = __archive_read_filter_ahead(filter, 8, &avail);
1016c95142eSMartin Matuska if (b == NULL)
1026c95142eSMartin Matuska return (0);
1036c95142eSMartin Matuska
1046c95142eSMartin Matuska bits_checked = 0;
1056c95142eSMartin Matuska /*
1066c95142eSMartin Matuska * Verify Header Magic Bytes : 0XED 0XAB 0XEE 0XDB
1076c95142eSMartin Matuska */
1086c95142eSMartin Matuska if (memcmp(b, "\xED\xAB\xEE\xDB", 4) != 0)
1096c95142eSMartin Matuska return (0);
1106c95142eSMartin Matuska bits_checked += 32;
1116c95142eSMartin Matuska /*
1126c95142eSMartin Matuska * Check major version.
1136c95142eSMartin Matuska */
1146c95142eSMartin Matuska if (b[4] != 3 && b[4] != 4)
1156c95142eSMartin Matuska return (0);
1166c95142eSMartin Matuska bits_checked += 8;
1176c95142eSMartin Matuska /*
1186c95142eSMartin Matuska * Check package type; binary or source.
1196c95142eSMartin Matuska */
1206c95142eSMartin Matuska if (b[6] != 0)
1216c95142eSMartin Matuska return (0);
1226c95142eSMartin Matuska bits_checked += 8;
1236c95142eSMartin Matuska if (b[7] != 0 && b[7] != 1)
1246c95142eSMartin Matuska return (0);
1256c95142eSMartin Matuska bits_checked += 8;
1266c95142eSMartin Matuska
1276c95142eSMartin Matuska return (bits_checked);
1286c95142eSMartin Matuska }
1296c95142eSMartin Matuska
130833a452eSMartin Matuska static const struct archive_read_filter_vtable
131833a452eSMartin Matuska rpm_reader_vtable = {
132833a452eSMartin Matuska .read = rpm_filter_read,
133833a452eSMartin Matuska .close = rpm_filter_close,
134833a452eSMartin Matuska };
135833a452eSMartin Matuska
1366c95142eSMartin Matuska static int
rpm_bidder_init(struct archive_read_filter * self)1376c95142eSMartin Matuska rpm_bidder_init(struct archive_read_filter *self)
1386c95142eSMartin Matuska {
1396c95142eSMartin Matuska struct rpm *rpm;
1406c95142eSMartin Matuska
141acc60b03SMartin Matuska self->code = ARCHIVE_FILTER_RPM;
1426c95142eSMartin Matuska self->name = "rpm";
1436c95142eSMartin Matuska
144*b9128a37SMartin Matuska rpm = (struct rpm *)calloc(1, sizeof(*rpm));
1456c95142eSMartin Matuska if (rpm == NULL) {
1466c95142eSMartin Matuska archive_set_error(&self->archive->archive, ENOMEM,
1476c95142eSMartin Matuska "Can't allocate data for rpm");
1486c95142eSMartin Matuska return (ARCHIVE_FATAL);
1496c95142eSMartin Matuska }
1506c95142eSMartin Matuska
1516c95142eSMartin Matuska self->data = rpm;
1526c95142eSMartin Matuska rpm->state = ST_LEAD;
153833a452eSMartin Matuska self->vtable = &rpm_reader_vtable;
1546c95142eSMartin Matuska
1556c95142eSMartin Matuska return (ARCHIVE_OK);
1566c95142eSMartin Matuska }
1576c95142eSMartin Matuska
1586c95142eSMartin Matuska static ssize_t
rpm_filter_read(struct archive_read_filter * self,const void ** buff)1596c95142eSMartin Matuska rpm_filter_read(struct archive_read_filter *self, const void **buff)
1606c95142eSMartin Matuska {
1616c95142eSMartin Matuska struct rpm *rpm;
1626c95142eSMartin Matuska const unsigned char *b;
1636c95142eSMartin Matuska ssize_t avail_in, total;
1646c95142eSMartin Matuska size_t used, n;
1656c95142eSMartin Matuska uint32_t section;
1666c95142eSMartin Matuska uint32_t bytes;
1676c95142eSMartin Matuska
1686c95142eSMartin Matuska rpm = (struct rpm *)self->data;
1696c95142eSMartin Matuska *buff = NULL;
1706c95142eSMartin Matuska total = avail_in = 0;
1716c95142eSMartin Matuska b = NULL;
1726c95142eSMartin Matuska used = 0;
1736c95142eSMartin Matuska do {
1746c95142eSMartin Matuska if (b == NULL) {
1756c95142eSMartin Matuska b = __archive_read_filter_ahead(self->upstream, 1,
1766c95142eSMartin Matuska &avail_in);
1776c95142eSMartin Matuska if (b == NULL) {
1786c95142eSMartin Matuska if (avail_in < 0)
1796c95142eSMartin Matuska return (ARCHIVE_FATAL);
1806c95142eSMartin Matuska else
1816c95142eSMartin Matuska break;
1826c95142eSMartin Matuska }
1836c95142eSMartin Matuska }
1846c95142eSMartin Matuska
1856c95142eSMartin Matuska switch (rpm->state) {
1866c95142eSMartin Matuska case ST_LEAD:
1876c95142eSMartin Matuska if (rpm->total_in + avail_in < RPM_LEAD_SIZE)
1886c95142eSMartin Matuska used += avail_in;
1896c95142eSMartin Matuska else {
190fd082e96SMartin Matuska n = (size_t)(RPM_LEAD_SIZE - rpm->total_in);
1916c95142eSMartin Matuska used += n;
1926c95142eSMartin Matuska b += n;
1936c95142eSMartin Matuska rpm->state = ST_HEADER;
1946c95142eSMartin Matuska rpm->hpos = 0;
1956c95142eSMartin Matuska rpm->hlen = 0;
1966c95142eSMartin Matuska rpm->first_header = 1;
1976c95142eSMartin Matuska }
1986c95142eSMartin Matuska break;
1996c95142eSMartin Matuska case ST_HEADER:
2006c95142eSMartin Matuska n = 16 - rpm->hpos;
2016c95142eSMartin Matuska if (n > avail_in - used)
2026c95142eSMartin Matuska n = avail_in - used;
2036c95142eSMartin Matuska memcpy(rpm->header+rpm->hpos, b, n);
2046c95142eSMartin Matuska b += n;
2056c95142eSMartin Matuska used += n;
2066c95142eSMartin Matuska rpm->hpos += n;
2076c95142eSMartin Matuska
2086c95142eSMartin Matuska if (rpm->hpos == 16) {
2096c95142eSMartin Matuska if (rpm->header[0] != 0x8e ||
2106c95142eSMartin Matuska rpm->header[1] != 0xad ||
2116c95142eSMartin Matuska rpm->header[2] != 0xe8 ||
2126c95142eSMartin Matuska rpm->header[3] != 0x01) {
2136c95142eSMartin Matuska if (rpm->first_header) {
2146c95142eSMartin Matuska archive_set_error(
2156c95142eSMartin Matuska &self->archive->archive,
2166c95142eSMartin Matuska ARCHIVE_ERRNO_FILE_FORMAT,
217ddce862aSMartin Matuska "Unrecognized rpm header");
2186c95142eSMartin Matuska return (ARCHIVE_FATAL);
2196c95142eSMartin Matuska }
2206c95142eSMartin Matuska rpm->state = ST_ARCHIVE;
2216c95142eSMartin Matuska *buff = rpm->header;
2226c95142eSMartin Matuska total = rpm->hpos;
2236c95142eSMartin Matuska break;
2246c95142eSMartin Matuska }
2256c95142eSMartin Matuska /* Calculate 'Header' length. */
2266c95142eSMartin Matuska section = archive_be32dec(rpm->header+8);
2276c95142eSMartin Matuska bytes = archive_be32dec(rpm->header+12);
2286c95142eSMartin Matuska rpm->hlen = 16 + section * 16 + bytes;
2296c95142eSMartin Matuska rpm->state = ST_HEADER_DATA;
2306c95142eSMartin Matuska rpm->first_header = 0;
2316c95142eSMartin Matuska }
2326c95142eSMartin Matuska break;
2336c95142eSMartin Matuska case ST_HEADER_DATA:
2346c95142eSMartin Matuska n = rpm->hlen - rpm->hpos;
2356c95142eSMartin Matuska if (n > avail_in - used)
2366c95142eSMartin Matuska n = avail_in - used;
2376c95142eSMartin Matuska b += n;
2386c95142eSMartin Matuska used += n;
2396c95142eSMartin Matuska rpm->hpos += n;
2406c95142eSMartin Matuska if (rpm->hpos == rpm->hlen)
2416c95142eSMartin Matuska rpm->state = ST_PADDING;
2426c95142eSMartin Matuska break;
2436c95142eSMartin Matuska case ST_PADDING:
2446c95142eSMartin Matuska while (used < (size_t)avail_in) {
2456c95142eSMartin Matuska if (*b != 0) {
2466c95142eSMartin Matuska /* Read next header. */
2476c95142eSMartin Matuska rpm->state = ST_HEADER;
2486c95142eSMartin Matuska rpm->hpos = 0;
2496c95142eSMartin Matuska rpm->hlen = 0;
2506c95142eSMartin Matuska break;
2516c95142eSMartin Matuska }
2526c95142eSMartin Matuska b++;
2536c95142eSMartin Matuska used++;
2546c95142eSMartin Matuska }
2556c95142eSMartin Matuska break;
2566c95142eSMartin Matuska case ST_ARCHIVE:
2576c95142eSMartin Matuska *buff = b;
2586c95142eSMartin Matuska total = avail_in;
2596c95142eSMartin Matuska used = avail_in;
2606c95142eSMartin Matuska break;
2616c95142eSMartin Matuska }
2626c95142eSMartin Matuska if (used == (size_t)avail_in) {
2636c95142eSMartin Matuska rpm->total_in += used;
2646c95142eSMartin Matuska __archive_read_filter_consume(self->upstream, used);
2656c95142eSMartin Matuska b = NULL;
2666c95142eSMartin Matuska used = 0;
2676c95142eSMartin Matuska }
2686c95142eSMartin Matuska } while (total == 0 && avail_in > 0);
2696c95142eSMartin Matuska
2706c95142eSMartin Matuska if (used > 0 && b != NULL) {
2716c95142eSMartin Matuska rpm->total_in += used;
2726c95142eSMartin Matuska __archive_read_filter_consume(self->upstream, used);
2736c95142eSMartin Matuska }
2746c95142eSMartin Matuska return (total);
2756c95142eSMartin Matuska }
2766c95142eSMartin Matuska
2776c95142eSMartin Matuska static int
rpm_filter_close(struct archive_read_filter * self)2786c95142eSMartin Matuska rpm_filter_close(struct archive_read_filter *self)
2796c95142eSMartin Matuska {
2806c95142eSMartin Matuska struct rpm *rpm;
2816c95142eSMartin Matuska
2826c95142eSMartin Matuska rpm = (struct rpm *)self->data;
2836c95142eSMartin Matuska free(rpm);
2846c95142eSMartin Matuska
2856c95142eSMartin Matuska return (ARCHIVE_OK);
2866c95142eSMartin Matuska }
2876c95142eSMartin Matuska
288