razor: rpm.c@ec850cbd6935 (annotated)

krh@74	1	#include <stdio.h>
krh@84	2	#include <stddef.h>
krh@74	3	#include <string.h>
krh@74	4	#include <sys/stat.h>
krh@74	5	#include <sys/mman.h>
krh@85	6	#include <sys/types.h>
krh@85	7	#include <sys/wait.h>
krh@74	8	#include <fcntl.h>
krh@74	9	#include <unistd.h>
krh@74	10	#include <arpa/inet.h>
krh@74	11	#include <rpm/rpmlib.h>
krh@77	12	#include <zlib.h>
krh@74	13
krh@75	14	#include "razor.h"
krh@91	15	#include "razor-internal.h"
krh@75	16
krh@74	17	#define RPM_LEAD_SIZE 96
krh@74	18
krh@74	19	struct rpm_header {
krh@74	20	unsigned char magic[4];
krh@74	21	unsigned char reserved[4];
krh@74	22	int nindex;
krh@74	23	int hsize;
krh@74	24	};
krh@74	25
krh@74	26	struct rpm_header_index {
krh@74	27	int tag;
krh@74	28	int type;
krh@74	29	int offset;
krh@74	30	int count;
krh@74	31	};
krh@74	32
krh@77	33	struct razor_rpm {
krh@75	34	struct rpm_header *signature;
krh@75	35	struct rpm_header *header;
krh@77	36	const char **dirs;
krh@75	37	const char *pool;
krh@75	38	void *map;
krh@75	39	size_t size;
krh@77	40	void *payload;
krh@75	41	};
krh@75	42
krh@85	43	static struct rpm_header_index *
krh@85	44	razor_rpm_get_header(struct razor_rpm *rpm, unsigned int tag)
krh@85	45	{
krh@85	46	struct rpm_header_index index, end;
krh@85	47
krh@85	48	index = (struct rpm_header_index *) (rpm->header + 1);
krh@85	49	end = index + ntohl(rpm->header->nindex);
krh@85	50	while (index < end) {
krh@85	51	if (ntohl(index->tag) == tag)
krh@85	52	return index;
krh@85	53	index++;
krh@85	54	}
krh@85	55
krh@85	56	return NULL;
krh@85	57	}
krh@85	58
krh@87	59	static const void *
krh@87	60	razor_rpm_get_indirect(struct razor_rpm *rpm,
krh@87	61	unsigned int tag, unsigned int *count)
krh@87	62	{
krh@87	63	struct rpm_header_index *index;
krh@87	64
krh@87	65	index = razor_rpm_get_header(rpm, tag);
krh@87	66	if (index != NULL) {
krh@87	67	if (count)
krh@87	68	*count = ntohl(index->count);
krh@87	69
krh@87	70	return rpm->pool + ntohl(index->offset);
krh@87	71	}
krh@87	72
krh@87	73	return NULL;
krh@87	74	}
krh@87	75
krh@87	76	static void
krh@87	77	import_properties(struct razor_importer *importer, unsigned long type,
krh@87	78	struct razor_rpm *rpm,
krh@87	79	int name_tag, int version_tag, int flags_tag)
krh@87	80	{
krh@87	81	const char name, version;
krh@87	82	unsigned int i, count;
krh@87	83
krh@87	84	name = razor_rpm_get_indirect(rpm, name_tag, &count);
krh@87	85	if (name == NULL)
krh@87	86	return;
krh@87	87
krh@87	88	/* FIXME: Concat version and release. */
krh@87	89	version = razor_rpm_get_indirect(rpm, version_tag, &count);
krh@87	90	for (i = 0; i < count; i++) {
krh@87	91	razor_importer_add_property(importer, name, version, type);
krh@87	92	name += strlen(name) + 1;
krh@87	93	version += strlen(version) + 1;
krh@87	94	}
krh@87	95	}
krh@87	96
krh@87	97	static void
krh@87	98	import_files(struct razor_importer importer, struct razor_rpm rpm)
krh@87	99	{
krh@87	100	const char *name;
krh@87	101	const unsigned long *index;
krh@87	102	unsigned int i, count;
krh@87	103	char buffer[256];
krh@87	104
krh@87	105	/* assert: count is the same for all arrays */
krh@87	106
krh@87	107	index = razor_rpm_get_indirect(rpm, RPMTAG_DIRINDEXES, &count);
krh@87	108	name = razor_rpm_get_indirect(rpm, RPMTAG_BASENAMES, &count);
krh@87	109	for (i = 0; i < count; i++) {
krh@87	110	snprintf(buffer, sizeof buffer,
krh@87	111	"%s%s", rpm->dirs[ntohl(*index)], name);
krh@87	112	razor_importer_add_file(importer, buffer);
krh@87	113	name += strlen(name) + 1;
krh@87	114	index++;
krh@87	115	}
krh@87	116	}
krh@87	117
krh@77	118	struct razor_rpm *
krh@77	119	razor_rpm_open(const char *filename)
krh@74	120	{
krh@77	121	struct razor_rpm *rpm;
krh@74	122	struct rpm_header_index base, index;
krh@75	123	struct stat buf;
krh@87	124	unsigned int count, i, nindex, hsize;
krh@77	125	const char *name;
krh@87	126	int fd;
krh@74	127
krh@77	128	rpm = malloc(sizeof *rpm);
krh@75	129	memset(rpm, 0, sizeof *rpm);
krh@74	130
krh@75	131	fd = open(filename, O_RDONLY);
krh@75	132	if (fd < 0) {
krh@75	133	fprintf(stderr, "couldn't open %s\n", filename);
krh@77	134	return NULL;
krh@75	135	}
krh@88	136
krh@88	137	if (fstat(fd, &buf) < 0) {
krh@88	138	fprintf(stderr, "failed to stat %s (%m)\n", filename);
krh@88	139	return NULL;
krh@88	140	}
krh@88	141
krh@75	142	rpm->size = buf.st_size;
krh@75	143	rpm->map = mmap(NULL, rpm->size, PROT_READ, MAP_PRIVATE, fd, 0);
krh@75	144	if (rpm->map == MAP_FAILED) {
krh@75	145	fprintf(stderr, "couldn't mmap %s\n", filename);
krh@77	146	return NULL;
krh@75	147	}
krh@75	148	close(fd);
krh@75	149
krh@75	150	rpm->signature = rpm->map + RPM_LEAD_SIZE;
krh@75	151	nindex = ntohl(rpm->signature->nindex);
krh@75	152	hsize = ntohl(rpm->signature->hsize);
krh@75	153	rpm->header = (void *) (rpm->signature + 1) +
krh@75	154	ALIGN(nindex * sizeof *index + hsize, 8);
krh@77	155	nindex = ntohl(rpm->header->nindex);
krh@77	156	hsize = ntohl(rpm->header->hsize);
krh@77	157	rpm->payload = (void *) (rpm->header + 1) +
krh@77	158	nindex * sizeof *index + hsize;
krh@75	159
krh@75	160	base = (struct rpm_header_index *) (rpm->header + 1);
krh@75	161	rpm->pool = (void ) base + nindex sizeof *index;
krh@75	162
krh@87	163	/* Look up dir names now so we can index them directly. */
krh@87	164	name = razor_rpm_get_indirect(rpm, RPMTAG_DIRNAMES, &count);
krh@87	165	if (name == NULL) {
krh@87	166	fprintf(stderr, "old filename style not handled\n");
krh@87	167	return NULL;
krh@74	168	}
krh@75	169
krh@87	170	rpm->dirs = calloc(count, sizeof *rpm->dirs);
krh@87	171	for (i = 0; i < count; i++) {
krh@87	172	rpm->dirs[i] = name;
krh@87	173	name += strlen(name) + 1;
krh@77	174	}
krh@77	175
krh@77	176	return rpm;
krh@77	177	}
krh@77	178
krh@77	179	struct cpio_file_header {
krh@77	180	char magic[6];
krh@77	181	char inode[8];
krh@77	182	char mode[8];
krh@77	183	char uid[8];
krh@77	184	char gid[8];
krh@77	185	char nlink[8];
krh@77	186	char mtime[8];
krh@77	187	char filesize[8];
krh@77	188	char devmajor[8];
krh@77	189	char devminor[8];
krh@77	190	char rdevmajor[8];
krh@77	191	char rdevminor[8];
krh@77	192	char namesize[8];
krh@77	193	char checksum[8];
krh@77	194	char filename[0];
krh@77	195	};
krh@77	196
krh@77	197	/* gzip flags */
krh@77	198	#define ASCII_FLAG 0x01 /* bit 0 set: file probably ascii text */
krh@77	199	#define HEAD_CRC 0x02 /* bit 1 set: header CRC present */
krh@77	200	#define EXTRA_FIELD 0x04 /* bit 2 set: extra field present */
krh@77	201	#define ORIG_NAME 0x08 /* bit 3 set: original file name present */
krh@77	202	#define COMMENT 0x10 /* bit 4 set: file comment present */
krh@77	203	#define RESERVED 0xE0 /* bits 5..7: reserved */
krh@77	204
krh@86	205	struct installer {
krh@86	206	const char *root;
krh@86	207	struct razor_rpm *rpm;
krh@86	208	z_stream stream;
krh@86	209	unsigned char buffer[32768];
krh@86	210	size_t rest, length;
krh@86	211	};
krh@86	212
krh@77	213	static int
krh@86	214	installer_inflate(struct installer *installer)
krh@86	215	{
krh@86	216	size_t length;
krh@86	217	int err;
krh@86	218
krh@86	219	if (ALIGN(installer->rest, 4) > sizeof installer->buffer)
krh@86	220	length = sizeof installer->buffer;
krh@86	221	else
krh@86	222	length = installer->rest;
krh@86	223
krh@86	224	installer->stream.next_out = installer->buffer;
krh@86	225	installer->stream.avail_out = ALIGN(length, 4);
krh@86	226	err = inflate(&installer->stream, Z_SYNC_FLUSH);
krh@86	227	if (err != Z_OK && err != Z_STREAM_END) {
krh@86	228	fprintf(stderr, "inflate error: %d (%m)\n", err);
krh@86	229	return -1;
krh@86	230	}
krh@86	231
krh@86	232	installer->rest -= length;
krh@86	233	installer->length = length;
krh@86	234
krh@86	235	return 0;
krh@86	236	}
krh@86	237
krh@86	238	static int
krh@86	239	create_path(struct installer *installer,
krh@86	240	const char path, const char name, unsigned int mode)
krh@77	241	{
krh@91	242	char buffer[PATH_MAX];
krh@86	243	int fd;
krh@77	244
krh@91	245	if (razor_create_dir(installer->root, path) < 0)
krh@91	246	return -1;
krh@77	247
krh@91	248	snprintf(buffer, sizeof buffer, "%s%s/%s",
krh@91	249	installer->root, path, name);
krh@77	250
krh@77	251	switch (mode >> 12) {
krh@77	252	case REG:
krh@86	253	fd = open(buffer, O_WRONLY \| O_CREAT \| O_TRUNC, mode & 0x1ff);
krh@86	254	if (fd < 0){
krh@86	255	fprintf(stderr, "failed to create file %s\n", buffer);
krh@86	256	return -1;
krh@86	257	}
krh@86	258	while (installer->rest > 0) {
krh@86	259	if (installer_inflate(installer)) {
krh@86	260	fprintf(stderr, "failed to inflate\n");
krh@86	261	return -1;
krh@86	262	}
krh@91	263	if (razor_write(fd, installer->buffer,
krh@91	264	installer->length)) {
krh@86	265	fprintf(stderr, "failed to write payload\n");
krh@86	266	return -1;
krh@86	267	}
krh@86	268	}
krh@86	269	if (close(fd) < 0) {
krh@86	270	fprintf(stderr, "failed to close %s: %m\n", buffer);
krh@86	271	return -1;
krh@86	272	}
krh@86	273	return 0;
krh@86	274	case XDIR:
krh@86	275	return mkdir(buffer, mode & 0x1ff);
krh@86	276	case PIPE:
krh@86	277	case CDEV:
krh@86	278	case BDEV:
krh@86	279	case SOCK:
krh@86	280	printf("%s: unhandled file type %d\n", buffer, mode >> 12);
krh@86	281	return 0;
krh@86	282	case LINK:
krh@86	283	if (installer_inflate(installer)) {
krh@86	284	fprintf(stderr, "failed to inflate\n");
krh@86	285	return -1;
krh@86	286	}
krh@86	287	if (installer->length >= sizeof installer->buffer) {
krh@86	288	fprintf(stderr, "link name too long\n");
krh@86	289	return -1;
krh@86	290	}
krh@86	291	installer->buffer[installer->length] = '\0';
krh@86	292	if (symlink((const char *) installer->buffer, buffer)) {
krh@86	293	fprintf(stderr, "failed to create symlink, %m\n");
krh@86	294	return -1;
krh@86	295	}
krh@86	296	return 0;
krh@77	297	default:
krh@86	298	printf("%s: unknown file type %d\n", buffer, mode >> 12);
krh@86	299	return 0;
krh@77	300	}
krh@77	301	}
krh@77	302
krh@85	303	static int
krh@86	304	run_script(struct installer *installer,
krh@86	305	unsigned int program_tag, unsigned int script_tag)
krh@85	306	{
krh@85	307	int pid, status, fd[2];
krh@86	308	const char script = NULL, program = NULL;
krh@85	309
krh@87	310	program = razor_rpm_get_indirect(installer->rpm, program_tag, NULL);
krh@87	311	script = razor_rpm_get_indirect(installer->rpm, script_tag, NULL);
krh@86	312	if (program == NULL && script == NULL) {
krh@86	313	printf("no script or program for tags %d and %d\n",
krh@86	314	program_tag, script_tag);
krh@86	315	return -1;
krh@86	316	} else if (program == NULL) {
krh@86	317	program = "/bin/sh";
krh@85	318	}
krh@85	319
krh@85	320	if (pipe(fd) < 0) {
krh@85	321	fprintf(stderr, "failed to create pipe\n");
krh@85	322	return -1;
krh@85	323	}
krh@85	324	pid = fork();
krh@85	325	if (pid < 0) {
krh@85	326	fprintf(stderr, "failed to fork, %m\n");
krh@85	327	} else if (pid == 0) {
krh@85	328	if (dup2(fd[0], STDIN_FILENO) < 0) {
krh@85	329	fprintf(stderr, "failed redirect stdin, %m\n");
krh@85	330	return -1;
krh@85	331	}
krh@85	332	if (close(fd[0]) < 0 \|\| close(fd[1]) < 0) {
krh@85	333	fprintf(stderr, "failed to close pipe, %m\n");
krh@89	334	return -1;
krh@85	335	}
krh@86	336	if (chroot(installer->root) < 0) {
krh@86	337	fprintf(stderr, "failed to chroot to %s, %m\n",
krh@86	338	installer->root);
krh@85	339	return -1;
krh@85	340	}
krh@86	341	printf("executing program %s in chroot %s\n",
krh@86	342	program, installer->root);
krh@86	343	if (execl(program, program, NULL)) {
krh@86	344	fprintf(stderr, "failed to exec %s, %m\n", program);
krh@89	345	exit(-1);
krh@85	346	}
krh@85	347	} else {
krh@86	348	if (script && write(fd[1], script, strlen(script)) < 0) {
krh@85	349	fprintf(stderr, "failed to pipe script, %m\n");
krh@85	350	return -1;
krh@85	351	}
krh@85	352	if (close(fd[0]) \|\| close(fd[1])) {
krh@85	353	fprintf(stderr, "failed to close pipe, %m\n");
krh@85	354	return -1;
krh@85	355	}
krh@85	356	if (wait(&status) < 0) {
krh@85	357	fprintf(stderr, "wait for child failed, %m");
krh@85	358	return -1;
krh@85	359	}
krh@85	360	printf("script exited with status %d\n", status);
krh@85	361	}
krh@85	362
krh@85	363	return 0;
krh@85	364	}
krh@85	365
krh@86	366	static int
krh@86	367	installer_init(struct installer *installer)
krh@77	368	{
krh@86	369	unsigned char *gz_header;
krh@86	370	int method, flags, err;
krh@77	371
krh@86	372	gz_header = installer->rpm->payload;
krh@77	373	if (gz_header[0] != 0x1f \|\| gz_header[1] != 0x8b) {
krh@77	374	fprintf(stderr, "payload section doesn't have gz header\n");
krh@77	375	return -1;
krh@77	376	}
krh@77	377
krh@77	378	method = gz_header[2];
krh@77	379	flags = gz_header[3];
krh@77	380
krh@77	381	if (method != Z_DEFLATED \|\| flags != 0) {
krh@77	382	fprintf(stderr,
krh@77	383	"unknown payload compression method or flags set\n");
krh@77	384	return -1;
krh@77	385	}
krh@77	386
krh@86	387	installer->stream.zalloc = NULL;
krh@86	388	installer->stream.zfree = NULL;
krh@86	389	installer->stream.opaque = NULL;
krh@85	390
krh@86	391	installer->stream.next_in = gz_header + 10;
krh@86	392	installer->stream.avail_in =
krh@86	393	(installer->rpm->map + installer->rpm->size) -
krh@86	394	(void *) installer->stream.next_in;
krh@86	395	installer->stream.next_out = NULL;
krh@86	396	installer->stream.avail_out = 0;
krh@77	397
krh@86	398	err = inflateInit2(&installer->stream, -MAX_WBITS);
krh@77	399	if (err != Z_OK) {
krh@77	400	fprintf(stderr, "inflateInit error: %d\n", err);
krh@77	401	return -1;
krh@77	402	}
krh@77	403
krh@86	404	return 0;
krh@86	405	}
krh@77	406
krh@86	407	static int
krh@86	408	installer_finish(struct installer *installer)
krh@86	409	{
krh@86	410	int err;
krh@77	411
krh@86	412	err = inflateEnd(&installer->stream);
krh@77	413
krh@77	414	if (err != Z_OK) {
krh@77	415	fprintf(stderr, "inflateEnd error: %d\n", err);
krh@77	416	return -1;
krh@77	417	}
krh@77	418
krh@86	419	return 0;
krh@86	420	}
krh@86	421
krh@86	422	int
krh@86	423	razor_rpm_install(struct razor_rpm rpm, const char root)
krh@86	424	{
krh@86	425	struct installer installer;
krh@87	426	unsigned int count, i, length;
krh@86	427	struct cpio_file_header *header;
krh@87	428	const unsigned long size, index, *flags;
krh@87	429	const unsigned short *mode;
krh@86	430	const char name, dir;
krh@86	431	struct stat buf;
krh@86	432
krh@86	433	installer.rpm = rpm;
krh@86	434	installer.root = root;
krh@86	435
krh@86	436	/* FIXME: Only do this before a transaction, not per rpm. */
krh@86	437	if (stat(root, &buf) < 0 \|\| !S_ISDIR(buf.st_mode)) {
krh@86	438	fprintf(stderr,
krh@86	439	"root installation directory \"%s\" does not exist\n",
krh@86	440	root);
krh@86	441	return -1;
krh@86	442	}
krh@86	443
krh@86	444	if (installer_init(&installer))
krh@86	445	return -1;
krh@86	446
krh@86	447	run_script(&installer, RPMTAG_PREINPROG, RPMTAG_PREIN);
krh@86	448
krh@87	449	name = razor_rpm_get_indirect(rpm, RPMTAG_BASENAMES, &count);
krh@87	450	size = razor_rpm_get_indirect(rpm, RPMTAG_FILESIZES, &count);
krh@87	451	index = razor_rpm_get_indirect(rpm, RPMTAG_DIRINDEXES, &count);
krh@87	452	mode = razor_rpm_get_indirect(rpm, RPMTAG_FILEMODES, &count);
krh@87	453	flags = razor_rpm_get_indirect(rpm, RPMTAG_FILEFLAGS, &count);
krh@86	454
krh@86	455	for (i = 0; i < count; i++) {
krh@86	456	dir = rpm->dirs[ntohl(*index)];
krh@86	457
krh@86	458	/* Skip past the cpio header block unless it's a ghost file,
krh@86	459	* in which case doesn't appear in the cpio archive. */
krh@86	460	if (!(ntohl(*flags) & RPMFILE_GHOST)) {
krh@86	461	/* Plus two for the leading '.' and the terminating NUL. */
krh@86	462	length = sizeof *header + strlen(dir) + strlen(name) + 2;
krh@86	463	installer.rest = ALIGN(length, 4);
krh@86	464	if (installer_inflate(&installer))
krh@86	465	return -1;
krh@86	466	}
krh@86	467
krh@86	468	installer.rest = ntohl(*size);
krh@86	469	if (create_path(&installer, dir, name, ntohs(*mode)) < 0)
krh@86	470	return -1;
krh@86	471
krh@86	472	name += strlen(name) + 1;
krh@86	473	index++;
krh@86	474	size++;
krh@86	475	mode++;
krh@86	476	flags++;
krh@86	477	}
krh@86	478
krh@86	479	if (installer_finish(&installer))
krh@86	480	return -1;
krh@86	481
krh@86	482	run_script(&installer, RPMTAG_POSTINPROG, RPMTAG_POSTIN);
krh@85	483
krh@75	484	return 0;
krh@74	485	}
krh@74	486
krh@77	487	int
krh@77	488	razor_rpm_close(struct razor_rpm *rpm)
krh@74	489	{
krh@77	490	int err;
krh@77	491
krh@77	492	free(rpm->dirs);
krh@77	493	err = munmap(rpm->map, rpm->size);
krh@77	494	free(rpm);
krh@77	495
krh@77	496	return err;
krh@75	497	}
krh@74	498
krh@75	499	int
krh@77	500	razor_importer_add_rpm(struct razor_importer importer, struct razor_rpm rpm)
krh@75	501	{
krh@87	502	const char name, version, *release;
krh@75	503
krh@87	504	name = razor_rpm_get_indirect(rpm, RPMTAG_NAME, NULL);
krh@87	505	version = razor_rpm_get_indirect(rpm, RPMTAG_VERSION, NULL);
krh@87	506	release = razor_rpm_get_indirect(rpm, RPMTAG_RELEASE, NULL);
krh@87	507
krh@87	508	/* FIXME: Concatenate version and release. */
krh@87	509	razor_importer_begin_package(importer, name, version);
krh@87	510
krh@87	511	import_properties(importer, RAZOR_PROPERTY_REQUIRES, rpm,
krh@87	512	RPMTAG_REQUIRENAME,
krh@87	513	RPMTAG_REQUIREVERSION,
krh@87	514	RPMTAG_REQUIREFLAGS);
krh@87	515
krh@87	516	import_properties(importer, RAZOR_PROPERTY_PROVIDES, rpm,
krh@87	517	RPMTAG_PROVIDENAME,
krh@87	518	RPMTAG_PROVIDEVERSION,
krh@87	519	RPMTAG_PROVIDEFLAGS);
krh@87	520
krh@87	521	import_properties(importer, RAZOR_PROPERTY_OBSOLETES, rpm,
krh@87	522	RPMTAG_OBSOLETENAME,
krh@87	523	RPMTAG_OBSOLETEVERSION,
krh@87	524	RPMTAG_OBSOLETEFLAGS);
krh@87	525
krh@87	526	import_properties(importer, RAZOR_PROPERTY_CONFLICTS, rpm,
krh@87	527	RPMTAG_CONFLICTNAME,
krh@87	528	RPMTAG_CONFLICTVERSION,
krh@87	529	RPMTAG_CONFLICTFLAGS);
krh@87	530
krh@77	531	import_files(importer, rpm);
krh@74	532
krh@75	533	razor_importer_finish_package(importer);
krh@74	534
krh@75	535	return 0;
krh@74	536	}

author	Kristian H?gsberg <krh@redhat.com>
	Thu Jan 10 23:03:03 2008 -0500 (2008-01-10)
changeset 98	ec850cbd6935
parent 89	feaf8640e017
child 109	313b0a615c14
permissions	-rw-r--r--