package.c 22.5 KB
Newer Older
1 2 3 4 5 6
/* 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.
 *
7
 * This program is free software; you can redistribute it and/or modify it
8 9 10 11
 * 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
#include <fcntl.h>
#include <ctype.h>
#include <stdio.h>
16
#include <limits.h>
17 18 19 20 21
#include <malloc.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/wait.h>
Natanael Copa's avatar
Natanael Copa committed
22
#include <sys/stat.h>
23

Timo Teräs's avatar
Timo Teräs committed
24 25
#include <openssl/pem.h>

26 27 28 29 30
#include "apk_defines.h"
#include "apk_archive.h"
#include "apk_package.h"
#include "apk_database.h"
#include "apk_state.h"
Natanael Copa's avatar
Natanael Copa committed
31
#include "apk_print.h"
32

33 34 35 36 37
void apk_pkg_format_plain(struct apk_package *pkg, apk_blob_t to)
{
	/* pkgname-1.0.apk */
	apk_blob_push_blob(&to, APK_BLOB_STR(pkg->name->name));
	apk_blob_push_blob(&to, APK_BLOB_STR("-"));
38
	apk_blob_push_blob(&to, *pkg->version);
39 40 41 42 43 44 45 46 47
	apk_blob_push_blob(&to, APK_BLOB_STR(".apk"));
	apk_blob_push_blob(&to, APK_BLOB_PTR_LEN("", 1));
}

void apk_pkg_format_cache(struct apk_package *pkg, apk_blob_t to)
{
	/* pkgname-1.0_alpha1.12345678.apk */
	apk_blob_push_blob(&to, APK_BLOB_STR(pkg->name->name));
	apk_blob_push_blob(&to, APK_BLOB_STR("-"));
48
	apk_blob_push_blob(&to, *pkg->version);
49 50 51 52 53 54 55
	apk_blob_push_blob(&to, APK_BLOB_STR("."));
	apk_blob_push_hexdump(&to, APK_BLOB_PTR_LEN((char *) pkg->csum.data,
						    APK_CACHE_CSUM_BYTES));
	apk_blob_push_blob(&to, APK_BLOB_STR(".apk"));
	apk_blob_push_blob(&to, APK_BLOB_PTR_LEN("", 1));
}

56
struct apk_package *apk_pkg_new(void)
57
{
Timo Teräs's avatar
Timo Teräs committed
58 59 60 61 62 63 64
	struct apk_package *pkg;

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

	return pkg;
65
}
66

67 68 69 70 71 72 73 74 75 76
struct apk_installed_package *apk_pkg_install(struct apk_database *db,
					      struct apk_package *pkg)
{
	struct apk_installed_package *ipkg;

	if (pkg->ipkg != NULL)
		return pkg->ipkg;

	pkg->ipkg = ipkg = calloc(1, sizeof(struct apk_installed_package));
	ipkg->pkg = pkg;
Timo Teräs's avatar
Timo Teräs committed
77 78
	apk_string_array_init(&ipkg->triggers);
	apk_string_array_init(&ipkg->pending_triggers);
Timo Teräs's avatar
Timo Teräs committed
79 80 81 82 83 84 85

	/* Overlay override information resides in a nameless package */
	if (pkg->name != NULL) {
		db->installed.stats.packages++;
		list_add_tail(&ipkg->installed_pkgs_list,
			      &db->installed.packages);
	}
86 87 88 89 90 91 92 93 94 95 96 97 98 99

	return ipkg;
}

void apk_pkg_uninstall(struct apk_database *db, struct apk_package *pkg)
{
	struct apk_installed_package *ipkg = pkg->ipkg;
	int i;

	if (ipkg == NULL)
		return;

	if (db != NULL)
		db->installed.stats.packages--;
100

101 102
	list_del(&ipkg->installed_pkgs_list);

Timo Teräs's avatar
Timo Teräs committed
103
	if (ipkg->triggers->num) {
104
		list_del(&ipkg->trigger_pkgs_list);
105
		list_init(&ipkg->trigger_pkgs_list);
106 107 108
		for (i = 0; i < ipkg->triggers->num; i++)
			free(ipkg->triggers->item[i]);
	}
Timo Teräs's avatar
Timo Teräs committed
109 110
	apk_string_array_free(&ipkg->triggers);
	apk_string_array_free(&ipkg->pending_triggers);
111 112 113 114 115 116

	for (i = 0; i < APK_SCRIPT_MAX; i++)
		if (ipkg->script[i].ptr != NULL)
			free(ipkg->script[i].ptr);
	free(ipkg);
	pkg->ipkg = NULL;
117 118
}

119 120 121 122 123 124
int apk_pkg_parse_name(apk_blob_t apkname,
		       apk_blob_t *name,
		       apk_blob_t *version)
{
	int i, dash = 0;

125 126 127
	if (APK_BLOB_IS_NULL(apkname))
		return -1;

128 129 130 131 132 133 134 135
	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;
	}
136 137 138
	if (i < 0)
		return -1;

139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155
	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;
}

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++) {
Timo Teräs's avatar
Timo Teräs committed
156 157
			if (deps->item[i].name == dep->name) {
				deps->item[i] = *dep;
158
				return 0;
Timo Teräs's avatar
Timo Teräs committed
159
			}
160 161 162 163 164 165
		}
	}

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

167 168 169 170 171 172 173 174 175 176 177 178 179 180
void apk_deps_del(struct apk_dependency_array **pdeps,
		  struct apk_name *name)
{
	struct apk_dependency_array *deps = *pdeps;
	int i;

	if (deps == NULL)
		return;

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

		deps->item[i] = deps->item[deps->num-1];
Timo Teräs's avatar
Timo Teräs committed
181
		apk_dependency_array_resize(pdeps, deps->num-1);
182 183 184 185
		break;
	}
}

186 187 188 189 190
struct parse_depend_ctx {
	struct apk_database *db;
	struct apk_dependency_array **depends;
};

Timo Teräs's avatar
Timo Teräs committed
191 192
int apk_dep_from_blob(struct apk_dependency *dep, struct apk_database *db,
		      apk_blob_t blob)
193 194
{
	struct apk_name *name;
195
	apk_blob_t bname, bop, bver = APK_BLOB_NULL;
196
	int mask = APK_DEPMASK_REQUIRE;
197

198 199 200 201 202 203 204 205 206 207
	/* [!]name[<,<=,=,>=,>]ver */
	if (blob.ptr[0] == '!') {
		mask = 0;
		blob.ptr++;
		blob.len--;
	}
	if (apk_blob_cspn(blob, "<>=", &bname, &bop)) {
		int i;

		if (mask == 0)
Timo Teräs's avatar
Timo Teräs committed
208
			return -EINVAL;
209
		if (!apk_blob_spn(bop, "<>=", &bop, &bver))
Timo Teräs's avatar
Timo Teräs committed
210 211 212 213
			return -EINVAL;
		mask = 0;
		for (i = 0; i < bop.len; i++) {
			switch (bop.ptr[i]) {
214 215 216 217 218 219 220 221 222 223 224
			case '<':
				mask |= APK_VERSION_LESS;
				break;
			case '>':
				mask |= APK_VERSION_GREATER;
				break;
			case '=':
				mask |= APK_VERSION_EQUAL;
				break;
			}
		}
225 226
		if ((mask & APK_DEPMASK_CHECKSUM) != APK_DEPMASK_CHECKSUM &&
		    !apk_version_validate(bver))
Timo Teräs's avatar
Timo Teräs committed
227 228 229
			return -EINVAL;

		blob = bname;
230 231
	}

Timo Teräs's avatar
Timo Teräs committed
232
	name = apk_db_get_name(db, blob);
233
	if (name == NULL)
Timo Teräs's avatar
Timo Teräs committed
234
		return -ENOENT;
235 236 237

	*dep = (struct apk_dependency){
		.name = name,
238
		.version = apk_blob_atomize_dup(bver),
239
		.result_mask = mask,
240
	};
Timo Teräs's avatar
Timo Teräs committed
241 242 243 244 245 246
	return 0;
}

void apk_dep_from_pkg(struct apk_dependency *dep, struct apk_database *db,
		      struct apk_package *pkg)
{
247 248 249 250 251 252
	char buf[64];
	apk_blob_t b = APK_BLOB_BUF(buf);

	apk_blob_push_csum(&b, &pkg->csum);
	b = apk_blob_pushed(APK_BLOB_BUF(buf), b);

Timo Teräs's avatar
Timo Teräs committed
253
	*dep = (struct apk_dependency) {
254
		.name = pkg->name,
255 256
		.version = apk_blob_atomize_dup(b),
		.result_mask = APK_DEPMASK_CHECKSUM,
Timo Teräs's avatar
Timo Teräs committed
257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274
	};
}

static int parse_depend(void *ctx, apk_blob_t blob)
{
	struct parse_depend_ctx *pctx = (struct parse_depend_ctx *) ctx;
	struct apk_dependency *dep, p;

	if (blob.len == 0)
		return 0;

	if (apk_dep_from_blob(&p, pctx->db, blob) < 0)
		return -1;

	dep = apk_dependency_array_add(pctx->depends);
	if (dep == NULL)
		return -1;
	*dep = p;
275 276

	return 0;
277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295
}

int apk_dep_is_satisfied(struct apk_dependency *dep, struct apk_package *pkg)
{
	if (dep->name != pkg->name)
		return 0;
	if (dep->result_mask == APK_DEPMASK_CHECKSUM) {
		struct apk_checksum csum;
		apk_blob_t b = *dep->version;

		apk_blob_pull_csum(&b, &csum);
		if (apk_checksum_compare(&csum, &pkg->csum) == 0)
			return 1;
	} else {
		if (apk_version_compare_blob(*pkg->version, *dep->version)
		    & dep->result_mask)
			return 1;
	}
	return 0;
296 297
}

298 299 300 301
void apk_deps_parse(struct apk_database *db,
		    struct apk_dependency_array **depends,
		    apk_blob_t blob)
{
302
	struct parse_depend_ctx ctx = { db, depends };
303

304
	if (blob.len > 0 && blob.ptr[blob.len-1] == '\n')
305
		blob.len--;
306

307
	apk_blob_for_each_segment(blob, " ", parse_depend, &ctx);
308 309
}

310 311 312 313 314 315 316 317 318 319
void apk_blob_push_dep(apk_blob_t *to, struct apk_dependency *dep)
{
	if (dep->result_mask == APK_DEPMASK_CONFLICT)
		apk_blob_push_blob(to, APK_BLOB_PTR_LEN("!", 1));

	apk_blob_push_blob(to, APK_BLOB_STR(dep->name->name));

	if (dep->result_mask != APK_DEPMASK_CONFLICT &&
	    dep->result_mask != APK_DEPMASK_REQUIRE) {
		apk_blob_push_blob(to, APK_BLOB_STR(apk_version_op_string(dep->result_mask)));
320
		apk_blob_push_blob(to, *dep->version);
321 322 323
	}
}

324 325
int apk_deps_write(struct apk_dependency_array *deps, struct apk_ostream *os)
{
326 327 328
	apk_blob_t blob;
	char tmp[256];
	int i, n = 0;
329 330 331 332 333

	if (deps == NULL)
		return 0;

	for (i = 0; i < deps->num; i++) {
334 335 336 337 338 339 340 341 342
		blob = APK_BLOB_BUF(tmp);
		if (i)
			apk_blob_push_blob(&blob, APK_BLOB_PTR_LEN(" ", 1));
		apk_blob_push_dep(&blob, &deps->item[i]);

		blob = apk_blob_pushed(APK_BLOB_BUF(tmp), blob);
		if (APK_BLOB_IS_NULL(blob) || 
		    os->write(os, blob.ptr, blob.len) != blob.len)
			return -1;
343

344
		n += blob.len;
345 346 347 348 349
	}

	return n;
}

Timo Teräs's avatar
Timo Teräs committed
350
const char *apk_script_types[] = {
351 352 353 354 355 356
	[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",
357
	[APK_SCRIPT_TRIGGER]		= "trigger",
358 359 360 361 362 363
};

int apk_script_type(const char *name)
{
	int i;

Timo Teräs's avatar
Timo Teräs committed
364 365 366
	for (i = 0; i < ARRAY_SIZE(apk_script_types); i++)
		if (apk_script_types[i] &&
		    strcmp(apk_script_types[i], name) == 0)
367 368
			return i;

369
	return APK_SCRIPT_INVALID;
370 371
}

372
void apk_sign_ctx_init(struct apk_sign_ctx *ctx, int action,
373
		       struct apk_checksum *identity, int keys_fd)
Timo Teräs's avatar
Timo Teräs committed
374 375
{
	memset(ctx, 0, sizeof(struct apk_sign_ctx));
376
	ctx->keys_fd = keys_fd;
Timo Teräs's avatar
Timo Teräs committed
377 378
	ctx->action = action;
	switch (action) {
Timo Teräs's avatar
Timo Teräs committed
379 380 381
	case APK_SIGN_VERIFY:
		ctx->md = EVP_md_null();
		break;
382 383 384 385 386 387 388 389 390 391
	case APK_SIGN_VERIFY_IDENTITY:
		if (identity->type == APK_CHECKSUM_MD5) {
			ctx->md = EVP_md5();
			ctx->control_started = 1;
			ctx->data_started = 1;
		} else {
			ctx->md = EVP_sha1();
		}
		memcpy(&ctx->identity, identity, sizeof(ctx->identity));
		break;
Timo Teräs's avatar
Timo Teräs committed
392
	case APK_SIGN_GENERATE:
393
	case APK_SIGN_VERIFY_AND_GENERATE:
Timo Teräs's avatar
Timo Teräs committed
394 395
		ctx->md = EVP_sha1();
		break;
396 397 398 399 400 401
	default:
		action = APK_SIGN_NONE;
		ctx->md = EVP_md_null();
		ctx->control_started = 1;
		ctx->data_started = 1;
		break;
Timo Teräs's avatar
Timo Teräs committed
402
	}
403 404
	EVP_MD_CTX_init(&ctx->mdctx);
	EVP_DigestInit_ex(&ctx->mdctx, ctx->md, NULL);
405
	EVP_MD_CTX_set_flags(&ctx->mdctx, EVP_MD_CTX_FLAG_ONESHOT);
Timo Teräs's avatar
Timo Teräs committed
406 407 408 409 410 411 412 413
}

void apk_sign_ctx_free(struct apk_sign_ctx *ctx)
{
	if (ctx->signature.data.ptr != NULL)
		free(ctx->signature.data.ptr);
	if (ctx->signature.pkey != NULL)
		EVP_PKEY_free(ctx->signature.pkey);
414
	EVP_MD_CTX_cleanup(&ctx->mdctx);
Timo Teräs's avatar
Timo Teräs committed
415 416 417 418 419 420 421 422 423
}

int apk_sign_ctx_process_file(struct apk_sign_ctx *ctx,
			      const struct apk_file_info *fi,
			      struct apk_istream *is)
{
	if (ctx->data_started)
		return 1;

424
	if (fi->name[0] != '.' || strchr(fi->name, '/') != NULL) {
Timo Teräs's avatar
Timo Teräs committed
425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441
		ctx->data_started = 1;
		ctx->control_started = 1;
		return 1;
	}

	if (ctx->control_started)
		return 1;

	if (strncmp(fi->name, ".SIGN.", 6) != 0) {
		ctx->control_started = 1;
		return 1;
	}

	/* A signature file */
	ctx->num_signatures++;

	/* Found already a trusted key */
442 443
	if ((ctx->action != APK_SIGN_VERIFY &&
	     ctx->action != APK_SIGN_VERIFY_AND_GENERATE) ||
444
	    ctx->signature.pkey != NULL)
Timo Teräs's avatar
Timo Teräs committed
445 446
		return 0;

447 448 449
	if (ctx->keys_fd < 0)
		return 0;

Timo Teräs's avatar
Timo Teräs committed
450 451
	if (strncmp(&fi->name[6], "RSA.", 4) == 0 ||
	    strncmp(&fi->name[6], "DSA.", 4) == 0) {
452
		int fd = openat(ctx->keys_fd, &fi->name[10], O_RDONLY|O_CLOEXEC);
453 454 455 456 457
	        BIO *bio;

		if (fd < 0)
			return 0;

458
		bio = BIO_new_fp(fdopen(fd, "r"), BIO_CLOSE);
459
		ctx->signature.pkey = PEM_read_bio_PUBKEY(bio, NULL, NULL, NULL);
Timo Teräs's avatar
Timo Teräs committed
460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475
		if (ctx->signature.pkey != NULL) {
			if (fi->name[6] == 'R')
				ctx->md = EVP_sha1();
			else
				ctx->md = EVP_dss1();
		}
		BIO_free(bio);
	} else
		return 0;

	if (ctx->signature.pkey != NULL)
		ctx->signature.data = apk_blob_from_istream(is, fi->size);

	return 0;
}

476
int apk_sign_ctx_parse_pkginfo_line(void *ctx, apk_blob_t line)
477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508
{
	struct apk_sign_ctx *sctx = (struct apk_sign_ctx *) ctx;
	apk_blob_t l, r;

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

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

	if (sctx->data_started == 0 &&
	    apk_blob_compare(APK_BLOB_STR("datahash"), l) == 0) {
		sctx->has_data_checksum = 1;
		sctx->md = EVP_sha256();
		apk_blob_pull_hexdump(
			&r, APK_BLOB_PTR_LEN(sctx->data_checksum,
					     EVP_MD_size(sctx->md)));
	}

	return 0;
}

int apk_sign_ctx_verify_tar(void *sctx, const struct apk_file_info *fi,
			    struct apk_istream *is)
{
	struct apk_sign_ctx *ctx = (struct apk_sign_ctx *) sctx;

	if (apk_sign_ctx_process_file(ctx, fi, is) == 0)
		return 0;

	if (strcmp(fi->name, ".PKGINFO") == 0) {
		apk_blob_t blob = apk_blob_from_istream(is, fi->size);
509 510 511
		apk_blob_for_each_segment(
			blob, "\n",
			apk_sign_ctx_parse_pkginfo_line, ctx);
512 513 514 515 516 517
		free(blob.ptr);
	}

	return 0;
}

518
int apk_sign_ctx_mpart_cb(void *ctx, int part, apk_blob_t data)
Timo Teräs's avatar
Timo Teräs committed
519 520 521
{
	struct apk_sign_ctx *sctx = (struct apk_sign_ctx *) ctx;
	unsigned char calculated[EVP_MAX_MD_SIZE];
Timo Teräs's avatar
Timo Teräs committed
522 523 524 525 526 527 528
	int r, end_of_control;

	if ((part == APK_MPART_DATA) ||
	    (part == APK_MPART_BOUNDARY && sctx->data_started))
		goto update_digest;

	/* Still in signature blocks? */
529 530 531
	if (!sctx->control_started) {
		if (part == APK_MPART_END)
			return -EKEYREJECTED;
Timo Teräs's avatar
Timo Teräs committed
532
		goto reset_digest;
533
	}
Timo Teräs's avatar
Timo Teräs committed
534 535 536 537 538 539

	/* Grab state and mark all remaining block as data */
	end_of_control = (sctx->data_started == 0);
	sctx->data_started = 1;

	/* End of control-block and control does not have data checksum? */
540 541
	if (sctx->has_data_checksum == 0 && end_of_control &&
	    part != APK_MPART_END)
Timo Teräs's avatar
Timo Teräs committed
542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562
		goto update_digest;

	/* Drool in the remaining of the digest block now, we will finish
	 * it on all cases */
	EVP_DigestUpdate(&sctx->mdctx, data.ptr, data.len);

	/* End of control-block and checking control hash/signature or
	 * end of data-block and checking its hash/signature */
	if (sctx->has_data_checksum && !end_of_control) {
		/* End of control-block and check it's hash */
		EVP_DigestFinal_ex(&sctx->mdctx, calculated, NULL);
		if (EVP_MD_CTX_size(&sctx->mdctx) == 0 ||
		    memcmp(calculated, sctx->data_checksum,
		           EVP_MD_CTX_size(&sctx->mdctx)) != 0)
			return -EKEYREJECTED;
		sctx->data_verified = 1;
		if (!(apk_flags & APK_ALLOW_UNTRUSTED) &&
		    !sctx->control_verified)
			return -ENOKEY;
		return 0;
	}
Timo Teräs's avatar
Timo Teräs committed
563

Timo Teräs's avatar
Timo Teräs committed
564 565
	switch (sctx->action) {
	case APK_SIGN_VERIFY:
566
	case APK_SIGN_VERIFY_AND_GENERATE:
Timo Teräs's avatar
Timo Teräs committed
567 568 569 570
		if (sctx->signature.pkey == NULL) {
			if (apk_flags & APK_ALLOW_UNTRUSTED)
				break;
			return -ENOKEY;
Timo Teräs's avatar
Timo Teräs committed
571 572
		}

Timo Teräs's avatar
Timo Teräs committed
573 574 575 576 577 578 579 580
		r = EVP_VerifyFinal(&sctx->mdctx,
			(unsigned char *) sctx->signature.data.ptr,
			sctx->signature.data.len,
			sctx->signature.pkey);
		if (r != 1)
			return -EKEYREJECTED;
		sctx->control_verified = 1;
		if (!sctx->has_data_checksum && part == APK_MPART_END)
581
			sctx->data_verified = 1;
Timo Teräs's avatar
Timo Teräs committed
582 583 584 585 586 587 588 589 590
		break;
	case APK_SIGN_VERIFY_IDENTITY:
		/* Reset digest for hashing data */
		EVP_DigestFinal_ex(&sctx->mdctx, calculated, NULL);
		if (memcmp(calculated, sctx->identity.data,
			   sctx->identity.type) != 0)
			return -EKEYREJECTED;
		sctx->control_verified = 1;
		if (!sctx->has_data_checksum && part == APK_MPART_END)
591
			sctx->data_verified = 1;
Timo Teräs's avatar
Timo Teräs committed
592 593 594 595 596 597 598 599
		break;
	case APK_SIGN_GENERATE:
		/* Package identity is the checksum */
		sctx->identity.type = EVP_MD_CTX_size(&sctx->mdctx);
		EVP_DigestFinal_ex(&sctx->mdctx, sctx->identity.data, NULL);
		if (sctx->action == APK_SIGN_GENERATE &&
		    sctx->has_data_checksum)
			return -ECANCELED;
600
		break;
Timo Teräs's avatar
Timo Teräs committed
601
	}
602 603 604 605
	if (sctx->action == APK_SIGN_VERIFY_AND_GENERATE) {
		sctx->identity.type = EVP_MD_CTX_size(&sctx->mdctx);
		EVP_DigestFinal_ex(&sctx->mdctx, sctx->identity.data, NULL);
	}
Timo Teräs's avatar
Timo Teräs committed
606 607 608 609 610 611 612 613
reset_digest:
	EVP_DigestInit_ex(&sctx->mdctx, sctx->md, NULL);
	EVP_MD_CTX_set_flags(&sctx->mdctx, EVP_MD_CTX_FLAG_ONESHOT);
	return 0;

update_digest:
	EVP_MD_CTX_clear_flags(&sctx->mdctx, EVP_MD_CTX_FLAG_ONESHOT);
	EVP_DigestUpdate(&sctx->mdctx, data.ptr, data.len);
Timo Teräs's avatar
Timo Teräs committed
614 615 616
	return 0;
}

617 618 619
struct read_info_ctx {
	struct apk_database *db;
	struct apk_package *pkg;
Timo Teräs's avatar
Timo Teräs committed
620
	struct apk_sign_ctx *sctx;
621 622
};

623 624
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
625 626 627
{
	switch (field) {
	case 'P':
628
		pkg->name = apk_db_get_name(db, value);
Timo Teräs's avatar
Timo Teräs committed
629 630
		break;
	case 'V':
631
		pkg->version = apk_blob_atomize_dup(value);
Timo Teräs's avatar
Timo Teräs committed
632 633 634 635 636 637 638 639
		break;
	case 'T':
		pkg->description = apk_blob_cstr(value);
		break;
	case 'U':
		pkg->url = apk_blob_cstr(value);
		break;
	case 'L':
640
		pkg->license = apk_blob_atomize_dup(value);
Timo Teräs's avatar
Timo Teräs committed
641
		break;
642
	case 'A':
643
		pkg->arch = apk_blob_atomize_dup(value);
644
		break;
Timo Teräs's avatar
Timo Teräs committed
645 646 647 648
	case 'D':
		apk_deps_parse(db, &pkg->depends, value);
		break;
	case 'C':
Timo Teräs's avatar
Timo Teräs committed
649
		apk_blob_pull_csum(&value, &pkg->csum);
Timo Teräs's avatar
Timo Teräs committed
650 651
		break;
	case 'S':
652
		pkg->size = apk_blob_pull_uint(&value, 10);
Timo Teräs's avatar
Timo Teräs committed
653 654
		break;
	case 'I':
655
		pkg->installed_size = apk_blob_pull_uint(&value, 10);
Timo Teräs's avatar
Timo Teräs committed
656
		break;
657 658
	default:
		return -1;
Timo Teräs's avatar
Timo Teräs committed
659
	}
660 661
	if (APK_BLOB_IS_NULL(value))
		return -1;
Timo Teräs's avatar
Timo Teräs committed
662 663 664 665 666 667 668 669 670 671 672 673 674 675 676
	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' },
677
		{ "arch",    'A' },
Timo Teräs's avatar
Timo Teräs committed
678 679 680 681 682 683 684 685 686
		{ "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;

687
	if (!apk_blob_split(line, APK_BLOB_STR(" = "), &l, &r))
Timo Teräs's avatar
Timo Teräs committed
688 689 690
		return 0;

	for (i = 0; i < ARRAY_SIZE(fields); i++) {
691
		if (apk_blob_compare(APK_BLOB_STR(fields[i].str), l) == 0) {
692
			apk_pkg_add_info(ri->db, ri->pkg, fields[i].field, r);
693
			return 0;
Timo Teräs's avatar
Timo Teräs committed
694 695
		}
	}
696
	apk_sign_ctx_parse_pkginfo_line(ri->sctx, line);
697

Timo Teräs's avatar
Timo Teräs committed
698 699 700
	return 0;
}

701
static int read_info_entry(void *ctx, const struct apk_file_info *ae,
702
			   struct apk_istream *is)
703 704 705 706
{
	struct read_info_ctx *ri = (struct read_info_ctx *) ctx;
	struct apk_package *pkg = ri->pkg;

Timo Teräs's avatar
Timo Teräs committed
707
	/* Meta info and scripts */
Timo Teräs's avatar
Timo Teräs committed
708 709
	if (apk_sign_ctx_process_file(ri->sctx, ae, is) == 0)
		return 0;
710

Timo Teräs's avatar
Timo Teräs committed
711
	if (ae->name[0] == '.') {
Timo Teräs's avatar
Timo Teräs committed
712 713
		/* APK 2.0 format */
		if (strcmp(ae->name, ".PKGINFO") == 0) {
714
			apk_blob_t blob = apk_blob_from_istream(is, ae->size);
Timo Teräs's avatar
Timo Teräs committed
715 716
			apk_blob_for_each_segment(blob, "\n", read_info_line, ctx);
			free(blob.ptr);
717
		} else if (strcmp(ae->name, ".INSTALL") == 0) {
718 719
			apk_warning("Package '%s-%s' contains deprecated .INSTALL",
				    pkg->name->name, pkg->version);
Timo Teräs's avatar
Timo Teräs committed
720
		}
721 722 723
		return 0;
	}

724 725 726
	return 0;
}

727 728
int apk_pkg_read(struct apk_database *db, const char *file,
	         struct apk_sign_ctx *sctx, struct apk_package **pkg)
729 730
{
	struct read_info_ctx ctx;
731
	struct apk_file_info fi;
732
	struct apk_bstream *bs;
733
	struct apk_istream *tar;
734
	int r;
735

736
	r = apk_file_get_info(AT_FDCWD, file, APK_CHECKSUM_NONE, &fi);
737 738
	if (r != 0)
		return r;
739

740
	memset(&ctx, 0, sizeof(ctx));
Timo Teräs's avatar
Timo Teräs committed
741
	ctx.sctx = sctx;
742
	ctx.pkg = apk_pkg_new();
743
	r = -ENOMEM;
744
	if (ctx.pkg == NULL)
745
		goto err;
746
	bs = apk_bstream_from_file(AT_FDCWD, file);
Timo Teräs's avatar
Timo Teräs committed
747
	if (bs == NULL)
748 749 750
		goto err;

	ctx.db = db;
751
	ctx.pkg->size = fi.size;
752

Timo Teräs's avatar
Timo Teräs committed
753
	tar = apk_bstream_gunzip_mpart(bs, apk_sign_ctx_mpart_cb, sctx);
754
	r = apk_tar_parse(tar, read_info_entry, &ctx, FALSE, &db->id_cache);
755
	tar->close(tar);
Timo Teräs's avatar
Timo Teräs committed
756
	if (r < 0 && r != -ECANCELED)
757
		goto err;
758
	if (ctx.pkg->name == NULL) {
759
		r = -ENOMSG;
760
		goto err;
761
	}
Timo Teräs's avatar
Timo Teräs committed
762 763
	if (sctx->action != APK_SIGN_VERIFY)
		ctx.pkg->csum = sctx->identity;
764
	ctx.pkg->filename = strdup(file);
765

766 767 768
	ctx.pkg = apk_db_pkg_add(db, ctx.pkg);
	if (pkg != NULL)
		*pkg = ctx.pkg;
769
	return 0;
770 771
err:
	apk_pkg_free(ctx.pkg);
772
	return r;
773 774 775 776 777 778 779
}

void apk_pkg_free(struct apk_package *pkg)
{
	if (pkg == NULL)
		return;

780
	apk_pkg_uninstall(NULL, pkg);
Timo Teräs's avatar
Timo Teräs committed
781
	apk_dependency_array_free(&pkg->depends);
782 783 784 785 786 787 788
	if (pkg->url)
		free(pkg->url);
	if (pkg->description)
		free(pkg->description);
	free(pkg);
}

789 790 791
int apk_ipkg_add_script(struct apk_installed_package *ipkg,
			struct apk_istream *is,
			unsigned int type, unsigned int size)
792
{
793
	void *ptr;
794 795
	int r;

796 797 798 799 800
	if (type >= APK_SCRIPT_MAX)
		return -1;

	ptr = malloc(size);
	r = is->read(is, ptr, size);
801
	if (r < 0) {
802
		free(ptr);
803 804 805
		return r;
	}

806 807 808 809 810
	if (ipkg->script[type].ptr)
		free(ipkg->script[type].ptr);
	ipkg->script[type].ptr = ptr;
	ipkg->script[type].len = size;
	return 0;
811 812
}

813 814
int apk_ipkg_run_script(struct apk_installed_package *ipkg,
			struct apk_database *db,
815
			unsigned int type, char **argv)
816
{
817
	static char * const environment[] = {
818 819 820
		"PATH=/usr/sbin:/usr/bin:/sbin:/bin",
		NULL
	};
821 822
	struct apk_package *pkg = ipkg->pkg;
	char fn[PATH_MAX];
823
	int fd, status, root_fd = db->root_fd;
824 825
	pid_t pid;

826 827
	if (type >= APK_SCRIPT_MAX)
		return -1;
828

829 830 831 832
	if (ipkg->script[type].ptr == NULL)
		return 0;

	argv[0] = (char *) apk_script_types[type];
833

834
	/* Avoid /tmp as it can be mounted noexec */
835 836
	snprintf(fn, sizeof(fn), "var/cache/misc/" PKG_VER_FMT ".%s",
		PKG_VER_PRINTF(pkg),
837 838
		apk_script_types[type]);

839 840 841 842
	apk_message("Executing %s", &fn[15]);
	if (apk_flags & APK_SIMULATE)
		return 0;

843
	fd = openat(root_fd, fn, O_CREAT|O_RDWR|O_TRUNC|O_CLOEXEC, 0755);
844 845
	if (fd < 0) {
		mkdirat(root_fd, "var/cache/misc", 0755);
846
		fd = openat(root_fd, fn, O_CREAT|O_RDWR|O_TRUNC|O_CLOEXEC, 0755);
847 848 849
		if (fd < 0)
			return -errno;
	}
850 851 852 853
	if (write(fd, ipkg->script[type].ptr, ipkg->script[type].len) < 0) {
		close(fd);
		return -errno;
	}
854
	close(fd);
855

856 857
	pid = fork();
	if (pid == -1)
858
		return -1;
859
	if (pid == 0) {
860
		if (fchdir(root_fd) < 0 || chroot(".") < 0) {
861 862 863 864 865
			apk_error("chroot: %s", strerror(errno));
		} else {
			execve(fn, argv, environment);
		}
		exit(1);
866
	}
867 868
	waitpid(pid, &status, 0);
	unlinkat(root_fd, fn, 0);
869
	apk_id_cache_reset(&db->id_cache);
870

871 872 873
	if (WIFEXITED(status))
		return WEXITSTATUS(status);
	return -1;
874 875
}

Timo Teräs's avatar
Timo Teräs committed
876
static int parse_index_line(void *ctx, apk_blob_t line)
877
{
Timo Teräs's avatar
Timo Teräs committed
878
	struct read_info_ctx *ri = (struct read_info_ctx *) ctx;
879

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

883
	apk_pkg_add_info(ri->db, ri->pkg, line.ptr[0], APK_BLOB_PTR_LEN(line.ptr+2, line.len-2));
884 885 886 887 888
	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
889
	struct read_info_ctx ctx;
890

891
	ctx.pkg = apk_pkg_new();
Timo Teräs's avatar
Timo Teräs committed
892
	if (ctx.pkg == NULL)
893 894
		return NULL;

Timo Teräs's avatar
Timo Teräs committed
895
	ctx.db = db;
896

Timo Teräs's avatar
Timo Teräs committed
897 898 899 900
	apk_blob_for_each_segment(blob, "\n", parse_index_line, &ctx);

	if (ctx.pkg->name == NULL) {
		apk_pkg_free(ctx.pkg);
901 902
		apk_error("Failed to parse index entry: " BLOB_FMT,
			  BLOB_PRINTF(blob));
Timo Teräs's avatar
Timo Teräs committed
903
		ctx.pkg = NULL;
904 905
	}

Timo Teräs's avatar
Timo Teräs committed
906
	return ctx.pkg;
907 908
}

909 910
int apk_pkg_write_index_entry(struct apk_package *info,
			      struct apk_ostream *os)
911
{
912
	char buf[512];
913 914 915 916
	apk_blob_t bbuf = APK_BLOB_BUF(buf);
	int r;

	apk_blob_push_blob(&bbuf, APK_BLOB_STR("C:"));
Timo Teräs's avatar
Timo Teräs committed
917
	apk_blob_push_csum(&bbuf, &info->csum);
918 919 920
	apk_blob_push_blob(&bbuf, APK_BLOB_STR("\nP:"));
	apk_blob_push_blob(&bbuf, APK_BLOB_STR(info->name->name));
	apk_blob_push_blob(&bbuf, APK_BLOB_STR("\nV:"));
921
	apk_blob_push_blob(&bbuf, *info->version);
922 923
	if (info->arch != NULL) {
		apk_blob_push_blob(&bbuf, APK_BLOB_STR("\nA:"));
924
		apk_blob_push_blob(&bbuf, *info->arch);
925
	}
926 927 928 929 930 931 932 933 934
	apk_blob_push_blob(&bbuf, APK_BLOB_STR("\nS:"));
	apk_blob_push_uint(&bbuf, info->size, 10);
	apk_blob_push_blob(&bbuf, APK_BLOB_STR("\nI:"));
	apk_blob_push_uint(&bbuf, info->installed_size, 10);
	apk_blob_push_blob(&bbuf, APK_BLOB_STR("\nT:"));
	apk_blob_push_blob(&bbuf, APK_BLOB_STR(info->description));
	apk_blob_push_blob(&bbuf, APK_BLOB_STR("\nU:"));
	apk_blob_push_blob(&bbuf, APK_BLOB_STR(info->url));
	apk_blob_push_blob(&bbuf, APK_BLOB_STR("\nL:"));
935
	apk_blob_push_blob(&bbuf, *info->license);
936 937
	apk_blob_push_blob(&bbuf, APK_BLOB_STR("\n"));

Timo Teräs's avatar
Timo Teräs committed
938 939 940
	if (APK_BLOB_IS_NULL(bbuf))
		return -1;

941 942
	bbuf = apk_blob_pushed(APK_BLOB_BUF(buf), bbuf);
	if (os->write(os, bbuf.ptr, bbuf.len) != bbuf.len)
943
		return -1;
944

Timo Teräs's avatar
Timo Teräs committed
945
	if (info->depends->num > 0) {
946 947 948 949 950 951 952
		if (os->write(os, "D:", 2) != 2)
			return -1;
		r = apk_deps_write(info->depends, os);
		if (r < 0)
			return r;
		if (os->write(os, "\n", 1) != 1)
			return -1;
953 954
	}

955
	return 0;
956
}
957

958 959
int apk_pkg_version_compare(struct apk_package *a, struct apk_package *b)
{
960 961 962 963
	if (a->version == b->version)
		return APK_VERSION_EQUAL;

	return apk_version_compare_blob(*a->version, *b->version);
964
}