mirror of
https://github.com/Thealexbarney/LibHac.git
synced 2024-11-14 10:49:41 +01:00
314 lines
11 KiB
C#
314 lines
11 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using LibHac;
|
|
using LibHac.IO;
|
|
using LibHac.IO.RomFs;
|
|
using LibHac.IO.Save;
|
|
|
|
namespace hactoolnet
|
|
{
|
|
internal static class ProcessSwitchFs
|
|
{
|
|
public static void Process(Context ctx)
|
|
{
|
|
SwitchFs switchFs;
|
|
var baseFs = new LocalFileSystem(ctx.Options.InFile);
|
|
|
|
if (Directory.Exists(Path.Combine(ctx.Options.InFile, "Nintendo", "Contents", "registered")))
|
|
{
|
|
ctx.Logger.LogMessage("Treating path as SD card storage");
|
|
switchFs = SwitchFs.OpenSdCard(ctx.Keyset, baseFs);
|
|
}
|
|
else if (Directory.Exists(Path.Combine(ctx.Options.InFile, "Contents", "registered")))
|
|
{
|
|
ctx.Logger.LogMessage("Treating path as NAND storage");
|
|
switchFs = SwitchFs.OpenNandPartition(ctx.Keyset, baseFs);
|
|
}
|
|
else
|
|
{
|
|
ctx.Logger.LogMessage("Treating path as a directory of loose NCAs");
|
|
switchFs = SwitchFs.OpenNcaDirectory(ctx.Keyset, baseFs);
|
|
}
|
|
|
|
if (ctx.Options.ListNcas)
|
|
{
|
|
ctx.Logger.LogMessage(ListNcas(switchFs));
|
|
}
|
|
|
|
if (ctx.Options.ListTitles)
|
|
{
|
|
ctx.Logger.LogMessage(ListTitles(switchFs));
|
|
}
|
|
|
|
if (ctx.Options.ListApps)
|
|
{
|
|
ctx.Logger.LogMessage(ListApplications(switchFs));
|
|
}
|
|
|
|
if (ctx.Options.ExefsOutDir != null || ctx.Options.ExefsOut != null)
|
|
{
|
|
ulong id = ctx.Options.TitleId;
|
|
if (id == 0)
|
|
{
|
|
ctx.Logger.LogMessage("Title ID must be specified to dump ExeFS");
|
|
return;
|
|
}
|
|
|
|
if (!switchFs.Titles.TryGetValue(id, out Title title))
|
|
{
|
|
ctx.Logger.LogMessage($"Could not find title {id:X16}");
|
|
return;
|
|
}
|
|
|
|
if (title.MainNca == null)
|
|
{
|
|
ctx.Logger.LogMessage($"Could not find main data for title {id:X16}");
|
|
return;
|
|
}
|
|
|
|
NcaSection section = title.MainNca.Sections[(int)ProgramPartitionType.Code];
|
|
|
|
if (section == null)
|
|
{
|
|
ctx.Logger.LogMessage($"Main NCA for title {id:X16} has no ExeFS section");
|
|
return;
|
|
}
|
|
|
|
if (ctx.Options.ExefsOutDir != null)
|
|
{
|
|
title.MainNca.ExtractSection(section.SectionNum, ctx.Options.ExefsOutDir, ctx.Options.IntegrityLevel, ctx.Logger);
|
|
}
|
|
|
|
if (ctx.Options.ExefsOut != null)
|
|
{
|
|
title.MainNca.ExportSection(section.SectionNum, ctx.Options.ExefsOut, ctx.Options.Raw, ctx.Options.IntegrityLevel, ctx.Logger);
|
|
}
|
|
}
|
|
|
|
if (ctx.Options.RomfsOutDir != null || ctx.Options.RomfsOut != null)
|
|
{
|
|
ulong id = ctx.Options.TitleId;
|
|
if (id == 0)
|
|
{
|
|
ctx.Logger.LogMessage("Title ID must be specified to dump RomFS");
|
|
return;
|
|
}
|
|
|
|
if (!switchFs.Titles.TryGetValue(id, out Title title))
|
|
{
|
|
ctx.Logger.LogMessage($"Could not find title {id:X16}");
|
|
return;
|
|
}
|
|
|
|
if (title.MainNca == null)
|
|
{
|
|
ctx.Logger.LogMessage($"Could not find main data for title {id:X16}");
|
|
return;
|
|
}
|
|
|
|
NcaSection section = title.MainNca.Sections.FirstOrDefault(x => x?.Type == SectionType.Romfs || x?.Type == SectionType.Bktr);
|
|
|
|
if (section == null)
|
|
{
|
|
ctx.Logger.LogMessage($"Main NCA for title {id:X16} has no RomFS section");
|
|
return;
|
|
}
|
|
|
|
if (ctx.Options.RomfsOutDir != null)
|
|
{
|
|
var romfs = new RomFsFileSystem(title.MainNca.OpenSection(section.SectionNum, false, ctx.Options.IntegrityLevel, true));
|
|
romfs.Extract(ctx.Options.RomfsOutDir, ctx.Logger);
|
|
}
|
|
|
|
if (ctx.Options.RomfsOut != null)
|
|
{
|
|
title.MainNca.ExportSection(section.SectionNum, ctx.Options.RomfsOut, ctx.Options.Raw, ctx.Options.IntegrityLevel, ctx.Logger);
|
|
}
|
|
}
|
|
|
|
if (ctx.Options.OutDir != null)
|
|
{
|
|
SaveTitle(ctx, switchFs);
|
|
}
|
|
|
|
if (ctx.Options.NspOut != null)
|
|
{
|
|
ProcessPfs.CreateNsp(ctx, switchFs);
|
|
}
|
|
|
|
if (ctx.Options.SaveOutDir != null)
|
|
{
|
|
ExportSdSaves(ctx, switchFs);
|
|
}
|
|
|
|
if (ctx.Options.Validate)
|
|
{
|
|
ValidateSwitchFs(ctx, switchFs);
|
|
}
|
|
}
|
|
|
|
private static void ValidateSwitchFs(Context ctx, SwitchFs switchFs)
|
|
{
|
|
if (ctx.Options.TitleId != 0)
|
|
{
|
|
ulong id = ctx.Options.TitleId;
|
|
|
|
if (!switchFs.Titles.TryGetValue(id, out Title title))
|
|
{
|
|
ctx.Logger.LogMessage($"Could not find title {id:X16}");
|
|
return;
|
|
}
|
|
|
|
ValidateTitle(ctx, title, "");
|
|
|
|
return;
|
|
}
|
|
|
|
foreach (Application app in switchFs.Applications.Values)
|
|
{
|
|
ctx.Logger.LogMessage($"Checking {app.Name}...");
|
|
|
|
Title mainTitle = app.Patch ?? app.Main;
|
|
|
|
if (mainTitle != null)
|
|
{
|
|
ValidateTitle(ctx, mainTitle, "Main title");
|
|
}
|
|
|
|
foreach (Title title in app.AddOnContent)
|
|
{
|
|
ValidateTitle(ctx, title, "Add-on content");
|
|
}
|
|
}
|
|
}
|
|
|
|
private static void ValidateTitle(Context ctx, Title title, string caption)
|
|
{
|
|
try
|
|
{
|
|
ctx.Logger.LogMessage($" {caption} {title.Id:x16}");
|
|
|
|
foreach (Nca nca in title.Ncas)
|
|
{
|
|
ctx.Logger.LogMessage($" {nca.Header.ContentType.ToString()}");
|
|
|
|
Validity validity = nca.VerifyNca(ctx.Logger, true);
|
|
|
|
ctx.Logger.LogMessage($" {validity.ToString()}");
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
ctx.Logger.LogMessage($"Error processing title {title.Id:x16}:\n{ex.Message}");
|
|
}
|
|
}
|
|
|
|
private static void SaveTitle(Context ctx, SwitchFs switchFs)
|
|
{
|
|
ulong id = ctx.Options.TitleId;
|
|
if (id == 0)
|
|
{
|
|
ctx.Logger.LogMessage("Title ID must be specified to save title");
|
|
return;
|
|
}
|
|
|
|
if (!switchFs.Titles.TryGetValue(id, out Title title))
|
|
{
|
|
ctx.Logger.LogMessage($"Could not find title {id:X16}");
|
|
return;
|
|
}
|
|
|
|
string saveDir = Path.Combine(ctx.Options.OutDir, $"{title.Id:X16}v{title.Version.Version}");
|
|
Directory.CreateDirectory(saveDir);
|
|
|
|
foreach (Nca nca in title.Ncas)
|
|
{
|
|
Stream stream = nca.GetStorage().AsStream();
|
|
string outFile = Path.Combine(saveDir, nca.Filename);
|
|
ctx.Logger.LogMessage(nca.Filename);
|
|
using (var outStream = new FileStream(outFile, FileMode.Create, FileAccess.ReadWrite))
|
|
{
|
|
stream.CopyStream(outStream, stream.Length, ctx.Logger);
|
|
}
|
|
}
|
|
}
|
|
|
|
static string ListTitles(SwitchFs sdfs)
|
|
{
|
|
var table = new TableBuilder("Title ID", "Version", "", "Type", "Size", "Display Version", "Name");
|
|
|
|
foreach (Title title in sdfs.Titles.Values.OrderBy(x => x.Id))
|
|
{
|
|
table.AddRow($"{title.Id:X16}",
|
|
$"v{title.Version?.Version}",
|
|
title.Version?.ToString(),
|
|
title.Metadata?.Type.ToString(),
|
|
Util.GetBytesReadable(title.GetSize()),
|
|
title.Control?.DisplayVersion,
|
|
title.Name);
|
|
}
|
|
|
|
return table.Print();
|
|
}
|
|
|
|
static string ListNcas(SwitchFs sdfs)
|
|
{
|
|
var table = new TableBuilder("NCA ID", "Type", "Title ID");
|
|
|
|
foreach (Nca nca in sdfs.Ncas.Values.OrderBy(x => x.NcaId))
|
|
{
|
|
table.AddRow(nca.NcaId, nca.Header.ContentType.ToString(), nca.Header.TitleId.ToString("X16"));
|
|
}
|
|
|
|
return table.Print();
|
|
}
|
|
|
|
static string ListApplications(SwitchFs sdfs)
|
|
{
|
|
var sb = new StringBuilder();
|
|
|
|
foreach (Application app in sdfs.Applications.Values.OrderBy(x => x.Name))
|
|
{
|
|
sb.AppendLine($"{app.Name} v{app.DisplayVersion}");
|
|
|
|
if (app.Main != null)
|
|
{
|
|
sb.AppendLine($"Software: {Util.GetBytesReadable(app.Main.GetSize())}");
|
|
}
|
|
|
|
if (app.Patch != null)
|
|
{
|
|
sb.AppendLine($"Update Data: {Util.GetBytesReadable(app.Patch.GetSize())}");
|
|
}
|
|
|
|
if (app.AddOnContent.Count > 0)
|
|
{
|
|
sb.AppendLine($"DLC: {Util.GetBytesReadable(app.AddOnContent.Sum(x => x.GetSize()))}");
|
|
}
|
|
|
|
if (app.Nacp?.UserTotalSaveDataSize > 0)
|
|
sb.AppendLine($"User save: {Util.GetBytesReadable(app.Nacp.UserTotalSaveDataSize)}");
|
|
if (app.Nacp?.DeviceTotalSaveDataSize > 0)
|
|
sb.AppendLine($"System save: {Util.GetBytesReadable(app.Nacp.DeviceTotalSaveDataSize)}");
|
|
if (app.Nacp?.BcatDeliveryCacheStorageSize > 0)
|
|
sb.AppendLine($"BCAT save: {Util.GetBytesReadable(app.Nacp.BcatDeliveryCacheStorageSize)}");
|
|
|
|
sb.AppendLine();
|
|
}
|
|
|
|
return sb.ToString();
|
|
}
|
|
|
|
private static void ExportSdSaves(Context ctx, SwitchFs switchFs)
|
|
{
|
|
foreach (KeyValuePair<string, SaveDataFileSystem> save in switchFs.Saves)
|
|
{
|
|
string outDir = Path.Combine(ctx.Options.SaveOutDir, save.Key);
|
|
save.Value.Extract(outDir, ctx.Logger);
|
|
}
|
|
}
|
|
}
|
|
}
|