mirror of
https://github.com/CTCaer/RetroArch.git
synced 2024-12-27 05:05:51 +00:00
914 lines
20 KiB
C
914 lines
20 KiB
C
/* SSNES - A Super Nintendo Entertainment System (SNES) Emulator frontend for libsnes.
|
|
* Copyright (C) 2010-2011 - Hans-Kristian Arntzen
|
|
*
|
|
* Some code herein may be based on code found in BSNES.
|
|
*
|
|
* SSNES 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 Found-
|
|
* ation, either version 3 of the License, or (at your option) any later version.
|
|
*
|
|
* SSNES 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 SSNES.
|
|
* If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "file.h"
|
|
#include "general.h"
|
|
#include <stdlib.h>
|
|
#include <stdbool.h>
|
|
#include <libsnes.hpp>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
#include <time.h>
|
|
#include "dynamic.h"
|
|
#include "movie.h"
|
|
#include "ups.h"
|
|
#include "bps.h"
|
|
#include "strl.h"
|
|
|
|
#ifdef HAVE_XML
|
|
#include "sha256.h"
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
#include <io.h>
|
|
#include <fcntl.h>
|
|
#include <windows.h>
|
|
#include <shlwapi.h>
|
|
#else
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <dirent.h>
|
|
#endif
|
|
|
|
// Generic file loader.
|
|
ssize_t read_file(const char *path, void **buf)
|
|
{
|
|
void *rom_buf = NULL;
|
|
FILE *file = fopen(path, "rb");
|
|
if (!file)
|
|
goto error;
|
|
|
|
fseek(file, 0, SEEK_END);
|
|
long len = ftell(file);
|
|
ssize_t rc = 0;
|
|
rewind(file);
|
|
rom_buf = malloc(len + 1);
|
|
if (!rom_buf)
|
|
{
|
|
SSNES_ERR("Couldn't allocate memory!\n");
|
|
goto error;
|
|
}
|
|
|
|
if ((rc = fread(rom_buf, 1, len, file)) < len)
|
|
SSNES_WARN("Didn't read whole file.\n");
|
|
|
|
*buf = rom_buf;
|
|
// Allow for easy reading of strings to be safe.
|
|
// Will only work with sane character formatting (Unix).
|
|
((char*)rom_buf)[len] = '\0';
|
|
fclose(file);
|
|
return rc;
|
|
|
|
error:
|
|
if (file)
|
|
fclose(file);
|
|
free(rom_buf);
|
|
*buf = NULL;
|
|
return -1;
|
|
}
|
|
|
|
// Reads file content as one string.
|
|
bool read_file_string(const char *path, char **buf)
|
|
{
|
|
*buf = NULL;
|
|
FILE *file = fopen(path, "r");
|
|
if (!file)
|
|
goto error;
|
|
|
|
fseek(file, 0, SEEK_END);
|
|
long len = ftell(file) + 2; // Takes account of being able to read in EOF and '\0' at end.
|
|
rewind(file);
|
|
|
|
*buf = calloc(len, sizeof(char));
|
|
if (!*buf)
|
|
goto error;
|
|
|
|
char *ptr = *buf;
|
|
|
|
while (ptr && !feof(file))
|
|
{
|
|
size_t bufsize = (size_t)(((ptrdiff_t)*buf + (ptrdiff_t)len) - (ptrdiff_t)ptr);
|
|
fgets(ptr, bufsize, file);
|
|
|
|
ptr = strchr(ptr, '\0');
|
|
}
|
|
|
|
ptr = strchr(ptr, EOF);
|
|
if (ptr) *ptr = '\0';
|
|
|
|
fclose(file);
|
|
return true;
|
|
|
|
error:
|
|
if (file)
|
|
fclose(file);
|
|
if (*buf)
|
|
free(*buf);
|
|
return false;
|
|
}
|
|
|
|
enum patch_type
|
|
{
|
|
PATCH_NONE,
|
|
PATCH_UPS,
|
|
PATCH_BPS
|
|
};
|
|
|
|
static void patch_rom(uint8_t **buf, ssize_t *size)
|
|
{
|
|
uint8_t *ret_buf = *buf;
|
|
ssize_t ret_size = *size;
|
|
|
|
ssize_t patch_size = 0;
|
|
void *patch_data = NULL;
|
|
enum patch_type type = PATCH_NONE;
|
|
|
|
if (g_extern.ups_pref && g_extern.bps_pref)
|
|
{
|
|
SSNES_WARN("Both UPS and BPS patch explicitly defined, ignoring both ...\n");
|
|
return;
|
|
}
|
|
|
|
if (!g_extern.bps_pref && *g_extern.ups_name && (patch_size = read_file(g_extern.ups_name, &patch_data)) >= 0)
|
|
type = PATCH_UPS;
|
|
else if (!g_extern.ups_pref && *g_extern.bps_name && (patch_size = read_file(g_extern.bps_name, &patch_data)) >= 0)
|
|
type = PATCH_BPS;
|
|
|
|
if (type == PATCH_NONE)
|
|
{
|
|
SSNES_LOG("Did not find a valid ROM patch.\n");
|
|
return;
|
|
}
|
|
|
|
switch (type)
|
|
{
|
|
case PATCH_UPS:
|
|
SSNES_LOG("Found UPS file in \"%s\", attempting to patch ...\n", g_extern.ups_name);
|
|
break;
|
|
case PATCH_BPS:
|
|
SSNES_LOG("Found BPS file in \"%s\", attempting to patch ...\n", g_extern.bps_name);
|
|
break;
|
|
|
|
default:
|
|
return; // Should not happen, but.
|
|
}
|
|
|
|
size_t target_size = ret_size * 4; // Just to be sure ...
|
|
uint8_t *patched_rom = malloc(target_size);
|
|
if (!patched_rom)
|
|
{
|
|
SSNES_ERR("Failed to allocate memory for patched ROM ...\n");
|
|
goto error;
|
|
}
|
|
|
|
bool success = false;
|
|
switch (type)
|
|
{
|
|
case PATCH_UPS:
|
|
{
|
|
ups_error_t err = ups_apply_patch(patch_data, patch_size, ret_buf, ret_size, patched_rom, &target_size);
|
|
if (err == UPS_SUCCESS)
|
|
{
|
|
SSNES_LOG("ROM patched successfully (UPS)!\n");
|
|
success = true;
|
|
}
|
|
else
|
|
SSNES_ERR("Failed to patch UPS: Error #%u\n", (unsigned)err);
|
|
|
|
break;
|
|
}
|
|
|
|
case PATCH_BPS:
|
|
{
|
|
bps_error_t err = bps_apply_patch(patch_data, patch_size, ret_buf, ret_size, patched_rom, &target_size);
|
|
if (err == BPS_SUCCESS)
|
|
{
|
|
SSNES_LOG("ROM patched successfully (BPS)!\n");
|
|
success = true;
|
|
}
|
|
else
|
|
SSNES_ERR("Failed to patch BPS: Error #%u\n", (unsigned)err);
|
|
|
|
break;
|
|
}
|
|
|
|
default:
|
|
return;
|
|
}
|
|
|
|
if (success)
|
|
{
|
|
free(ret_buf);
|
|
*buf = patched_rom;
|
|
*size = target_size;
|
|
}
|
|
|
|
if (patch_data)
|
|
free(patch_data);
|
|
|
|
return;
|
|
|
|
error:
|
|
*buf = ret_buf;
|
|
*size = ret_size;
|
|
if (patch_data)
|
|
free(patch_data);
|
|
}
|
|
|
|
// Load SNES rom only. Applies a hack for headered ROMs.
|
|
static ssize_t read_rom_file(FILE* file, void** buf)
|
|
{
|
|
ssize_t ret = 0;
|
|
uint8_t *ret_buf = NULL;
|
|
|
|
if (file == NULL) // stdin
|
|
{
|
|
#ifdef _WIN32
|
|
setmode(0, O_BINARY);
|
|
#endif
|
|
|
|
SSNES_LOG("Reading ROM from stdin ...\n");
|
|
size_t buf_size = 0xFFFFF; // Some initial guesstimate.
|
|
size_t buf_ptr = 0;
|
|
uint8_t *rom_buf = malloc(buf_size);
|
|
if (rom_buf == NULL)
|
|
{
|
|
SSNES_ERR("Couldn't allocate memory!\n");
|
|
return -1;
|
|
}
|
|
|
|
for(;;)
|
|
{
|
|
size_t ret = fread(rom_buf + buf_ptr, 1, buf_size - buf_ptr, stdin);
|
|
buf_ptr += ret;
|
|
|
|
// We've reached the end
|
|
if (buf_ptr < buf_size)
|
|
break;
|
|
|
|
rom_buf = realloc(rom_buf, buf_size * 2);
|
|
if (rom_buf == NULL)
|
|
{
|
|
SSNES_ERR("Couldn't allocate memory!\n");
|
|
return -1;
|
|
}
|
|
|
|
buf_size *= 2;
|
|
}
|
|
|
|
ret_buf = rom_buf;
|
|
ret = buf_ptr;
|
|
}
|
|
else
|
|
{
|
|
fseek(file, 0, SEEK_END);
|
|
ret = ftell(file);
|
|
rewind(file);
|
|
|
|
void *rom_buf = malloc(ret);
|
|
if (rom_buf == NULL)
|
|
{
|
|
SSNES_ERR("Couldn't allocate memory!\n");
|
|
return -1;
|
|
}
|
|
|
|
if (fread(rom_buf, 1, ret, file) < ret)
|
|
{
|
|
SSNES_ERR("Didn't read whole file.\n");
|
|
free(rom_buf);
|
|
return -1;
|
|
}
|
|
|
|
ret_buf = rom_buf;
|
|
}
|
|
|
|
// Remove copier header if present (512 first bytes).
|
|
if ((ret & 0x7fff) == 512)
|
|
{
|
|
memmove(ret_buf, ret_buf + 512, ret - 512);
|
|
ret -= 512;
|
|
}
|
|
|
|
// Attempt to apply a patch :)
|
|
patch_rom(&ret_buf, &ret);
|
|
|
|
g_extern.cart_crc = crc32_calculate(ret_buf, ret);
|
|
#ifdef HAVE_XML
|
|
sha256_hash(g_extern.sha256, ret_buf, ret);
|
|
SSNES_LOG("SHA256 sum: %s\n", g_extern.sha256);
|
|
#endif
|
|
*buf = ret_buf;
|
|
return ret;
|
|
}
|
|
|
|
|
|
// Dump stuff to file.
|
|
static bool dump_to_file(const char *path, const void *data, size_t size)
|
|
{
|
|
FILE *file = fopen(path, "wb");
|
|
if (!file)
|
|
return false;
|
|
else
|
|
{
|
|
bool ret = fwrite(data, 1, size, file) == size;
|
|
fclose(file);
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
static const char *ramtype2str(int type)
|
|
{
|
|
switch (type)
|
|
{
|
|
case SNES_MEMORY_CARTRIDGE_RAM:
|
|
case SNES_MEMORY_GAME_BOY_RAM:
|
|
case SNES_MEMORY_BSX_RAM:
|
|
return ".srm";
|
|
|
|
case SNES_MEMORY_CARTRIDGE_RTC:
|
|
case SNES_MEMORY_GAME_BOY_RTC:
|
|
return ".rtc";
|
|
|
|
case SNES_MEMORY_BSX_PRAM:
|
|
return ".pram";
|
|
|
|
case SNES_MEMORY_SUFAMI_TURBO_A_RAM:
|
|
return ".aram";
|
|
case SNES_MEMORY_SUFAMI_TURBO_B_RAM:
|
|
return ".bram";
|
|
|
|
default:
|
|
return "";
|
|
}
|
|
}
|
|
|
|
// Attempt to save valuable RAM data somewhere ...
|
|
static void dump_to_file_desperate(const void *data, size_t size, int type)
|
|
{
|
|
#ifdef _WIN32
|
|
const char *base = getenv("APPDATA");
|
|
#else
|
|
const char *base = getenv("HOME");
|
|
#endif
|
|
|
|
if (!base)
|
|
goto error;
|
|
|
|
char path[MAXPATHLEN];
|
|
snprintf(path, sizeof(path), "%s/SSNES-recovery-", base);
|
|
char timebuf[MAXPATHLEN];
|
|
|
|
time_t time_;
|
|
time(&time_);
|
|
strftime(timebuf, sizeof(timebuf), "%Y-%m-%d-%H-%M-%S", localtime(&time_));
|
|
strlcat(path, timebuf, sizeof(path));
|
|
strlcat(path, ramtype2str(type), sizeof(path));
|
|
|
|
if (dump_to_file(path, data, size))
|
|
SSNES_WARN("Succeeded in saving RAM data to \"%s\". Phew! :D\n", path);
|
|
else
|
|
goto error;
|
|
|
|
return;
|
|
|
|
error:
|
|
SSNES_WARN("Failed ... Tough luck ... :(\n");
|
|
}
|
|
|
|
bool save_state(const char *path)
|
|
{
|
|
SSNES_LOG("Saving state: \"%s\".\n", path);
|
|
size_t size = psnes_serialize_size();
|
|
if (size == 0)
|
|
return false;
|
|
|
|
void *data = malloc(size);
|
|
if (!data)
|
|
{
|
|
SSNES_ERR("Failed to allocate memory for save state buffer.\n");
|
|
return false;
|
|
}
|
|
|
|
SSNES_LOG("State size: %d bytes.\n", (int)size);
|
|
psnes_serialize(data, size);
|
|
bool ret = dump_to_file(path, data, size);
|
|
free(data);
|
|
if (!ret)
|
|
SSNES_ERR("Failed to save state to \"%s\".\n", path);
|
|
return ret;
|
|
}
|
|
|
|
bool load_state(const char *path)
|
|
{
|
|
SSNES_LOG("Loading state: \"%s\".\n", path);
|
|
void *buf = NULL;
|
|
ssize_t size = read_file(path, &buf);
|
|
if (size < 0)
|
|
{
|
|
SSNES_ERR("Failed to load state from \"%s\".\n", path);
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
SSNES_LOG("State size: %d bytes.\n", (int)size);
|
|
psnes_unserialize(buf, size);
|
|
}
|
|
|
|
free(buf);
|
|
return true;
|
|
}
|
|
|
|
void load_ram_file(const char *path, int type)
|
|
{
|
|
size_t size = psnes_get_memory_size(type);
|
|
uint8_t *data = psnes_get_memory_data(type);
|
|
|
|
if (size == 0 || !data)
|
|
return;
|
|
|
|
void *buf = NULL;
|
|
ssize_t rc = read_file(path, &buf);
|
|
if (rc <= size)
|
|
memcpy(data, buf, size);
|
|
|
|
free(buf);
|
|
}
|
|
|
|
void save_ram_file(const char *path, int type)
|
|
{
|
|
size_t size = psnes_get_memory_size(type);
|
|
uint8_t *data = psnes_get_memory_data(type);
|
|
|
|
if (data && size > 0)
|
|
{
|
|
if (!dump_to_file(path, data, size))
|
|
{
|
|
SSNES_ERR("Failed to save SNES RAM!\n");
|
|
SSNES_WARN("Attempting to recover ...\n");
|
|
dump_to_file_desperate(data, size, type);
|
|
}
|
|
}
|
|
}
|
|
|
|
static char* load_xml_map(const char *path)
|
|
{
|
|
char *xml_buf = NULL;
|
|
if (*path)
|
|
{
|
|
if (!read_file_string(path, &xml_buf))
|
|
SSNES_LOG("Did not find XML memory map in \"%s\"\n", path);
|
|
else
|
|
SSNES_LOG("Found XML memory map in \"%s\"\n", path);
|
|
}
|
|
|
|
return xml_buf;
|
|
}
|
|
|
|
static bool load_sgb_rom(void)
|
|
{
|
|
void *rom_buf = NULL;
|
|
ssize_t rom_len = 0;
|
|
|
|
FILE *extra_rom = NULL;
|
|
void *extra_rom_buf = NULL;
|
|
ssize_t extra_rom_len = 0;
|
|
|
|
if ((rom_len = read_rom_file(g_extern.rom_file, &rom_buf)) == -1)
|
|
{
|
|
SSNES_ERR("Could not read ROM file.\n");
|
|
goto error;
|
|
}
|
|
|
|
if ((extra_rom_len = read_file(g_extern.gb_rom_path, &extra_rom_buf)) == -1)
|
|
{
|
|
SSNES_ERR("Cannot read GameBoy rom.\n");
|
|
goto error;
|
|
}
|
|
|
|
char *xml_buf = load_xml_map(g_extern.xml_name);
|
|
|
|
if (!psnes_load_cartridge_super_game_boy(
|
|
xml_buf, rom_buf, rom_len,
|
|
NULL, extra_rom_buf, extra_rom_len))
|
|
{
|
|
SSNES_ERR("Cannot load SGB/GameBoy rom.\n");
|
|
goto error;
|
|
}
|
|
|
|
if (xml_buf)
|
|
free(xml_buf);
|
|
|
|
if (g_extern.rom_file)
|
|
fclose(g_extern.rom_file);
|
|
if (extra_rom)
|
|
fclose(extra_rom);
|
|
free(rom_buf);
|
|
free(extra_rom_buf);
|
|
return true;
|
|
|
|
error:
|
|
if (g_extern.rom_file)
|
|
fclose(g_extern.rom_file);
|
|
if (extra_rom)
|
|
fclose(extra_rom);
|
|
free(rom_buf);
|
|
free(extra_rom_buf);
|
|
return false;
|
|
}
|
|
|
|
static bool load_bsx_rom(bool slotted)
|
|
{
|
|
void *rom_buf = NULL;
|
|
ssize_t rom_len = 0;
|
|
|
|
FILE *extra_rom = NULL;
|
|
void *extra_rom_buf = NULL;
|
|
ssize_t extra_rom_len = 0;
|
|
|
|
if ((rom_len = read_rom_file(g_extern.rom_file, &rom_buf)) == -1)
|
|
{
|
|
SSNES_ERR("Could not read ROM file.\n");
|
|
goto error;
|
|
}
|
|
|
|
if ((extra_rom_len = read_file(g_extern.bsx_rom_path, &extra_rom_buf)) == -1)
|
|
{
|
|
SSNES_ERR("Cannot read BSX game rom.\n");
|
|
goto error;
|
|
}
|
|
|
|
char *xml_buf = load_xml_map(g_extern.xml_name);
|
|
|
|
if (slotted)
|
|
{
|
|
if (!psnes_load_cartridge_bsx_slotted(
|
|
xml_buf, rom_buf, rom_len,
|
|
NULL, extra_rom_buf, extra_rom_len))
|
|
{
|
|
SSNES_ERR("Cannot load BSX slotted rom.\n");
|
|
goto error;
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
if (!psnes_load_cartridge_bsx(
|
|
NULL, rom_buf, rom_len,
|
|
NULL, extra_rom_buf, extra_rom_len))
|
|
{
|
|
SSNES_ERR("Cannot load BSX rom.\n");
|
|
goto error;
|
|
}
|
|
}
|
|
|
|
if (xml_buf)
|
|
free(xml_buf);
|
|
|
|
if (g_extern.rom_file)
|
|
fclose(g_extern.rom_file);
|
|
if (extra_rom)
|
|
fclose(extra_rom);
|
|
free(rom_buf);
|
|
free(extra_rom_buf);
|
|
return true;
|
|
|
|
error:
|
|
if (g_extern.rom_file)
|
|
fclose(g_extern.rom_file);
|
|
if (extra_rom)
|
|
fclose(extra_rom);
|
|
free(rom_buf);
|
|
free(extra_rom_buf);
|
|
return false;
|
|
}
|
|
|
|
static bool load_sufami_rom(void)
|
|
{
|
|
void *rom_buf = NULL;
|
|
ssize_t rom_len = 0;
|
|
|
|
FILE *extra_rom[2] = {NULL};
|
|
void *extra_rom_buf[2] = {NULL};
|
|
ssize_t extra_rom_len[2] = {0};
|
|
|
|
if ((rom_len = read_rom_file(g_extern.rom_file, &rom_buf)) == -1)
|
|
{
|
|
SSNES_ERR("Could not read ROM file.\n");
|
|
goto error;
|
|
}
|
|
|
|
const char *roms[2] = { g_extern.sufami_rom_path[0], g_extern.sufami_rom_path[1] };
|
|
|
|
for (int i = 0; i < 2; i++)
|
|
{
|
|
if (strlen(roms[i]) > 0)
|
|
{
|
|
if ((extra_rom_len[i] = read_file(roms[i], &extra_rom_buf[i])) == -1)
|
|
{
|
|
SSNES_ERR("Cannot read Sufami game rom.\n");
|
|
goto error;
|
|
}
|
|
}
|
|
}
|
|
|
|
char *xml_buf = load_xml_map(g_extern.xml_name);
|
|
|
|
if (!psnes_load_cartridge_sufami_turbo(
|
|
xml_buf, rom_buf, rom_len,
|
|
NULL, extra_rom_buf[0], extra_rom_len[0],
|
|
NULL, extra_rom_buf[1], extra_rom_len[1]))
|
|
{
|
|
SSNES_ERR("Cannot load Sufami Turbo rom.\n");
|
|
goto error;
|
|
}
|
|
|
|
if (xml_buf)
|
|
free(xml_buf);
|
|
|
|
if (g_extern.rom_file)
|
|
fclose(g_extern.rom_file);
|
|
for (int i = 0; i < 2; i++)
|
|
{
|
|
if (extra_rom[i])
|
|
fclose(extra_rom[i]);
|
|
free(extra_rom_buf[i]);
|
|
}
|
|
free(rom_buf);
|
|
return true;
|
|
|
|
error:
|
|
if (g_extern.rom_file)
|
|
fclose(g_extern.rom_file);
|
|
for (int i = 0; i < 2; i++)
|
|
{
|
|
if (extra_rom[i])
|
|
fclose(extra_rom[i]);
|
|
free(extra_rom_buf[i]);
|
|
}
|
|
free(rom_buf);
|
|
return false;
|
|
}
|
|
|
|
static bool load_normal_rom(void)
|
|
{
|
|
void *rom_buf = NULL;
|
|
ssize_t rom_len = 0;
|
|
|
|
if ((rom_len = read_rom_file(g_extern.rom_file, &rom_buf)) == -1)
|
|
{
|
|
SSNES_ERR("Could not read ROM file.\n");
|
|
return false;
|
|
}
|
|
|
|
if (g_extern.rom_file != NULL)
|
|
fclose(g_extern.rom_file);
|
|
|
|
SSNES_LOG("ROM size: %d bytes\n", (int)rom_len);
|
|
|
|
char *xml_buf = load_xml_map(g_extern.xml_name);
|
|
|
|
if (!psnes_load_cartridge_normal(xml_buf, rom_buf, rom_len))
|
|
{
|
|
SSNES_ERR("ROM file is not valid!\n");
|
|
free(rom_buf);
|
|
return false;
|
|
}
|
|
|
|
if (xml_buf)
|
|
free(xml_buf);
|
|
|
|
free(rom_buf);
|
|
return true;
|
|
}
|
|
|
|
|
|
bool init_rom_file(enum ssnes_game_type type)
|
|
{
|
|
switch (type)
|
|
{
|
|
case SSNES_CART_SGB:
|
|
if (!load_sgb_rom())
|
|
return false;
|
|
break;
|
|
|
|
case SSNES_CART_NORMAL:
|
|
if (!load_normal_rom())
|
|
return false;
|
|
break;
|
|
|
|
case SSNES_CART_BSX:
|
|
if (!load_bsx_rom(false))
|
|
return false;
|
|
break;
|
|
|
|
case SSNES_CART_BSX_SLOTTED:
|
|
if (!load_bsx_rom(true))
|
|
return false;
|
|
break;
|
|
|
|
case SSNES_CART_SUFAMI:
|
|
if (!load_sufami_rom())
|
|
return false;
|
|
break;
|
|
|
|
default:
|
|
SSNES_ERR("Invalid ROM type!\n");
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
// Yep, this is C alright ;)
|
|
char** dir_list_new(const char *dir, const char *ext)
|
|
{
|
|
size_t cur_ptr = 0;
|
|
size_t cur_size = 32;
|
|
char **dir_list = NULL;
|
|
|
|
#ifdef _WIN32
|
|
WIN32_FIND_DATAW ffd;
|
|
HANDLE hFind = INVALID_HANDLE_VALUE;
|
|
|
|
wchar_t wchar_buf[MAXPATHLEN];
|
|
char utf8_buf[MAXPATHLEN];
|
|
|
|
if (strlcpy(utf8_buf, dir, sizeof(utf8_buf)) >= sizeof(utf8_buf))
|
|
goto error;
|
|
if (strlcat(utf8_buf, "/*", sizeof(utf8_buf)) >= sizeof(utf8_buf))
|
|
goto error;
|
|
if (strlcat(utf8_buf, ext, sizeof(utf8_buf)) >= sizeof(utf8_buf))
|
|
goto error;
|
|
|
|
if (MultiByteToWideChar(CP_UTF8, 0, utf8_buf, -1, wchar_buf, MAXPATHLEN) == 0)
|
|
goto error;
|
|
|
|
hFind = FindFirstFileW(wchar_buf, &ffd);
|
|
if (hFind == INVALID_HANDLE_VALUE)
|
|
goto error;
|
|
#else
|
|
DIR *directory = NULL;
|
|
const struct dirent *entry = NULL;
|
|
|
|
directory = opendir(dir);
|
|
if (!directory)
|
|
goto error;
|
|
#endif
|
|
|
|
dir_list = calloc(cur_size, sizeof(char*));
|
|
if (!dir_list)
|
|
goto error;
|
|
|
|
#ifdef _WIN32 // Hard to read? Blame non-POSIX heathens!
|
|
do
|
|
#else
|
|
while ((entry = readdir(directory)))
|
|
#endif
|
|
{
|
|
// Not a perfect search of course, but hopefully good enough in practice.
|
|
#ifdef _WIN32
|
|
if (ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
|
|
continue;
|
|
if (WideCharToMultiByte(CP_UTF8, 0, ffd.cFileName, -1, utf8_buf, MAX_PATH, NULL, NULL) == 0)
|
|
continue;
|
|
if (ext && !strstr(utf8_buf, ext))
|
|
continue;
|
|
#else
|
|
if (ext && !strstr(entry->d_name, ext))
|
|
continue;
|
|
#endif
|
|
|
|
dir_list[cur_ptr] = malloc(MAXPATHLEN);
|
|
if (!dir_list[cur_ptr])
|
|
goto error;
|
|
|
|
strlcpy(dir_list[cur_ptr], dir, MAXPATHLEN);
|
|
strlcat(dir_list[cur_ptr], "/", MAXPATHLEN);
|
|
#ifdef _WIN32
|
|
strlcat(dir_list[cur_ptr], utf8_buf, MAXPATHLEN);
|
|
#else
|
|
strlcat(dir_list[cur_ptr], entry->d_name, MAXPATHLEN);
|
|
#endif
|
|
|
|
cur_ptr++;
|
|
if (cur_ptr + 1 == cur_size) // Need to reserve for NULL.
|
|
{
|
|
cur_size *= 2;
|
|
dir_list = realloc(dir_list, cur_size * sizeof(char*));
|
|
if (!dir_list)
|
|
goto error;
|
|
|
|
// Make sure it's all NULL'd out since we cannot rely on realloc to do this.
|
|
memset(dir_list + cur_ptr, 0, (cur_size - cur_ptr) * sizeof(char*));
|
|
}
|
|
}
|
|
#ifdef _WIN32
|
|
while (FindNextFileW(hFind, &ffd) != 0);
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
FindClose(hFind);
|
|
#else
|
|
closedir(directory);
|
|
#endif
|
|
return dir_list;
|
|
|
|
error:
|
|
SSNES_ERR("Failed to open directory: \"%s\"\n", dir);
|
|
#ifdef _WIN32
|
|
if (hFind != INVALID_HANDLE_VALUE)
|
|
FindClose(hFind);
|
|
#else
|
|
if (directory)
|
|
closedir(directory);
|
|
#endif
|
|
dir_list_free(dir_list);
|
|
return NULL;
|
|
}
|
|
|
|
void dir_list_free(char **dir_list)
|
|
{
|
|
if (!dir_list)
|
|
return;
|
|
|
|
char **orig = dir_list;
|
|
while (*dir_list)
|
|
free(*dir_list++);
|
|
free(orig);
|
|
}
|
|
|
|
bool path_is_directory(const char *path)
|
|
{
|
|
#ifdef _WIN32
|
|
wchar_t buf[MAXPATHLEN];
|
|
if (MultiByteToWideChar(CP_UTF8, 0, path, -1, buf, MAXPATHLEN) == 0)
|
|
return false;
|
|
return PathIsDirectoryW(buf) == FILE_ATTRIBUTE_DIRECTORY;
|
|
#else
|
|
struct stat buf;
|
|
if (stat(path, &buf) < 0)
|
|
return false;
|
|
|
|
return S_ISDIR(buf.st_mode);
|
|
#endif
|
|
}
|
|
|
|
bool path_file_exists(const char *path)
|
|
{
|
|
FILE *dummy = fopen(path, "rb");
|
|
if (dummy)
|
|
{
|
|
fclose(dummy);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void fill_pathname(char *out_path, const char *in_path, const char *replace, size_t size)
|
|
{
|
|
char tmp_path[strlen(in_path) + 1];
|
|
strlcpy(tmp_path, in_path, sizeof(tmp_path));
|
|
char *tok = strrchr(tmp_path, '.');
|
|
if (tok != NULL)
|
|
*tok = '\0';
|
|
assert(strlcpy(out_path, tmp_path, size) < size);
|
|
assert(strlcat(out_path, replace, size) < size);
|
|
}
|
|
|
|
void fill_pathname_noext(char *out_path, const char *in_path, const char *replace, size_t size)
|
|
{
|
|
assert(strlcpy(out_path, in_path, size) < size);
|
|
assert(strlcat(out_path, replace, size) < size);
|
|
}
|
|
|
|
void fill_pathname_dir(char *in_dir, const char *in_basename, const char *replace, size_t size)
|
|
{
|
|
assert(strlcat(in_dir, "/", size) < size);
|
|
|
|
const char *base = strrchr(in_basename, '/');
|
|
if (!base)
|
|
base = strrchr(in_basename, '\\');
|
|
|
|
if (base)
|
|
base++;
|
|
else
|
|
base = in_basename;
|
|
|
|
assert(strlcat(in_dir, base, size) < size);
|
|
assert(strlcat(in_dir, replace, size) < size);
|
|
}
|