database.c 67 KB
Newer Older
1 2 3
/* database.c - Alpine Package Keeper (APK)
 *
 * Copyright (C) 2005-2008 Natanael Copa <n@tanael.org>
4
 * Copyright (C) 2008-2011 Timo Teräs <timo.teras@iki.fi>
5 6
 * All rights reserved.
 *
7
 * This program is free software; you can redistribute it and/or modify it
8 9 10 11 12 13 14
 * 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.
 */

#include <errno.h>
#include <stdio.h>
#include <fcntl.h>
15
#include <mntent.h>
16
#include <limits.h>
17 18 19
#include <unistd.h>
#include <malloc.h>
#include <string.h>
20
#include <stdlib.h>
21
#include <signal.h>
22
#include <fnmatch.h>
Timo Teräs's avatar
Timo Teräs committed
23
#include <sys/vfs.h>
24
#include <sys/file.h>
25
#include <sys/wait.h>
Natanael Copa's avatar
Natanael Copa committed
26
#include <sys/stat.h>
27
#include <sys/statvfs.h>
28 29 30 31

#include "apk_defines.h"
#include "apk_package.h"
#include "apk_database.h"
32
#include "apk_applet.h"
33
#include "apk_archive.h"
Natanael Copa's avatar
Natanael Copa committed
34 35
#include "apk_print.h"

Timo Teräs's avatar
Timo Teräs committed
36 37 38 39 40
static const apk_spn_match_def apk_spn_repo_separators = {
	[4] = (1<<0) /* */,
	[7] = (1<<2) /*:*/,
};

41 42 43 44 45
enum {
	APK_DISALLOW_RMDIR = 0,
	APK_ALLOW_RMDIR = 1
};

Natanael Copa's avatar
Natanael Copa committed
46 47
int apk_verbosity = 1;
unsigned int apk_flags = 0;
48

49
const char * const apkindex_tar_gz = "APKINDEX.tar.gz";
Timo Teräs's avatar
Timo Teräs committed
50
static const char * const apk_static_cache_dir = "var/cache/apk";
Timo Teräs's avatar
Timo Teräs committed
51
static const char * const apk_linked_cache_dir = "etc/apk/cache";
Timo Teräs's avatar
Timo Teräs committed
52

Timo Teräs's avatar
Timo Teräs committed
53
static const char * const apk_lock_file = "var/lock/apkdb";
Timo Teräs's avatar
Timo Teräs committed
54

Timo Teräs's avatar
Timo Teräs committed
55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
static const char * const apk_world_file = "etc/apk/world";
static const char * const apk_world_file_tmp = "etc/apk/world.new";
static const char * const apk_world_file_old = "var/lib/apk/world";

static const char * const apk_scripts_file = "lib/apk/db/scripts.tar";
static const char * const apk_scripts_file_tmp = "lib/apk/db/scripts.tar.new";
static const char * const apk_scripts_file_old = "var/lib/apk/scripts.tar";

static const char * const apk_triggers_file = "lib/apk/db/triggers";
static const char * const apk_triggers_file_tmp = "lib/apk/db/triggers.new";
static const char * const apk_triggers_file_old = "var/lib/apk/triggers";

static const char * const apk_installed_file = "lib/apk/db/installed";
static const char * const apk_installed_file_tmp = "lib/apk/db/installed.new";
static const char * const apk_installed_file_old = "var/lib/apk/installed";

71 72 73
struct install_ctx {
	struct apk_database *db;
	struct apk_package *pkg;
74
	struct apk_installed_package *ipkg;
75 76

	int script;
77
	char **script_args;
78
	int script_pending : 1;
79

80
	struct apk_db_dir_instance *diri;
Timo Teräs's avatar
Timo Teräs committed
81
	struct apk_checksum data_csum;
82
	struct apk_sign_ctx sctx;
83

84 85 86 87 88
	apk_progress_cb cb;
	void *cb_ctx;
	size_t installed_size;
	size_t current_file_size;

89 90
	struct hlist_node **diri_node;
	struct hlist_node **file_diri_node;
91 92
};

93
static apk_blob_t pkg_name_get_key(apk_hash_item item)
94
{
95
	return APK_BLOB_STR(((struct apk_name *) item)->name);
96 97
}

98 99 100
static void pkg_name_free(struct apk_name *name)
{
	free(name->name);
Timo Teräs's avatar
Timo Teräs committed
101 102
	apk_package_array_free(&name->pkgs);
	apk_name_array_free(&name->rdepends);
Timo Teräs's avatar
Timo Teräs committed
103
	apk_name_array_free(&name->rinstall_if);
104 105 106
	free(name);
}

107 108 109
static const struct apk_hash_ops pkg_name_hash_ops = {
	.node_offset = offsetof(struct apk_name, hash_node),
	.get_key = pkg_name_get_key,
110 111
	.hash_key = apk_blob_hash,
	.compare = apk_blob_compare,
112
	.delete_item = (apk_hash_delete_f) pkg_name_free,
113 114
};

115
static apk_blob_t pkg_info_get_key(apk_hash_item item)
116
{
Timo Teräs's avatar
Timo Teräs committed
117
	return APK_BLOB_CSUM(((struct apk_package *) item)->csum);
118 119
}

120
static unsigned long csum_hash(apk_blob_t csum)
121
{
122 123
	/* Checksum's highest bits have the most "randomness", use that
	 * directly as hash */
124
	return *(unsigned long *) csum.ptr;
125 126 127 128 129
}

static const struct apk_hash_ops pkg_info_hash_ops = {
	.node_offset = offsetof(struct apk_package, hash_node),
	.get_key = pkg_info_get_key,
130 131
	.hash_key = csum_hash,
	.compare = apk_blob_compare,
132 133 134
	.delete_item = (apk_hash_delete_f) apk_pkg_free,
};

135
static apk_blob_t apk_db_dir_get_key(apk_hash_item item)
136
{
Timo Teräs's avatar
Timo Teräs committed
137 138
	struct apk_db_dir *dir = (struct apk_db_dir *) item;
	return APK_BLOB_PTR_LEN(dir->name, dir->namelen);
139 140 141 142 143
}

static const struct apk_hash_ops dir_hash_ops = {
	.node_offset = offsetof(struct apk_db_dir, hash_node),
	.get_key = apk_db_dir_get_key,
144 145
	.hash_key = apk_blob_hash,
	.compare = apk_blob_compare,
146 147 148
	.delete_item = (apk_hash_delete_f) free,
};

149 150 151 152 153 154
struct apk_db_file_hash_key {
	apk_blob_t dirname;
	apk_blob_t filename;
};

static unsigned long apk_db_file_hash_key(apk_blob_t _key)
155
{
156 157
	struct apk_db_file_hash_key *key = (struct apk_db_file_hash_key *) _key.ptr;

Timo Teräs's avatar
Timo Teräs committed
158
	return apk_blob_hash_seed(key->filename, apk_blob_hash(key->dirname));
159 160 161 162 163 164
}

static unsigned long apk_db_file_hash_item(apk_hash_item item)
{
	struct apk_db_file *dbf = (struct apk_db_file *) item;

Timo Teräs's avatar
Timo Teräs committed
165 166
	return apk_blob_hash_seed(APK_BLOB_PTR_LEN(dbf->name, dbf->namelen),
				  dbf->diri->dir->hash);
167 168 169 170 171 172
}

static int apk_db_file_compare_item(apk_hash_item item, apk_blob_t _key)
{
	struct apk_db_file *dbf = (struct apk_db_file *) item;
	struct apk_db_file_hash_key *key = (struct apk_db_file_hash_key *) _key.ptr;
Timo Teräs's avatar
Timo Teräs committed
173
	struct apk_db_dir *dir = dbf->diri->dir;
174 175
	int r;

Timo Teräs's avatar
Timo Teräs committed
176 177
	r = apk_blob_compare(key->filename,
			     APK_BLOB_PTR_LEN(dbf->name, dbf->namelen));
178 179 180
	if (r != 0)
		return r;

Timo Teräs's avatar
Timo Teräs committed
181 182 183
	r = apk_blob_compare(key->dirname,
			     APK_BLOB_PTR_LEN(dir->name, dir->namelen));
	return r;
184 185 186 187
}

static const struct apk_hash_ops file_hash_ops = {
	.node_offset = offsetof(struct apk_db_file, hash_node),
188 189 190
	.hash_key = apk_db_file_hash_key,
	.hash_item = apk_db_file_hash_item,
	.compare_item = apk_db_file_compare_item,
191 192 193
	.delete_item = (apk_hash_delete_f) free,
};

Timo Teräs's avatar
Timo Teräs committed
194 195 196 197 198
struct apk_name *apk_db_query_name(struct apk_database *db, apk_blob_t name)
{
	return (struct apk_name *) apk_hash_get(&db->available.names, name);
}

199
struct apk_name *apk_db_get_name(struct apk_database *db, apk_blob_t name)
200 201
{
	struct apk_name *pn;
202
	unsigned long hash = apk_hash_from_key(&db->available.names, name);
203

204
	pn = (struct apk_name *) apk_hash_get_hashed(&db->available.names, name, hash);
205 206 207 208 209 210 211
	if (pn != NULL)
		return pn;

	pn = calloc(1, sizeof(struct apk_name));
	if (pn == NULL)
		return NULL;

212
	pn->name = apk_blob_cstr(name);
Timo Teräs's avatar
Timo Teräs committed
213 214
	apk_package_array_init(&pn->pkgs);
	apk_name_array_init(&pn->rdepends);
Timo Teräs's avatar
Timo Teräs committed
215
	apk_name_array_init(&pn->rinstall_if);
216
	apk_hash_insert_hashed(&db->available.names, pn, hash);
217 218 219 220

	return pn;
}

221 222
static void apk_db_dir_mkdir(struct apk_database *db, struct apk_db_dir *dir)
{
223 224 225 226 227 228 229
	if (apk_flags & APK_SIMULATE)
		return;

	/* Don't mess with root, as no package provides it directly */
	if (dir->namelen == 0)
		return;

230 231 232 233 234 235 236 237 238 239 240 241
	if ((dir->refs == 1) ||
	    (fchmodat(db->root_fd, dir->name, dir->mode, AT_SYMLINK_NOFOLLOW) != 0 &&
	     errno == ENOENT))
		if ((mkdirat(db->root_fd, dir->name, dir->mode) != 0 &&
		     errno == EEXIST))
			if (fchmodat(db->root_fd, dir->name, dir->mode, AT_SYMLINK_NOFOLLOW) != 0)
				;

	if (fchownat(db->root_fd, dir->name, dir->uid, dir->gid, 0) != 0)
		;
}

242 243
static void apk_db_dir_unref(struct apk_database *db, struct apk_db_dir *dir,
			     int allow_rmdir)
244 245
{
	dir->refs--;
246 247
	if (dir->refs > 0) {
		if (allow_rmdir) {
248
			dir->recalc_mode = 1;
249 250 251
			dir->mode = 0;
			dir->uid = (uid_t) -1;
			dir->gid = (gid_t) -1;
252
		}
253
		return;
254 255
	}

256
	db->installed.stats.dirs--;
257 258

	if (allow_rmdir) {
259 260
		/* The final instance of this directory was removed,
		 * so this directory gets deleted in reality too. */
261
		dir->recalc_mode = 0;
262 263 264 265
		dir->mode = 0;
		dir->uid = (uid_t) -1;
		dir->gid = (gid_t) -1;

266
		unlinkat(db->root_fd, dir->name, AT_REMOVEDIR);
267
	} else if (dir->recalc_mode) {
268 269
		/* Directory permissions need a reset. */
		apk_db_dir_mkdir(db, dir);
270
	}
271 272

	if (dir->parent != NULL)
273
		apk_db_dir_unref(db, dir->parent, allow_rmdir);
274 275
}

276
static struct apk_db_dir *apk_db_dir_ref(struct apk_db_dir *dir)
277 278 279 280 281
{
	dir->refs++;
	return dir;
}

282 283
struct apk_db_dir *apk_db_dir_query(struct apk_database *db,
				    apk_blob_t name)
284 285 286 287
{
	return (struct apk_db_dir *) apk_hash_get(&db->installed.dirs, name);
}

288
struct apk_db_dir *apk_db_dir_get(struct apk_database *db, apk_blob_t name)
289 290
{
	struct apk_db_dir *dir;
291
	struct apk_protected_path_array *ppaths;
292
	apk_blob_t bparent;
293
	unsigned long hash = apk_hash_from_key(&db->installed.dirs, name);
294
	char *relative_name;
295
	int i;
296

297
	if (name.len && name.ptr[name.len-1] == '/')
298 299
		name.len--;

300
	dir = (struct apk_db_dir *) apk_hash_get_hashed(&db->installed.dirs, name, hash);
301
	if (dir != NULL)
302
		return apk_db_dir_ref(dir);
303

304
	db->installed.stats.dirs++;
305 306
	dir = malloc(sizeof(*dir) + name.len + 1);
	memset(dir, 0, sizeof(*dir));
307
	dir->refs = 1;
308 309
	dir->uid = (uid_t) -1;
	dir->gid = (gid_t) -1;
310
	dir->rooted_name[0] = '/';
Timo Teräs's avatar
Timo Teräs committed
311 312 313 314
	memcpy(dir->name, name.ptr, name.len);
	dir->name[name.len] = 0;
	dir->namelen = name.len;
	dir->hash = hash;
315
	apk_protected_path_array_init(&dir->protected_paths);
316
	apk_hash_insert_hashed(&db->installed.dirs, dir, hash);
317

318
	if (name.len == 0) {
319
		dir->parent = NULL;
320 321 322
		dir->has_protected_children = 1;
		ppaths = NULL;
	} else if (apk_blob_rsplit(name, '/', &bparent, NULL)) {
323
		dir->parent = apk_db_dir_get(db, bparent);
324 325 326 327 328
		dir->protected = dir->parent->protected;
		dir->has_protected_children = dir->protected;
		dir->symlinks_only = dir->parent->symlinks_only;
		ppaths = dir->parent->protected_paths;
	} else {
329
		dir->parent = apk_db_dir_get(db, APK_BLOB_NULL);
330 331
		ppaths = db->protected_paths;
	}
332

333 334
	if (ppaths == NULL)
		return dir;
335

336 337 338 339
	relative_name = strrchr(dir->rooted_name, '/') + 1;
	for (i = 0; i < ppaths->num; i++) {
		struct apk_protected_path *ppath = &ppaths->item[i];
		char *slash;
340

341 342 343 344 345
		slash = strchr(ppath->relative_pattern, '/');
		if (slash != NULL) {
			*slash = 0;
			if (fnmatch(ppath->relative_pattern, relative_name, FNM_PATHNAME) != 0) {
				*slash = '/';
346 347
				continue;
			}
348 349 350 351 352 353 354 355 356 357 358
			*slash = '/';

			*apk_protected_path_array_add(&dir->protected_paths) = (struct apk_protected_path) {
				.relative_pattern = slash + 1,
				.protected = ppath->protected,
				.symlinks_only = ppath->symlinks_only,
			};
			dir->has_protected_children |= ppath->protected;
		} else {
			if (fnmatch(ppath->relative_pattern, relative_name, FNM_PATHNAME) != 0)
				continue;
359

360 361 362
			dir->protected = ppath->protected;
			dir->symlinks_only = ppath->symlinks_only;
		}
363 364
	}

365 366 367
	return dir;
}

368 369 370
static struct apk_db_dir_instance *apk_db_diri_new(struct apk_database *db,
						   struct apk_package *pkg,
						   apk_blob_t name,
371
						   struct hlist_node ***after)
372 373 374 375
{
	struct apk_db_dir_instance *diri;

	diri = calloc(1, sizeof(struct apk_db_dir_instance));
376
	if (diri != NULL) {
377 378
		hlist_add_after(&diri->pkg_dirs_list, *after);
		*after = &diri->pkg_dirs_list.next;
379
		diri->dir = apk_db_dir_get(db, name);
380 381
		diri->pkg = pkg;
	}
382 383 384 385

	return diri;
}

386 387 388 389
static void apk_db_dir_apply_diri_permissions(struct apk_db_dir_instance *diri)
{
	struct apk_db_dir *dir = diri->dir;

390 391
	if (diri->uid < dir->uid ||
	    (diri->uid == dir->uid && diri->gid < dir->gid)) {
392 393
		dir->uid = diri->uid;
		dir->gid = diri->gid;
394 395 396
		dir->mode = diri->mode;
	} else if (diri->uid == dir->uid && diri->gid == dir->gid) {
		dir->mode &= diri->mode;
397 398 399
	}
}

400 401 402
static void apk_db_diri_set(struct apk_db_dir_instance *diri, mode_t mode,
			    uid_t uid, gid_t gid)
{
403
	diri->mode = mode & 07777;
404 405
	diri->uid = uid;
	diri->gid = gid;
406
	apk_db_dir_apply_diri_permissions(diri);
407 408
}

409
static void apk_db_diri_free(struct apk_database *db,
410 411
			     struct apk_db_dir_instance *diri,
			     int allow_rmdir)
412
{
413 414
	if (allow_rmdir == APK_DISALLOW_RMDIR &&
	    diri->dir->recalc_mode)
415 416
		apk_db_dir_apply_diri_permissions(diri);

417
	apk_db_dir_unref(db, diri->dir, allow_rmdir);
418 419 420
	free(diri);
}

421 422 423 424 425 426
struct apk_db_file *apk_db_file_query(struct apk_database *db,
				      apk_blob_t dir,
				      apk_blob_t name)
{
	struct apk_db_file_hash_key key;

427 428 429
	if (dir.len && dir.ptr[dir.len-1] == '/')
		dir.len--;

430 431 432 433 434 435 436 437 438
	key = (struct apk_db_file_hash_key) {
		.dirname = dir,
		.filename = name,
	};

	return (struct apk_db_file *) apk_hash_get(&db->installed.files,
						   APK_BLOB_BUF(&key));
}

Timo Teräs's avatar
Timo Teräs committed
439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460
static struct apk_db_file *apk_db_file_new(struct apk_db_dir_instance *diri,
					   apk_blob_t name,
					   struct hlist_node ***after)
{
	struct apk_db_file *file;

	file = malloc(sizeof(*file) + name.len + 1);
	if (file == NULL)
		return NULL;

	memset(file, 0, sizeof(*file));
	memcpy(file->name, name.ptr, name.len);
	file->name[name.len] = 0;
	file->namelen = name.len;

	file->diri = diri;
	hlist_add_after(&file->diri_files_list, *after);
	*after = &file->diri_files_list.next;

	return file;
}

461 462 463 464 465 466 467
static void apk_db_file_set(struct apk_db_file *file, mode_t mode, uid_t uid, gid_t gid)
{
	file->mode = mode & 07777;
	file->uid = uid;
	file->gid = gid;
}

468
static struct apk_db_file *apk_db_file_get(struct apk_database *db,
469
					   struct apk_db_dir_instance *diri,
470 471
					   apk_blob_t name,
					   struct hlist_node ***after)
472 473
{
	struct apk_db_file *file;
474
	struct apk_db_file_hash_key key;
Timo Teräs's avatar
Timo Teräs committed
475 476
	struct apk_db_dir *dir = diri->dir;
	unsigned long hash;
477 478

	key = (struct apk_db_file_hash_key) {
Timo Teräs's avatar
Timo Teräs committed
479
		.dirname = APK_BLOB_PTR_LEN(dir->name, dir->namelen),
480 481
		.filename = name,
	};
482

Timo Teräs's avatar
Timo Teräs committed
483 484 485
	hash = apk_blob_hash_seed(name, dir->hash);
	file = (struct apk_db_file *) apk_hash_get_hashed(
		&db->installed.files, APK_BLOB_BUF(&key), hash);
486 487 488
	if (file != NULL)
		return file;

Timo Teräs's avatar
Timo Teräs committed
489
	file = apk_db_file_new(diri, name, after);
Timo Teräs's avatar
Timo Teräs committed
490
	apk_hash_insert_hashed(&db->installed.files, file, hash);
491
	db->installed.stats.files++;
492 493 494 495

	return file;
}

Timo Teräs's avatar
Timo Teräs committed
496 497 498
static void apk_db_pkg_rdepends(struct apk_database *db, struct apk_package *pkg)
{
	int i, j;
Timo Teräs's avatar
Timo Teräs committed
499
	struct apk_name *rname;
Timo Teräs's avatar
Timo Teräs committed
500 501

	for (i = 0; i < pkg->depends->num; i++) {
Timo Teräs's avatar
Timo Teräs committed
502
		rname = pkg->depends->item[i].name;
Timo Teräs's avatar
Timo Teräs committed
503 504
		for (j = 0; j < rname->rdepends->num; j++)
			if (rname->rdepends->item[j] == pkg->name)
Timo Teräs's avatar
Timo Teräs committed
505
				goto rdeps_done;
Timo Teräs's avatar
Timo Teräs committed
506 507
		*apk_name_array_add(&rname->rdepends) = pkg->name;
	}
Timo Teräs's avatar
Timo Teräs committed
508 509 510 511 512 513 514 515 516 517
rdeps_done:
	for (i = 0; i < pkg->install_if->num; i++) {
		rname = pkg->install_if->item[i].name;
		for (j = 0; j < rname->rinstall_if->num; j++)
			if (rname->rinstall_if->item[j] == pkg->name)
				goto riif_done;
		*apk_name_array_add(&rname->rinstall_if) = pkg->name;
	}
riif_done:
	return;
Timo Teräs's avatar
Timo Teräs committed
518 519
}

520
struct apk_package *apk_db_pkg_add(struct apk_database *db, struct apk_package *pkg)
521 522 523
{
	struct apk_package *idb;

524
	if (pkg->license == NULL)
525
		pkg->license = apk_blob_atomize(APK_BLOB_NULL);
526

Timo Teräs's avatar
Timo Teräs committed
527
	idb = apk_hash_get(&db->available.packages, APK_BLOB_CSUM(pkg->csum));
528 529 530 531
	if (idb == NULL) {
		idb = pkg;
		apk_hash_insert(&db->available.packages, pkg);
		*apk_package_array_add(&pkg->name->pkgs) = pkg;
Timo Teräs's avatar
Timo Teräs committed
532
		apk_db_pkg_rdepends(db, pkg);
533 534
	} else {
		idb->repos |= pkg->repos;
535 536 537 538
		if (idb->filename == NULL && pkg->filename != NULL) {
			idb->filename = pkg->filename;
			pkg->filename = NULL;
		}
539 540 541 542 543
		if (idb->ipkg == NULL && pkg->ipkg != NULL) {
			idb->ipkg = pkg->ipkg;
			idb->ipkg->pkg = idb;
			pkg->ipkg = NULL;
		}
544 545 546 547 548
		apk_pkg_free(pkg);
	}
	return idb;
}

549
void apk_cache_format_index(apk_blob_t to, struct apk_repository *repo)
550
{
551
	/* APKINDEX.12345678.tar.gz */
552
	apk_blob_push_blob(&to, APK_BLOB_STR("APKINDEX."));
553 554
	apk_blob_push_hexdump(&to, APK_BLOB_PTR_LEN((char *) repo->csum.data,
						    APK_CACHE_CSUM_BYTES));
555
	apk_blob_push_blob(&to, APK_BLOB_STR(".tar.gz"));
556 557 558
	apk_blob_push_blob(&to, APK_BLOB_PTR_LEN("", 1));
}

559
int apk_cache_download(struct apk_database *db, const char *url, apk_blob_t *arch,
560 561 562 563 564
		       const char *item, const char *cacheitem, int verify)
{
	char fullurl[PATH_MAX];
	int r;

565
	apk_repo_format_filename(fullurl, sizeof(fullurl), url, arch, item);
566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583
	apk_message("fetch %s", fullurl);

	if (apk_flags & APK_SIMULATE)
		return 0;

	r = apk_url_download(fullurl, db->cachetmp_fd, cacheitem);
	if (r < 0)
		return r;

	if (verify != APK_SIGN_NONE) {
		struct apk_istream *is;
		struct apk_sign_ctx sctx;

		apk_sign_ctx_init(&sctx, APK_SIGN_VERIFY, NULL, db->keys_fd);
		is = apk_bstream_gunzip_mpart(
			apk_bstream_from_file(db->cachetmp_fd, cacheitem),
			apk_sign_ctx_mpart_cb, &sctx);

584
		r = apk_tar_parse(is, apk_sign_ctx_verify_tar, &sctx, FALSE, &db->id_cache);
585 586 587 588 589 590 591
		is->close(is);
		apk_sign_ctx_free(&sctx);
		if (r != 0) {
			unlinkat(db->cachetmp_fd, cacheitem, 0);
			return r;
		}
	}
592

593 594 595 596
	if (db->cachetmp_fd != db->cache_fd) {
		if (renameat(db->cachetmp_fd, cacheitem, db->cache_fd, cacheitem) < 0)
			return -errno;
	}
597

598
	return 0;
599 600
}

601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625
static struct apk_db_dir_instance *find_diri(struct apk_installed_package *ipkg,
					     apk_blob_t dirname,
					     struct apk_db_dir_instance *curdiri,
					     struct hlist_node ***tail)
{
	struct hlist_node *n;
	struct apk_db_dir_instance *diri;

	if (curdiri != NULL &&
	    apk_blob_compare(APK_BLOB_PTR_LEN(curdiri->dir->name,
					      curdiri->dir->namelen),
			     dirname) == 0)
		return curdiri;

	hlist_for_each_entry(diri, n, &ipkg->owned_dirs, pkg_dirs_list) {
		if (apk_blob_compare(APK_BLOB_PTR_LEN(diri->dir->name,
						      diri->dir->namelen), dirname) == 0) {
			if (tail != NULL)
				*tail = hlist_tail_ptr(&diri->owned_files);
			return diri;
		}
	}
	return NULL;
}

Timo Teräs's avatar
Timo Teräs committed
626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651
int apk_db_read_overlay(struct apk_database *db, struct apk_bstream *bs)
{
	struct apk_db_dir_instance *diri = NULL;
	struct hlist_node **diri_node = NULL, **file_diri_node = NULL;
	struct apk_package *pkg;
	struct apk_installed_package *ipkg;
	apk_blob_t token = APK_BLOB_STR("\n"), line, bdir, bfile;

	pkg = apk_pkg_new();
	if (pkg == NULL)
		return -1;

	ipkg = apk_pkg_install(db, pkg);
	if (ipkg == NULL)
		return -1;

	diri_node = hlist_tail_ptr(&ipkg->owned_dirs);

	while (!APK_BLOB_IS_NULL(line = bs->read(bs, token))) {
		if (!apk_blob_rsplit(line, '/', &bdir, &bfile))
			break;

		if (bfile.len == 0) {
			diri = apk_db_diri_new(db, pkg, bdir, &diri_node);
			file_diri_node = &diri->owned_files.first;
		} else {
652 653
			diri = find_diri(ipkg, bdir, diri, &file_diri_node);
			if (diri == NULL) {
654 655
				diri = apk_db_diri_new(db, pkg, bdir, &diri_node);
				file_diri_node = &diri->owned_files.first;
656
			}
Timo Teräs's avatar
Timo Teräs committed
657
			(void) apk_db_file_get(db, diri, bfile, &file_diri_node);
Timo Teräs's avatar
Timo Teräs committed
658 659 660 661 662 663
		}
	}

	return 0;
}

664
int apk_db_index_read(struct apk_database *db, struct apk_bstream *bs, int repo)
665 666
{
	struct apk_package *pkg = NULL;
667
	struct apk_installed_package *ipkg = NULL;
668
	struct apk_db_dir_instance *diri = NULL;
669
	struct apk_db_file *file = NULL;
670 671
	struct hlist_node **diri_node = NULL;
	struct hlist_node **file_diri_node = NULL;
672
	apk_blob_t token = APK_BLOB_STR("\n"), l;
673 674 675
	mode_t mode;
	uid_t uid;
	gid_t gid;
676
	int field, r;
677

678 679 680 681
	while (!APK_BLOB_IS_NULL(l = bs->read(bs, token))) {
		if (l.len < 2 || l.ptr[1] != ':') {
			if (pkg == NULL)
				continue;
682

683
			if (repo >= 0) {
684
				pkg->repos |= BIT(repo);
685 686 687 688
			} else if (repo == -1 && ipkg == NULL) {
				/* Installed package without files */
				ipkg = apk_pkg_install(db, pkg);
			}
689

690
			if (apk_db_pkg_add(db, pkg) == NULL) {
691 692
				apk_error("Installed database load failed");
				return -1;
693
			}
694
			pkg = NULL;
695
			ipkg = NULL;
696 697
			continue;
		}
698

699 700 701 702 703 704 705 706
		/* Get field */
		field = l.ptr[0];
		l.ptr += 2;
		l.len -= 2;

		/* If no package, create new */
		if (pkg == NULL) {
			pkg = apk_pkg_new();
707
			ipkg = NULL;
708 709 710
			diri = NULL;
			file_diri_node = NULL;
		}
711

712
		/* Standard index line? */
713 714
		r = apk_pkg_add_info(db, pkg, field, l);
		if (r == 0) {
715 716
			continue;
		}
717
		if (r == 1 && repo == -1 && ipkg == NULL) {
718 719 720 721 722 723
			/* Instert to installed database; this needs to
			 * happen after package name has been read, but
			 * before first FDB entry. */
			ipkg = apk_pkg_install(db, pkg);
			diri_node = hlist_tail_ptr(&ipkg->owned_dirs);
		}
724 725
		if (repo != -1 || ipkg == NULL)
			continue;
726

727 728 729 730 731
		/* Check FDB special entries */
		switch (field) {
		case 'F':
			if (pkg->name == NULL) {
				apk_error("FDB directory entry before package entry");
732 733
				return -1;
			}
734 735 736
			diri = apk_db_diri_new(db, pkg, l, &diri_node);
			file_diri_node = &diri->owned_files.first;
			break;
737 738 739 740 741
		case 'a':
			if (file == NULL) {
				apk_error("FDB file attribute metadata entry before file entry");
				return -1;
			}
742 743 744 745 746
		case 'M':
			if (diri == NULL) {
				apk_error("FDB directory metadata entry before directory entry");
				return -1;
			}
747
			uid = apk_blob_pull_uint(&l, 10);
748
			apk_blob_pull_char(&l, ':');
749
			gid = apk_blob_pull_uint(&l, 10);
750
			apk_blob_pull_char(&l, ':');
751
			mode = apk_blob_pull_uint(&l, 8);
752 753 754 755
			if (field == 'M')
				apk_db_diri_set(diri, mode, uid, gid);
			else
				apk_db_file_set(file, mode, uid, gid);
756 757 758 759 760 761
			break;
		case 'R':
			if (diri == NULL) {
				apk_error("FDB file entry before directory entry");
				return -1;
			}
Timo Teräs's avatar
Timo Teräs committed
762
			file = apk_db_file_get(db, diri, l, &file_diri_node);
763 764 765 766 767 768
			break;
		case 'Z':
			if (file == NULL) {
				apk_error("FDB checksum entry before file entry");
				return -1;
			}
Timo Teräs's avatar
Timo Teräs committed
769
			apk_blob_pull_csum(&l, &file->csum);
770
			break;
771
		case 'r':
772
			apk_blob_pull_deps(&l, db, &ipkg->replaces);
773
			break;
774
		case 'q':
775
			ipkg->replaces_priority = apk_blob_pull_uint(&l, 10);
776
			break;
777 778 779
		case 'p':
			ipkg->repository_tag = apk_db_get_tag_id(db, l);
			break;
780
		default:
781 782 783 784 785 786 787 788
			if (r != 0 && !(apk_flags & APK_FORCE)) {
				/* Installed db should not have unsupported fields */
				apk_error("This apk-tools is too old to handle installed packages");
				return -1;
			}
			/* Installed. So mark the package as installable. */
			pkg->filename = NULL;
			continue;
789
		}
790 791 792 793
		if (APK_BLOB_IS_NULL(l)) {
			apk_error("FDB format error in entry '%c'", field);
			return -1;
		}
794 795
	}

796
	return 0;
797 798
}

Timo Teräs's avatar
Timo Teräs committed
799
static int apk_db_write_fdb(struct apk_database *db, struct apk_ostream *os)
800
{
801
	struct apk_installed_package *ipkg;
802
	struct apk_package *pkg;
803
	struct apk_db_dir_instance *diri;
804
	struct apk_db_file *file;
805
	struct hlist_node *c1, *c2;
806
	char buf[1024];
807 808
	apk_blob_t bbuf = APK_BLOB_BUF(buf);
	int r;
809

810 811
	list_for_each_entry(ipkg, &db->installed.packages, installed_pkgs_list) {
		pkg = ipkg->pkg;
812 813 814
		r = apk_pkg_write_index_entry(pkg, os);
		if (r < 0)
			return r;
815

816 817
		if (ipkg->replaces->num) {
			apk_blob_push_blob(&bbuf, APK_BLOB_STR("r:"));
818
			apk_blob_push_deps(&bbuf, db, ipkg->replaces);
819 820
			apk_blob_push_blob(&bbuf, APK_BLOB_STR("\n"));
		}
821 822 823 824 825
		if (ipkg->replaces_priority) {
			apk_blob_push_blob(&bbuf, APK_BLOB_STR("q:"));
			apk_blob_push_uint(&bbuf, ipkg->replaces_priority, 10);
			apk_blob_push_blob(&bbuf, APK_BLOB_STR("\n"));
		}
826 827 828 829 830
		if (ipkg->repository_tag) {
			apk_blob_push_blob(&bbuf, APK_BLOB_STR("p:"));
			apk_blob_push_blob(&bbuf, *db->repo_tags[ipkg->repository_tag].name);
			apk_blob_push_blob(&bbuf, APK_BLOB_STR("\n"));
		}
831
		hlist_for_each_entry(diri, c1, &ipkg->owned_dirs, pkg_dirs_list) {
832 833 834 835 836 837 838 839 840
			apk_blob_push_blob(&bbuf, APK_BLOB_STR("F:"));
			apk_blob_push_blob(&bbuf, APK_BLOB_PTR_LEN(diri->dir->name, diri->dir->namelen));
			apk_blob_push_blob(&bbuf, APK_BLOB_STR("\nM:"));
			apk_blob_push_uint(&bbuf, diri->uid, 10);
			apk_blob_push_blob(&bbuf, APK_BLOB_STR(":"));
			apk_blob_push_uint(&bbuf, diri->gid, 10);
			apk_blob_push_blob(&bbuf, APK_BLOB_STR(":"));
			apk_blob_push_uint(&bbuf, diri->mode, 8);
			apk_blob_push_blob(&bbuf, APK_BLOB_STR("\n"));
841

842
			hlist_for_each_entry(file, c2, &diri->owned_files, diri_files_list) {
843 844
				apk_blob_push_blob(&bbuf, APK_BLOB_STR("R:"));
				apk_blob_push_blob(&bbuf, APK_BLOB_PTR_LEN(file->name, file->namelen));
845 846 847 848 849 850 851 852
				if (file->mode != 0 || file->uid != 0 || file->gid != 0) {
					apk_blob_push_blob(&bbuf, APK_BLOB_STR("\na:"));
					apk_blob_push_uint(&bbuf, file->uid, 10);
					apk_blob_push_blob(&bbuf, APK_BLOB_STR(":"));
					apk_blob_push_uint(&bbuf, file->gid, 10);
					apk_blob_push_blob(&bbuf, APK_BLOB_STR(":"));
					apk_blob_push_uint(&bbuf, file->mode, 8);
				}
Timo Teräs's avatar
Timo Teräs committed
853
				if (file->csum.type != APK_CHECKSUM_NONE) {
854
					apk_blob_push_blob(&bbuf, APK_BLOB_STR("\nZ:"));
Timo Teräs's avatar
Timo Teräs committed
855
					apk_blob_push_csum(&bbuf, &file->csum);
856
				}
857
				apk_blob_push_blob(&bbuf, APK_BLOB_STR("\n"));
858

859
				if (os->write(os, buf, bbuf.ptr - buf) != bbuf.ptr - buf)
860
					return -1;
861
				bbuf = APK_BLOB_BUF(buf);
862
			}
863
			if (os->write(os, buf, bbuf.ptr - buf) != bbuf.ptr - buf)
864
				return -1;
865
			bbuf = APK_BLOB_BUF(buf);
866
		}
Timo Teräs's avatar
Timo Teräs committed
867
		os->write(os, "\n", 1);
868 869 870 871 872
	}

	return 0;
}

Timo Teräs's avatar
Timo Teräs committed
873
static int apk_db_scriptdb_write(struct apk_database *db, struct apk_ostream *os)
874
{
875
	struct apk_installed_package *ipkg;
876
	struct apk_package *pkg;
Timo Teräs's avatar
Timo Teräs committed
877 878 879
	struct apk_file_info fi;
	char filename[256];
	apk_blob_t bfn;
880
	int r, i;
881
	time_t now = time(NULL);
882 883 884 885 886 887 888

	list_for_each_entry(ipkg, &db->installed.packages, installed_pkgs_list) {
		pkg = ipkg->pkg;

		for (i = 0; i < APK_SCRIPT_MAX; i++) {
			if (ipkg->script[i].ptr == NULL)
				continue;
889

Timo Teräs's avatar
Timo Teräs committed
890 891
			fi = (struct apk_file_info) {
				.name = filename,
892
				.size = ipkg->script[i].len,
Timo Teräs's avatar
Timo Teräs committed
893
				.mode = 0755 | S_IFREG,
894
				.mtime = now,
Timo Teräs's avatar
Timo Teräs committed
895 896 897 898 899 900
			};
			/* The scripts db expects file names in format:
			 * pkg-version.<hexdump of package checksum>.action */
			bfn = APK_BLOB_BUF(filename);
			apk_blob_push_blob(&bfn, APK_BLOB_STR(pkg->name->name));
			apk_blob_push_blob(&bfn, APK_BLOB_STR("-"));
901
			apk_blob_push_blob(&bfn, *pkg->version);
Timo Teräs's avatar
Timo Teräs committed
902 903 904
			apk_blob_push_blob(&bfn, APK_BLOB_STR("."));
			apk_blob_push_csum(&bfn, &pkg->csum);
			apk_blob_push_blob(&bfn, APK_BLOB_STR("."));
905
			apk_blob_push_blob(&bfn, APK_BLOB_STR(apk_script_types[i]));
Timo Teräs's avatar
Timo Teräs committed
906 907
			apk_blob_push_blob(&bfn, APK_BLOB_PTR_LEN("", 1));

908
			r = apk_tar_write_entry(os, &fi, ipkg->script[i].ptr);
Timo Teräs's avatar
Timo Teräs committed
909 910
			if (r < 0)
				return r;
911 912 913
		}
	}