Handle enable rollback for the apk session part of a staged install (ac0d0f9b) · Commits · e / os / android_frameworks_base

services/core/java/com/android/server/rollback/RollbackManagerServiceImpl.java

+51 −0

Original line number	Diff line number	Diff line
		@@ -725,6 +725,7 @@ class RollbackManagerServiceImpl extends IRollbackManager.Stub {
		// ourselves.
		PackageInstaller.SessionInfo session = null;

		int parentSessionId = -1;
		PackageInstaller installer = mContext.getPackageManager().getPackageInstaller();
		for (PackageInstaller.SessionInfo info : installer.getAllSessions()) {
		if (info.isMultiPackage()) {
		@@ -732,12 +733,14 @@ class RollbackManagerServiceImpl extends IRollbackManager.Stub {
		PackageInstaller.SessionInfo child = installer.getSessionInfo(childId);
		if (sessionMatchesForEnableRollback(child, installFlags, newPackageCodePath)) {
		// TODO: Check we only have one matching session?
		parentSessionId = info.getSessionId();
		session = child;
		break;
		}
		}
		} else if (sessionMatchesForEnableRollback(info, installFlags, newPackageCodePath)) {
		// TODO: Check we only have one matching session?
		parentSessionId = info.getSessionId();
		session = info;
		break;
		}
		@@ -748,6 +751,54 @@ class RollbackManagerServiceImpl extends IRollbackManager.Stub {
		return false;
		}

		// Check to see if this is the apk session for a staged session with
		// rollback enabled.
		// TODO: This check could be made more efficient.
		RollbackData rd = null;
		synchronized (mLock) {
		ensureRollbackDataLoadedLocked();
		for (int i = 0; i < mAvailableRollbacks.size(); ++i) {
		RollbackData data = mAvailableRollbacks.get(i);
		if (data.apkSessionId == parentSessionId) {
		rd = data;
		break;
		}
		}
		}

		if (rd != null) {
		// This is the apk session for a staged session. We have already
		// backed up the apks, we just need to do user data backup.
		PackageParser.PackageLite newPackage = null;
		try {
		newPackage = PackageParser.parsePackageLite(
		new File(session.resolvedBaseCodePath), 0);
		} catch (PackageParser.PackageParserException e) {
		Log.e(TAG, "Unable to parse new package", e);
		return false;
		}
		String packageName = newPackage.packageName;
		for (PackageRollbackInfo info : rd.packages) {
		if (info.getPackageName().equals(packageName)) {
		IntArray pendingBackups = mUserdataHelper.snapshotAppData(
		packageName, installedUsers);
		info.getPendingBackups().addAll(pendingBackups);
		try {
		mRollbackStore.saveAvailableRollback(rd);
		} catch (IOException ioe) {
		// TODO: Hopefully this is okay because we will try
		// again to save the rollback when the staged session
		// is applied. Just so long as the device doesn't
		// reboot before then.
		Log.e(TAG, "Unable to save rollback info for : " + rd.rollbackId, ioe);
		}
		return true;
		}
		}
		Log.e(TAG, "Unable to find package in apk session");
		return false;
		}

		return enableRollbackForSession(session, installedUsers, true);
		}