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

author	Kristian H?gsberg <krh@redhat.com>
	Thu Dec 27 15:47:09 2007 -0500 (2007-12-27)
changeset 85	c5041a93b91a
parent 84	6883222205ad
child 86	e44ccd213756
permissions	-rw-r--r--