VitaShell/utils.c
2018-03-14 17:11:55 +01:00

427 lines
11 KiB
C

/*
VitaShell
Copyright (C) 2015-2018, 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 "init.h"
#include "file.h"
#include "message_dialog.h"
#include "uncommon_dialog.h"
#include "theme.h"
#include "language.h"
#include "utils.h"
#include "bm.h"
SceCtrlData pad;
Pad old_pad, current_pad, pressed_pad, released_pad, hold_pad, hold2_pad;
Pad hold_count, hold2_count;
static int netdbg_sock = -1;
static void *net_memory = NULL;
static int net_init = -1;
static int lock_power = 0;
float easeOut(float x0, float x1, float a, float b) {
float dx = (x1 - x0);
return ((dx * a) > b) ? (dx * a) : dx;
}
void startDrawing(vita2d_texture *bg) {
vita2d_start_drawing();
vita2d_set_clear_color(BACKGROUND_COLOR);
vita2d_clear_screen();
if (wallpaper_image) {
vita2d_draw_texture(wallpaper_image, 0.0f, 0.0f);
} else if (bg) {
vita2d_draw_texture(bg, 0.0f, 0.0f);
}
}
void endDrawing() {
drawUncommonDialog();
vita2d_end_drawing();
vita2d_common_dialog_update();
vita2d_swap_buffers();
sceDisplayWaitVblankStart();
}
void closeWaitDialog() {
sceMsgDialogClose();
while (updateMessageDialog() != MESSAGE_DIALOG_RESULT_NONE) {
sceKernelDelayThread(1000);
}
}
void errorDialog(int error) {
if (error < 0) {
initMessageDialog(SCE_MSG_DIALOG_BUTTON_TYPE_OK, language_container[ERROR], error);
setDialogStep(DIALOG_STEP_ERROR);
}
}
void infoDialog(const char *msg, ...) {
va_list list;
char string[512];
va_start(list, msg);
vsnprintf(string, sizeof(string), msg, list);
va_end(list);
initMessageDialog(SCE_MSG_DIALOG_BUTTON_TYPE_OK, string);
setDialogStep(DIALOG_STEP_INFO);
}
int checkMemoryCardFreeSpace(const char *path, uint64_t size) {
char device[8];
uint64_t free_size = 0, max_size = 0, extra_space = 0;
char *p = strchr(path, ':');
if (p) {
strncpy(device, path, p - path + 1);
device[p - path + 1] = '\0';
}
if (strcmp(device, "ux0:") == 0) {
extra_space = 40 * 1024 * 1024;
}
if (is_safe_mode) {
sceAppMgrGetDevInfo(device, &max_size, &free_size);
} else {
SceIoDevInfo info;
memset(&info, 0, sizeof(SceIoDevInfo));
sceIoDevctl(device, 0x3001, NULL, 0, &info, sizeof(SceIoDevInfo));
free_size = info.free_size;
max_size = info.max_size;
}
if (size >= (free_size + extra_space)) {
closeWaitDialog();
char size_string[16];
getSizeString(size_string, size - (free_size + extra_space));
infoDialog(language_container[NO_SPACE_ERROR], size_string);
return 1;
}
return 0;
}
static int power_tick_thread(SceSize args, void *argp) {
while (1) {
if (lock_power > 0) {
sceKernelPowerTick(SCE_KERNEL_POWER_TICK_DISABLE_AUTO_SUSPEND);
}
sceKernelDelayThread(10 * 1000 * 1000);
}
return 0;
}
void initPowerTickThread() {
SceUID thid = sceKernelCreateThread("power_tick_thread", power_tick_thread, 0x10000100, 0x40000, 0, 0, NULL);
if (thid >= 0)
sceKernelStartThread(thid, 0, NULL);
}
void powerLock() {
if (!lock_power)
sceShellUtilLock(SCE_SHELL_UTIL_LOCK_TYPE_PS_BTN);
lock_power++;
}
void powerUnlock() {
if (lock_power)
sceShellUtilUnlock(SCE_SHELL_UTIL_LOCK_TYPE_PS_BTN);
lock_power--;
if (lock_power < 0)
lock_power = 0;
}
void readPad() {
memset(&pad, 0, sizeof(SceCtrlData));
sceCtrlPeekBufferPositive(0, &pad, 1);
memcpy(&old_pad, current_pad, sizeof(Pad));
memset(&current_pad, 0, sizeof(Pad));
if (pad.buttons & SCE_CTRL_UP)
current_pad[PAD_UP] = 1;
if (pad.buttons & SCE_CTRL_DOWN)
current_pad[PAD_DOWN] = 1;
if (pad.buttons & SCE_CTRL_LEFT)
current_pad[PAD_LEFT] = 1;
if (pad.buttons & SCE_CTRL_RIGHT)
current_pad[PAD_RIGHT] = 1;
if (pad.buttons & SCE_CTRL_LTRIGGER)
current_pad[PAD_LTRIGGER] = 1;
if (pad.buttons & SCE_CTRL_RTRIGGER)
current_pad[PAD_RTRIGGER] = 1;
if (pad.buttons & SCE_CTRL_TRIANGLE)
current_pad[PAD_TRIANGLE] = 1;
if (pad.buttons & SCE_CTRL_CIRCLE)
current_pad[PAD_CIRCLE] = 1;
if (pad.buttons & SCE_CTRL_CROSS)
current_pad[PAD_CROSS] = 1;
if (pad.buttons & SCE_CTRL_SQUARE)
current_pad[PAD_SQUARE] = 1;
if (pad.buttons & SCE_CTRL_START)
current_pad[PAD_START] = 1;
if (pad.buttons & SCE_CTRL_SELECT)
current_pad[PAD_SELECT] = 1;
if (pad.ly < ANALOG_CENTER - ANALOG_THRESHOLD) {
current_pad[PAD_LEFT_ANALOG_UP] = 1;
} else if (pad.ly > ANALOG_CENTER + ANALOG_THRESHOLD) {
current_pad[PAD_LEFT_ANALOG_DOWN] = 1;
}
if (pad.lx < ANALOG_CENTER - ANALOG_THRESHOLD) {
current_pad[PAD_LEFT_ANALOG_LEFT] = 1;
} else if (pad.lx > ANALOG_CENTER + ANALOG_THRESHOLD) {
current_pad[PAD_LEFT_ANALOG_RIGHT] = 1;
}
if (pad.ry < ANALOG_CENTER - ANALOG_THRESHOLD) {
current_pad[PAD_RIGHT_ANALOG_UP] = 1;
} else if (pad.ry > ANALOG_CENTER + ANALOG_THRESHOLD) {
current_pad[PAD_RIGHT_ANALOG_DOWN] = 1;
}
if (pad.rx < ANALOG_CENTER - ANALOG_THRESHOLD) {
current_pad[PAD_RIGHT_ANALOG_LEFT] = 1;
} else if (pad.rx > ANALOG_CENTER + ANALOG_THRESHOLD) {
current_pad[PAD_RIGHT_ANALOG_RIGHT] = 1;
}
int i;
for (i = 0; i < PAD_N_BUTTONS; i++) {
pressed_pad[i] = current_pad[i] & ~old_pad[i];
released_pad[i] = ~current_pad[i] & old_pad[i];
hold_pad[i] = pressed_pad[i];
hold2_pad[i] = pressed_pad[i];
if (current_pad[i]) {
if (hold_count[i] >= 10) {
hold_pad[i] = 1;
hold_count[i] = 6;
}
if (hold2_count[i] >= 10) {
hold2_pad[i] = 1;
hold2_count[i] = 10;
}
hold_count[i]++;
hold2_count[i]++;
} else {
hold_count[i] = 0;
hold2_count[i] = 0;
}
}
if (enter_button == SCE_SYSTEM_PARAM_ENTER_BUTTON_CIRCLE) {
old_pad[PAD_ENTER] = old_pad[PAD_CIRCLE];
current_pad[PAD_ENTER] = current_pad[PAD_CIRCLE];
pressed_pad[PAD_ENTER] = pressed_pad[PAD_CIRCLE];
released_pad[PAD_ENTER] = released_pad[PAD_CIRCLE];
hold_pad[PAD_ENTER] = hold_pad[PAD_CIRCLE];
hold2_pad[PAD_ENTER] = hold2_pad[PAD_CIRCLE];
old_pad[PAD_CANCEL] = old_pad[PAD_CROSS];
current_pad[PAD_CANCEL] = current_pad[PAD_CROSS];
pressed_pad[PAD_CANCEL] = pressed_pad[PAD_CROSS];
released_pad[PAD_CANCEL] = released_pad[PAD_CROSS];
hold_pad[PAD_CANCEL] = hold_pad[PAD_CROSS];
hold2_pad[PAD_CANCEL] = hold2_pad[PAD_CROSS];
} else {
old_pad[PAD_ENTER] = old_pad[PAD_CROSS];
current_pad[PAD_ENTER] = current_pad[PAD_CROSS];
pressed_pad[PAD_ENTER] = pressed_pad[PAD_CROSS];
released_pad[PAD_ENTER] = released_pad[PAD_CROSS];
hold_pad[PAD_ENTER] = hold_pad[PAD_CROSS];
hold2_pad[PAD_ENTER] = hold2_pad[PAD_CROSS];
old_pad[PAD_CANCEL] = old_pad[PAD_CIRCLE];
current_pad[PAD_CANCEL] = current_pad[PAD_CIRCLE];
pressed_pad[PAD_CANCEL] = pressed_pad[PAD_CIRCLE];
released_pad[PAD_CANCEL] = released_pad[PAD_CIRCLE];
hold_pad[PAD_CANCEL] = hold_pad[PAD_CIRCLE];
hold2_pad[PAD_CANCEL] = hold2_pad[PAD_CIRCLE];
}
}
int holdButtons(SceCtrlData *pad, uint32_t buttons, uint64_t time) {
if ((pad->buttons & buttons) == buttons) {
uint64_t time_start = sceKernelGetProcessTimeWide();
while ((pad->buttons & buttons) == buttons) {
sceKernelDelayThread(10 * 1000);
sceCtrlPeekBufferPositive(0, pad, 1);
if ((sceKernelGetProcessTimeWide() - time_start) >= time) {
return 1;
}
}
}
return 0;
}
int hasEndSlash(const char *path) {
return path[strlen(path) - 1] == '/';
}
int removeEndSlash(char *path) {
int len = strlen(path);
if (path[len - 1] == '/') {
path[len - 1] = '\0';
return 1;
}
return 0;
}
int addEndSlash(char *path) {
int len = strlen(path);
if (len < MAX_PATH_LENGTH - 2) {
if (path[len - 1] != '/') {
path[len] = '/';
path[len + 1] = '\0';
return 1;
}
}
return 0;
}
void getSizeString(char string[16], uint64_t size) {
double double_size = (double)size;
int i = 0;
static char *units[] = { "B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB" };
while (double_size >= 1024.0) {
double_size /= 1024.0;
i++;
}
snprintf(string, 16, "%.*f %s", (i == 0) ? 0 : 2, double_size, units[i]);
}
void convertUtcToLocalTime(SceDateTime *time_local, SceDateTime *time_utc) {
SceRtcTick tick;
sceRtcGetTick(time_utc, &tick);
sceRtcConvertUtcToLocalTime(&tick, &tick);
sceRtcSetTick(time_local, &tick);
}
void convertLocalTimeToUtc(SceDateTime *time_utc, SceDateTime *time_local) {
SceRtcTick tick;
sceRtcGetTick(time_local, &tick);
sceRtcConvertLocalTimeToUtc(&tick, &tick);
sceRtcSetTick(time_utc, &tick);
}
void getDateString(char string[24], int date_format, SceDateTime *time) {
SceDateTime time_local;
convertUtcToLocalTime(&time_local, time);
switch (date_format) {
case SCE_SYSTEM_PARAM_DATE_FORMAT_YYYYMMDD:
snprintf(string, 24, "%04d/%02d/%02d", time_local.year, time_local.month, time_local.day);
break;
case SCE_SYSTEM_PARAM_DATE_FORMAT_DDMMYYYY:
snprintf(string, 24, "%02d/%02d/%04d", time_local.day, time_local.month, time_local.year);
break;
case SCE_SYSTEM_PARAM_DATE_FORMAT_MMDDYYYY:
snprintf(string, 24, "%02d/%02d/%04d", time_local.month, time_local.day, time_local.year);
break;
}
}
void getTimeString(char string[16], int time_format, SceDateTime *time) {
SceDateTime time_local;
convertUtcToLocalTime(&time_local, time);
switch(time_format) {
case SCE_SYSTEM_PARAM_TIME_FORMAT_12HR:
snprintf(string, 16, "%02d:%02d %s", (time_local.hour > 12) ? (time_local.hour - 12) : ((time_local.hour == 0) ? 12 : time_local.hour), time_local.minute, time_local.hour >= 12 ? "PM" : "AM");
break;
case SCE_SYSTEM_PARAM_TIME_FORMAT_24HR:
snprintf(string, 16, "%02d:%02d", time_local.hour, time_local.minute);
break;
}
}
int debugPrintf(const char *text, ...) {
va_list list;
char string[512];
va_start(list, text);
vsnprintf(string, sizeof(string), text, list);
va_end(list);
SceUID fd = sceIoOpen("ux0:data/vitashell_log.txt", SCE_O_WRONLY | SCE_O_CREAT | SCE_O_APPEND, 0777);
if (fd >= 0) {
sceIoWrite(fd, string, strlen(string));
sceIoClose(fd);
}
return 0;
}
int launchAppByUriExit(const char *titleid) {
char uri[32];
snprintf(uri, sizeof(uri), "psgm:play?titleid=%s", titleid);
sceAppMgrLaunchAppByUri(0xFFFFF, uri);
sceKernelExitProcess(0);
return 0;
}
char *strcasestr(const char *haystack, const char *needle) {
return boyer_moore(haystack, needle);
}
int vshIoMount(int id, const char *path, int permission, int a4, int a5, int a6) {
uint32_t buf[3];
buf[0] = a4;
buf[1] = a5;
buf[2] = a6;
return _vshIoMount(id, path, permission, buf);
}
void remount(int id) {
vshIoUmount(id, 0, 0, 0);
vshIoUmount(id, 1, 0, 0);
vshIoMount(id, NULL, 0, 0, 0, 0);
}