diff options
Diffstat (limited to 'src/core/file_sys/fsmitm_romfsbuild.hpp')
-rw-r--r-- | src/core/file_sys/fsmitm_romfsbuild.hpp | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/src/core/file_sys/fsmitm_romfsbuild.hpp b/src/core/file_sys/fsmitm_romfsbuild.hpp new file mode 100644 index 000000000..b897aab21 --- /dev/null +++ b/src/core/file_sys/fsmitm_romfsbuild.hpp @@ -0,0 +1,106 @@ +/* + * Copyright (c) 2018 Atmosphère-NX + * + * This program is free software; you can redistribute it and/or modify it + * under the terms and conditions of the GNU General Public License, + * version 2, as published by the Free Software Foundation. + * + * This program is distributed in the hope 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/>. + */ + +/* + * Adapted by DarkLordZach for use/interaction with yuzu + * + * Modifications Copyright 2018 yuzu emulator team + * Licensed under GPLv2 or any later version + * Refer to the license.txt file included. + */ + +#pragma once +#include <map> +#include <boost/detail/container_fwd.hpp> +#include "common/common_types.h" +#include "vfs.h" + +namespace FileSys { + +/* Used as comparator for std::map<char *, RomFSBuild*Context> */ +struct build_ctx_cmp { + bool operator()(const char* a, const char* b) const { + return strcmp(a, b) < 0; + } +}; + +struct RomFSDirectoryEntry; +struct RomFSFileEntry; +struct RomFSBuildDirectoryContext; +struct RomFSBuildFileContext; + +class RomFSBuildContext { +private: + VirtualDir base; + RomFSBuildDirectoryContext* root; + std::map<char*, RomFSBuildDirectoryContext*, build_ctx_cmp> directories; + std::map<char*, RomFSBuildFileContext*, build_ctx_cmp> files; + u64 num_dirs = 0; + u64 num_files = 0; + u64 dir_table_size = 0; + u64 file_table_size = 0; + u64 dir_hash_table_size = 0; + u64 file_hash_table_size = 0; + u64 file_partition_size = 0; + + void VisitDirectory(VirtualDir filesys, RomFSBuildDirectoryContext* parent); + + bool AddDirectory(RomFSBuildDirectoryContext* parent_dir_ctx, + RomFSBuildDirectoryContext* dir_ctx, + RomFSBuildDirectoryContext** out_dir_ctx); + bool AddFile(RomFSBuildDirectoryContext* parent_dir_ctx, RomFSBuildFileContext* file_ctx); + +public: + explicit RomFSBuildContext(VirtualDir base); + + /* This finalizes the context. */ + std::map<u64, VirtualFile> Build(); +}; + +static inline RomFSDirectoryEntry* romfs_get_direntry(void* directories, uint32_t offset) { + return (RomFSDirectoryEntry*)((uintptr_t)directories + offset); +} + +static inline RomFSFileEntry* romfs_get_fentry(void* files, uint32_t offset) { + return (RomFSFileEntry*)((uintptr_t)files + offset); +} + +static inline uint32_t romfs_calc_path_hash(uint32_t parent, const unsigned char* path, + uint32_t start, size_t path_len) { + uint32_t hash = parent ^ 123456789; + for (uint32_t i = 0; i < path_len; i++) { + hash = (hash >> 5) | (hash << 27); + hash ^= path[start + i]; + } + + return hash; +} + +static inline uint32_t romfs_get_hash_table_count(uint32_t num_entries) { + if (num_entries < 3) { + return 3; + } else if (num_entries < 19) { + return num_entries | 1; + } + uint32_t count = num_entries; + while (count % 2 == 0 || count % 3 == 0 || count % 5 == 0 || count % 7 == 0 || + count % 11 == 0 || count % 13 == 0 || count % 17 == 0) { + count++; + } + return count; +} + +} // namespace FileSys |