2020-07-08 21:31:15 -07:00
using LibHac.Common ;
using LibHac.Fs ;
2020-09-01 13:08:59 -07:00
using LibHac.Fs.Fsa ;
2020-07-08 21:31:15 -07:00
using LibHac.FsSystem ;
using LibHac.FsSystem.RomFs ;
2021-08-12 14:56:24 -07:00
using LibHac.Loader ;
2020-08-30 09:51:53 -07:00
using Ryujinx.Common.Configuration ;
2020-07-08 21:31:15 -07:00
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 ;
2021-03-27 07:12:05 -07:00
using Ryujinx.HLE.HOS.Kernel.Process ;
using System.Globalization ;
2021-12-23 08:55:50 -08:00
using Path = System . IO . Path ;
2020-07-08 21:31:15 -07:00
namespace Ryujinx.HLE.HOS
{
public class ModLoader
{
private const string RomfsDir = "romfs" ;
private const string ExefsDir = "exefs" ;
2021-03-27 07:12:05 -07:00
private const string CheatDir = "cheats" ;
2020-07-08 21:31:15 -07:00
private const string RomfsContainer = "romfs.bin" ;
private const string ExefsContainer = "exefs.nsp" ;
private const string StubExtension = ".stub" ;
2021-03-27 07:12:05 -07:00
private const string CheatExtension = ".txt" ;
private const string DefaultCheatName = "<default>" ;
2020-07-08 21:31:15 -07:00
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 ;
}
}
2021-03-27 07:12:05 -07:00
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 ;
}
}
2020-07-08 21:31:15 -07:00
// 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 ; }
2021-03-27 07:12:05 -07:00
public List < Cheat > Cheats { get ; }
2020-07-08 21:31:15 -07:00
public ModCache ( )
{
RomfsContainers = new List < Mod < FileInfo > > ( ) ;
ExefsContainers = new List < Mod < FileInfo > > ( ) ;
RomfsDirs = new List < Mod < DirectoryInfo > > ( ) ;
ExefsDirs = new List < Mod < DirectoryInfo > > ( ) ;
2021-03-27 07:12:05 -07:00
Cheats = new List < Cheat > ( ) ;
2020-07-08 21:31:15 -07:00
}
}
// Title independent mods
public class PatchCache
{
public List < Mod < DirectoryInfo > > NsoPatches { get ; }
public List < Mod < DirectoryInfo > > NroPatches { get ; }
public List < Mod < DirectoryInfo > > KipPatches { get ; }
2021-02-19 16:25:01 -08:00
internal bool Initialized { get ; set ; }
2020-07-08 21:31:15 -07:00
public PatchCache ( )
{
NsoPatches = new List < Mod < DirectoryInfo > > ( ) ;
NroPatches = new List < Mod < DirectoryInfo > > ( ) ;
KipPatches = new List < Mod < DirectoryInfo > > ( ) ;
2021-02-19 16:25:01 -08:00
Initialized = false ;
2020-07-08 21:31:15 -07:00
}
}
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 ) ;
2020-08-30 09:51:53 -07:00
public string GetModsBasePath ( ) = > EnsureBaseDirStructure ( AppDataManager . GetModsPath ( ) ) ;
private string EnsureBaseDirStructure ( string modsBasePath )
2020-07-08 21:31:15 -07:00
{
var modsDir = new DirectoryInfo ( modsBasePath ) ;
modsDir . CreateSubdirectory ( AmsContentsDir ) ;
modsDir . CreateSubdirectory ( AmsNsoPatchDir ) ;
modsDir . CreateSubdirectory ( AmsNroPatchDir ) ;
// modsDir.CreateSubdirectory(AmsKipPatchDir); // uncomment when KIPs are supported
2020-08-30 09:51:53 -07:00
return modsDir . FullName ;
2020-07-08 21:31:15 -07:00
}
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 )
{
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , $"Creating mods dir for Title {titleId.ToUpper()}" ) ;
2020-07-08 21:31:15 -07:00
titleModsPath = contentsDir . CreateSubdirectory ( titleId ) ;
}
return titleModsPath . FullName ;
}
// Static Query Methods
2021-02-19 16:25:01 -08:00
public static void QueryPatchDirs ( PatchCache cache , DirectoryInfo patchDir )
2020-07-08 21:31:15 -07:00
{
2021-02-19 16:25:01 -08:00
if ( cache . Initialized | | ! patchDir . Exists ) return ;
2020-07-08 21:31:15 -07:00
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 ) ) ;
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , $"Found {type} patch '{modDir.Name}'" ) ;
2020-07-08 21:31:15 -07:00
}
}
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' ) ;
}
2021-03-27 07:12:05 -07:00
else if ( StrEquals ( CheatDir , modDir . Name ) )
{
for ( int i = 0 ; i < QueryCheatsDir ( mods , modDir ) ; i + + )
{
types . Append ( 'C' ) ;
}
}
2020-07-08 21:31:15 -07:00
else
{
var romfs = new DirectoryInfo ( Path . Combine ( modDir . FullName , RomfsDir ) ) ;
var exefs = new DirectoryInfo ( Path . Combine ( modDir . FullName , ExefsDir ) ) ;
2021-03-27 07:12:05 -07:00
var cheat = new DirectoryInfo ( Path . Combine ( modDir . FullName , CheatDir ) ) ;
2020-07-08 21:31:15 -07:00
if ( romfs . Exists )
{
mods . RomfsDirs . Add ( mod = new Mod < DirectoryInfo > ( modDir . Name , romfs ) ) ;
types . Append ( 'R' ) ;
}
2021-03-27 07:12:05 -07:00
2020-07-08 21:31:15 -07:00
if ( exefs . Exists )
{
mods . ExefsDirs . Add ( mod = new Mod < DirectoryInfo > ( modDir . Name , exefs ) ) ;
types . Append ( 'E' ) ;
}
2021-03-27 07:12:05 -07:00
if ( cheat . Exists )
{
for ( int i = 0 ; i < QueryCheatsDir ( mods , cheat ) ; i + + )
{
types . Append ( 'C' ) ;
}
}
2020-07-08 21:31:15 -07:00
}
2020-08-03 16:32:53 -07:00
if ( types . Length > 0 ) Logger . Info ? . Print ( LogClass . ModLoader , $"Found mod '{mod.Name}' [{types}]" ) ;
2020-07-08 21:31:15 -07:00
}
}
public static void QueryContentsDir ( ModCache mods , DirectoryInfo contentsDir , ulong titleId )
{
if ( ! contentsDir . Exists ) return ;
2021-02-19 16:25:01 -08:00
Logger . Info ? . Print ( LogClass . ModLoader , $"Searching mods for {((titleId & 0x1000) != 0 ? " DLC " : " Title ")} {titleId:X16}" ) ;
2020-07-08 21:31:15 -07:00
var titleDir = FindTitleDir ( contentsDir , $"{titleId:x16}" ) ;
if ( titleDir ! = null )
{
QueryTitleDir ( mods , titleDir ) ;
}
}
2021-03-27 07:12:05 -07:00
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 ;
}
2021-02-19 16:25:01 -08:00
// Assumes searchDirPaths don't overlap
public static void CollectMods ( Dictionary < ulong , ModCache > modCaches , PatchCache patches , params string [ ] searchDirPaths )
2020-07-08 21:31:15 -07:00
{
static bool IsPatchesDir ( string name ) = > StrEquals ( AmsNsoPatchDir , name ) | |
StrEquals ( AmsNroPatchDir , name ) | |
StrEquals ( AmsKipPatchDir , name ) ;
2021-02-19 16:25:01 -08:00
static bool IsContentsDir ( string name ) = > StrEquals ( AmsContentsDir , name ) ;
static bool TryQuery ( DirectoryInfo searchDir , PatchCache patches , Dictionary < ulong , ModCache > modCaches )
2020-07-08 21:31:15 -07:00
{
2021-02-19 16:25:01 -08:00
if ( IsContentsDir ( searchDir . Name ) )
2020-07-08 21:31:15 -07:00
{
2021-02-19 16:25:01 -08:00
foreach ( var ( titleId , cache ) in modCaches )
2020-07-08 21:31:15 -07:00
{
2021-02-19 16:25:01 -08:00
QueryContentsDir ( cache , searchDir , titleId ) ;
2020-07-08 21:31:15 -07:00
}
2021-02-19 16:25:01 -08:00
return true ;
2020-07-08 21:31:15 -07:00
}
2021-02-19 16:25:01 -08:00
else if ( IsPatchesDir ( searchDir . Name ) )
2020-07-08 21:31:15 -07:00
{
2021-02-19 16:25:01 -08:00
QueryPatchDirs ( patches , searchDir ) ;
2020-07-08 21:31:15 -07:00
return true ;
}
return false ;
}
foreach ( var path in searchDirPaths )
{
2021-02-19 16:25:01 -08:00
var searchDir = new DirectoryInfo ( path ) ;
if ( ! searchDir . Exists )
2020-07-08 21:31:15 -07:00
{
2021-02-19 16:25:01 -08:00
Logger . Warning ? . Print ( LogClass . ModLoader , $"Mod Search Dir '{searchDir.FullName}' doesn't exist" ) ;
2020-07-08 21:31:15 -07:00
continue ;
}
2021-02-19 16:25:01 -08:00
if ( ! TryQuery ( searchDir , patches , modCaches ) )
2020-07-08 21:31:15 -07:00
{
2021-02-19 16:25:01 -08:00
foreach ( var subdir in searchDir . EnumerateDirectories ( ) )
2020-07-08 21:31:15 -07:00
{
2021-02-19 16:25:01 -08:00
TryQuery ( subdir , patches , modCaches ) ;
2020-07-08 21:31:15 -07:00
}
}
}
2021-02-19 16:25:01 -08:00
patches . Initialized = true ;
2020-07-08 21:31:15 -07:00
}
2021-02-19 16:25:01 -08:00
public void CollectMods ( IEnumerable < ulong > titles , params string [ ] searchDirPaths )
2020-07-08 21:31:15 -07:00
{
2021-02-19 16:25:01 -08:00
Clear ( ) ;
foreach ( ulong titleId in titles )
2020-07-08 21:31:15 -07:00
{
2021-02-19 16:25:01 -08:00
AppMods [ titleId ] = new ModCache ( ) ;
2020-07-08 21:31:15 -07:00
}
2021-02-19 16:25:01 -08:00
CollectMods ( AppMods , Patches , searchDirPaths ) ;
2020-07-08 21:31:15 -07:00
}
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 ;
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , $"Applying RomFS mods for Title {titleId:X16}" ) ;
2020-07-08 21:31:15 -07:00
// 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 )
{
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , $"Found 'romfs.bin' for Title {titleId:X16}" ) ;
2020-07-08 21:31:15 -07:00
using ( IFileSystem fs = new RomFsFileSystem ( mod . Path . OpenRead ( ) . AsStorage ( ) ) )
{
AddFiles ( fs , mod . Name , fileSet , builder ) ;
}
count + + ;
}
if ( fileSet . Count = = 0 )
{
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , "No files found. Using base RomFS" ) ;
2020-07-08 21:31:15 -07:00
return baseStorage ;
}
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , $"Replaced {fileSet.Count} file(s) over {count} mod(s). Processing base storage..." ) ;
2020-07-08 21:31:15 -07:00
// 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 ) )
{
2021-12-23 08:55:50 -08:00
using var file = new UniqueRef < IFile > ( ) ;
baseRom . OpenFile ( ref file . Ref ( ) , entry . FullPath . ToU8Span ( ) , OpenMode . Read ) . ThrowIfFailure ( ) ;
builder . AddFile ( entry . FullPath , file . Release ( ) ) ;
2020-07-08 21:31:15 -07:00
}
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , "Building new RomFS..." ) ;
2020-07-08 21:31:15 -07:00
IStorage newStorage = builder . Build ( ) ;
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , "Using modded RomFS" ) ;
2020-07-08 21:31:15 -07:00
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 ) )
{
2021-12-23 08:55:50 -08:00
using var file = new UniqueRef < IFile > ( ) ;
fs . OpenFile ( ref file . Ref ( ) , entry . FullPath . ToU8Span ( ) , OpenMode . Read ) . ThrowIfFailure ( ) ;
2020-07-08 21:31:15 -07:00
if ( fileSet . Add ( entry . FullPath ) )
{
2021-12-23 08:55:50 -08:00
builder . AddFile ( entry . FullPath , file . Release ( ) ) ;
2020-07-08 21:31:15 -07:00
}
else
{
2020-08-03 16:32:53 -07:00
Logger . Warning ? . Print ( LogClass . ModLoader , $" Skipped duplicate file '{entry.FullPath}' from '{modName}'" , "ApplyRomFsMods" ) ;
2020-07-08 21:31:15 -07:00
}
}
}
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 )
{
2020-08-03 16:32:53 -07:00
Logger . Warning ? . Print ( LogClass . ModLoader , "Multiple ExeFS partition replacements detected" ) ;
2020-07-08 21:31:15 -07:00
}
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , $"Using replacement ExeFS partition" ) ;
2020-07-08 21:31:15 -07:00
exefs = new PartitionFileSystem ( mods . ExefsContainers [ 0 ] . Path . OpenRead ( ) . AsStorage ( ) ) ;
return true ;
}
2020-12-29 11:54:32 -08:00
public struct ModLoadResult
2020-07-08 21:31:15 -07:00
{
2020-12-29 11:54:32 -08:00
public BitVector32 Stubs ;
public BitVector32 Replaces ;
2021-08-12 14:56:24 -07:00
public MetaLoader Npdm ;
2020-12-29 11:54:32 -08:00
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 ( )
} ;
2020-07-08 21:31:15 -07:00
if ( ! AppMods . TryGetValue ( titleId , out ModCache mods ) | | mods . ExefsDirs . Count = = 0 )
{
2020-12-29 11:54:32 -08:00
return modLoadResult ;
2020-07-08 21:31:15 -07:00
}
2020-12-29 11:54:32 -08:00
if ( nsos . Length ! = ApplicationLoader . ExeFsPrefixes . Length )
2020-07-08 21:31:15 -07:00
{
2020-12-29 11:54:32 -08:00
throw new ArgumentOutOfRangeException ( "NSO Count is incorrect" ) ;
2020-07-08 21:31:15 -07:00
}
var exeMods = mods . ExefsDirs ;
foreach ( var mod in exeMods )
{
2020-12-29 11:54:32 -08:00
for ( int i = 0 ; i < ApplicationLoader . ExeFsPrefixes . Length ; + + i )
2020-07-08 21:31:15 -07:00
{
2020-12-29 11:54:32 -08:00
var nsoName = ApplicationLoader . ExeFsPrefixes [ i ] ;
2020-07-08 21:31:15 -07:00
FileInfo nsoFile = new FileInfo ( Path . Combine ( mod . Path . FullName , nsoName ) ) ;
if ( nsoFile . Exists )
{
2020-12-29 11:54:32 -08:00
if ( modLoadResult . Replaces [ 1 < < i ] )
2020-07-08 21:31:15 -07:00
{
2020-08-03 16:32:53 -07:00
Logger . Warning ? . Print ( LogClass . ModLoader , $"Multiple replacements to '{nsoName}'" ) ;
2020-12-29 11:54:32 -08:00
2020-07-08 21:31:15 -07:00
continue ;
}
2020-12-29 11:54:32 -08:00
modLoadResult . Replaces [ 1 < < i ] = true ;
2020-07-08 21:31:15 -07:00
nsos [ i ] = new NsoExecutable ( nsoFile . OpenRead ( ) . AsStorage ( ) , nsoName ) ;
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , $"NSO '{nsoName}' replaced" ) ;
2020-12-29 11:54:32 -08:00
}
2020-07-08 21:31:15 -07:00
2020-12-29 11:54:32 -08:00
modLoadResult . Stubs [ 1 < < i ] | = File . Exists ( Path . Combine ( mod . Path . FullName , nsoName + StubExtension ) ) ;
}
FileInfo npdmFile = new FileInfo ( Path . Combine ( mod . Path . FullName , "main.npdm" ) ) ;
2021-01-03 03:30:31 -08:00
if ( npdmFile . Exists )
2020-12-29 11:54:32 -08:00
{
2021-01-03 03:30:31 -08:00
if ( modLoadResult . Npdm ! = null )
2020-12-29 11:54:32 -08:00
{
Logger . Warning ? . Print ( LogClass . ModLoader , "Multiple replacements to 'main.npdm'" ) ;
2020-07-08 21:31:15 -07:00
continue ;
}
2021-08-12 14:56:24 -07:00
modLoadResult . Npdm = new MetaLoader ( ) ;
modLoadResult . Npdm . Load ( File . ReadAllBytes ( npdmFile . FullName ) ) ;
2021-02-19 16:25:01 -08:00
2021-08-12 14:56:24 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , "main.npdm replaced" ) ;
2020-07-08 21:31:15 -07:00
}
}
2020-12-29 11:54:32 -08:00
for ( int i = ApplicationLoader . ExeFsPrefixes . Length - 1 ; i > = 0 ; - - i )
2020-07-08 21:31:15 -07:00
{
2020-12-29 11:54:32 -08:00
if ( modLoadResult . Stubs [ 1 < < i ] & & ! modLoadResult . Replaces [ 1 < < i ] ) // Prioritizes replacements over stubs
2020-07-08 21:31:15 -07:00
{
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , $" NSO '{nsos[i].Name}' stubbed" ) ;
2020-12-29 11:54:32 -08:00
nsos [ i ] = null ;
2020-07-08 21:31:15 -07:00
}
}
2020-12-29 11:54:32 -08:00
return modLoadResult ;
2020-07-08 21:31:15 -07:00
}
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 )
{
2020-07-14 16:40:17 -07:00
IEnumerable < Mod < DirectoryInfo > > nsoMods = Patches . NsoPatches ;
if ( AppMods . TryGetValue ( titleId , out ModCache mods ) )
{
nsoMods = nsoMods . Concat ( mods . ExefsDirs ) ;
}
2020-07-08 21:31:15 -07:00
// 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 ) ;
}
2021-03-27 07:12:05 -07:00
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" ) ;
2021-04-02 06:42:25 -07:00
return ;
2021-03-27 07:12:05 -07:00
}
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}'" ) ;
2022-01-03 00:39:43 -08:00
tamperMachine . InstallAtmosphereCheat ( cheat . Name , cheatId , cheat . Instructions , tamperInfo , exeAddress ) ;
}
EnableCheats ( titleId , tamperMachine ) ;
}
internal void EnableCheats ( ulong titleId , TamperMachine tamperMachine )
{
var contentDirectory = FindTitleDir ( new DirectoryInfo ( Path . Combine ( GetModsBasePath ( ) , AmsContentsDir ) ) , $"{titleId:x16}" ) ;
string enabledCheatsPath = Path . Combine ( contentDirectory . FullName , CheatDir , "enabled.txt" ) ;
if ( File . Exists ( enabledCheatsPath ) )
{
tamperMachine . EnableCheats ( File . ReadAllLines ( enabledCheatsPath ) ) ;
2021-03-27 07:12:05 -07:00
}
}
2020-07-08 21:31:15 -07:00
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 ;
}
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , $"Matching IPS patch '{patchFile.Name}' in '{mod.Name}' bid={buildId}" ) ;
2020-07-08 21:31:15 -07:00
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 ;
}
2020-08-03 16:32:53 -07:00
Logger . Info ? . Print ( LogClass . ModLoader , $"Matching IPSwitch patch '{patchFile.Name}' in '{mod.Name}' bid={patcher.BuildId}" ) ;
2020-07-08 21:31:15 -07:00
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 ;
}
}
2020-12-29 11:54:32 -08:00
}