state.c 24.7 KB
Newer Older
1 2 3 4 5 6
/* state.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.
 *
Timo Teräs's avatar
Timo Teräs committed
7
 * This program is free software; you can redistribute it and/or modify it
8 9 10 11 12
 * 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 <stdio.h>
13
#include <stdlib.h>
14
#include <unistd.h>
15 16 17 18
#include <malloc.h>

#include "apk_state.h"
#include "apk_database.h"
Natanael Copa's avatar
Natanael Copa committed
19
#include "apk_print.h"
20

21 22 23 24 25 26 27 28 29 30 31 32
struct apk_name_choices {
	unsigned short refs, num;
	struct apk_package *pkgs[];
};

#if 0
struct apk_deferred_state {
	unsigned int preference;
	struct apk_state *state;
};
#endif

33 34 35 36 37
int apk_state_prune_dependency(struct apk_state *state,
			       struct apk_dependency *dep);

#define APK_NS_LOCKED			0x00000001
#define APK_NS_PENDING			0x00000002
38 39 40 41 42 43 44
#define APK_NS_ERROR			0x00000004

static void apk_state_record_conflict(struct apk_state *state,
				      struct apk_package *pkg)
{
	struct apk_name *name = pkg->name;

Timo Teräs's avatar
Timo Teräs committed
45
	state->name[name->id] = (void*) (((intptr_t) pkg) | APK_NS_ERROR | APK_NS_LOCKED);
46 47
	*apk_package_array_add(&state->conflicts) = pkg;
}
48

49 50
static int inline ns_locked(apk_name_state_t name)
{
51 52 53 54 55 56 57 58
	if (((intptr_t) name) & APK_NS_LOCKED)
		return TRUE;
	return FALSE;
}

static int inline ns_pending(apk_name_state_t name)
{
	if (((intptr_t) name) & APK_NS_PENDING)
59 60 61 62
		return TRUE;
	return FALSE;
}

63 64 65 66 67 68 69
static int inline ns_error(apk_name_state_t name)
{
	if (((intptr_t) name) & APK_NS_ERROR)
		return TRUE;
	return FALSE;
}

70 71 72 73 74 75 76
static int ns_empty(apk_name_state_t name)
{
	return name == NULL;
}

static apk_name_state_t ns_from_pkg(struct apk_package *pkg)
{
77 78 79 80 81 82
	return (apk_name_state_t) (((intptr_t) pkg) | APK_NS_LOCKED | APK_NS_PENDING);
}

static apk_name_state_t ns_from_pkg_non_pending(struct apk_package *pkg)
{
	return (apk_name_state_t) (((intptr_t) pkg) | APK_NS_LOCKED);
83 84 85 86
}

static struct apk_package *ns_to_pkg(apk_name_state_t name)
{
87
	return (struct apk_package *)
88
		(((intptr_t) name) & ~(APK_NS_LOCKED | APK_NS_PENDING | APK_NS_ERROR));
89 90 91 92 93 94 95 96 97 98 99 100 101 102
}

static apk_name_state_t ns_from_choices(struct apk_name_choices *nc)
{
	if (nc == NULL)
		return ns_from_pkg(NULL);
	return (apk_name_state_t) nc;
}

static struct apk_name_choices *ns_to_choices(apk_name_state_t name)
{
	return (struct apk_name_choices *) name;
}

103 104
static struct apk_name_choices *name_choices_new(struct apk_database *db,
						 struct apk_name *name)
105 106
{
	struct apk_name_choices *nc;
107
	int i, j;
108

Timo Teräs's avatar
Timo Teräs committed
109
	if (name->pkgs->num == 0)
110 111 112 113 114 115 116 117 118 119 120
		return NULL;

	nc = malloc(sizeof(struct apk_name_choices) +
		    name->pkgs->num * sizeof(struct apk_package *));
	if (nc == NULL)
		return NULL;

	nc->refs = 1;
	nc->num = name->pkgs->num;
	memcpy(nc->pkgs, name->pkgs->item,
	       name->pkgs->num * sizeof(struct apk_package *));
121

122 123 124 125 126 127 128 129 130
	for (j = 0; j < nc->num; ) {
		if (nc->pkgs[j]->filename != APK_PKG_UNINSTALLABLE) {
			j++;
		} else {
			nc->pkgs[j] = nc->pkgs[nc->num - 1];
			nc->num--;
		}
	}

131 132 133
	if (name->flags & APK_NAME_TOPLEVEL_OVERRIDE)
		return nc;

134
	/* Check for global dependencies */
Timo Teräs's avatar
Timo Teräs committed
135
	for (i = 0; i < db->world->num; i++) {
136 137 138 139 140
		struct apk_dependency *dep = &db->world->item[i];

		if (dep->name != name)
			continue;

141 142 143 144 145 146 147 148 149 150 151
		if (apk_flags & APK_PREFER_AVAILABLE) {
			dep->version = apk_blob_atomize(APK_BLOB_NULL);
			dep->result_mask = APK_DEPMASK_REQUIRE;
		} else {
			for (j = 0; j < nc->num; ) {
				if (apk_dep_is_satisfied(dep, nc->pkgs[j])) {
					j++;
				} else {
					nc->pkgs[j] = nc->pkgs[nc->num - 1];
					nc->num--;
				}
152 153
			}
		}
154
		break;
155 156
	}

157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191
	return nc;
}

static void name_choices_unref(struct apk_name_choices *nc)
{
	if (--nc->refs == 0)
		free(nc);
}

static struct apk_name_choices *name_choices_writable(struct apk_name_choices *nc)
{
	struct apk_name_choices *n;

	if (nc->refs == 1)
		return nc;

	n = malloc(sizeof(struct apk_name_choices) +
		   nc->num * sizeof(struct apk_package *));
	if (n == NULL)
		return NULL;

	n->refs = 1;
	n->num = nc->num;
	memcpy(n->pkgs, nc->pkgs, nc->num * sizeof(struct apk_package *));
	name_choices_unref(nc);

	return n;
}

static void ns_free(apk_name_state_t name)
{
	if (!ns_empty(name) && !ns_locked(name))
		name_choices_unref(ns_to_choices(name));
}

192 193 194
static inline int apk_state_pkg_available(struct apk_state *state,
					  struct apk_package *pkg)
{
195 196
	if (pkg->ipkg != NULL)
		return TRUE;
197 198
	if (pkg->installed_size == 0)
		return TRUE;
199 200 201 202 203 204 205
	if (pkg->filename != NULL)
		return TRUE;
	if (apk_db_select_repo(state->db, pkg) != NULL)
		return TRUE;
	return FALSE;
}

206 207 208
struct apk_state *apk_state_new(struct apk_database *db)
{
	struct apk_state *state;
209
	int num_bytes;
210

211
	num_bytes = sizeof(struct apk_state) + db->name_id * sizeof(char *);
212 213
	state = (struct apk_state*) calloc(1, num_bytes);
	state->refs = 1;
214
	state->num_names = db->name_id;
215
	state->db = db;
216
	state->print_ok = 1;
217
	list_init(&state->change_list_head);
218

Timo Teräs's avatar
Timo Teräs committed
219 220 221
	apk_name_array_init(&state->missing);
	apk_package_array_init(&state->conflicts);

222 223 224 225 226 227 228 229 230 231 232 233 234
	return state;
}

struct apk_state *apk_state_dup(struct apk_state *state)
{
	state->refs++;
	return state;
}

void apk_state_unref(struct apk_state *state)
{
	if (--state->refs > 0)
		return;
Timo Teräs's avatar
Timo Teräs committed
235 236 237

	apk_package_array_free(&state->conflicts);
	apk_name_array_free(&state->missing);
238 239 240
	free(state);
}

241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267
static struct apk_package *get_locked_or_installed_package(
					struct apk_state *state,
					struct apk_name *name)
{
	int i;

	if (ns_locked(state->name[name->id]))
		return ns_to_pkg(state->name[name->id]);

	if (!ns_empty(state->name[name->id])) {
		struct apk_name_choices *ns =
			ns_to_choices(state->name[name->id]);

		for (i = 0; i < ns->num; i++) {
			if (ns->pkgs[i]->ipkg != NULL)
				return ns->pkgs[i];
		}
		return NULL;
	}

	for (i = 0; i < name->pkgs->num; i++) {
		if (name->pkgs->item[i]->ipkg != NULL)
			return name->pkgs->item[i];
	}
	return NULL;
}

268 269 270 271 272 273 274 275 276 277 278 279 280 281
static int check_dependency(struct apk_state *state,
			    struct apk_dependency *dep)
{
	struct apk_package *pkg;

	pkg = get_locked_or_installed_package(state, dep->name);
	if (pkg == NULL && dep->result_mask != APK_DEPMASK_CONFLICT)
		return 0;
	if (!apk_dep_is_satisfied(dep, pkg))
		return 0;

	return 1;
}

282 283 284 285 286 287
static int check_dependency_array(struct apk_state *state,
				  struct apk_dependency_array *da)
{
	int i;

	for (i = 0; i < da->num; i++) {
288
		if (!check_dependency(state, &da->item[i]))
289 290 291 292 293 294
			return 0;
	}

	return da->num;
}

295 296 297 298 299 300 301 302 303 304 305
static int apk_state_add_change(struct apk_state *state,
				struct apk_package *oldpkg,
				struct apk_package *newpkg)
{
	struct apk_change *change;

	change = (struct apk_change *) malloc(sizeof(struct apk_change));
	if (change == NULL)
		return -1;

	list_init(&change->change_list);
306
	list_add_tail(&change->change_list, &state->change_list_head);
307
	state->num_changes++;
308 309 310 311 312 313
	change->oldpkg = oldpkg;
	change->newpkg = newpkg;

	return 0;
}

314 315 316 317 318 319 320
/* returns:
 *   -1 error
 *    0 locked entry matches and is ok
 *   +n this many candidates on apk_name_choices for the name
 */
int apk_state_prune_dependency(struct apk_state *state,
			       struct apk_dependency *dep)
321
{
322 323 324 325
	struct apk_name *name = dep->name;
	struct apk_name_choices *c;
	int i;

326 327 328
	if (name->id >= state->num_names)
		return -1;

329
	if (ns_empty(state->name[name->id])) {
330 331 332 333
		if (dep->result_mask == APK_DEPMASK_CONFLICT) {
			state->name[name->id] = ns_from_pkg(NULL);
			return 1;
		}
334 335 336

		/* This name has not been visited yet.
		 * Construct list of candidates. */
337
		state->name[name->id] = ns_from_choices(name_choices_new(state->db, name));
338 339 340 341 342 343 344 345
	}

	if (ns_locked(state->name[name->id])) {
		/* Locked: check that selected package provides
		 * requested version. */
		struct apk_package *pkg = ns_to_pkg(state->name[name->id]);

		/* Locked to not-installed / remove? */
346 347 348
		if (ns_error(state->name[name->id])) {
			return -1;
		} else if (pkg == NULL) {
349 350 351 352 353
			if (dep->result_mask != APK_DEPMASK_CONFLICT) {
				if (ns_pending(state->name[name->id])) {
					state->name[name->id] = ns_from_pkg_non_pending(NULL);
					*apk_name_array_add(&state->missing) = name;
				}
354
				return -1;
355
			}
356
		} else {
357
			if (!apk_dep_is_satisfied(dep, pkg))
358
				return -1;
359 360
		}

361 362
		if (ns_pending(state->name[name->id]))
			return 1;
363

364
		return 0;
365 366 367 368
	}

	/* Multiple candidates: prune incompatible versions. */
	c = ns_to_choices(state->name[name->id]);
369 370
	i = 0;
	while (i < c->num) {
371 372
		if (apk_dep_is_satisfied(dep, c->pkgs[i])) {
			i++;
373
			continue;
374
		}
375 376 377 378 379

		c = name_choices_writable(c);
		c->pkgs[i] = c->pkgs[c->num - 1];
		c->num--;
	}
380
	if (c->num == 1 && apk_state_pkg_available(state, c->pkgs[0])) {
381 382
		struct apk_package *pkg = c->pkgs[0];
		name_choices_unref(c);
383 384
		state->name[name->id] = ns_from_pkg(pkg);
		return 1;
385
	}
386 387
	if (c->num <= 1) {
		name_choices_unref(c);
Timo Teräs's avatar
Timo Teräs committed
388
		state->name[name->id] = ns_from_pkg(NULL);
389
		*apk_name_array_add(&state->missing) = name;
390 391
		return -1;
	}
392

393
	state->name[name->id] = ns_from_choices(c);
394 395
	return c->num;
}
396

397 398
int apk_state_autolock_name(struct apk_state *state, struct apk_name *name,
			    int install_if)
399 400
{
	struct apk_name_choices *c;
401 402
	struct apk_package *installed = NULL, *latest = NULL, *use;
	int i;
403 404 405

	if (ns_pending(state->name[name->id]))
		return apk_state_lock_name(state, name, ns_to_pkg(state->name[name->id]));
406 407 408 409 410 411 412
	if (ns_locked(state->name[name->id]))
		return 0;
	if (ns_empty(state->name[name->id])) {
		/* This name has not been visited yet.
		 * Construct list of candidates. */
		state->name[name->id] = ns_from_choices(name_choices_new(state->db, name));
	}
413 414

	c = ns_to_choices(state->name[name->id]);
415 416
#if 1
	/* Get latest and installed packages */
417
	for (i = 0; i < c->num; i++) {
418 419
		struct apk_package *pkg = c->pkgs[i];

420 421 422 423
		if (install_if &&
		    !check_dependency_array(state, pkg->install_if))
			continue;

424
		if (pkg->ipkg != NULL)
425
			installed = pkg;
426
		else if (!apk_state_pkg_available(state, pkg))
427 428
			continue;

Timo Teräs's avatar
Timo Teräs committed
429 430 431 432 433
		if (latest == NULL) {
			latest = pkg;
			continue;
		}

434 435
		if ((apk_flags & APK_PREFER_AVAILABLE) ||
		    (name->flags & APK_NAME_REINSTALL)) {
Timo Teräs's avatar
Timo Teräs committed
436 437 438 439 440 441 442 443 444 445 446 447 448
			if (latest->repos != 0 && pkg->repos == 0)
				continue;

			if (latest->repos == 0 && pkg->repos != 0) {
				latest = pkg;
				continue;
			}

			/* Otherwise both are not available, or both are
			 * available and we just compare the versions then */
		}

		if (apk_pkg_version_compare(pkg, latest) == APK_VERSION_GREATER)
449 450
			latest = pkg;
	}
451

452 453 454 455 456
	/* Choose the best looking candidate.
	 * FIXME: We should instead try all alternatives. */
	if (apk_flags & APK_UPGRADE) {
		use = latest;
	} else {
457 458 459
		if (installed != NULL &&
		    (installed->repos != 0 ||
		     !(name->flags & APK_NAME_REINSTALL)))
460 461 462 463 464
			use = installed;
		else
			use = latest;
	}
	if (use == NULL)
465
		return -2;
466

467 468 469 470 471 472
	/* Install_if check did not result in package selection change:
	 * do not lock the package yet as the preferency might change
	 * later. */
	if (install_if && use->ipkg != NULL)
		return 0;

473 474 475 476 477 478 479 480 481 482 483 484 485
	return apk_state_lock_name(state, name, use);
#else
	/* If any of the choices is installed, we are good. Otherwise,
	 * the caller needs to install this dependency. */
	for (i = 0; i < c->num; i++)
		if (apk_pkg_get_state(c->pkgs[i]) == APK_PKG_INSTALLED)
			return 0;

	/* Queue for deferred solution. */
	return 0;
#endif
}

486 487 488 489 490 491 492 493 494 495 496 497
int apk_state_lock_dependency(struct apk_state *state,
			      struct apk_dependency *dep)
{
	int r;

	r = apk_state_prune_dependency(state, dep);
	if (r <= 0)
		return r;

	return apk_state_autolock_name(state, dep->name, FALSE);
}

498 499 500
static int apk_state_fix_package(struct apk_state *state,
				 struct apk_package *pkg)
{
501 502
	int i, r, ret = 0;

Timo Teräs's avatar
Timo Teräs committed
503
	if (pkg == NULL)
504
		return 0;
505 506

	for (i = 0; i < pkg->depends->num; i++) {
507 508
		if ((pkg->depends->item[i].name->flags & APK_NAME_TOPLEVEL_OVERRIDE) &&
		    check_dependency(state, &pkg->depends->item[i])) {
509 510 511 512 513 514 515 516 517 518
			r = apk_state_prune_dependency(state,
						       &pkg->depends->item[i]);
			if (r < 0)
				ret = -1;
		} else {
			r = apk_state_lock_dependency(state,
						      &pkg->depends->item[i]);
			if (r != 0)
				ret = -1;
		}
519
	}
520
	return ret;
521 522
}

Timo Teräs's avatar
Timo Teräs committed
523 524 525 526
static int call_if_dependency_broke(struct apk_state *state,
				    struct apk_package *pkg,
				    struct apk_name *dep_name,
				    int (*cb)(struct apk_state *state,
527 528 529 530
					      struct apk_package *pkg,
					      struct apk_dependency *dep,
					      void *ctx),
				    void *ctx)
Timo Teräs's avatar
Timo Teräs committed
531 532 533 534 535 536 537 538 539 540 541 542 543
{
	struct apk_package *dep_pkg;
	int k;

	dep_pkg = ns_to_pkg(state->name[dep_name->id]);
	for (k = 0; k < pkg->depends->num; k++) {
		struct apk_dependency *dep = &pkg->depends->item[k];
		if (dep->name != dep_name)
			continue;
		if (dep_pkg == NULL &&
		    dep->result_mask == APK_DEPMASK_CONFLICT)
			continue;
		if (dep_pkg != NULL &&
544
		    apk_dep_is_satisfied(dep, dep_pkg))
Timo Teräs's avatar
Timo Teräs committed
545
			continue;
546
		return cb(state, pkg, dep, ctx);
Timo Teräs's avatar
Timo Teräs committed
547 548 549 550 551 552 553 554
	}

	return 0;
}

static int for_each_broken_reverse_depency(struct apk_state *state,
					   struct apk_name *name,
					   int (*cb)(struct apk_state *state,
555 556 557 558
						     struct apk_package *pkg,
						     struct apk_dependency *dep,
						     void *ctx),
					   void *ctx)
Timo Teräs's avatar
Timo Teräs committed
559 560
{
	struct apk_package *pkg0;
561
	int i, r;
Timo Teräs's avatar
Timo Teräs committed
562 563 564 565

	for (i = 0; i < name->rdepends->num; i++) {
		struct apk_name *name0 = name->rdepends->item[i];

566 567 568 569 570 571 572 573
		pkg0 = get_locked_or_installed_package(state, name0);
		if (pkg0 == NULL)
			continue;

		r = call_if_dependency_broke(state, pkg0, name,
					     cb, ctx);
		if (r != 0)
			return r;
Timo Teräs's avatar
Timo Teräs committed
574 575 576 577 578 579
	}

	return 0;
}

static int delete_broken_package(struct apk_state *state,
580 581 582
				 struct apk_package *pkg,
				 struct apk_dependency *dep,
				 void *ctx)
Timo Teräs's avatar
Timo Teräs committed
583 584 585 586 587
{
	return apk_state_lock_name(state, pkg->name, NULL);
}

static int reinstall_broken_package(struct apk_state *state,
588 589 590
				    struct apk_package *pkg,
				    struct apk_dependency *dep,
				    void *ctx)
Timo Teräs's avatar
Timo Teräs committed
591 592

{
593
	struct apk_dependency dep0 = {
Timo Teräs's avatar
Timo Teräs committed
594
		.name = pkg->name,
595
		.version = apk_blob_atomize(APK_BLOB_NULL),
Timo Teräs's avatar
Timo Teräs committed
596 597
		.result_mask = APK_DEPMASK_REQUIRE,
	};
598
	return apk_state_lock_dependency(state, &dep0);
Timo Teräs's avatar
Timo Teräs committed
599 600
}

601 602 603
int apk_state_lock_name(struct apk_state *state,
			struct apk_name *name,
			struct apk_package *newpkg)
604
{
605
	struct apk_package *oldpkg = NULL;
Timo Teräs's avatar
Timo Teräs committed
606
	int i, r;
607

608 609 610
	if (name->id >= state->num_names)
		return -1;

611
	ns_free(state->name[name->id]);
612
	state->name[name->id] = ns_from_pkg_non_pending(newpkg);
613

Timo Teräs's avatar
Timo Teräs committed
614 615 616 617 618
	for (i = 0; i < name->pkgs->num; i++) {
		struct apk_package *pkg = name->pkgs->item[i];
		if (name->pkgs->item[i]->name == name &&
		    pkg->ipkg != NULL)
			oldpkg = pkg;
619 620 621 622
	}

	/* First we need to make sure the dependants of the old package
	 * still have their dependencies ok. */
Timo Teräs's avatar
Timo Teräs committed
623 624 625 626
	if (oldpkg != NULL) {
		r = for_each_broken_reverse_depency(state, name,
						    newpkg == NULL ?
						    delete_broken_package :
627 628 629 630
						    reinstall_broken_package,
						    NULL);
		if (r != 0) {
			apk_state_record_conflict(state, newpkg);
Timo Teräs's avatar
Timo Teräs committed
631
			return r;
632
		}
633 634 635
	}

	/* Check that all other dependencies hold for the new package. */
636 637 638 639
	r = apk_state_fix_package(state, newpkg);
	if (r != 0) {
		apk_state_record_conflict(state, newpkg);
		return r;
640 641
	}

642
	/* If the chosen package is installed, all is done here */
643 644 645 646 647 648 649
	if ((oldpkg != newpkg) ||
	    (newpkg != NULL && (newpkg->name->flags & APK_NAME_REINSTALL))) {
		/* Track change */
		r = apk_state_add_change(state, oldpkg, newpkg);
		if (r != 0)
			return r;
	}
650

651 652 653 654 655
	/* Check all reverse install_if's */
	if (newpkg != NULL) {
		for (i = 0; i < newpkg->name->rinstall_if->num; i++)
			apk_state_autolock_name(state, newpkg->name->rinstall_if->item[i], TRUE);
	}
656 657

	return 0;
658 659
}

660 661
static void apk_print_change(struct apk_database *db,
			     struct apk_package *oldpkg,
662 663
			     struct apk_package *newpkg,
			     int num, int total)
664 665
{
	const char *msg = NULL;
666 667
	int r;
	struct apk_name *name;
668 669 670 671
	char status[64];

	snprintf(status, sizeof(status), "(%i/%i)", num, total);
	status[sizeof(status) - 1] = '\0';
672

673 674 675 676
	if (oldpkg != NULL)
		name = oldpkg->name;
	else
		name = newpkg->name;
677 678

	if (oldpkg == NULL) {
679 680 681
		apk_message("%s Installing %s (" BLOB_FMT ")",
			    status, name->name,
			    BLOB_PRINTF(*newpkg->version));
682
	} else if (newpkg == NULL) {
683 684 685
		apk_message("%s Purging %s (" BLOB_FMT ")",
			    status, name->name,
			    BLOB_PRINTF(*oldpkg->version));
686
	} else {
687
		r = apk_pkg_version_compare(newpkg, oldpkg);
688 689 690 691 692
		switch (r) {
		case APK_VERSION_LESS:
			msg = "Downgrading";
			break;
		case APK_VERSION_EQUAL:
693
			if (newpkg == oldpkg)
694
				msg = "Re-installing";
695 696 697
			else
				msg = "Replacing";
			break;
698 699 700 701
		case APK_VERSION_GREATER:
			msg = "Upgrading";
			break;
		}
702 703 704 705
		apk_message("%s %s %s (" BLOB_FMT " -> " BLOB_FMT ")",
			    status, msg, name->name,
			    BLOB_PRINTF(*oldpkg->version),
			    BLOB_PRINTF(*newpkg->version));
706 707 708
	}
}

709 710 711 712
struct apk_stats {
	unsigned int bytes;
	unsigned int packages;
};
713

714 715 716 717 718
static void apk_count_change(struct apk_change *change, struct apk_stats *stats)
{
	if (change->newpkg != NULL) {
		stats->bytes += change->newpkg->installed_size;
		stats->packages ++;
719
	}
720 721 722
	if (change->oldpkg != NULL)
		stats->packages ++;
}
723

Timo Teräs's avatar
Timo Teräs committed
724
static void apk_draw_progress(int percent)
725
{
Timo Teräs's avatar
Timo Teräs committed
726
	const int bar_width = apk_get_screen_width() - 7;
727
	int i;
728

729
	fprintf(stderr, "\e7%3i%% [", percent);
Timo Teräs's avatar
Timo Teräs committed
730 731 732 733
	for (i = 0; i < bar_width * percent / 100; i++)
		fputc('#', stderr);
	for (; i < bar_width; i++)
		fputc(' ', stderr);
734
	fputc(']', stderr);
735
	fflush(stderr);
Timo Teräs's avatar
Timo Teräs committed
736
	fputs("\e8\e[0K", stderr);
737 738
}

739 740 741 742 743 744 745 746
struct progress {
	struct apk_stats done;
	struct apk_stats total;
	struct apk_package *pkg;
	size_t count;
};

static void progress_cb(void *ctx, size_t progress)
747
{
748 749
	struct progress *prog = (struct progress *) ctx;
	size_t partial = 0, count;
750

751 752
	if (prog->pkg != NULL)
		partial = muldiv(progress, prog->pkg->installed_size, APK_PROGRESS_SCALE);
753

754
        count = muldiv(100, prog->done.bytes + prog->done.packages + partial,
755
		       prog->total.bytes + prog->total.packages);
756

757
	if (prog->count != count)
758
		apk_draw_progress(count);
759
	prog->count = count;
760 761
}

762 763 764
static int dump_packages(struct apk_state *state,
			 int (*cmp)(struct apk_change *change),
			 const char *msg)
765
{
766
	struct apk_change *change;
767
	struct apk_name *name;
Timo Teräs's avatar
Timo Teräs committed
768 769 770
	struct apk_indent indent = { 0, 2 };
	char tmp[256];
	int match = 0, i;
771

772
	list_for_each_entry(change, &state->change_list_head, change_list) {
773 774 775
		if (!cmp(change))
			continue;
		if (match == 0)
Timo Teräs's avatar
Timo Teräs committed
776
			printf("%s:\n ", msg);
777 778 779 780 781
		if (change->newpkg != NULL)
			name = change->newpkg->name;
		else
			name = change->oldpkg->name;

Timo Teräs's avatar
Timo Teräs committed
782 783 784
		i = snprintf(tmp, sizeof(tmp), "%s%s", name->name,
			     (name->flags & APK_NAME_TOPLEVEL) ? "*" : "");
		apk_print_indented(&indent, APK_BLOB_PTR_LEN(tmp, i));
785
		match++;
786
	}
787
	if (match)
Timo Teräs's avatar
Timo Teräs committed
788
		printf("\n");
789
	return match;
790 791
}

792
static int cmp_remove(struct apk_change *change)
793
{
794 795
	return change->newpkg == NULL;
}
796

797 798 799 800
static int cmp_new(struct apk_change *change)
{
	return change->oldpkg == NULL;
}
801

802 803 804
static int cmp_downgrade(struct apk_change *change)
{
	if (change->newpkg == NULL || change->oldpkg == NULL)
805
		return 0;
806
	if (apk_pkg_version_compare(change->newpkg, change->oldpkg)
807 808
	    & APK_VERSION_LESS)
		return 1;
809
	return 0;
810 811
}

812
static int cmp_upgrade(struct apk_change *change)
813
{
814
	if (change->newpkg == NULL || change->oldpkg == NULL)
815
		return 0;
816

817 818 819 820
	/* Count swapping package as upgrade too - this can happen if
	 * same package version is used after it was rebuilt against
	 * newer libraries. Basically, different (and probably newer)
	 * package, but equal version number. */
821 822 823
	if ((apk_pkg_version_compare(change->newpkg, change->oldpkg) &
	     (APK_VERSION_GREATER | APK_VERSION_EQUAL)) &&
	    (change->newpkg != change->oldpkg))
824
		return 1;
825

826 827 828
	return 0;
}

Timo Teräs's avatar
Timo Teräs committed
829
static int fail_if_something_broke(struct apk_state *state,
830 831 832
				   struct apk_package *pkg,
				   struct apk_dependency *dep,
				   void *ctx)
Timo Teräs's avatar
Timo Teräs committed
833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854

{
	return 1;
}

static int apk_state_autoclean(struct apk_state *state,
			       struct apk_package *pkg)
{
	apk_name_state_t oldns;
	int i, r;

	for (i = 0; i < pkg->depends->num; i++) {
		struct apk_name *n = pkg->depends->item[i].name;

		if (ns_locked(state->name[n->id]))
			continue;
		if (n->flags & APK_NAME_TOPLEVEL)
			continue;

		oldns = state->name[n->id];
		state->name[n->id] = ns_from_pkg(NULL);
		r = for_each_broken_reverse_depency(state, n,
855 856
						    fail_if_something_broke,
						    NULL);
Timo Teräs's avatar
Timo Teräs committed
857 858 859 860 861 862 863 864
		state->name[n->id] = oldns;

		if (r == 0) {
			r = apk_state_lock_name(state, n, NULL);
			if (r != 0)
				return r;
		}
	}
865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881

	for (i = 0; i < pkg->name->rinstall_if->num; i++) {
		struct apk_name *n = pkg->name->rinstall_if->item[i];

		if (ns_locked(state->name[n->id]))
			continue;
		if (n->flags & APK_NAME_TOPLEVEL)
			continue;

		r = apk_state_autolock_name(state, n, TRUE);
		if (r == -2) {
			r = apk_state_lock_name(state, n, NULL);
			if (r != 0)
				return r;
		}
	}

Timo Teräs's avatar
Timo Teräs committed
882 883 884
	return 0;
}

885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902
struct error_state {
	struct apk_indent indent;
	struct apk_package *prevpkg;
};

static int print_dep(struct apk_state *state,
		     struct apk_package *pkg,
		     struct apk_dependency *dep,
		     void *ctx)
{
	struct error_state *es = (struct error_state *) ctx;
	apk_blob_t blob;
	char buf[256];
	int len;

	if (pkg != es->prevpkg) {
		printf("\n");
		es->indent.x = 0;
903 904
		len = snprintf(buf, sizeof(buf), PKG_VER_FMT ":",
			       PKG_VER_PRINTF(pkg));
905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922
		apk_print_indented(&es->indent, APK_BLOB_PTR_LEN(buf, len));
		es->prevpkg = pkg;
	}

	blob = APK_BLOB_BUF(buf);
	apk_blob_push_dep(&blob, dep);
	blob = apk_blob_pushed(APK_BLOB_BUF(buf), blob);
	apk_print_indented(&es->indent, blob);

	return 0;
}

void apk_state_print_errors(struct apk_state *state)
{
	struct apk_package *pkg;
	struct error_state es;
	int i, j, r;

Timo Teräs's avatar
Timo Teräs committed
923
	for (i = 0; i < state->conflicts->num; i++) {
924 925
		if (i == 0)
			apk_error("Unable to satisfy all dependencies:");
926 927

		es.prevpkg = pkg = state->conflicts->item[i];
Timo Teräs's avatar
Timo Teräs committed
928
		es.indent.x =
929
		printf("  " PKG_VER_FMT ":", PKG_VER_PRINTF(pkg));
Timo Teräs's avatar
Timo Teräs committed
930
		es.indent.indent = es.indent.x + 1;
Timo Teräs's avatar
Timo Teräs committed
931
		for (j = 0; j < pkg->depends->num; j++) {
932 933 934 935 936 937 938 939 940 941 942
			r = apk_state_lock_dependency(state,
						      &pkg->depends->item[j]);
			if (r != 0)
				print_dep(state, pkg, &pkg->depends->item[j], &es);
		}

		/* Print conflicting reverse deps */
		for_each_broken_reverse_depency(state, pkg->name,
						print_dep, &es);
		printf("\n");
	}
943

Timo Teräs's avatar
Timo Teräs committed
944
	for (i = 0; i < state->missing->num; i++) {
945 946 947 948 949 950 951 952 953 954
		struct apk_name *name = state->missing->item[i];
		if (i == 0) {
			apk_error("Missing packages:");
			es.indent.x = 0;
			es.indent.indent = 2;
		}
		apk_print_indented(&es.indent, APK_BLOB_STR(name->name));
	}
	if (i != 0)
		printf("\n");
955 956
}

957
int apk_state_commit(struct apk_state *state)
958
{
959 960
	struct progress prog;
	struct apk_change *change;
961
	struct apk_database *db = state->db;
962
	int n = 0, r = 0, size_diff = 0, toplevel = FALSE, deleteonly = TRUE;
963

964 965
	/* Count what needs to be done */
	memset(&prog, 0, sizeof(prog));
Timo Teräs's avatar
Timo Teräs committed
966
	list_for_each_entry(change, &state->change_list_head, change_list) {
967 968 969 970 971
		if (change->newpkg == NULL) {
			if (change->oldpkg->name->flags & APK_NAME_TOPLEVEL)
				toplevel = TRUE;
		} else
			deleteonly = FALSE;
972 973
		if (change->oldpkg != NULL)
			apk_state_autoclean(state, change->oldpkg);
974
		apk_count_change(change, &prog.total);
975 976 977 978
		if (change->newpkg)
			size_diff += change->newpkg->installed_size;
		if (change->oldpkg)
			size_diff -= change->oldpkg->installed_size;
Timo Teräs's avatar
Timo Teräs committed
979
	}
980
	size_diff /= 1024;
981

982 983 984 985 986 987 988 989 990 991 992
	if (toplevel &&
	    (apk_flags & (APK_INTERACTIVE | APK_RECURSIVE_DELETE)) == 0) {
		if (!deleteonly)
			return -1;

		dump_packages(state, cmp_remove,
			      "The top-level dependencies have been updated "
			      "but the following packages are not removed");
		goto update_state;
	}

993
	if (apk_verbosity > 1 || (apk_flags & APK_INTERACTIVE)) {
994 995 996 997
		r = dump_packages(state, cmp_remove,
				  "The following packages will be REMOVED");
		r += dump_packages(state, cmp_downgrade,
				   "The following packages will be DOWNGRADED");
998
		if (r || (apk_flags & APK_INTERACTIVE) || apk_verbosity > 2) {
999 1000 1001 1002
			dump_packages(state, cmp_new,
				      "The following NEW packages will be installed");
			dump_packages(state, cmp_upgrade,
				      "The following packages will be upgraded");
1003
			printf("After this operation, %d kB of %s\n", abs(size_diff),
1004
			       (size_diff < 0) ?
1005 1006
			       "disk space will be freed." :
			       "additional disk space will be used.");
1007 1008
		}
		if (apk_flags & APK_INTERACTIVE) {
Timo Teräs's avatar
Timo Teräs committed
1009 1010
			printf("Do you want to continue [Y/n]? ");
			fflush(stdout);
1011
			r = fgetc(stdin);
1012
			if (r != 'y' && r != 'Y' && r != '\n')
1013 1014 1015 1016 1017
				return -1;
		}
	}

	/* Go through changes */
1018
	r = 0;
1019
	n = 0;
Timo Teräs's avatar