2018-09-07 17:00:13 +02:00
|
|
|
/*
|
2019-04-08 04:00:49 +02:00
|
|
|
* Copyright (c) 2018-2019 Atmosphère-NX
|
2018-09-07 17:00:13 +02:00
|
|
|
*
|
|
|
|
* 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/>.
|
|
|
|
*/
|
|
|
|
|
2018-06-15 01:50:01 +02:00
|
|
|
#pragma once
|
2019-03-08 16:25:33 +01:00
|
|
|
#include <cstdlib>
|
2018-06-15 01:50:01 +02:00
|
|
|
#include <switch.h>
|
2018-06-29 06:40:18 +02:00
|
|
|
#include <map>
|
2018-06-15 01:50:01 +02:00
|
|
|
|
|
|
|
#include "fsmitm_romstorage.hpp"
|
|
|
|
|
2019-01-29 11:29:29 +01:00
|
|
|
#include "../debug.hpp"
|
2018-06-28 07:02:06 +02:00
|
|
|
|
2018-06-15 01:50:01 +02:00
|
|
|
#define ROMFS_ENTRY_EMPTY 0xFFFFFFFF
|
|
|
|
#define ROMFS_FILEPARTITION_OFS 0x200
|
|
|
|
|
2018-11-04 22:56:07 +01:00
|
|
|
#define ROMFS_METADATA_FILE_PATH "romfs_metadata.bin"
|
|
|
|
|
2018-06-15 01:50:01 +02:00
|
|
|
/* Types for RomFS Meta construction. */
|
2018-06-19 20:07:31 +02:00
|
|
|
enum class RomFSDataSource {
|
|
|
|
BaseRomFS,
|
|
|
|
FileRomFS,
|
|
|
|
LooseFile,
|
2018-11-04 22:56:07 +01:00
|
|
|
MetaData,
|
2018-06-19 20:07:31 +02:00
|
|
|
Memory,
|
2018-06-15 01:50:01 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
struct RomFSBaseSourceInfo {
|
|
|
|
u64 offset;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct RomFSFileSourceInfo {
|
|
|
|
u64 offset;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct RomFSLooseSourceInfo {
|
|
|
|
const char *path;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct RomFSMemorySourceInfo {
|
|
|
|
const u8 *data;
|
|
|
|
};
|
|
|
|
|
2018-11-04 22:56:07 +01:00
|
|
|
struct RomFSMetaDataSourceInfo {
|
|
|
|
|
|
|
|
};
|
|
|
|
|
2018-06-28 07:02:06 +02:00
|
|
|
struct RomFSSourceInfo {
|
|
|
|
u64 virtual_offset;
|
|
|
|
u64 size;
|
|
|
|
union {
|
|
|
|
RomFSBaseSourceInfo base_source_info;
|
|
|
|
RomFSFileSourceInfo file_source_info;
|
|
|
|
RomFSLooseSourceInfo loose_source_info;
|
|
|
|
RomFSMemorySourceInfo memory_source_info;
|
2018-11-04 22:56:07 +01:00
|
|
|
RomFSMemorySourceInfo metadata_source_info;
|
2018-06-28 07:02:06 +02:00
|
|
|
};
|
|
|
|
RomFSDataSource type;
|
|
|
|
|
|
|
|
RomFSSourceInfo(u64 v_o, u64 s, u64 offset, RomFSDataSource t) : virtual_offset(v_o), size(s), type(t) {
|
|
|
|
switch (this->type) {
|
2018-06-19 20:07:31 +02:00
|
|
|
case RomFSDataSource::BaseRomFS:
|
2018-06-28 07:02:06 +02:00
|
|
|
this->base_source_info.offset = offset;
|
|
|
|
break;
|
2018-06-19 20:07:31 +02:00
|
|
|
case RomFSDataSource::FileRomFS:
|
2018-06-28 07:02:06 +02:00
|
|
|
this->file_source_info.offset = offset;
|
|
|
|
break;
|
|
|
|
case RomFSDataSource::LooseFile:
|
2018-11-04 22:56:07 +01:00
|
|
|
case RomFSDataSource::MetaData:
|
2018-06-28 07:02:06 +02:00
|
|
|
case RomFSDataSource::Memory:
|
2018-06-15 01:50:01 +02:00
|
|
|
default:
|
2019-03-29 00:57:18 +01:00
|
|
|
/* TODO: Better error. */
|
|
|
|
fatalSimple(ResultKernelConnectionClosed);
|
|
|
|
break;
|
2018-06-15 01:50:01 +02:00
|
|
|
}
|
|
|
|
}
|
2018-06-28 07:02:06 +02:00
|
|
|
|
|
|
|
RomFSSourceInfo(u64 v_o, u64 s, const void *arg, RomFSDataSource t) : virtual_offset(v_o), size(s), type(t) {
|
|
|
|
switch (this->type) {
|
2018-06-19 20:07:31 +02:00
|
|
|
case RomFSDataSource::LooseFile:
|
2018-06-28 07:02:06 +02:00
|
|
|
this->loose_source_info.path = (decltype(this->loose_source_info.path))arg;
|
|
|
|
break;
|
2018-06-19 20:07:31 +02:00
|
|
|
case RomFSDataSource::Memory:
|
2018-06-28 07:02:06 +02:00
|
|
|
this->memory_source_info.data = (decltype(this->memory_source_info.data))arg;
|
|
|
|
break;
|
2018-11-04 22:56:07 +01:00
|
|
|
case RomFSDataSource::MetaData:
|
|
|
|
case RomFSDataSource::BaseRomFS:
|
|
|
|
case RomFSDataSource::FileRomFS:
|
|
|
|
default:
|
2019-03-29 00:57:18 +01:00
|
|
|
/* TODO: Better error. */
|
|
|
|
fatalSimple(ResultKernelConnectionClosed);
|
|
|
|
break;
|
2018-11-04 22:56:07 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
RomFSSourceInfo(u64 v_o, u64 s, RomFSDataSource t) : virtual_offset(v_o), size(s), type(t) {
|
|
|
|
switch (this->type) {
|
|
|
|
case RomFSDataSource::MetaData:
|
|
|
|
break;
|
|
|
|
case RomFSDataSource::LooseFile:
|
|
|
|
case RomFSDataSource::Memory:
|
2018-06-28 07:02:06 +02:00
|
|
|
case RomFSDataSource::BaseRomFS:
|
|
|
|
case RomFSDataSource::FileRomFS:
|
2018-06-15 01:50:01 +02:00
|
|
|
default:
|
2019-03-29 00:57:18 +01:00
|
|
|
/* TODO: Better error. */
|
|
|
|
fatalSimple(ResultKernelConnectionClosed);
|
|
|
|
break;
|
2018-06-15 01:50:01 +02:00
|
|
|
}
|
|
|
|
}
|
2018-06-28 07:02:06 +02:00
|
|
|
|
|
|
|
void Cleanup() {
|
|
|
|
switch (this->type) {
|
|
|
|
case RomFSDataSource::BaseRomFS:
|
|
|
|
case RomFSDataSource::FileRomFS:
|
2018-11-04 22:56:07 +01:00
|
|
|
case RomFSDataSource::MetaData:
|
2018-06-28 07:02:06 +02:00
|
|
|
break;
|
|
|
|
case RomFSDataSource::LooseFile:
|
2019-03-08 16:25:33 +01:00
|
|
|
delete[] this->loose_source_info.path;
|
2018-06-28 07:02:06 +02:00
|
|
|
break;
|
|
|
|
case RomFSDataSource::Memory:
|
2019-03-08 16:25:33 +01:00
|
|
|
std::free((void*)this->memory_source_info.data);
|
2018-06-28 07:02:06 +02:00
|
|
|
break;
|
|
|
|
default:
|
2019-03-29 00:57:18 +01:00
|
|
|
/* TODO: Better error. */
|
|
|
|
fatalSimple(ResultKernelConnectionClosed);
|
|
|
|
break;
|
2018-06-19 20:07:31 +02:00
|
|
|
}
|
|
|
|
}
|
2018-06-28 07:02:06 +02:00
|
|
|
|
2018-06-15 01:50:01 +02:00
|
|
|
static bool Compare(RomFSSourceInfo *a, RomFSSourceInfo *b) {
|
|
|
|
return (a->virtual_offset < b->virtual_offset);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
/* Types for building a RomFS. */
|
|
|
|
struct RomFSHeader {
|
|
|
|
u64 header_size;
|
|
|
|
u64 dir_hash_table_ofs;
|
|
|
|
u64 dir_hash_table_size;
|
|
|
|
u64 dir_table_ofs;
|
|
|
|
u64 dir_table_size;
|
|
|
|
u64 file_hash_table_ofs;
|
|
|
|
u64 file_hash_table_size;
|
|
|
|
u64 file_table_ofs;
|
|
|
|
u64 file_table_size;
|
|
|
|
u64 file_partition_ofs;
|
|
|
|
};
|
|
|
|
|
|
|
|
static_assert(sizeof(RomFSHeader) == 0x50, "Incorrect RomFS Header definition!");
|
|
|
|
|
|
|
|
struct RomFSDirectoryEntry {
|
|
|
|
u32 parent;
|
|
|
|
u32 sibling;
|
|
|
|
u32 child;
|
|
|
|
u32 file;
|
|
|
|
u32 hash;
|
|
|
|
u32 name_size;
|
|
|
|
char name[];
|
|
|
|
};
|
|
|
|
|
|
|
|
static_assert(sizeof(RomFSDirectoryEntry) == 0x18, "Incorrect RomFSDirectoryEntry definition!");
|
|
|
|
|
|
|
|
struct RomFSFileEntry {
|
|
|
|
u32 parent;
|
|
|
|
u32 sibling;
|
|
|
|
u64 offset;
|
|
|
|
u64 size;
|
|
|
|
u32 hash;
|
|
|
|
u32 name_size;
|
|
|
|
char name[];
|
|
|
|
};
|
|
|
|
|
|
|
|
static_assert(sizeof(RomFSFileEntry) == 0x20, "Incorrect RomFSFileEntry definition!");
|
|
|
|
|
|
|
|
struct RomFSBuildFileContext;
|
|
|
|
|
2018-06-29 06:40:18 +02:00
|
|
|
/* 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;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2018-06-15 01:50:01 +02:00
|
|
|
struct RomFSBuildDirectoryContext {
|
2018-06-29 04:39:23 +02:00
|
|
|
char *path;
|
2018-06-15 01:50:01 +02:00
|
|
|
u32 cur_path_ofs;
|
|
|
|
u32 path_len;
|
2018-06-28 07:02:06 +02:00
|
|
|
u32 entry_offset = 0;
|
|
|
|
RomFSBuildDirectoryContext *parent = NULL;
|
|
|
|
RomFSBuildDirectoryContext *child = NULL;
|
|
|
|
RomFSBuildDirectoryContext *sibling = NULL;
|
|
|
|
RomFSBuildFileContext *file = NULL;
|
2018-06-15 01:50:01 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
struct RomFSBuildFileContext {
|
2018-06-29 04:39:23 +02:00
|
|
|
char *path;
|
2018-06-15 01:50:01 +02:00
|
|
|
u32 cur_path_ofs;
|
|
|
|
u32 path_len;
|
2018-06-28 07:02:06 +02:00
|
|
|
u32 entry_offset = 0;
|
|
|
|
u64 offset = 0;
|
|
|
|
u64 size = 0;
|
|
|
|
RomFSBuildDirectoryContext *parent = NULL;
|
|
|
|
RomFSBuildFileContext *sibling = NULL;
|
|
|
|
RomFSDataSource source{0};
|
|
|
|
u64 orig_offset = 0;
|
2018-06-15 01:50:01 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
class RomFSBuildContext {
|
|
|
|
private:
|
|
|
|
u64 title_id;
|
|
|
|
RomFSBuildDirectoryContext *root;
|
2018-06-29 06:40:18 +02:00
|
|
|
std::map<char *, RomFSBuildDirectoryContext *, build_ctx_cmp> directories;
|
|
|
|
std::map<char *, RomFSBuildFileContext *, build_ctx_cmp> files;
|
2018-07-02 16:26:03 +02:00
|
|
|
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;
|
2018-06-15 01:50:01 +02:00
|
|
|
|
|
|
|
FsDirectoryEntry dir_entry;
|
|
|
|
RomFSDataSource cur_source_type;
|
|
|
|
|
|
|
|
void VisitDirectory(FsFileSystem *filesys, RomFSBuildDirectoryContext *parent);
|
|
|
|
void VisitDirectory(RomFSBuildDirectoryContext *parent, u32 parent_offset, void *dir_table, size_t dir_table_size, void *file_table, size_t file_table_size);
|
|
|
|
|
|
|
|
bool AddDirectory(RomFSBuildDirectoryContext *parent_dir_ctx, RomFSBuildDirectoryContext *dir_ctx, RomFSBuildDirectoryContext **out_dir_ctx);
|
|
|
|
bool AddFile(RomFSBuildDirectoryContext *parent_dir_ctx, RomFSBuildFileContext *file_ctx);
|
|
|
|
public:
|
2018-07-02 16:26:03 +02:00
|
|
|
RomFSBuildContext(u64 tid) : title_id(tid) {
|
2018-06-28 07:02:06 +02:00
|
|
|
this->root = new RomFSBuildDirectoryContext({0});
|
2018-06-29 04:39:23 +02:00
|
|
|
this->root->path = new char[1];
|
|
|
|
this->root->path[0] = '\x00';
|
2018-06-29 06:40:18 +02:00
|
|
|
this->directories.insert({this->root->path, this->root});
|
2018-06-15 01:50:01 +02:00
|
|
|
this->num_dirs = 1;
|
|
|
|
this->dir_table_size = 0x18;
|
|
|
|
}
|
|
|
|
|
|
|
|
void MergeSdFiles();
|
|
|
|
void MergeRomStorage(IROStorage *storage, RomFSDataSource source);
|
|
|
|
|
|
|
|
/* This finalizes the context. */
|
|
|
|
void Build(std::vector<RomFSSourceInfo> *out_infos);
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
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;
|
|
|
|
}
|