让Squirrel脚本支持Win32DLL调用。

人呢就是受不住表扬，居然有人说我是Squirrel脚本高手！为了不“浪得虚名”特发写东东出来与大伙共享，希望看者有用。

 

下面代码用VS2008+Squirrel2.1以上+SQPluss编译通过

脚本调用例子如下：

    // Demo2. 显示消息对话框
    {
        local dllUser32 = CWin32Dll( "user32.dll" );
        local eMB =
        { // 枚举: 消息框按钮类型
            MB_OK                       =0x00000000,
            MB_OKCANCEL                 =0x00000001,
            MB_ABORTRETRYIGNORE         =0x00000002,
            MB_YESNOCANCEL              =0x00000003,
            MB_YESNO                    =0x00000004,
            MB_RETRYCANCEL              =0x00000005,  
        }
        local eID =
        { // 枚举: 消息框返回值
            IDOK                =1,
            IDCANCEL            =2,
            IDABORT             =3,
            IDRETRY             =4,
            IDIGNORE            =5,
            IDYES               =6,
            IDNO                =7, 
        }   
        local iRet = dllUser32.MessageBoxA( 0, "脚本调用动态库!", "提示信息", eMB.MB_YESNOCANCEL );
        print( iRet == eID.IDYES );
    }

======================================================================

下面是源代码：

// 头文件ScriptImportWin32Dll.h

//======================================================================= // 类 名: CWin32Dll // 头 文 件: ScriptImportWin32Dll.h // 实现文件: ScriptImportWin32Dll.cpp // 功能描述: 脚本导入并执行dll // 依 赖 于: // 被引用于: // 创建日期: 2010年10月11日 10时45分15秒 // 用法举例: // 修改说明: // 修改日期 修改人 修改说明 //======================================================================= #pragma once #include "sqplus.h" _DECL_CLASS(CWin32Dll); class CWin32Dll { public: CWin32Dll(void); ~CWin32Dll(void); BOOL LoadDLL( LPCTSTR lpszDllFilePath ); BOOL RunFunction( DWORD* dwFuncs,UINT nSize ); HMODULE m_hMod; }; // end 头文件

// 下面是cpp文件

#include "stdafx.h" #include "ScriptImportWin32Dll.h" // class CSmartScriptFunctionParam { public: ~CSmartScriptFunctionParam(){}; CSmartScriptFunctionParam( const SQChar* vIn ){ m_strString = vIn; }; CSmartScriptFunctionParam( SQFloat vIn ) { m_fFloat = vIn; }; CSmartScriptFunctionParam( SQBool vIn ) { m_bBool = vIn; }; CSmartScriptFunctionParam( SQInteger vIn ) { m_iInt = vIn; }; LPCTSTR GetString() { return ( LPCTSTR )m_strString; }; SQFloat GetFloat() { return m_fFloat; }; SQInteger GetInteger(){ return m_iInt; }; SQBool GetBool() { return m_bBool; }; private: SQObjectType m_eParamType; CString m_strString; SQFloat m_fFloat; SQInteger m_iInt; SQBool m_bBool; }; CWin32Dll::CWin32Dll() { m_hMod = NULL; } BOOL CWin32Dll::LoadDLL( LPCTSTR lpszDllFilePath ) { if ( strcmp( lpszDllFilePath, _T( "" ) ) == 0 ) { return FALSE; } if (m_hMod) { ::FreeLibrary ( m_hMod ); m_hMod = NULL; } m_hMod = ::LoadLibrary( lpszDllFilePath ); return m_hMod != NULL; } CWin32Dll::~CWin32Dll() { if (m_hMod) { ::FreeLibrary ( m_hMod ); } } BOOL CWin32Dll::RunFunction(DWORD* dwFuncs,UINT nSize) { if( !dwFuncs || !nSize) { return FALSE; } if ( !m_hMod ) { return FALSE; } DWORD dwFunction = (DWORD)::GetProcAddress( m_hMod /*GetModuleHandleA( (const char*)dwFuncs[0] )*/, (const char*)dwFuncs[1] ); if( !dwFunction ) { return FALSE; } DWORD dwTemp(0); _asm { pushad; mov ebx, nSize; mov edx, dword ptr dwFuncs; cmp ebx,3 jb LEXIT; L1: dec ebx cmp ebx,2 je LCALL mov eax, [edx + 4*ebx] push eax loop L1; LCALL: mov dword ptr[dwTemp],edx call dword ptr [dwFunction] mov edx, dword ptr[dwTemp] mov dword ptr[edx + 2*4],eax LEXIT: popad; } return TRUE; } DECLARE_INSTANCE_TYPE( CWin32Dll ); _IMPL_NATIVE_CONSTRUCTION(CWin32Dll,CWin32Dll); _MEMBER_FUNCTION_IMPL(CWin32Dll,constructor) { StackHandler sa(v); CWin32Dll * newv = NULL; int nparams = sa.GetParamCount(); CString strDllFiePath( _T( "" ) ), strTmp; CWin32Dll *pstrIn; if ( sa.GetParamCount() == 2 ) // 必须带一个参数 { if ( sa.GetType( 2 ) == OT_STRING ) { strDllFiePath = sa.GetString( 2 ); } else { strTmp.Format ( _T( "函数名称必须为字符串!" ), strDllFiePath ); return sq_throwerror( v, ( LPCTSTR )strTmp ); } } newv = new CWin32Dll( ); if ( !newv ) { strTmp.Format ( _T( "创建 CWin32Dll 对象失败!" ) ); return sq_throwerror( v, ( LPCTSTR )strTmp ); } BOOL bRet = newv->LoadDLL ( strDllFiePath ); if ( !bRet ) { strTmp.Format ( _T( "载入[%s]失败!" ), strDllFiePath ); return sq_throwerror( v, ( LPCTSTR )strTmp ); } return construct_CWin32Dll(newv); return 1; } //======================================================================= // 函数原型: // 功能描述: // 参数说明: 名称 类型 说明 // 返 回 值: // 依 赖 于: 无 // 引 用 表: 无 // 更 新 表: 无 // 被引用于: 无 // 创建日期: 2010年10月18日 9时49分54秒 // 修改记录: // 修改日期 修改人 修改说明 // //======================================================================= int _CWin32Dll_Invoke(HSQUIRRELVM v) { StackHandler sa(v); _CHECK_SELF(CWin32Dll,CWin32Dll); CString strTmp; int iRet = SQ_OK; const SQChar* funcname = NULL; if (SQ_FAILED(sq_getstring(v,-1,&funcname))) { return sa.ThrowError("internal Invoke error function name expected"); } unsigned int iPassedParamsCount = sa.GetParamCount() - 2; // -2 is 'this' and -1 is funcname int iParamCountCall = iPassedParamsCount + 3; // 0.DLL名称, 2.返回值 DWORD *pdwParamCall = new DWORD[ iParamCountCall ]; if ( !pdwParamCall ) { strTmp.Format ( _T( "为 %s 参数申请内存失败!" ) ); return sq_throwerror( v, ( LPCTSTR )strTmp ); } CPtrList listParams; memset( pdwParamCall, 0x00, sizeof( DWORD ) * iParamCountCall ); pdwParamCall[ 0 ] = (DWORD)( _T( "" ) ); // Dll File Path pdwParamCall[ 1 ] = (DWORD)( ( LPCTSTR )funcname ); // FunctionName pdwParamCall[ 2 ] = (DWORD)(-1); // Return value CSmartScriptFunctionParam *pNewParam = NULL; for ( int iPassedParamsIndex = 0; iPassedParamsIndex < iPassedParamsCount; iPassedParamsIndex ++ ) { SquirrelObject sqVal = SquirrelObject( sa.GetObjectHandle( iPassedParamsIndex + 2 ) ); switch( sqVal.GetType () ) { case OT_FLOAT: { pNewParam = new CSmartScriptFunctionParam( sqVal.ToFloat( ) ); if ( !pNewParam ) { strTmp.Format ( _T( "为参数申请内存失败!" ) ); return sq_throwerror( v, ( LPCTSTR )strTmp ); } listParams.AddTail ( pNewParam ); pdwParamCall[ iPassedParamsIndex + 3 ] = (DWORD)( pNewParam->GetFloat () ); } break; case OT_BOOL: { pNewParam = new CSmartScriptFunctionParam( sqVal.ToBool( ) ); if ( !pNewParam ) { strTmp.Format ( _T( "为参数申请内存失败!" ) ); return sq_throwerror( v, ( LPCTSTR )strTmp ); } listParams.AddTail ( pNewParam ); pdwParamCall[ iPassedParamsIndex + 3 ] = (DWORD)( pNewParam->GetBool () ); } break; case OT_INTEGER: { pNewParam = new CSmartScriptFunctionParam( sqVal.ToInteger( ) ); if ( !pNewParam ) { strTmp.Format ( _T( "为参数申请内存失败!" ) ); return sq_throwerror( v, ( LPCTSTR )strTmp ); } listParams.AddTail ( pNewParam ); pdwParamCall[ iPassedParamsIndex + 3 ] = (DWORD)( pNewParam->GetInteger () ); } break; case OT_STRING: { pNewParam = new CSmartScriptFunctionParam( sqVal.ToString( ) ); if ( !pNewParam ) { strTmp.Format ( _T( "为参数申请内存失败!" ) ); return sq_throwerror( v, ( LPCTSTR )strTmp ); } listParams.AddTail ( pNewParam ); pdwParamCall[ iPassedParamsIndex + 3 ] = (DWORD)( pNewParam->GetString () ); } break; default: strTmp.Format ( _T( "函数[%s]的第[%d]个参数错误:目前函数如参只支持OT_FLOAT,OT_BOOL,OT_INTEGER,OT_STRING" ), sa.GetString( 2 ), iPassedParamsIndex ); sq_throwerror( v, ( LPCTSTR )strTmp ); iRet = SQ_ERROR; goto __finished; break; } } // 调用 BOOL bRet = self->RunFunction( pdwParamCall, iParamCountCall ); if ( !bRet ) { strTmp.Format ( _T( "函数[%s]调用失败!" ), sa.GetString( 2 ) ); sq_throwerror( v, ( LPCTSTR )strTmp ); iRet = SQ_ERROR; goto __finished; } int iRetCall = int( pdwParamCall[ 2 ] ); __finished: if ( pdwParamCall ) { delete [] pdwParamCall; } // 释放 listParams 内存! while( !listParams.IsEmpty () ) { CSmartScriptFunctionParam *pTmp = ( CSmartScriptFunctionParam * )listParams.RemoveHead (); if ( pTmp ) { delete pTmp; } } if ( iRet == SQ_OK ) { return sa.Return( iRetCall ); } return iRet; }// end _CWin32Dll_Invoke //======================================================================= // 函数原型: // 功能描述: // 参数说明: 名称 类型 说明 // 返 回 值: // 依 赖 于: 无 // 引 用 表: 无 // 更 新 表: 无 // 被引用于: 无 // 创建日期: 2010年10月18日 8时43分54秒 // 修改记录: // 修改日期 修改人 修改说明 // //======================================================================= _MEMBER_FUNCTION_IMPL(CWin32Dll,_get) { StackHandler sa(v); _CHECK_SELF(CWin32Dll,CWin32Dll); const SQChar *szFunctionName = sa.GetString(2); sq_pushstring(v, szFunctionName,-1); sq_newclosure( SquirrelVM::GetVMPtr(), _CWin32Dll_Invoke,1); return 1; }// end _get _BEGIN_CLASS(CWin32Dll) _MEMBER_FUNCTION(CWin32Dll,constructor,2,_T("xs")) _MEMBER_FUNCTION(CWin32Dll,_get, 2, _T("xs|x")) _END_CLASS(CWin32Dll) //end ScriptImportWin32Dll.cpp