RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-25 18:27:35 +00:00
Code Issues Activity Actions

Kernel: Use an ArmedScopeGuard to revert changes after failed mmap

Browse source
This commit is contained in:
Hendiadyoin1 2022-02-16 23:33:18 +01:00 • committed by Brian Gianforcaro
parent a4e3ae0ee9
commit 790d620b39
1 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

13
Kernel/Syscalls/mmap.cpp
View file

@ -5,6 +5,7 @@
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <AK/kmalloc.h>
#include <Kernel/Arch/SmapDisabler.h>
#include <Kernel/Arch/x86/MSR.h>
#include <Kernel/Arch/x86/SafeMem.h>
@ -184,8 +185,16 @@ ErrorOr<FlatPtr> Process::sys$mmap(Userspace<const Syscall::SC_mmap_params*> use
return EINVAL;
Memory::Region* region = nullptr;
Memory::VirtualRange range { {}, 0 };
auto range = TRY([&]() -> ErrorOr<Memory::VirtualRange> {
ArmedScopeGuard scope_guard = [&] {
if (region)
address_space().deallocate_region(*region);
else if (range.is_valid())
address_space().page_directory().range_allocator().deallocate(range);
};
range = TRY([&]() -> ErrorOr<Memory::VirtualRange> {
if (map_randomized)
return address_space().page_directory().range_allocator().try_allocate_randomized(rounded_size, alignment);
@ -248,6 +257,8 @@ ErrorOr<FlatPtr> Process::sys$mmap(Userspace<const Syscall::SC_mmap_params*> use
PerformanceManager::add_mmap_perf_event(*this, *region);
scope_guard.disarm();
return region->vaddr().get();
}