diff --git a/examples/ZydisWinKernel.c b/examples/ZydisWinKernel.c index 6c0e2b2..c50ee01 100644 --- a/examples/ZydisWinKernel.c +++ b/examples/ZydisWinKernel.c @@ -58,10 +58,13 @@ RtlImageNtHeader( _In_ PVOID ImageBase ); +#if defined(ZYDIS_CLANG) || defined(ZYDIS_GNUC) +__attribute__((section("INIT"))) +#endif DRIVER_INITIALIZE DriverEntry; -#ifdef ALLOC_PRAGMA +#if defined(ALLOC_PRAGMA) && !(defined(ZYDIS_CLANG) || defined(ZYDIS_GNUC)) #pragma alloc_text(INIT, DriverEntry) #endif