1 #ifndef __INCLUDE_INTERNAL_NTOSKRNL_H
2 #define __INCLUDE_INTERNAL_NTOSKRNL_H
5 * Use these to place a function in a specific section of the executable
7 #define PLACE_IN_SECTION(s) __attribute__((section (s)))
9 #define INIT_FUNCTION PLACE_IN_SECTION("init")
10 #define PAGE_LOCKED_FUNCTION PLACE_IN_SECTION("pagelk")
11 #define PAGE_UNLOCKED_FUNCTION PLACE_IN_SECTION("pagepo")
14 #define PAGE_LOCKED_FUNCTION
15 #define PAGE_UNLOCKED_FUNCTION
20 #define KeGetCurrentThread _KeGetCurrentThread
21 #define KeGetPreviousMode _KeGetPreviousMode
22 #undef PsGetCurrentProcess
23 #define PsGetCurrentProcess _PsGetCurrentProcess
25 #include "arch/intrin_i.h"
29 #include "i386/v86m.h"
48 #include "../kdbg/kdb.h"
59 * Defines a descriptor as it appears in the processor tables
61 typedef struct __DESCRIPTOR
65 } IDT_DESCRIPTOR
, GDT_DESCRIPTOR
;
68 //extern GDT_DESCRIPTOR KiGdt[256];
71 * Initalization functions (called once by main())
73 BOOLEAN NTAPI
ObInit(VOID
);
74 BOOLEAN NTAPI
CmInitSystem1(VOID
);
75 VOID
CmShutdownRegistry(VOID
);
76 //BOOLEAN CmImportSystemHive(PCHAR ChunkBase, ULONG ChunkSize);
77 //BOOLEAN CmImportHardwareHive(PCHAR ChunkBase, ULONG ChunkSize);
78 BOOLEAN NTAPI
KdInitSystem(ULONG Reserved
, PLOADER_PARAMETER_BLOCK LoaderBlock
);
80 /* FIXME - RtlpCreateUnicodeString is obsolete and should be removed ASAP! */
82 RtlpCreateUnicodeString(
83 IN OUT PUNICODE_STRING UniDest
,
85 IN POOL_TYPE PoolType
);
89 RtlpLogException(IN PEXCEPTION_RECORD ExceptionRecord
,
90 IN PCONTEXT ContextRecord
,
94 /* FIXME: Interlocked functions that need to be made into a public header */
98 InterlockedAnd(IN OUT LONG
volatile *Target
,
107 j
= InterlockedCompareExchange((PLONG
)Target
,
118 InterlockedOr(IN OUT LONG
volatile *Target
,
127 j
= InterlockedCompareExchange((PLONG
)Target
,
138 * generic information class probing code
141 #define ICIF_QUERY 0x1
143 #define ICIF_QUERY_SIZE_VARIABLE 0x4
144 #define ICIF_SET_SIZE_VARIABLE 0x8
145 #define ICIF_SIZE_VARIABLE (ICIF_QUERY_SIZE_VARIABLE | ICIF_SET_SIZE_VARIABLE)
147 typedef struct _INFORMATION_CLASS_INFO
149 ULONG RequiredSizeQUERY
;
150 ULONG RequiredSizeSET
;
152 ULONG AlignmentQUERY
;
154 } INFORMATION_CLASS_INFO
, *PINFORMATION_CLASS_INFO
;
156 #define ICI_SQ_SAME(Type, Alignment, Flags) \
157 { Type, Type, Alignment, Alignment, Flags }
159 #define ICI_SQ(TypeQuery, TypeSet, AlignmentQuery, AlignmentSet, Flags) \
160 { TypeQuery, TypeSet, AlignmentQuery, AlignmentSet, Flags }
165 #define IQS_SAME(Type, Alignment, Flags) \
166 { sizeof(Type), sizeof(Type), sizeof(Alignment), sizeof(Alignment), Flags }
168 #define IQS(TypeQuery, TypeSet, AlignmentQuery, AlignmentSet, Flags) \
169 { sizeof(TypeQuery), sizeof(TypeSet), sizeof(AlignmentQuery), sizeof(AlignmentSet), Flags }
173 DefaultSetInfoBufferCheck(ULONG Class
,
174 const INFORMATION_CLASS_INFO
*ClassList
,
175 ULONG ClassListEntries
,
178 KPROCESSOR_MODE PreviousMode
)
180 NTSTATUS Status
= STATUS_SUCCESS
;
182 if (Class
< ClassListEntries
)
184 if (!(ClassList
[Class
].Flags
& ICIF_SET
))
186 Status
= STATUS_INVALID_INFO_CLASS
;
188 else if (ClassList
[Class
].RequiredSizeSET
> 0 &&
189 BufferLength
!= ClassList
[Class
].RequiredSizeSET
)
191 if (!(ClassList
[Class
].Flags
& ICIF_SET_SIZE_VARIABLE
))
193 Status
= STATUS_INFO_LENGTH_MISMATCH
;
197 if (NT_SUCCESS(Status
))
199 if (PreviousMode
!= KernelMode
)
205 ClassList
[Class
].AlignmentSET
);
209 Status
= _SEH_GetExceptionCode();
216 Status
= STATUS_INVALID_INFO_CLASS
;
223 DefaultQueryInfoBufferCheck(ULONG Class
,
224 const INFORMATION_CLASS_INFO
*ClassList
,
225 ULONG ClassListEntries
,
229 KPROCESSOR_MODE PreviousMode
)
231 NTSTATUS Status
= STATUS_SUCCESS
;
233 if (Class
< ClassListEntries
)
235 if (!(ClassList
[Class
].Flags
& ICIF_QUERY
))
237 Status
= STATUS_INVALID_INFO_CLASS
;
239 else if (ClassList
[Class
].RequiredSizeQUERY
> 0 &&
240 BufferLength
!= ClassList
[Class
].RequiredSizeQUERY
)
242 if (!(ClassList
[Class
].Flags
& ICIF_QUERY_SIZE_VARIABLE
))
244 Status
= STATUS_INFO_LENGTH_MISMATCH
;
248 if (NT_SUCCESS(Status
))
250 if (PreviousMode
!= KernelMode
)
256 ProbeForWrite(Buffer
,
258 ClassList
[Class
].AlignmentQUERY
);
261 if (ReturnLength
!= NULL
)
263 ProbeForWriteUlong(ReturnLength
);
268 Status
= _SEH_GetExceptionCode();
275 Status
= STATUS_INVALID_INFO_CLASS
;
281 * Use IsPointerOffset to test whether a pointer should be interpreted as an offset
284 #if defined(_X86_) || defined(_M_AMD64) || defined(_MIPS_) || defined(_PPC_)
286 /* for x86 and x86-64 the MSB is 1 so we can simply test on that */
287 #define IsPointerOffset(Ptr) ((LONG_PTR)(Ptr) >= 0)
289 #elif defined(_IA64_)
291 /* on Itanium if the 24 most significant bits are set, we're not dealing with
293 #define IsPointerOffset(Ptr) (((ULONG_PTR)(Ptr) & 0xFFFFFF0000000000ULL) == 0)
296 #error IsPointerOffset() needs to be defined for this architecture
301 C_ASSERT(FIELD_OFFSET(KUSER_SHARED_DATA
, SystemCall
) == 0x300);
302 C_ASSERT(FIELD_OFFSET(KTHREAD
, InitialStack
) == KTHREAD_INITIAL_STACK
);
303 C_ASSERT(FIELD_OFFSET(KTHREAD
, Teb
) == KTHREAD_TEB
);
304 C_ASSERT(FIELD_OFFSET(KTHREAD
, KernelStack
) == KTHREAD_KERNEL_STACK
);
305 C_ASSERT(FIELD_OFFSET(KTHREAD
, NpxState
) == KTHREAD_NPX_STATE
);
306 C_ASSERT(FIELD_OFFSET(KTHREAD
, ServiceTable
) == KTHREAD_SERVICE_TABLE
);
307 C_ASSERT(FIELD_OFFSET(KTHREAD
, PreviousMode
) == KTHREAD_PREVIOUS_MODE
);
308 C_ASSERT(FIELD_OFFSET(KTHREAD
, TrapFrame
) == KTHREAD_TRAP_FRAME
);
309 C_ASSERT(FIELD_OFFSET(KTHREAD
, CallbackStack
) == KTHREAD_CALLBACK_STACK
);
310 C_ASSERT(FIELD_OFFSET(KTHREAD
, ApcState
.Process
) == KTHREAD_APCSTATE_PROCESS
);
311 C_ASSERT(FIELD_OFFSET(KPROCESS
, DirectoryTableBase
) == KPROCESS_DIRECTORY_TABLE_BASE
);
312 //C_ASSERT(FIELD_OFFSET(KPCR, Tib.ExceptionList) == KPCR_EXCEPTION_LIST);
313 //C_ASSERT(FIELD_OFFSET(KPCR, Self) == KPCR_SELF);
314 C_ASSERT(FIELD_OFFSET(KPCR
, IRR
) == KPCR_IRR
);
315 C_ASSERT(FIELD_OFFSET(KPCR
, IDR
) == KPCR_IDR
);
316 C_ASSERT(FIELD_OFFSET(KPCR
, Irql
) == KPCR_IRQL
);
318 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, CurrentThread
) == KPCR_CURRENT_THREAD
);
319 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, NextThread
) == KPCR_PRCB_NEXT_THREAD
);
320 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, NpxThread
) == KPCR_NPX_THREAD
);
321 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) == KPCR_PRCB_DATA
);
322 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, KeSystemCalls
) == KPCR_SYSTEM_CALLS
);
323 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, DpcData
) + /*FIELD_OFFSET(KDPC_DATA, DpcQueuDepth)*/12 == KPCR_PRCB_DPC_QUEUE_DEPTH
);
324 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, DpcData
) + 16 == KPCR_PRCB_DPC_COUNT
);
325 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, DpcStack
) == KPCR_PRCB_DPC_STACK
);
326 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, TimerRequest
) == KPCR_PRCB_TIMER_REQUEST
);
327 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, MaximumDpcQueueDepth
) == KPCR_PRCB_MAXIMUM_DPC_QUEUE_DEPTH
);
328 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, DpcRequestRate
) == KPCR_PRCB_DPC_REQUEST_RATE
);
329 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, DpcInterruptRequested
) == KPCR_PRCB_DPC_INTERRUPT_REQUESTED
);
330 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, DpcRoutineActive
) == KPCR_PRCB_DPC_ROUTINE_ACTIVE
);
331 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, DpcLastCount
) == KPCR_PRCB_DPC_LAST_COUNT
);
332 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, TimerRequest
) == KPCR_PRCB_TIMER_REQUEST
);
333 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, QuantumEnd
) == KPCR_PRCB_QUANTUM_END
);
334 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, DeferredReadyListHead
) == KPCR_PRCB_DEFERRED_READY_LIST_HEAD
);
335 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, PowerState
) == KPCR_PRCB_POWER_STATE_IDLE_FUNCTION
);
336 //C_ASSERT(FIELD_OFFSET(KIPCR, PrcbData) + FIELD_OFFSET(KPRCB, PrcbLock) == KPCR_PRCB_PRCB_LOCK);
337 C_ASSERT(FIELD_OFFSET(KIPCR
, PrcbData
) + FIELD_OFFSET(KPRCB
, DpcStack
) == KPCR_PRCB_DPC_STACK
);
338 C_ASSERT(sizeof(FX_SAVE_AREA
) == SIZEOF_FX_SAVE_AREA
);
340 /* Platform specific checks */
341 C_ASSERT(FIELD_OFFSET(KPROCESS
, IopmOffset
) == KPROCESS_IOPM_OFFSET
);
342 C_ASSERT(FIELD_OFFSET(KPROCESS
, LdtDescriptor
) == KPROCESS_LDT_DESCRIPTOR0
);
343 C_ASSERT(FIELD_OFFSET(KV86M_TRAP_FRAME
, SavedExceptionStack
) == TF_SAVED_EXCEPTION_STACK
);
344 C_ASSERT(FIELD_OFFSET(KV86M_TRAP_FRAME
, regs
) == TF_REGS
);
345 C_ASSERT(FIELD_OFFSET(KV86M_TRAP_FRAME
, orig_ebp
) == TF_ORIG_EBP
);
346 C_ASSERT(FIELD_OFFSET(KTSS
, Esp0
) == KTSS_ESP0
);
347 C_ASSERT(FIELD_OFFSET(KTSS
, IoMapBase
) == KTSS_IOMAPBASE
);
350 #endif /* INCLUDE_INTERNAL_NTOSKRNL_H */