mirror of
https://github.com/SerenityOS/serenity
synced 2024-07-22 18:46:18 +00:00
fdc0328ce3
Sticking this to the function source has multiple benefits: - We instrument more code, by not excluding entire files. - NO_SANITIZE_COVERAGE can be used in Header files. - Keeping the info with the source code, means if a function or file is moved around, the NO_SANITIZE_COVERAGE moves with it.
155 lines
2.7 KiB
C++
155 lines
2.7 KiB
C++
/*
|
|
* Copyright (c) 2018-2021, Andreas Kling <kling@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/Types.h>
|
|
|
|
#include <AK/Platform.h>
|
|
VALIDATE_IS_X86()
|
|
|
|
namespace Kernel {
|
|
|
|
ALWAYS_INLINE void cli()
|
|
{
|
|
asm volatile("cli" ::
|
|
: "memory");
|
|
}
|
|
ALWAYS_INLINE void sti()
|
|
{
|
|
asm volatile("sti" ::
|
|
: "memory");
|
|
}
|
|
ALWAYS_INLINE NO_SANITIZE_COVERAGE FlatPtr cpu_flags()
|
|
{
|
|
FlatPtr flags;
|
|
asm volatile(
|
|
"pushf\n"
|
|
"pop %0\n"
|
|
: "=rm"(flags)::"memory");
|
|
return flags;
|
|
}
|
|
|
|
template<typename T>
|
|
ALWAYS_INLINE T read_gs_value(FlatPtr offset)
|
|
{
|
|
T val;
|
|
asm volatile(
|
|
"mov %%gs:%a[off], %[val]"
|
|
: [val] "=r"(val)
|
|
: [off] "ir"(offset));
|
|
return val;
|
|
}
|
|
|
|
template<typename T>
|
|
ALWAYS_INLINE void write_gs_value(FlatPtr offset, T val)
|
|
{
|
|
asm volatile(
|
|
"mov %[val], %%gs:%a[off]" ::[off] "ir"(offset), [val] "r"(val)
|
|
: "memory");
|
|
}
|
|
|
|
ALWAYS_INLINE NO_SANITIZE_COVERAGE FlatPtr read_gs_ptr(FlatPtr offset)
|
|
{
|
|
FlatPtr val;
|
|
asm volatile(
|
|
"mov %%gs:%a[off], %[val]"
|
|
: [val] "=r"(val)
|
|
: [off] "ir"(offset));
|
|
return val;
|
|
}
|
|
|
|
ALWAYS_INLINE void write_gs_ptr(u32 offset, FlatPtr val)
|
|
{
|
|
asm volatile(
|
|
"mov %[val], %%gs:%a[off]" ::[off] "ir"(offset), [val] "r"(val)
|
|
: "memory");
|
|
}
|
|
|
|
ALWAYS_INLINE NO_SANITIZE_COVERAGE bool are_interrupts_enabled()
|
|
{
|
|
return (cpu_flags() & 0x200) != 0;
|
|
}
|
|
|
|
FlatPtr read_cr0();
|
|
FlatPtr read_cr2();
|
|
FlatPtr read_cr3();
|
|
FlatPtr read_cr4();
|
|
u64 read_xcr0();
|
|
|
|
void write_cr0(FlatPtr);
|
|
void write_cr3(FlatPtr);
|
|
void write_cr4(FlatPtr);
|
|
void write_xcr0(u64);
|
|
|
|
void flush_idt();
|
|
|
|
ALWAYS_INLINE void load_task_register(u16 selector)
|
|
{
|
|
asm("ltr %0" ::"r"(selector));
|
|
}
|
|
|
|
FlatPtr read_dr0();
|
|
void write_dr0(FlatPtr);
|
|
FlatPtr read_dr1();
|
|
void write_dr1(FlatPtr);
|
|
FlatPtr read_dr2();
|
|
void write_dr2(FlatPtr);
|
|
FlatPtr read_dr3();
|
|
void write_dr3(FlatPtr);
|
|
FlatPtr read_dr6();
|
|
void write_dr6(FlatPtr);
|
|
FlatPtr read_dr7();
|
|
void write_dr7(FlatPtr);
|
|
|
|
ALWAYS_INLINE void read_tsc(u32& lsw, u32& msw)
|
|
{
|
|
asm volatile("rdtsc"
|
|
: "=d"(msw), "=a"(lsw));
|
|
}
|
|
|
|
ALWAYS_INLINE u64 read_tsc()
|
|
{
|
|
u32 lsw;
|
|
u32 msw;
|
|
read_tsc(lsw, msw);
|
|
return ((u64)msw << 32) | lsw;
|
|
}
|
|
|
|
ALWAYS_INLINE u32 read_rdrand()
|
|
{
|
|
u32 value;
|
|
asm volatile(
|
|
"1:\n"
|
|
"rdrand %0\n"
|
|
"jnc 1b\n"
|
|
: "=r"(value)::"cc");
|
|
return value;
|
|
}
|
|
|
|
ALWAYS_INLINE u32 read_rdseed()
|
|
{
|
|
u32 value;
|
|
asm volatile(
|
|
"1:\n"
|
|
"rdseed %0\n"
|
|
"jnc 1b\n"
|
|
: "=r"(value)::"cc");
|
|
return value;
|
|
}
|
|
|
|
void stac();
|
|
void clac();
|
|
|
|
[[noreturn]] ALWAYS_INLINE void halt_this()
|
|
{
|
|
for (;;) {
|
|
asm volatile("cli; hlt");
|
|
}
|
|
}
|
|
|
|
}
|