Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

481 lines
12 KiB

/*************** ************************************************************/
// tsappcmp.c
//
// Copyright (C) 1997-1999 Microsoft Corp.
/****************************************************************************/
#include "precomp.h"
#pragma hdrstop
extern BOOL InitRegisterSupport();
PWINSTATIONQUERYINFORMATIONW pWinStationQueryInformationW;
BOOL GetDefaultUserProfileName(LPWSTR lpProfileDir, LPDWORD lpcchSize);
extern WCHAR gpwszDefaultUserName[];
extern void FreeLDRTable();
DWORD g_dwFlags=0;
/*
* Read flags, if flags don't exit, then assume default behavior.
* The default behavior is the same as dwFlags = 0x0
* The default behavior will result is the loadlib func calls to be patched by our
* redirected func TLoadLibraryExW().
*
*/
void ReadImportTablePatchFLagsAndAppCompatMode( DWORD *pdwFlags, BOOLEAN *pInAppCompatMode )
{
NTSTATUS NtStatus;
UNICODE_STRING KeyName;
OBJECT_ATTRIBUTES ObjectAttributes;
HANDLE KeyHandle;
UCHAR Buffer[100];
PKEY_VALUE_PARTIAL_INFORMATION KeyValueInformation = (PKEY_VALUE_PARTIAL_INFORMATION) Buffer;
ULONG KeyValueLength = 100;
ULONG ResultLength;
*pdwFlags=0;
*pInAppCompatMode=FALSE;
RtlInitUnicodeString(
&KeyName,
L"\\Registry\\Machine\\System\\CurrentControlSet\\Control\\Terminal Server"
);
InitializeObjectAttributes(
&ObjectAttributes,
&KeyName,
OBJ_CASE_INSENSITIVE,
0,
NULL
);
NtStatus = NtOpenKey(
&KeyHandle,
KEY_READ,
&ObjectAttributes
);
if (!NT_SUCCESS(NtStatus)) {
return; // so nothing found, just return since we do have a default behavior.
}
RtlInitUnicodeString(
&KeyName,
REG_TERMSRV_APPCOMPAT
);
NtStatus = NtQueryValueKey(
KeyHandle,
&KeyName,
KeyValuePartialInformation,
KeyValueInformation,
KeyValueLength,
&ResultLength
);
if (NT_SUCCESS(NtStatus))
{
//
// Check that the data is the correct size and type - a DWORD.
//
if ((KeyValueInformation->DataLength >= sizeof(DWORD)) &&
(KeyValueInformation->Type == REG_DWORD))
{
*pInAppCompatMode = * (PBOOLEAN) KeyValueInformation->Data;
}
}
RtlInitUnicodeString(
&KeyName,
TERMSRV_COMPAT_IAT_FLAGS
);
NtStatus = NtQueryValueKey(
KeyHandle,
&KeyName,
KeyValuePartialInformation,
KeyValueInformation,
KeyValueLength,
&ResultLength
);
if (NT_SUCCESS(NtStatus))
{
//
// Check that the data is the correct size and type - a DWORD.
//
if ((KeyValueInformation->DataLength >= sizeof(DWORD)) &&
(KeyValueInformation->Type == REG_DWORD))
{
*pdwFlags = * (PDWORD) KeyValueInformation->Data;
}
}
NtClose(KeyHandle);
}
BOOL WINAPI LibMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
{
static ULONG attachCount=0;
static BOOLEAN inAppCompatMode=FALSE;
switch (dwReason)
{
case DLL_PROCESS_ATTACH:
{
PWCHAR pwch, pwchext;
WCHAR pwcAppName[MAX_PATH+1];
PRTL_USER_PROCESS_PARAMETERS pUserParam;
DWORD dwSize;
attachCount++;
ReadImportTablePatchFLagsAndAppCompatMode( &g_dwFlags , &inAppCompatMode ); // this will initialize our global flag for IAT and debug
if ( g_dwFlags & DEBUG_IAT )
{
DbgPrint("tsappcmp: LibMain: DLL_PROCESS_ATTACH called, attach count = %d \n", attachCount );
}
if ( inAppCompatMode )
{
// Get the path of the executable name
pUserParam = NtCurrentPeb()->ProcessParameters;
// Get the executable name, if there's no \ just use the name as it is
pwch = wcsrchr(pUserParam->ImagePathName.Buffer, L'\\');
if (pwch) {
pwch++;
} else {
pwch = pUserParam->ImagePathName.Buffer;
}
wcscpy(pwcAppName, pwch);
pwch = pwcAppName;
#if DBGX
DbgPrint("\nApp-name : %ws\n", pwch );
#endif
// Check if it's a DOS or Win16 app by checking if the app is ntvdm.exe
// Only disable ThreadLibrary calls if not ntvdm.
if (_wcsicmp(pwch, L"ntvdm.exe")) {
DisableThreadLibraryCalls (hInstance);
}
// Init support for the register command
if (!InitRegisterSupport())
{
DbgPrint("TSAPPCMP: LibMain: DLL_PROCESS_ATTACH - failed\n");
return FALSE;
}
dwSize = MAX_PATH;
if (!GetDefaultUserProfileName(gpwszDefaultUserName, &dwSize)) {
gpwszDefaultUserName[0] = L'\0';
}
}
break;
}
case DLL_THREAD_ATTACH:
break;
case DLL_PROCESS_DETACH:
{
attachCount--;
if ( g_dwFlags & DEBUG_IAT )
{
DbgPrint("tsappcmp: LibMain: DLL_PROCESS_DETACH called, attach count = %d \n", attachCount );
}
if (inAppCompatMode )
{
if (attachCount==0 )
{
FreeLDRTable();
}
}
}
break;
}
return TRUE;
}
PWCHAR TermsrvFormatObjectName(LPCWSTR OldName)
{
PWCHAR pstrNewObjName = NULL;
#if 0
SIZE_T Size;
Size = ( wcslen(OldName) * sizeof(WCHAR)) + sizeof(L"Local\\") + sizeof(WCHAR);
pstrNewObjName = RtlAllocateHeap(RtlProcessHeap(),
LMEM_FIXED | LMEM_ZEROINIT,
Size);
if (pstrNewObjName) {
swprintf(pstrNewObjName,L"Local\\%ws",OldName);
}
#endif
return pstrNewObjName;
}
DWORD TermsrvGetComputerName( LPWSTR lpBuffer, LPDWORD nSize )
{
ULONG ulCompatFlags=0, ulAppType = 0;
WINSTATIONINFORMATIONW WSInfo;
ULONG ValueLength;
HMODULE hwinsta = NULL;
GetCtxAppCompatFlags(&ulCompatFlags, &ulAppType);
// Return the username instead of the computername?
if ((ulCompatFlags & TERMSRV_COMPAT_USERNAME) &&
(ulCompatFlags & ulAppType)) {
if ( !pWinStationQueryInformationW ) {
/*
* Get handle to winsta.dll
*/
if ( (hwinsta = LoadLibraryA( "WINSTA" )) != NULL ) {
pWinStationQueryInformationW = (PWINSTATIONQUERYINFORMATIONW)
GetProcAddress( hwinsta, "WinStationQueryInformationW" );
}
}
// Fetch the WinStation's basic information
if ( pWinStationQueryInformationW ) {
if ( (*pWinStationQueryInformationW)(SERVERNAME_CURRENT,
LOGONID_CURRENT,
WinStationInformation,
&WSInfo,
sizeof(WSInfo),
&ValueLength ) ) {
// Check if username will fit in buffer
if (wcslen(WSInfo.UserName) >= *nSize) {
return ERROR_BUFFER_OVERFLOW;
} else {
wcscpy(lpBuffer, WSInfo.UserName);
return ERROR_SUCCESS;
}
}
}
}
return ERROR_RETRY;
}
void TermsrvAdjustPhyMemLimits (
IN OUT LPDWORD TotalPhys,
IN OUT LPDWORD AvailPhys,
IN DWORD SysPageSize
)
{
ULONG ulAppType = 0;
DWORD PhysicalMemory;
if ( GetCtxPhysMemoryLimits(&ulAppType, &PhysicalMemory) ) {
if (*TotalPhys > PhysicalMemory ) {
*TotalPhys = PhysicalMemory;
}
}
if ( *AvailPhys > *TotalPhys ) {
// Reset the Available Physical Memory to be smaller than the
// Total Physical Memory. It is made smaller to avoid
// possible divide by zero errors when Available and Total are
// equal
*AvailPhys = *TotalPhys - SysPageSize;
}
return;
}
UINT
APIENTRY
TermsrvGetWindowsDirectoryA(
LPSTR lpBuffer,
UINT uSize
)
{
ANSI_STRING AnsiString;
NTSTATUS Status;
ULONG cbAnsiString;
UNICODE_STRING Path;
//
// If in install mode return the system windows dir
//
if (TermsrvAppInstallMode()) {
return 0;
}
if (!TermsrvPerUserWinDirMapping()) {
return 0;
}
// if buffer looks real, then init it to zero
if ( lpBuffer ) {
*lpBuffer = '\0'; // in case we have an error, the shell folks want this to be null
// BUG 453487
}
Path.Length = 0;
Path.MaximumLength = (USHORT)(uSize * sizeof( WCHAR ));
if ( Path.Buffer = LocalAlloc( LPTR, Path.MaximumLength ) ) {
Status = GetPerUserWindowsDirectory( &Path );
if ( NT_SUCCESS(Status) ) {
AnsiString.MaximumLength = (USHORT)(uSize);
AnsiString.Buffer = lpBuffer;
Status = RtlUnicodeStringToAnsiString(
&AnsiString,
&Path,
FALSE
);
} else if ( (Status == STATUS_BUFFER_TOO_SMALL) || (Status == STATUS_BUFFER_OVERFLOW ) ) {
DbgPrint( "KERNEL32: GetWindowsDirectoryA: User buffer too small (%u) need(%u)\n",
uSize, Path.Length >> 1 );
return( Path.Length >> 1 );
}
LocalFree( Path.Buffer );
} else {
Status = STATUS_NO_MEMORY;
DbgPrint( "KERNEL32: GetWindowsDirectoryA: No memory\n" );
}
if ( Status == STATUS_BUFFER_TOO_SMALL ) {
DbgPrint( "KERNEL32: GetWindowsDirectoryA: User buffer too small (%u) need(%u)\n",
uSize, Path.Length >> 1 );
return( Path.Length >> 1 );
} else if ( !NT_SUCCESS(Status) ) {
return 0;
}
return AnsiString.Length;
}
UINT
APIENTRY
TermsrvGetWindowsDirectoryW(
LPWSTR lpBuffer,
UINT uSize
)
{
UNICODE_STRING Path;
NTSTATUS Status;
//
// If in install mode return the system windows dir
//
if (TermsrvAppInstallMode()) {
return 0;
}
if (!TermsrvPerUserWinDirMapping()) {
return 0;
}
// if buffer looks real, then init it to zero
if ( lpBuffer ) {
*lpBuffer = '\0'; // in case we have an error, the shell folks want this to be null
// BUG 453487
}
/*
* If it fails, return 0
* If buffer too small, return len (not includding NULL)
* If buffer ok, return len (not inc. NULL) and fill buffer
* (GetPerUserWindowsDirectory will do all of this for us!)
*/
Path.Length = 0;
Path.MaximumLength = (USHORT)(uSize * sizeof( WCHAR ));
Path.Buffer = lpBuffer;
Status = GetPerUserWindowsDirectory( &Path );
if ( Status == STATUS_SUCCESS ) {
/*
* Add a NULL to the end (if it fits!)
*/
if ( Path.Length + sizeof( WCHAR ) <= Path.MaximumLength ) {
lpBuffer[(Path.Length>>1)] = UNICODE_NULL;
}
}
return( Path.Length / sizeof(WCHAR) );
}