This commit is contained in:
Anurag Saxena 2022-07-26 00:39:16 -07:00
Родитель 7e79b67fa8
Коммит 21debced20
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: FDFD0EEA29F8CA8A
2 изменённых файлов: 5 добавлений и 5 удалений

Просмотреть файл

@ -87,7 +87,7 @@ _clean_up_thread_pool()
class _ebpf_emulated_dpc;
thread_local static bool _ebpf_non_preemptible = false;
bool _ebpf_non_preemptible = false;
typedef struct _ebpf_non_preemptible_work_item
{

Просмотреть файл

@ -19,7 +19,7 @@ using namespace std::chrono_literals;
#include "test_helper.hpp"
extern "C" bool ebpf_fuzzing_enabled;
extern bool _ebpf_platform_is_preemptible;
extern bool _ebpf_non_preemptible;
static bool _is_platform_preemptible = false;
@ -243,10 +243,10 @@ _unload_all_native_modules()
static void
_preprocess_load_native_module(_Inout_ service_context_t* context)
{
// Every time a native module is loaded, flip the bit for _ebpf_platform_is_preemptible.
// Every time a native module is loaded, flip the bit for _ebpf_non_preemptible.
// This ensures both the code paths are executed in the native module code, when the
// test cases are executed.
_ebpf_platform_is_preemptible = _is_platform_preemptible;
_ebpf_non_preemptible = _is_platform_preemptible;
_is_platform_preemptible = !_is_platform_preemptible;
context->dll = LoadLibraryW(context->file_path.c_str());
@ -566,7 +566,7 @@ _test_helper_end_to_end::~_test_helper_end_to_end()
_expect_native_module_load_failures = false;
// Change back to original value.
_ebpf_platform_is_preemptible = true;
_ebpf_non_preemptible = false;
set_verification_in_progress(false);
}