public override unsafe Native.NtStatus NtReadFileImpl(IntPtr handle, IntPtr hEvent, IntPtr *apcRoutine, IntPtr *apcContext, ref Native.IO_STATUS_BLOCK ioStatus,
byte *buffer, uint length, Native.LARGE_INTEGER *byteOffset, IntPtr key)
{
if (!mWaveBankByHandle.TryGetValue(handle, out var waveBank))
{
return(mHooks.NtReadFileHook.OriginalFunction(handle, hEvent, apcRoutine, apcContext, ref ioStatus, buffer, length, byteOffset, key));
}
var offset = waveBank.FilePointer;
var reqOffset = (byteOffset != null || (byteOffset != null && byteOffset->HighPart == -1 && byteOffset->LowPart == FILE_USE_FILE_POINTER_POSITION)) ?
byteOffset->QuadPart : -1;
var effOffset = reqOffset == -1 ? offset : reqOffset;
var result = NtStatus.Success;
var waveDataOffset = waveBank.Native.Header->Segments[(int)WaveBankSegmentIndex.EntryWaveData].Offset;
if ((effOffset + length) <= waveDataOffset)
{
// Header read
Unsafe.CopyBlock(buffer, waveBank.Native.Ptr + effOffset, length);
SetBytesRead(handle, (int)offset, (int)length, ref ioStatus);
result = NtStatus.Success;
}
else if (effOffset >= waveDataOffset && effOffset < waveBank.VirtualFileSize)
{
if ((waveBank.Native.Data->Flags & WaveBankFlags.Compact) != 0)
{
// TODO: compact format
result = mHooks.NtReadFileHook.OriginalFunction(handle, hEvent, apcRoutine, apcContext, ref ioStatus, buffer, length, byteOffset, key);
}
else
{
ReadEntryWaveData(waveBank, effOffset, handle, hEvent, apcRoutine, apcContext, ref ioStatus, buffer, length, byteOffset, key);
}
}
else
{
mLogger.Error($"{waveBank.FileName} Hnd: {handle} Unexpected read request!! Offset: {effOffset:X8} Length: {length:X8}");
result = mHooks.NtReadFileHook.OriginalFunction(handle, hEvent, apcRoutine, apcContext, ref ioStatus, buffer, length, byteOffset, key);
}
if (result != NtStatus.Success)
{
mLogger.Error($"{waveBank.FileName} Hnd: {handle} NtReadFile failed with {result}!!!");
}
return(result);
}
private NtStatus ReadEntryWaveData(VirtualWaveBank waveBank, long absOffset,
IntPtr handle, IntPtr hEvent, IntPtr *apcRoutine, IntPtr *apcContext, ref Native.IO_STATUS_BLOCK ioStatus, byte *buffer, uint length, Native.LARGE_INTEGER *byteOffset, IntPtr key)
{
var status = NtStatus.Success;
var handled = false;
var segBaseOffset = waveBank.Native.Header->Segments[(int)WaveBankSegmentIndex.EntryWaveData].Offset;
var segOffset = absOffset - segBaseOffset;
for (int i = 0; i < waveBank.Entries.Count; i++)
{
var entry = waveBank.Entries[i];
if (segOffset < entry.Native->PlayRegion.Offset || segOffset >= (entry.Native->PlayRegion.Offset + entry.Native->PlayRegion.Length))
{
continue;
}
var fileDataOffset = segOffset - entry.Native->PlayRegion.Offset;
var readEndOffset = fileDataOffset + length;
var nextDataOffset = i < waveBank.Entries.Count - 1 ? waveBank.Entries[i + 1].Native->PlayRegion.Offset : (waveBank.VirtualFileSize - segBaseOffset);
if (readEndOffset > nextDataOffset)
{
continue;
}
handled = true;
if (!entry.IsRedirected)
{
// Trigger cache miss
for (int j = 0; j < mCache.Length; j++)
{
var cacheEntry = mCache[j].Entry;
if (mCache[j].Miss())
{
mLogger.Debug($"{waveBank.FileName} Hnd: {handle} Index: {j} {cacheEntry.CueName} removed from cache");
}
}
mLogger.Info($"{waveBank.FileName} Hnd: {handle} Index: {i} {entry.CueName} Data access Offset: 0x{absOffset:X8} Length: 0x{length:X8}");
status = mHooks.NtReadFileHook.OriginalFunction(handle, hEvent, apcRoutine, apcContext, ref ioStatus, buffer, length, byteOffset, key);
}
else
{
mLogger.Info($"{waveBank.FileName} Hnd: {handle} Index: {i} {entry.CueName} Data access Offset: 0x{absOffset:X8} Length: 0x{length:X8} redirected to {entry.FilePath}");
status = NtStatus.Success;
if (fileDataOffset < 0)
{
mLogger.Error($"{waveBank.FileName} Hnd: {handle} Index: {i} {entry.CueName} Offset is before start of data!!!");
continue;
}
else if (fileDataOffset > entry.FileSize)
{
mLogger.Error($"{waveBank.FileName} Hnd: {handle} Index: {i} {entry.CueName} Offset is after end of data!!!");
//continue;
}
mLogger.Debug($"{waveBank.FileName} Hnd: {handle} Index: {i} {entry.CueName} Reading 0x{length:X8} bytes from redirected file at offset 0x{fileDataOffset:X8}");
// Get cached file stream if the file was previously opened or open a new file
Stream redirectedStream = null;
for (int j = 0; j < mCache.Length; j++)
{
if (mCache[j].Entry == entry)
{
// Found entry in cache, increase score
mCache[j].Hit();
mLogger.Debug($"{waveBank.FileName} Hnd: {handle} Index: {i} {entry.CueName} loaded from cache");
redirectedStream = mCache[j].Stream;
break;
}
else
{
// Entry is not the one we're looking for, so we lower its score
var cacheEntry = mCache[j].Entry;
if (mCache[j].Miss())
{
mLogger.Debug($"{waveBank.FileName} Hnd: {handle} Index: {j} {cacheEntry.CueName} removed from cache");
}
}
}
if (redirectedStream == null)
{
// Wasn't found in cache
mLogger.Debug($"{waveBank.FileName} Hnd: {handle} Index: {i} {entry.CueName} added to cache");
redirectedStream = entry.OpenRead();
for (int j = 0; j < mCache.Length; j++)
{
if (mCache[j].Entry == null)
{
mCache[j] = new CacheEntry()
{
Entry = entry, Score = 1, Stream = redirectedStream
};
break;;
}
}
}
// Read from redirected file into the buffer
try
{
redirectedStream.Seek(fileDataOffset, SeekOrigin.Begin);
var readBytes = redirectedStream.Read(new Span <byte>(( void * )buffer, ( int )length));
SetBytesRead(handle, ( int )waveBank.FilePointer, ( int )length, ref ioStatus);
if (readBytes != length)
{
mLogger.Error($"{waveBank.FileName} Hnd: {handle} Index: {i} {entry.CueName} File read length doesnt match requested read length!! Expected 0x{length:X8}, Actual 0x{readBytes:X8}");
}
mLogger.Debug($"{waveBank.FileName} Hnd: {handle} Index: {i} {entry.CueName} Wrote redirected file to buffer");
}
catch (Exception e)
{
mLogger.Debug($"{waveBank.FileName} Hnd: {handle} Index: {i} {entry.CueName} Unhandled exception thrown during reading {entry.FileName}: {e}");
}
}
}
if (!handled)
{
mLogger.Error($"{waveBank.FileName} Hnd: {handle} Unhandled file data read request!! Offset: 0x{absOffset:X8} Length: 0x{length:X8}");
status = mHooks.NtReadFileHook.OriginalFunction(handle, hEvent, apcRoutine, apcContext, ref ioStatus, buffer, length, byteOffset, key);
}
return(status);
}
