问题描述: 我们开发过程中可能会经常遇到,需要屏蔽EXE的双击运行功能,只能通过宿主程序(Service或EXE)来启动。比如腾讯的迷你弹窗,就只能通过主程序来启动,而不能直接通过双击来运行。
实现原理:宿主程序可以通过向子程序命令行参数可以来启动,而子程序在初始化的时候,会去检测当前的命令行参数是否匹配;如果匹配,就启动;如果不匹配,就不启动。
因为宿主程序可以通过CreateProcess来传递命令行参数,所以可以顺利执行;而双击程序,没有传递任何命令行参数,自然程序也就没法启动了。
程序示例:#define CMD_LINE _T("CommandLine")
子程序在程序初始化中进行判断,判断部分源码如下:(返回TRUE-运行程序; 返回FALSE-不运行程序)
BOOL IsExecuteWhenDbClk() { BOOL bRt = FALSE; int cmdLineCount = 0; LPWSTR* pCmdLine = ::CommandLineToArgvW(GetCommandLine(), &cmdLineCount); for ( int i = 0; i < cmdLineCount; i++ ) { if ( 0 == CString(pCmdLine[i]).CompareNoCase( CString(CMD_LINE) ) ) { bRt = TRUE; break; } } GlobalFree( HGLOBAL(pCmdLine) ); return bRt; }
上面的例子只支持MFC框架,下面提供一个跨平台使用的版本:
#ifndef TOOLS_H_ #define TOOLS_H_ #include <Windows.h> #include <string> #include <algorithm> using namespace std; #define CMD_LINE L"AUTOINSTALL" BOOL IsExecuteWhenDbClk() { BOOL bRt = FALSE; int cmdLineCount = 0; LPWSTR* pCmdLine = ::CommandLineToArgvW(GetCommandLine(), &cmdLineCount); for ( int i = 0; i < cmdLineCount; i++ ) { wstring cmdLine(pCmdLine[i]); transform(cmdLine.begin(), cmdLine.end(), cmdLine.begin(), toupper); if ( 0 == cmdLine.compare(CMD_LINE) ) { bRt = TRUE; break; } } GlobalFree( HGLOBAL(pCmdLine) ); return bRt; } #endif // TOOLS_H_
宿主程序中,通过CreateProcess来启动,代码如下:
STARTUPINFO si; PROCESS_INFORMATION pi; ZeroMemory( &si, sizeof(si) ); si.cb = sizeof(si); ZeroMemory( &pi, sizeof(pi) ); // Start the child process. wchar_t cmdLine[] = CMD_LINE; if( !CreateProcess( NULL, // No module name (use command line) cmdLine, // Command line NULL, // Process handle not inheritable NULL, // Thread handle not inheritable FALSE, // Set handle inheritance to FALSE 0, // No creation flags NULL, // Use parent's environment block NULL, // Use parent's starting directory &si, // Pointer to STARTUPINFO structure &pi ) // Pointer to PROCESS_INFORMATION structure ) { printf( "CreateProcess failed (%d). ", GetLastError() ); return; } // Wait until child process exits. //WaitForSingleObject( pi.hProcess, INFINITE ); // Close process and thread handles. CloseHandle( pi.hProcess ); CloseHandle( pi.hThread );