RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-05-30 21:58:10 +00:00
Code Issues Activity Actions

kcov-example: Port to LibMain

Browse source
This commit is contained in:
Kenneth Myhra 2022-03-28 22:12:28 +02:00 • committed by Brian Gianforcaro
parent 234025ee53
commit e302fac34b
2 changed files with 13 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

1
Userland/Utilities/CMakeLists.txt
View file

@ -124,6 +124,7 @@ target_link_libraries(install-bin LibMain)
target_link_libraries(jp LibMain) target_link_libraries(jp LibMain)
target_link_libraries(js LibJS LibLine LibMain) target_link_libraries(js LibJS LibLine LibMain)
link_with_unicode_data(js) link_with_unicode_data(js)
target_link_libraries(kcov-example LibMain)
target_link_libraries(keymap LibKeyboard LibMain) target_link_libraries(keymap LibKeyboard LibMain)
target_link_libraries(kill LibMain) target_link_libraries(kill LibMain)
target_link_libraries(killall LibCore LibMain) target_link_libraries(killall LibCore LibMain)

42
Userland/Utilities/kcov-example.cpp
View file

@ -4,6 +4,8 @@
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
*/ */
#include <LibCore/System.h>
#include <LibMain/Main.h>
#include <fcntl.h> #include <fcntl.h>
#include <stdio.h> #include <stdio.h>
#include <sys/ioctl.h> #include <sys/ioctl.h>
@ -14,47 +16,27 @@
// Note: This program requires serenity to be built with the CMake build option // Note: This program requires serenity to be built with the CMake build option
// ENABLE_KERNEL_COVERAGE_COLLECTION // ENABLE_KERNEL_COVERAGE_COLLECTION
int main(void) ErrorOr<int> serenity_main(Main::Arguments)
{ {
constexpr size_t num_entries = 1024 * 100; constexpr size_t num_entries = 1024 * 100;
int fd = open("/dev/kcov0", O_RDWR); int fd = TRY(Core::System::open("/dev/kcov0", O_RDWR));
if (fd == -1) { TRY(Core::System::ioctl(fd, KCOV_SETBUFSIZE, num_entries));
perror("open"); kcov_pc_t* cover = (kcov_pc_t*)TRY(Core::System::mmap(NULL, num_entries * KCOV_ENTRY_SIZE, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0));
fprintf(stderr, "Could not open /dev/kcov0 - is ENABLE_KERNEL_COVERAGE_COLLECTION enabled?\n"); TRY(Core::System::ioctl(fd, KCOV_ENABLE));
return 1;
}
if (ioctl(fd, KCOV_SETBUFSIZE, num_entries) == -1) {
perror("ioctl: KCOV_SETBUFSIZE");
return 1;
}
kcov_pc_t* cover = (kcov_pc_t*)mmap(NULL, num_entries * KCOV_ENTRY_SIZE,
PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
if (cover == MAP_FAILED) {
perror("mmap");
return 1;
}
if (ioctl(fd, KCOV_ENABLE) == -1) {
perror("ioctl: KCOV_ENABLE");
return 1;
}
cover[0] = 0; cover[0] = 0;
// Example syscall so we actually cover some kernel code. // Example syscall so we actually cover some kernel code.
getppid(); getppid();
if (ioctl(fd, KCOV_DISABLE) == -1) { TRY(Core::System::ioctl(fd, KCOV_DISABLE));
perror("ioctl: KCOV_DISABLE");
return 1;
}
u64 cov_idx = cover[0]; u64 cov_idx = cover[0];
for (size_t idx = 1; idx <= cov_idx; idx++) for (size_t idx = 1; idx <= cov_idx; idx++)
printf("%p\n", (void*)cover[idx]); printf("%p\n", (void*)cover[idx]);
if (munmap(const_cast<u64*>(cover), num_entries * KCOV_ENTRY_SIZE) == -1) {
perror("munmap"); TRY(Core::System::munmap(const_cast<u64*>(cover), num_entries * KCOV_ENTRY_SIZE));
return 1; TRY(Core::System::close(fd));
}
close(fd);
return 0; return 0;
} }