libsupp.c File Reference

#include <ntdll.h>
#include <debug.h>

Include dependency graph for libsupp.c:

Go to the source code of this file.

Classes
struct	_RTL_ATOM_HANDLE

Macros
#define	NDEBUG

Typedefs
typedef struct _RTL_ATOM_HANDLE	RTL_ATOM_HANDLE
typedef struct _RTL_ATOM_HANDLE *	PRTL_ATOM_HANDLE

Functions
BOOLEAN NTAPI	RtlpCheckForActiveDebugger (VOID)
BOOLEAN NTAPI	RtlpSetInDbgPrint (VOID)
VOID NTAPI	RtlpClearInDbgPrint (VOID)
KPROCESSOR_MODE NTAPI	RtlpGetMode (VOID)
PPEB NTAPI	RtlGetCurrentPeb (VOID)
VOID NTAPI	RtlAcquirePebLock (VOID)
VOID NTAPI	RtlReleasePebLock (VOID)
ULONG NTAPI	RtlGetNtGlobalFlags (VOID)
NTSTATUS NTAPI	RtlDeleteHeapLock (IN OUT PHEAP_LOCK Lock)
NTSTATUS NTAPI	RtlEnterHeapLock (IN OUT PHEAP_LOCK Lock, IN BOOLEAN Exclusive)
BOOLEAN NTAPI	RtlTryEnterHeapLock (IN OUT PHEAP_LOCK Lock, IN BOOLEAN Exclusive)
NTSTATUS NTAPI	RtlInitializeHeapLock (IN OUT PHEAP_LOCK *Lock)
NTSTATUS NTAPI	RtlLeaveHeapLock (IN OUT PHEAP_LOCK Lock)
PVOID NTAPI	RtlpAllocateMemory (UINT Bytes, ULONG Tag)
VOID NTAPI	RtlpFreeMemory (PVOID Mem, ULONG Tag)
VOID NTAPI	RtlpSetHeapParameters (IN PRTL_HEAP_PARAMETERS Parameters)
BOOLEAN NTAPI	RtlpHandleDpcStackException (IN PEXCEPTION_REGISTRATION_RECORD RegistrationFrame, IN ULONG_PTR RegistrationFrameEnd, IN OUT PULONG_PTR StackLow, IN OUT PULONG_PTR StackHigh)
VOID NTAPI	RtlpCheckLogException (IN PEXCEPTION_RECORD ExceptionRecord, IN PCONTEXT ContextRecord, IN PVOID ContextData, IN ULONG Size)
BOOLEAN NTAPI	RtlpCaptureStackLimits (IN ULONG_PTR Ebp, IN ULONG_PTR *StackBegin, IN ULONG_PTR *StackEnd)
ULONG NTAPI	RtlWalkFrameChain (OUT PVOID *Callers, IN ULONG Count, IN ULONG Flags)
BOOLEAN NTAPI	RtlIsThreadWithinLoaderCallout (VOID)
NTSTATUS	RtlpInitAtomTableLock (PRTL_ATOM_TABLE AtomTable)
VOID	RtlpDestroyAtomTableLock (PRTL_ATOM_TABLE AtomTable)
BOOLEAN	RtlpLockAtomTable (PRTL_ATOM_TABLE AtomTable)
VOID	RtlpUnlockAtomTable (PRTL_ATOM_TABLE AtomTable)
BOOLEAN	RtlpCreateAtomHandleTable (PRTL_ATOM_TABLE AtomTable)
VOID	RtlpDestroyAtomHandleTable (PRTL_ATOM_TABLE AtomTable)
PRTL_ATOM_TABLE	RtlpAllocAtomTable (ULONG Size)
VOID	RtlpFreeAtomTable (PRTL_ATOM_TABLE AtomTable)
PRTL_ATOM_TABLE_ENTRY	RtlpAllocAtomTableEntry (ULONG Size)
VOID	RtlpFreeAtomTableEntry (PRTL_ATOM_TABLE_ENTRY Entry)
VOID	RtlpFreeAtomHandle (PRTL_ATOM_TABLE AtomTable, PRTL_ATOM_TABLE_ENTRY Entry)
BOOLEAN	RtlpCreateAtomHandle (PRTL_ATOM_TABLE AtomTable, PRTL_ATOM_TABLE_ENTRY Entry)
PRTL_ATOM_TABLE_ENTRY	RtlpGetAtomEntry (PRTL_ATOM_TABLE AtomTable, ULONG Index)
NTSTATUS NTAPI	RtlpImageNtHeaderEx (_In_ ULONG Flags, _In_ PVOID Base, _In_ ULONG64 Size, _Out_ PIMAGE_NT_HEADERS *OutHeaders)
NTSTATUS NTAPI	RtlImageNtHeaderEx (_In_ ULONG Flags, _In_ PVOID Base, _In_ ULONG64 Size, _Out_ PIMAGE_NT_HEADERS *OutHeaders)
IMAGE_RESOURCE_DIRECTORY *	find_entry_by_name (IMAGE_RESOURCE_DIRECTORY *dir, LPCWSTR name, void *root, int want_dir)
IMAGE_RESOURCE_DIRECTORY *	find_entry_by_id (IMAGE_RESOURCE_DIRECTORY *dir, WORD id, void *root, int want_dir)
IMAGE_RESOURCE_DIRECTORY *	find_first_entry (IMAGE_RESOURCE_DIRECTORY *dir, void *root, int want_dir)
int	push_language (USHORT *list, ULONG pos, WORD lang)
NTSTATUS	find_entry (PVOID BaseAddress, LDR_RESOURCE_INFO *info, ULONG level, void **ret, int want_dir)
PVOID NTAPI	RtlPcToFileHeader (IN PVOID PcValue, PVOID *BaseOfImage)
NTSTATUS	get_buffer (LPWSTR *buffer, SIZE_T needed, PUNICODE_STRING CallerBuffer, BOOLEAN bAllocateBuffer)
NTSTATUS	find_actctx_dll (PUNICODE_STRING pnameW, LPWSTR *fullname, PUNICODE_STRING CallerBuffer, BOOLEAN bAllocateBuffer)
NTSYSAPI NTSTATUS NTAPI	RtlDosApplyFileIsolationRedirection_Ustr (IN ULONG Flags, IN PUNICODE_STRING OriginalName, IN PUNICODE_STRING Extension, IN OUT PUNICODE_STRING StaticString, IN OUT PUNICODE_STRING DynamicString, IN OUT PUNICODE_STRING *NewName, IN PULONG NewFlags, IN PSIZE_T FileNameSize, IN PSIZE_T RequiredLength)
NTSYSAPI NTSTATUS NTAPI	RtlWow64EnableFsRedirection (IN BOOLEAN Wow64FsEnableRedirection)
NTSYSAPI NTSTATUS NTAPI	RtlWow64EnableFsRedirectionEx (IN PVOID Wow64FsEnableRedirection, OUT PVOID *OldFsRedirectionLevel)
NTSYSAPI NTSTATUS NTAPI	RtlComputeImportTableHash (IN HANDLE FileHandle, OUT PCHAR Hash, IN ULONG ImportTableHashSize)
NTSTATUS NTAPI	RtlpSafeCopyMemory (_Out_writes_bytes_all_(Length) VOID UNALIGNED *Destination, _In_reads_bytes_(Length) CONST VOID UNALIGNED *Source, _In_ SIZE_T Length)
ULONG NTAPI	RtlGetTickCount (VOID)

Variables
SIZE_T	RtlpAllocDeallocQueryBufferSize = PAGE_SIZE
PTEB	LdrpTopLevelDllBeingLoadedTeb = NULL
PVOID	MmHighestUserAddress = (PVOID)MI_HIGHEST_USER_ADDRESS

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 14 of file libsupp.c.

Typedef Documentation

PRTL_ATOM_HANDLE

typedef struct _RTL_ATOM_HANDLE * PRTL_ATOM_HANDLE

RTL_ATOM_HANDLE

typedef struct _RTL_ATOM_HANDLE RTL_ATOM_HANDLE

Function Documentation

find_actctx_dll()

NTSTATUS find_actctx_dll	(	PUNICODE_STRING	pnameW,
		LPWSTR *	fullname,
		PUNICODE_STRING	CallerBuffer,
		BOOLEAN	bAllocateBuffer
	)

Definition at line 715 of file libsupp.c.

{
    static const WCHAR winsxsW[] = {'\\','w','i','n','s','x','s','\\'};
    static const WCHAR dotManifestW[] = {'.','m','a','n','i','f','e','s','t',0};

    ACTIVATION_CONTEXT_ASSEMBLY_DETAILED_INFORMATION *info;
    ACTCTX_SECTION_KEYED_DATA data;
    NTSTATUS status;
    SIZE_T needed, size = 1024;
    WCHAR *p;

    data.cbSize = sizeof(data);
    status = RtlFindActivationContextSectionString( FIND_ACTCTX_SECTION_KEY_RETURN_HACTCTX, NULL,
                                                    ACTIVATION_CONTEXT_SECTION_DLL_REDIRECTION,
                                                    pnameW, &data );
    if (status != STATUS_SUCCESS)
    {
        //DPRINT1("RtlFindActivationContextSectionString returned 0x%x for %wZ\n", status, pnameW);
        return status;
    }

    for (;;)
    {
        if (!(info = RtlAllocateHeap( RtlGetProcessHeap(), 0, size )))
        {
            status = STATUS_NO_MEMORY;
            goto done;
        }
        status = RtlQueryInformationActivationContext( 0, data.hActCtx, &data.ulAssemblyRosterIndex,
                                                       AssemblyDetailedInformationInActivationContext,
                                                       info, size, &needed );
        if (status == STATUS_SUCCESS) break;
        if (status != STATUS_BUFFER_TOO_SMALL) goto done;
        RtlFreeHeap( RtlGetProcessHeap(), 0, info );
        size = needed;
    }

    DPRINT("manifestpath === %S\n", info->lpAssemblyManifestPath);
    DPRINT("DirectoryName === %S\n", info->lpAssemblyDirectoryName);
    if (!info->lpAssemblyManifestPath /*|| !info->lpAssemblyDirectoryName*/)
    {
        status = STATUS_SXS_KEY_NOT_FOUND;
        goto done;
    }    

    if ((p = wcsrchr( info->lpAssemblyManifestPath, '\\' )))
    {
        DWORD dirlen = info->ulAssemblyDirectoryNameLength / sizeof(WCHAR);

        p++;
        if (!info->lpAssemblyDirectoryName || _wcsnicmp( p, info->lpAssemblyDirectoryName, dirlen ) || wcsicmp( p + dirlen, dotManifestW ))
        {
            /* manifest name does not match directory name, so it's not a global
             * windows/winsxs manifest; use the manifest directory name instead */
            dirlen = p - info->lpAssemblyManifestPath;
            needed = (dirlen + 1) * sizeof(WCHAR) + pnameW->Length;

            status = get_buffer(fullname, needed, CallerBuffer, bAllocateBuffer);
            if (!NT_SUCCESS(status))
                goto done;

            p = *fullname;

            memcpy( p, info->lpAssemblyManifestPath, dirlen * sizeof(WCHAR) );
            p += dirlen;
            memcpy( p, pnameW->Buffer, pnameW->Length);
            p += (pnameW->Length / sizeof(WCHAR));
            *p = L'\0';

            goto done;
        }
    }

    needed = (wcslen(SharedUserData->NtSystemRoot) * sizeof(WCHAR) +
              sizeof(winsxsW) + info->ulAssemblyDirectoryNameLength + pnameW->Length + 2*sizeof(WCHAR));

    status = get_buffer(fullname, needed, CallerBuffer, bAllocateBuffer);
    if (!NT_SUCCESS(status))
        goto done;

    p = *fullname;

    wcscpy( p, SharedUserData->NtSystemRoot );
    p += wcslen(p);
    memcpy( p, winsxsW, sizeof(winsxsW) );
    p += sizeof(winsxsW) / sizeof(WCHAR);
    memcpy( p, info->lpAssemblyDirectoryName, info->ulAssemblyDirectoryNameLength );
    p += info->ulAssemblyDirectoryNameLength / sizeof(WCHAR);
    *p++ = L'\\';
    memcpy( p, pnameW->Buffer, pnameW->Length);
    p += (pnameW->Length / sizeof(WCHAR));
    *p = L'\0';

done:
    RtlFreeHeap( RtlGetProcessHeap(), 0, info );
    RtlReleaseActivationContext( data.hActCtx );
    DPRINT("%S\n", fullname);
    return status;
}

Referenced by RtlDosApplyFileIsolationRedirection_Ustr().

find_entry()

NTSTATUS find_entry	(	PVOID	BaseAddress,
		LDR_RESOURCE_INFO *	info,
		ULONG	level,
		void **	ret,
		int	want_dir
	)

Definition at line 570 of file libsupp.c.

{
    ULONG size;
    void *root;
    IMAGE_RESOURCE_DIRECTORY *resdirptr;
    USHORT list[9];  /* list of languages to try */
    int i, pos = 0;
    LCID user_lcid, system_lcid;

    root = RtlImageDirectoryEntryToData( BaseAddress, TRUE, IMAGE_DIRECTORY_ENTRY_RESOURCE, &size );
    if (!root) return STATUS_RESOURCE_DATA_NOT_FOUND;
    if (size < sizeof(*resdirptr)) return STATUS_RESOURCE_DATA_NOT_FOUND;
    resdirptr = root;

    if (!level--) goto done;
    if (!(*ret = find_entry_by_name( resdirptr, (LPCWSTR)info->Type, root, want_dir || level )))
        return STATUS_RESOURCE_TYPE_NOT_FOUND;
    if (!level--) return STATUS_SUCCESS;

    resdirptr = *ret;
    if (!(*ret = find_entry_by_name( resdirptr, (LPCWSTR)info->Name, root, want_dir || level )))
        return STATUS_RESOURCE_NAME_NOT_FOUND;
    if (!level--) return STATUS_SUCCESS;
    if (level) return STATUS_INVALID_PARAMETER;  /* level > 3 */

    /* 1. specified language */
    pos = push_language( list, pos, info->Language );

    /* 2. specified language with neutral sublanguage */
    pos = push_language( list, pos, MAKELANGID( PRIMARYLANGID(info->Language), SUBLANG_NEUTRAL ) );

    /* 3. neutral language with neutral sublanguage */
    pos = push_language( list, pos, MAKELANGID( LANG_NEUTRAL, SUBLANG_NEUTRAL ) );

    /* if no explicitly specified language, try some defaults */
    if (PRIMARYLANGID(info->Language) == LANG_NEUTRAL)
    {
        /* user defaults, unless SYS_DEFAULT sublanguage specified  */
        if (SUBLANGID(info->Language) != SUBLANG_SYS_DEFAULT)
        {
            /* 4. current thread locale language */
            pos = push_language( list, pos, LANGIDFROMLCID(NtCurrentTeb()->CurrentLocale) );

            if (NT_SUCCESS(NtQueryDefaultLocale(TRUE, &user_lcid)))
            {
                /* 5. user locale language */
                pos = push_language( list, pos, LANGIDFROMLCID(user_lcid) );

                /* 6. user locale language with neutral sublanguage  */
                pos = push_language( list, pos, MAKELANGID( PRIMARYLANGID(user_lcid), SUBLANG_NEUTRAL ) );
            }
        }

        /* now system defaults */

        if (NT_SUCCESS(NtQueryDefaultLocale(FALSE, &system_lcid)))
        {
            /* 7. system locale language */
            pos = push_language( list, pos, LANGIDFROMLCID( system_lcid ) );

            /* 8. system locale language with neutral sublanguage */
            pos = push_language( list, pos, MAKELANGID( PRIMARYLANGID(system_lcid), SUBLANG_NEUTRAL ) );
        }

        /* 9. English */
        pos = push_language( list, pos, MAKELANGID( LANG_ENGLISH, SUBLANG_DEFAULT ) );
    }

    resdirptr = *ret;
    for (i = 0; i < pos; i++)
        if ((*ret = find_entry_by_id( resdirptr, list[i], root, want_dir ))) return STATUS_SUCCESS;

    /* if no explicitly specified language, return the first entry */
    if (PRIMARYLANGID(info->Language) == LANG_NEUTRAL)
    {
        if ((*ret = find_first_entry( resdirptr, root, want_dir ))) return STATUS_SUCCESS;
    }
    return STATUS_RESOURCE_LANG_NOT_FOUND;

done:
    *ret = resdirptr;
    return STATUS_SUCCESS;
}

find_entry_by_id()

IMAGE_RESOURCE_DIRECTORY* find_entry_by_id	(	IMAGE_RESOURCE_DIRECTORY *	dir,
		WORD	id,
		void *	root,
		int	want_dir
	)

Definition at line 95 of file res.c.

{
    const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry;
    int min, max, pos;

    entry = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(dir + 1);
    min = dir->NumberOfNamedEntries;
    max = min + dir->NumberOfIdEntries - 1;
    while (min <= max)
    {
        pos = (min + max) / 2;
        if (entry[pos].Id == id)
        {
            if (!entry[pos].DataIsDirectory == !want_dir)
            {
                DPRINT("root %p dir %p id %04x ret %p\n",
                       root, dir, id, (const char*)root + entry[pos].OffsetToDirectory);
                return (IMAGE_RESOURCE_DIRECTORY *)((char *)root + entry[pos].OffsetToDirectory);
            }
            break;
        }
        if (entry[pos].Id > id) max = pos - 1;
        else min = pos + 1;
    }
    DPRINT("root %p dir %p id %04x not found\n", root, dir, id );
    return NULL;
}

Referenced by find_entry().

find_entry_by_name()

IMAGE_RESOURCE_DIRECTORY* find_entry_by_name	(	IMAGE_RESOURCE_DIRECTORY *	dir,
		LPCWSTR	name,
		void *	root,
		int	want_dir
	)

Definition at line 130 of file res.c.

{
    const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry;
    const IMAGE_RESOURCE_DIR_STRING_U *str;
    int min, max, res, pos;
    size_t namelen;

    if (!((ULONG_PTR)name & 0xFFFF0000)) return find_entry_by_id( dir, (ULONG_PTR)name & 0xFFFF, root, want_dir );
    entry = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(dir + 1);
    namelen = wcslen(name);
    min = 0;
    max = dir->NumberOfNamedEntries - 1;
    while (min <= max)
    {
        pos = (min + max) / 2;
        str = (const IMAGE_RESOURCE_DIR_STRING_U *)((const char *)root + entry[pos].NameOffset);
        res = _wcsnicmp( name, str->NameString, str->Length );
        if (!res && namelen == str->Length)
        {
            if (!entry[pos].DataIsDirectory == !want_dir)
            {
                DPRINT("root %p dir %p name %ws ret %p\n",
                       root, dir, name, (const char*)root + entry[pos].OffsetToDirectory);
                return (IMAGE_RESOURCE_DIRECTORY *)((char *)root + entry[pos].OffsetToDirectory);
            }
            break;
        }
        if (res < 0) max = pos - 1;
        else min = pos + 1;
    }
    DPRINT("root %p dir %p name %ws not found\n", root, dir, name);
    return NULL;
}

Referenced by find_entry().

find_first_entry()

IMAGE_RESOURCE_DIRECTORY* find_first_entry	(	IMAGE_RESOURCE_DIRECTORY *	dir,
		void *	root,
		int	want_dir
	)

Definition at line 75 of file res.c.

{
    const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(dir + 1);
    int pos;

    for (pos = 0; pos < dir->NumberOfNamedEntries + dir->NumberOfIdEntries; pos++)
    {
        if (!entry[pos].DataIsDirectory == !want_dir)
            return (IMAGE_RESOURCE_DIRECTORY *)((char *)root + entry[pos].OffsetToDirectory);
    }
    return NULL;
}

Referenced by find_entry().

get_buffer()

NTSTATUS get_buffer	(	LPWSTR *	buffer,
		SIZE_T	needed,
		PUNICODE_STRING	CallerBuffer,
		BOOLEAN	bAllocateBuffer
	)

Definition at line 688 of file libsupp.c.

{
    WCHAR *p;

    if (CallerBuffer && CallerBuffer->MaximumLength > needed)
    {
        p = CallerBuffer->Buffer;
        CallerBuffer->Length = needed - sizeof(WCHAR);
    }
    else
    {
        if (!bAllocateBuffer)
            return STATUS_BUFFER_TOO_SMALL;

        if (CallerBuffer)
            CallerBuffer->Buffer[0] = 0;

        p = RtlAllocateHeap(RtlGetProcessHeap(), 0, needed );
        if (!p)
            return STATUS_NO_MEMORY;
    }
    *buffer = p;

    return STATUS_SUCCESS;
}

Referenced by find_actctx_dll(), jpeg_fill_bit_buffer(), jpeg_huff_decode(), and RtlDosApplyFileIsolationRedirection_Ustr().

push_language()

int push_language	(	USHORT *	list,
		ULONG	pos,
		WORD	lang
	)

Definition at line 61 of file res.c.

{
    ULONG i;
    for (i = 0; i < pos; i++) if (list[i] == lang) return pos;
    list[pos++] = lang;
    return pos;
}

Referenced by find_entry().

RtlAcquirePebLock()

VOID NTAPI RtlAcquirePebLock

(

VOID

)

Definition at line 72 of file libsupp.c.

{
   PPEB Peb = NtCurrentPeb ();
   RtlEnterCriticalSection(Peb->FastPebLock);
}

Referenced by BaseGetNamedObjectDirectory(), BasepComputeProcessPath(), BasePushProcessParameters(), BaseRundownFls(), CheckForSameCurdir(), ExitProcess(), FlsAlloc(), FlsFree(), FlsSetValue(), GetEnvironmentStringsA(), GetEnvironmentStringsW(), GetEnvironmentVariableA(), LdrShutdownThread(), RtlCreateEnvironment(), RtlCreateProcessParameters(), RtlGetCurrentDirectory_U(), RtlGetFullPathName_Ustr(), RtlGetFullPathName_UstrEx(), RtlpDosPathNameToRelativeNtPathName_Ustr(), RtlQueryEnvironmentVariable_U(), RtlSetCurrentDirectory_U(), RtlSetCurrentEnvironment(), RtlSetEnvironmentVariable(), TlsAlloc(), TlsFree(), and TlsSetValue().

RtlComputeImportTableHash()

NTSYSAPI NTSTATUS NTAPI RtlComputeImportTableHash	(	IN HANDLE	FileHandle,
		OUT PCHAR	Hash,
		IN ULONG	ImportTableHashSize
	)

Definition at line 1027 of file libsupp.c.

{
    UNIMPLEMENTED;
    return STATUS_NOT_IMPLEMENTED;
}

Referenced by BasepCheckWebBladeHashes().

RtlDeleteHeapLock()

NTSTATUS NTAPI RtlDeleteHeapLock

(

IN OUT PHEAP_LOCK

Lock

)

Definition at line 101 of file libsupp.c.

{
    return RtlDeleteCriticalSection(&Lock->CriticalSection);
}

Referenced by RtlDestroyHeap(), and RtlpPageHeapDestroy().

RtlDosApplyFileIsolationRedirection_Ustr()

NTSYSAPI NTSTATUS NTAPI RtlDosApplyFileIsolationRedirection_Ustr	(	IN ULONG	Flags,
		IN PUNICODE_STRING	OriginalName,
		IN PUNICODE_STRING	Extension,
		IN OUT PUNICODE_STRING	StaticString,
		IN OUT PUNICODE_STRING	DynamicString,
		IN OUT PUNICODE_STRING *	NewName,
		IN PULONG	NewFlags,
		IN PSIZE_T	FileNameSize,
		IN PSIZE_T	RequiredLength
	)

Definition at line 821 of file libsupp.c.

{
    NTSTATUS Status;
    LPWSTR fullname;
    WCHAR buffer [MAX_PATH];
    UNICODE_STRING localStr, localStr2, *pstrParam;
    WCHAR *p;
    BOOLEAN GotExtension;
    WCHAR c;
    C_ASSERT(sizeof(UNICODE_NULL) == sizeof(WCHAR));


    /* Check for invalid parameters */
    if (!OriginalName)
    {
        return STATUS_INVALID_PARAMETER;
    }

    if (!DynamicString && !StaticString)
    {
        return STATUS_INVALID_PARAMETER;
    }

    if ((DynamicString) && (StaticString) && !(NewName))
    {
        return STATUS_INVALID_PARAMETER;
    }

    if (!OriginalName->Buffer || OriginalName->Length == 0)
    {
        return STATUS_SXS_KEY_NOT_FOUND;
    }

    if (StaticString && (OriginalName == StaticString || OriginalName->Buffer == StaticString->Buffer))
    {
        return STATUS_SXS_KEY_NOT_FOUND;
    }

    if (NtCurrentPeb()->ProcessParameters &&
        (NtCurrentPeb()->ProcessParameters->Flags & RTL_USER_PROCESS_PARAMETERS_PRIVATE_DLL_PATH))
    {
        UNICODE_STRING RealName, LocalName;
        WCHAR RealNameBuf[MAX_PATH], LocalNameBuf[MAX_PATH];

        RtlInitEmptyUnicodeString(&RealName, RealNameBuf, sizeof(RealNameBuf));
        RtlInitEmptyUnicodeString(&LocalName, LocalNameBuf, sizeof(LocalNameBuf));

        Status = RtlComputePrivatizedDllName_U(OriginalName, &RealName, &LocalName);
        if (!NT_SUCCESS(Status))
        {
            DPRINT1("RtlComputePrivatizedDllName_U failed for %wZ: 0x%lx\n", OriginalName, Status);
            return Status;
        }

        if (RtlDoesFileExists_UStr(&LocalName))
        {
            Status = get_buffer(&fullname, LocalName.Length + sizeof(UNICODE_NULL), StaticString, DynamicString != NULL);
            if (NT_SUCCESS(Status))
            {
                RtlCopyMemory(fullname, LocalName.Buffer, LocalName.Length + sizeof(UNICODE_NULL));
            }
            else
            {
                DPRINT1("Error while retrieving buffer for %wZ: 0x%lx\n", OriginalName, Status);
            }
        }
        else if (RtlDoesFileExists_UStr(&RealName))
        {
            Status = get_buffer(&fullname, RealName.Length + sizeof(UNICODE_NULL), StaticString, DynamicString != NULL);
            if (NT_SUCCESS(Status))
            {
                RtlCopyMemory(fullname, RealName.Buffer, RealName.Length + sizeof(UNICODE_NULL));
            }
            else
            {
                DPRINT1("Error while retrieving buffer for %wZ: 0x%lx\n", OriginalName, Status);
            }
        }
        else
        {
            Status = STATUS_NOT_FOUND;
        }

        if (RealName.Buffer != RealNameBuf)
            RtlFreeUnicodeString(&RealName);
        if (LocalName.Buffer != LocalNameBuf)
            RtlFreeUnicodeString(&LocalName);

        if (NT_SUCCESS(Status))
        {
            DPRINT("Redirecting %wZ to %S\n", OriginalName, fullname);
            if (!StaticString || StaticString->Buffer != fullname)
            {
                RtlInitUnicodeString(DynamicString, fullname);
                *NewName = DynamicString;
            }
            else
            {
                *NewName = StaticString;
            }
            return Status;
        }
    }

    pstrParam = OriginalName;

    /* Get the file name with an extension */
    p = OriginalName->Buffer + OriginalName->Length / sizeof(WCHAR) - 1;
    GotExtension = FALSE;
    while (p >= OriginalName->Buffer)
    {
        c = *p--;
        if (c == L'.')
        {
            GotExtension = TRUE;
        }
        else if (c == L'\\')
        {
            localStr.Buffer = p + 2;
            localStr.Length = OriginalName->Length - ((ULONG_PTR)localStr.Buffer - (ULONG_PTR)OriginalName->Buffer);
            localStr.MaximumLength = OriginalName->MaximumLength - ((ULONG_PTR)localStr.Buffer - (ULONG_PTR)OriginalName->Buffer);
            pstrParam = &localStr;
            break;
        }
    }

    if (!GotExtension)
    {
        if (!Extension)
        {
            return STATUS_SXS_KEY_NOT_FOUND;
        }

        if (pstrParam->Length + Extension->Length > sizeof(buffer))
        {
            //FIXME!
            return STATUS_NO_MEMORY;
        }

        RtlInitEmptyUnicodeString(&localStr2, buffer, sizeof(buffer));
        RtlAppendUnicodeStringToString(&localStr2, pstrParam);
        RtlAppendUnicodeStringToString(&localStr2, Extension);
        pstrParam = &localStr2;
    }

    /* Use wine's function as long as we use wine's sxs implementation in ntdll */
    Status = find_actctx_dll(pstrParam, &fullname, StaticString, DynamicString != NULL);
    if (!NT_SUCCESS(Status))
    {
        return Status;
    }

    DPRINT("Redirecting %wZ to %S\n", OriginalName, fullname);

    if (!StaticString || StaticString->Buffer != fullname)
    {
        RtlInitUnicodeString(DynamicString, fullname);
        *NewName = DynamicString;
    }
    else
    {
        *NewName = StaticString;
    }

    return Status;
}

Referenced by LdrGetDllHandleEx(), LdrLoadDll(), LdrpLoadImportModule(), LdrpSnapThunk(), LdrpUpdateLoadCount3(), RtlDosSearchPath_Ustr(), TestBuffers(), and TestRedirection().

RtlEnterHeapLock()

NTSTATUS NTAPI RtlEnterHeapLock	(	IN OUT PHEAP_LOCK	Lock,
		IN BOOLEAN	Exclusive
	)

Definition at line 108 of file libsupp.c.

{
    UNREFERENCED_PARAMETER(Exclusive);

    return RtlEnterCriticalSection(&Lock->CriticalSection);
}

Referenced by RtlAllocateHeap(), RtlDebugAllocateHeap(), RtlDebugFreeHeap(), RtlDebugGetUserInfoHeap(), RtlDebugReAllocateHeap(), RtlDebugSetUserFlagsHeap(), RtlDebugSetUserValueHeap(), RtlDebugSizeHeap(), RtlFreeHeap(), RtlGetUserInfoHeap(), RtlLockHeap(), RtlpDphEnterCriticalSection(), RtlpDphFreeDelayedBlocksFromHeap(), RtlpPageHeapCreate(), RtlpPageHeapDestroy(), RtlReAllocateHeap(), RtlSetUserFlagsHeap(), RtlSetUserValueHeap(), and RtlValidateHeap().

RtlGetCurrentPeb()

PPEB NTAPI RtlGetCurrentPeb

(

VOID

)

Definition at line 63 of file libsupp.c.

{
    return NtCurrentPeb();
}

Referenced by LdrpLoadImportModule(), RtlComputePrivatizedDllName_U(), RtlDebugDestroyHeap(), RtlInitializeHeapManager(), RtlpAddHeapToProcessList(), RtlpRemoveHeapFromProcessList(), RtlpSetHeapParameters(), and RtlQueryEnvironmentVariable_U().

RtlGetNtGlobalFlags()

ULONG NTAPI RtlGetNtGlobalFlags

(

VOID

)

Definition at line 93 of file libsupp.c.

{
    PPEB pPeb = NtCurrentPeb();
    return pPeb->NtGlobalFlag;
}

Referenced by RtlCreateHeap(), and RtlCreateUserProcess().

RtlGetTickCount()

ULONG NTAPI RtlGetTickCount

(

VOID

)

Definition at line 1058 of file libsupp.c.

{
    ULARGE_INTEGER TickCount;

#ifdef _WIN64
    TickCount.QuadPart = *((volatile ULONG64*)&SharedUserData->TickCount);
#else
    while (TRUE)
    {
        TickCount.HighPart = (ULONG)SharedUserData->TickCount.High1Time;
        TickCount.LowPart = SharedUserData->TickCount.LowPart;

        if (TickCount.HighPart == (ULONG)SharedUserData->TickCount.High2Time)
            break;

        YieldProcessor();
    }
#endif

    return (ULONG)((UInt32x32To64(TickCount.LowPart,
                                  SharedUserData->TickCountMultiplier) >> 24) +
                    UInt32x32To64((TickCount.HighPart << 8) & 0xFFFFFFFF,
                                  SharedUserData->TickCountMultiplier));
}

RtlImageNtHeaderEx()

NTSTATUS NTAPI RtlImageNtHeaderEx	(	_In_ ULONG	Flags,
		_In_ PVOID	Base,
		_In_ ULONG64	Size,
		_Out_ PIMAGE_NT_HEADERS *	OutHeaders
	)

Definition at line 526 of file libsupp.c.

{
    NTSTATUS Status;

    /* Assume failure. This is also done in RtlpImageNtHeaderEx, but this is guarded by SEH. */
    if (OutHeaders != NULL)
        *OutHeaders = NULL;

    _SEH2_TRY
    {
        Status = RtlpImageNtHeaderEx(Flags, Base, Size, OutHeaders);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        /* Fail with the SEH error */
        Status = _SEH2_GetExceptionCode();
    }
    _SEH2_END;

    return Status;
}

RtlInitializeHeapLock()

NTSTATUS NTAPI RtlInitializeHeapLock

(

IN OUT PHEAP_LOCK *

Lock

)

Definition at line 126 of file libsupp.c.

{
    return RtlInitializeCriticalSection(&(*Lock)->CriticalSection);
}

Referenced by RtlpDphInitializeDelayedFreeQueue(), RtlpDphProcessStartupInitialization(), RtlpInitializeHeap(), and RtlpPageHeapCreate().

RtlIsThreadWithinLoaderCallout()

BOOLEAN NTAPI RtlIsThreadWithinLoaderCallout

(

VOID

)

Definition at line 348 of file libsupp.c.

{
    return LdrpTopLevelDllBeingLoadedTeb == NtCurrentTeb();
}

Referenced by ForceResetUserApiHook(), and LoadUserApiHook().

RtlLeaveHeapLock()

NTSTATUS NTAPI RtlLeaveHeapLock

(

IN OUT PHEAP_LOCK

Lock

)

Definition at line 133 of file libsupp.c.

{
    return RtlLeaveCriticalSection(&Lock->CriticalSection);
}

Referenced by RtlAllocateHeap(), RtlDebugAllocateHeap(), RtlDebugFreeHeap(), RtlDebugGetUserInfoHeap(), RtlDebugReAllocateHeap(), RtlDebugSetUserFlagsHeap(), RtlDebugSetUserValueHeap(), RtlDebugSizeHeap(), RtlFreeHeap(), RtlGetUserInfoHeap(), RtlpAllocateNonDedicated(), RtlpDphFreeDelayedBlocksFromHeap(), RtlpDphLeaveCriticalSection(), RtlpPageHeapCreate(), RtlpPageHeapDestroy(), RtlReAllocateHeap(), RtlSetUserFlagsHeap(), RtlSetUserValueHeap(), RtlUnlockHeap(), and RtlValidateHeap().

RtlpAllocateMemory()

PVOID NTAPI RtlpAllocateMemory	(	UINT	Bytes,
		ULONG	Tag
	)

Definition at line 140 of file libsupp.c.

{
    UNREFERENCED_PARAMETER(Tag);

    return RtlAllocateHeap(RtlGetProcessHeap(),
                           0,
                           Bytes);
}

RtlpAllocAtomTable()

PRTL_ATOM_TABLE RtlpAllocAtomTable

(

ULONG

Size

)

Definition at line 410 of file libsupp.c.

{
   return (PRTL_ATOM_TABLE)RtlAllocateHeap(RtlGetProcessHeap(),
                                           HEAP_ZERO_MEMORY,
                                           Size);
}

RtlpAllocAtomTableEntry()

PRTL_ATOM_TABLE_ENTRY RtlpAllocAtomTableEntry

(

ULONG

Size

)

Definition at line 426 of file libsupp.c.

{
   return (PRTL_ATOM_TABLE_ENTRY)RtlAllocateHeap(RtlGetProcessHeap(),
                                                 HEAP_ZERO_MEMORY,
                                                 Size);
}

RtlpCaptureStackLimits()

BOOLEAN NTAPI RtlpCaptureStackLimits	(	IN ULONG_PTR	Ebp,
		IN ULONG_PTR *	StackBegin,
		IN ULONG_PTR *	StackEnd
	)

Definition at line 211 of file libsupp.c.

{
    /* FIXME: Verify */
    *StackBegin = (ULONG_PTR)NtCurrentTeb()->NtTib.StackLimit;
    *StackEnd = (ULONG_PTR)NtCurrentTeb()->NtTib.StackBase;
    return TRUE;
}

Referenced by RtlWalkFrameChain().

RtlpCheckForActiveDebugger()

BOOLEAN NTAPI RtlpCheckForActiveDebugger

(

VOID

)

Definition at line 25 of file libsupp.c.

{
    /* Return the flag in the PEB */
    return NtCurrentPeb()->BeingDebugged;
}

Referenced by RtlRaiseException(), RtlRaiseStatus(), and vDbgPrintExWithPrefixInternal().

RtlpCheckLogException()

VOID NTAPI RtlpCheckLogException	(	IN PEXCEPTION_RECORD	ExceptionRecord,
		IN PCONTEXT	ContextRecord,
		IN PVOID	ContextData,
		IN ULONG	Size
	)

Definition at line 201 of file libsupp.c.

{
    /* Exception logging is not done in user-mode */
}

Referenced by RtlDispatchException().

RtlpClearInDbgPrint()

VOID NTAPI RtlpClearInDbgPrint

(

VOID

)

Definition at line 45 of file libsupp.c.

{
    /* Clear the flag */
    NtCurrentTeb()->InDbgPrint = FALSE;
}

Referenced by vDbgPrintExWithPrefixInternal().

RtlpCreateAtomHandle()

BOOLEAN RtlpCreateAtomHandle	(	PRTL_ATOM_TABLE	AtomTable,
		PRTL_ATOM_TABLE_ENTRY	Entry
	)

Definition at line 456 of file libsupp.c.

{
   ULONG HandleIndex;
   PRTL_HANDLE_TABLE_ENTRY RtlHandle;

   RtlHandle = RtlAllocateHandle(&AtomTable->RtlHandleTable,
                                 &HandleIndex);
   if (RtlHandle != NULL)
   {
      PRTL_ATOM_HANDLE AtomHandle = (PRTL_ATOM_HANDLE)RtlHandle;

      /* FIXME - Handle Indexes >= 0xC000 ?! */
      if (HandleIndex < 0xC000)
      {
         Entry->HandleIndex = (USHORT)HandleIndex;
         Entry->Atom = 0xC000 + (USHORT)HandleIndex;

         AtomHandle->AtomEntry = Entry;
         AtomHandle->Handle.Flags = RTL_HANDLE_VALID;

         return TRUE;
      }
      else
      {
         /* set the valid flag, otherwise RtlFreeHandle will fail! */
         AtomHandle->Handle.Flags = RTL_HANDLE_VALID;

         RtlFreeHandle(&AtomTable->RtlHandleTable,
                       RtlHandle);
      }
   }

   return FALSE;
}

RtlpCreateAtomHandleTable()

BOOLEAN RtlpCreateAtomHandleTable

(

PRTL_ATOM_TABLE

AtomTable

)

Definition at line 394 of file libsupp.c.

{
   RtlInitializeHandleTable(0xCFFF,
                            sizeof(RTL_ATOM_HANDLE),
                            &AtomTable->RtlHandleTable);

   return TRUE;
}

RtlPcToFileHeader()

PVOID NTAPI RtlPcToFileHeader	(	IN PVOID	PcValue,
		PVOID *	BaseOfImage
	)

Definition at line 659 of file libsupp.c.

{
    PLIST_ENTRY ModuleListHead;
    PLIST_ENTRY Entry;
    PLDR_DATA_TABLE_ENTRY Module;
    PVOID ImageBase = NULL;

    RtlEnterCriticalSection (NtCurrentPeb()->LoaderLock);
    ModuleListHead = &NtCurrentPeb()->Ldr->InLoadOrderModuleList;
    Entry = ModuleListHead->Flink;
    while (Entry != ModuleListHead)
    {
        Module = CONTAINING_RECORD(Entry, LDR_DATA_TABLE_ENTRY, InLoadOrderLinks);

        if ((ULONG_PTR)PcValue >= (ULONG_PTR)Module->DllBase &&
                (ULONG_PTR)PcValue < (ULONG_PTR)Module->DllBase + Module->SizeOfImage)
        {
            ImageBase = Module->DllBase;
            break;
        }
        Entry = Entry->Flink;
    }
    RtlLeaveCriticalSection (NtCurrentPeb()->LoaderLock);

    *BaseOfImage = ImageBase;
    return ImageBase;
}

Referenced by BasepGetModuleHandleExW(), RtlLookupFunctionTable(), and SeiGetModuleFromAddress().

RtlpDestroyAtomHandleTable()

VOID RtlpDestroyAtomHandleTable

(

PRTL_ATOM_TABLE

AtomTable

)

Definition at line 404 of file libsupp.c.

{
   RtlDestroyHandleTable(&AtomTable->RtlHandleTable);
}

RtlpDestroyAtomTableLock()

VOID RtlpDestroyAtomTableLock

(

PRTL_ATOM_TABLE

AtomTable

)

Definition at line 370 of file libsupp.c.

{
   RtlDeleteCriticalSection(&AtomTable->CriticalSection);
}

RtlpFreeAtomHandle()

VOID RtlpFreeAtomHandle	(	PRTL_ATOM_TABLE	AtomTable,
		PRTL_ATOM_TABLE_ENTRY	Entry
	)

Definition at line 442 of file libsupp.c.

{
   PRTL_HANDLE_TABLE_ENTRY RtlHandleEntry;

   if (RtlIsValidIndexHandle(&AtomTable->RtlHandleTable,
                             (ULONG)Entry->HandleIndex,
                             &RtlHandleEntry))
   {
      RtlFreeHandle(&AtomTable->RtlHandleTable,
                    RtlHandleEntry);
   }
}

RtlpFreeAtomTable()

VOID RtlpFreeAtomTable

(

PRTL_ATOM_TABLE

AtomTable

)

Definition at line 418 of file libsupp.c.

{
   RtlFreeHeap(RtlGetProcessHeap(),
               0,
               AtomTable);
}

RtlpFreeAtomTableEntry()

VOID RtlpFreeAtomTableEntry

(

PRTL_ATOM_TABLE_ENTRY

Entry

)

Definition at line 434 of file libsupp.c.

{
   RtlFreeHeap(RtlGetProcessHeap(),
               0,
               Entry);
}

RtlpFreeMemory()

VOID NTAPI RtlpFreeMemory	(	PVOID	Mem,
		ULONG	Tag
	)

Definition at line 153 of file libsupp.c.

{
    UNREFERENCED_PARAMETER(Tag);

    RtlFreeHeap(RtlGetProcessHeap(),
                0,
                Mem);
}

RtlpGetAtomEntry()

PRTL_ATOM_TABLE_ENTRY RtlpGetAtomEntry	(	PRTL_ATOM_TABLE	AtomTable,
		ULONG	Index
	)

Definition at line 492 of file libsupp.c.

{
   PRTL_HANDLE_TABLE_ENTRY RtlHandle;

   if (RtlIsValidIndexHandle(&AtomTable->RtlHandleTable,
                             Index,
                             &RtlHandle))
   {
      PRTL_ATOM_HANDLE AtomHandle = (PRTL_ATOM_HANDLE)RtlHandle;

      return AtomHandle->AtomEntry;
   }

   return NULL;
}

RtlpGetMode()

KPROCESSOR_MODE NTAPI RtlpGetMode

(

VOID

)

Definition at line 53 of file libsupp.c.

{
   return UserMode;
}

Referenced by RtlCreateHeap(), RtlCreateProcessParameters(), RtlDestroyHeap(), RtlFreeHeap(), RtlpExtendHeap(), and RtlpInitializeHeap().

RtlpHandleDpcStackException()

BOOLEAN NTAPI RtlpHandleDpcStackException	(	IN PEXCEPTION_REGISTRATION_RECORD	RegistrationFrame,
		IN ULONG_PTR	RegistrationFrameEnd,
		IN OUT PULONG_PTR	StackLow,
		IN OUT PULONG_PTR	StackHigh
	)

Definition at line 190 of file libsupp.c.

{
    /* There's no such thing as a DPC stack in user-mode */
    return FALSE;
}

Referenced by RtlDispatchException(), and RtlUnwind().

RtlpImageNtHeaderEx()

NTSTATUS NTAPI RtlpImageNtHeaderEx	(	_In_ ULONG	Flags,
		_In_ PVOID	Base,
		_In_ ULONG64	Size,
		_Out_ PIMAGE_NT_HEADERS *	OutHeaders
	)

Definition at line 140 of file image.c.

{
    PIMAGE_NT_HEADERS NtHeaders;
    PIMAGE_DOS_HEADER DosHeader;
    BOOLEAN WantsRangeCheck;
    ULONG NtHeaderOffset;

    /* You must want NT Headers, no? */
    if (OutHeaders == NULL)
    {
        DPRINT1("OutHeaders is NULL\n");
        return STATUS_INVALID_PARAMETER;
    }

    /* Assume failure */
    *OutHeaders = NULL;

    /* Validate Flags */
    if (Flags & ~RTL_IMAGE_NT_HEADER_EX_FLAG_NO_RANGE_CHECK)
    {
        DPRINT1("Invalid flags: 0x%lx\n", Flags);
        return STATUS_INVALID_PARAMETER;
    }

    /* Validate base */
    if ((Base == NULL) || (Base == (PVOID)-1))
    {
        DPRINT1("Invalid base address: %p\n", Base);
        return STATUS_INVALID_PARAMETER;
    }

    /* Check if the caller wants range checks */
    WantsRangeCheck = !(Flags & RTL_IMAGE_NT_HEADER_EX_FLAG_NO_RANGE_CHECK);
    if (WantsRangeCheck)
    {
        /* Make sure the image size is at least big enough for the DOS header */
        if (Size < sizeof(IMAGE_DOS_HEADER))
        {
            DPRINT1("Size too small\n");
            return STATUS_INVALID_IMAGE_FORMAT;
        }
    }

    /* Check if the DOS Signature matches */
    DosHeader = Base;
    if (DosHeader->e_magic != IMAGE_DOS_SIGNATURE)
    {
        /* Not a valid COFF */
        DPRINT1("Invalid image DOS signature!\n");
        return STATUS_INVALID_IMAGE_FORMAT;
    }

    /* Get the offset to the NT headers (and copy from LONG to ULONG) */
    NtHeaderOffset = DosHeader->e_lfanew;

    /* The offset must not be larger than 256MB, as a hard-coded check.
       In Windows this check is only done in user mode, not in kernel mode,
       but it shouldn't harm to have it anyway. Note that without this check,
       other overflow checks would become necessary! */
    if (NtHeaderOffset >= (256 * 1024 * 1024))
    {
        /* Fail */
        DPRINT1("NT headers offset is larger than 256MB!\n");
        return STATUS_INVALID_IMAGE_FORMAT;
    }

    /* Check if the caller wants validation */
    if (WantsRangeCheck)
    {
        /* Make sure the file header fits into the size */
        if ((NtHeaderOffset +
             RTL_SIZEOF_THROUGH_FIELD(IMAGE_NT_HEADERS, FileHeader)) >= Size)
        {
            /* Fail */
            DPRINT1("NT headers beyond image size!\n");
            return STATUS_INVALID_IMAGE_FORMAT;
        }
    }

    /* Now get a pointer to the NT Headers */
    NtHeaders = (PIMAGE_NT_HEADERS)((ULONG_PTR)Base + NtHeaderOffset);

    /* Check if the mapping is in user space */
    if (Base <= MmHighestUserAddress)
    {
        /* Make sure we don't overflow into kernel space */
        if ((PVOID)(NtHeaders + 1) > MmHighestUserAddress)
        {
            DPRINT1("Image overflows from user space into kernel space!\n");
            return STATUS_INVALID_IMAGE_FORMAT;
        }
    }

    /* Verify the PE Signature */
    if (NtHeaders->Signature != IMAGE_NT_SIGNATURE)
    {
        /* Fail */
        DPRINT1("Invalid image NT signature!\n");
        return STATUS_INVALID_IMAGE_FORMAT;
    }

    /* Now return success and the NT header */
    *OutHeaders = NtHeaders;
    return STATUS_SUCCESS;
}

RtlpInitAtomTableLock()

NTSTATUS RtlpInitAtomTableLock

(

PRTL_ATOM_TABLE

AtomTable

)

Definition at line 362 of file libsupp.c.

{
   RtlInitializeCriticalSection(&AtomTable->CriticalSection);
   return STATUS_SUCCESS;
}

RtlpLockAtomTable()

BOOLEAN RtlpLockAtomTable

(

PRTL_ATOM_TABLE

AtomTable

)

Definition at line 377 of file libsupp.c.

{
   RtlEnterCriticalSection(&AtomTable->CriticalSection);
   return TRUE;
}

RtlpSafeCopyMemory()

NTSTATUS NTAPI RtlpSafeCopyMemory	(	_Out_writes_bytes_all_(Length) VOID UNALIGNED *	Destination,
		_In_reads_bytes_(Length) CONST VOID UNALIGNED *	Source,
		_In_ SIZE_T	Length
	)

Definition at line 1037 of file libsupp.c.

{
    _SEH2_TRY
    {
        RtlCopyMemory(Destination, Source, Length);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        _SEH2_YIELD(return _SEH2_GetExceptionCode());
    }
    _SEH2_END;

    return STATUS_SUCCESS;
}

RtlpSetHeapParameters()

VOID NTAPI RtlpSetHeapParameters

(

IN PRTL_HEAP_PARAMETERS

Parameters

)

Definition at line 174 of file libsupp.c.

{
    PPEB Peb;

    /* Get PEB */
    Peb = RtlGetCurrentPeb();

    /* Apply defaults for non-set parameters */
    if (!Parameters->SegmentCommit) Parameters->SegmentCommit = Peb->HeapSegmentCommit;
    if (!Parameters->SegmentReserve) Parameters->SegmentReserve = Peb->HeapSegmentReserve;
    if (!Parameters->DeCommitFreeBlockThreshold) Parameters->DeCommitFreeBlockThreshold = Peb->HeapDeCommitFreeBlockThreshold;
    if (!Parameters->DeCommitTotalFreeThreshold) Parameters->DeCommitTotalFreeThreshold = Peb->HeapDeCommitTotalFreeThreshold;
}

Referenced by RtlCreateHeap().

RtlpSetInDbgPrint()

BOOLEAN NTAPI RtlpSetInDbgPrint

(

VOID

)

Definition at line 33 of file libsupp.c.

{
    /* Check if it's already set and return TRUE if so */
    if (NtCurrentTeb()->InDbgPrint) return TRUE;

    /* Set it and return */
    NtCurrentTeb()->InDbgPrint = TRUE;
    return FALSE;
}

Referenced by vDbgPrintExWithPrefixInternal().

RtlpUnlockAtomTable()

VOID RtlpUnlockAtomTable

(

PRTL_ATOM_TABLE

AtomTable

)

Definition at line 385 of file libsupp.c.

{
   RtlLeaveCriticalSection(&AtomTable->CriticalSection);
}

RtlReleasePebLock()

VOID NTAPI RtlReleasePebLock

(

VOID

)

Definition at line 82 of file libsupp.c.

{
   PPEB Peb = NtCurrentPeb ();
   RtlLeaveCriticalSection(Peb->FastPebLock);
}

Referenced by BaseGetNamedObjectDirectory(), BasepComputeProcessPath(), BasePushProcessParameters(), BaseRundownFls(), CheckForSameCurdir(), ExitProcess(), FlsAlloc(), FlsFree(), FlsSetValue(), GetEnvironmentStringsA(), GetEnvironmentStringsW(), GetEnvironmentVariableA(), LdrShutdownThread(), RtlCreateEnvironment(), RtlCreateProcessParameters(), RtlGetCurrentDirectory_U(), RtlGetFullPathName_Ustr(), RtlGetFullPathName_UstrEx(), RtlpDosPathNameToRelativeNtPathName_Ustr(), RtlQueryEnvironmentVariable_U(), RtlSetCurrentDirectory_U(), RtlSetCurrentEnvironment(), RtlSetEnvironmentVariable(), TlsAlloc(), TlsFree(), and TlsSetValue().

RtlTryEnterHeapLock()

BOOLEAN NTAPI RtlTryEnterHeapLock	(	IN OUT PHEAP_LOCK	Lock,
		IN BOOLEAN	Exclusive
	)

Definition at line 117 of file libsupp.c.

{
    UNREFERENCED_PARAMETER(Exclusive);

    return RtlTryEnterCriticalSection(&Lock->CriticalSection);
}

Referenced by RtlpDphEnterCriticalSection().

RtlWalkFrameChain()

ULONG NTAPI RtlWalkFrameChain	(	OUT PVOID *	Callers,
		IN ULONG	Count,
		IN ULONG	Flags
	)

Definition at line 227 of file libsupp.c.

{
    ULONG_PTR Stack, NewStack, StackBegin, StackEnd = 0;
    ULONG Eip;
    BOOLEAN Result, StopSearch = FALSE;
    ULONG i = 0;

    /* Get current EBP */
#if defined(_M_IX86)
#if defined __GNUC__
    __asm__("mov %%ebp, %0" : "=r" (Stack) : );
#elif defined(_MSC_VER)
    __asm mov Stack, ebp
#endif
#elif defined(_M_MIPS)
        __asm__("move $sp, %0" : "=r" (Stack) : );
#elif defined(_M_PPC)
    __asm__("mr %0,1" : "=r" (Stack) : );
#elif defined(_M_ARM)
#if defined __GNUC__
    __asm__("mov sp, %0" : "=r"(Stack) : );
#elif defined(_MSC_VER)
    // FIXME: Hack. Probably won't work if this ever actually manages to run someday.
    Stack = (ULONG_PTR)&Stack;
#endif
#else
#error Unknown architecture
#endif

    /* Set it as the stack begin limit as well */
    StackBegin = (ULONG_PTR)Stack;

    /* Check if we're called for non-logging mode */
    if (!Flags)
    {
        /* Get the actual safe limits */
        Result = RtlpCaptureStackLimits((ULONG_PTR)Stack,
                                        &StackBegin,
                                        &StackEnd);
        if (!Result) return 0;
    }

    /* Use a SEH block for maximum protection */
    _SEH2_TRY
    {
        /* Loop the frames */
        for (i = 0; i < Count; i++)
        {
            /*
             * Leave if we're past the stack,
             * if we're before the stack,
             * or if we've reached ourselves.
             */
            if ((Stack >= StackEnd) ||
                (!i ? (Stack < StackBegin) : (Stack <= StackBegin)) ||
                ((StackEnd - Stack) < (2 * sizeof(ULONG_PTR))))
            {
                /* We're done or hit a bad address */
                break;
            }

            /* Get new stack and EIP */
            NewStack = *(PULONG_PTR)Stack;
            Eip = *(PULONG_PTR)(Stack + sizeof(ULONG_PTR));

            /* Check if the new pointer is above the oldone and past the end */
            if (!((Stack < NewStack) && (NewStack < StackEnd)))
            {
                /* Stop searching after this entry */
                StopSearch = TRUE;
            }

            /* Also make sure that the EIP isn't a stack address */
            if ((StackBegin < Eip) && (Eip < StackEnd)) break;

            /* FIXME: Check that EIP is inside a loaded module */

            /* Save this frame */
            Callers[i] = (PVOID)Eip;

            /* Check if we should continue */
            if (StopSearch)
            {
                /* Return the next index */
                i++;
                break;
            }

            /* Move to the next stack */
            Stack = NewStack;
        }
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        /* No index */
        i = 0;
    }
    _SEH2_END;

    /* Return frames parsed */
    return i;
}

Referenced by KeRosCaptureUserStackBackTrace(), lie_about_fs_type(), RtlCaptureStackBackTrace(), and TestStackWalk4().

RtlWow64EnableFsRedirection()

NTSYSAPI NTSTATUS NTAPI RtlWow64EnableFsRedirection

(

IN BOOLEAN

Wow64FsEnableRedirection

)

Definition at line 1002 of file libsupp.c.

{
    /* This is what Windows returns on x86 */
    return STATUS_NOT_IMPLEMENTED;
}

Referenced by Wow64EnableWow64FsRedirection().

RtlWow64EnableFsRedirectionEx()

NTSYSAPI NTSTATUS NTAPI RtlWow64EnableFsRedirectionEx	(	IN PVOID	Wow64FsEnableRedirection,
		OUT PVOID *	OldFsRedirectionLevel
	)

Definition at line 1014 of file libsupp.c.

{
    /* This is what Windows returns on x86 */
    return STATUS_NOT_IMPLEMENTED;
}

Referenced by Wow64DisableWow64FsRedirection(), and Wow64RevertWow64FsRedirection().

Variable Documentation

LdrpTopLevelDllBeingLoadedTeb

PTEB LdrpTopLevelDllBeingLoadedTeb = NULL

Definition at line 18 of file libsupp.c.

Referenced by LdrpRunInitializeRoutines(), and RtlIsThreadWithinLoaderCallout().

MmHighestUserAddress

PVOID MmHighestUserAddress = (PVOID)MI_HIGHEST_USER_ADDRESS

Definition at line 19 of file libsupp.c.

RtlpAllocDeallocQueryBufferSize

SIZE_T RtlpAllocDeallocQueryBufferSize = PAGE_SIZE

Definition at line 17 of file libsupp.c.