有些程序运行的时候,需要调用外部的资源文件如DLL,EXE等,用户使用时可能会不小心丢失这些资源文件,导致程序无法正常运行,因此可以考虑将这些文件嵌入到资源中,程序启动时自动释放到可执行程序目录(或者其他环境变量目录)。这里介绍将文件到资源后再运行释放的小技巧。
开发环境C++ VS2013。
1.加载文件到c++程序。
a.资源文件右击->添加->资源->自定义,输入资源类型如:restxt
b.将要添加的资源文件导入到restxt类型的资源里。
2.释放文件(本文介绍txt文件):源码如下
releaseHelper.cpp
#include "stdafx.h"
#include "releaseHelper.h"
#include <cstdio>
#include <string.h>
#include <direct.h>
#include <exception>
CReleaseDLL::CReleaseDLL()
{
this->m_hModule = GetSelfModuleHandle();
if (m_hModule == NULL)
{
throw std::exception("Error:获取基址失败");
}
//获取目录
memset(this->m_filePath, 0, MAX_DLL_PATH);
_getcwd(this->m_filePath, MAX_DLL_PATH);
}
CReleaseDLL::~CReleaseDLL()
{
}
bool CReleaseDLL::FreeResFile(unsigned long m_lResourceID, const char* m_strResourceType, const char* m_strFileName)
{
//构造完整释放文件路径
char strFullPath[MAX_DLL_PATH] = { 0 };
sprintf_s(strFullPath, "%s\\%s", this->m_filePath, m_strFileName);
//查找资源
HRSRC hResID = ::FindResource(this->m_hModule, MAKEINTRESOURCE(m_lResourceID), m_strResourceType);
//加载资源
HGLOBAL hRes = ::LoadResource(this->m_hModule, hResID);
//锁定资源
LPVOID pRes = ::LockResource(hRes);
if (pRes == NULL)
{
return FALSE;
}
//得到待释放资源文件大小
unsigned long dwResSize = ::SizeofResource(this->m_hModule, hResID);
//创建文件
HANDLE hResFile = CreateFile(strFullPath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if (INVALID_HANDLE_VALUE == hResFile)
{
return FALSE;
}
DWORD dwWrited = 0;
if (FALSE == WriteFile(hResFile, pRes, dwResSize, &dwWrited, NULL))
{
return FALSE;
}
CloseHandle(hResFile);
UnlockResource(hRes);
FreeResource(hRes);
return TRUE;
}
HMODULE CReleaseDLL::GetSelfModuleHandle()
{
try
{
#ifdef _USER_RELEASEDLL_
//如果释放的帮助类定义在DLL中,将调用下面的方式获取基址
MEMORY_BASIC_INFORMATION mbi;
return ((::VirtualQuery((LPCVOID)&CReleaseDLL::GetSelfModuleHandle, &mbi, sizeof(mbi)) != 0)
? (HMODULE)mbi.AllocationBase : NULL);
#else
//如果直接定义在exe本身的代码中
return ::GetModuleHandle(NULL);
#endif
}
catch (...)
{
return NULL;
}
}
releaseHelper.h
#ifndef _RELEASEHELPER_H_
#define _RELEASEHELPER_H_
#include <windows.h>
#define MAX_DLL_PATH 1024
//如果操作代码位于DLL工程中,就定义下面的宏,否则注释掉
//#define _USER_RELEASEDLL_
//释放DLL帮助类
class CReleaseDLL
{
public:
CReleaseDLL();
~CReleaseDLL();
/*
*function:释放DLL文件帮助函数
*[IN]:
m_lResourceID:资源的ID
m_strResourceType:资源类型
m_strReleasePath:需要释放的文件名称
*[OUT]:
释放成功,返回TRUE,否则返回FALSE
*/
bool FreeResFile(unsigned long m_lResourceID, const char* m_strResourceType, const char* m_strFileName);
private:
/*
*function:获取模块的基址
*[IN]:
*[OUT]:
返回获取的基址
*/
HMODULE GetSelfModuleHandle();
private:
//模块基址
HMODULE m_hModule;
//程序当前目录
char m_filePath[MAX_DLL_PATH];
};
#endif
main 文件
// 资源释放示例.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include <iostream>
#include "resource.h"
#include "releaseHelper.h"
using namespace std;
#define PAUSE cout<<"Please Entry Any Code..."<<endl;getchar();
int main()
{
//定义操作类对象
CReleaseDLL releasehelper;
bool blRes;
blRes = releasehelper.FreeResFile(IDR_RESTXT2, "RESTXT", "520.txt");
blRes = releasehelper.FreeResFile(IDR_RESTXT3, "RESTXT", "521.txt");
if (blRes)
{
cout << "DLL文件释放成功" << endl;
}
else
{
cout << "DLL文件释放失败" << endl;
}
PAUSE;
return 0;
}
3.总结
把文件嵌入到程序资源中并运行释放有一定的现实意义,特别是DLL文件,可以减少开发程序对外部环境的依懒性。当然了,上面的方法不仅仅局限于释放DLL文件,理论上什么文件都是可以的。上面的做法可能会被某些杀毒软件拦截,原因是很多远控木马等恶意程序采用了类似的技术,导致杀软禁止了这种加载行为。