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

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

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

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

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

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

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

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

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

	return 0;
}

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

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

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

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

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

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

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

Timo Teräs's avatar
Timo Teräs committed
528
	return ctx.pkg;
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
558
559
}

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