diff --git a/Ryujinx.Cpu/Jit/MappingTree.cs b/Ryujinx.Cpu/Jit/MappingTree.cs deleted file mode 100644 index 63908a83..00000000 --- a/Ryujinx.Cpu/Jit/MappingTree.cs +++ /dev/null @@ -1,328 +0,0 @@ -using Ryujinx.Common.Collections; -using Ryujinx.Memory; -using Ryujinx.Memory.Range; -using System; -using System.Collections.Generic; -using System.Threading; - -namespace Ryujinx.Cpu.Jit -{ - class MappingTree - { - private const ulong PageSize = 0x1000; - - private enum MappingState : byte - { - Unmapped, - Mapped, - MappedWithMirror - } - - private class Mapping : IntrusiveRedBlackTreeNode, IComparable - { - public ulong Address { get; private set; } - public ulong Size { get; private set; } - public ulong EndAddress => Address + Size; - public ulong BackingOffset { get; private set; } - public MappingState State { get; private set; } - - public Mapping(ulong address, ulong size, ulong backingOffset, MappingState state) - { - Address = address; - Size = size; - BackingOffset = backingOffset; - State = state; - } - - public Mapping Split(ulong splitAddress) - { - ulong leftSize = splitAddress - Address; - ulong rightSize = EndAddress - splitAddress; - - Mapping left = new Mapping(Address, leftSize, BackingOffset, State); - - Address = splitAddress; - Size = rightSize; - - if (State != MappingState.Unmapped) - { - BackingOffset += leftSize; - } - - return left; - } - - public void UpdateState(ulong newBackingOffset, MappingState newState) - { - BackingOffset = newBackingOffset; - State = newState; - } - - public void Extend(ulong sizeDelta) - { - Size += sizeDelta; - } - - public int CompareTo(Mapping other) - { - if (Address < other.Address) - { - return -1; - } - else if (Address <= other.EndAddress - 1UL) - { - return 0; - } - else - { - return 1; - } - } - } - - private readonly IntrusiveRedBlackTree _tree; - private readonly ReaderWriterLock _treeLock; - - public MappingTree(ulong addressSpaceSize) - { - _tree = new IntrusiveRedBlackTree(); - _treeLock = new ReaderWriterLock(); - - _tree.Add(new Mapping(0UL, addressSpaceSize, 0UL, MappingState.Unmapped)); - } - - public void Map(ulong va, ulong pa, ulong size) - { - _treeLock.AcquireWriterLock(Timeout.Infinite); - Update(va, pa, size, MappingState.Mapped); - _treeLock.ReleaseWriterLock(); - } - - public void Unmap(ulong va, ulong size) - { - _treeLock.AcquireWriterLock(Timeout.Infinite); - Update(va, 0UL, size, MappingState.Unmapped); - _treeLock.ReleaseWriterLock(); - } - - public IEnumerable GetPhysicalRegions(ulong va, ulong size) - { - _treeLock.AcquireReaderLock(Timeout.Infinite); - var regions = GetPhysicalRegionsImpl(va, size); - _treeLock.ReleaseReaderLock(); - - return regions; - } - - public (MemoryBlock, ulong) GetContiguousBlock(MemoryBlock backingMemory, MemoryBlock mirror, ulong va, ulong size) - { - _treeLock.AcquireReaderLock(Timeout.Infinite); - var result = GetContiguousBlockImpl(backingMemory, mirror, va, size); - _treeLock.ReleaseReaderLock(); - - return result; - } - - private void Update(ulong va, ulong pa, ulong size, MappingState state) - { - Mapping map = _tree.GetNode(new Mapping(va, 1UL, 0UL, MappingState.Unmapped)); - - Update(map, va, pa, size, state); - } - - private Mapping Update(Mapping map, ulong va, ulong pa, ulong size, MappingState state) - { - ulong endAddress = va + size; - - for (; map != null; map = map.Successor) - { - if (map.Address < va) - { - _tree.Add(map.Split(va)); - } - - if (map.EndAddress > endAddress) - { - Mapping newMap = map.Split(endAddress); - _tree.Add(newMap); - map = newMap; - } - - map.UpdateState(pa, state); - map = TryCoalesce(map); - - if (map.EndAddress >= endAddress) - { - break; - } - } - - return map; - } - - private Mapping TryCoalesce(Mapping map) - { - Mapping previousMap = map.Predecessor; - Mapping nextMap = map.Successor; - - if (previousMap != null && CanCoalesce(previousMap, map)) - { - previousMap.Extend(map.Size); - _tree.Remove(map); - map = previousMap; - } - - if (nextMap != null && CanCoalesce(map, nextMap)) - { - map.Extend(nextMap.Size); - _tree.Remove(nextMap); - } - - return map; - } - - private static bool CanCoalesce(Mapping left, Mapping right) - { - if (left.State != right.State) - { - return false; - } - - return left.State == MappingState.Unmapped || (left.BackingOffset + left.Size == right.BackingOffset); - } - - private IEnumerable GetPhysicalRegionsImpl(ulong va, ulong size) - { - Mapping map = _tree.GetNode(new Mapping(va, 1UL, 0UL, MappingState.Unmapped)); - - if (map == null) - { - ThrowInvalidMemoryRegionException($"Not mapped: va=0x{va:X16}, size=0x{size:X16}"); - } - - var regions = new List(); - - ulong endAddress = va + size; - ulong regionStart = 0; - ulong regionSize = 0; - - for (; map != null; map = map.Successor) - { - if (map.State == MappingState.Unmapped) - { - ThrowInvalidMemoryRegionException($"Not mapped: va=0x{va:X16}, size=0x{size:X16}"); - } - - ulong clampedAddress = Math.Max(map.Address, va); - ulong clampedEndAddress = Math.Min(map.EndAddress, endAddress); - ulong clampedSize = clampedEndAddress - clampedAddress; - - ulong pa = map.BackingOffset + (clampedAddress - map.Address); - - if (pa != regionStart + regionSize) - { - if (regionSize != 0) - { - regions.Add(new MemoryRange(regionStart, regionSize)); - } - - regionStart = pa; - regionSize = clampedSize; - } - else - { - regionSize += clampedSize; - } - - if (map.EndAddress >= endAddress) - { - break; - } - } - - if (regionSize != 0) - { - regions.Add(new MemoryRange(regionStart, regionSize)); - } - - return regions; - } - - private (MemoryBlock, ulong) GetContiguousBlockImpl(MemoryBlock backingMemory, MemoryBlock mirror, ulong va, ulong size) - { - Mapping map = _tree.GetNode(new Mapping(va, 1UL, 0UL, MappingState.Unmapped)); - - ulong endAddress = va + size; - - if (map != null && map.Address <= va && map.EndAddress >= endAddress) - { - ulong pa = map.BackingOffset + (va - map.Address); - return (backingMemory, pa); - } - - if (map != null) - { - Mapping firstMap = map; - - bool contiguous = true; - ulong expectedPa = map.BackingOffset + map.Size; - - while ((map = map.Successor) != null && map.Address < endAddress) - { - if (map.State == MappingState.Unmapped || map.BackingOffset != expectedPa) - { - contiguous = false; - break; - } - - if (map.EndAddress >= endAddress) - { - break; - } - - expectedPa = map.BackingOffset + map.Size; - } - - if (contiguous && map != null) - { - ulong pa = firstMap.BackingOffset + (va - firstMap.Address); - return (backingMemory, pa); - } - - map = firstMap; - } - - ulong endVaAligned = (endAddress + PageSize - 1) & ~(PageSize - 1); - ulong vaAligned = va & ~(PageSize - 1); - - // Make sure the range that will be accessed on the mirror is fully mapped. - for (; map != null; map = map.Successor) - { - if (map.State == MappingState.Mapped) - { - ulong clampedAddress = Math.Max(map.Address, vaAligned); - ulong clampedEndAddress = Math.Min(map.EndAddress, endVaAligned); - ulong clampedSize = clampedEndAddress - clampedAddress; - ulong backingOffset = map.BackingOffset + (clampedAddress - map.Address); - - LockCookie lockCookie = _treeLock.UpgradeToWriterLock(Timeout.Infinite); - - mirror.MapView(backingMemory, backingOffset, clampedAddress, clampedSize); - - map = Update(map, clampedAddress, backingOffset, clampedSize, MappingState.MappedWithMirror); - - _treeLock.DowngradeFromWriterLock(ref lockCookie); - } - - if (map.EndAddress >= endAddress) - { - break; - } - } - - return (mirror, va); - } - - private static void ThrowInvalidMemoryRegionException(string message) => throw new InvalidMemoryRegionException(message); - } -} \ No newline at end of file diff --git a/Ryujinx.Cpu/Jit/MemoryManagerHostMapped.cs b/Ryujinx.Cpu/Jit/MemoryManagerHostMapped.cs index a183f95e..8994e9c0 100644 --- a/Ryujinx.Cpu/Jit/MemoryManagerHostMapped.cs +++ b/Ryujinx.Cpu/Jit/MemoryManagerHostMapped.cs @@ -42,7 +42,7 @@ namespace Ryujinx.Cpu.Jit private readonly ulong _addressSpaceSize; private readonly MemoryBlock _backingMemory; - private readonly MappingTree _mappingTree; + private readonly PageTable _pageTable; private readonly MemoryEhMeilleure _memoryEh; @@ -68,6 +68,7 @@ namespace Ryujinx.Cpu.Jit public MemoryManagerHostMapped(MemoryBlock backingMemory, ulong addressSpaceSize, bool unsafeMode, InvalidAccessHandler invalidAccessHandler = null) { _backingMemory = backingMemory; + _pageTable = new PageTable(); _invalidAccessHandler = invalidAccessHandler; _unsafeMode = unsafeMode; _addressSpaceSize = addressSpaceSize; @@ -83,8 +84,6 @@ namespace Ryujinx.Cpu.Jit AddressSpaceBits = asBits; - _mappingTree = new MappingTree(asSize); - _pageBitmap = new ulong[1 << (AddressSpaceBits - (PageBits + PageToPteShift))]; MemoryAllocationFlags asFlags = MemoryAllocationFlags.Reserve | MemoryAllocationFlags.ViewCompatible; @@ -151,8 +150,9 @@ namespace Ryujinx.Cpu.Jit AssertValidAddressAndSize(va, size); _addressSpace.MapView(_backingMemory, pa, va, size); + _addressSpaceMirror.MapView(_backingMemory, pa, va, size); AddMapping(va, size); - _mappingTree.Map(va, pa, size); + PtMap(va, pa, size); Tracking.Map(va, size); } @@ -166,11 +166,34 @@ namespace Ryujinx.Cpu.Jit Tracking.Unmap(va, size); RemoveMapping(va, size); - _mappingTree.Unmap(va, size); + PtUnmap(va, size); _addressSpace.UnmapView(_backingMemory, va, size); _addressSpaceMirror.UnmapView(_backingMemory, va, size); } + private void PtMap(ulong va, ulong pa, ulong size) + { + while (size != 0) + { + _pageTable.Map(va, pa); + + va += PageSize; + pa += PageSize; + size -= PageSize; + } + } + + private void PtUnmap(ulong va, ulong size) + { + while (size != 0) + { + _pageTable.Unmap(va); + + va += PageSize; + size -= PageSize; + } + } + /// public T Read(ulong va) where T : unmanaged { @@ -178,8 +201,7 @@ namespace Ryujinx.Cpu.Jit { AssertMapped(va, (ulong)Unsafe.SizeOf()); - (MemoryBlock block, ulong offset) = GetContiguousBlock(va, (ulong)Unsafe.SizeOf()); - return block.Read(offset); + return _addressSpaceMirror.Read(va); } catch (InvalidMemoryRegionException) { @@ -219,8 +241,7 @@ namespace Ryujinx.Cpu.Jit { AssertMapped(va, (ulong)data.Length); - (MemoryBlock block, ulong offset) = GetContiguousBlock(va, (ulong)data.Length); - block.Read(offset, data); + _addressSpaceMirror.Read(va, data); } catch (InvalidMemoryRegionException) { @@ -239,8 +260,7 @@ namespace Ryujinx.Cpu.Jit { SignalMemoryTracking(va, (ulong)Unsafe.SizeOf(), write: true); - (MemoryBlock block, ulong offset) = GetContiguousBlock(va, (ulong)Unsafe.SizeOf()); - block.Write(offset, value); + _addressSpaceMirror.Write(va, value); } catch (InvalidMemoryRegionException) { @@ -258,8 +278,7 @@ namespace Ryujinx.Cpu.Jit { SignalMemoryTracking(va, (ulong)data.Length, write: true); - (MemoryBlock block, ulong offset) = GetContiguousBlock(va, (ulong)data.Length); - block.Write(offset, data); + _addressSpaceMirror.Write(va, data); } catch (InvalidMemoryRegionException) { @@ -277,8 +296,7 @@ namespace Ryujinx.Cpu.Jit { AssertMapped(va, (ulong)data.Length); - (MemoryBlock block, ulong offset) = GetContiguousBlock(va, (ulong)data.Length); - block.Write(offset, data); + _addressSpaceMirror.Write(va, data); } catch (InvalidMemoryRegionException) { @@ -301,8 +319,7 @@ namespace Ryujinx.Cpu.Jit AssertMapped(va, (ulong)size); } - (MemoryBlock block, ulong offset) = GetContiguousBlock(va, (ulong)size); - return block.GetSpan(offset, size); + return _addressSpaceMirror.GetSpan(va, size); } /// @@ -317,8 +334,7 @@ namespace Ryujinx.Cpu.Jit AssertMapped(va, (ulong)size); } - (MemoryBlock block, ulong offset) = GetContiguousBlock(va, (ulong)size); - return block.GetWritableRegion(offset, size); + return _addressSpaceMirror.GetWritableRegion(va, size); } /// @@ -326,8 +342,7 @@ namespace Ryujinx.Cpu.Jit { SignalMemoryTracking(va, (ulong)Unsafe.SizeOf(), true); - (MemoryBlock block, ulong offset) = GetContiguousBlock(va, (ulong)Unsafe.SizeOf()); - return ref block.GetRef(offset); + return ref _addressSpaceMirror.GetRef(va); } /// @@ -414,7 +429,51 @@ namespace Ryujinx.Cpu.Jit /// public IEnumerable GetPhysicalRegions(ulong va, ulong size) { - return _mappingTree.GetPhysicalRegions(va, size); + int pages = GetPagesCount(va, (uint)size, out va); + + var regions = new List(); + + ulong regionStart = GetPhysicalAddressChecked(va); + ulong regionSize = PageSize; + + for (int page = 0; page < pages - 1; page++) + { + if (!ValidateAddress(va + PageSize)) + { + return null; + } + + ulong newPa = GetPhysicalAddressChecked(va + PageSize); + + if (GetPhysicalAddressChecked(va) + PageSize != newPa) + { + regions.Add(new MemoryRange(regionStart, regionSize)); + regionStart = newPa; + regionSize = 0; + } + + va += PageSize; + regionSize += PageSize; + } + + regions.Add(new MemoryRange(regionStart, regionSize)); + + return regions; + } + + private ulong GetPhysicalAddressChecked(ulong va) + { + if (!IsMapped(va)) + { + ThrowInvalidMemoryRegionException($"Not mapped: va=0x{va:X16}"); + } + + return GetPhysicalAddressInternal(va); + } + + private ulong GetPhysicalAddressInternal(ulong va) + { + return _pageTable.Read(va) + (va & PageMask); } /// @@ -626,11 +685,6 @@ namespace Ryujinx.Cpu.Jit return new CpuSmartMultiRegionHandle(Tracking.BeginSmartGranularTracking(address, size, granularity)); } - private (MemoryBlock, ulong) GetContiguousBlock(ulong va, ulong size) - { - return _mappingTree.GetContiguousBlock(_backingMemory, _addressSpaceMirror, va, size); - } - /// /// Adds the given address mapping to the page table. ///