librazor/rpm.c
author J. Ali Harlow <ali@juiblex.co.uk>
Wed Jan 14 12:21:38 2009 +0000 (2009-01-14)
changeset 350 1555934cfb04
parent 345 edd9b0fa63ca
child 351 48b0adfe3059
permissions -rw-r--r--
Fix bug causing "razor remove" to quietly fail.
     1 /*
     2  * Copyright (C) 2008  Kristian Høgsberg <krh@redhat.com>
     3  * Copyright (C) 2008  Red Hat, Inc
     4  *
     5  * This program is free software; you can redistribute it and/or modify
     6  * it under the terms of the GNU General Public License as published by
     7  * the Free Software Foundation; either version 2 of the License, or
     8  * (at your option) any later version.
     9  *
    10  * This program is distributed in the hope that it will be useful,
    11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    13  * GNU General Public License for more details.
    14  *
    15  * You should have received a copy of the GNU General Public License along
    16  * with this program; if not, write to the Free Software Foundation, Inc.,
    17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
    18  */
    19 
    20 #include "config.h"
    21 
    22 #include <stdio.h>
    23 #include <stddef.h>
    24 #include <stdlib.h>
    25 #include <string.h>
    26 #include <errno.h>
    27 #include <sys/stat.h>
    28 #include <sys/types.h>
    29 #if HAVE_SYS_WAIT_H
    30 #include <sys/wait.h>
    31 #endif
    32 #include <fcntl.h>
    33 #include <dirent.h>
    34 #include <unistd.h>
    35 #if MSWIN_API
    36 #include <winsock2.h>	/* For ntohl() */
    37 #else
    38 #include <arpa/inet.h>
    39 #endif
    40 #include <limits.h>
    41 #include <zlib.h>
    42 #include <assert.h>
    43 
    44 #include "razor.h"
    45 #include "razor-internal.h"
    46 
    47 #ifndef O_BINARY
    48 #define O_BINARY	0
    49 #endif
    50 
    51 #define	RPM_LEAD_SIZE 96
    52 
    53 enum {
    54     PIPE	=  1,	/*!< pipe/fifo */
    55     CDEV	=  2,	/*!< character device */
    56     XDIR	=  4,	/*!< directory */
    57     BDEV	=  6,	/*!< block device */
    58     REG		=  8,	/*!< regular file */
    59     LINK	= 10,	/*!< hard link */
    60     SOCK	= 12	/*!< socket */
    61 };
    62 
    63 enum {
    64     RPMSENSE_LESS		= 1 << 1,
    65     RPMSENSE_GREATER		= 1 << 2,
    66     RPMSENSE_EQUAL		= 1 << 3,
    67     RPMSENSE_PREREQ		= 1 << 6,
    68     RPMSENSE_SCRIPT_PRE		= 1 << 9,
    69     RPMSENSE_SCRIPT_POST	= 1 << 10,
    70     RPMSENSE_SCRIPT_PREUN	= 1 << 11,
    71     RPMSENSE_SCRIPT_POSTUN	= 1 << 12,
    72 };
    73 
    74 enum {
    75     RPMTAG_NAME  		= 1000,	/* s */
    76     RPMTAG_VERSION		= 1001,	/* s */
    77     RPMTAG_RELEASE		= 1002,	/* s */
    78     RPMTAG_EPOCH   		= 1003,	/* i */
    79     RPMTAG_SUMMARY		= 1004,	/* s{} */
    80     RPMTAG_DESCRIPTION		= 1005,	/* s{} */
    81     RPMTAG_BUILDTIME		= 1006,	/* i */
    82     RPMTAG_BUILDHOST		= 1007,	/* s */
    83     RPMTAG_INSTALLTIME		= 1008,	/* i */
    84     RPMTAG_SIZE			= 1009,	/* i */
    85     RPMTAG_DISTRIBUTION		= 1010,	/* s */
    86     RPMTAG_VENDOR		= 1011,	/* s */
    87     RPMTAG_GIF			= 1012,	/* x */
    88     RPMTAG_XPM			= 1013,	/* x */
    89     RPMTAG_LICENSE		= 1014,	/* s */
    90     RPMTAG_PACKAGER		= 1015,	/* s */
    91     RPMTAG_GROUP		= 1016,	/* s{} */
    92     RPMTAG_CHANGELOG		= 1017, /*!< s[] internal */
    93     RPMTAG_SOURCE		= 1018,	/* s[] */
    94     RPMTAG_PATCH		= 1019,	/* s[] */
    95     RPMTAG_URL			= 1020,	/* s */
    96     RPMTAG_OS			= 1021,	/* s legacy used int */
    97     RPMTAG_ARCH			= 1022,	/* s legacy used int */
    98     RPMTAG_PREIN		= 1023,	/* s */
    99     RPMTAG_POSTIN		= 1024,	/* s */
   100     RPMTAG_PREUN		= 1025,	/* s */
   101     RPMTAG_POSTUN		= 1026,	/* s */
   102     RPMTAG_OLDFILENAMES		= 1027, /* s[] obsolete */
   103     RPMTAG_FILESIZES		= 1028,	/* i */
   104     RPMTAG_FILESTATES		= 1029, /* c */
   105     RPMTAG_FILEMODES		= 1030,	/* h */
   106     RPMTAG_FILEUIDS		= 1031, /*!< internal */
   107     RPMTAG_FILEGIDS		= 1032, /*!< internal */
   108     RPMTAG_FILERDEVS		= 1033,	/* h */
   109     RPMTAG_FILEMTIMES		= 1034, /* i */
   110     RPMTAG_FILEMD5S		= 1035,	/* s[] */
   111     RPMTAG_FILELINKTOS		= 1036,	/* s[] */
   112     RPMTAG_FILEFLAGS		= 1037,	/* i */
   113     RPMTAG_ROOT			= 1038, /*!< internal - obsolete */
   114     RPMTAG_FILEUSERNAME		= 1039,	/* s[] */
   115     RPMTAG_FILEGROUPNAME	= 1040,	/* s[] */
   116     RPMTAG_EXCLUDE		= 1041, /*!< internal - obsolete */
   117     RPMTAG_EXCLUSIVE		= 1042, /*!< internal - obsolete */
   118     RPMTAG_ICON			= 1043,
   119     RPMTAG_SOURCERPM		= 1044,	/* s */
   120     RPMTAG_FILEVERIFYFLAGS	= 1045,	/* i */
   121     RPMTAG_ARCHIVESIZE		= 1046,	/* i */
   122     RPMTAG_PROVIDENAME		= 1047,	/* s[] */
   123     RPMTAG_REQUIREFLAGS		= 1048,	/* i */
   124     RPMTAG_REQUIRENAME		= 1049,	/* s[] */
   125     RPMTAG_REQUIREVERSION	= 1050,	/* s[] */
   126     RPMTAG_NOSOURCE		= 1051, /*!< internal */
   127     RPMTAG_NOPATCH		= 1052, /*!< internal */
   128     RPMTAG_CONFLICTFLAGS	= 1053, /* i */
   129     RPMTAG_CONFLICTNAME		= 1054,	/* s[] */
   130     RPMTAG_CONFLICTVERSION	= 1055,	/* s[] */
   131     RPMTAG_DEFAULTPREFIX	= 1056, /*!< internal - deprecated */
   132     RPMTAG_BUILDROOT		= 1057, /*!< internal */
   133     RPMTAG_INSTALLPREFIX	= 1058, /*!< internal - deprecated */
   134     RPMTAG_EXCLUDEARCH		= 1059,
   135     RPMTAG_EXCLUDEOS		= 1060,
   136     RPMTAG_EXCLUSIVEARCH	= 1061,
   137     RPMTAG_EXCLUSIVEOS		= 1062,
   138     RPMTAG_AUTOREQPROV		= 1063, /*!< internal */
   139     RPMTAG_RPMVERSION		= 1064,	/* s */
   140     RPMTAG_TRIGGERSCRIPTS	= 1065,	/* s[] */
   141     RPMTAG_TRIGGERNAME		= 1066,	/* s[] */
   142     RPMTAG_TRIGGERVERSION	= 1067,	/* s[] */
   143     RPMTAG_TRIGGERFLAGS		= 1068,	/* i */
   144     RPMTAG_TRIGGERINDEX		= 1069,	/* i */
   145     RPMTAG_VERIFYSCRIPT		= 1079,	/* s */
   146     RPMTAG_CHANGELOGTIME	= 1080,	/* i */
   147     RPMTAG_CHANGELOGNAME	= 1081,	/* s[] */
   148     RPMTAG_CHANGELOGTEXT	= 1082,	/* s[] */
   149     RPMTAG_BROKENMD5		= 1083, /*!< internal - obsolete */
   150     RPMTAG_PREREQ		= 1084, /*!< internal */
   151     RPMTAG_PREINPROG		= 1085,	/* s */
   152     RPMTAG_POSTINPROG		= 1086,	/* s */
   153     RPMTAG_PREUNPROG		= 1087,	/* s */
   154     RPMTAG_POSTUNPROG		= 1088,	/* s */
   155     RPMTAG_BUILDARCHS		= 1089,
   156     RPMTAG_OBSOLETENAME		= 1090,	/* s[] */
   157     RPMTAG_VERIFYSCRIPTPROG	= 1091,	/* s */
   158     RPMTAG_TRIGGERSCRIPTPROG	= 1092,	/* s */
   159     RPMTAG_DOCDIR		= 1093, /*!< internal */
   160     RPMTAG_COOKIE		= 1094,	/* s */
   161     RPMTAG_FILEDEVICES		= 1095,	/* i */
   162     RPMTAG_FILEINODES		= 1096,	/* i */
   163     RPMTAG_FILELANGS		= 1097,	/* s[] */
   164     RPMTAG_PREFIXES		= 1098,	/* s[] */
   165     RPMTAG_INSTPREFIXES		= 1099,	/* s[] */
   166     RPMTAG_TRIGGERIN		= 1100, /*!< internal */
   167     RPMTAG_TRIGGERUN		= 1101, /*!< internal */
   168     RPMTAG_TRIGGERPOSTUN	= 1102, /*!< internal */
   169     RPMTAG_AUTOREQ		= 1103, /*!< internal */
   170     RPMTAG_AUTOPROV		= 1104, /*!< internal */
   171     RPMTAG_CAPABILITY		= 1105, /*!< internal - obsolete */
   172     RPMTAG_SOURCEPACKAGE	= 1106, /*!< i src.rpm header marker */
   173     RPMTAG_OLDORIGFILENAMES	= 1107, /*!< internal - obsolete */
   174     RPMTAG_BUILDPREREQ		= 1108, /*!< internal */
   175     RPMTAG_BUILDREQUIRES	= 1109, /*!< internal */
   176     RPMTAG_BUILDCONFLICTS	= 1110, /*!< internal */
   177     RPMTAG_BUILDMACROS		= 1111, /*!< internal - unused */
   178     RPMTAG_PROVIDEFLAGS		= 1112,	/* i */
   179     RPMTAG_PROVIDEVERSION	= 1113,	/* s[] */
   180     RPMTAG_OBSOLETEFLAGS	= 1114,	/* i */
   181     RPMTAG_OBSOLETEVERSION	= 1115,	/* s[] */
   182     RPMTAG_DIRINDEXES		= 1116,	/* i */
   183     RPMTAG_BASENAMES		= 1117,	/* s[] */
   184     RPMTAG_DIRNAMES		= 1118,	/* s[] */
   185     RPMTAG_ORIGDIRINDEXES	= 1119, /*!< internal */
   186     RPMTAG_ORIGBASENAMES	= 1120, /*!< internal */
   187     RPMTAG_ORIGDIRNAMES		= 1121, /*!< internal */
   188     RPMTAG_OPTFLAGS		= 1122,	/* s */
   189     RPMTAG_DISTURL		= 1123,	/* s */
   190     RPMTAG_PAYLOADFORMAT	= 1124,	/* s */
   191     RPMTAG_PAYLOADCOMPRESSOR	= 1125,	/* s */
   192     RPMTAG_PAYLOADFLAGS		= 1126,	/* s */
   193     RPMTAG_INSTALLCOLOR		= 1127, /*!< i transaction color when installed */
   194     RPMTAG_INSTALLTID		= 1128,	/* i */
   195     RPMTAG_REMOVETID		= 1129,	/* i */
   196     RPMTAG_SHA1RHN		= 1130, /*!< internal - obsolete */
   197     RPMTAG_RHNPLATFORM		= 1131,	/* s */
   198     RPMTAG_PLATFORM		= 1132,	/* s */
   199     RPMTAG_PATCHESNAME		= 1133, /*!< placeholder (SuSE) */
   200     RPMTAG_PATCHESFLAGS		= 1134, /*!< placeholder (SuSE) */
   201     RPMTAG_PATCHESVERSION	= 1135, /*!< placeholder (SuSE) */
   202     RPMTAG_CACHECTIME		= 1136,	/* i */
   203     RPMTAG_CACHEPKGPATH		= 1137,	/* s */
   204     RPMTAG_CACHEPKGSIZE		= 1138,	/* i */
   205     RPMTAG_CACHEPKGMTIME	= 1139,	/* i */
   206     RPMTAG_FILECOLORS		= 1140,	/* i */
   207     RPMTAG_FILECLASS		= 1141,	/* i */
   208     RPMTAG_CLASSDICT		= 1142,	/* s[] */
   209     RPMTAG_FILEDEPENDSX		= 1143,	/* i */
   210     RPMTAG_FILEDEPENDSN		= 1144,	/* i */
   211     RPMTAG_DEPENDSDICT		= 1145,	/* i */
   212     RPMTAG_SOURCEPKGID		= 1146,	/* x */
   213     RPMTAG_FILECONTEXTS		= 1147,	/* s[] */
   214     RPMTAG_FSCONTEXTS		= 1148,	/*!< s[] extension */
   215     RPMTAG_RECONTEXTS		= 1149,	/*!< s[] extension */
   216     RPMTAG_POLICIES		= 1150,	/*!< s[] selinux *.te policy file. */
   217     RPMTAG_PRETRANS		= 1151,	/* s */
   218     RPMTAG_POSTTRANS		= 1152,	/* s */
   219     RPMTAG_PRETRANSPROG		= 1153,	/* s */
   220     RPMTAG_POSTTRANSPROG	= 1154,	/* s */
   221     RPMTAG_DISTTAG		= 1155,	/* s */
   222     RPMTAG_SUGGESTSNAME		= 1156,	/* s[] extension placeholder */
   223     RPMTAG_SUGGESTSVERSION	= 1157,	/* s[] extension placeholder */
   224     RPMTAG_SUGGESTSFLAGS	= 1158,	/* i   extension placeholder */
   225     RPMTAG_ENHANCESNAME		= 1159,	/* s[] extension placeholder */
   226     RPMTAG_ENHANCESVERSION	= 1160,	/* s[] extension placeholder */
   227     RPMTAG_ENHANCESFLAGS	= 1161,	/* i   extension placeholder */
   228     RPMTAG_PRIORITY		= 1162, /* i   extension placeholder */
   229     RPMTAG_CVSID		= 1163, /* s */
   230     RPMTAG_TRIGGERPREIN		= 1171, /*!< internal */
   231 };
   232 
   233 struct rpm_header {
   234 	unsigned char magic[4];
   235 	unsigned char reserved[4];
   236 	int nindex;
   237 	int hsize;
   238 };
   239 
   240 struct rpm_header_index {
   241 	int tag;
   242 	int type;
   243 	int offset;
   244 	int count;
   245 };
   246 
   247 struct razor_rpm {
   248 	struct rpm_header *signature;
   249 	struct rpm_header *header;
   250 	const char **dirs;
   251 	const char *pool;
   252 	void *map;
   253 	size_t size;
   254 	void *payload;
   255 };
   256 
   257 static struct rpm_header_index *
   258 razor_rpm_get_header(struct razor_rpm *rpm, unsigned int tag)
   259 {
   260 	struct rpm_header_index *index, *end;
   261 
   262 	index = (struct rpm_header_index *) (rpm->header + 1);
   263 	end = index + ntohl(rpm->header->nindex);
   264 	while (index < end) {
   265 		if (ntohl(index->tag) == tag)
   266 			return index;
   267 		index++;
   268 	}
   269 
   270 	return NULL;
   271 }
   272 
   273 static const void *
   274 razor_rpm_get_indirect(struct razor_rpm *rpm,
   275 		       unsigned int tag, unsigned int *count)
   276 {
   277 	struct rpm_header_index *index;
   278 
   279 	index = razor_rpm_get_header(rpm, tag);
   280 	if (index != NULL) {
   281 		if (count)
   282 			*count = ntohl(index->count);
   283 
   284 		return rpm->pool + ntohl(index->offset);
   285 	}
   286 
   287 	return NULL;
   288 }
   289 
   290 static uint32_t
   291 rpm_to_razor_flags(uint32_t flags)
   292 {
   293 	uint32_t razor_flags;
   294 
   295 	razor_flags = 0;
   296 	if (flags & RPMSENSE_LESS)
   297 		razor_flags |= RAZOR_PROPERTY_LESS;
   298 	if (flags & RPMSENSE_EQUAL)
   299 		razor_flags |= RAZOR_PROPERTY_EQUAL;
   300 	if (flags & RPMSENSE_GREATER)
   301 		razor_flags |= RAZOR_PROPERTY_GREATER;
   302 
   303 	if (flags & RPMSENSE_SCRIPT_PRE)
   304 		razor_flags |= RAZOR_PROPERTY_PRE;
   305 	if (flags & RPMSENSE_SCRIPT_POST)
   306 		razor_flags |= RAZOR_PROPERTY_POST;
   307 	if (flags & RPMSENSE_SCRIPT_PREUN)
   308 		razor_flags |= RAZOR_PROPERTY_PREUN;
   309 	if (flags & RPMSENSE_SCRIPT_POSTUN)
   310 		razor_flags |= RAZOR_PROPERTY_POSTUN;
   311 	
   312 	return razor_flags;
   313 }
   314 
   315 static void
   316 import_properties(struct razor_importer *importer, uint32_t type,
   317 		  struct razor_rpm *rpm,
   318 		  int name_tag, int version_tag, int flags_tag)
   319 {
   320 	const char *name, *version;
   321 	const uint32_t *flags;
   322 	uint32_t f;
   323 	unsigned int i, count;
   324 
   325 	name = razor_rpm_get_indirect(rpm, name_tag, &count);
   326 	if (name == NULL)
   327 		return;
   328 
   329 	flags = razor_rpm_get_indirect(rpm, flags_tag, &count);
   330 
   331 	version = razor_rpm_get_indirect(rpm, version_tag, &count);
   332 	for (i = 0; i < count; i++) {
   333 		f = rpm_to_razor_flags(ntohl(flags[i]));
   334 		razor_importer_add_property(importer, name, f | type, version);
   335 		name += strlen(name) + 1;
   336 		version += strlen(version) + 1;
   337 	}
   338 }
   339 
   340 static void
   341 import_files(struct razor_importer *importer, struct razor_rpm *rpm)
   342 {
   343 	const char *name;
   344 	const uint32_t *index;
   345 	unsigned int i, count;
   346 	char buffer[256];
   347 
   348 	if (rpm->dirs == NULL)
   349 		return;
   350 
   351 	/* assert: count is the same for all arrays */
   352 	index = razor_rpm_get_indirect(rpm, RPMTAG_DIRINDEXES, &count);
   353 	name = razor_rpm_get_indirect(rpm, RPMTAG_BASENAMES, &count);
   354 	for (i = 0; i < count; i++) {
   355 		snprintf(buffer, sizeof buffer,
   356 			 "%s%s", rpm->dirs[ntohl(*index)], name);
   357 		razor_importer_add_file(importer, buffer);
   358 		name += strlen(name) + 1;
   359 		index++;
   360 	}
   361 }
   362 
   363 RAZOR_EXPORT struct razor_rpm *
   364 razor_rpm_open(const char *filename)
   365 {
   366 	struct razor_rpm *rpm;
   367 	struct rpm_header_index *base, *index;
   368 	unsigned int count, i, nindex, hsize;
   369 	const char *name;
   370 
   371 	assert (filename != NULL);
   372 
   373 	rpm = malloc(sizeof *rpm);
   374 	if (rpm == NULL)
   375 		return NULL;
   376 	memset(rpm, 0, sizeof *rpm);
   377 
   378 	rpm->map = razor_file_get_contents(filename, &rpm->size);
   379 	if (!rpm->map) {
   380 		fprintf(stderr, "couldn't get contents of %s (%s)\n", filename,
   381 			strerror(errno));
   382 		return NULL;
   383 	}
   384 
   385 	rpm->signature = rpm->map + RPM_LEAD_SIZE;
   386 	nindex = ntohl(rpm->signature->nindex);
   387 	hsize = ntohl(rpm->signature->hsize);
   388 	rpm->header = (void *) (rpm->signature + 1) +
   389 		ALIGN(nindex * sizeof *index + hsize, 8);
   390 	nindex = ntohl(rpm->header->nindex);
   391 	hsize = ntohl(rpm->header->hsize);
   392 	rpm->payload = (void *) (rpm->header + 1) +
   393 		nindex * sizeof *index + hsize;
   394 
   395 	base = (struct rpm_header_index *) (rpm->header + 1);
   396 	rpm->pool = (void *) base + nindex * sizeof *index;
   397 
   398 	/* Look up dir names now so we can index them directly. */
   399 	name = razor_rpm_get_indirect(rpm, RPMTAG_DIRNAMES, &count);
   400 	if (name) {
   401 		rpm->dirs = calloc(count, sizeof *rpm->dirs);
   402 		for (i = 0; i < count; i++) {
   403 			rpm->dirs[i] = name;
   404 			name += strlen(name) + 1;
   405 		}
   406 	} else {
   407 		name = razor_rpm_get_indirect(rpm, RPMTAG_OLDFILENAMES,
   408 					      &count);
   409 		if (name) {
   410 			fprintf(stderr, "old filenames not supported\n");
   411 			return NULL;
   412 		}
   413 	}
   414 
   415 	return rpm;
   416 }
   417 
   418 struct cpio_file_header {
   419 	char magic[6];
   420 	char inode[8];
   421 	char mode[8];
   422 	char uid[8];
   423 	char gid[8];
   424 	char nlink[8];
   425 	char mtime[8];
   426 	char filesize[8];
   427 	char devmajor[8];
   428 	char devminor[8];
   429 	char rdevmajor[8];
   430 	char rdevminor[8];
   431 	char namesize[8];
   432 	char checksum[8];
   433 	char filename[0];
   434 };
   435 
   436 /* gzip flags */
   437 #define ASCII_FLAG   0x01 /* bit 0 set: file probably ascii text */
   438 #define HEAD_CRC     0x02 /* bit 1 set: header CRC present */
   439 #define EXTRA_FIELD  0x04 /* bit 2 set: extra field present */
   440 #define ORIG_NAME    0x08 /* bit 3 set: original file name present */
   441 #define COMMENT      0x10 /* bit 4 set: file comment present */
   442 #define RESERVED     0xE0 /* bits 5..7: reserved */
   443 
   444 struct installer {
   445 	const char *root;
   446 	struct razor_rpm *rpm;
   447 	z_stream stream;
   448 	unsigned char buffer[32768];
   449 	size_t rest, length;
   450 };
   451 
   452 static int
   453 installer_inflate(struct installer *installer)
   454 {
   455 	size_t length;
   456 	int err;
   457 
   458 	if (installer->rest > sizeof installer->buffer)
   459 		length = sizeof installer->buffer;
   460 	else
   461 		length = installer->rest;
   462 
   463 	installer->stream.next_out = installer->buffer;
   464 	installer->stream.avail_out = length;
   465 	err = inflate(&installer->stream, Z_SYNC_FLUSH);
   466 	if (err != Z_OK && err != Z_STREAM_END) {
   467 		fprintf(stderr, "inflate error: %d (%s)\n", err,
   468 			strerror(errno));
   469 		return -1;
   470 	}
   471 
   472 	installer->rest -= length;
   473 	installer->length = length;
   474 
   475 	return 0;
   476 }
   477 
   478 static int
   479 installer_align(struct installer *installer, size_t size)
   480 {
   481 	unsigned char buffer[4];
   482 	int err;
   483 
   484 	installer->stream.next_out = buffer;
   485 	installer->stream.avail_out =
   486 		(size - installer->stream.total_out) & (size - 1);
   487 
   488 	if (installer->stream.avail_out == 0)
   489 		return 0;
   490 
   491 	err = inflate(&installer->stream, Z_SYNC_FLUSH);
   492 	if (err != Z_OK && err != Z_STREAM_END) {
   493 		fprintf(stderr, "inflate error: %d (%s)\n", err,
   494 			strerror(errno));
   495 		return -1;
   496 	}
   497 
   498 	return 0;
   499 }
   500 
   501 static int
   502 create_path(struct installer *installer, const char *path, unsigned int mode)
   503 {
   504 	char buffer[PATH_MAX];
   505 	struct stat buf;
   506 	int fd, ret;
   507 
   508 	if (razor_create_dir(installer->root, path) < 0)
   509 		return -1;
   510 
   511 	snprintf(buffer, sizeof buffer, "%s%s", installer->root, path);
   512 
   513 	switch (mode >> 12) {
   514 	case REG:
   515 		/* FIXME: handle the case where a file is already there. */
   516 		fd = open(buffer, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY,
   517 			  mode & 0x1ff);
   518 		if (fd < 0){
   519 			fprintf(stderr, "failed to create file %s\n", buffer);
   520 			return -1;
   521 		}
   522 		while (installer->rest > 0) {
   523 			if (installer_inflate(installer)) {
   524 				fprintf(stderr, "failed to inflate\n");
   525 				return -1;
   526 			}
   527 			if (razor_write(fd, installer->buffer,
   528 					installer->length)) {
   529 				fprintf(stderr, "failed to write payload\n");
   530 				return -1;
   531 			}
   532 		}
   533 		if (close(fd) < 0) {
   534 			fprintf(stderr, "failed to close %s: %s\n", buffer,
   535 				strerror(errno));
   536 			return -1;
   537 		}
   538 		return 0;
   539 	case XDIR:
   540 		ret = mkdir(buffer, mode & 0x1ff);
   541 		if (ret == 0 || errno != EEXIST)
   542 			return ret;
   543 		if (stat(buffer, &buf) || !S_ISDIR(buf.st_mode)) {
   544 			/* FIXME: also check that mode match. */
   545 			fprintf(stderr,
   546 				"%s exists but is not a directory\n", buffer);
   547 			return -1;
   548 		}
   549 		return 0;
   550 	case LINK:
   551 #if HAVE_SYMLINK
   552 		if (installer_inflate(installer)) {
   553 			fprintf(stderr, "failed to inflate\n");
   554 			return -1;
   555 		}
   556 		if (installer->length >= sizeof installer->buffer) {
   557 			fprintf(stderr, "link name too long\n");
   558 			return -1;
   559 		}
   560 		installer->buffer[installer->length] = '\0';
   561 		if (symlink((const char *) installer->buffer, buffer)) {
   562 			perror("failed to create symlink");
   563 			return -1;
   564 		}
   565 		return 0;
   566 #else
   567 		/* fall through */
   568 #endif
   569 	case PIPE:
   570 	case CDEV:
   571 	case BDEV:
   572 	case SOCK:
   573 		printf("%s: unhandled file type %d\n", buffer, mode >> 12);
   574 		return 0;
   575 	default:
   576 		printf("%s: unknown file type %d\n", buffer, mode >> 12);
   577 		return 0;
   578 	}
   579 }
   580 
   581 static int
   582 run_script(struct installer *installer,
   583 	   unsigned int program_tag, unsigned int script_tag)
   584 {
   585 #if HAVE_CHROOT
   586 	int pid, status, fd[2];
   587 #else
   588 	FILE *fp;
   589 #endif
   590 	const char *script = NULL, *program = NULL;
   591 
   592 	program = razor_rpm_get_indirect(installer->rpm, program_tag, NULL);
   593 	script = razor_rpm_get_indirect(installer->rpm, script_tag, NULL);
   594 	if (program == NULL && script == NULL) {
   595 		return 0;
   596 	} else if (program == NULL) {
   597 #if MSWIN_API
   598 		program = getenv("COMSPEC");
   599 		if (program) {
   600 			program = strchr(program, '=');
   601 			if (program)
   602 				program++;
   603 		}
   604 		if (!program)
   605 			program = "c:\\windows\\system32\\cmd.exe";
   606 #else
   607 		program = "/bin/sh";
   608 #endif
   609 	}
   610 
   611 #if HAVE_CHROOT
   612 	if (pipe(fd) < 0) {
   613 		fprintf(stderr, "failed to create pipe\n");
   614 		return -1;
   615 	}
   616 	pid = fork();
   617 	if (pid < 0) {
   618 		perror("failed to fork");
   619 	} else if (pid == 0) {
   620 		if (dup2(fd[0], STDIN_FILENO) < 0) {
   621 			perror("failed redirect stdin");
   622 			exit(-1);
   623 		}
   624 		if (close(fd[0]) < 0 || close(fd[1]) < 0) {
   625 			perror("failed to close pipe");
   626 			exit(-1);
   627 		}
   628 		if (chroot(installer->root) < 0) {
   629 			fprintf(stderr, "failed to chroot to %s: %s\n",
   630 				installer->root, strerror(errno));
   631 			exit(-1);
   632 		}
   633 		printf("executing program %s in chroot %s\n",
   634 		       program, installer->root);
   635 		if (execl(program, program, NULL)) {
   636 			fprintf(stderr, "failed to exec %s: %s\n", program,
   637 				strerror(errno));
   638 			exit(-1);
   639 		}
   640 	} else {
   641 		if (script && razor_write(fd[1], script, strlen(script)) < 0) {
   642 			perror("failed to pipe script");
   643 			return -1;
   644 		}
   645 		if (close(fd[0]) || close(fd[1])) {
   646 			perror("failed to close pipe");
   647 			return -1;
   648 		}
   649 		if (wait(&status) < 0) {
   650 			perror("wait for child failed");
   651 			return -1;
   652 		}
   653 		if (status)
   654 			printf("script exited with status %d\n", status);
   655 	}
   656 #else
   657 	fp = popen(program, "w");
   658 	if (fwrite(script, strlen(script), 1, fp) != 1) {
   659 		perror("failed to pipe script");
   660 		return -1;
   661 	}
   662 	pclose(fp);
   663 #endif
   664 
   665 	return 0;
   666 }
   667 
   668 static int
   669 installer_init(struct installer *installer)
   670 {
   671 	unsigned char *gz_header;
   672 	int method, flags, err;
   673 
   674 	gz_header = installer->rpm->payload;
   675 	if (gz_header[0] != 0x1f || gz_header[1] != 0x8b) {
   676 		fprintf(stderr, "payload section doesn't have gz header\n");
   677 		return -1;
   678 	}
   679 
   680 	method = gz_header[2];
   681 	flags = gz_header[3];
   682 
   683 	if (method != Z_DEFLATED || flags != 0) {
   684 		fprintf(stderr,
   685 			"unknown payload compression method or flags set\n");
   686 		return -1;
   687 	}
   688 
   689 	installer->stream.zalloc = NULL;
   690 	installer->stream.zfree = NULL;
   691 	installer->stream.opaque = NULL;
   692 
   693 	installer->stream.next_in  = gz_header + 10;
   694 	installer->stream.avail_in =
   695 		(installer->rpm->map + installer->rpm->size) -
   696 		(void *) installer->stream.next_in;
   697 	installer->stream.next_out = NULL;
   698 	installer->stream.avail_out = 0;
   699 
   700 	err = inflateInit2(&installer->stream, -MAX_WBITS);
   701 	if (err != Z_OK) {
   702 		fprintf(stderr, "inflateInit error: %d\n", err);
   703 		return -1;
   704 	}
   705 
   706 	return 0;
   707 }
   708 
   709 static int
   710 installer_finish(struct installer *installer)
   711 {
   712 	int err;
   713 
   714 	err = inflateEnd(&installer->stream);
   715 
   716 	if (err != Z_OK) {
   717 		fprintf(stderr, "inflateEnd error: %d\n", err);
   718 		return -1;
   719 	}
   720 
   721 	return 0;
   722 }
   723 
   724 static unsigned long
   725 fixed_hex_to_ulong(const char *hex, int length)
   726 {
   727 	long l;
   728 	int i;
   729 
   730 	for (i = 0, l = 0; i < length; i++) {
   731 		if (hex[i] < 'a')
   732 			l = l * 16 + hex[i] - '0';
   733 		else
   734 			l = l * 16 + hex[i] - 'a' + 10;
   735 	}
   736 
   737 	return l;
   738 }
   739 
   740 RAZOR_EXPORT int
   741 razor_rpm_install(struct razor_rpm *rpm, const char *root)
   742 {
   743 	struct installer installer;
   744 	struct cpio_file_header *header;
   745 	struct stat buf;
   746 	unsigned int mode;
   747 	char *path;
   748 	size_t filesize;
   749 
   750 	assert (rpm != NULL);
   751 	assert (root != NULL);
   752 
   753 	installer.rpm = rpm;
   754 	installer.root = root;
   755 
   756 	/* FIXME: Only do this before a transaction, not per rpm. */
   757 	if (*root && (stat(root, &buf) < 0 || !S_ISDIR(buf.st_mode))) {
   758 		fprintf(stderr,
   759 			"root installation directory \"%s\" does not exist\n",
   760 			root);
   761 		return -1;
   762 	}
   763 
   764 	if (installer_init(&installer))
   765 		return -1;
   766 
   767 	run_script(&installer, RPMTAG_PREINPROG, RPMTAG_PREIN);
   768 
   769 	while (installer.stream.avail_in > 0) {
   770 		installer.rest = sizeof *header;
   771 		if (installer_inflate(&installer))
   772 			return -1;
   773 
   774 		header = (struct cpio_file_header *) installer.buffer;
   775 		mode = fixed_hex_to_ulong(header->mode, sizeof header->mode);
   776 		filesize = fixed_hex_to_ulong(header->filesize,
   777 					      sizeof header->filesize);
   778 
   779 		installer.rest = fixed_hex_to_ulong(header->namesize,
   780 						    sizeof header->namesize);
   781 
   782 		if (installer_inflate(&installer) ||
   783 		    installer_align(&installer, 4))
   784 			return -1;
   785 
   786 		path = (char *) installer.buffer;
   787 		/* This convention is so lame... */
   788 		if (strcmp(path, "TRAILER!!!") == 0)
   789 			break;
   790 
   791 		installer.rest = filesize;
   792 		if (create_path(&installer, path + 1, mode) < 0)
   793 			return -1;
   794 		if (installer_align(&installer, 4))
   795 			return -1;
   796 	}
   797 
   798 	if (installer_finish(&installer))
   799 		return -1;
   800 
   801 	run_script(&installer, RPMTAG_POSTINPROG, RPMTAG_POSTIN);
   802 
   803 	return 0;
   804 }
   805 
   806 RAZOR_EXPORT int
   807 razor_rpm_close(struct razor_rpm *rpm)
   808 {
   809 	int err;
   810 
   811 	assert (rpm != NULL);
   812 
   813 	free(rpm->dirs);
   814 	err = razor_file_free_contents(rpm->map, rpm->size);
   815 	free(rpm);
   816 
   817 	return err;
   818 }
   819 
   820 RAZOR_EXPORT int
   821 razor_importer_add_rpm(struct razor_importer *importer, struct razor_rpm *rpm)
   822 {
   823 	const char *name, *version, *release, *arch;
   824 	const char *summary, *description, *url, *license;
   825 	const uint32_t *epoch;
   826 	char evr[128], buf[16];
   827 
   828 	assert (importer != NULL);
   829 	assert (rpm != NULL);
   830 
   831 	name = razor_rpm_get_indirect(rpm, RPMTAG_NAME, NULL);
   832 	epoch = razor_rpm_get_indirect(rpm, RPMTAG_EPOCH, NULL);
   833 	version = razor_rpm_get_indirect(rpm, RPMTAG_VERSION, NULL);
   834 	release = razor_rpm_get_indirect(rpm, RPMTAG_RELEASE, NULL);
   835 	arch = razor_rpm_get_indirect(rpm, RPMTAG_ARCH, NULL);
   836 
   837 	summary = razor_rpm_get_indirect(rpm, RPMTAG_SUMMARY, NULL);
   838 	description = razor_rpm_get_indirect(rpm, RPMTAG_DESCRIPTION, NULL);
   839 	url = razor_rpm_get_indirect(rpm, RPMTAG_URL, NULL);
   840 	license = razor_rpm_get_indirect(rpm, RPMTAG_LICENSE, NULL);
   841 
   842 	if (epoch) {
   843 		snprintf(buf, sizeof buf, "%lu", ntohl(*epoch));
   844 		razor_build_evr(evr, sizeof evr, buf, version, release);
   845 	} else {
   846 		razor_build_evr(evr, sizeof evr, NULL, version, release);
   847 	}
   848 	razor_importer_begin_package(importer, name, evr, arch);
   849 
   850 	razor_importer_add_details(importer, summary, description, url,
   851 				   license);
   852 
   853 	import_properties(importer, RAZOR_PROPERTY_REQUIRES, rpm,
   854 			  RPMTAG_REQUIRENAME,
   855 			  RPMTAG_REQUIREVERSION,
   856 			  RPMTAG_REQUIREFLAGS);
   857 
   858 	import_properties(importer, RAZOR_PROPERTY_PROVIDES, rpm,
   859 			  RPMTAG_PROVIDENAME,
   860 			  RPMTAG_PROVIDEVERSION,
   861 			  RPMTAG_PROVIDEFLAGS);
   862 
   863 	import_properties(importer, RAZOR_PROPERTY_OBSOLETES, rpm,
   864 			  RPMTAG_OBSOLETENAME,
   865 			  RPMTAG_OBSOLETEVERSION,
   866 			  RPMTAG_OBSOLETEFLAGS);
   867 
   868 	import_properties(importer, RAZOR_PROPERTY_CONFLICTS, rpm,
   869 			  RPMTAG_CONFLICTNAME,
   870 			  RPMTAG_CONFLICTVERSION,
   871 			  RPMTAG_CONFLICTFLAGS);
   872 
   873 	import_files(importer, rpm);
   874 
   875 	razor_importer_finish_package(importer);
   876 
   877 	return 0;
   878 }