package.c 22 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 225 226
			case '<':
				mask |= APK_VERSION_LESS;
				break;
			case '>':
				mask |= APK_VERSION_GREATER;
				break;
			case '=':
				mask |= APK_VERSION_EQUAL;
				break;
			}
		}
		if ((mask & (APK_VERSION_LESS|APK_VERSION_GREATER))
		    == (APK_VERSION_LESS|APK_VERSION_GREATER))
Timo Teräs's avatar
Timo Teräs committed
227
			return -EINVAL;
228 229

		if (!apk_version_validate(bver))
Timo Teräs's avatar
Timo Teräs committed
230 231 232
			return -EINVAL;

		blob = bname;
233 234
	}

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

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

void apk_dep_from_pkg(struct apk_dependency *dep, struct apk_database *db,
		      struct apk_package *pkg)
{
	*dep = (struct apk_dependency) {
251
		.name = pkg->name,
Timo Teräs's avatar
Timo Teräs committed
252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271
		.version = pkg->version,
		.result_mask = APK_VERSION_EQUAL,
	};
}

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;
272 273 274 275

	return 0;
}

276 277 278 279
void apk_deps_parse(struct apk_database *db,
		    struct apk_dependency_array **depends,
		    apk_blob_t blob)
{
280
	struct parse_depend_ctx ctx = { db, depends };
281

282
	if (blob.len > 0 && blob.ptr[blob.len-1] == '\n')
283
		blob.len--;
284

285
	apk_blob_for_each_segment(blob, " ", parse_depend, &ctx);
286 287
}

288 289 290 291 292 293 294 295 296 297
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)));
298
		apk_blob_push_blob(to, *dep->version);
299 300 301
	}
}

302 303
int apk_deps_write(struct apk_dependency_array *deps, struct apk_ostream *os)
{
304 305 306
	apk_blob_t blob;
	char tmp[256];
	int i, n = 0;
307 308 309 310 311

	if (deps == NULL)
		return 0;

	for (i = 0; i < deps->num; i++) {
312 313 314 315 316 317 318 319 320
		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;
321

322
		n += blob.len;
323 324 325 326 327
	}

	return n;
}

Timo Teräs's avatar
Timo Teräs committed
328
const char *apk_script_types[] = {
329 330 331 332 333 334
	[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",
335
	[APK_SCRIPT_TRIGGER]		= "trigger",
336 337 338 339 340 341
};

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

Timo Teräs's avatar
Timo Teräs committed
342 343 344
	for (i = 0; i < ARRAY_SIZE(apk_script_types); i++)
		if (apk_script_types[i] &&
		    strcmp(apk_script_types[i], name) == 0)
345 346
			return i;

347
	return APK_SCRIPT_INVALID;
348 349
}

350
void apk_sign_ctx_init(struct apk_sign_ctx *ctx, int action,
351
		       struct apk_checksum *identity, int keys_fd)
Timo Teräs's avatar
Timo Teräs committed
352 353
{
	memset(ctx, 0, sizeof(struct apk_sign_ctx));
354
	ctx->keys_fd = keys_fd;
Timo Teräs's avatar
Timo Teräs committed
355 356
	ctx->action = action;
	switch (action) {
Timo Teräs's avatar
Timo Teräs committed
357 358 359
	case APK_SIGN_VERIFY:
		ctx->md = EVP_md_null();
		break;
360 361 362 363 364 365 366 367 368 369
	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
370
	case APK_SIGN_GENERATE:
371
	case APK_SIGN_VERIFY_AND_GENERATE:
Timo Teräs's avatar
Timo Teräs committed
372 373
		ctx->md = EVP_sha1();
		break;
374 375 376 377 378 379
	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
380
	}
381 382
	EVP_MD_CTX_init(&ctx->mdctx);
	EVP_DigestInit_ex(&ctx->mdctx, ctx->md, NULL);
383
	EVP_MD_CTX_set_flags(&ctx->mdctx, EVP_MD_CTX_FLAG_ONESHOT);
Timo Teräs's avatar
Timo Teräs committed
384 385 386 387 388 389 390 391
}

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);
392
	EVP_MD_CTX_cleanup(&ctx->mdctx);
Timo Teräs's avatar
Timo Teräs committed
393 394 395 396 397 398 399 400 401
}

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;

402
	if (fi->name[0] != '.' || strchr(fi->name, '/') != NULL) {
Timo Teräs's avatar
Timo Teräs committed
403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419
		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 */
420 421
	if ((ctx->action != APK_SIGN_VERIFY &&
	     ctx->action != APK_SIGN_VERIFY_AND_GENERATE) ||
422
	    ctx->signature.pkey != NULL)
Timo Teräs's avatar
Timo Teräs committed
423 424
		return 0;

425 426 427
	if (ctx->keys_fd < 0)
		return 0;

Timo Teräs's avatar
Timo Teräs committed
428 429
	if (strncmp(&fi->name[6], "RSA.", 4) == 0 ||
	    strncmp(&fi->name[6], "DSA.", 4) == 0) {
430
		int fd = openat(ctx->keys_fd, &fi->name[10], O_RDONLY|O_CLOEXEC);
431 432 433 434 435
	        BIO *bio;

		if (fd < 0)
			return 0;

436
		bio = BIO_new_fp(fdopen(fd, "r"), BIO_CLOSE);
437
		ctx->signature.pkey = PEM_read_bio_PUBKEY(bio, NULL, NULL, NULL);
Timo Teräs's avatar
Timo Teräs committed
438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453
		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;
}

454
int apk_sign_ctx_parse_pkginfo_line(void *ctx, apk_blob_t line)
455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486
{
	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);
487 488 489
		apk_blob_for_each_segment(
			blob, "\n",
			apk_sign_ctx_parse_pkginfo_line, ctx);
490 491 492 493 494 495
		free(blob.ptr);
	}

	return 0;
}

496
int apk_sign_ctx_mpart_cb(void *ctx, int part, apk_blob_t data)
Timo Teräs's avatar
Timo Teräs committed
497 498 499
{
	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
500 501 502 503 504 505 506
	int r, end_of_control;

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

	/* Still in signature blocks? */
507 508 509
	if (!sctx->control_started) {
		if (part == APK_MPART_END)
			return -EKEYREJECTED;
Timo Teräs's avatar
Timo Teräs committed
510
		goto reset_digest;
511
	}
Timo Teräs's avatar
Timo Teräs committed
512 513 514 515 516 517

	/* 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? */
518 519
	if (sctx->has_data_checksum == 0 && end_of_control &&
	    part != APK_MPART_END)
Timo Teräs's avatar
Timo Teräs committed
520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540
		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
541

Timo Teräs's avatar
Timo Teräs committed
542 543
	switch (sctx->action) {
	case APK_SIGN_VERIFY:
544
	case APK_SIGN_VERIFY_AND_GENERATE:
Timo Teräs's avatar
Timo Teräs committed
545 546 547 548
		if (sctx->signature.pkey == NULL) {
			if (apk_flags & APK_ALLOW_UNTRUSTED)
				break;
			return -ENOKEY;
Timo Teräs's avatar
Timo Teräs committed
549 550
		}

Timo Teräs's avatar
Timo Teräs committed
551 552 553 554 555 556 557 558
		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)
559
			sctx->data_verified = 1;
Timo Teräs's avatar
Timo Teräs committed
560 561 562 563 564 565 566 567 568
		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)
569
			sctx->data_verified = 1;
Timo Teräs's avatar
Timo Teräs committed
570 571 572 573 574 575 576 577
		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;
578
		break;
Timo Teräs's avatar
Timo Teräs committed
579
	}
580 581 582 583
	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
584 585 586 587 588 589 590 591
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
592 593 594
	return 0;
}

595 596 597
struct read_info_ctx {
	struct apk_database *db;
	struct apk_package *pkg;
Timo Teräs's avatar
Timo Teräs committed
598
	struct apk_sign_ctx *sctx;
599 600
};

601 602
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
603 604 605
{
	switch (field) {
	case 'P':
606
		pkg->name = apk_db_get_name(db, value);
Timo Teräs's avatar
Timo Teräs committed
607 608
		break;
	case 'V':
609
		pkg->version = apk_blob_atomize_dup(value);
Timo Teräs's avatar
Timo Teräs committed
610 611 612 613 614 615 616 617
		break;
	case 'T':
		pkg->description = apk_blob_cstr(value);
		break;
	case 'U':
		pkg->url = apk_blob_cstr(value);
		break;
	case 'L':
618
		pkg->license = apk_blob_atomize_dup(value);
Timo Teräs's avatar
Timo Teräs committed
619
		break;
620
	case 'A':
621
		pkg->arch = apk_blob_atomize_dup(value);
622
		break;
Timo Teräs's avatar
Timo Teräs committed
623 624 625 626
	case 'D':
		apk_deps_parse(db, &pkg->depends, value);
		break;
	case 'C':
Timo Teräs's avatar
Timo Teräs committed
627
		apk_blob_pull_csum(&value, &pkg->csum);
Timo Teräs's avatar
Timo Teräs committed
628 629
		break;
	case 'S':
630
		pkg->size = apk_blob_pull_uint(&value, 10);
Timo Teräs's avatar
Timo Teräs committed
631 632
		break;
	case 'I':
633
		pkg->installed_size = apk_blob_pull_uint(&value, 10);
Timo Teräs's avatar
Timo Teräs committed
634
		break;
635 636
	default:
		return -1;
Timo Teräs's avatar
Timo Teräs committed
637
	}
638 639
	if (APK_BLOB_IS_NULL(value))
		return -1;
Timo Teräs's avatar
Timo Teräs committed
640 641 642 643 644 645 646 647 648 649 650 651 652 653 654
	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' },
655
		{ "arch",    'A' },
Timo Teräs's avatar
Timo Teräs committed
656 657 658 659 660 661 662 663 664
		{ "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;

665
	if (!apk_blob_split(line, APK_BLOB_STR(" = "), &l, &r))
Timo Teräs's avatar
Timo Teräs committed
666 667 668
		return 0;

	for (i = 0; i < ARRAY_SIZE(fields); i++) {
669
		if (apk_blob_compare(APK_BLOB_STR(fields[i].str), l) == 0) {
670
			apk_pkg_add_info(ri->db, ri->pkg, fields[i].field, r);
671
			return 0;
Timo Teräs's avatar
Timo Teräs committed
672 673
		}
	}
674
	apk_sign_ctx_parse_pkginfo_line(ri->sctx, line);
675

Timo Teräs's avatar
Timo Teräs committed
676 677 678
	return 0;
}

679
static int read_info_entry(void *ctx, const struct apk_file_info *ae,
680
			   struct apk_istream *is)
681 682 683 684
{
	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
685
	/* Meta info and scripts */
Timo Teräs's avatar
Timo Teräs committed
686 687
	if (apk_sign_ctx_process_file(ri->sctx, ae, is) == 0)
		return 0;
688

Timo Teräs's avatar
Timo Teräs committed
689
	if (ae->name[0] == '.') {
Timo Teräs's avatar
Timo Teräs committed
690 691
		/* APK 2.0 format */
		if (strcmp(ae->name, ".PKGINFO") == 0) {
692
			apk_blob_t blob = apk_blob_from_istream(is, ae->size);
Timo Teräs's avatar
Timo Teräs committed
693 694
			apk_blob_for_each_segment(blob, "\n", read_info_line, ctx);
			free(blob.ptr);
695
		} else if (strcmp(ae->name, ".INSTALL") == 0) {
696 697
			apk_warning("Package '%s-%s' contains deprecated .INSTALL",
				    pkg->name->name, pkg->version);
Timo Teräs's avatar
Timo Teräs committed
698
		}
699 700 701
		return 0;
	}

702 703 704
	return 0;
}

705 706
int apk_pkg_read(struct apk_database *db, const char *file,
	         struct apk_sign_ctx *sctx, struct apk_package **pkg)
707 708
{
	struct read_info_ctx ctx;
709
	struct apk_file_info fi;
710
	struct apk_bstream *bs;
711
	struct apk_istream *tar;
712
	int r;
713

714
	r = apk_file_get_info(AT_FDCWD, file, APK_CHECKSUM_NONE, &fi);
715 716
	if (r != 0)
		return r;
717

718
	memset(&ctx, 0, sizeof(ctx));
Timo Teräs's avatar
Timo Teräs committed
719
	ctx.sctx = sctx;
720
	ctx.pkg = apk_pkg_new();
721
	r = -ENOMEM;
722
	if (ctx.pkg == NULL)
723
		goto err;
724
	bs = apk_bstream_from_file(AT_FDCWD, file);
Timo Teräs's avatar
Timo Teräs committed
725
	if (bs == NULL)
726 727 728
		goto err;

	ctx.db = db;
729
	ctx.pkg->size = fi.size;
730

Timo Teräs's avatar
Timo Teräs committed
731
	tar = apk_bstream_gunzip_mpart(bs, apk_sign_ctx_mpart_cb, sctx);
732
	r = apk_tar_parse(tar, read_info_entry, &ctx, FALSE, &db->id_cache);
733
	tar->close(tar);
Timo Teräs's avatar
Timo Teräs committed
734
	if (r < 0 && r != -ECANCELED)
735
		goto err;
736
	if (ctx.pkg->name == NULL) {
737
		r = -ENOMSG;
738
		goto err;
739
	}
Timo Teräs's avatar
Timo Teräs committed
740 741
	if (sctx->action != APK_SIGN_VERIFY)
		ctx.pkg->csum = sctx->identity;
742
	ctx.pkg->filename = strdup(file);
743

744 745 746
	ctx.pkg = apk_db_pkg_add(db, ctx.pkg);
	if (pkg != NULL)
		*pkg = ctx.pkg;
747
	return 0;
748 749
err:
	apk_pkg_free(ctx.pkg);
750
	return r;
751 752 753 754 755 756 757
}

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

758
	apk_pkg_uninstall(NULL, pkg);
Timo Teräs's avatar
Timo Teräs committed
759
	apk_dependency_array_free(&pkg->depends);
760 761 762 763 764 765 766
	if (pkg->url)
		free(pkg->url);
	if (pkg->description)
		free(pkg->description);
	free(pkg);
}

767 768 769
int apk_ipkg_add_script(struct apk_installed_package *ipkg,
			struct apk_istream *is,
			unsigned int type, unsigned int size)
770
{
771
	void *ptr;
772 773
	int r;

774 775 776 777 778
	if (type >= APK_SCRIPT_MAX)
		return -1;

	ptr = malloc(size);
	r = is->read(is, ptr, size);
779
	if (r < 0) {
780
		free(ptr);
781 782 783
		return r;
	}

784 785 786 787 788
	if (ipkg->script[type].ptr)
		free(ipkg->script[type].ptr);
	ipkg->script[type].ptr = ptr;
	ipkg->script[type].len = size;
	return 0;
789 790
}

791 792
int apk_ipkg_run_script(struct apk_installed_package *ipkg,
			struct apk_database *db,
793
			unsigned int type, char **argv)
794
{
795
	static char * const environment[] = {
796 797 798
		"PATH=/usr/sbin:/usr/bin:/sbin:/bin",
		NULL
	};
799 800
	struct apk_package *pkg = ipkg->pkg;
	char fn[PATH_MAX];
801
	int fd, status, root_fd = db->root_fd;
802 803
	pid_t pid;

804 805
	if (type >= APK_SCRIPT_MAX)
		return -1;
806

807 808 809 810
	if (ipkg->script[type].ptr == NULL)
		return 0;

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

812
	/* Avoid /tmp as it can be mounted noexec */
813 814
	snprintf(fn, sizeof(fn), "var/cache/misc/" PKG_VER_FMT ".%s",
		PKG_VER_PRINTF(pkg),
815 816
		apk_script_types[type]);

817 818 819 820
	apk_message("Executing %s", &fn[15]);
	if (apk_flags & APK_SIMULATE)
		return 0;

821
	fd = openat(root_fd, fn, O_CREAT|O_RDWR|O_TRUNC|O_CLOEXEC, 0755);
822 823
	if (fd < 0) {
		mkdirat(root_fd, "var/cache/misc", 0755);
824
		fd = openat(root_fd, fn, O_CREAT|O_RDWR|O_TRUNC|O_CLOEXEC, 0755);
825 826 827
		if (fd < 0)
			return -errno;
	}
828 829 830 831
	if (write(fd, ipkg->script[type].ptr, ipkg->script[type].len) < 0) {
		close(fd);
		return -errno;
	}
832
	close(fd);
833

834 835
	pid = fork();
	if (pid == -1)
836
		return -1;
837
	if (pid == 0) {
838
		if (fchdir(root_fd) < 0 || chroot(".") < 0) {
839 840 841 842 843
			apk_error("chroot: %s", strerror(errno));
		} else {
			execve(fn, argv, environment);
		}
		exit(1);
844
	}
845 846
	waitpid(pid, &status, 0);
	unlinkat(root_fd, fn, 0);
847
	apk_id_cache_reset(&db->id_cache);
848

849 850 851
	if (WIFEXITED(status))
		return WEXITSTATUS(status);
	return -1;
852 853
}

Timo Teräs's avatar
Timo Teräs committed
854
static int parse_index_line(void *ctx, apk_blob_t line)
855
{
Timo Teräs's avatar
Timo Teräs committed
856
	struct read_info_ctx *ri = (struct read_info_ctx *) ctx;
857

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

861
	apk_pkg_add_info(ri->db, ri->pkg, line.ptr[0], APK_BLOB_PTR_LEN(line.ptr+2, line.len-2));
862 863 864 865 866
	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
867
	struct read_info_ctx ctx;
868

869
	ctx.pkg = apk_pkg_new();
Timo Teräs's avatar
Timo Teräs committed
870
	if (ctx.pkg == NULL)
871 872
		return NULL;

Timo Teräs's avatar
Timo Teräs committed
873
	ctx.db = db;
874

Timo Teräs's avatar
Timo Teräs committed
875 876 877 878
	apk_blob_for_each_segment(blob, "\n", parse_index_line, &ctx);

	if (ctx.pkg->name == NULL) {
		apk_pkg_free(ctx.pkg);
879 880
		apk_error("Failed to parse index entry: " BLOB_FMT,
			  BLOB_PRINTF(blob));
Timo Teräs's avatar
Timo Teräs committed
881
		ctx.pkg = NULL;
882 883
	}

Timo Teräs's avatar
Timo Teräs committed
884
	return ctx.pkg;
885 886
}

887 888
int apk_pkg_write_index_entry(struct apk_package *info,
			      struct apk_ostream *os)
889
{
890
	char buf[512];
891 892 893 894
	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
895
	apk_blob_push_csum(&bbuf, &info->csum);
896 897 898
	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:"));
899
	apk_blob_push_blob(&bbuf, *info->version);
900 901
	if (info->arch != NULL) {
		apk_blob_push_blob(&bbuf, APK_BLOB_STR("\nA:"));
902
		apk_blob_push_blob(&bbuf, *info->arch);
903
	}
904 905 906 907 908 909 910 911 912
	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:"));
913
	apk_blob_push_blob(&bbuf, *info->license);
914 915
	apk_blob_push_blob(&bbuf, APK_BLOB_STR("\n"));

Timo Teräs's avatar
Timo Teräs committed
916 917 918
	if (APK_BLOB_IS_NULL(bbuf))
		return -1;

919 920
	bbuf = apk_blob_pushed(APK_BLOB_BUF(buf), bbuf);
	if (os->write(os, bbuf.ptr, bbuf.len) != bbuf.len)
921
		return -1;
922

Timo Teräs's avatar
Timo Teräs committed
923
	if (info->depends->num > 0) {
924 925 926 927 928 929 930
		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;
931 932
	}

933
	return 0;
934
}
935

936 937
int apk_pkg_version_compare(struct apk_package *a, struct apk_package *b)
{
938 939 940 941
	if (a->version == b->version)
		return APK_VERSION_EQUAL;

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