1
0
mirror of synced 2024-12-13 16:21:10 +01:00
Switch-Toolbox/File_Format_Library/FileFormats/Rom/NSP.cs

217 lines
7.0 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, ILeaveOpenOnLoad
{
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; }
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)
{
var Keys = Forms.SwitchKeySelectionForm.ShowKeySelector();
if (Keys == null)
throw new Exception("Failed to get keys. Please select valid paths!");
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());
foreach (var entry in Cnmt.ContentEntries)
{
if (entry.Type == CnmtContentType.Program) {
var Program = entry;
string ncaFileName = $"{Program.NcaId.ToHexString().ToLower()}.nca";
Stream Input = Pfs.OpenFile(ncaFileName).AsStream();
var Nca = new Nca(Keys, Input.AsStorage(), true);
string root = Nca.Header.TitleId.ToString("X");
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, root));
}
for (int i = 0; i < romfs.Files.Count; i++)
files.Add(new FileEntry(romfs, romfs.Files[i], root));
}
}
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);
}
public void Unload()
{
files.Clear();
Control?.Dispose();
}
public void Save(System.IO.Stream stream)
{
}
public bool AddFile(ArchiveFileInfo archiveFileInfo)
{
return false;
}
public bool DeleteFile(ArchiveFileInfo archiveFileInfo)
{
return false;
}
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 PfsFileEntry fileEntry;
public override Stream FileDataStream
{
get
{
return ParentPfs.OpenFile(fileEntry).AsStream();
}
}
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)
{
FileDataStream.ExportToFile(sfd.FileName);
}
}
public PartEntry(Pfs pfs, PfsFileEntry entry)
{
ParentPfs = pfs;
fileEntry = entry;
FileName = fileEntry.Name;
}
}
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);
public override Stream FileDataStream
{
get
{
return ParentROMFS.OpenFile(File).AsStream();
}
}
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()
{
}
public FileEntry(Romfs romfs, RomfsFile romfsFile, string root = "")
{
ParentROMFS = romfs;
File = romfsFile;
if (root != string.Empty)
FileName = $"{root}/Romfs/{File.FullPath}";
else
FileName = $"Romfs/{File.FullPath}";
}
}
}
}