mirrors/LibHac
1
0
Fork 0
mirror of https://github.com/Thealexbarney/LibHac.git synced 2024-11-14 10:49:41 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix regression caused by decrypting multiple AES blocks at a time

Browse source
This commit is contained in:
Alex Barney 2018-08-28 13:22:19 -05:00
parent dcb25713ba
commit 7f016f875b
5 changed files with 51 additions and 51 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
hactoolnet/Program.cs
View file

@ -379,29 +379,26 @@ namespace hactoolnet
private static void ProcessRomFs(Context ctx, Romfs romfs)
{
using (var file = new FileStream(ctx.Options.InFile, FileMode.Open, FileAccess.Read))
if (ctx.Options.ListRomFs)
{
if (ctx.Options.ListRomFs)
foreach (var romfsFile in romfs.Files)
{
foreach (var romfsFile in romfs.Files)
{
ctx.Logger.LogMessage(romfsFile.FullPath);
}
ctx.Logger.LogMessage(romfsFile.FullPath);
}
}
if (ctx.Options.RomfsOut != null)
if (ctx.Options.RomfsOut != null)
{
using (var outFile = new FileStream(ctx.Options.RomfsOut, FileMode.Create, FileAccess.ReadWrite))
{
using (var outFile = new FileStream(ctx.Options.RomfsOut, FileMode.Create, FileAccess.ReadWrite))
{
var romfsStream = romfs.OpenRawStream();
romfsStream.CopyStream(outFile, romfsStream.Length, ctx.Logger);
}
var romfsStream = romfs.OpenRawStream();
romfsStream.CopyStream(outFile, romfsStream.Length, ctx.Logger);
}
}
if (ctx.Options.RomfsOutDir != null)
{
romfs.Extract(ctx.Options.RomfsOutDir, ctx.Logger);
}
if (ctx.Options.RomfsOutDir != null)
{
romfs.Extract(ctx.Options.RomfsOutDir, ctx.Logger);
}
}

6
libhac/Aes128CtrStream.cs
View file

@ -134,5 +134,11 @@ namespace libhac
return _decryptor.TransformBlock(_tempBuffer, 0, bytesRead, buffer, offset);
}
protected override void ValidateSizeMultiple(long value)
{
if (value % 0x10 != 0)
throw new ArgumentException($"Value needs to be a multiple of {SectorSize}");
}
}
}

63
libhac/Nca.cs
View file

@ -87,62 +87,59 @@ namespace libhac
long offset = sect.Offset;
long size = sect.Size;
if (!raw)
{
switch (sect.Header.Type)
{
case SectionType.Pfs0:
offset = sect.Offset + sect.Pfs0.Superblock.Pfs0Offset;
size = sect.Pfs0.Superblock.Pfs0Size;
break;
case SectionType.Romfs:
offset = sect.Offset + sect.Header.Romfs.IvfcHeader.LevelHeaders[Romfs.IvfcMaxLevel - 1].LogicalOffset;
size = sect.Header.Romfs.IvfcHeader.LevelHeaders[Romfs.IvfcMaxLevel - 1].HashDataSize;
break;
case SectionType.Bktr:
break;
default:
throw new ArgumentOutOfRangeException();
}
}
var sectionStream = StreamSource.CreateStream(offset, size);
Stream rawStream = StreamSource.CreateStream(offset, size);
switch (sect.Header.CryptType)
{
case SectionCryptType.None:
return sectionStream;
break;
case SectionCryptType.XTS:
break;
case SectionCryptType.CTR:
return new RandomAccessSectorStream(new Aes128CtrStream(sectionStream, DecryptedKeys[2], offset, sect.Header.Ctr), false);
rawStream = new RandomAccessSectorStream(new Aes128CtrStream(rawStream, DecryptedKeys[2], offset, sect.Header.Ctr), false);
break;
case SectionCryptType.BKTR:
var patchStream = new RandomAccessSectorStream(
new BktrCryptoStream(sectionStream, DecryptedKeys[2], 0, size, offset, sect.Header.Ctr, sect.Header.Bktr),
rawStream = new RandomAccessSectorStream(
new BktrCryptoStream(rawStream, DecryptedKeys[2], 0, size, offset, sect.Header.Ctr, sect.Header.Bktr),
false);
if (BaseNca == null)
{
return patchStream;
return rawStream;
}
else
{
var dataLevel = sect.Header.Bktr.IvfcHeader.LevelHeaders[Romfs.IvfcMaxLevel - 1];
var baseSect = BaseNca.Sections.FirstOrDefault(x => x.Type == SectionType.Romfs);
if (baseSect == null) throw new InvalidDataException("Base NCA has no RomFS section");
var baseStream = BaseNca.OpenSection(baseSect.SectionNum, true);
var virtStreamRaw = new Bktr(patchStream, baseStream, sect);
if (raw) return virtStreamRaw;
var virtStream = new SubStream(virtStreamRaw, dataLevel.LogicalOffset, dataLevel.HashDataSize);
return virtStream;
rawStream = new Bktr(rawStream, baseStream, sect);
}
break;
default:
throw new ArgumentOutOfRangeException();
}
return sectionStream;
if (raw) return rawStream;
switch (sect.Header.Type)
{
case SectionType.Pfs0:
offset = sect.Pfs0.Superblock.Pfs0Offset;
size = sect.Pfs0.Superblock.Pfs0Size;
break;
case SectionType.Romfs:
offset = sect.Header.Romfs.IvfcHeader.LevelHeaders[Romfs.IvfcMaxLevel - 1].LogicalOffset;
size = sect.Header.Romfs.IvfcHeader.LevelHeaders[Romfs.IvfcMaxLevel - 1].HashDataSize;
break;
case SectionType.Bktr:
offset = sect.Header.Bktr.IvfcHeader.LevelHeaders[Romfs.IvfcMaxLevel - 1].LogicalOffset;
size = sect.Header.Bktr.IvfcHeader.LevelHeaders[Romfs.IvfcMaxLevel - 1].HashDataSize;
break;
default:
throw new ArgumentOutOfRangeException();
}
return new SubStream(rawStream, offset, size);
}
public void SetBaseNca(Nca baseNca) => BaseNca = baseNca;

2
libhac/XTSSharp/RandomAccessSectorStream.cs
View file

@ -334,7 +334,7 @@ namespace libhac.XTSSharp
_bufferDirty = false;
_currentBufferSize = sector.Length;
CacheHits++;
_s.Position += _bufferSize;
_s.Position += _currentBufferSize;
return;
}
}

2
libhac/XTSSharp/SectorStream.cs
View file

@ -145,7 +145,7 @@ namespace libhac.XTSSharp
/// Validates that the size is a multiple of the sector size
/// </summary>
// ReSharper disable once ParameterOnlyUsedForPreconditionCheck.Local
private void ValidateSizeMultiple(long value)
protected virtual void ValidateSizeMultiple(long value)
{
if (value % SectorSize != 0)
throw new ArgumentException(string.Format("Value needs to be a multiple of {0}", SectorSize));