[SDK][NTOSKRNL] Simple fixes

Author: DarkFire01

ntoskrnl/include/ntoskrnl.h

@@ -17,6 +17,12 @@
 #endif
 
 /* WDK hacks */
+#ifdef _M_ARM64
+#define IoAllocateAdapterChannel _IoAllocateAdapterChannel
+#define KeGetCurrentThread _KeGetCurrentThread
+#define RtlFillMemoryUlong _RtlFillMemoryUlong
+#endif
+
 #ifdef _M_AMD64
 #define IoAllocateAdapterChannel _IoAllocateAdapterChannel
 #define KeGetCurrentThread _KeGetCurrentThread

sdk/include/ndk/psfuncs.h

@@ -427,7 +427,7 @@ FORCEINLINE struct _TEB * NtCurrentTeb(VOID)
     // return (struct _TEB *)KeGetPcr()->Used_Self;
     return (struct _TEB *)(ULONG_PTR)_MoveFromCoprocessor(CP15_TPIDRURW);
 #elif defined (_M_ARM64)
-    return (struct _TEB *)__getReg(18);
+    return (struct _TEB *)__readx18qword(FIELD_OFFSET(NT_TIB, Self));
 #else
 #error Unsupported architecture
 #endif

sdk/include/xdk/winnt_old.h

@@ -4612,12 +4612,15 @@ FORCEINLINE PVOID GetCurrentFiber(VOID)
 #elif defined (_M_ARM64)
 FORCEINLINE struct _TEB * NtCurrentTeb(VOID)
 {
-    return (struct _TEB *)__getReg(18);
+  return (struct _TEB *)__readx18qword(FIELD_OFFSET(NT_TIB, Self));
 }
 FORCEINLINE PVOID GetCurrentFiber(VOID)
 {
-    //UNIMPLEMENTED;
-    return 0;
+#ifdef NONAMELESSUNION
+    return (PVOID)__readgsqword(FIELD_OFFSET(NT_TIB, DUMMYUNIONNAME.FiberData));
+#else
+    return (PVOID)__readgsqword(FIELD_OFFSET(NT_TIB, FiberData));
+#endif
 }
 #elif defined(_M_PPC)
 FORCEINLINE unsigned long _read_teb_dword(const unsigned long Offset)

sdk/lib/rtl/exception.c

@@ -22,7 +22,7 @@ PRTLP_UNHANDLED_EXCEPTION_FILTER RtlpUnhandledExceptionFilter;
 
 /* FUNCTIONS ***************************************************************/
 
-#if !defined(_M_IX86) && !defined(_M_AMD64)
+#if !defined(_M_IX86) && !defined(_M_AMD64) && !defined(_M_ARM64)
 
 /*
  * @implemented

sdk/lib/rtl/slist.c

@@ -208,7 +208,7 @@ RtlInterlockedPushListSList(
 }
 
 
-#if !defined(_M_IX86) && !defined(_M_AMD64)
+#if !defined(_M_IX86) && !defined(_M_AMD64) && !defined(_M_ARM64)
 
 _WARN("C based S-List functions can bugcheck, if not handled properly in kernel")