1*b1cdbd2cSJim Jagielski /************************************************************** 2*b1cdbd2cSJim Jagielski * 3*b1cdbd2cSJim Jagielski * Licensed to the Apache Software Foundation (ASF) under one 4*b1cdbd2cSJim Jagielski * or more contributor license agreements. See the NOTICE file 5*b1cdbd2cSJim Jagielski * distributed with this work for additional information 6*b1cdbd2cSJim Jagielski * regarding copyright ownership. The ASF licenses this file 7*b1cdbd2cSJim Jagielski * to you under the Apache License, Version 2.0 (the 8*b1cdbd2cSJim Jagielski * "License"); you may not use this file except in compliance 9*b1cdbd2cSJim Jagielski * with the License. You may obtain a copy of the License at 10*b1cdbd2cSJim Jagielski * 11*b1cdbd2cSJim Jagielski * http://www.apache.org/licenses/LICENSE-2.0 12*b1cdbd2cSJim Jagielski * 13*b1cdbd2cSJim Jagielski * Unless required by applicable law or agreed to in writing, 14*b1cdbd2cSJim Jagielski * software distributed under the License is distributed on an 15*b1cdbd2cSJim Jagielski * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 16*b1cdbd2cSJim Jagielski * KIND, either express or implied. See the License for the 17*b1cdbd2cSJim Jagielski * specific language governing permissions and limitations 18*b1cdbd2cSJim Jagielski * under the License. 19*b1cdbd2cSJim Jagielski * 20*b1cdbd2cSJim Jagielski *************************************************************/ 21*b1cdbd2cSJim Jagielski 22*b1cdbd2cSJim Jagielski #include "macros.h" 23*b1cdbd2cSJim Jagielski #ifdef _MSC_VER 24*b1cdbd2cSJim Jagielski #pragma warning(push,1) // disable warnings within system headers 25*b1cdbd2cSJim Jagielski #endif 26*b1cdbd2cSJim Jagielski #include <psapi.h> 27*b1cdbd2cSJim Jagielski #ifdef _MSC_VER 28*b1cdbd2cSJim Jagielski #pragma warning(pop) 29*b1cdbd2cSJim Jagielski #endif 30*b1cdbd2cSJim Jagielski #include <tlhelp32.h> 31*b1cdbd2cSJim Jagielski 32*b1cdbd2cSJim Jagielski IMPLEMENT_THUNK( psapi, WINDOWS, DWORD, WINAPI, GetModuleFileNameExA, (HANDLE hProcess, HMODULE hModule, LPSTR lpFileName, DWORD nSize ) ) 33*b1cdbd2cSJim Jagielski { 34*b1cdbd2cSJim Jagielski DWORD dwProcessId = 0; 35*b1cdbd2cSJim Jagielski DWORD dwResult = 0; 36*b1cdbd2cSJim Jagielski 37*b1cdbd2cSJim Jagielski if ( !hProcess || hProcess == GetCurrentProcess() || GetCurrentProcessId() == (dwProcessId = GetProcessId( hProcess )) ) 38*b1cdbd2cSJim Jagielski return GetModuleFileNameA( hModule, lpFileName, nSize ); 39*b1cdbd2cSJim Jagielski 40*b1cdbd2cSJim Jagielski HANDLE hSnapshot = CreateToolhelp32Snapshot( TH32CS_SNAPMODULE, dwProcessId ); 41*b1cdbd2cSJim Jagielski 42*b1cdbd2cSJim Jagielski if ( IsValidHandle( hSnapshot ) ) 43*b1cdbd2cSJim Jagielski { 44*b1cdbd2cSJim Jagielski MODULEENTRY32 me; 45*b1cdbd2cSJim Jagielski 46*b1cdbd2cSJim Jagielski me.dwSize = sizeof(me); 47*b1cdbd2cSJim Jagielski if ( Module32First( hSnapshot, &me ) ) 48*b1cdbd2cSJim Jagielski { 49*b1cdbd2cSJim Jagielski BOOL fFound = FALSE; 50*b1cdbd2cSJim Jagielski 51*b1cdbd2cSJim Jagielski if ( NULL == hModule ) 52*b1cdbd2cSJim Jagielski fFound = TRUE; 53*b1cdbd2cSJim Jagielski else do 54*b1cdbd2cSJim Jagielski { 55*b1cdbd2cSJim Jagielski fFound = (me.hModule == hModule); 56*b1cdbd2cSJim Jagielski } while ( !fFound && Module32Next( hSnapshot, &me ) ); 57*b1cdbd2cSJim Jagielski 58*b1cdbd2cSJim Jagielski if ( fFound ) 59*b1cdbd2cSJim Jagielski { 60*b1cdbd2cSJim Jagielski dwResult = _tcslen( me.szExePath ); 61*b1cdbd2cSJim Jagielski 62*b1cdbd2cSJim Jagielski if ( dwResult > nSize && nSize > 0 ) 63*b1cdbd2cSJim Jagielski lpFileName[nSize -1] = 0; 64*b1cdbd2cSJim Jagielski 65*b1cdbd2cSJim Jagielski _tcsncpy( lpFileName, me.szExePath, nSize ); 66*b1cdbd2cSJim Jagielski } 67*b1cdbd2cSJim Jagielski } 68*b1cdbd2cSJim Jagielski 69*b1cdbd2cSJim Jagielski CloseHandle( hSnapshot ); 70*b1cdbd2cSJim Jagielski } 71*b1cdbd2cSJim Jagielski 72*b1cdbd2cSJim Jagielski return dwResult; 73*b1cdbd2cSJim Jagielski } 74*b1cdbd2cSJim Jagielski 75