rtlinit.c File Reference

#include "precomp.h"
#include "ntimage.h"

Go to the source code of this file.

Defines
#define	MOVE_TO_RTL
#define	AllocateFromZone(Zone)
#define	FreeToZone(Zone, Block)
Functions
BOOL	RtlCaptureAnsiString (PIN_STRING pstr, LPCSTR psz, BOOL fForceAlloc)
BOOL	RtlCaptureLargeAnsiString (PLARGE_IN_STRING plstr, LPCSTR psz, BOOL fForceAlloc)
NTSTATUS	InitLookaside (PLOOKASIDE pla, DWORD cbEntry, DWORD cEntries)
PVOID	AllocLookasideEntry (PLOOKASIDE pla)
void	FreeLookasideEntry (PLOOKASIDE pla, PVOID pEntry)

---

Define Documentation

#define AllocateFromZone (  Zone   )

Value: (PVOID)((Zone)->FreeList.Next); \ if ( (Zone)->FreeList.Next ) (Zone)->FreeList.Next = (Zone)->FreeList.Next->Next Definition at line 171 of file rtlinit.c. Referenced by AllocLookasideEntry().

#define FreeToZone (  Zone, Block   )

Value: ( ((PSINGLE_LIST_ENTRY)(Block))->Next = (Zone)->FreeList.Next, \ (Zone)->FreeList.Next = ((PSINGLE_LIST_ENTRY)(Block)), \ ((PSINGLE_LIST_ENTRY)(Block))->Next \ ) Definition at line 205 of file rtlinit.c. Referenced by FreeLookasideEntry().

#define MOVE_TO_RTL

Definition at line 14 of file rtlinit.c.

---

Function Documentation

PVOID AllocLookasideEntry (  PLOOKASIDE  pla  )

Definition at line 299 of file rtlinit.c. References AllocateFromZone, _LOOKASIDE::EntrySize, gcsLookaside, _LOOKASIDE::LookasideZone, NULL, and UserLocalAlloc. Referenced by ValidateDialogPwnd(). { PVOID pEntry; // // Attempt to get an entry from the zone. If this fails, then // LocalAlloc the entry // RtlEnterCriticalSection(&gcsLookaside); pEntry = AllocateFromZone(&pla->LookasideZone); RtlLeaveCriticalSection(&gcsLookaside); if ( !pEntry ) { /* * Allocate a local structure. */ #if DBG pla->AllocSlowCalls++; #endif // DBG if ((pEntry = UserLocalAlloc(0, pla->EntrySize)) == NULL) return NULL; } RtlZeroMemory(pEntry, pla->EntrySize); #if DBG pla->AllocCalls++; if (pla->AllocCalls - pla->DelCalls > pla->AllocHiWater ) { pla->AllocHiWater = pla->AllocCalls - pla->DelCalls; } #endif // DBG return pEntry; }

void FreeLookasideEntry (  PLOOKASIDE  pla, PVOID  pEntry )

Definition at line 344 of file rtlinit.c. References FreeToZone, gcsLookaside, _LOOKASIDE::LookasideBase, _LOOKASIDE::LookasideBounds, _LOOKASIDE::LookasideZone, and UserLocalFree. Referenced by ButtonWndProcWorker(), DefDlgProcWorker(), ECNcCreate(), ECNcDestroyHandler(), ImeWndProcWorker(), StaticWndProcWorker(), xxxCBNcDestroyHandler(), and xxxDestroyLBox(). { #if DBG pla->DelCalls++; #endif // DBG // // If the pEntry was from zone, then free to zone // if ( (PVOID)pEntry >= pla->LookasideBase && (PVOID)pEntry < pla->LookasideBounds ) { RtlEnterCriticalSection(&gcsLookaside); FreeToZone(&pla->LookasideZone,pEntry); RtlLeaveCriticalSection(&gcsLookaside); } else { #if DBG pla->DelSlowCalls++; #endif // DBG UserLocalFree(pEntry); } }

NTSTATUS InitLookaside (  PLOOKASIDE  pla, DWORD  cbEntry, DWORD  cEntries )

Definition at line 221 of file rtlinit.c. References BlockSize, _ZONE_HEADER::BlockSize, _LOOKASIDE::EntrySize, _ZONE_HEADER::FreeList, gcsLookaside, _LOOKASIDE::LookasideBase, _LOOKASIDE::LookasideBounds, _LOOKASIDE::LookasideZone, NULL, PLOOKASIDE, PZONE_HEADER, _ZONE_HEADER::SegmentList, _ZONE_HEADER::TotalSegmentSize, UserLocalAlloc, UserLocalFree, and ZONE_SEGMENT_HEADER. Referenced by ValidateDialogPwnd(). { ULONG i; PCH p; ULONG BlockSize; PZONE_HEADER Zone; PVOID InitialSegment; ULONG InitialSegmentSize; InitialSegmentSize = (cEntries * cbEntry) + sizeof(ZONE_SEGMENT_HEADER); p = (PCH)UserLocalAlloc(0, InitialSegmentSize); if ( !p ) { return STATUS_NO_MEMORY; } RtlEnterCriticalSection(&gcsLookaside); // // If the lookaside list has already been initialized, we're done. // if (pla->LookasideBase != NULL && pla->EntrySize == cbEntry) { RtlLeaveCriticalSection(&gcsLookaside); UserLocalFree(p); return STATUS_SUCCESS; } pla->LookasideBase = (PVOID)p; pla->LookasideBounds = (PVOID)(p + InitialSegmentSize); pla->EntrySize = cbEntry; // // Using the ExZone-like code, slice up the page into QMSG's // Zone = &pla->LookasideZone; BlockSize = cbEntry; InitialSegment = pla->LookasideBase; Zone->BlockSize = BlockSize; Zone->SegmentList.Next = &((PZONE_SEGMENT_HEADER) InitialSegment)->SegmentList; ((PZONE_SEGMENT_HEADER) InitialSegment)->SegmentList.Next = NULL; ((PZONE_SEGMENT_HEADER) InitialSegment)->Reserved = NULL; Zone->FreeList.Next = NULL; p = (PCH)InitialSegment + sizeof(ZONE_SEGMENT_HEADER); for (i = sizeof(ZONE_SEGMENT_HEADER); i <= InitialSegmentSize - BlockSize; i += BlockSize ) { ((PSINGLE_LIST_ENTRY)p)->Next = Zone->FreeList.Next; Zone->FreeList.Next = (PSINGLE_LIST_ENTRY)p; p += BlockSize; } Zone->TotalSegmentSize = i; RtlLeaveCriticalSection(&gcsLookaside); return STATUS_SUCCESS; }

BOOL RtlCaptureAnsiString (  PIN_STRING  pstr, LPCSTR  psz, BOOL  fForceAlloc )

Definition at line 30 of file rtlinit.c. References BOOL, _IN_STRING::fAllocated, FALSE, NULL, _IN_STRING::pstr, pUserHeap, RtlAllocateHeap, RtlFreeHeap, RtlMultiByteToUnicodeN(), _IN_STRING::strCapture, strlen(), TRUE, and USHORT. { int cbSrc; int cbDst; pstr->fAllocated = FALSE; if (psz) { cbSrc = strlen(psz) + 1; if (cbSrc > MAXUSHORT) { RIPMSG0(RIP_WARNING, "String too long for standard string"); return FALSE; } /* * If the allocation is forced or if the string is * too long to fit in the TEB, allocate a buffer. * Otherwise, store the result in the TEB. */ if (fForceAlloc || cbSrc > (STATIC_UNICODE_BUFFER_LENGTH / sizeof(WCHAR))) { pstr->strCapture.Buffer = RtlAllocateHeap(pUserHeap, 0, cbSrc * sizeof(WCHAR)); if (pstr->strCapture.Buffer == NULL) return FALSE; pstr->fAllocated = TRUE; pstr->pstr = &pstr->strCapture; pstr->strCapture.MaximumLength = (USHORT)(cbSrc * sizeof(WCHAR)); } else { pstr->pstr = &NtCurrentTeb()->StaticUnicodeString; } /* * Convert the string to Unicode */ if (RtlMultiByteToUnicodeN(pstr->pstr->Buffer, (ULONG)pstr->pstr->MaximumLength, &cbDst, (LPSTR)psz, cbSrc)) { RIPMSG0(RIP_WARNING, "Unicode conversion failed"); if (pstr->fAllocated) { RtlFreeHeap(pUserHeap, 0, pstr->strCapture.Buffer); pstr->fAllocated = FALSE; } return FALSE; } pstr->pstr->Length = (USHORT)cbDst - sizeof(WCHAR); } else { pstr->pstr = &pstr->strCapture; pstr->strCapture.Length = pstr->strCapture.MaximumLength = 0; pstr->strCapture.Buffer = NULL; } return TRUE; }

BOOL RtlCaptureLargeAnsiString (  PLARGE_IN_STRING  plstr, LPCSTR  psz, BOOL  fForceAlloc )

Definition at line 95 of file rtlinit.c. References BOOL, _LARGE_UNICODE_STRING::Buffer, _LARGE_IN_STRING::fAllocated, FALSE, _LARGE_UNICODE_STRING::Length, _LARGE_UNICODE_STRING::MaximumLength, NULL, PLARGE_IN_STRING, _LARGE_IN_STRING::pstr, pUserHeap, RtlAllocateHeap, RtlFreeHeap, RtlMultiByteToUnicodeN(), _LARGE_IN_STRING::strCapture, strlen(), TRUE, and UINT. Referenced by _CreateWindowEx(). { int cchSrc; UINT uLength; plstr->fAllocated = FALSE; plstr->pstr = &plstr->strCapture; if (psz) { cchSrc = strlen(psz) + 1; /* * If the allocation is forced or if the string is * too long to fit in the TEB, allocate a buffer. * Otherwise, store the result in the TEB. */ if (fForceAlloc || cchSrc > STATIC_UNICODE_BUFFER_LENGTH) { plstr->strCapture.Buffer = RtlAllocateHeap(pUserHeap, 0, cchSrc * sizeof(WCHAR)); if (plstr->strCapture.Buffer == NULL) return FALSE; plstr->fAllocated = TRUE; plstr->strCapture.MaximumLength = cchSrc * sizeof(WCHAR); } else { plstr->strCapture.Buffer = NtCurrentTeb()->StaticUnicodeBuffer; plstr->strCapture.MaximumLength = (UINT)(STATIC_UNICODE_BUFFER_LENGTH * sizeof(WCHAR)); } /* * Convert the string to Unicode */ if (RtlMultiByteToUnicodeN(plstr->pstr->Buffer, plstr->pstr->MaximumLength, &uLength, (LPSTR)psz, cchSrc)) { RIPMSG0(RIP_WARNING, "Unicode conversion failed"); if (plstr->fAllocated) { RtlFreeHeap(pUserHeap, 0, plstr->strCapture.Buffer); plstr->fAllocated = FALSE; } return FALSE; } plstr->pstr->Length = uLength - sizeof(WCHAR); } else { plstr->strCapture.Length = plstr->strCapture.MaximumLength = 0; plstr->strCapture.Buffer = NULL; } return TRUE; }

---