2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS NT User-Mode DLL
4 * FILE: lib/ntdll/rtl/libsup.c
5 * PURPOSE: RTL Support Routines
6 * PROGRAMMERS: Alex Ionescu (alex@relsoft.net)
10 /* INCLUDES *****************************************************************/
16 SIZE_T RtlpAllocDeallocQueryBufferSize
= PAGE_SIZE
;
17 PTEB LdrpTopLevelDllBeingLoadedTeb
= NULL
;
19 /* FUNCTIONS ***************************************************************/
22 // FIXME: Why "Not implemented"???
28 RtlWalkFrameChain(OUT PVOID
*Callers
,
32 /* Not implemented for user-mode */
39 RtlpCheckForActiveDebugger(VOID
)
41 /* Return the flag in the PEB */
42 return NtCurrentPeb()->BeingDebugged
;
47 RtlpSetInDbgPrint(VOID
)
49 /* Check if it's already set and return TRUE if so */
50 if (NtCurrentTeb()->InDbgPrint
) return TRUE
;
52 /* Set it and return */
53 NtCurrentTeb()->InDbgPrint
= TRUE
;
59 RtlpClearInDbgPrint(VOID
)
62 NtCurrentTeb()->InDbgPrint
= FALSE
;
77 RtlGetCurrentPeb(VOID
)
79 return NtCurrentPeb();
86 RtlAcquirePebLock(VOID
)
88 PPEB Peb
= NtCurrentPeb ();
89 RtlEnterCriticalSection(Peb
->FastPebLock
);
96 RtlReleasePebLock(VOID
)
98 PPEB Peb
= NtCurrentPeb ();
99 RtlLeaveCriticalSection(Peb
->FastPebLock
);
107 RtlGetNtGlobalFlags(VOID
)
109 PPEB pPeb
= NtCurrentPeb();
110 return pPeb
->NtGlobalFlag
;
118 return RtlDeleteCriticalSection(&Lock
->CriticalSection
);
126 return RtlEnterCriticalSection(&Lock
->CriticalSection
);
131 RtlInitializeHeapLock(
134 return RtlInitializeCriticalSection(&Lock
->CriticalSection
);
142 return RtlLeaveCriticalSection(&Lock
->CriticalSection
);
147 RtlpAllocateMemory(UINT Bytes
,
150 UNREFERENCED_PARAMETER(Tag
);
152 return RtlAllocateHeap(RtlGetProcessHeap(),
160 RtlpFreeMemory(PVOID Mem
,
163 UNREFERENCED_PARAMETER(Tag
);
165 RtlFreeHeap(RtlGetProcessHeap(),
173 CHECK_PAGED_CODE_RTL(char *file
, int line
)
175 /* meaningless in user mode */
181 RtlpHandleDpcStackException(IN PEXCEPTION_REGISTRATION_RECORD RegistrationFrame
,
182 IN ULONG_PTR RegistrationFrameEnd
,
183 IN OUT PULONG_PTR StackLow
,
184 IN OUT PULONG_PTR StackHigh
)
186 /* There's no such thing as a DPC stack in user-mode */
192 RtlpCheckLogException(IN PEXCEPTION_RECORD ExceptionRecord
,
193 IN PCONTEXT ContextRecord
,
194 IN PVOID ContextData
,
197 /* Exception logging is not done in user-mode */
202 RtlpCaptureStackLimits(IN ULONG_PTR Ebp
,
203 IN ULONG_PTR
*StackBegin
,
204 IN ULONG_PTR
*StackEnd
)
207 *StackBegin
= (ULONG_PTR
)NtCurrentTeb()->NtTib
.StackLimit
;
208 *StackEnd
= (ULONG_PTR
)NtCurrentTeb()->NtTib
.StackBase
;
216 OUT PULONG_PTR LowLimit
,
217 OUT PULONG_PTR HighLimit
)
219 *LowLimit
= (ULONG_PTR
)NtCurrentTeb()->NtTib
.StackLimit
;
220 *HighLimit
= (ULONG_PTR
)NtCurrentTeb()->NtTib
.StackBase
;
227 RtlIsThreadWithinLoaderCallout(VOID
)
229 return LdrpTopLevelDllBeingLoadedTeb
== NtCurrentTeb();
232 /* RTL Atom Tables ************************************************************/
234 typedef struct _RTL_ATOM_HANDLE
236 RTL_HANDLE_TABLE_ENTRY Handle
;
237 PRTL_ATOM_TABLE_ENTRY AtomEntry
;
238 } RTL_ATOM_HANDLE
, *PRTL_ATOM_HANDLE
;
241 RtlpInitAtomTableLock(PRTL_ATOM_TABLE AtomTable
)
243 RtlInitializeCriticalSection(&AtomTable
->CriticalSection
);
244 return STATUS_SUCCESS
;
249 RtlpDestroyAtomTableLock(PRTL_ATOM_TABLE AtomTable
)
251 RtlDeleteCriticalSection(&AtomTable
->CriticalSection
);
256 RtlpLockAtomTable(PRTL_ATOM_TABLE AtomTable
)
258 RtlEnterCriticalSection(&AtomTable
->CriticalSection
);
264 RtlpUnlockAtomTable(PRTL_ATOM_TABLE AtomTable
)
266 RtlLeaveCriticalSection(&AtomTable
->CriticalSection
);
270 /* handle functions */
273 RtlpCreateAtomHandleTable(PRTL_ATOM_TABLE AtomTable
)
275 RtlInitializeHandleTable(0xCFFF,
276 sizeof(RTL_ATOM_HANDLE
),
277 &AtomTable
->RtlHandleTable
);
283 RtlpDestroyAtomHandleTable(PRTL_ATOM_TABLE AtomTable
)
285 RtlDestroyHandleTable(&AtomTable
->RtlHandleTable
);
289 RtlpAllocAtomTable(ULONG Size
)
291 return (PRTL_ATOM_TABLE
)RtlAllocateHeap(RtlGetProcessHeap(),
297 RtlpFreeAtomTable(PRTL_ATOM_TABLE AtomTable
)
299 RtlFreeHeap(RtlGetProcessHeap(),
304 PRTL_ATOM_TABLE_ENTRY
305 RtlpAllocAtomTableEntry(ULONG Size
)
307 return (PRTL_ATOM_TABLE_ENTRY
)RtlAllocateHeap(RtlGetProcessHeap(),
313 RtlpFreeAtomTableEntry(PRTL_ATOM_TABLE_ENTRY Entry
)
315 RtlFreeHeap(RtlGetProcessHeap(),
321 RtlpFreeAtomHandle(PRTL_ATOM_TABLE AtomTable
, PRTL_ATOM_TABLE_ENTRY Entry
)
323 PRTL_HANDLE_TABLE_ENTRY RtlHandleEntry
;
325 if (RtlIsValidIndexHandle(&AtomTable
->RtlHandleTable
,
326 (ULONG
)Entry
->HandleIndex
,
329 RtlFreeHandle(&AtomTable
->RtlHandleTable
,
335 RtlpCreateAtomHandle(PRTL_ATOM_TABLE AtomTable
, PRTL_ATOM_TABLE_ENTRY Entry
)
338 PRTL_HANDLE_TABLE_ENTRY RtlHandle
;
340 RtlHandle
= RtlAllocateHandle(&AtomTable
->RtlHandleTable
,
342 if (RtlHandle
!= NULL
)
344 PRTL_ATOM_HANDLE AtomHandle
= (PRTL_ATOM_HANDLE
)RtlHandle
;
346 /* FIXME - Handle Indexes >= 0xC000 ?! */
347 if (HandleIndex
< 0xC000)
349 Entry
->HandleIndex
= (USHORT
)HandleIndex
;
350 Entry
->Atom
= 0xC000 + (USHORT
)HandleIndex
;
352 AtomHandle
->AtomEntry
= Entry
;
353 AtomHandle
->Handle
.Flags
= RTL_HANDLE_VALID
;
359 /* set the valid flag, otherwise RtlFreeHandle will fail! */
360 AtomHandle
->Handle
.Flags
= RTL_HANDLE_VALID
;
362 RtlFreeHandle(&AtomTable
->RtlHandleTable
,
370 PRTL_ATOM_TABLE_ENTRY
371 RtlpGetAtomEntry(PRTL_ATOM_TABLE AtomTable
, ULONG Index
)
373 PRTL_HANDLE_TABLE_ENTRY RtlHandle
;
375 if (RtlIsValidIndexHandle(&AtomTable
->RtlHandleTable
,
379 PRTL_ATOM_HANDLE AtomHandle
= (PRTL_ATOM_HANDLE
)RtlHandle
;
381 return AtomHandle
->AtomEntry
;
389 * Ldr Resource support code
392 IMAGE_RESOURCE_DIRECTORY
*find_entry_by_name( IMAGE_RESOURCE_DIRECTORY
*dir
,
393 LPCWSTR name
, void *root
,
395 IMAGE_RESOURCE_DIRECTORY
*find_entry_by_id( IMAGE_RESOURCE_DIRECTORY
*dir
,
396 WORD id
, void *root
, int want_dir
);
397 IMAGE_RESOURCE_DIRECTORY
*find_first_entry( IMAGE_RESOURCE_DIRECTORY
*dir
,
398 void *root
, int want_dir
);
399 int push_language( USHORT
*list
, ULONG pos
, WORD lang
);
401 /**********************************************************************
404 * Find a resource entry
406 NTSTATUS
find_entry( PVOID BaseAddress
, LDR_RESOURCE_INFO
*info
,
407 ULONG level
, void **ret
, int want_dir
)
411 IMAGE_RESOURCE_DIRECTORY
*resdirptr
;
412 USHORT list
[9]; /* list of languages to try */
414 LCID user_lcid
, system_lcid
;
416 root
= RtlImageDirectoryEntryToData( BaseAddress
, TRUE
, IMAGE_DIRECTORY_ENTRY_RESOURCE
, &size
);
417 if (!root
) return STATUS_RESOURCE_DATA_NOT_FOUND
;
420 if (!level
--) goto done
;
421 if (!(*ret
= find_entry_by_name( resdirptr
, (LPCWSTR
)info
->Type
, root
, want_dir
|| level
)))
422 return STATUS_RESOURCE_TYPE_NOT_FOUND
;
423 if (!level
--) return STATUS_SUCCESS
;
426 if (!(*ret
= find_entry_by_name( resdirptr
, (LPCWSTR
)info
->Name
, root
, want_dir
|| level
)))
427 return STATUS_RESOURCE_NAME_NOT_FOUND
;
428 if (!level
--) return STATUS_SUCCESS
;
429 if (level
) return STATUS_INVALID_PARAMETER
; /* level > 3 */
431 /* 1. specified language */
432 pos
= push_language( list
, pos
, info
->Language
);
434 /* 2. specified language with neutral sublanguage */
435 pos
= push_language( list
, pos
, MAKELANGID( PRIMARYLANGID(info
->Language
), SUBLANG_NEUTRAL
) );
437 /* 3. neutral language with neutral sublanguage */
438 pos
= push_language( list
, pos
, MAKELANGID( LANG_NEUTRAL
, SUBLANG_NEUTRAL
) );
440 /* if no explicitly specified language, try some defaults */
441 if (PRIMARYLANGID(info
->Language
) == LANG_NEUTRAL
)
443 /* user defaults, unless SYS_DEFAULT sublanguage specified */
444 if (SUBLANGID(info
->Language
) != SUBLANG_SYS_DEFAULT
)
446 /* 4. current thread locale language */
447 pos
= push_language( list
, pos
, LANGIDFROMLCID(NtCurrentTeb()->CurrentLocale
) );
449 if (NT_SUCCESS(NtQueryDefaultLocale(TRUE
, &user_lcid
)))
451 /* 5. user locale language */
452 pos
= push_language( list
, pos
, LANGIDFROMLCID(user_lcid
) );
454 /* 6. user locale language with neutral sublanguage */
455 pos
= push_language( list
, pos
, MAKELANGID( PRIMARYLANGID(user_lcid
), SUBLANG_NEUTRAL
) );
459 /* now system defaults */
461 if (NT_SUCCESS(NtQueryDefaultLocale(FALSE
, &system_lcid
)))
463 /* 7. system locale language */
464 pos
= push_language( list
, pos
, LANGIDFROMLCID( system_lcid
) );
466 /* 8. system locale language with neutral sublanguage */
467 pos
= push_language( list
, pos
, MAKELANGID( PRIMARYLANGID(system_lcid
), SUBLANG_NEUTRAL
) );
471 pos
= push_language( list
, pos
, MAKELANGID( LANG_ENGLISH
, SUBLANG_DEFAULT
) );
475 for (i
= 0; i
< pos
; i
++)
476 if ((*ret
= find_entry_by_id( resdirptr
, list
[i
], root
, want_dir
))) return STATUS_SUCCESS
;
478 /* if no explicitly specified language, return the first entry */
479 if (PRIMARYLANGID(info
->Language
) == LANG_NEUTRAL
)
481 if ((*ret
= find_first_entry( resdirptr
, root
, want_dir
))) return STATUS_SUCCESS
;
483 return STATUS_RESOURCE_LANG_NOT_FOUND
;
487 return STATUS_SUCCESS
;
494 RtlPcToFileHeader(IN PVOID PcValue
,
497 PLIST_ENTRY ModuleListHead
;
499 PLDR_DATA_TABLE_ENTRY Module
;
500 PVOID ImageBase
= NULL
;
502 RtlEnterCriticalSection (NtCurrentPeb()->LoaderLock
);
503 ModuleListHead
= &NtCurrentPeb()->Ldr
->InLoadOrderModuleList
;
504 Entry
= ModuleListHead
->Flink
;
505 while (Entry
!= ModuleListHead
)
507 Module
= CONTAINING_RECORD(Entry
, LDR_DATA_TABLE_ENTRY
, InLoadOrderLinks
);
509 if ((ULONG_PTR
)PcValue
>= (ULONG_PTR
)Module
->DllBase
&&
510 (ULONG_PTR
)PcValue
< (ULONG_PTR
)Module
->DllBase
+ Module
->SizeOfImage
)
512 ImageBase
= Module
->DllBase
;
515 Entry
= Entry
->Flink
;
517 RtlLeaveCriticalSection (NtCurrentPeb()->LoaderLock
);
519 *BaseOfImage
= ImageBase
;
526 RtlDosApplyFileIsolationRedirection_Ustr(
528 IN PUNICODE_STRING OriginalName
,
529 IN PUNICODE_STRING Extension
,
530 IN OUT PUNICODE_STRING RedirectedName
,
531 IN OUT PUNICODE_STRING RedirectedName2
,
532 IN OUT PUNICODE_STRING
*OriginalName2
,
538 return STATUS_SXS_KEY_NOT_FOUND
;