using System; using System.Collections.Generic; using System.Linq; using System.IO; using System.Threading.Tasks; using Toolbox.Library; using Toolbox.Library.IO; using Toolbox.Library.Forms; using LibHac; using LibHac.IO; using System.Windows.Forms; using System.Runtime.InteropServices; using System.Reflection; using System.ComponentModel; namespace FirstPlugin { public class NSP : IArchiveFile, IFileFormat { public FileType FileType { get; set; } = FileType.Rom; public bool CanSave { get; set; } public string[] Description { get; set; } = new string[] { "NSP" }; public string[] Extension { get; set; } = new string[] { "*.nsp" }; public string FileName { get; set; } public string FilePath { get; set; } public IFileInfo IFileInfo { get; set; } public Type[] Types { get { List types = new List(); return types.ToArray(); } } Nca Control { get; set; } RomfsNodeWrapper romfsWrapper; public bool CanAddFiles { get; set; } public bool CanRenameFiles { get; set; } public bool CanReplaceFiles { get; set; } public bool CanDeleteFiles { get; set; } public List files = new List(); public IEnumerable Files => files; public void ClearFiles() { files.Clear(); } public bool Identify(System.IO.Stream stream) { return Utils.HasExtension(FileName, ".nsp"); } public void Load(System.IO.Stream stream) { string homeFolder = Environment.GetFolderPath(Environment.SpecialFolder.UserProfile); string KeyFile = Path.Combine(homeFolder, ".switch", "prod.keys"); string TitleKeyFile = Path.Combine(homeFolder, ".switch", "title.keys"); var Keys = ExternalKeys.ReadKeyFile(KeyFile, TitleKeyFile); Stream Input; var Pfs = new Pfs(stream.AsStorage()); 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 Cnmt = new Cnmt(CnmtPfs.OpenFile(CnmtPfs.Files[0]).AsStream()); var Program = Cnmt.ContentEntries.FirstOrDefault(c => c.Type == CnmtContentType.Program); var CtrlEntry = Cnmt.ContentEntries.FirstOrDefault(c => c.Type == CnmtContentType.Control); if (CtrlEntry != null) 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)); for (int i = 0; i < romfs.Files.Count; i++) files.Add(new FileEntry(romfs,romfs.Files[i])); } public void Unload() { files.Clear(); Control = null; } public byte[] Save() { return null; } public bool AddFile(ArchiveFileInfo archiveFileInfo) { return false; } public bool DeleteFile(ArchiveFileInfo archiveFileInfo) { return false; } public class FileEntry : ArchiveFileInfo { private Romfs ParentROMFS; private RomfsFile File; [Browsable(false)] public override object DisplayProperties => this; public override string FileSize => STMath.GetFileSize(File.DataLength, 4); [Browsable(false)] public override byte[] FileData { get { using (var stream = ParentROMFS.OpenFile(File).AsStream()) { var mem = new MemoryStream(); stream.CopyTo(mem); return mem.ToArray(); } } } public override void Export() { string fileName = Path.GetFileName(FileName.RemoveIllegaleFolderNameCharacters()); SaveFileDialog sfd = new SaveFileDialog(); sfd.FileName = fileName; sfd.DefaultExt = Path.GetExtension(fileName); sfd.Filter = "Raw Data (*.*)|*.*"; if (sfd.ShowDialog() == DialogResult.OK) { ParentROMFS.OpenFile(File).WriteAllBytes($"{sfd.FileName}"); } } public FileEntry(Romfs romfs, RomfsFile romfsFile) { ParentROMFS = romfs; File = romfsFile; FileName = File.FullPath; } } } }