mirror of
https://github.com/SerenityOS/serenity.git
synced 2026-11-07 12:38:38 -10:00
Sönke Holz
38c162aeea
This is more useful than passing a reference to the Process, which can be easily obtained with Process::current(). Only KCOVDevice previously used this argument. We will need this new argument in the next commit.
89 lines
2.7 KiB
C++
89 lines
2.7 KiB
C++
/*
|
|
* Copyright (c) 2021, Patrick Meyer <[email protected]>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <AK/Assertions.h>
|
|
#include <AK/NonnullOwnPtr.h>
|
|
#include <Kernel/API/Ioctl.h>
|
|
#include <Kernel/API/MajorNumberAllocation.h>
|
|
#include <Kernel/Devices/Device.h>
|
|
#include <Kernel/Devices/KCOVDevice.h>
|
|
#include <Kernel/Devices/KCOVInstance.h>
|
|
#include <Kernel/FileSystem/OpenFileDescription.h>
|
|
|
|
#include <Kernel/Library/Panic.h>
|
|
|
|
namespace Kernel {
|
|
|
|
UNMAP_AFTER_INIT NonnullRefPtr<KCOVDevice> KCOVDevice::must_create()
|
|
{
|
|
return MUST(Device::try_create_device<KCOVDevice>());
|
|
}
|
|
|
|
UNMAP_AFTER_INIT KCOVDevice::KCOVDevice()
|
|
: BlockDevice(MajorAllocation::BlockDeviceFamily::KCOV, 0)
|
|
{
|
|
dbgln("KCOVDevice created");
|
|
}
|
|
|
|
ErrorOr<NonnullRefPtr<OpenFileDescription>> KCOVDevice::open(int options)
|
|
{
|
|
auto& proc = Process::current();
|
|
auto* kcov_instance = proc.kcov_instance();
|
|
if (kcov_instance != nullptr)
|
|
return EBUSY; // This process already open()ed the kcov device
|
|
|
|
proc.set_kcov_instance(new KCOVInstance(proc.pid()));
|
|
return Device::open(options);
|
|
}
|
|
|
|
ErrorOr<void> KCOVDevice::ioctl(OpenFileDescription&, unsigned request, Userspace<void*> arg)
|
|
{
|
|
auto& proc = Process::current();
|
|
auto* thread = Thread::current();
|
|
auto* kcov_instance = proc.kcov_instance();
|
|
if (kcov_instance == nullptr) {
|
|
VERIFY(!Process::is_kcov_busy()); // No thread should be tracing at this point.
|
|
return ENXIO; // This proc hasn't opened the kcov dev yet
|
|
}
|
|
|
|
SpinlockLocker locker(kcov_instance->spinlock());
|
|
switch (request) {
|
|
case KCOV_SETBUFSIZE:
|
|
if (Process::is_kcov_busy())
|
|
// Buffer is shared among all proc threads, hence we need to check if any of them
|
|
// are currently tracing. If so, don't mess with the buffer.
|
|
return EBUSY;
|
|
return kcov_instance->buffer_allocate((FlatPtr)arg.unsafe_userspace_ptr());
|
|
case KCOV_ENABLE:
|
|
if (!kcov_instance->has_buffer())
|
|
return ENOBUFS;
|
|
thread->m_kcov_enabled = true;
|
|
return {};
|
|
case KCOV_DISABLE: {
|
|
thread->m_kcov_enabled = false;
|
|
return {};
|
|
}
|
|
default:
|
|
return EINVAL;
|
|
}
|
|
}
|
|
|
|
ErrorOr<File::VMObjectAndMemoryType> KCOVDevice::vmobject_and_memory_type_for_mmap(OpenFileDescription&, Memory::VirtualRange const&, u64&, bool)
|
|
{
|
|
auto* kcov_instance = Process::current().kcov_instance();
|
|
VERIFY(kcov_instance != nullptr); // Should have happened on fd open()
|
|
|
|
if (!kcov_instance->vmobject())
|
|
return ENOBUFS; // mmaped, before KCOV_SETBUFSIZE
|
|
|
|
return VMObjectAndMemoryType {
|
|
.vmobject = *kcov_instance->vmobject(),
|
|
.memory_type = Memory::MemoryType::Normal,
|
|
};
|
|
}
|
|
|
|
}
|