GerbilSoft/rvthtool
1
Fork 0
mirror of https://github.com/GerbilSoft/rvthtool.git synced 2025-06-18 19:45:35 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

[librvth] rvth.c: Added support for writing sparse files on Windows.

Browse Source
This commit is contained in:
David Korth 2018-01-17 19:52:11 -05:00
parent 2fc8580b0d
commit fcbb936b66
2 changed files with 56 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/librvth/CMakeLists.txt
View File

@ -1,8 +1,10 @@
PROJECT(librvth)
# Check for C library functions.
INCLUDE(CheckFunctionExists)
CHECK_FUNCTION_EXISTS(ftruncate HAVE_FTRUNCATE)
IF(NOT WIN32)
INCLUDE(CheckFunctionExists)
CHECK_FUNCTION_EXISTS(ftruncate HAVE_FTRUNCATE)
ENDIF(NOT WIN32)
# Write the config.h file.
CONFIGURE_FILE("${CMAKE_CURRENT_SOURCE_DIR}/config.librvth.h.in" "${CMAKE_CURRENT_BINARY_DIR}/config.librvth.h")

56
src/librvth/rvth.c
View File

@ -32,10 +32,14 @@
#include <stdlib.h>
#include <string.h>
#ifdef HAVE_FTRUNCATE
#if defined(_WIN32)
# include <windows.h>
# include <io.h>
# include <winioctl.h>
#elif defined(HAVE_FTRUNCATE)
# include <unistd.h>
# include <sys/types.h>
#endif /* HAVE_FTRUNCATE */
#endif
// RVT-H main struct.
struct _RvtH {
@ -692,7 +696,6 @@ static bool is_block_empty(const uint8_t *block, unsigned int size)
*/
int rvth_extract(const RvtH *rvth, unsigned int bank, const char *filename, RvtH_Progress_Callback callback)
{
// TODO: Windows file I/O for sparse file handling.
FILE *f_extract;
const RvtH_BankEntry *entry;
uint8_t *buf;
@ -701,6 +704,13 @@ int rvth_extract(const RvtH *rvth, unsigned int bank, const char *filename, RvtH
unsigned int sprs; // Sparse counter.
int ret;
#ifdef _WIN32
wchar_t root_dir[4]; // Root directory.
wchar_t *p_root_dir; // Pointer to root_dir, or NULL if relative.
DWORD dwFileSystemFlags; // Flags from GetVolumeInformation().
BOOL bRet;
#endif /* _WIN32 */
if (!rvth || !filename || filename[0] == 0) {
errno = EINVAL;
return -EINVAL;
@ -744,7 +754,45 @@ int rvth_extract(const RvtH *rvth, unsigned int bank, const char *filename, RvtH
return -err;
}
#ifdef HAVE_FTRUNCATE
#if defined(_WIN32)
// Check if the file system supports sparse files.
// TODO: Handle mount points?
if (isalpha(filename[0]) && filename[1] == ':' &&
(filename[2] == '\\' || filename[2] == '/'))
{
// Absolute pathname.
root_dir[0] = (wchar_t)filename[0];
root_dir[1] = L':';
root_dir[2] = L'\\';
root_dir[3] = 0;
p_root_dir = root_dir;
}
else
{
// Relative pathname.
p_root_dir = NULL;
}
bRet = GetVolumeInformation(p_root_dir, NULL, 0, NULL, NULL,
&dwFileSystemFlags, NULL, 0);
if (bRet != 0 && (dwFileSystemFlags & FILE_SUPPORTS_SPARSE_FILES)) {
// File system supports sparse files.
// Mark the file as sparse.
HANDLE h_extract = (HANDLE)_get_osfhandle(_fileno(f_extract));
if (h_extract != NULL && h_extract != INVALID_HANDLE_VALUE) {
DWORD bytesReturned;
FILE_SET_SPARSE_BUFFER fssb;
fssb.SetSparse = TRUE;
// TODO: Use SetEndOfFile() if this succeeds?
// Note that SetEndOfFile() isn't guaranteed to fill the
// resulting file with zero bytes...
DeviceIoControl(h_extract, FSCTL_SET_SPARSE,
&fssb, sizeof(fssb),
NULL, 0, &bytesReturned, NULL);
}
}
#elif defined(HAVE_FTRUNCATE)
// Set the file size.
// NOTE: If the underlying file system doesn't support sparse files,
// this may take a long time. (TODO: Check this.)