mirror of https://github.com/tongzx/nt5src
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.
398 lines
10 KiB
398 lines
10 KiB
#include <stdio.h>
|
|
#include <windows.h>
|
|
#include <tchar.h>
|
|
#include <oleauto.h>
|
|
|
|
int __cdecl main(int ,char *argv[]);
|
|
void ShowHelp(void);
|
|
int DoesFileExist(char *input_filespec);
|
|
int ProcessFile(char szTheFileName[]);
|
|
DWORD CallProcedureInDll(LPCTSTR lpszDLLFile, LPCTSTR lpszProcedureToCall, BOOL bInitOleFlag);
|
|
|
|
int g_Param_A = FALSE;
|
|
int g_Param_B = FALSE;
|
|
|
|
//-------------------------------------------------------------------
|
|
// purpose: main
|
|
//-------------------------------------------------------------------
|
|
int __cdecl main(int argc,char *argv[])
|
|
{
|
|
int argno;
|
|
int nflags=0;
|
|
char szTempFileName1[_MAX_PATH];
|
|
char szTempFileName2[_MAX_PATH];
|
|
int iLen = 0;
|
|
|
|
szTempFileName1[0] = '\0';
|
|
szTempFileName2[0] = '\0';
|
|
|
|
// process command line arguments
|
|
for(argno=1; argno<argc; argno++)
|
|
{
|
|
if ( argv[argno][0] == '-' || argv[argno][0] == '/' )
|
|
{
|
|
nflags++;
|
|
switch (argv[argno][1])
|
|
{
|
|
case 'a':
|
|
g_Param_A = TRUE;
|
|
break;
|
|
case 'b':
|
|
g_Param_B = TRUE;
|
|
break;
|
|
case '?':
|
|
goto exit_with_help;
|
|
break;
|
|
}
|
|
} // if switch character found
|
|
else
|
|
{
|
|
if ( *szTempFileName1 == '\0' )
|
|
{
|
|
strcpy(szTempFileName1, argv[argno]);
|
|
}
|
|
else
|
|
{
|
|
if ( *szTempFileName2 == '\0' )
|
|
{
|
|
strcpy(szTempFileName2, argv[argno]);
|
|
}
|
|
|
|
}
|
|
} // non-switch char found
|
|
} // for all arguments
|
|
|
|
|
|
// check if filename was specified
|
|
if ( *szTempFileName1 == '\0')
|
|
{
|
|
_tprintf("Too few arguments, argc=%d\n\n",argc);
|
|
goto exit_with_help;
|
|
}
|
|
|
|
// Check if the file exists!
|
|
if (FALSE == DoesFileExist(szTempFileName1))
|
|
{
|
|
_tprintf("File '%s', does not exist!\r\n", szTempFileName1);
|
|
goto exit_gracefully;
|
|
}
|
|
|
|
if (g_Param_A)
|
|
{
|
|
if ( *szTempFileName2 == '\0')
|
|
{
|
|
_tprintf("Calling DllRegisterServer() in %s\r\n",szTempFileName1);
|
|
CallProcedureInDll(szTempFileName1,_T("DllRegisterServer"),TRUE);
|
|
}
|
|
else
|
|
{
|
|
_tprintf("Calling %s() in %s\r\n", szTempFileName2,szTempFileName1);
|
|
CallProcedureInDll(szTempFileName1,szTempFileName2,TRUE);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// run the function to do everything
|
|
ProcessFile(szTempFileName1);
|
|
}
|
|
|
|
|
|
exit_gracefully:
|
|
_tprintf("Done.\n");
|
|
return TRUE;
|
|
|
|
exit_with_help:
|
|
ShowHelp();
|
|
return FALSE;
|
|
}
|
|
|
|
//-------------------------------------------------------------------
|
|
// purpose:
|
|
//-------------------------------------------------------------------
|
|
int ProcessFile(char szTheFileName[])
|
|
{
|
|
HINSTANCE hLibHandle;
|
|
if ( ( hLibHandle = LoadLibrary( szTheFileName ) ) != NULL )
|
|
{
|
|
_tprintf("LoadLibarary Successfull.\n");
|
|
{
|
|
FreeLibrary(hLibHandle);
|
|
return TRUE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
_tprintf("LoadLibarary Failed.\n");
|
|
return FALSE;
|
|
}
|
|
|
|
}
|
|
|
|
//-------------------------------------------------------------------
|
|
// purpose:
|
|
//-------------------------------------------------------------------
|
|
void ShowHelp()
|
|
{
|
|
_tprintf("\n");
|
|
_tprintf("LoadLib - does a loadlibrary call on the specified file.\n");
|
|
_tprintf("Usage: LoadLib <input file> \n");
|
|
return;
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------
|
|
// purpose:
|
|
//-------------------------------------------------------------------
|
|
int DoesFileExist(char *input_filespec)
|
|
{
|
|
if (GetFileAttributes(input_filespec) == -1) {return(FALSE);}
|
|
return (TRUE);
|
|
}
|
|
|
|
|
|
BOOL IsFileExist(LPCTSTR szFile)
|
|
{
|
|
// Check if the file has expandable Environment strings
|
|
LPTSTR pch = NULL;
|
|
pch = _tcschr( (LPTSTR) szFile, _T('%'));
|
|
if (pch)
|
|
{
|
|
TCHAR szValue[_MAX_PATH];
|
|
_tcscpy(szValue,szFile);
|
|
if (!ExpandEnvironmentStrings( (LPCTSTR)szFile, szValue, sizeof(szValue)/sizeof(TCHAR)))
|
|
{_tcscpy(szValue,szFile);}
|
|
|
|
return (GetFileAttributes(szValue) != 0xFFFFFFFF);
|
|
}
|
|
else
|
|
{
|
|
return (GetFileAttributes(szFile) != 0xFFFFFFFF);
|
|
}
|
|
}
|
|
|
|
|
|
|
|
int iOleInitialize(void)
|
|
{
|
|
int iBalanceOLE = FALSE;
|
|
HRESULT hInitRes = NULL;
|
|
hInitRes = OleInitialize(NULL);
|
|
if ( SUCCEEDED(hInitRes) || hInitRes == RPC_E_CHANGED_MODE )
|
|
{
|
|
if ( SUCCEEDED(hInitRes))
|
|
{
|
|
iBalanceOLE = TRUE;
|
|
}
|
|
else
|
|
{
|
|
_tprintf("iOleInitialize failed 1\n");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
_tprintf("iOleInitialize failed 2\n");
|
|
}
|
|
|
|
return iBalanceOLE;
|
|
}
|
|
|
|
|
|
void iOleUnInitialize(int iBalanceOLE)
|
|
{
|
|
if (iBalanceOLE)
|
|
{
|
|
OleUninitialize();
|
|
}
|
|
return;
|
|
}
|
|
|
|
void InetGetFilePath(LPCTSTR szFile, LPTSTR szPath)
|
|
{
|
|
// if UNC name \\computer\share\local1\local2
|
|
if (*szFile == _T('\\') && *(_tcsinc(szFile)) == _T('\\')) {
|
|
TCHAR szTemp[_MAX_PATH], szLocal[_MAX_PATH];
|
|
TCHAR *p = NULL;
|
|
int i = 0;
|
|
|
|
_tcscpy(szTemp, szFile);
|
|
p = szTemp;
|
|
while (*p) {
|
|
if (*p == _T('\\'))
|
|
i++;
|
|
if (i == 4) {
|
|
*p = _T('\0');
|
|
p = _tcsinc(p); // p is now pointing at local1\local2
|
|
break;
|
|
}
|
|
p = _tcsinc(p);
|
|
}
|
|
_tcscpy(szPath, szTemp); // now szPath contains \\computer\share
|
|
|
|
if (i == 4 && *p) { // p is pointing the local path now
|
|
_tcscpy(szLocal, p);
|
|
p = _tcsrchr(szLocal, _T('\\'));
|
|
if (p)
|
|
*p = _T('\0');
|
|
_tcscat(szPath, _T("\\"));
|
|
_tcscat(szPath, szLocal); // szPath contains \\computer\share\local1
|
|
}
|
|
} else { // NOT UNC name
|
|
TCHAR *p;
|
|
if (GetFullPathName(szFile, _MAX_PATH, szPath, &p)) {
|
|
p = _tcsrchr(szPath, _T('\\'));
|
|
if (p)
|
|
{
|
|
TCHAR *p2 = NULL;
|
|
p2 = _tcsdec(szPath, p);
|
|
if (p2)
|
|
{
|
|
if (*p2 == _T(':') )
|
|
{p = _tcsinc(p);}
|
|
}
|
|
*p = _T('\0');
|
|
}
|
|
} else {
|
|
_tprintf("InetGetFilePath failed\n");
|
|
}
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
typedef HRESULT (CALLBACK *HCRET)(void);
|
|
|
|
DWORD CallProcedureInDll(LPCTSTR lpszDLLFile, LPCTSTR lpszProcedureToCall, BOOL bInitOleFlag)
|
|
{
|
|
TCHAR szErrorString[100];
|
|
|
|
DWORD dwReturn = ERROR_SUCCESS;
|
|
HINSTANCE hDll = NULL;
|
|
|
|
// Diferent function prototypes...
|
|
HCRET hProc = NULL;
|
|
int iTempProcGood = FALSE;
|
|
HRESULT hRes = 0;
|
|
|
|
BOOL bBalanceOLE = FALSE;
|
|
HRESULT hInitRes = NULL;
|
|
|
|
int err = NOERROR;
|
|
|
|
// Variables to changing and saving dirs
|
|
TCHAR szDirName[_MAX_PATH], szFilePath[_MAX_PATH];
|
|
// Variable to set error string
|
|
TCHAR szErrString[256];
|
|
|
|
_tcscpy(szDirName, _T(""));
|
|
_tcscpy(szErrString, _T(""));
|
|
|
|
_tprintf("CallProcedureInDll start\n");
|
|
|
|
// If we need to initialize the ole library then init it.
|
|
if (bInitOleFlag)
|
|
{
|
|
bBalanceOLE = iOleInitialize();
|
|
if (FALSE == bBalanceOLE)
|
|
{
|
|
hInitRes = OleInitialize(NULL);
|
|
// Ole Failed.
|
|
dwReturn = hInitRes;
|
|
SetLastError(dwReturn);
|
|
_tprintf("CallProcedureInDll failed 1\n");
|
|
goto CallProcedureInDll_Exit;
|
|
}
|
|
}
|
|
|
|
// Check if the file exists
|
|
if (!IsFileExist(lpszDLLFile))
|
|
{
|
|
dwReturn = ERROR_FILE_NOT_FOUND;
|
|
_tprintf("CallProcedureInDll failed 2\n");
|
|
SetLastError(dwReturn);
|
|
goto CallProcedureInDll_Exit;
|
|
}
|
|
|
|
// Change Directory
|
|
GetCurrentDirectory( _MAX_PATH, szDirName );
|
|
InetGetFilePath(lpszDLLFile, szFilePath);
|
|
|
|
// Change to The Drive.
|
|
if (SetCurrentDirectory(szFilePath) == 0) {}
|
|
|
|
// Try to load the module,dll,ocx.
|
|
hDll = LoadLibraryEx(lpszDLLFile, NULL, LOAD_WITH_ALTERED_SEARCH_PATH );
|
|
if (!hDll)
|
|
{
|
|
// Failed to load library, Probably because some .dll file is missing.
|
|
// Show the error message.
|
|
dwReturn = TYPE_E_CANTLOADLIBRARY;
|
|
_tprintf("CallProcedureInDll failed 3\n");
|
|
SetLastError(dwReturn);
|
|
|
|
goto CallProcedureInDll_Exit;
|
|
}
|
|
|
|
// Ok module was successfully loaded. now let's try to get the Address of the Procedure
|
|
// Convert the function name to ascii before passing it to GetProcAddress()
|
|
char AsciiProcedureName[255];
|
|
#if defined(UNICODE) || defined(_UNICODE)
|
|
// convert to ascii
|
|
WideCharToMultiByte( CP_ACP, 0, (TCHAR *)lpszProcedureToCall, -1, AsciiProcedureName, 255, NULL, NULL );
|
|
#else
|
|
// the is already ascii so just copy
|
|
strcpy(AsciiProcedureName, lpszProcedureToCall);
|
|
#endif
|
|
|
|
iTempProcGood = TRUE;
|
|
hProc = (HCRET)GetProcAddress(hDll, AsciiProcedureName);
|
|
if (!hProc){iTempProcGood = FALSE;}
|
|
if (!iTempProcGood)
|
|
{
|
|
// failed to load,find or whatever this function.
|
|
dwReturn = ERROR_PROC_NOT_FOUND;
|
|
_tprintf("CallProcedureInDll failed 4\n");
|
|
SetLastError(dwReturn);
|
|
goto CallProcedureInDll_Exit;
|
|
}
|
|
|
|
// Call the function that we got the handle to
|
|
__try
|
|
{
|
|
hRes = (*hProc)();
|
|
}
|
|
__except(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
_tprintf(szErrorString, _T("\r\n\r\nException Caught in CallProcedureInDll(). GetExceptionCode()=0x%x.\r\n\r\n"), GetExceptionCode());
|
|
_tprintf(szErrorString);
|
|
}
|
|
|
|
if (FAILED(hRes))
|
|
{
|
|
dwReturn = E_FAIL;
|
|
_tprintf("CallProcedureInDll failed 5\n");
|
|
// this function returns E_FAIL but
|
|
// the actual error is in GetLastError()
|
|
// set the last error to whatever was returned from the function call
|
|
SetLastError(hRes);
|
|
}
|
|
else
|
|
{
|
|
}
|
|
|
|
CallProcedureInDll_Exit:
|
|
if (hDll)
|
|
{
|
|
FreeLibrary(hDll);
|
|
}
|
|
else
|
|
{
|
|
}
|
|
if (_tcscmp(szDirName, _T("")) != 0){SetCurrentDirectory(szDirName);}
|
|
// To close the library gracefully, each successful call to OleInitialize,
|
|
// including those that return S_FALSE, must be balanced by a corresponding
|
|
// call to the OleUninitialize function.
|
|
iOleUnInitialize(bBalanceOLE);
|
|
return dwReturn;
|
|
}
|
|
|