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
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++) {
127
		if (blob.ptr[i] != ' ' && blob.ptr[i] != '\n')
128
129
130
131
			continue;

		parse_depend(db, depends,
			     APK_BLOB_PTR_PTR(start, &blob.ptr[i-1]));
132
		start = &blob.ptr[i];
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
	}
	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,
148
			      "%s ",
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
			      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
	struct read_info_ctx *ri = (struct read_info_ctx *) ctx;
	struct apk_database *db = ri->db;
	struct apk_package *pkg = ri->pkg;
	apk_blob_t name, version;
279
	char *slash;
Timo Teräs's avatar
Timo Teräs committed
280
	int i;
281

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

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

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

305
306
		if (pkg->name == NULL)
			pkg->name = apk_db_get_name(db, name);
Timo Teräs's avatar
Timo Teräs committed
307
308
309
310
311
		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) {
312
				apk_blob_t blob = apk_blob_from_istream(is, ae->size);
313
314
				apk_pkg_add_info(ri->db, ri->pkg, fields[i].field,
						 trim(blob));
Timo Teräs's avatar
Timo Teräs committed
315
316
317
				free(blob.ptr);
				break;
			}
318
		}
Timo Teräs's avatar
Timo Teräs committed
319
320
321
322
		if (apk_script_type(slash+1) == APK_SCRIPT_POST_INSTALL ||
		    apk_script_type(slash+1) == APK_SCRIPT_PRE_INSTALL)
			ri->has_install = 1;
	} else {
323
		pkg->installed_size += apk_calc_installed_size(ae->size);
Timo Teräs's avatar
Timo Teräs committed
324
	}
325
326
327
328
329
330
331

	return 0;
}

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

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

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

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

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

	/* Add implicit busybox dependency if there is scripts */
	if (ctx.has_install) {
		struct apk_dependency dep = {
359
			.name = apk_db_get_name(db, APK_BLOB_STR("busybox")),
360
361
362
		};
		apk_deps_add(&ctx.pkg->depends, &dep);
	}
363
	ctx.pkg->filename = strdup(file);
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381

	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);

382
383
	if (pkg->depends)
		free(pkg->depends);
384
385
386
387
388
389
390
391
392
393
394
395
396
	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)
{
397
	if (list_hashed(&pkg->installed_pkgs_list))
398
399
400
401
		return APK_STATE_INSTALL;
	return APK_STATE_NO_INSTALL;
}

402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
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;
	}
}

421
int apk_pkg_add_script(struct apk_package *pkg, struct apk_istream *is,
422
423
424
425
426
427
428
429
		       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;
430
	r = is->read(is, script->script, size);
431
432
433
434
435
436
437
438
439
	if (r < 0) {
		free(script);
		return r;
	}

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

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

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

		snprintf(fn, sizeof(fn),
			"tmp/%s-%s.%s",
			pkg->name->name, pkg->version,
Timo Teräs's avatar
Timo Teräs committed
462
			script_types[type]);
463
464
465
466
467
468
469
470
471
472
473
474
		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) {
475
476
			if (chroot(".") < 0) {
				apk_error("chroot: %s", strerror(errno));
Timo Teräs's avatar
Timo Teräs committed
477
478
479
480
481
			} else if (script->type == APK_SCRIPT_GENERIC) {
				execle(fn, "INSTALL", script_types2[type],
				       pkg->version, "", NULL, environment);
			} else {
				execle(fn, script_types[type],
482
				       pkg->version, "", NULL, environment);
Timo Teräs's avatar
Timo Teräs committed
483
			}
484
485
486
			exit(1);
		}
		waitpid(pid, &status, 0);
Timo Teräs's avatar
Timo Teräs committed
487
		unlink(fn);
488
489
490
491
492
493
494
495
		if (WIFEXITED(status))
			return WEXITSTATUS(status);
		return -1;
	}

	return 0;
}

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

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

503
	apk_pkg_add_info(ri->db, ri->pkg, line.ptr[0], APK_BLOB_PTR_LEN(line.ptr+2, line.len-2));
504
505
506
507
508
	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
509
	struct read_info_ctx ctx;
510

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

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

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

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

Timo Teräs's avatar
Timo Teräs committed
527
	return ctx.pkg;
528
529
530
531
532
533
534
535
536
537
}

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"
538
539
		      "S:%zu\n"
		      "I:%zu\n"
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
		      "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);
}