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.
430 lines
13 KiB
430 lines
13 KiB
//
|
|
// 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
|
|
}
|
|
|
|
|
|
|