forked from Mirror/Ryujinx
19afb3209c
Update the LibHac dependency to version 0.13.1. This brings a ton of improvements and changes such as: - Refactor `FsSrv` to match the official refactoring done in FS. - Change how the `Horizon` and `HorizonClient` classes are handled. Each client created represents a different process with its own process ID and client state. - Add FS access control to handle permissions for FS service method calls. - Add FS program registry to keep track of the program ID, location and permissions of each process. - Add FS program index map info manager to track the program IDs and indexes of multi-application programs. - Add all FS IPC interfaces. - Rewrite `Fs.Fsa` code to be more accurate. - Rewrite a lot of `FsSrv` code to be more accurate. - Extend directory save data to store `SaveDataExtraData` - Extend directory save data to lock the save directory to allow only one accessor at a time. - Improve waiting and retrying when encountering access issues in `LocalFileSystem` and `DirectorySaveDataFileSystem`. - More `IFileSystemProxy` methods should work now. - Probably a bunch more stuff. On the Ryujinx side: - Forward most `IFileSystemProxy` methods to LibHac. - Register programs and program index map info when launching an application. - Remove hacks and workarounds for missing LibHac functionality. - Recreate missing save data extra data found on emulator startup. - Create system save data that wasn't indexed correctly on an older LibHac version. `FsSrv` now enforces access control for each process. When a process tries to open a save data file system, FS reads the save's extra data to determine who the save owner is and if the caller has permission to open the save data. Previously-created save data did not have extra data created when the save was created. With access control checks in place, this means that processes with no permissions (most games) wouldn't be able to access their own save data. The extra data can be partially created from data in the save data indexer, which should be enough for access control purposes.
740 lines
27 KiB
C#
740 lines
27 KiB
C#
using LibHac.Common;
|
|
using LibHac.Fs;
|
|
using LibHac.Fs.Fsa;
|
|
using LibHac.FsSystem;
|
|
using LibHac.FsSystem.RomFs;
|
|
using LibHac.Loader;
|
|
using Ryujinx.Common.Configuration;
|
|
using Ryujinx.Common.Logging;
|
|
using Ryujinx.HLE.Loaders.Mods;
|
|
using Ryujinx.HLE.Loaders.Executables;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Specialized;
|
|
using System.Linq;
|
|
using System.IO;
|
|
using Ryujinx.HLE.HOS.Kernel.Process;
|
|
using System.Globalization;
|
|
|
|
namespace Ryujinx.HLE.HOS
|
|
{
|
|
public class ModLoader
|
|
{
|
|
private const string RomfsDir = "romfs";
|
|
private const string ExefsDir = "exefs";
|
|
private const string CheatDir = "cheats";
|
|
private const string RomfsContainer = "romfs.bin";
|
|
private const string ExefsContainer = "exefs.nsp";
|
|
private const string StubExtension = ".stub";
|
|
private const string CheatExtension = ".txt";
|
|
private const string DefaultCheatName = "<default>";
|
|
|
|
private const string AmsContentsDir = "contents";
|
|
private const string AmsNsoPatchDir = "exefs_patches";
|
|
private const string AmsNroPatchDir = "nro_patches";
|
|
private const string AmsKipPatchDir = "kip_patches";
|
|
|
|
public struct Mod<T> where T : FileSystemInfo
|
|
{
|
|
public readonly string Name;
|
|
public readonly T Path;
|
|
|
|
public Mod(string name, T path)
|
|
{
|
|
Name = name;
|
|
Path = path;
|
|
}
|
|
}
|
|
|
|
public struct Cheat
|
|
{
|
|
// Atmosphere identifies the executables with the first 8 bytes
|
|
// of the build id, which is equivalent to 16 hex digits.
|
|
public const int CheatIdSize = 16;
|
|
|
|
public readonly string Name;
|
|
public readonly FileInfo Path;
|
|
public readonly IEnumerable<String> Instructions;
|
|
|
|
public Cheat(string name, FileInfo path, IEnumerable<String> instructions)
|
|
{
|
|
Name = name;
|
|
Path = path;
|
|
Instructions = instructions;
|
|
}
|
|
}
|
|
|
|
// Title dependent mods
|
|
public class ModCache
|
|
{
|
|
public List<Mod<FileInfo>> RomfsContainers { get; }
|
|
public List<Mod<FileInfo>> ExefsContainers { get; }
|
|
|
|
public List<Mod<DirectoryInfo>> RomfsDirs { get; }
|
|
public List<Mod<DirectoryInfo>> ExefsDirs { get; }
|
|
|
|
public List<Cheat> Cheats { get; }
|
|
|
|
public ModCache()
|
|
{
|
|
RomfsContainers = new List<Mod<FileInfo>>();
|
|
ExefsContainers = new List<Mod<FileInfo>>();
|
|
RomfsDirs = new List<Mod<DirectoryInfo>>();
|
|
ExefsDirs = new List<Mod<DirectoryInfo>>();
|
|
Cheats = new List<Cheat>();
|
|
}
|
|
}
|
|
|
|
// Title independent mods
|
|
public class PatchCache
|
|
{
|
|
public List<Mod<DirectoryInfo>> NsoPatches { get; }
|
|
public List<Mod<DirectoryInfo>> NroPatches { get; }
|
|
public List<Mod<DirectoryInfo>> KipPatches { get; }
|
|
|
|
internal bool Initialized { get; set; }
|
|
|
|
public PatchCache()
|
|
{
|
|
NsoPatches = new List<Mod<DirectoryInfo>>();
|
|
NroPatches = new List<Mod<DirectoryInfo>>();
|
|
KipPatches = new List<Mod<DirectoryInfo>>();
|
|
|
|
Initialized = false;
|
|
}
|
|
}
|
|
|
|
public Dictionary<ulong, ModCache> AppMods; // key is TitleId
|
|
public PatchCache Patches;
|
|
|
|
private static readonly EnumerationOptions _dirEnumOptions;
|
|
|
|
static ModLoader()
|
|
{
|
|
_dirEnumOptions = new EnumerationOptions
|
|
{
|
|
MatchCasing = MatchCasing.CaseInsensitive,
|
|
MatchType = MatchType.Simple,
|
|
RecurseSubdirectories = false,
|
|
ReturnSpecialDirectories = false
|
|
};
|
|
}
|
|
|
|
public ModLoader()
|
|
{
|
|
AppMods = new Dictionary<ulong, ModCache>();
|
|
Patches = new PatchCache();
|
|
}
|
|
|
|
public void Clear()
|
|
{
|
|
AppMods.Clear();
|
|
Patches = new PatchCache();
|
|
}
|
|
|
|
private static bool StrEquals(string s1, string s2) => string.Equals(s1, s2, StringComparison.OrdinalIgnoreCase);
|
|
|
|
public string GetModsBasePath() => EnsureBaseDirStructure(AppDataManager.GetModsPath());
|
|
|
|
private string EnsureBaseDirStructure(string modsBasePath)
|
|
{
|
|
var modsDir = new DirectoryInfo(modsBasePath);
|
|
|
|
modsDir.CreateSubdirectory(AmsContentsDir);
|
|
modsDir.CreateSubdirectory(AmsNsoPatchDir);
|
|
modsDir.CreateSubdirectory(AmsNroPatchDir);
|
|
// modsDir.CreateSubdirectory(AmsKipPatchDir); // uncomment when KIPs are supported
|
|
|
|
return modsDir.FullName;
|
|
}
|
|
|
|
private static DirectoryInfo FindTitleDir(DirectoryInfo contentsDir, string titleId)
|
|
=> contentsDir.EnumerateDirectories($"{titleId}*", _dirEnumOptions).FirstOrDefault();
|
|
|
|
public string GetTitleDir(string modsBasePath, string titleId)
|
|
{
|
|
var contentsDir = new DirectoryInfo(Path.Combine(modsBasePath, AmsContentsDir));
|
|
var titleModsPath = FindTitleDir(contentsDir, titleId);
|
|
|
|
if (titleModsPath == null)
|
|
{
|
|
Logger.Info?.Print(LogClass.ModLoader, $"Creating mods dir for Title {titleId.ToUpper()}");
|
|
titleModsPath = contentsDir.CreateSubdirectory(titleId);
|
|
}
|
|
|
|
return titleModsPath.FullName;
|
|
}
|
|
|
|
// Static Query Methods
|
|
public static void QueryPatchDirs(PatchCache cache, DirectoryInfo patchDir)
|
|
{
|
|
if (cache.Initialized || !patchDir.Exists) return;
|
|
|
|
var patches = cache.KipPatches;
|
|
string type = null;
|
|
|
|
if (StrEquals(AmsNsoPatchDir, patchDir.Name)) { patches = cache.NsoPatches; type = "NSO"; }
|
|
else if (StrEquals(AmsNroPatchDir, patchDir.Name)) { patches = cache.NroPatches; type = "NRO"; }
|
|
else if (StrEquals(AmsKipPatchDir, patchDir.Name)) { patches = cache.KipPatches; type = "KIP"; }
|
|
else return;
|
|
|
|
foreach (var modDir in patchDir.EnumerateDirectories())
|
|
{
|
|
patches.Add(new Mod<DirectoryInfo>(modDir.Name, modDir));
|
|
Logger.Info?.Print(LogClass.ModLoader, $"Found {type} patch '{modDir.Name}'");
|
|
}
|
|
}
|
|
|
|
public static void QueryTitleDir(ModCache mods, DirectoryInfo titleDir)
|
|
{
|
|
if (!titleDir.Exists) return;
|
|
|
|
var fsFile = new FileInfo(Path.Combine(titleDir.FullName, RomfsContainer));
|
|
if (fsFile.Exists)
|
|
{
|
|
mods.RomfsContainers.Add(new Mod<FileInfo>($"<{titleDir.Name} RomFs>", fsFile));
|
|
}
|
|
|
|
fsFile = new FileInfo(Path.Combine(titleDir.FullName, ExefsContainer));
|
|
if (fsFile.Exists)
|
|
{
|
|
mods.ExefsContainers.Add(new Mod<FileInfo>($"<{titleDir.Name} ExeFs>", fsFile));
|
|
}
|
|
|
|
System.Text.StringBuilder types = new System.Text.StringBuilder(5);
|
|
|
|
foreach (var modDir in titleDir.EnumerateDirectories())
|
|
{
|
|
types.Clear();
|
|
Mod<DirectoryInfo> mod = new Mod<DirectoryInfo>("", null);
|
|
|
|
if (StrEquals(RomfsDir, modDir.Name))
|
|
{
|
|
mods.RomfsDirs.Add(mod = new Mod<DirectoryInfo>($"<{titleDir.Name} RomFs>", modDir));
|
|
types.Append('R');
|
|
}
|
|
else if (StrEquals(ExefsDir, modDir.Name))
|
|
{
|
|
mods.ExefsDirs.Add(mod = new Mod<DirectoryInfo>($"<{titleDir.Name} ExeFs>", modDir));
|
|
types.Append('E');
|
|
}
|
|
else if (StrEquals(CheatDir, modDir.Name))
|
|
{
|
|
for (int i = 0; i < QueryCheatsDir(mods, modDir); i++)
|
|
{
|
|
types.Append('C');
|
|
}
|
|
}
|
|
else
|
|
{
|
|
var romfs = new DirectoryInfo(Path.Combine(modDir.FullName, RomfsDir));
|
|
var exefs = new DirectoryInfo(Path.Combine(modDir.FullName, ExefsDir));
|
|
var cheat = new DirectoryInfo(Path.Combine(modDir.FullName, CheatDir));
|
|
|
|
if (romfs.Exists)
|
|
{
|
|
mods.RomfsDirs.Add(mod = new Mod<DirectoryInfo>(modDir.Name, romfs));
|
|
types.Append('R');
|
|
}
|
|
|
|
if (exefs.Exists)
|
|
{
|
|
mods.ExefsDirs.Add(mod = new Mod<DirectoryInfo>(modDir.Name, exefs));
|
|
types.Append('E');
|
|
}
|
|
|
|
if (cheat.Exists)
|
|
{
|
|
for (int i = 0; i < QueryCheatsDir(mods, cheat); i++)
|
|
{
|
|
types.Append('C');
|
|
}
|
|
}
|
|
}
|
|
|
|
if (types.Length > 0) Logger.Info?.Print(LogClass.ModLoader, $"Found mod '{mod.Name}' [{types}]");
|
|
}
|
|
}
|
|
|
|
public static void QueryContentsDir(ModCache mods, DirectoryInfo contentsDir, ulong titleId)
|
|
{
|
|
if (!contentsDir.Exists) return;
|
|
|
|
Logger.Info?.Print(LogClass.ModLoader, $"Searching mods for {((titleId & 0x1000) != 0 ? "DLC" : "Title")} {titleId:X16}");
|
|
|
|
var titleDir = FindTitleDir(contentsDir, $"{titleId:x16}");
|
|
|
|
if (titleDir != null)
|
|
{
|
|
QueryTitleDir(mods, titleDir);
|
|
}
|
|
}
|
|
|
|
private static int QueryCheatsDir(ModCache mods, DirectoryInfo cheatsDir)
|
|
{
|
|
if (!cheatsDir.Exists)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
int numMods = 0;
|
|
|
|
foreach (FileInfo file in cheatsDir.EnumerateFiles())
|
|
{
|
|
if (!StrEquals(CheatExtension, file.Extension))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
string cheatId = Path.GetFileNameWithoutExtension(file.Name);
|
|
|
|
if (cheatId.Length != Cheat.CheatIdSize)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (!ulong.TryParse(cheatId, NumberStyles.HexNumber, CultureInfo.InvariantCulture, out _))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// A cheat file can contain several cheats for the same executable, so the file must be parsed in
|
|
// order to properly enumerate them.
|
|
mods.Cheats.AddRange(GetCheatsInFile(file));
|
|
}
|
|
|
|
return numMods;
|
|
}
|
|
|
|
private static IEnumerable<Cheat> GetCheatsInFile(FileInfo cheatFile)
|
|
{
|
|
string cheatName = DefaultCheatName;
|
|
List<string> instructions = new List<string>();
|
|
List<Cheat> cheats = new List<Cheat>();
|
|
|
|
using (StreamReader cheatData = cheatFile.OpenText())
|
|
{
|
|
string line;
|
|
while ((line = cheatData.ReadLine()) != null)
|
|
{
|
|
line = line.Trim();
|
|
|
|
if (line.StartsWith('['))
|
|
{
|
|
// This line starts a new cheat section.
|
|
if (!line.EndsWith(']') || line.Length < 3)
|
|
{
|
|
// Skip the entire file if there's any error while parsing the cheat file.
|
|
|
|
Logger.Warning?.Print(LogClass.ModLoader, $"Ignoring cheat '{cheatFile.FullName}' because it is malformed");
|
|
|
|
return new List<Cheat>();
|
|
}
|
|
|
|
// Add the previous section to the list.
|
|
if (instructions.Count != 0)
|
|
{
|
|
cheats.Add(new Cheat($"<{cheatName} Cheat>", cheatFile, instructions));
|
|
}
|
|
|
|
// Start a new cheat section.
|
|
cheatName = line.Substring(1, line.Length - 2);
|
|
instructions = new List<string>();
|
|
}
|
|
else if (line.Length > 0)
|
|
{
|
|
// The line contains an instruction.
|
|
instructions.Add(line);
|
|
}
|
|
}
|
|
|
|
// Add the last section being processed.
|
|
if (instructions.Count != 0)
|
|
{
|
|
cheats.Add(new Cheat($"<{cheatName} Cheat>", cheatFile, instructions));
|
|
}
|
|
}
|
|
|
|
return cheats;
|
|
}
|
|
|
|
// Assumes searchDirPaths don't overlap
|
|
public static void CollectMods(Dictionary<ulong, ModCache> modCaches, PatchCache patches, params string[] searchDirPaths)
|
|
{
|
|
static bool IsPatchesDir(string name) => StrEquals(AmsNsoPatchDir, name) ||
|
|
StrEquals(AmsNroPatchDir, name) ||
|
|
StrEquals(AmsKipPatchDir, name);
|
|
|
|
static bool IsContentsDir(string name) => StrEquals(AmsContentsDir, name);
|
|
|
|
static bool TryQuery(DirectoryInfo searchDir, PatchCache patches, Dictionary<ulong, ModCache> modCaches)
|
|
{
|
|
if (IsContentsDir(searchDir.Name))
|
|
{
|
|
foreach (var (titleId, cache) in modCaches)
|
|
{
|
|
QueryContentsDir(cache, searchDir, titleId);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
else if (IsPatchesDir(searchDir.Name))
|
|
{
|
|
QueryPatchDirs(patches, searchDir);
|
|
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
foreach (var path in searchDirPaths)
|
|
{
|
|
var searchDir = new DirectoryInfo(path);
|
|
if (!searchDir.Exists)
|
|
{
|
|
Logger.Warning?.Print(LogClass.ModLoader, $"Mod Search Dir '{searchDir.FullName}' doesn't exist");
|
|
continue;
|
|
}
|
|
|
|
if (!TryQuery(searchDir, patches, modCaches))
|
|
{
|
|
foreach (var subdir in searchDir.EnumerateDirectories())
|
|
{
|
|
TryQuery(subdir, patches, modCaches);
|
|
}
|
|
}
|
|
}
|
|
|
|
patches.Initialized = true;
|
|
}
|
|
|
|
public void CollectMods(IEnumerable<ulong> titles, params string[] searchDirPaths)
|
|
{
|
|
Clear();
|
|
|
|
foreach (ulong titleId in titles)
|
|
{
|
|
AppMods[titleId] = new ModCache();
|
|
}
|
|
|
|
CollectMods(AppMods, Patches, searchDirPaths);
|
|
}
|
|
|
|
internal IStorage ApplyRomFsMods(ulong titleId, IStorage baseStorage)
|
|
{
|
|
if (!AppMods.TryGetValue(titleId, out ModCache mods) || mods.RomfsDirs.Count + mods.RomfsContainers.Count == 0)
|
|
{
|
|
return baseStorage;
|
|
}
|
|
|
|
var fileSet = new HashSet<string>();
|
|
var builder = new RomFsBuilder();
|
|
int count = 0;
|
|
|
|
Logger.Info?.Print(LogClass.ModLoader, $"Applying RomFS mods for Title {titleId:X16}");
|
|
|
|
// Prioritize loose files first
|
|
foreach (var mod in mods.RomfsDirs)
|
|
{
|
|
using (IFileSystem fs = new LocalFileSystem(mod.Path.FullName))
|
|
{
|
|
AddFiles(fs, mod.Name, fileSet, builder);
|
|
}
|
|
count++;
|
|
}
|
|
|
|
// Then files inside images
|
|
foreach (var mod in mods.RomfsContainers)
|
|
{
|
|
Logger.Info?.Print(LogClass.ModLoader, $"Found 'romfs.bin' for Title {titleId:X16}");
|
|
using (IFileSystem fs = new RomFsFileSystem(mod.Path.OpenRead().AsStorage()))
|
|
{
|
|
AddFiles(fs, mod.Name, fileSet, builder);
|
|
}
|
|
count++;
|
|
}
|
|
|
|
if (fileSet.Count == 0)
|
|
{
|
|
Logger.Info?.Print(LogClass.ModLoader, "No files found. Using base RomFS");
|
|
|
|
return baseStorage;
|
|
}
|
|
|
|
Logger.Info?.Print(LogClass.ModLoader, $"Replaced {fileSet.Count} file(s) over {count} mod(s). Processing base storage...");
|
|
|
|
// And finally, the base romfs
|
|
var baseRom = new RomFsFileSystem(baseStorage);
|
|
foreach (var entry in baseRom.EnumerateEntries()
|
|
.Where(f => f.Type == DirectoryEntryType.File && !fileSet.Contains(f.FullPath))
|
|
.OrderBy(f => f.FullPath, StringComparer.Ordinal))
|
|
{
|
|
baseRom.OpenFile(out IFile file, entry.FullPath.ToU8Span(), OpenMode.Read).ThrowIfFailure();
|
|
builder.AddFile(entry.FullPath, file);
|
|
}
|
|
|
|
Logger.Info?.Print(LogClass.ModLoader, "Building new RomFS...");
|
|
IStorage newStorage = builder.Build();
|
|
Logger.Info?.Print(LogClass.ModLoader, "Using modded RomFS");
|
|
|
|
return newStorage;
|
|
}
|
|
|
|
private static void AddFiles(IFileSystem fs, string modName, HashSet<string> fileSet, RomFsBuilder builder)
|
|
{
|
|
foreach (var entry in fs.EnumerateEntries()
|
|
.Where(f => f.Type == DirectoryEntryType.File)
|
|
.OrderBy(f => f.FullPath, StringComparer.Ordinal))
|
|
{
|
|
fs.OpenFile(out IFile file, entry.FullPath.ToU8Span(), OpenMode.Read).ThrowIfFailure();
|
|
if (fileSet.Add(entry.FullPath))
|
|
{
|
|
builder.AddFile(entry.FullPath, file);
|
|
}
|
|
else
|
|
{
|
|
Logger.Warning?.Print(LogClass.ModLoader, $" Skipped duplicate file '{entry.FullPath}' from '{modName}'", "ApplyRomFsMods");
|
|
}
|
|
}
|
|
}
|
|
|
|
internal bool ReplaceExefsPartition(ulong titleId, ref IFileSystem exefs)
|
|
{
|
|
if (!AppMods.TryGetValue(titleId, out ModCache mods) || mods.ExefsContainers.Count == 0)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (mods.ExefsContainers.Count > 1)
|
|
{
|
|
Logger.Warning?.Print(LogClass.ModLoader, "Multiple ExeFS partition replacements detected");
|
|
}
|
|
|
|
Logger.Info?.Print(LogClass.ModLoader, $"Using replacement ExeFS partition");
|
|
|
|
exefs = new PartitionFileSystem(mods.ExefsContainers[0].Path.OpenRead().AsStorage());
|
|
|
|
return true;
|
|
}
|
|
|
|
public struct ModLoadResult
|
|
{
|
|
public BitVector32 Stubs;
|
|
public BitVector32 Replaces;
|
|
public MetaLoader Npdm;
|
|
|
|
public bool Modified => (Stubs.Data | Replaces.Data) != 0;
|
|
}
|
|
|
|
internal ModLoadResult ApplyExefsMods(ulong titleId, NsoExecutable[] nsos)
|
|
{
|
|
ModLoadResult modLoadResult = new ModLoadResult
|
|
{
|
|
Stubs = new BitVector32(),
|
|
Replaces = new BitVector32()
|
|
};
|
|
|
|
if (!AppMods.TryGetValue(titleId, out ModCache mods) || mods.ExefsDirs.Count == 0)
|
|
{
|
|
return modLoadResult;
|
|
}
|
|
|
|
if (nsos.Length != ApplicationLoader.ExeFsPrefixes.Length)
|
|
{
|
|
throw new ArgumentOutOfRangeException("NSO Count is incorrect");
|
|
}
|
|
|
|
var exeMods = mods.ExefsDirs;
|
|
|
|
foreach (var mod in exeMods)
|
|
{
|
|
for (int i = 0; i < ApplicationLoader.ExeFsPrefixes.Length; ++i)
|
|
{
|
|
var nsoName = ApplicationLoader.ExeFsPrefixes[i];
|
|
|
|
FileInfo nsoFile = new FileInfo(Path.Combine(mod.Path.FullName, nsoName));
|
|
if (nsoFile.Exists)
|
|
{
|
|
if (modLoadResult.Replaces[1 << i])
|
|
{
|
|
Logger.Warning?.Print(LogClass.ModLoader, $"Multiple replacements to '{nsoName}'");
|
|
|
|
continue;
|
|
}
|
|
|
|
modLoadResult.Replaces[1 << i] = true;
|
|
|
|
nsos[i] = new NsoExecutable(nsoFile.OpenRead().AsStorage(), nsoName);
|
|
Logger.Info?.Print(LogClass.ModLoader, $"NSO '{nsoName}' replaced");
|
|
}
|
|
|
|
modLoadResult.Stubs[1 << i] |= File.Exists(Path.Combine(mod.Path.FullName, nsoName + StubExtension));
|
|
}
|
|
|
|
FileInfo npdmFile = new FileInfo(Path.Combine(mod.Path.FullName, "main.npdm"));
|
|
if (npdmFile.Exists)
|
|
{
|
|
if (modLoadResult.Npdm != null)
|
|
{
|
|
Logger.Warning?.Print(LogClass.ModLoader, "Multiple replacements to 'main.npdm'");
|
|
|
|
continue;
|
|
}
|
|
|
|
modLoadResult.Npdm = new MetaLoader();
|
|
modLoadResult.Npdm.Load(File.ReadAllBytes(npdmFile.FullName));
|
|
|
|
Logger.Info?.Print(LogClass.ModLoader, "main.npdm replaced");
|
|
}
|
|
}
|
|
|
|
for (int i = ApplicationLoader.ExeFsPrefixes.Length - 1; i >= 0; --i)
|
|
{
|
|
if (modLoadResult.Stubs[1 << i] && !modLoadResult.Replaces[1 << i]) // Prioritizes replacements over stubs
|
|
{
|
|
Logger.Info?.Print(LogClass.ModLoader, $" NSO '{nsos[i].Name}' stubbed");
|
|
nsos[i] = null;
|
|
}
|
|
}
|
|
|
|
return modLoadResult;
|
|
}
|
|
|
|
internal void ApplyNroPatches(NroExecutable nro)
|
|
{
|
|
var nroPatches = Patches.NroPatches;
|
|
|
|
if (nroPatches.Count == 0) return;
|
|
|
|
// NRO patches aren't offset relative to header unlike NSO
|
|
// according to Atmosphere's ro patcher module
|
|
ApplyProgramPatches(nroPatches, 0, nro);
|
|
}
|
|
|
|
internal bool ApplyNsoPatches(ulong titleId, params IExecutable[] programs)
|
|
{
|
|
IEnumerable<Mod<DirectoryInfo>> nsoMods = Patches.NsoPatches;
|
|
|
|
if (AppMods.TryGetValue(titleId, out ModCache mods))
|
|
{
|
|
nsoMods = nsoMods.Concat(mods.ExefsDirs);
|
|
}
|
|
|
|
// NSO patches are created with offset 0 according to Atmosphere's patcher module
|
|
// But `Program` doesn't contain the header which is 0x100 bytes. So, we adjust for that here
|
|
return ApplyProgramPatches(nsoMods, 0x100, programs);
|
|
}
|
|
|
|
internal void LoadCheats(ulong titleId, ProcessTamperInfo tamperInfo, TamperMachine tamperMachine)
|
|
{
|
|
if (tamperInfo == null || tamperInfo.BuildIds == null || tamperInfo.CodeAddresses == null)
|
|
{
|
|
Logger.Error?.Print(LogClass.ModLoader, "Unable to install cheat because the associated process is invalid");
|
|
|
|
return;
|
|
}
|
|
|
|
Logger.Info?.Print(LogClass.ModLoader, $"Build ids found for title {titleId:X16}:\n {String.Join("\n ", tamperInfo.BuildIds)}");
|
|
|
|
if (!AppMods.TryGetValue(titleId, out ModCache mods) || mods.Cheats.Count == 0)
|
|
{
|
|
return;
|
|
}
|
|
|
|
var cheats = mods.Cheats;
|
|
var processExes = tamperInfo.BuildIds.Zip(tamperInfo.CodeAddresses, (k, v) => new { k, v })
|
|
.ToDictionary(x => x.k.Substring(0, Math.Min(Cheat.CheatIdSize, x.k.Length)), x => x.v);
|
|
|
|
foreach (var cheat in cheats)
|
|
{
|
|
string cheatId = Path.GetFileNameWithoutExtension(cheat.Path.Name).ToUpper();
|
|
|
|
if (!processExes.TryGetValue(cheatId, out ulong exeAddress))
|
|
{
|
|
Logger.Warning?.Print(LogClass.ModLoader, $"Skipping cheat '{cheat.Name}' because no executable matches its BuildId {cheatId} (check if the game title and version are correct)");
|
|
|
|
continue;
|
|
}
|
|
|
|
Logger.Info?.Print(LogClass.ModLoader, $"Installing cheat '{cheat.Name}'");
|
|
|
|
tamperMachine.InstallAtmosphereCheat(cheat.Instructions, tamperInfo, exeAddress);
|
|
}
|
|
}
|
|
|
|
private static bool ApplyProgramPatches(IEnumerable<Mod<DirectoryInfo>> mods, int protectedOffset, params IExecutable[] programs)
|
|
{
|
|
int count = 0;
|
|
|
|
MemPatch[] patches = new MemPatch[programs.Length];
|
|
|
|
for (int i = 0; i < patches.Length; ++i)
|
|
{
|
|
patches[i] = new MemPatch();
|
|
}
|
|
|
|
var buildIds = programs.Select(p => p switch
|
|
{
|
|
NsoExecutable nso => BitConverter.ToString(nso.BuildId.Bytes.ToArray()).Replace("-", "").TrimEnd('0'),
|
|
NroExecutable nro => BitConverter.ToString(nro.Header.BuildId).Replace("-", "").TrimEnd('0'),
|
|
_ => string.Empty
|
|
}).ToList();
|
|
|
|
int GetIndex(string buildId) => buildIds.FindIndex(id => id == buildId); // O(n) but list is small
|
|
|
|
// Collect patches
|
|
foreach (var mod in mods)
|
|
{
|
|
var patchDir = mod.Path;
|
|
foreach (var patchFile in patchDir.EnumerateFiles())
|
|
{
|
|
if (StrEquals(".ips", patchFile.Extension)) // IPS|IPS32
|
|
{
|
|
string filename = Path.GetFileNameWithoutExtension(patchFile.FullName).Split('.')[0];
|
|
string buildId = filename.TrimEnd('0');
|
|
|
|
int index = GetIndex(buildId);
|
|
if (index == -1)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
Logger.Info?.Print(LogClass.ModLoader, $"Matching IPS patch '{patchFile.Name}' in '{mod.Name}' bid={buildId}");
|
|
|
|
using var fs = patchFile.OpenRead();
|
|
using var reader = new BinaryReader(fs);
|
|
|
|
var patcher = new IpsPatcher(reader);
|
|
patcher.AddPatches(patches[index]);
|
|
}
|
|
else if (StrEquals(".pchtxt", patchFile.Extension)) // IPSwitch
|
|
{
|
|
using var fs = patchFile.OpenRead();
|
|
using var reader = new StreamReader(fs);
|
|
|
|
var patcher = new IPSwitchPatcher(reader);
|
|
|
|
int index = GetIndex(patcher.BuildId);
|
|
if (index == -1)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
Logger.Info?.Print(LogClass.ModLoader, $"Matching IPSwitch patch '{patchFile.Name}' in '{mod.Name}' bid={patcher.BuildId}");
|
|
|
|
patcher.AddPatches(patches[index]);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Apply patches
|
|
for (int i = 0; i < programs.Length; ++i)
|
|
{
|
|
count += patches[i].Patch(programs[i].Program, protectedOffset);
|
|
}
|
|
|
|
return count > 0;
|
|
}
|
|
}
|
|
}
|