一个Win32 C++ 动态连接库的模板 --- 调用方可管理DLL分配的内存

 一个Win32 C++  动态连接库的模板

cheungmine

  一般情形下，使用C++编写动态库（DLL），在方法参数中不能输出STL对象类型， 如 std::string，这给我们的工作带来极大的麻烦。我参考一些资料，写了个C++ DLL模板，专门解决了这个问题。

     使用VS2003新建一个Win32 DLL工程——DllCppTmpl，选择空项目，即什么也不自动创建。然后把下面2个文件加入到你的工程中，编译之。

     文件1：DllCppTmpl.h

 

//
// DllCppTmpl.h - Win32 CPP Dll Template
// cheungmine
// under licence BSD
// 2007
//
#ifndef _DLLCPPTMPL_H_
#define _DLLCPPTMPL_H_

//
// DllExport
//
#ifdef DLLCPPTMPL_DLLEXPORT
    #define DLLCPPTMPL_DLL __declspec(dllexport)
#elif defined(DLLCPPTMPL_IMPORTS)
    #define DLLCPPTMPL_DLL __declspec(dllimport)
#else
    #define DLLCPPTMPL_DLL
#endif

//
// 公共头文件
//
#include <string>
using namespace std;

#include <new>    // for new_handler

#define DLLCPPTMPL_API    __cdecl

typedef void * (DLLCPPTMPL_API * PtrNew)(size_t);
typedef void (DLLCPPTMPL_API * PtrDelete)(void *);
typedef void (DLLCPPTMPL_API * PtrGetNewAndDelete)(PtrNew &, PtrDelete &);
typedef new_handler (DLLCPPTMPL_API * PtrSetNewHandler)(new_handler);
typedef void (DLLCPPTMPL_API * PtrSetNewAndDelete)(PtrNew, PtrDelete, PtrSetNewHandler);

//
// DLLCPPTMPL_IMPORTS
//
#ifdef DLLCPPTMPL_IMPORTS
    #ifdef _DLL
        // cause CRT DLL to be initialized before Crypto++ so that we can use malloc and free during DllMain()
        #ifdef NDEBUG
            #pragma comment(lib, "msvcrt")
        #else
            #pragma comment(lib, "msvcrtd")
        #endif
    #endif

    #if !(defined(_MSC_VER) && (_MSC_VER < 1300))
        using std::new_handler;
    #endif

    
    // When DllCppTmpl attaches to a new process, it searches all modules loaded 
    // into the process space for exported functions "GetNewAndDeleteFor_DllCppTmpl" 
    // and "SetNewAndDeleteFrom_DllCppTmpl". If one of these functions is found, 
    // Crypto++ uses methods 1 or 2, respectively, by calling the function. 
    // Otherwise, method 3 is used. 
    static PtrNew _s_pNew = NULL;
    static PtrDelete _s_pDelete = NULL;

    extern "C" __declspec(dllexport) void __cdecl SetNewAndDeleteFrom_DllCppTmpl(PtrNew pNew, PtrDelete pDelete, PtrSetNewHandler pSetNewHandler)
    ...{
        _s_pNew = pNew;
        _s_pDelete = pDelete;
    }

    void * __cdecl operator new (size_t size)
    ...{
        return _s_pNew(size);
    }

    void __cdecl operator delete (void * p)
    ...{
        _s_pDelete(p);
    }
#endif    // DLLCPPTMPL_IMPORTS


//
// DLL接口方法
//
namespace DllCppTmpl
...{        

void  DLLCPPTMPL_DLL Test1(const char *in, char** out);

void  DLLCPPTMPL_DLL Test2(const string& in, string& out);

};    // Can be ignored!
/**///////////////////////////////////////////////////////////////////////////////////
#endif /* ndef _DLLCPPTMPL_H_ */

          文件2：DllCppTmpl.cpp

#define DLLCPPTMPL_DLLEXPORT  __declspec(dllexport)

#include "DllCppTmpl.h"

#include <string.h>

#include <iostream>
#include <time.h>
#include <windows.h>


#if (_MSC_VER >= 1000)
#include <crtdbg.h>        // for the debug heap
#endif


/**/////////////////////////////////////////////////////
// 如果使用动态库: DllCppTmpl
using namespace DllCppTmpl;

//
// DLL标准输出
//
#ifdef DLLCPPTMPL_EXPORTS
    #if !(defined(_MSC_VER) && (_MSC_VER < 1300))
        using std::new_handler;
        using std::set_new_handler;
    #endif

    void _CallNewHandler()
    ...{
        new_handler newHandler = set_new_handler(NULL);
        if (newHandler)
            set_new_handler(newHandler);

        if (newHandler)
            newHandler();
        else
            throw std::bad_alloc();
    }

    static PtrNew        _s_pNew = NULL;
    static PtrDelete    _s_pDelete = NULL;

    static void * New (size_t size)
    ...{
        void *p;
        while (!(p = malloc(size)))
            _CallNewHandler();

        return p;
    }

    static void SetNewAndDeleteFunctionPointers()
    ...{
        void *p = NULL;
        HMODULE hModule = NULL;
        MEMORY_BASIC_INFORMATION mbi;

        while (true)
        ...{
            VirtualQuery(p, &mbi, sizeof(mbi));

            if (p >= (char *)mbi.BaseAddress + mbi.RegionSize)
                break;

            p = (char *)mbi.BaseAddress + mbi.RegionSize;

            if (!mbi.AllocationBase || mbi.AllocationBase == hModule)
                continue;

            hModule = HMODULE(mbi.AllocationBase);

            PtrGetNewAndDelete pGetNewAndDelete = (PtrGetNewAndDelete)GetProcAddress(hModule, "GetNewAndDeleteFor_DllCppTmpl");
            if (pGetNewAndDelete)
            ...{
                pGetNewAndDelete(_s_pNew, _s_pDelete);
                return;
            }

            PtrSetNewAndDelete pSetNewAndDelete = (PtrSetNewAndDelete)GetProcAddress(hModule, "SetNewAndDeleteFrom_DllCppTmpl");
            if (pSetNewAndDelete)
            ...{
                _s_pNew = &New;
                _s_pDelete = &free;
                pSetNewAndDelete(_s_pNew, _s_pDelete, &set_new_handler);
                return;
            }
        }

        hModule = GetModuleHandle("msvcrtd");
        if (!hModule)
            hModule = GetModuleHandle("msvcrt");
        if (hModule)
        ...{
            _s_pNew = (PtrNew)GetProcAddress(hModule, "??2@YAPAXI@Z");        // operator new
            _s_pDelete = (PtrDelete)GetProcAddress(hModule, "??3@YAXPAX@Z");    // operator delete
            return;
        }

        OutputDebugString("DllCppTmpl was not able to obtain new and delete function pointers. ");
        throw 0;
    }

    void * operator new (size_t size)
    ...{
        if (!_s_pNew)
            SetNewAndDeleteFunctionPointers();

        return _s_pNew(size);
    }

    void operator delete (void * p)
    ...{
        _s_pDelete(p);
    }

    void * operator new [] (size_t size)
    ...{
        return operator new (size);
    }

    void operator delete [] (void * p)
    ...{
        operator delete (p);
    }
#endif    // #ifdef DLLCPPTMPL_EXPORTS


/**//////////////////////////////////////////
// Test
void  DLLCPPTMPL_DLL DllCppTmpl::Test1(const char *in, char** out)
...{
    *out = new char[strlen(in) + 20];
    strcpy(*out, "Hello ");
    strcat(*out, in);
}

void  DLLCPPTMPL_DLL DllCppTmpl::Test2(const string& in, string& out)
...{
    out = "hello ";
    out += in;
}

 

      好了，开始写客户端。建一个Win32 console项目，如下：

 

// testDllCppTmpl.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"

#include <windows.h>

#define DLLCPPTMPL_IMPORTS    // 必须定义在前

#include "../DllCppTmpl/DllCppTmpl.h"
#pragma comment(lib, "../DllCppTmpl/Debug/DllCppTmpl.lib")

/**///////////////////////////////////////////////////////////////////////
using namespace DllCppTmpl;
/**/////////////////////////////////////////////////////////
int _tmain(int argc, _TCHAR* argv[])
...{
    char in[] = "cheungmine";
    char* out = 0;
    
    DllCppTmpl::Test1(in, &out);
    
    printf("Test1: %s ", out);
    delete out;


    string in2="cheungmine";
    string out2;

    DllCppTmpl::Test2(in2, out2);

    printf("Test2: %s ", out2.c_str());


    return 0;
}

 

注意路径的正确，和DllCppTmpl.dll要放到system32下。