- private const ulong SystemUpdateTitleId = 0x0100000000000816;

- private Dictionary<string, ulong> _sharedFontTitleDictionary;

- private Dictionary<ulong, string> _systemTitlesNameDictionary;

- private Dictionary<string, string> _sharedFontFilenameDictionary;

- private struct AocItem

- private SortedList<ulong, AocItem> _aocData { get; }

- private VirtualFileSystem _virtualFileSystem;

- _locationEntries = new Dictionary<StorageId, LinkedList<LocationEntry>>();

- { "FontNintendoExtended", 0x0100000000000810 }

- { "FontNintendoExtended", "nintendo_ext_003.bfttf" }

- _aocData = new SortedList<ulong, AocItem>();

- _locationEntries = new Dictionary<StorageId, LinkedList<LocationEntry>>();

- string contentDirectory = null;

- string contentPathString = null;

- contentPathString = ContentPath.GetContentPath(storageId);

- contentDirectory = ContentPath.GetRealPath(_virtualFileSystem, contentPathString);

- LinkedList<LocationEntry> locationList = new LinkedList<LocationEntry>();

- using (FileStream ncaFile = File.OpenRead(Directory.GetFiles(directoryPath)[0]))

- {

- Nca nca = new Nca(_virtualFileSystem.KeySet, ncaFile.AsStorage());

- string switchPath = contentPathString + ":/" + ncaFile.Name.Replace(contentDirectory, string.Empty).TrimStart(Path.DirectorySeparatorChar);

- // Change path format to switch's

- switchPath = switchPath.Replace('\\', '/');

- LocationEntry entry = new LocationEntry(switchPath,

- 0,

- nca.Header.TitleId,

- nca.Header.ContentType);

- AddEntry(entry);

- _contentDictionary.Add((nca.Header.TitleId, nca.Header.ContentType), ncaName);

- }

- using (FileStream ncaFile = new FileStream(filePath, FileMode.Open, FileAccess.Read))

- {

- Nca nca = new Nca(_virtualFileSystem.KeySet, ncaFile.AsStorage());

- string switchPath = contentPathString + ":/" + filePath.Replace(contentDirectory, string.Empty).TrimStart(Path.DirectorySeparatorChar);

- // Change path format to switch's

- switchPath = switchPath.Replace('\\', '/');

- LocationEntry entry = new LocationEntry(switchPath,

- 0,

- nca.Header.TitleId,

- nca.Header.ContentType);

- AddEntry(entry);

- _contentDictionary.Add((nca.Header.TitleId, nca.Header.ContentType), ncaName);

- }

- if (!_locationEntries.ContainsKey(storageId))

- {

- _locationEntries.Add(storageId, locationList);

- }

- if (!_aocData.TryAdd(titleId, new AocItem(containerPath, ncaPath)))

- using FileStream fileStream = File.OpenRead(containerPath);

- public void ClearAocData() => _aocData.Clear();

- public int GetAocCount() => _aocData.Count;

- public IList<ulong> GetAocTitleIds() => _aocData.Select(e => e.Key).ToList();

- if (_aocData.TryGetValue(aocTitleId, out AocItem aoc))

- LinkedList<LocationEntry> locationList = _locationEntries[storageId];

- ulong titleId = content.Key.Item1;

- string installedPath = _virtualFileSystem.SwitchPathToSystemPath(locationEntry.ContentPath);

- using (FileStream file = new FileStream(installedPath, FileMode.Open, FileAccess.Read))

- {

- Nca nca = new Nca(_virtualFileSystem.KeySet, file.AsStorage());

- bool contentCheck = nca.Header.ContentType == contentType;

- return contentCheck;

- }

- if (_locationEntries.ContainsKey(storageId))

- locationList = _locationEntries[storageId];

- if (_locationEntries.ContainsKey(storageId))

- locationList = _locationEntries[storageId];

- string contentPathString = ContentPath.GetContentPath(StorageId.BuiltInSystem);

- string contentDirectory = ContentPath.GetRealPath(_virtualFileSystem, contentPathString);

- string temporaryDirectory = Path.Combine(contentDirectory, "temp");

- FileInfo info = new FileInfo(firmwareSource);

- using (FileStream file = File.OpenRead(firmwareSource))

- {

- switch (info.Extension)

- {

- case ".zip":

- using (ZipArchive archive = ZipFile.OpenRead(firmwareSource))

- {

- InstallFromZip(archive, temporaryDirectory);

- }

- break;

- case ".xci":

- Xci xci = new Xci(_virtualFileSystem.KeySet, file.AsStorage());

- InstallFromCart(xci, temporaryDirectory);

- break;

- default:

- throw new InvalidFirmwarePackageException("Input file is not a valid firmware package");

- }

- FinishInstallation(temporaryDirectory, registeredDirectory);

- Nca nca = new Nca(_virtualFileSystem.KeySet, OpenPossibleFragmentedFile(filesystem, entry.FullPath, OpenMode.Read).AsStorage());

- private void InstallFromZip(ZipArchive archive, string temporaryDirectory)

- using (archive)

- foreach (var entry in archive.Entries)

- if (entry.FullName.EndsWith(".nca") || entry.FullName.EndsWith(".nca/00"))

- {

- // Clean up the name and get the NcaId

- string[] pathComponents = entry.FullName.Replace(".cnmt", "").Split('/');

- string ncaId = pathComponents[pathComponents.Length - 1];

- // If this is a fragmented nca, we need to get the previous element.GetZip

- if (ncaId.Equals("00"))

- {

- ncaId = pathComponents[pathComponents.Length - 2];

- }

- if (ncaId.Contains(".nca"))

- {

- string newPath = Path.Combine(temporaryDirectory, ncaId);

- Directory.CreateDirectory(newPath);

- entry.ExtractToFile(Path.Combine(newPath, "00"));

- }

- public void SaveNca(Nca nca, string ncaId, string temporaryDirectory)

- using (FileStream file = File.Create(Path.Combine(newPath, "00")))

- {

- nca.BaseStorage.AsStream().CopyTo(file);

- }

- private IFile OpenPossibleFragmentedFile(IFileSystem filesystem, string path, OpenMode mode)

- private Stream GetZipStream(ZipArchiveEntry entry)

- using (Stream src = entry.Open())

- {

- src.CopyTo(dest);

- }

- Dictionary<ulong, List<(NcaContentType type, string path)>> updateNcas = new Dictionary<ulong, List<(NcaContentType, string)>>();

- FileInfo info = new FileInfo(firmwarePackage);

- using (FileStream file = File.OpenRead(firmwarePackage))

- switch (info.Extension)

- {

- case ".zip":

- using (ZipArchive archive = ZipFile.OpenRead(firmwarePackage))

- {

- return VerifyAndGetVersionZip(archive);

- }

- case ".xci":

- Xci xci = new Xci(_virtualFileSystem.KeySet, file.AsStorage());

- if (xci.HasPartition(XciPartitionType.Update))

- {

- XciPartition partition = xci.OpenPartition(XciPartitionType.Update);

- return VerifyAndGetVersion(partition);

- }

- else

- {

- throw new InvalidFirmwarePackageException("Update not found in xci file.");

- }

- default:

- break;

- }

- using (Stream ncaStream = GetZipStream(entry))

- {

- IStorage storage = ncaStream.AsStorage();

- Nca nca = new Nca(_virtualFileSystem.KeySet, storage);

- if (updateNcas.TryGetValue(nca.Header.TitleId, out var updateNcasItem))

- {

- updateNcasItem.Add((nca.Header.ContentType, entry.FullName));

- }

- else

- {

- updateNcas.Add(nca.Header.TitleId, new List<(NcaContentType, string)>());

- updateNcas[nca.Header.TitleId].Add((nca.Header.ContentType, entry.FullName));

- }

- Nca metaNca = new Nca(_virtualFileSystem.KeySet, ncaStream.AsStorage());

- using (Stream ncaStream = GetZipStream(archive.GetEntry(versionEntry)))

- {

- Nca nca = new Nca(_virtualFileSystem.KeySet, ncaStream.AsStorage());

- var romfs = nca.OpenFileSystem(NcaSectionType.Data, IntegrityCheckLevel.ErrorOnInvalid);

- using var systemVersionFile = new UniqueRef<IFile>();

- if (romfs.OpenFile(ref systemVersionFile.Ref, "/file".ToU8Span(), OpenMode.Read).IsSuccess())

- {

- systemVersion = new SystemVersion(systemVersionFile.Get.AsStream());

- }

- ZipArchiveEntry metaZipEntry = archive.GetEntry(metaPath);

- using (Stream metaNcaStream = GetZipStream(metaZipEntry))

- {

- using (Stream contentNcaStream = GetZipStream(contentZipEntry))

- {

- Nca metaNca = new Nca(_virtualFileSystem.KeySet, metaNcaStream.AsStorage());

- IFileSystem fs = metaNca.OpenFileSystem(NcaSectionType.Data, IntegrityCheckLevel.ErrorOnInvalid);

- string cnmtPath = fs.EnumerateEntries("/", "*.cnmt").Single().FullPath;

- using var metaFile = new UniqueRef<IFile>();

- if (fs.OpenFile(ref metaFile.Ref, cnmtPath.ToU8Span(), OpenMode.Read).IsSuccess())

- {

- var meta = new Cnmt(metaFile.Get.AsStream());

- IStorage contentStorage = contentNcaStream.AsStorage();

- if (contentStorage.GetSize(out long size).IsSuccess())

- {

- byte[] contentData = new byte[size];

- Span<byte> content = new Span<byte>(contentData);

- contentStorage.Read(0, content);

- Span<byte> hash = new Span<byte>(new byte[32]);

- LibHac.Crypto.Sha256.GenerateSha256Hash(content, hash);

- if (LibHac.Common.Utilities.ArraysEqual(hash.ToArray(), meta.ContentEntries[0].Hash))

- {

- updateNcas.Remove(metaEntry.TitleId);

- }

- }

- foreach (var nca in entry.Value)

- extraNcas += nca.path + Environment.NewLine;

- Nca nca = new Nca(_virtualFileSystem.KeySet, ncaStorage);

- var metaNcaEntry = ncaEntry.Find(x => x.type == NcaContentType.Meta);

- string contentPath = ncaEntry.Find(x => x.type != NcaContentType.Meta).path;

- IStorage metaStorage = OpenPossibleFragmentedFile(filesystem, metaNcaEntry.path, OpenMode.Read).AsStorage();

- Nca metaNca = new Nca(_virtualFileSystem.KeySet, metaStorage);

- Span<byte> content = new Span<byte>(contentData);

- Span<byte> hash = new Span<byte>(new byte[32]);

- var path = _virtualFileSystem.SwitchPathToSystemPath(entry.ContentPath);

- using (FileStream fileStream = File.OpenRead(path))

- {

- Nca nca = new Nca(_virtualFileSystem.KeySet, fileStream.AsStorage());

- if (nca.Header.TitleId == SystemVersionTitleId && nca.Header.ContentType == NcaContentType.Data)

- {

- var romfs = nca.OpenFileSystem(NcaSectionType.Data, IntegrityCheckLevel.ErrorOnInvalid);

- using var systemVersionFile = new UniqueRef<IFile>();

- if (romfs.OpenFile(ref systemVersionFile.Ref, "/file".ToU8Span(), OpenMode.Read).IsSuccess())

- {

- return new SystemVersion(systemVersionFile.Get.AsStream());

- }

-