(C++) DLL 管理類別

MSN Space、Google Doc 、Google Blog
Chui-Wen Chiu
2007.06.23

將動態載入 DLL 並取得函數指標的動作封裝成專屬類別，透過這個類別可以省去許多瑣碎步驟又可避免忘記釋放 DLL 的問題。

原始碼

LibraryMgr.h

#include <windows.h> #include <string> #ifndef LIBRARYMGR_H_INCLUDED #define LIBRARYMGR_H_INCLUDED class LibraryMgr_t { public:     LibraryMgr_t(void);     LibraryMgr_t(const TCHAR* ptcFileName);     ~LibraryMgr_t(void);     bool LoadLibrary( const TCHAR* ptcFileName );     bool FreeLibrary();     FARPROC GetProcAddress( const TCHAR* ptcProcName );     HMODULE GetModuleHandle();     bool GetModuleFileName(std::basic_string<TCHAR>& sName);     bool FreeLibraryAndExitThread( DWORD dwExitCode );     void SetAutoFree(bool bFree) { m_bAutoFree = bFree; } private:     HMODULE m_hModule;     bool m_bAutoFree;     std::basic_string<TCHAR> m_sFileNAme; }; #endif //LIBRARYMGR_H_INCLUDED

LibraryMgr.cpp

#include "LibraryMgr.h" LibraryMgr_t::LibraryMgr_t(void) : m_hModule(0), m_bAutoFree(true) { } /**  * 建構子  *   * @param ptcFileName DLL 名稱  */ LibraryMgr_t::LibraryMgr_t(const TCHAR* ptcFileName) : m_hModule(0), m_bAutoFree(true) {     LoadLibrary( ptcFileName ); } /**  * 解構子  *   * @param ptcFileName DLL 名稱  */ LibraryMgr_t::~LibraryMgr_t(void) {     if ( m_hModule && m_bAutoFree )     {         FreeLibrary();     } } /**  * 載入 DLL  *  * @param ptcFileName DLL 名稱  * ＠return true 成功, false = 失敗  */ bool LibraryMgr_t::LoadLibrary(                              const TCHAR* ptcFileName                              ) {     if (! ptcFileName)         return false;     if ( m_hModule )     {         FreeLibrary();     }     m_sFileNAme.assign( ptcFileName );     m_hModule = ::LoadLibrary( m_sFileNAme.c_str() );     return m_hModule != NULL; } /**  * 釋放 DLL  *  * ＠return true 成功, false = 失敗  */ bool LibraryMgr_t::FreeLibrary() {     if( ! m_hModule )         return false;     ::FreeLibrary( m_hModule );     m_hModule = NULL;     m_sFileNAme.clear();     return true; } /**  * 取得指定函數名稱的函數指標  *  * @param ptcProcName 函數名稱  * ＠return 成功回傳函數指標，失敗回傳 NULL  */ FARPROC LibraryMgr_t::GetProcAddress(         const TCHAR* ptcProcName         ) {     if( ! m_hModule )         return NULL;     return ::GetProcAddress( m_hModule, ptcProcName ); } /**  * 取得模組識別碼  *  * ＠return 成功回傳模組識別碼，否則回傳 NULL  */ HMODULE LibraryMgr_t::GetModuleHandle() {     if( ! m_hModule )         return NULL;     return ::GetModuleHandle( m_sFileNAme.c_str() ); } /**  * 取得模組名稱  *  * @param sName 回傳模組名稱  * ＠return true 成功, false = 失敗  */ bool LibraryMgr_t::GetModuleFileName(std::basic_string<TCHAR>& sName) {     if( ! m_hModule )         return false;     sName.assign( m_sFileNAme );     return true; } /**  * 釋放 DLL 並離開執行緒  *  * @param dwExitCode 結束代碼  * ＠return true 成功, false = 失敗  */ bool LibraryMgr_t::FreeLibraryAndExitThread( DWORD dwExitCode ) {     if( ! m_hModule )         return false;     ::FreeLibraryAndExitThread( m_hModule, dwExitCode );     m_hModule = 0;     return true; }

範例

// 引入 LibraryMrg 類別標頭檔 #include "LibraryMgr.h" int _tmain(int argc, _TCHAR* argv[]){ // 載入指定的 DLL LibraryMgr_t lm( _T("C:\MyProg\MyLib.dll") ); typedef int (*Test_t)(void); Test_t TestFn; // 取得 DLL 中的函數指標 TestFn = (Test_t) lm.GetProcAddress( _T("Test") ); if(TestFn){ // 執行 DLL 中的函數 TestFn(); } return 0; }

參考資料
[1] Yet another DLL manager class

廣告