mirror of
https://github.com/KillzXGaming/Switch-Toolbox
synced 2024-12-24 03:43:08 +00:00
141 lines
4.6 KiB
C#
141 lines
4.6 KiB
C#
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<Type> types = new List<Type>();
|
|
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<FileEntry> files = new List<FileEntry>();
|
|
public IEnumerable<ArchiveFileInfo> 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 FileEntry(Romfs romfs, RomfsFile romfsFile)
|
|
{
|
|
ParentROMFS = romfs;
|
|
File = romfsFile;
|
|
FileName = File.FullPath;
|
|
}
|
|
}
|
|
}
|
|
}
|