您的位置:首页 > 编程语言 > C语言/C++

C++嵌入DLL到资源运行释放的问题

2017-09-16 18:29 417 查看
以前写过一篇博文《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文件,理论上什么文件都是可以的。上面的做法可能会被某些杀毒软件拦截,原因是很多远控木马等恶意程序采用了类似的技术,导致杀软禁止了这种加载行为。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息