以前写过一篇博文《C#嵌入dll到资源释放的问题》。虽然相对于C#,C++中嵌入DLL到程序资源中,然后再释放出来的应用场合并没有那么多,但是还是有必要了解下一般的过程。结合本人在实际工作中解决此类问题的实践思路,这里介绍下最基本的一种解决方案。
1 嵌入DLL到资源
有些程序运行的时候,可能调用外部的DLL,用户使用时可能会不小心丢失这些DLL,导致程序无法正常运行,因此可以考虑将这些DLL嵌入到资源中,启动时自动释放到可执行程序目录(或者其他环境变量目录)。这里以嵌入FFmpeg+SDL开发所需要的DLL文件为例,介绍下C++程序(EXE或者DLL)嵌入DLL到资源后再运行释放的小技巧。
需要嵌入的DLL文件如下:
利用Visual Studio新建C++项目后,会自动生成一个“资源文件”的筛选器:
右键->“添加”->“资源”->”自定义”:
新建一种新的资源类型“dlls”并保存。然后将需要嵌入的DLL文件加入资源文件中,重命名相应的ID,结果如下:
上面DLL的ID定义位于resource.h头文件中:
resource.h
#define IDR_avcodec_55 102
#define IDR_avdevice_55 103
#define IDR_avfilter_4 104
#define IDR_avformat_55 105
#define IDR_avutil_52 106
#define IDR_postproc_52 107
#define IDR_SDL 108
#define IDR_swresample_0 109
#define IDR_DLLS9 110
#define IDR_swscale_2 110
DLL资源与实际物理路径的映射关系都保存在*.rc文件中。这里给个小建议,为了方便程序的移植,最好把路径改成相对路径(相对于工程的路径):
*.rc
2 释放DLL文件
从资源中释放文件,整个过程并不复杂。程序运行后,在内存中定位程序的基址,然后加载资源文件,并将其释放到指定路径即可。下面是我封装的一个释放DLL文件帮助类:
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
releaseHelper.cpp
#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;
}
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;
}
}
初始化类对象后,直接调用FreeResFile函数就可以完成DLL文件的释放。这里需要说明的一点就是,将CReleaseDLL定义在一个动态链接库中供程序来调用,和直接将CReleaseDLL定义在程序本身的代码中,这两种获取程序基址的方式是不一样的,所以上述代码中增加了_USER_RELEASEDLL_
宏,用来兼容这两种模式。
下面是调用上述封装类完成DLL释放的示例代码:
main.cpp
#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_avcodec_55,"dlls","avcodec-55.dll");
blRes=releasehelper.FreeResFile(IDR_avdevice_55,"DLLS","avdevice-55.dll");
blRes=releasehelper.FreeResFile(IDR_avfilter_4,"DLLS","avfilter-4.dll");
blRes=releasehelper.FreeResFile(IDR_avformat_55,"DLLS","avformat-55.dll");
blRes=releasehelper.FreeResFile(IDR_avutil_52,"DLLS","avutil-52.dll");
blRes=releasehelper.FreeResFile(IDR_postproc_52,"DLLS","postproc-52.dll");
blRes=releasehelper.FreeResFile(IDR_SDL,"DLLS","SDL.dll");
blRes=releasehelper.FreeResFile(IDR_swresample_0,"DLLS","swresample-0.dll");
blRes=releasehelper.FreeResFile(IDR_swscale_2,"DLLS","swscale-2.dll");
if(blRes)
{
cout<<"DLL文件释放成功"<<endl;
}
else
{
cout<<"DLL文件释放失败"<<endl;
}
PAUSE;
return 0;
}
3 后记
把DLL文件嵌入到程序资源中并运行释放有一定的现实意义,可以减少开发程序对外部环境的依懒性。当然了,上面的方法不仅仅局限于释放DLL文件,理论上什么文件都是可以的。上面的做法可能会被某些杀毒软件拦截,原因是很多远控木马等恶意程序采用了类似的技术,导致杀软禁止了这种加载行为。