博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
应用程序启动后修改自身EXE文件或自删除EXE文件(附VC++6.0源码)
阅读量:5133 次
发布时间:2019-06-13

本文共 8331 字,大约阅读时间需要 27 分钟。

在CSDN论坛看到这么一个问题:如何为第三方工具加上使用限制次数?问题的答案很简单,重新做一个应用程序,将第三方程序打包进这个应用程序,启动应用程序时可以检查第三方工具的使用次数,检查通过,可运行第三方工具。至于如果检查使用次数,答案也是五花八门,写注册表、写配置文件、读取硬盘、网卡信息等,都是常规做法。我也思考了下,还可以这样:

也是重写应用程序,但将使用次数写入到应用程序文件里面,每使用一次,就更新一下应用程序文件里的计数。

这种方案想来有几个问题:1.如果运行应用程序前做了备份,则每当使用次数完后,还可以用备份应用程序运行;2.我们知道,一个应用程序运行后,是不能修改自身EXE文件的。因此这个方案也只是想想而已,并不实用。但是要实现这种方案还是需要一定的技术,我闲来无事,就写了一个DEMO。

每2个问题,如何在应用程序运行时候修改自身EXE文件?答案也很简单,应用程序运行时加载另一个应用程序,自己退出,让另一个应用程序来修改EXE文件。既然应用程序运行时可以修改本身EXE文件,那么删除自身的EXE文件也是可以的,因此这种方法还可以用于卸载软件中,卸载软件时,可以使用此方法删除卸载软件本身。

好了,我们把目标明确一下:

1.第三方软件为SourceExe.exe,为了演示,我们还是自己模拟第三方,自己写;

2.加载的应用程序为ExePackage.exe;

3.修改ExePackage.exe的程序为ExeHelper.exe。

SourceExe.exe和ExeHelper.exe以资源形式加载进ExePackage.exe。启动ExePackage.exe时,ExePackage.exe并不显示界面,只将SourceExe.exe和ExeHelper.exe释放到临时目录,然后启动ExeHelper.exe,并告诉ExeHelper.exe一些必要的信息(DEMO中通过参数传递)。

ExeHelper.exe启动时,首先等待ExePackage.exe结束,这样才能修改ExePackage.exe文件。然后从ExePackage.exe文件末尾读取16个字节,测试是否匹配我们指定的数据(以区别这些数据是EXE文件本身的数据,还是我们添加的数据),这16个字节的最后4个字节保存的是使用次数,这样我们就可得到使用次数了,使用一次后,再将数据更新到EXE文件,最后就可以启动SourceExe.exe了。

原理很简单,但用到的技术还是不少的:

1.子进程等待父进程结束再处理事务(应用程序的自删除);

2.父进程将句柄等信息传递给子进程;

3.将EXE中加载的资源释放为本地文件。

下面就开始上代码,代码很简单,就不一一解释了,只在需要说明的地方说明一下:

首先是SourceExe.exe的源码,作为第三方工具,代码很简单,就不多说了。

SourceExe.cpp:

 

// SourceExe.cpp : Defines the entry point for the console application.//#include "stdafx.h"#include 
#include
using namespace std;int main(int argc, char* argv[]){ cout << "Hello world!" << endl; getch(); return 0;}

然后是ExeHelper.exe的源码,仔细看,也不太难。

 

ExeHelper.cpp:

 

// ExeHelper.cpp : Defines the entry point for the console application.//#include "stdafx.h"#include 
#include
#include
using namespace std;BYTE g_byCompare[16] = {'1', '2', '0', '6', '8', '0', '4', '5', '1', 0, 0, 0, 0, 0, 0, 0};DWORD dwDefaultTimes = 30;int main(int argc, char* argv[]){ if (NULL == argv[0] || 0 == strlen(argv[0]) || NULL == argv[1] || 0 == strlen(argv[1]) || NULL == argv[2] || 0 == strlen(argv[2]) || NULL == argv[3] || 0 == strlen(argv[3])) { cout << "argv error!" << endl; getch(); return 0; } cout << "argv[0]=" << argv[0] << endl; cout << "argv[1]=" << argv[1] << endl; cout << "argv[2]=" << argv[2] << endl; cout << "argv[3]=" << argv[3] << endl; // 保证父进程已退出,才能操作父进程的EXE文件 HANDLE handle=(HANDLE)(atoi(argv[1])); if (WAIT_FAILED == WaitForSingleObject(handle, INFINITE)) { cout << "WaitForSingleObject Error (" << GetLastError() << ")" << endl; getch(); return 0; } HANDLE hFile = CreateFile( argv[2], GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL); BYTE pBuffer[16] = {0}; DWORD dwFileSizeLow = 0; DWORD dwFileSizeHigh = 0; dwFileSizeLow = GetFileSize(hFile, &dwFileSizeHigh); if (dwFileSizeLow < 16) { cout << "The size of " << argv[2] << " is " << dwFileSizeLow << ", less than 16 bytes." << endl; getch(); return 0; } SetFilePointer(hFile, -16, NULL, FILE_END); DWORD dwBytesRead = 0; ReadFile(hFile, pBuffer, 16, &dwBytesRead, NULL); DWORD dwTimes = 0; memcpy(&dwTimes, pBuffer + 12, sizeof(DWORD)); // pBuffer的前12个字节应该与g_byCompare的前12个字节相同,后4个字节为使用次数 if (0 != memcmp(pBuffer, g_byCompare, 12)) { // pBuffer前12个字节不是g_byCompare的前12个字节,则认为是第一次启动 // 在EXE文件末尾添加上默认的30次使用次数 dwTimes = dwDefaultTimes; memcpy(pBuffer, g_byCompare, 16); memcpy(pBuffer + 12, &dwTimes, sizeof(DWORD)); SetFilePointer(hFile, 0, NULL, FILE_END); DWORD dwBytesWrite = 0; WriteFile(hFile, pBuffer, 16, &dwBytesWrite, NULL); } cout << "Can use " << dwTimes << " times" << endl; if (0 == dwTimes) { getch(); return 0; } cout << "Press any key to run" << endl; getch(); // 减少使用次数,并将使用次数再更新到EXE文件 dwTimes--; memcpy(pBuffer + 12, &dwTimes, sizeof(DWORD)); SetFilePointer(hFile, -16, NULL, FILE_END); DWORD dwBytesWrite = 0; WriteFile(hFile, pBuffer, 16, &dwBytesWrite, NULL); CloseHandle(hFile); STARTUPINFO si; memset(&si, 0, sizeof(si)); si.cb = sizeof(si); PROCESS_INFORMATION pi; memset(&pi, 0, sizeof(pi)); // 然后启动第三方EXE CreateProcess( argv[3], NULL, NULL, NULL, FALSE, NORMAL_PRIORITY_CLASS | CREATE_NEW_CONSOLE, NULL, NULL, &si, &pi); CloseHandle(pi.hThread); CloseHandle(pi.hProcess); return 0;}

最后是ExePackage.exe的源码,ExePackage.cpp:

 

 

CExePackageApp theApp;/// CExePackageApp initialization/*  函数功能:释放资源文件         参数说明:DWORD   dwResID			:指定要释放的资源ID号,如IDR_EXE       LPCTSTR lpszResType		:指定释放的资源的资源类型       LPCTSTR lpszFilePathName	:指定释放后的目标文件名         返回值:成功则返回TRUE,失败返回FALSE  使用方法:BOOL bRet = FreeResFile(IDR_SOURCE_EXE, _T("EXE"), _T("D:\\123.txt"));		  其中,IDR_SOURCE_EXE为EXE类型的资源ID*/BOOL FreeResFile(DWORD dwResID, LPCTSTR lpszResType, LPCTSTR lpszFilePathName)  {         HMODULE hInstance = ::GetModuleHandle(NULL);//得到自身实例句柄                 HRSRC   hResID = ::FindResource(hInstance, MAKEINTRESOURCE(dwResID), lpszResType); //查找资源    HGLOBAL hRes   = ::LoadResource(hInstance, hResID); //加载资源      LPVOID  pRes   = ::LockResource(hRes); //锁定资源                 if (NULL == pRes) //锁定失败      {          return FALSE;      }      DWORD  dwResSize = ::SizeofResource(hInstance, hResID); //得到待释放资源文件大小      HANDLE hResFile  = CreateFile(		lpszFilePathName,		GENERIC_WRITE,		0,		NULL,		CREATE_ALWAYS,		FILE_ATTRIBUTE_NORMAL,		NULL);       if (INVALID_HANDLE_VALUE == hResFile)      {          return FALSE;      }             DWORD dwWritten = 0; //写入文件的大小         WriteFile(hResFile, pRes, dwResSize, &dwWritten, NULL); //写入文件      CloseHandle(hResFile); //关闭文件句柄		FreeResource(hRes);               return (dwResSize == dwWritten); //若写入大小等于文件大小,返回成功,否则失败  }BOOL CExePackageApp::InitInstance(){	AfxEnableControlContainer();	// Standard initialization	// If you are not using these features and wish to reduce the size	//  of your final executable, you should remove from the following	//  the specific initialization routines you do not need.#ifdef _AFXDLL	Enable3dControls();			// Call this when using MFC in a shared DLL#else	Enable3dControlsStatic();	// Call this when linking to MFC statically#endif	TCHAR szTmpDir[MAX_PATH] = {0};	GetTempPath(MAX_PATH, szTmpDir);		TCHAR szSourceExeFilePathName[MAX_PATH] = {0};	_stprintf(szSourceExeFilePathName, _T("%sSourceExe.exe"), szTmpDir);		TCHAR szExeHelperFilePathName[MAX_PATH] = {0};	_stprintf(szExeHelperFilePathName, _T("%sExeHelper.exe"), szTmpDir);		DeleteFile(szSourceExeFilePathName);	DeleteFile(szExeHelperFilePathName);	FreeResFile(IDR_SOURCE_EXE, "EXE", szSourceExeFilePathName);	FreeResFile(IDR_EXE_HELPER, "EXE", szExeHelperFilePathName);		TCHAR szModuleName[MAX_PATH] = {0};	GetModuleFileName(NULL, szModuleName, MAX_PATH);		STARTUPINFO si;	memset(&si, 0, sizeof(si));	si.cb = sizeof(si);	PROCESS_INFORMATION pi;	memset(&pi, 0, sizeof(pi));	    SECURITY_ATTRIBUTES sa;  	memset(&sa, 0, sizeof(sa));    sa.nLength = sizeof(sa);      sa.lpSecurityDescriptor = NULL;      sa.bInheritHandle = TRUE; 		// 得到此进程的句柄,传递给ExeHelper.exe,因此这儿不能关闭hParentProcess	// 需要在子进程中关闭    HANDLE hParentProcess = OpenProcess(PROCESS_ALL_ACCESS, TRUE, GetCurrentProcessId());    TCHAR szCommandLine[MAX_PATH * 3] = {0};	// 注意此处的参数,调用CreateProcess时,并不会将EXE的文件名也当作参数	// 因此,需要在参数列表中加上EXE的文件名(ExeHelper.exe并不使用)    _stprintf(szCommandLine, _T("\"%s\" %d \"%s\" \"%s\""),		szExeHelperFilePathName, (DWORD)hParentProcess, szModuleName, szSourceExeFilePathName);	    CreateProcess(		szExeHelperFilePathName,		szCommandLine,		&sa,		NULL,		TRUE,		NULL,		NULL,		NULL,		&si,		&pi);    CloseHandle(pi.hProcess);    CloseHandle(pi.hThread);// 	CExePackageDlg dlg;// 	m_pMainWnd = &dlg;// 	int nResponse = dlg.DoModal();// 	if (nResponse == IDOK)// 	{// 		// TODO: Place code here to handle when the dialog is// 		//  dismissed with OK// 	}// 	else if (nResponse == IDCANCEL)// 	{// 		// TODO: Place code here to handle when the dialog is// 		//  dismissed with Cancel// 	}	// Since the dialog has been closed, return FALSE so that we exit the	//  application, rather than start the application's message pump.	return FALSE;}

需要说明:

创建的是MFC对话框程序,因为方便导入SourceExe.exe和ExeHelper.exe作为资源。SourceExe.exe和ExeHelper.exe都是导入的EXE资源。ID可见代码或下图。代码中使用了FreeResFile函数来释放资源到本地文件。注释掉了启动对话框的代码,只将SourceExe.exe和ExeHelper.exe释放到临时目录,运行ExeHelper.exe后立即退出。SourceExe和ExeHelper编译后,会自动将生成的EXE复制到ExePackage\res目录中,因此SourceExe或ExeHelper重新编译后,都需重新编译ExePackage,重新将SourceExe.exe和ExeHelper.exe打包进ExePackage.exe。

运行的效果图如下:

当然,你也可做成非Console的界面,获取参数就需使用GetCommandLine()。

源码下载地址:

 

转载于:https://www.cnblogs.com/jiangu66/p/3170462.html

你可能感兴趣的文章
日常训练
查看>>
1030: [JSOI2007]文本生成器 ac自动机+dp
查看>>
如何在普通 UIViewController 中使用 UITableView
查看>>
HDU 3651 A Simple Problem
查看>>
十全干货:核心游戏系统架构设计
查看>>
64位CentOS源码编译方式安装wine
查看>>
【luogu P3931 SAC E#1 - 一道难题 Tree】 题解
查看>>
Spring Boot实战笔记(一)-- Spring简介
查看>>
java基础(1)
查看>>
Java对象的序列化和反序列化实践
查看>>
window 常用软件
查看>>
第八届河南省赛D.引水工程(kruthcra+prime)
查看>>
例子:点击滑块滚动,左侧逐渐变大,右侧变小
查看>>
python学习笔记(二十七)多线程与多进程
查看>>
简单照片浏览器
查看>>
split(v1,v2)用于把一个字符串分割成字符串数组
查看>>
python学习笔记 -- map() 操作可迭代序列
查看>>
删除一个带有文件的文件夹
查看>>
content-providers
查看>>
冒泡排序及优化
查看>>