1*cdf0e10cSrcweir #include "quickstarter.hxx" 2*cdf0e10cSrcweir #ifdef _MSC_VER 3*cdf0e10cSrcweir #pragma warning(push, 1) /* disable warnings within system headers */ 4*cdf0e10cSrcweir #endif 5*cdf0e10cSrcweir #include <psapi.h> 6*cdf0e10cSrcweir #ifdef _MSC_VER 7*cdf0e10cSrcweir #pragma warning(pop) 8*cdf0e10cSrcweir #endif 9*cdf0e10cSrcweir #include <tlhelp32.h> 10*cdf0e10cSrcweir #include <malloc.h> 11*cdf0e10cSrcweir 12*cdf0e10cSrcweir std::string GetOfficeInstallationPath(MSIHANDLE handle) 13*cdf0e10cSrcweir { 14*cdf0e10cSrcweir std::string progpath; 15*cdf0e10cSrcweir DWORD sz = 0; 16*cdf0e10cSrcweir LPTSTR dummy = TEXT(""); 17*cdf0e10cSrcweir 18*cdf0e10cSrcweir if (MsiGetProperty(handle, TEXT("INSTALLLOCATION"), dummy, &sz) == ERROR_MORE_DATA) 19*cdf0e10cSrcweir { 20*cdf0e10cSrcweir sz++; // space for the final '\0' 21*cdf0e10cSrcweir DWORD nbytes = sz * sizeof(TCHAR); 22*cdf0e10cSrcweir LPTSTR buff = reinterpret_cast<LPTSTR>(_alloca(nbytes)); 23*cdf0e10cSrcweir ZeroMemory(buff, nbytes); 24*cdf0e10cSrcweir MsiGetProperty(handle, TEXT("INSTALLLOCATION"), buff, &sz); 25*cdf0e10cSrcweir progpath = buff; 26*cdf0e10cSrcweir } 27*cdf0e10cSrcweir return progpath; 28*cdf0e10cSrcweir } 29*cdf0e10cSrcweir 30*cdf0e10cSrcweir std::string GetOfficeProductName(MSIHANDLE handle) 31*cdf0e10cSrcweir { 32*cdf0e10cSrcweir std::string productname; 33*cdf0e10cSrcweir DWORD sz = 0; 34*cdf0e10cSrcweir LPTSTR dummy = TEXT(""); 35*cdf0e10cSrcweir 36*cdf0e10cSrcweir if (MsiGetProperty(handle, TEXT("ProductName"), dummy, &sz) == ERROR_MORE_DATA) 37*cdf0e10cSrcweir { 38*cdf0e10cSrcweir sz++; // space for the final '\0' 39*cdf0e10cSrcweir DWORD nbytes = sz * sizeof(TCHAR); 40*cdf0e10cSrcweir LPTSTR buff = reinterpret_cast<LPTSTR>(_alloca(nbytes)); 41*cdf0e10cSrcweir ZeroMemory(buff, nbytes); 42*cdf0e10cSrcweir MsiGetProperty(handle, TEXT("ProductName"), buff, &sz); 43*cdf0e10cSrcweir productname = buff; 44*cdf0e10cSrcweir } 45*cdf0e10cSrcweir return productname; 46*cdf0e10cSrcweir } 47*cdf0e10cSrcweir 48*cdf0e10cSrcweir std::string GetQuickstarterLinkName(MSIHANDLE handle) 49*cdf0e10cSrcweir { 50*cdf0e10cSrcweir std::string quickstarterlinkname; 51*cdf0e10cSrcweir DWORD sz = 0; 52*cdf0e10cSrcweir LPTSTR dummy = TEXT(""); 53*cdf0e10cSrcweir 54*cdf0e10cSrcweir if (MsiGetProperty(handle, TEXT("Quickstarterlinkname"), dummy, &sz) == ERROR_MORE_DATA) 55*cdf0e10cSrcweir { 56*cdf0e10cSrcweir sz++; // space for the final '\0' 57*cdf0e10cSrcweir DWORD nbytes = sz * sizeof(TCHAR); 58*cdf0e10cSrcweir LPTSTR buff = reinterpret_cast<LPTSTR>(_alloca(nbytes)); 59*cdf0e10cSrcweir ZeroMemory(buff, nbytes); 60*cdf0e10cSrcweir MsiGetProperty(handle, TEXT("Quickstarterlinkname"), buff, &sz); 61*cdf0e10cSrcweir quickstarterlinkname = buff; 62*cdf0e10cSrcweir } 63*cdf0e10cSrcweir else if (MsiGetProperty(handle, TEXT("ProductName"), dummy, &sz) == ERROR_MORE_DATA) 64*cdf0e10cSrcweir { 65*cdf0e10cSrcweir sz++; // space for the final '\0' 66*cdf0e10cSrcweir DWORD nbytes = sz * sizeof(TCHAR); 67*cdf0e10cSrcweir LPTSTR buff = reinterpret_cast<LPTSTR>(_alloca(nbytes)); 68*cdf0e10cSrcweir ZeroMemory(buff, nbytes); 69*cdf0e10cSrcweir MsiGetProperty(handle, TEXT("ProductName"), buff, &sz); 70*cdf0e10cSrcweir quickstarterlinkname = buff; 71*cdf0e10cSrcweir } 72*cdf0e10cSrcweir return quickstarterlinkname; 73*cdf0e10cSrcweir } 74*cdf0e10cSrcweir 75*cdf0e10cSrcweir inline bool IsValidHandle( HANDLE handle ) 76*cdf0e10cSrcweir { 77*cdf0e10cSrcweir return NULL != handle && INVALID_HANDLE_VALUE != handle; 78*cdf0e10cSrcweir } 79*cdf0e10cSrcweir 80*cdf0e10cSrcweir 81*cdf0e10cSrcweir static HANDLE WINAPI _CreateToolhelp32Snapshot( DWORD dwFlags, DWORD th32ProcessID ) 82*cdf0e10cSrcweir { 83*cdf0e10cSrcweir typedef HANDLE (WINAPI *FN_PROC)( DWORD dwFlags, DWORD th32ProcessID ); 84*cdf0e10cSrcweir static FN_PROC lpProc = NULL; 85*cdf0e10cSrcweir 86*cdf0e10cSrcweir HANDLE hSnapshot = NULL; 87*cdf0e10cSrcweir 88*cdf0e10cSrcweir if ( !lpProc ) 89*cdf0e10cSrcweir { 90*cdf0e10cSrcweir HMODULE hLibrary = GetModuleHandle("KERNEL32.DLL"); 91*cdf0e10cSrcweir 92*cdf0e10cSrcweir if ( hLibrary ) 93*cdf0e10cSrcweir lpProc = reinterpret_cast< FN_PROC >(GetProcAddress( hLibrary, "CreateToolhelp32Snapshot" )); 94*cdf0e10cSrcweir } 95*cdf0e10cSrcweir 96*cdf0e10cSrcweir if ( lpProc ) 97*cdf0e10cSrcweir hSnapshot = lpProc( dwFlags, th32ProcessID ); 98*cdf0e10cSrcweir 99*cdf0e10cSrcweir return hSnapshot; 100*cdf0e10cSrcweir } 101*cdf0e10cSrcweir 102*cdf0e10cSrcweir static BOOL WINAPI _Process32First( HANDLE hSnapshot, PROCESSENTRY32 *lppe32 ) 103*cdf0e10cSrcweir { 104*cdf0e10cSrcweir typedef BOOL (WINAPI *FN_PROC)( HANDLE hSnapshot, PROCESSENTRY32 *lppe32 ); 105*cdf0e10cSrcweir static FN_PROC lpProc = NULL; 106*cdf0e10cSrcweir 107*cdf0e10cSrcweir BOOL fSuccess = FALSE; 108*cdf0e10cSrcweir 109*cdf0e10cSrcweir if ( !lpProc ) 110*cdf0e10cSrcweir { 111*cdf0e10cSrcweir HMODULE hLibrary = GetModuleHandle("KERNEL32.DLL"); 112*cdf0e10cSrcweir 113*cdf0e10cSrcweir if ( hLibrary ) 114*cdf0e10cSrcweir lpProc = reinterpret_cast< FN_PROC >(GetProcAddress( hLibrary, "Process32First" )); 115*cdf0e10cSrcweir } 116*cdf0e10cSrcweir 117*cdf0e10cSrcweir if ( lpProc ) 118*cdf0e10cSrcweir fSuccess = lpProc( hSnapshot, lppe32 ); 119*cdf0e10cSrcweir 120*cdf0e10cSrcweir return fSuccess; 121*cdf0e10cSrcweir } 122*cdf0e10cSrcweir 123*cdf0e10cSrcweir static BOOL WINAPI _Process32Next( HANDLE hSnapshot, PROCESSENTRY32 *lppe32 ) 124*cdf0e10cSrcweir { 125*cdf0e10cSrcweir typedef BOOL (WINAPI *FN_PROC)( HANDLE hSnapshot, PROCESSENTRY32 *lppe32 ); 126*cdf0e10cSrcweir static FN_PROC lpProc = NULL; 127*cdf0e10cSrcweir 128*cdf0e10cSrcweir BOOL fSuccess = FALSE; 129*cdf0e10cSrcweir 130*cdf0e10cSrcweir if ( !lpProc ) 131*cdf0e10cSrcweir { 132*cdf0e10cSrcweir HMODULE hLibrary = GetModuleHandle("KERNEL32.DLL"); 133*cdf0e10cSrcweir 134*cdf0e10cSrcweir if ( hLibrary ) 135*cdf0e10cSrcweir lpProc = reinterpret_cast< FN_PROC >(GetProcAddress( hLibrary, "Process32Next" )); 136*cdf0e10cSrcweir } 137*cdf0e10cSrcweir 138*cdf0e10cSrcweir if ( lpProc ) 139*cdf0e10cSrcweir fSuccess = lpProc( hSnapshot, lppe32 ); 140*cdf0e10cSrcweir 141*cdf0e10cSrcweir return fSuccess; 142*cdf0e10cSrcweir } 143*cdf0e10cSrcweir 144*cdf0e10cSrcweir static std::string GetProcessImagePath_9x( DWORD dwProcessId ) 145*cdf0e10cSrcweir { 146*cdf0e10cSrcweir std::string sImagePath; 147*cdf0e10cSrcweir 148*cdf0e10cSrcweir HANDLE hSnapshot = _CreateToolhelp32Snapshot( TH32CS_SNAPPROCESS, 0 ); 149*cdf0e10cSrcweir 150*cdf0e10cSrcweir if ( IsValidHandle( hSnapshot ) ) 151*cdf0e10cSrcweir { 152*cdf0e10cSrcweir PROCESSENTRY32 pe32 = { 0 }; 153*cdf0e10cSrcweir 154*cdf0e10cSrcweir pe32.dwSize = sizeof(PROCESSENTRY32); 155*cdf0e10cSrcweir 156*cdf0e10cSrcweir BOOL fSuccess = _Process32First( hSnapshot, &pe32 ); 157*cdf0e10cSrcweir bool found = false; 158*cdf0e10cSrcweir 159*cdf0e10cSrcweir while ( !found && fSuccess ) 160*cdf0e10cSrcweir { 161*cdf0e10cSrcweir if ( pe32.th32ProcessID == dwProcessId ) 162*cdf0e10cSrcweir { 163*cdf0e10cSrcweir found = true; 164*cdf0e10cSrcweir sImagePath = pe32.szExeFile; 165*cdf0e10cSrcweir } 166*cdf0e10cSrcweir 167*cdf0e10cSrcweir if ( !found ) 168*cdf0e10cSrcweir fSuccess = _Process32Next( hSnapshot, &pe32 ); 169*cdf0e10cSrcweir } 170*cdf0e10cSrcweir 171*cdf0e10cSrcweir CloseHandle( hSnapshot ); 172*cdf0e10cSrcweir } 173*cdf0e10cSrcweir 174*cdf0e10cSrcweir return sImagePath; 175*cdf0e10cSrcweir } 176*cdf0e10cSrcweir 177*cdf0e10cSrcweir static DWORD WINAPI _GetModuleFileNameExA( HANDLE hProcess, HMODULE hModule, LPSTR lpFileName, DWORD nSize ) 178*cdf0e10cSrcweir { 179*cdf0e10cSrcweir typedef DWORD (WINAPI *FN_PROC)( HANDLE hProcess, HMODULE hModule, LPSTR lpFileName, DWORD nSize ); 180*cdf0e10cSrcweir 181*cdf0e10cSrcweir static FN_PROC lpProc = NULL; 182*cdf0e10cSrcweir 183*cdf0e10cSrcweir if ( !lpProc ) 184*cdf0e10cSrcweir { 185*cdf0e10cSrcweir HMODULE hLibrary = LoadLibrary("PSAPI.DLL"); 186*cdf0e10cSrcweir 187*cdf0e10cSrcweir if ( hLibrary ) 188*cdf0e10cSrcweir lpProc = reinterpret_cast< FN_PROC >(GetProcAddress( hLibrary, "GetModuleFileNameExA" )); 189*cdf0e10cSrcweir } 190*cdf0e10cSrcweir 191*cdf0e10cSrcweir if ( lpProc ) 192*cdf0e10cSrcweir return lpProc( hProcess, hModule, lpFileName, nSize ); 193*cdf0e10cSrcweir 194*cdf0e10cSrcweir return 0; 195*cdf0e10cSrcweir 196*cdf0e10cSrcweir } 197*cdf0e10cSrcweir 198*cdf0e10cSrcweir static std::string GetProcessImagePath_NT( DWORD dwProcessId ) 199*cdf0e10cSrcweir { 200*cdf0e10cSrcweir std::string sImagePath; 201*cdf0e10cSrcweir 202*cdf0e10cSrcweir HANDLE hProcess = OpenProcess( PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, dwProcessId ); 203*cdf0e10cSrcweir 204*cdf0e10cSrcweir if ( IsValidHandle( hProcess ) ) 205*cdf0e10cSrcweir { 206*cdf0e10cSrcweir CHAR szPathBuffer[MAX_PATH] = ""; 207*cdf0e10cSrcweir 208*cdf0e10cSrcweir if ( _GetModuleFileNameExA( hProcess, NULL, szPathBuffer, sizeof(szPathBuffer) ) ) 209*cdf0e10cSrcweir sImagePath = szPathBuffer; 210*cdf0e10cSrcweir 211*cdf0e10cSrcweir CloseHandle( hProcess ); 212*cdf0e10cSrcweir } 213*cdf0e10cSrcweir 214*cdf0e10cSrcweir return sImagePath; 215*cdf0e10cSrcweir } 216*cdf0e10cSrcweir 217*cdf0e10cSrcweir std::string GetProcessImagePath( DWORD dwProcessId ) 218*cdf0e10cSrcweir { 219*cdf0e10cSrcweir return (LONG)GetVersion() < 0 ? GetProcessImagePath_9x( dwProcessId ) : GetProcessImagePath_NT( dwProcessId ); 220*cdf0e10cSrcweir } 221*cdf0e10cSrcweir 222