mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-09-19 17:27:51 +03:00
b00476abac
Let's replace this bool with an `enum class` in order to enhance readability. This is done by repurposing `MappedFile`'s `OpenMode` into a shared `enum` simply called `Mode`.
75 lines
2.0 KiB
C++
75 lines
2.0 KiB
C++
/*
|
|
* Copyright (c) 2018-2021, Andreas Kling <kling@serenityos.org>
|
|
* Copyright (c) 2023, kleines Filmröllchen <filmroellchen@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <AK/ScopeGuard.h>
|
|
#include <LibCore/File.h>
|
|
#include <LibCore/MappedFile.h>
|
|
#include <LibCore/System.h>
|
|
#include <fcntl.h>
|
|
#include <sys/mman.h>
|
|
#include <unistd.h>
|
|
|
|
namespace Core {
|
|
|
|
ErrorOr<NonnullOwnPtr<MappedFile>> MappedFile::map(StringView path, Mode mode)
|
|
{
|
|
auto const file_mode = mode == Mode::ReadOnly ? O_RDONLY : O_RDWR;
|
|
auto fd = TRY(Core::System::open(path, file_mode | O_CLOEXEC, 0));
|
|
return map_from_fd_and_close(fd, path, mode);
|
|
}
|
|
|
|
ErrorOr<NonnullOwnPtr<MappedFile>> MappedFile::map_from_file(NonnullOwnPtr<Core::File> stream, StringView path)
|
|
{
|
|
return map_from_fd_and_close(stream->leak_fd(Badge<MappedFile> {}), path);
|
|
}
|
|
|
|
ErrorOr<NonnullOwnPtr<MappedFile>> MappedFile::map_from_fd_and_close(int fd, [[maybe_unused]] StringView path, Mode mode)
|
|
{
|
|
TRY(Core::System::fcntl(fd, F_SETFD, FD_CLOEXEC));
|
|
|
|
ScopeGuard fd_close_guard = [fd] {
|
|
::close(fd);
|
|
};
|
|
|
|
auto stat = TRY(Core::System::fstat(fd));
|
|
auto size = stat.st_size;
|
|
|
|
int protection;
|
|
int flags;
|
|
switch (mode) {
|
|
case Mode::ReadOnly:
|
|
protection = PROT_READ;
|
|
flags = MAP_SHARED;
|
|
break;
|
|
case Mode::ReadWrite:
|
|
protection = PROT_READ | PROT_WRITE;
|
|
// Don't map a read-write mapping shared as a precaution.
|
|
flags = MAP_PRIVATE;
|
|
break;
|
|
}
|
|
|
|
auto* ptr = TRY(Core::System::mmap(nullptr, size, protection, flags, fd, 0, 0, path));
|
|
|
|
return adopt_own(*new MappedFile(ptr, size, mode));
|
|
}
|
|
|
|
MappedFile::MappedFile(void* ptr, size_t size, Mode mode)
|
|
: FixedMemoryStream(Bytes { ptr, size }, mode)
|
|
, m_data(ptr)
|
|
, m_size(size)
|
|
{
|
|
}
|
|
|
|
MappedFile::~MappedFile()
|
|
{
|
|
auto res = Core::System::munmap(m_data, m_size);
|
|
if (res.is_error())
|
|
dbgln("Failed to unmap MappedFile (@ {:p}): {}", m_data, res.error());
|
|
}
|
|
|
|
}
|