VitaShell/network_download.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 "main.h"
#include "io_process.h"
#include "network_download.h"
#include "package_installer.h"
#include "archive.h"
#include "file.h"
#include "message_dialog.h"
#include "language.h"
#include "utils.h"

#define VITASHELL_USER_AGENT "VitaShell/1.00 libhttp/1.1"

int getDownloadFileSize(const char *src, uint64_t *size) {
	int res;
	int statusCode;
	int tmplId = -1, connId = -1, reqId = -1;

	res = sceHttpCreateTemplate(VITASHELL_USER_AGENT, SCE_HTTP_VERSION_1_1, SCE_TRUE);
	if (res < 0)
		goto ERROR_EXIT;

	tmplId = res;

	res = sceHttpCreateConnectionWithURL(tmplId, src, SCE_TRUE);
	if (res < 0)
		goto ERROR_EXIT;

	connId = res;

	res = sceHttpCreateRequestWithURL(connId, SCE_HTTP_METHOD_GET, src, 0);
	if (res < 0)
		goto ERROR_EXIT;

	reqId = res;

	res = sceHttpSendRequest(reqId, NULL, 0);
	if (res < 0)
		goto ERROR_EXIT;

	res = sceHttpGetStatusCode(reqId, &statusCode);
	if (res < 0)
		goto ERROR_EXIT;

	if (statusCode == 200) {
		res = sceHttpGetResponseContentLength(reqId, size);
	}

ERROR_EXIT:
	if (reqId >= 0)
		sceHttpDeleteRequest(reqId);

	if (connId >= 0)
		sceHttpDeleteConnection(connId);

	if (tmplId >= 0)
		sceHttpDeleteTemplate(tmplId);

	return res;
}

int getFieldFromHeader(const char *src, const char *field, const char **data, unsigned int *valueLen) {
	int res;
	char *header;
	unsigned int headerSize;
	int tmplId = -1, connId = -1, reqId = -1;

	res = sceHttpCreateTemplate(VITASHELL_USER_AGENT, SCE_HTTP_VERSION_1_1, SCE_TRUE);
	if (res < 0)
		goto ERROR_EXIT;

	tmplId = res;

	res = sceHttpCreateConnectionWithURL(tmplId, src, SCE_TRUE);
	if (res < 0)
		goto ERROR_EXIT;

	connId = res;
	
	res = sceHttpCreateRequestWithURL(connId, SCE_HTTP_METHOD_GET, src,  0);
	if (res < 0)
		goto ERROR_EXIT;
	
	reqId = res;
	
	res = sceHttpSendRequest(reqId, NULL, 0);
	if (res < 0)
		goto ERROR_EXIT;

	res = sceHttpGetAllResponseHeaders(reqId, &header, &headerSize);
	if (res < 0)
		goto ERROR_EXIT;

	res = sceHttpParseResponseHeader(header, headerSize, field, data, valueLen);
	if (res < 0) {
		*data = "";
		*valueLen = 0;
		res = 0;
	}
	
ERROR_EXIT:
	if (reqId >= 0)
		sceHttpDeleteRequest(reqId);

	if (connId >= 0)
		sceHttpDeleteConnection(connId);

	if (tmplId >= 0)
		sceHttpDeleteTemplate(tmplId);

	return res;
}

int downloadFile(const char *src, const char *dst, FileProcessParam *param) {
	int res;
	int statusCode;
	int tmplId = -1, connId = -1, reqId = -1;
	SceUID fd = -1;
	int ret = 1;

	res = sceHttpCreateTemplate(VITASHELL_USER_AGENT, SCE_HTTP_VERSION_1_1, SCE_TRUE);
	if (res < 0)
		goto ERROR_EXIT;

	tmplId = res;

	res = sceHttpCreateConnectionWithURL(tmplId, src, SCE_TRUE);
	if (res < 0)
		goto ERROR_EXIT;

	connId = res;

	res = sceHttpCreateRequestWithURL(connId, SCE_HTTP_METHOD_GET, src, 0);
	if (res < 0)
		goto ERROR_EXIT;

	reqId = res;

	res = sceHttpSendRequest(reqId, NULL, 0);
	if (res < 0)
		goto ERROR_EXIT;

	res = sceHttpGetStatusCode(reqId, &statusCode);
	if (res < 0)
		goto ERROR_EXIT;

	if (statusCode == 200) {		
		res = sceIoOpen(dst, SCE_O_WRONLY | SCE_O_CREAT | SCE_O_TRUNC, 0777);
		if (res < 0)
			goto ERROR_EXIT;

		fd = res;

		uint8_t buf[4096];

		while (1) {
			int read = sceHttpReadData(reqId, buf, sizeof(buf));
			
			if (read < 0) {
				res = read;
				break;
			}
			
			if (read == 0)
				break;

			int written = sceIoWrite(fd, buf, read);
			
			if (written < 0) {
				res = written;
				break;
			}

			if (param) {
				if (param->value)
					(*param->value) += read;

				if (param->SetProgress)
					param->SetProgress(param->value ? *param->value : 0, param->max);

				if (param->cancelHandler && param->cancelHandler()) {
					ret = 0;
					break;
				}
			}
		}
	}

ERROR_EXIT:
	if (fd >= 0)
		sceIoClose(fd);

	if (reqId >= 0)
		sceHttpDeleteRequest(reqId);

	if (connId >= 0)
		sceHttpDeleteConnection(connId);

	if (tmplId >= 0)
		sceHttpDeleteTemplate(tmplId);

	if (res < 0)
		return res;

	return ret;
}

int downloadFileProcess(const char *url, const char *dest, int successStep) {
	SceUID thid = -1;

	// 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

	// File size
	uint64_t size = 0;
	getDownloadFileSize(url, &size);

	// Update thread
	thid = createStartUpdateThread(size, 1);

	// Download
	uint64_t value = 0;
	
	FileProcessParam param;
	param.value = &value;
	param.max = size;
	param.SetProgress = SetProgress;
	param.cancelHandler = cancelHandler;

	int res = downloadFile(url, dest, &param);
	if (res <= 0) {
		sceIoRemove(dest);
		closeWaitDialog();
		setDialogStep(DIALOG_STEP_CANCELLED);
		errorDialog(res);
		goto EXIT;
	}

	// Set progress to 100%
	sceMsgDialogProgressBarSetValue(SCE_MSG_DIALOG_PROGRESSBAR_TARGET_BAR_DEFAULT, 100);
	sceKernelDelayThread(COUNTUP_WAIT);
	
	// Close
	if (successStep != 0) {
		sceMsgDialogClose();
		setDialogStep(successStep);
	}

EXIT:
	if (thid >= 0)
		sceKernelWaitThreadEnd(thid, NULL, NULL);
	
	// Unlock power timers
	powerUnlock();

	return sceKernelExitDeleteThread(0);
}