svc: Migrate address range checking functions to VMManager

Provides a bit of a more proper interface for these functions.
This commit is contained in:
Lioncash 2019-03-04 16:30:17 -05:00
parent 07e13d6728
commit 0be8fffc99
3 changed files with 30 additions and 23 deletions

View File

@ -47,23 +47,6 @@ constexpr bool IsValidAddressRange(VAddr address, u64 size) {
return address + size > address; return address + size > address;
} }
// Checks if a given address range lies within a larger address range.
constexpr bool IsInsideAddressRange(VAddr address, u64 size, VAddr address_range_begin,
VAddr address_range_end) {
const VAddr end_address = address + size - 1;
return address_range_begin <= address && end_address <= address_range_end - 1;
}
bool IsInsideAddressSpace(const VMManager& vm, VAddr address, u64 size) {
return IsInsideAddressRange(address, size, vm.GetAddressSpaceBaseAddress(),
vm.GetAddressSpaceEndAddress());
}
bool IsInsideNewMapRegion(const VMManager& vm, VAddr address, u64 size) {
return IsInsideAddressRange(address, size, vm.GetNewMapRegionBaseAddress(),
vm.GetNewMapRegionEndAddress());
}
// 8 GiB // 8 GiB
constexpr u64 MAIN_MEMORY_SIZE = 0x200000000; constexpr u64 MAIN_MEMORY_SIZE = 0x200000000;
@ -105,14 +88,14 @@ ResultCode MapUnmapMemorySanityChecks(const VMManager& vm_manager, VAddr dst_add
return ERR_INVALID_ADDRESS_STATE; return ERR_INVALID_ADDRESS_STATE;
} }
if (!IsInsideAddressSpace(vm_manager, src_addr, size)) { if (!vm_manager.IsWithinAddressSpace(src_addr, size)) {
LOG_ERROR(Kernel_SVC, LOG_ERROR(Kernel_SVC,
"Source is not within the address space, addr=0x{:016X}, size=0x{:016X}", "Source is not within the address space, addr=0x{:016X}, size=0x{:016X}",
src_addr, size); src_addr, size);
return ERR_INVALID_ADDRESS_STATE; return ERR_INVALID_ADDRESS_STATE;
} }
if (!IsInsideNewMapRegion(vm_manager, dst_addr, size)) { if (!vm_manager.IsWithinNewMapRegion(dst_addr, size)) {
LOG_ERROR(Kernel_SVC, LOG_ERROR(Kernel_SVC,
"Destination is not within the new map region, addr=0x{:016X}, size=0x{:016X}", "Destination is not within the new map region, addr=0x{:016X}, size=0x{:016X}",
dst_addr, size); dst_addr, size);
@ -238,7 +221,7 @@ static ResultCode SetMemoryPermission(VAddr addr, u64 size, u32 prot) {
auto* const current_process = Core::CurrentProcess(); auto* const current_process = Core::CurrentProcess();
auto& vm_manager = current_process->VMManager(); auto& vm_manager = current_process->VMManager();
if (!IsInsideAddressSpace(vm_manager, addr, size)) { if (!vm_manager.IsWithinAddressSpace(addr, size)) {
LOG_ERROR(Kernel_SVC, LOG_ERROR(Kernel_SVC,
"Source is not within the address space, addr=0x{:016X}, size=0x{:016X}", addr, "Source is not within the address space, addr=0x{:016X}, size=0x{:016X}", addr,
size); size);
@ -299,7 +282,7 @@ static ResultCode SetMemoryAttribute(VAddr address, u64 size, u32 mask, u32 attr
} }
auto& vm_manager = Core::CurrentProcess()->VMManager(); auto& vm_manager = Core::CurrentProcess()->VMManager();
if (!IsInsideAddressSpace(vm_manager, address, size)) { if (!vm_manager.IsWithinAddressSpace(address, size)) {
LOG_ERROR(Kernel_SVC, LOG_ERROR(Kernel_SVC,
"Given address (0x{:016X}) is outside the bounds of the address space.", address); "Given address (0x{:016X}) is outside the bounds of the address space.", address);
return ERR_INVALID_ADDRESS_STATE; return ERR_INVALID_ADDRESS_STATE;

View File

@ -17,8 +17,8 @@
#include "core/memory_setup.h" #include "core/memory_setup.h"
namespace Kernel { namespace Kernel {
namespace {
static const char* GetMemoryStateName(MemoryState state) { const char* GetMemoryStateName(MemoryState state) {
static constexpr const char* names[] = { static constexpr const char* names[] = {
"Unmapped", "Io", "Unmapped", "Io",
"Normal", "CodeStatic", "Normal", "CodeStatic",
@ -35,6 +35,14 @@ static const char* GetMemoryStateName(MemoryState state) {
return names[ToSvcMemoryState(state)]; return names[ToSvcMemoryState(state)];
} }
// Checks if a given address range lies within a larger address range.
constexpr bool IsInsideAddressRange(VAddr address, u64 size, VAddr address_range_begin,
VAddr address_range_end) {
const VAddr end_address = address + size - 1;
return address_range_begin <= address && end_address <= address_range_end - 1;
}
} // Anonymous namespace
bool VirtualMemoryArea::CanBeMergedWith(const VirtualMemoryArea& next) const { bool VirtualMemoryArea::CanBeMergedWith(const VirtualMemoryArea& next) const {
ASSERT(base + size == next.base); ASSERT(base + size == next.base);
if (permissions != next.permissions || state != next.state || attribute != next.attribute || if (permissions != next.permissions || state != next.state || attribute != next.attribute ||
@ -706,6 +714,11 @@ u64 VMManager::GetAddressSpaceWidth() const {
return address_space_width; return address_space_width;
} }
bool VMManager::IsWithinAddressSpace(VAddr address, u64 size) const {
return IsInsideAddressRange(address, size, GetAddressSpaceBaseAddress(),
GetAddressSpaceEndAddress());
}
VAddr VMManager::GetASLRRegionBaseAddress() const { VAddr VMManager::GetASLRRegionBaseAddress() const {
return aslr_region_base; return aslr_region_base;
} }
@ -786,6 +799,11 @@ u64 VMManager::GetNewMapRegionSize() const {
return new_map_region_end - new_map_region_base; return new_map_region_end - new_map_region_base;
} }
bool VMManager::IsWithinNewMapRegion(VAddr address, u64 size) const {
return IsInsideAddressRange(address, size, GetNewMapRegionBaseAddress(),
GetNewMapRegionEndAddress());
}
VAddr VMManager::GetTLSIORegionBaseAddress() const { VAddr VMManager::GetTLSIORegionBaseAddress() const {
return tls_io_region_base; return tls_io_region_base;
} }

View File

@ -432,6 +432,9 @@ public:
/// Gets the address space width in bits. /// Gets the address space width in bits.
u64 GetAddressSpaceWidth() const; u64 GetAddressSpaceWidth() const;
/// Determines whether or not the given address range lies within the address space.
bool IsWithinAddressSpace(VAddr address, u64 size) const;
/// Gets the base address of the ASLR region. /// Gets the base address of the ASLR region.
VAddr GetASLRRegionBaseAddress() const; VAddr GetASLRRegionBaseAddress() const;
@ -480,6 +483,9 @@ public:
/// Gets the total size of the new map region in bytes. /// Gets the total size of the new map region in bytes.
u64 GetNewMapRegionSize() const; u64 GetNewMapRegionSize() const;
/// Determines whether or not the given address range lies within the new map region
bool IsWithinNewMapRegion(VAddr address, u64 size) const;
/// Gets the base address of the TLS IO region. /// Gets the base address of the TLS IO region.
VAddr GetTLSIORegionBaseAddress() const; VAddr GetTLSIORegionBaseAddress() const;