NSP : Add support for loading multiple programs

This commit is contained in:
KillzXGaming 2020-09-16 20:55:48 -04:00
parent e6f4f0562d
commit 5e035ff15e

View file

@ -64,30 +64,35 @@ namespace FirstPlugin
var CnmtNca = new Nca(Keys, Pfs.OpenFile(Pfs.Files.FirstOrDefault(s => s.Name.Contains(".cnmt.nca"))), false); var CnmtNca = new Nca(Keys, Pfs.OpenFile(Pfs.Files.FirstOrDefault(s => s.Name.Contains(".cnmt.nca"))), false);
var CnmtPfs = new Pfs(CnmtNca.OpenSection(0, false, IntegrityCheckLevel.None, true)); var CnmtPfs = new Pfs(CnmtNca.OpenSection(0, false, IntegrityCheckLevel.None, true));
var Cnmt = new Cnmt(CnmtPfs.OpenFile(CnmtPfs.Files[0]).AsStream()); var Cnmt = new Cnmt(CnmtPfs.OpenFile(CnmtPfs.Files[0]).AsStream());
var Program = Cnmt.ContentEntries.FirstOrDefault(c => c.Type == CnmtContentType.Program); foreach (var entry in Cnmt.ContentEntries)
{
if (entry.Type == CnmtContentType.Program) {
var Program = entry;
Input = Pfs.OpenFile($"{Program.NcaId.ToHexString().ToLower()}.nca").AsStream();
var Nca = new Nca(Keys, Input.AsStorage(), true);
Romfs romfs = new Romfs(
Nca.OpenSection(Nca.Sections.FirstOrDefault
(s => s?.Type == SectionType.Romfs || s?.Type == SectionType.Bktr)
.SectionNum, false, IntegrityCheckLevel.None, true));
if (Nca.CanOpenSection((int)ProgramPartitionType.Code))
{
var exefs = new Pfs(Nca.OpenSection((int)ProgramPartitionType.Code,
false, IntegrityCheckLevel.None, true));
foreach (var file in exefs.Files)
files.Add(new ExefsEntry(exefs, file, Program.NcaId.ToHexString()));
}
for (int i = 0; i < romfs.Files.Count; i++)
files.Add(new FileEntry(romfs, romfs.Files[i], Program.NcaId.ToHexString()));
}
}
var CtrlEntry = Cnmt.ContentEntries.FirstOrDefault(c => c.Type == CnmtContentType.Control); var CtrlEntry = Cnmt.ContentEntries.FirstOrDefault(c => c.Type == CnmtContentType.Control);
if (CtrlEntry != null) if (CtrlEntry != null)
Control = new Nca(Keys, Pfs.OpenFile($"{CtrlEntry.NcaId.ToHexString().ToLower()}.nca"), false); Control = new Nca(Keys, Pfs.OpenFile($"{CtrlEntry.NcaId.ToHexString().ToLower()}.nca"), false);
Input = Pfs.OpenFile($"{Program.NcaId.ToHexString().ToLower()}.nca").AsStream();
var Nca = new Nca(Keys, Input.AsStorage(), true);
Romfs romfs = new Romfs(
Nca.OpenSection(Nca.Sections.FirstOrDefault
(s => s?.Type == SectionType.Romfs || s?.Type == SectionType.Bktr)
.SectionNum, false, IntegrityCheckLevel.None, true));
if (Nca.CanOpenSection((int)ProgramPartitionType.Code))
{
var exefs = new Pfs(Nca.OpenSection((int)ProgramPartitionType.Code,
false, IntegrityCheckLevel.None, true));
foreach (var file in exefs.Files)
files.Add(new ExefsEntry(exefs,file));
}
for (int i = 0; i < romfs.Files.Count; i++)
files.Add(new FileEntry(romfs,romfs.Files[i]));
} }
public void Unload() public void Unload()
@ -111,7 +116,18 @@ namespace FirstPlugin
return false; return false;
} }
public class ExefsEntry : FileEntry public class ExefsEntry : PartEntry
{
public ExefsEntry(Pfs pfs, PfsFileEntry entry, string root = "") : base(pfs, entry)
{
if (root != string.Empty)
FileName = $"{root}/Exefs/{entry.Name}";
else
FileName = $"Exefs/{entry.Name}";
}
}
public class PartEntry : FileEntry
{ {
private Pfs ParentPfs; private Pfs ParentPfs;
private PfsFileEntry fileEntry; private PfsFileEntry fileEntry;
@ -139,11 +155,11 @@ namespace FirstPlugin
} }
} }
public ExefsEntry(Pfs pfs, PfsFileEntry entry) public PartEntry(Pfs pfs, PfsFileEntry entry)
{ {
ParentPfs = pfs; ParentPfs = pfs;
fileEntry = entry; fileEntry = entry;
FileName = $"Exefs/{fileEntry.Name}"; FileName = fileEntry.Name;
} }
} }
@ -185,11 +201,14 @@ namespace FirstPlugin
} }
public FileEntry(Romfs romfs, RomfsFile romfsFile) public FileEntry(Romfs romfs, RomfsFile romfsFile, string root = "")
{ {
ParentROMFS = romfs; ParentROMFS = romfs;
File = romfsFile; File = romfsFile;
FileName = $"Romfs/{File.FullPath}"; if (root != string.Empty)
FileName = $"{root}/Romfs/{File.FullPath}";
else
FileName = $"Romfs/{File.FullPath}";
} }
} }
} }