package.c 12.1 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
/* package.c - Alpine Package Keeper (APK)
 *
 * Copyright (C) 2005-2008 Natanael Copa <n@tanael.org>
 * Copyright (C) 2008 Timo Teräs <timo.teras@iki.fi>
 * All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify it 
 * under the terms of the GNU General Public License version 2 as published
 * by the Free Software Foundation. See http://www.gnu.org/ for details.
 */

12
#include <errno.h>
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#include <fcntl.h>
#include <ctype.h>
#include <stdio.h>
#include <malloc.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/wait.h>

#include "apk_defines.h"
#include "apk_archive.h"
#include "apk_package.h"
#include "apk_database.h"
#include "apk_state.h"

28
struct apk_package *apk_pkg_new(void)
29
30
31
32
33
34
35
36
37
38
{
	struct apk_package *pkg;

	pkg = calloc(1, sizeof(struct apk_package));
	if (pkg != NULL)
		list_init(&pkg->installed_pkgs_list);

	return pkg;
}

39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
int apk_pkg_parse_name(apk_blob_t apkname,
		       apk_blob_t *name,
		       apk_blob_t *version)
{
	int i, dash = 0;

	for (i = apkname.len - 2; i >= 0; i--) {
		if (apkname.ptr[i] != '-')
			continue;
		if (isdigit(apkname.ptr[i+1]))
			break;
		if (++dash >= 2)
			return -1;
	}
	if (name != NULL)
		*name = APK_BLOB_PTR_LEN(apkname.ptr, i);
	if (version != NULL)
		*version = APK_BLOB_PTR_PTR(&apkname.ptr[i+1],
					    &apkname.ptr[apkname.len-1]);

	return 0;
}

Timo Teräs's avatar
Timo Teräs committed
62
static apk_blob_t trim(apk_blob_t str)
63
64
{
	if (str.ptr == NULL || str.len < 1)
Timo Teräs's avatar
Timo Teräs committed
65
		return str;
66

67
68
	if (str.ptr[str.len-1] == '\n') {
		str.ptr[str.len-1] = 0;
Timo Teräs's avatar
Timo Teräs committed
69
		return APK_BLOB_PTR_LEN(str.ptr, str.len-1);
70
	}
71

Timo Teräs's avatar
Timo Teräs committed
72
	return str;
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
}

static void parse_depend(struct apk_database *db,
			 struct apk_dependency_array **depends,
			 apk_blob_t blob)
{
	struct apk_dependency *dep;
	struct apk_name *name;

	while (blob.len && blob.ptr[0] == ' ')
		blob.ptr++, blob.len--;
	while (blob.len && (blob.ptr[blob.len-1] == ' ' ||
			    blob.ptr[blob.len-1] == 0))
		blob.len--;

	if (blob.len == 0)
		return;

91
	name = apk_db_get_name(db, blob);
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
	dep = apk_dependency_array_add(depends);
	*dep = (struct apk_dependency){
		.prefer_upgrade = 0,
		.version_mask = 0,
		.name = name,
		.version = NULL,
	};
}

int apk_deps_add(struct apk_dependency_array **depends,
		 struct apk_dependency *dep)
{
	struct apk_dependency_array *deps = *depends;
	int i;

	if (deps != NULL) {
		for (i = 0; i < deps->num; i++) {
			if (deps->item[i].name == dep->name)
				return 0;
		}
	}

	*apk_dependency_array_add(depends) = *dep;
	return 0;
}
Timo Teräs's avatar
Timo Teräs committed
117

118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
void apk_deps_parse(struct apk_database *db,
		    struct apk_dependency_array **depends,
		    apk_blob_t blob)
{
	char *start;
	int i;

	start = blob.ptr;
	for (i = 0; i < blob.len; i++) {
		if (blob.ptr[i] != ',' && blob.ptr[i] != '\n')
			continue;

		parse_depend(db, depends,
			     APK_BLOB_PTR_PTR(start, &blob.ptr[i-1]));
		start = &blob.ptr[i+1];
	}
	parse_depend(db, depends,
		     APK_BLOB_PTR_PTR(start, &blob.ptr[i-1]));
}

int apk_deps_format(char *buf, int size,
		    struct apk_dependency_array *depends)
{
	int i, n = 0;

	if (depends == NULL)
		return 0;

	for (i = 0; i < depends->num - 1; i++)
		n += snprintf(&buf[n], size-n,
			      "%s, ",
			      depends->item[i].name->name);
	n += snprintf(&buf[n], size-n,
		      "%s\n",
		      depends->item[i].name->name);
	return n;
}

static const char *script_types[] = {
	[APK_SCRIPT_PRE_INSTALL]	= "pre-install",
	[APK_SCRIPT_POST_INSTALL]	= "post-install",
	[APK_SCRIPT_PRE_DEINSTALL]	= "pre-deinstall",
	[APK_SCRIPT_POST_DEINSTALL]	= "post-deinstall",
	[APK_SCRIPT_PRE_UPGRADE]	= "pre-upgrade",
	[APK_SCRIPT_POST_UPGRADE]	= "post-upgrade",
};

Timo Teräs's avatar
Timo Teräs committed
165
166
167
168
169
170
171
172
173
static const char *script_types2[] = {
	[APK_SCRIPT_PRE_INSTALL]	= "pre_install",
	[APK_SCRIPT_POST_INSTALL]	= "post_install",
	[APK_SCRIPT_PRE_DEINSTALL]	= "pre_deinstall",
	[APK_SCRIPT_POST_DEINSTALL]	= "post_deinstall",
	[APK_SCRIPT_PRE_UPGRADE]	= "pre_upgrade",
	[APK_SCRIPT_POST_UPGRADE]	= "post_upgrade",
};

174
175
176
177
178
179
180
181
182
int apk_script_type(const char *name)
{
	int i;

	for (i = 0; i < ARRAY_SIZE(script_types); i++)
		if (script_types[i] &&
		    strcmp(script_types[i], name) == 0)
			return i;

183
	return APK_SCRIPT_INVALID;
184
185
186
187
188
189
190
191
}

struct read_info_ctx {
	struct apk_database *db;
	struct apk_package *pkg;
	int has_install;
};

192
193
int apk_pkg_add_info(struct apk_database *db, struct apk_package *pkg,
		     char field, apk_blob_t value)
Timo Teräs's avatar
Timo Teräs committed
194
195
196
{
	switch (field) {
	case 'P':
197
		pkg->name = apk_db_get_name(db, value);
Timo Teräs's avatar
Timo Teräs committed
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
		break;
	case 'V':
		pkg->version = apk_blob_cstr(value);
		break;
	case 'T':
		pkg->description = apk_blob_cstr(value);
		break;
	case 'U':
		pkg->url = apk_blob_cstr(value);
		break;
	case 'L':
		pkg->license = apk_blob_cstr(value);
		break;
	case 'D':
		apk_deps_parse(db, &pkg->depends, value);
		break;
	case 'C':
		apk_hexdump_parse(APK_BLOB_BUF(pkg->csum), value);
		break;
	case 'S':
		pkg->size = apk_blob_uint(value, 10);
		break;
	case 'I':
		pkg->installed_size = apk_blob_uint(value, 10);
		break;
223
224
	default:
		return -1;
Timo Teräs's avatar
Timo Teräs committed
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
	}
	return 0;
}

static int read_info_line(void *ctx, apk_blob_t line)
{
	static struct {
		const char *str;
		char field;
	} fields[] = {
		{ "pkgname", 'P' },
		{ "pkgver",  'V' },
		{ "pkgdesc", 'T' },
		{ "url",     'U' },
		{ "size",    'I' },
		{ "license", 'L' },
		{ "depend",  'D' },
	};
	struct read_info_ctx *ri = (struct read_info_ctx *) ctx;
	apk_blob_t l, r;
	int i;

	if (line.ptr == NULL || line.len < 1 || line.ptr[0] == '#')
		return 0;

	if (!apk_blob_splitstr(line, " = ", &l, &r))
		return 0;

	for (i = 0; i < ARRAY_SIZE(fields); i++) {
		if (strncmp(fields[i].str, l.ptr, l.len) == 0) {
255
			apk_pkg_add_info(ri->db, ri->pkg, fields[i].field, r);
Timo Teräs's avatar
Timo Teräs committed
256
257
258
259
260
261
262
			break;
		}
	}

	return 0;
}

263
static int read_info_entry(void *ctx, const struct apk_file_info *ae,
264
			   struct apk_istream *is)
265
{
Timo Teräs's avatar
Timo Teräs committed
266
267
268
269
270
271
272
273
274
	static struct {
		const char *str;
		char field;
	} fields[] = {
		{ "DESC",	'T' },
		{ "WW",		'U' },
		{ "LICENSE",	'L' },
		{ "DEPEND", 	'D' },
	};
275
276
277
278
279
	struct read_info_ctx *ri = (struct read_info_ctx *) ctx;
	struct apk_database *db = ri->db;
	struct apk_package *pkg = ri->pkg;
	const int bsize = 4 * 1024;
	apk_blob_t name, version;
280
	char *slash;
Timo Teräs's avatar
Timo Teräs committed
281
	int i;
282

Timo Teräs's avatar
Timo Teräs committed
283
284
285
286
	/* Meta info and scripts */
	if (ae->name[0] == '.') {
		/* APK 2.0 format */
		if (strcmp(ae->name, ".PKGINFO") == 0) {
287
			apk_blob_t blob = apk_blob_from_istream(is, ae->size);
Timo Teräs's avatar
Timo Teräs committed
288
289
			apk_blob_for_each_segment(blob, "\n", read_info_line, ctx);
			free(blob.ptr);
290
			return 1;
Timo Teräs's avatar
Timo Teräs committed
291
		}
292
		/* FIXME: Check for .INSTALL */
Timo Teräs's avatar
Timo Teräs committed
293
294
295
296
	} else if (strncmp(ae->name, "var/db/apk/", 11) == 0) {
		/* APK 1.0 format */
		if (!S_ISREG(ae->mode))
			return 0;
297

Timo Teräs's avatar
Timo Teräs committed
298
299
300
		slash = strchr(&ae->name[11], '/');
		if (slash == NULL)
			return 0;
301

Timo Teräs's avatar
Timo Teräs committed
302
303
304
		if (apk_pkg_parse_name(APK_BLOB_PTR_PTR(&ae->name[11], slash-1),
				       &name, &version) < 0)
			return -1;
305

306
307
		if (pkg->name == NULL)
			pkg->name = apk_db_get_name(db, name);
Timo Teräs's avatar
Timo Teräs committed
308
309
310
311
312
		if (pkg->version == NULL)
			pkg->version = apk_blob_cstr(version);

		for (i = 0; i < ARRAY_SIZE(fields); i++) {
			if (strcmp(fields[i].str, slash+1) == 0) {
313
				apk_blob_t blob = apk_blob_from_istream(is, ae->size);
314
315
				apk_pkg_add_info(ri->db, ri->pkg, fields[i].field,
						 trim(blob));
Timo Teräs's avatar
Timo Teräs committed
316
317
318
				free(blob.ptr);
				break;
			}
319
		}
Timo Teräs's avatar
Timo Teräs committed
320
321
322
323
324
325
		if (apk_script_type(slash+1) == APK_SCRIPT_POST_INSTALL ||
		    apk_script_type(slash+1) == APK_SCRIPT_PRE_INSTALL)
			ri->has_install = 1;
	} else {
		pkg->installed_size += (ae->size + bsize - 1) & ~(bsize - 1);
	}
326
327
328
329
330
331
332

	return 0;
}

struct apk_package *apk_pkg_read(struct apk_database *db, const char *file)
{
	struct read_info_ctx ctx;
333
	struct apk_bstream *bs;
334

335
	ctx.pkg = apk_pkg_new();
336
337
338
	if (ctx.pkg == NULL)
		return NULL;

Timo Teräs's avatar
Timo Teräs committed
339
340
	bs = apk_bstream_from_file(file);
	if (bs == NULL)
341
342
343
344
		goto err;

	ctx.db = db;
	ctx.has_install = 0;
345
	if (apk_parse_tar_gz(bs, read_info_entry, &ctx) < 0) {
Timo Teräs's avatar
Timo Teräs committed
346
		apk_error("File %s is not an APK archive", file);
Timo Teräs's avatar
Timo Teräs committed
347
		bs->close(bs, NULL, NULL);
348
349
		goto err;
	}
Timo Teräs's avatar
Timo Teräs committed
350
	bs->close(bs, ctx.pkg->csum, &ctx.pkg->size);
351

Timo Teräs's avatar
Timo Teräs committed
352
353
	if (ctx.pkg->name == NULL) {
		apk_error("File %s is corrupted", file);
354
		goto err;
Timo Teräs's avatar
Timo Teräs committed
355
	}
356
357
358
359

	/* Add implicit busybox dependency if there is scripts */
	if (ctx.has_install) {
		struct apk_dependency dep = {
360
			.name = apk_db_get_name(db, APK_BLOB_STR("busybox")),
361
362
363
		};
		apk_deps_add(&ctx.pkg->depends, &dep);
	}
364
	ctx.pkg->filename = strdup(file);
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395

	return ctx.pkg;
err:
	apk_pkg_free(ctx.pkg);
	return NULL;
}

void apk_pkg_free(struct apk_package *pkg)
{
	struct apk_script *script;
	struct hlist_node *c, *n;

	if (pkg == NULL)
		return;

	hlist_for_each_entry_safe(script, c, n, &pkg->scripts, script_list)
		free(script);

	if (pkg->version)
		free(pkg->version);
	if (pkg->url)
		free(pkg->url);
	if (pkg->description)
		free(pkg->description);
	if (pkg->license)
		free(pkg->license);
	free(pkg);
}

int apk_pkg_get_state(struct apk_package *pkg)
{
396
	if (list_hashed(&pkg->installed_pkgs_list))
397
398
399
400
		return APK_STATE_INSTALL;
	return APK_STATE_NO_INSTALL;
}

401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
void apk_pkg_set_state(struct apk_database *db, struct apk_package *pkg, int state)
{
	switch (state) {
	case APK_STATE_INSTALL:
		if (!list_hashed(&pkg->installed_pkgs_list)) {
			db->installed.stats.packages++;
			list_add_tail(&pkg->installed_pkgs_list,
				      &db->installed.packages);
		}
		break;
	case APK_STATE_NO_INSTALL:
		if (list_hashed(&pkg->installed_pkgs_list)) {
			db->installed.stats.packages--;
			list_del(&pkg->installed_pkgs_list);
		}
		break;
	}
}

420
int apk_pkg_add_script(struct apk_package *pkg, struct apk_istream *is,
421
422
423
424
425
426
427
428
		       unsigned int type, unsigned int size)
{
	struct apk_script *script;
	int r;

	script = malloc(sizeof(struct apk_script) + size);
	script->type = type;
	script->size = size;
429
	r = is->read(is, script->script, size);
430
431
432
433
434
435
436
437
438
	if (r < 0) {
		free(script);
		return r;
	}

	hlist_add_head(&script->script_list, &pkg->scripts);
	return r;
}

439
int apk_pkg_run_script(struct apk_package *pkg, int root_fd,
440
441
		       unsigned int type)
{
442
443
444
445
	static const char * const environment[] = {
		"PATH=/usr/sbin:/usr/bin:/sbin:/bin",
		NULL
	};
446
447
448
449
450
451
	struct apk_script *script;
	struct hlist_node *c;
	int fd, status;
	pid_t pid;
	char fn[1024];

452
	fchdir(root_fd);
453
	hlist_for_each_entry(script, c, &pkg->scripts, script_list) {
Timo Teräs's avatar
Timo Teräs committed
454
455
		if (script->type != type &&
		    script->type != APK_SCRIPT_GENERIC)
456
457
458
459
460
			continue;

		snprintf(fn, sizeof(fn),
			"tmp/%s-%s.%s",
			pkg->name->name, pkg->version,
Timo Teräs's avatar
Timo Teräs committed
461
			script_types[type]);
462
463
464
465
466
467
468
469
470
471
472
473
		fd = creat(fn, 0777);
		if (fd < 0)
			return fd;
		write(fd, script->script, script->size);
		close(fd);

		apk_message("Executing %s", &fn[4]);

		pid = fork();
		if (pid == -1)
			return -1;
		if (pid == 0) {
474
475
			if (chroot(".") < 0) {
				apk_error("chroot: %s", strerror(errno));
Timo Teräs's avatar
Timo Teräs committed
476
477
478
479
480
			} else if (script->type == APK_SCRIPT_GENERIC) {
				execle(fn, "INSTALL", script_types2[type],
				       pkg->version, "", NULL, environment);
			} else {
				execle(fn, script_types[type],
481
				       pkg->version, "", NULL, environment);
Timo Teräs's avatar
Timo Teräs committed
482
			}
483
484
485
			exit(1);
		}
		waitpid(pid, &status, 0);
Timo Teräs's avatar
Timo Teräs committed
486
		unlink(fn);
487
488
489
490
491
492
493
494
		if (WIFEXITED(status))
			return WEXITSTATUS(status);
		return -1;
	}

	return 0;
}

Timo Teräs's avatar
Timo Teräs committed
495
static int parse_index_line(void *ctx, apk_blob_t line)
496
{
Timo Teräs's avatar
Timo Teräs committed
497
	struct read_info_ctx *ri = (struct read_info_ctx *) ctx;
498

Timo Teräs's avatar
Timo Teräs committed
499
500
	if (line.len < 3 || line.ptr[1] != ':')
		return 0;
501

502
	apk_pkg_add_info(ri->db, ri->pkg, line.ptr[0], APK_BLOB_PTR_LEN(line.ptr+2, line.len-2));
503
504
505
506
507
	return 0;
}

struct apk_package *apk_pkg_parse_index_entry(struct apk_database *db, apk_blob_t blob)
{
Timo Teräs's avatar
Timo Teräs committed
508
	struct read_info_ctx ctx;
509

510
	ctx.pkg = apk_pkg_new();
Timo Teräs's avatar
Timo Teräs committed
511
	if (ctx.pkg == NULL)
512
513
		return NULL;

Timo Teräs's avatar
Timo Teräs committed
514
515
	ctx.db = db;
	ctx.has_install = 0;
516

Timo Teräs's avatar
Timo Teräs committed
517
518
519
520
	apk_blob_for_each_segment(blob, "\n", parse_index_line, &ctx);

	if (ctx.pkg->name == NULL) {
		apk_pkg_free(ctx.pkg);
521
522
		apk_error("Failed to parse index entry: %.*s",
			  blob.len, blob.ptr);
Timo Teräs's avatar
Timo Teräs committed
523
		ctx.pkg = NULL;
524
525
	}

Timo Teräs's avatar
Timo Teräs committed
526
	return ctx.pkg;
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
}

apk_blob_t apk_pkg_format_index_entry(struct apk_package *info, int size,
				      char *buf)
{
	int n = 0;

	n += snprintf(&buf[n], size-n,
		      "P:%s\n"
		      "V:%s\n"
		      "S:%u\n"
		      "I:%u\n"
		      "T:%s\n"
		      "U:%s\n"
		      "L:%s\n",
		      info->name->name, info->version,
		      info->size, info->installed_size,
		      info->description, info->url, info->license);

	if (info->depends != NULL) {
		n += snprintf(&buf[n], size-n, "D:");
		n += apk_deps_format(&buf[n], size-n, info->depends);
	}
	n += snprintf(&buf[n], size-n, "C:");
	n += apk_hexdump_format(size-n, &buf[n],
				APK_BLOB_BUF(info->csum));
	n += snprintf(&buf[n], size-n,
		      "\n\n");

	return APK_BLOB_PTR_LEN(buf, n);
}