[VFS] Add VFS dump tool (supporting STFS only as of now)

This commit is contained in:
Dr. Chat 2018-06-26 13:45:16 -05:00
parent 827e3a0be4
commit f89b4626a3
2 changed files with 138 additions and 0 deletions

View File

@ -15,3 +15,30 @@ project("xenia-vfs")
project_root.."/third_party/gflags/src",
})
recursive_platform_files()
removefiles({"vfs_dump.cc"})
project("xenia-vfs-dump")
uuid("2EF270C7-41A8-4D0E-ACC5-59693A9CCE32")
kind("ConsoleApp")
language("C++")
links({
"gflags",
"xenia-base",
"xenia-vfs",
})
flags({
-- "WinMain", -- Use WinMain instead of main.
})
defines({})
includedirs({
project_root.."/third_party/gflags/src",
})
files({
"vfs_dump.cc",
project_root.."/src/xenia/base/main_"..platform_suffix..".cc",
})
resincludedirs({
project_root,
})

111
src/xenia/vfs/vfs_dump.cc Normal file
View File

@ -0,0 +1,111 @@
/**
******************************************************************************
* Xenia : Xbox 360 Emulator Research Project *
******************************************************************************
* Copyright 2018 Ben Vanik. All rights reserved. *
* Released under the BSD license - see LICENSE in the root for more details. *
******************************************************************************
*/
#include <queue>
#include <string>
#include <vector>
#include "xenia/base/logging.h"
#include "xenia/base/main.h"
#include "xenia/base/math.h"
#include "xenia/vfs/devices/stfs_container_device.h"
#include "xenia/vfs/file.h"
namespace xe {
namespace vfs {
int vfs_dump_main(const std::vector<std::wstring>& args) {
if (args.size() <= 2) {
XELOGE("Usage: %s [source] [dump_path]", args[0].c_str());
return 1;
}
std::wstring base_path = args[2];
std::unique_ptr<vfs::Device> device;
// TODO: Flags specifying the type of device.
device = std::make_unique<vfs::StfsContainerDevice>("", args[1]);
if (!device->Initialize()) {
XELOGE("Failed to initialize device");
return 1;
}
// Run through all the files, breadth-first style.
std::queue<vfs::Entry*> queue;
auto root = device->ResolvePath("/");
queue.push(root);
// Allocate a buffer when needed.
size_t buffer_size = 0;
uint8_t* buffer = nullptr;
while (!queue.empty()) {
auto entry = queue.front();
queue.pop();
for (auto& entry : entry->children()) {
queue.push(entry.get());
}
XELOGI("%s", entry->path().c_str());
auto dest_name = xe::join_paths(base_path, xe::to_wstring(entry->path()));
if (entry->attributes() & kFileAttributeDirectory) {
xe::filesystem::CreateFolder(dest_name + L"\\");
continue;
}
vfs::File* in_file = nullptr;
if (entry->Open(FileAccess::kFileReadData, &in_file) != X_STATUS_SUCCESS) {
continue;
}
auto file = xe::filesystem::OpenFile(dest_name, "wb");
if (!file) {
in_file->Destroy();
continue;
}
if (entry->can_map()) {
auto map = entry->OpenMapped(xe::MappedMemory::Mode::kRead);
fwrite(map->data(), map->size(), 1, file);
map->Close();
} else {
// Can't map the file into memory. Read it into a temporary buffer.
if (!buffer || entry->size() > buffer_size) {
// Resize the buffer.
if (buffer) {
delete[] buffer;
}
// Allocate a buffer rounded up to the nearest 512MB.
buffer_size = xe::round_up(entry->size(), 512 * 1024 * 1024);
buffer = new uint8_t[buffer_size];
}
size_t bytes_read = 0;
in_file->ReadSync(buffer, entry->size(), 0, &bytes_read);
fwrite(buffer, bytes_read, 1, file);
}
fclose(file);
in_file->Destroy();
}
if (buffer) {
delete[] buffer;
}
return 0;
}
} // namespace vfs
} // namespace xe
DEFINE_ENTRY_POINT(L"xenia-vfs-dump", L"xenia-vfs-dump",
xe::vfs::vfs_dump_main);