|
|
//
// APITHK.C
//
// This file has API thunks that allow shell32 to load and run on
// multiple versions of NT or Win95. Since this component needs
// to load on the base-level NT 4.0 and Win95, any calls to system
// APIs introduced in later OS versions must be done via GetProcAddress.
//
// Also, any code that may need to access data structures that are
// post-4.0 specific can be added here.
//
// NOTE: this file does *not* use the standard precompiled header,
// so it can set _WIN32_WINNT to a later version.
//
#include "priv.h" // Don't use precompiled header here
#include "appwiz.h"
#define c_szARPJob TEXT("ARP Job")
// Return: hIOPort for the CompletionPort
HANDLE _SetJobCompletionPort(HANDLE hJob){ HANDLE hRet = NULL;
HANDLE hIOPort = CreateIoCompletionPort( INVALID_HANDLE_VALUE, NULL, (ULONG_PTR)hJob, 1 ); if ( hIOPort != NULL ) { JOBOBJECT_ASSOCIATE_COMPLETION_PORT CompletionPort; CompletionPort.CompletionKey = hJob ;
CompletionPort.CompletionPort = hIOPort;
if (SetInformationJobObject( hJob,JobObjectAssociateCompletionPortInformation, &CompletionPort, sizeof(CompletionPort) ) ) { hRet = hIOPort; } } return hRet;}
STDAPI_(DWORD) WaitingThreadProc(void *pv){ HANDLE hIOPort = (HANDLE)pv;
// RIP(hIOPort);
DWORD dwCompletionCode; PVOID pCompletionKey; LPOVERLAPPED lpOverlapped; while (TRUE) { // Wait for all the processes to finish...
if (!GetQueuedCompletionStatus( hIOPort, &dwCompletionCode, (PULONG_PTR) &pCompletionKey, &lpOverlapped, INFINITE ) || (dwCompletionCode == JOB_OBJECT_MSG_ACTIVE_PROCESS_ZERO)) { break; } } return 0;}
/*-------------------------------------------------------------------------
Purpose: Creates a process and waits for it to finish*/STDAPI_(BOOL) NT5_CreateAndWaitForProcess(LPTSTR pszExeName){ PROCESS_INFORMATION pi = {0}; STARTUPINFO si = {0}; BOOL fWorked = FALSE;#ifdef WX86
DWORD cchArch; WCHAR szArchValue[32];#endif
// CreateJobObject does not follow the win32 convention because even if the function succeeds, it can
// still SetLastError to ERROR_ALREADY_EXISTS
HANDLE hJob = CreateJobObject(NULL, c_szARPJob); if (hJob) { if (GetLastError() != ERROR_ALREADY_EXISTS) { HANDLE hIOPort = _SetJobCompletionPort(hJob); if (hIOPort) { DWORD dwCreationFlags = 0; // Create the install process
si.cb = sizeof(si);
#ifdef WX86
if (bWx86Enabled && bForceX86Env) { cchArch = GetEnvironmentVariableW(ProcArchName, szArchValue, sizeof(szArchValue) );
if (!cchArch || cchArch >= sizeof(szArchValue)) { szArchValue[0]=L'\0'; }
SetEnvironmentVariableW(ProcArchName, L"x86"); } #endif
dwCreationFlags = CREATE_SUSPENDED | CREATE_SEPARATE_WOW_VDM; // Create the process
fWorked = CreateProcess(NULL, pszExeName, NULL, NULL, FALSE, dwCreationFlags, NULL, NULL, &si, &pi); if (fWorked) { HANDLE hWait = NULL; if (AssignProcessToJobObject(hJob, pi.hProcess)) { hWait = CreateThread(NULL, 0, WaitingThreadProc, (LPVOID)hIOPort, 0, NULL); }
if (hWait == NULL) { // We might get here if the call to AssignProcessToJobObject has failed because
// the process already has a job assigned to it, or because we couldn't create the
// waiting thread. Try a more direct approach by just watching the process handle.
// This method won't catch spawned processes, but it is better than nothing.
hWait = pi.hProcess; } else { // we are not waiting on the process handle, so we are done /w it.
CloseHandle(pi.hProcess); }
ResumeThread(pi.hThread); CloseHandle(pi.hThread);
#ifdef WX86
if (bWx86Enabled && bForceX86Env) { SetEnvironmentVariableW(ProcArchName, szArchValue); } #endif
// we should have a valid handle at this point for sure
ASSERT(hWait && (hWait != INVALID_HANDLE_VALUE));
SHProcessSentMessagesUntilEvent(NULL, hWait, INFINITE); CloseHandle(hWait); }
CloseHandle(hIOPort); } }
CloseHandle(hJob); }
return fWorked;}
#define PFN_FIRSTTIME ((void *)-1)
// VerSetConditionMask
typedef ULONGLONG (WINAPI * PFNVERSETCONDITIONMASK)(ULONGLONG conditionMask, DWORD dwTypeMask, BYTE condition);
/*----------------------------------------------------------
Purpose: Thunk for NT 5's VerSetConditionMask*/ULONGLONG NT5_VerSetConditionMask(ULONGLONG conditionMask, DWORD dwTypeMask, BYTE condition){ static PFNVERSETCONDITIONMASK s_pfn = PFN_FIRSTTIME;
if (PFN_FIRSTTIME == s_pfn) { // It is safe to GetModuleHandle KERNEL32 because we implicitly link
// to it, so it is guaranteed to be loaded in every thread.
HINSTANCE hinst = GetModuleHandleA("KERNEL32.DLL");
if (hinst) s_pfn = (PFNVERSETCONDITIONMASK)GetProcAddress(hinst, "VerSetConditionMask"); else s_pfn = NULL; }
if (s_pfn) return s_pfn(conditionMask, dwTypeMask, condition);
return 0; // failure
}
typedef HRESULT (__stdcall * PFNRELEASEAPPCATEGORYINFOLIST)(APPCATEGORYINFOLIST *pAppCategoryList);
/*----------------------------------------------------------
Purpose: Thunk for NT 5's ReleaseAppCategoryInfoList*/HRESULT NT5_ReleaseAppCategoryInfoList(APPCATEGORYINFOLIST *pAppCategoryList){ static PFNRELEASEAPPCATEGORYINFOLIST s_pfn = PFN_FIRSTTIME;
if (PFN_FIRSTTIME == s_pfn) { HINSTANCE hinst = LoadLibraryA("APPMGMTS.DLL");
if (hinst) s_pfn = (PFNRELEASEAPPCATEGORYINFOLIST)GetProcAddress(hinst, "ReleaseAppCategoryInfoList"); else s_pfn = NULL; }
if (s_pfn) return s_pfn(pAppCategoryList);
return E_NOTIMPL; }
/*----------------------------------------------------------
Purpose: Thunk for NT 5's AllowSetForegroundWindow*/typedef UINT (WINAPI * PFNALLOWSFW) (DWORD dwPRocessID);
BOOL NT5_AllowSetForegroundWindow(DWORD dwProcessID){ static PFNALLOWSFW s_pfn = PFN_FIRSTTIME;
if (PFN_FIRSTTIME == s_pfn) { HINSTANCE hinst = LoadLibraryA("USER32.DLL");
if (hinst) { s_pfn = (PFNALLOWSFW)GetProcAddress(hinst, "AllowSetForegroundWindow"); } else s_pfn = NULL; }
if (s_pfn) return s_pfn(dwProcessID);
return FALSE;}
// InstallApplication
typedef DWORD (WINAPI * PFNINSTALLAPP)(PINSTALLDATA pInstallInfo);
/*----------------------------------------------------------
Purpose: Thunk for NT 5's InstallApplication*/DWORD NT5_InstallApplication(PINSTALLDATA pInstallInfo){ static PFNINSTALLAPP s_pfn = PFN_FIRSTTIME;
if (PFN_FIRSTTIME == s_pfn) { // It is safe to GetModuleHandle ADVAPI32 because we implicitly link
// to it, so it is guaranteed to be loaded in every thread.
HINSTANCE hinst = GetModuleHandleA("ADVAPI32.DLL");
if (hinst) s_pfn = (PFNINSTALLAPP)GetProcAddress(hinst, "InstallApplication"); else s_pfn = NULL; }
if (s_pfn) return s_pfn(pInstallInfo);
return ERROR_INVALID_FUNCTION; // failure
}
// UninstallApplication
typedef DWORD (WINAPI * PFNUNINSTALLAPP)(WCHAR * pszProductCode, DWORD dwStatus);
/*----------------------------------------------------------
Purpose: Thunk for NT 5's UninstallApplication*/DWORD NT5_UninstallApplication(WCHAR * pszProductCode, DWORD dwStatus){ static PFNUNINSTALLAPP s_pfn = PFN_FIRSTTIME;
if (PFN_FIRSTTIME == s_pfn) { // It is safe to GetModuleHandle ADVAPI32 because we implicitly link
// to it, so it is guaranteed to be loaded in every thread.
HINSTANCE hinst = GetModuleHandleA("ADVAPI32.DLL");
if (hinst) s_pfn = (PFNUNINSTALLAPP)GetProcAddress(hinst, "UninstallApplication"); else s_pfn = NULL; }
if (s_pfn) return s_pfn(pszProductCode, dwStatus);
return ERROR_INVALID_FUNCTION; // failure
}
// GetManagedApplications
typedef DWORD (WINAPI * PFNGETAPPS)(GUID * pCategory, DWORD dwQueryFlags, DWORD dwInfoLevel, LPDWORD pdwApps, PMANAGEDAPPLICATION* prgManagedApps);
/*----------------------------------------------------------
Purpose: Thunk for NT 5's GetManagedApplications*/DWORD NT5_GetManagedApplications(GUID * pCategory, DWORD dwQueryFlags, DWORD dwInfoLevel, LPDWORD pdwApps, PMANAGEDAPPLICATION* prgManagedApps){ static PFNGETAPPS s_pfn = PFN_FIRSTTIME;
if (PFN_FIRSTTIME == s_pfn) { // It is safe to GetModuleHandle ADVAPI32 because we implicitly link
// to it, so it is guaranteed to be loaded in every thread.
HINSTANCE hinst = GetModuleHandleA("ADVAPI32.DLL");
if (hinst) s_pfn = (PFNGETAPPS)GetProcAddress(hinst, "GetManagedApplications"); else s_pfn = NULL; }
if (s_pfn) return s_pfn(pCategory, dwQueryFlags, dwInfoLevel, pdwApps, prgManagedApps);
return ERROR_INVALID_FUNCTION; // failure
}
typedef DWORD (__stdcall * PFNGETMANAGEDAPPLICATIONCATEGORIES)(DWORD dwReserved, APPCATEGORYINFOLIST *pAppCategoryList);
/*----------------------------------------------------------
Purpose: Thunk for NT 5's CsGetAppCategories*/DWORD NT5_GetManagedApplicationCategories(DWORD dwReserved, APPCATEGORYINFOLIST *pAppCategoryList){ static PFNGETMANAGEDAPPLICATIONCATEGORIES s_pfn = PFN_FIRSTTIME;
if (PFN_FIRSTTIME == s_pfn) { HINSTANCE hinst = LoadLibraryA("ADVAPI32.DLL");
if (hinst) s_pfn = (PFNGETMANAGEDAPPLICATIONCATEGORIES)GetProcAddress(hinst, "GetManagedApplicationCategories"); else s_pfn = NULL; }
if (s_pfn) return s_pfn(dwReserved, pAppCategoryList);
return ERROR_INVALID_FUNCTION;}
// NetGetJoinInformation
typedef NET_API_STATUS (WINAPI * PFNGETJOININFO)(LPCWSTR lpServer, LPWSTR *lpNameBuffer, PNETSETUP_JOIN_STATUS BufferType);
/*----------------------------------------------------------
Purpose: Thunk for NT 5's NetGetJoinInformation*/NET_API_STATUS NT5_NetGetJoinInformation(LPCWSTR lpServer, LPWSTR *lpNameBuffer, PNETSETUP_JOIN_STATUS BufferType){ static PFNGETJOININFO s_pfn = PFN_FIRSTTIME;
if (PFN_FIRSTTIME == s_pfn) { HINSTANCE hinst = LoadLibraryA("NETAPI32.DLL");
if (hinst) s_pfn = (PFNGETJOININFO)GetProcAddress(hinst, "NetGetJoinInformation"); else s_pfn = NULL; }
if (s_pfn) return s_pfn(lpServer, lpNameBuffer, BufferType);
return NERR_NetNotStarted; // failure
}
// NetApiBufferFree
typedef NET_API_STATUS (WINAPI * PFNNETFREEBUFFER)(LPVOID lpBuffer);
/*----------------------------------------------------------
Purpose: Thunk for NT 5's NetApiBufferFree*/NET_API_STATUS NT5_NetApiBufferFree(LPVOID lpBuffer){ static PFNNETFREEBUFFER s_pfn = PFN_FIRSTTIME;
if (PFN_FIRSTTIME == s_pfn) { HINSTANCE hinst = LoadLibraryA("NETAPI32.DLL");
if (hinst) s_pfn = (PFNNETFREEBUFFER)GetProcAddress(hinst, "NetApiBufferFree"); else s_pfn = NULL; }
if (s_pfn) return s_pfn(lpBuffer);
return NERR_NetNotStarted; // failure
}
|
