VitaShell/package_installer.c

/*
	VitaShell
	Copyright (C) 2015-2017, TheFloW

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <psp2/sysmodule.h>
#include <psp2/promoterutil.h>

#include "main.h"
#include "io_process.h"
#include "package_installer.h"
#include "archive.h"
#include "file.h"
#include "message_dialog.h"
#include "language.h"
#include "utils.h"
#include "sfo.h"
#include "sha1.h"

#include "resources/base_head_bin.h"

static int patchRetailContents() {
	int res;
	
	SceIoStat stat;
	memset(&stat, 0, sizeof(SceIoStat));
	res = sceIoGetstat(PACKAGE_DIR "/sce_sys/retail/livearea", &stat);
	if (res < 0)
		return res;

	res = sceIoRename(PACKAGE_DIR "/sce_sys/livearea", PACKAGE_DIR "/sce_sys/livearea_org");
	if (res < 0)
		return res;

	res = sceIoRename(PACKAGE_DIR "/sce_sys/retail/livearea", PACKAGE_DIR "/sce_sys/livearea");
	if (res < 0)
		return res;

	return 0;
}

static int restoreRetailContents(const char *titleid) {
	int res;
	char src_path[128], dst_path[128];

	sprintf(src_path, "ux0:app/%s/sce_sys/livearea", titleid);
	sprintf(dst_path, "ux0:app/%s/sce_sys/retail/livearea", titleid);
	res = sceIoRename(src_path, dst_path);
	if (res < 0)
		return res;

	sprintf(src_path, "ux0:app/%s/sce_sys/livearea_org", titleid);
	sprintf(dst_path, "ux0:app/%s/sce_sys/livearea", titleid);
	res = sceIoRename(src_path, dst_path);
	if (res < 0)
		return res;

	return 0;
}

int promoteUpdate(const char *path, const char *titleid, const char *category, void *sfo_buffer, int sfo_size) {
	int res;

	// Update installation
	if (strcmp(category, "gp") == 0) {
		// Change category to 'gd'
		setSfoString(sfo_buffer, "CATEGORY", "gd");
		WriteFile(PACKAGE_DIR "/sce_sys/param.sfo", sfo_buffer, sfo_size);

		// App path
		char app_path[MAX_PATH_LENGTH];
		snprintf(app_path, MAX_PATH_LENGTH, "ux0:app/%s", titleid);

		/*
			Without the following trick, the livearea won't be updated and the game will even crash
		*/

		// Integrate patch to app
		res = movePath(path, app_path, MOVE_INTEGRATE | MOVE_REPLACE, NULL);
		if (res < 0)
			return res;

		// Move app to promotion directory
		res = movePath(app_path, path, 0, NULL);
		if (res < 0)
			return res;
	}

	return 0;
}

int promoteApp(const char *path) {
	int res;

	// Read param.sfo
	void *sfo_buffer = NULL;
	int sfo_size = allocateReadFile(PACKAGE_DIR "/sce_sys/param.sfo", &sfo_buffer);
	if (sfo_size < 0)
		return sfo_size;

	// Get titleid
	char titleid[12];
	getSfoString(sfo_buffer, "TITLE_ID", titleid, sizeof(titleid));

	// Get category
	char category[4];
	getSfoString(sfo_buffer, "CATEGORY", category, sizeof(category));

	// Promote update
	promoteUpdate(path, titleid, category, sfo_buffer, sfo_size);

	// Free sfo buffer
	free(sfo_buffer);

	// Patch to use retail contents so the game is not shown as test version
	int patch_retail_contents = patchRetailContents();

	res = scePromoterUtilityInit();
	if (res < 0)
		return res;

	res = scePromoterUtilityPromotePkg(path, 0);
	if (res < 0)
		return res;

	int state = 0;
	do {
		res = scePromoterUtilityGetState(&state);
		if (res < 0)
			return res;

		sceKernelDelayThread(100 * 1000);
	} while (state);

	int ret = 0;
	res = scePromoterUtilityGetResult(&ret);
	if (res < 0)
		return res;

	res = scePromoterUtilityExit();
	if (res < 0)
		return res;

	// Restore
	if (patch_retail_contents >= 0)
		restoreRetailContents(titleid);

	// Using the promoteUpdate trick, we get 0x80870005 as result, but it installed correctly though, so return ok
	return ret == 0x80870005 ? 0 : ret;
}

int deleteApp(const char *titleid) {
	int res;

	res = sceAppMgrDestroyOtherApp();
	if (res < 0)
		return res;

	res = scePromoterUtilityInit();
	if (res < 0)
		return res;

	res = scePromoterUtilityDeletePkg(titleid);
	if (res < 0)
		return res;

	int state = 0;
	do {
		res = scePromoterUtilityGetState(&state);
		if (res < 0)
			return res;

		sceKernelDelayThread(300 * 1000);
	} while (state);

	int ret = 0;
	res = scePromoterUtilityGetResult(&ret);
	if (res < 0)
		return res;

	res = scePromoterUtilityExit();
	if (res < 0)
		return res;

	return ret;
}

static void fpkg_hmac(const uint8_t *data, unsigned int len, uint8_t hmac[16]) {
	SHA1_CTX ctx;
	uint8_t sha1[20];
	uint8_t buf[64];

	sha1_init(&ctx);
	sha1_update(&ctx, data, len);
	sha1_final(&ctx, sha1);

	memset(buf, 0, 64);
	memcpy(&buf[0], &sha1[4], 8);
	memcpy(&buf[8], &sha1[4], 8);
	memcpy(&buf[16], &sha1[12], 4);
	buf[20] = sha1[16];
	buf[21] = sha1[1];
	buf[22] = sha1[2];
	buf[23] = sha1[3];
	memcpy(&buf[24], &buf[16], 8);

	sha1_init(&ctx);
	sha1_update(&ctx, buf, 64);
	sha1_final(&ctx, sha1);
	memcpy(hmac, sha1, 16);
}

int makeHeadBin() {
	uint8_t hmac[16];
	uint32_t off;
	uint32_t len;
	uint32_t out;

	// Read param.sfo
	void *sfo_buffer = NULL;
	int res = allocateReadFile(PACKAGE_DIR "/sce_sys/param.sfo", &sfo_buffer);
	if (res < 0)
		return res;

	// Get title id
	char titleid[12];
	memset(titleid, 0, sizeof(titleid));
	getSfoString(sfo_buffer, "TITLE_ID", titleid, sizeof(titleid));

	// Enforce TITLE_ID format
	if (strlen(titleid) != 9)
		return -1;

	// Get content id
	char contentid[48];
	memset(contentid, 0, sizeof(contentid));
	getSfoString(sfo_buffer, "CONTENT_ID", contentid, sizeof(contentid));

	// Free sfo buffer
	free(sfo_buffer);

	// Allocate head.bin buffer
	uint8_t *head_bin = malloc(sizeof(base_head_bin));
	memcpy(head_bin, base_head_bin, sizeof(base_head_bin));

	// Write full title id
	char full_title_id[48];
	snprintf(full_title_id, sizeof(full_title_id), "EP9000-%s_00-XXXXXXXXXXXXXXXX", titleid);
	strncpy((char *)&head_bin[0x30], strlen(contentid) > 0 ? contentid : full_title_id, 48);

	// hmac of pkg header
	len = ntohl(*(uint32_t *)&head_bin[0xD0]);
	fpkg_hmac(&head_bin[0], len, hmac);
	memcpy(&head_bin[len], hmac, 16);

	// hmac of pkg info
	off = ntohl(*(uint32_t *)&head_bin[0x8]);
	len = ntohl(*(uint32_t *)&head_bin[0x10]);
	out = ntohl(*(uint32_t *)&head_bin[0xD4]);
	fpkg_hmac(&head_bin[off], len - 64, hmac);
	memcpy(&head_bin[out], hmac, 16);

	// hmac of everything
	len = ntohl(*(uint32_t *)&head_bin[0xE8]);
	fpkg_hmac(&head_bin[0], len, hmac);
	memcpy(&head_bin[len], hmac, 16);

	// Make dir
	sceIoMkdir(PACKAGE_DIR "/sce_sys/package", 0777);

	// Write head.bin
	WriteFile(HEAD_BIN, head_bin, sizeof(base_head_bin));

	free(head_bin);

	return 0;
}

int installPackage(const char *file) {
	int res;

	// Recursively clean up pkg directory
	removePath(PACKAGE_DIR, NULL);

	// Open archive
	res = archiveOpen(file);
	if (res < 0)
		return res;

	// Src path
	char src_path[MAX_PATH_LENGTH];
	strcpy(src_path, file);
	addEndSlash(src_path);

	// Extract process
	res = extractArchivePath(src_path, PACKAGE_DIR "/", NULL);
	if (res < 0)
		return res;

	// Close archive
	res = archiveClose();
	if (res < 0)
		return res;

	// Make head.bin
	res = makeHeadBin();
	if (res < 0)
		return res;

	// Promote app
	res = promoteApp(PACKAGE_DIR);
	if (res < 0)
		return res;

	return 0;
}

int install_thread(SceSize args_size, InstallArguments *args) {
	int res;
	SceUID thid = -1;
	char path[MAX_PATH_LENGTH];
	SceIoStat stat;
	int isFolder = 0;

	// Lock power timers
	powerLock();

	// Set progress to 0%
	sceMsgDialogProgressBarSetValue(SCE_MSG_DIALOG_PROGRESSBAR_TARGET_BAR_DEFAULT, 0);
	sceKernelDelayThread(DIALOG_WAIT); // Needed to see the percentage

	// Recursively clean up pkg directory
	removePath(PACKAGE_DIR, NULL);

	res = sceIoGetstat(args->file, &stat);
	if (res < 0) {
		closeWaitDialog();
		errorDialog(res);
		goto EXIT;
	}

	if (SCE_S_ISDIR(stat.st_mode)) {
		// Check for param.sfo
		snprintf(path, MAX_PATH_LENGTH, "%s/sce_sys/param.sfo", args->file);
		if (sceIoGetstat(path, &stat) < 0 || SCE_S_ISDIR(stat.st_mode)) {
			closeWaitDialog();
			errorDialog(-2);
			goto EXIT;
		}

		// Check permissions
		snprintf(path, MAX_PATH_LENGTH, "%s/eboot.bin", args->file);
		SceUID fd = sceIoOpen(path, SCE_O_RDONLY, 0);
		if (fd >= 0) {
			char buffer[0x88];
			sceIoRead(fd, buffer, sizeof(buffer));
			sceIoClose(fd);

			// Team molecule's request: Full permission access warning
			uint64_t authid = *(uint64_t *)(buffer + 0x80);
			if (authid != 0x2F00000000000002) {
				closeWaitDialog();

				initMessageDialog(SCE_MSG_DIALOG_BUTTON_TYPE_YESNO, language_container[INSTALL_WARNING]);
				setDialogStep(DIALOG_STEP_INSTALL_WARNING);

				// Wait for response
				while (getDialogStep() == DIALOG_STEP_INSTALL_WARNING) {
					sceKernelDelayThread(10 * 1000);
				}

				// Cancelled
				if (getDialogStep() == DIALOG_STEP_CANCELLED) {
					closeWaitDialog();
					goto EXIT;
				}

				// Init again
				initMessageDialog(MESSAGE_DIALOG_PROGRESS_BAR, language_container[INSTALLING]);
				setDialogStep(DIALOG_STEP_INSTALLING);
			}
		}

		res = sceIoRename(args->file, PACKAGE_DIR);
		if (res < 0) {
			closeWaitDialog();
			setDialogStep(DIALOG_STEP_CANCELLED);
			errorDialog(res);
			goto EXIT;
		}
		sceMsgDialogProgressBarSetValue(SCE_MSG_DIALOG_PROGRESSBAR_TARGET_BAR_DEFAULT, 50);
		sceKernelDelayThread(COUNTUP_WAIT);

		isFolder = 1;
	} else {
		// Open archive
		res = archiveOpen(args->file);
		if (res < 0) {
			closeWaitDialog();
			errorDialog(res);
			goto EXIT;
		}

		// If you cancelled at the time archiveOpen was working,
		// it would still open the full permission dialog instead of termiating.
		// So terminate now
		if (cancelHandler()) {
			closeWaitDialog();
			setDialogStep(DIALOG_STEP_CANCELLED);
			goto EXIT;
		}

		// Check for param.sfo
		snprintf(path, MAX_PATH_LENGTH, "%s/sce_sys/param.sfo", args->file);
		if (archiveFileGetstat(path, NULL) < 0) {
			closeWaitDialog();
			errorDialog(-2);
			goto EXIT;
		}

		// Team molecule's request: Full permission access warning
		int unsafe = archiveCheckFilesForUnsafeFself(); // 0: Safe, 1: Unsafe, 2: Dangerous
		if (unsafe) {
			closeWaitDialog();

			initMessageDialog(SCE_MSG_DIALOG_BUTTON_TYPE_YESNO, language_container[unsafe == 2 ? INSTALL_BRICK_WARNING : INSTALL_WARNING]);
			setDialogStep(DIALOG_STEP_INSTALL_WARNING);

			// Wait for response
			while (getDialogStep() == DIALOG_STEP_INSTALL_WARNING) {
				sceKernelDelayThread(10 * 1000);
			}

			// Cancelled
			if (getDialogStep() == DIALOG_STEP_CANCELLED) {
				closeWaitDialog();
				goto EXIT;
			}

			// Init again
			initMessageDialog(MESSAGE_DIALOG_PROGRESS_BAR, language_container[INSTALLING]);
			setDialogStep(DIALOG_STEP_INSTALLING);
		}

		// Src path
		char src_path[MAX_PATH_LENGTH];
		strcpy(src_path, args->file);
		addEndSlash(src_path);

		// Get archive path info
		uint64_t size = 0;
		uint32_t folders = 0, files = 0;
		getArchivePathInfo(src_path, &size, &folders, &files);

		// Check memory card free space
		if (checkMemoryCardFreeSpace(size))
			goto EXIT;

		// Update thread
		thid = createStartUpdateThread(size + folders*DIRECTORY_SIZE, 1);

		// Extract process
		uint64_t value = 0;

		FileProcessParam param;
		param.value = &value;
		param.max = size + folders * DIRECTORY_SIZE;
		param.SetProgress = SetProgress;
		param.cancelHandler = cancelHandler;

		res = extractArchivePath(src_path, PACKAGE_DIR "/", &param);
		if (res <= 0) {
			closeWaitDialog();
			setDialogStep(DIALOG_STEP_CANCELLED);
			errorDialog(res);
			goto EXIT;
		}

		// Close archive
		res = archiveClose();
		if (res < 0) {
			closeWaitDialog();
			errorDialog(res);
			goto EXIT;
		}
	}

	// Make head.bin
	res = makeHeadBin();
	if (res < 0) {
		closeWaitDialog();
		errorDialog(res);
		// If failed, move package folder back
		if (isFolder) sceIoRename(PACKAGE_DIR, args->file);
		goto EXIT;
	}

	// Promote app
	res = promoteApp(PACKAGE_DIR);
	if (res < 0) {
		closeWaitDialog();
		errorDialog(res);
		// If failed, move package folder back
		if (isFolder) sceIoRename(PACKAGE_DIR, args->file);
		goto EXIT;
	}

	// Set progress to 100%
	sceMsgDialogProgressBarSetValue(SCE_MSG_DIALOG_PROGRESSBAR_TARGET_BAR_DEFAULT, 100);
	sceKernelDelayThread(COUNTUP_WAIT);

	// Close
	sceMsgDialogClose();

	setDialogStep(DIALOG_STEP_INSTALLED);

EXIT:
	if (thid >= 0)
		sceKernelWaitThreadEnd(thid, NULL, NULL);

	// Recursively clean up package_temp directory
	removePath(PACKAGE_DIR, NULL);

	// Unlock power timers
	powerUnlock();

	return sceKernelExitDeleteThread(0);
}