为Squirrel脚本语言增加COM支持

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

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

脚本代码样子如下：

local objCOM = CAutomationObject( "HZZSKJIM.Controler.1" ); // 新建 COM 对象

objCOM.FunName( ix, iy );

ix = objCOM.iX;

ScriptImportWin32COM.h文件如下：

#pragma once #include "sqplus.h" // Added by Lusp typedef SQChar NCHAR; typedef SquirrelObject ScriptObject; typedef SquirrelVM ScriptVM; typedef CString String; #define DECLARE_SMARTPTR(ifacename) typedef CComQIPtr<ifacename> ifacename##Pointer; _DECL_CLASS(CSmartVariant); _DECL_CLASS(CAutomationObject); // 1. _DECL_EXTERNAL_CONSTRUCTION #define _DECL_EXTERNAL_CONSTRUCTION(classname,cppclass) / BOOL push_##classname(cppclass &quat); / SquirrelObject new_##classname(cppclass *quat); #define _IMPL_EXTERNAL_CONSTRUCTION(classname,cppclass) / static int classname##_release_hook(SQUserPointer p, int size) / { / if(p) { / cppclass *pv = (cppclass *)p; / delete pv; / } / return 0; / } / BOOL push_##classname(cppclass *quat) / { / cppclass *newquat = new cppclass; / *newquat = *quat; / if(!CreateNativeClassInstance(SquirrelVM::GetVMPtr(),_SC(#classname),newquat,classname##_release_hook)) { / delete newquat; / return FALSE; / } / return TRUE; / } / SquirrelObject new_##classname(cppclass *quat) / { / SquirrelObject ret; / if(push_##classname(quat)) { / ret.AttachToStackObject(-1); / sq_pop(SquirrelVM::GetVMPtr(),1); / } / return ret; / } / int construct_##classname(cppclass *p) / { / sq_setinstanceup(SquirrelVM::GetVMPtr(),1,p); / sq_setreleasehook(SquirrelVM::GetVMPtr(),1,classname##_release_hook); / return 1; / } // 2. _DECL_SCALAR_CONSTRUCTION #define _DECL_SCALAR_CONSTRUCTION(classname,cppclass) / BOOL push_##classname(cppclass &quat); / SquirrelObject new_##classname(cppclass &quat); #define _IMPL_SCALAR_CONSTRUCTION(classname,cppclass) / static int classname##_release_hook(SQUserPointer p, int size) / { / if(p) { / cppclass *pv = (cppclass *)p; / delete pv; / } / return 0; / } / BOOL push_##classname(cppclass &quat) / { / cppclass *newquat = new cppclass; / *newquat = quat; / if(!CreateNativeClassInstance(SquirrelVM::GetVMPtr(),_SC(#classname),newquat,classname##_release_hook)) { / delete newquat; / return FALSE; / } / return TRUE; / } / SquirrelObject new_##classname(cppclass &quat) / { / SquirrelObject ret; / if(push_##classname(quat)) { / ret.AttachToStackObject(-1); / sq_pop(SquirrelVM::GetVMPtr(),1); / } / return ret; / } / int construct_##classname(cppclass *p) / { / sq_setinstanceup(SquirrelVM::GetVMPtr(),1,p); / sq_setreleasehook(SquirrelVM::GetVMPtr(),1,classname##_release_hook); / return 1; / } //end Added by Lusp ScriptObject OLE2SQ(VARIANT &v); BOOL SQ2OLE(VARIANTARG &arg,ScriptObject &sqo); typedef std::map<String,FUNCDESC*> FuncDescMap; DECLARE_SMARTPTR(IDispatch); DECLARE_SMARTPTR(ITypeInfo); DECLARE_SMARTPTR(IEnumVARIANT); class CSmartVariant : public VARIANT { public: CSmartVariant() { VariantInit(this); } CSmartVariant(const VARIANT &o) { VariantCopy((VARIANTARG*)this,(VARIANTARG*)&o); } void Set(const VARIANT &o) { VariantCopy((VARIANTARG*)this,(VARIANTARG*)&o); } ~CSmartVariant() { VariantClear(this); } }; class CAutomationObject { public: CAutomationObject(void); ~CAutomationObject(); BOOL Init(IDispatch *pdisp); FUNCDESC* GetFuncOrPropGet(const NCHAR *s); FUNCDESC* GetPropPut(const NCHAR *s); FuncDescMap methods; FuncDescMap get; FuncDescMap put; IDispatchPointer disp; ITypeInfoPointer tinfo; };

ScriptImportWin32COM.CPP文件如下：

 #include "stdafx.h" #include "ScriptImportWin32COM.h" // // EnumVARIANT int EnumVARIANT_release_hook(SQUserPointer p, int size) { IEnumVARIANT *ev = (IEnumVARIANT *)p; ev->Release(); return 0; } ScriptObject new_EnumVARIANT(IEnumVARIANT *p) { ScriptObject ev = ScriptVM::GetRootTable().GetValue("EnumVARIANT"); p->AddRef(); ScriptObject ret; // Modified by Lusp //ScriptVM::CreateInstance(ev,ret,p,EnumVARIANT_release_hook); if ( SQ_FAILED( sq_setinstanceup( ScriptVM::GetVMPtr (), 1, p ) ) ) { delete p; return sq_throwerror( ScriptVM::GetVMPtr (), _SC( "创建 ADODataset 实例失败!" )); } sq_setreleasehook( ScriptVM::GetVMPtr (), 1, EnumVARIANT_release_hook ); ret.AttachToStackObject( - 1 ); // end Modified by Lusp return ret; } int __EnumVARIANT__typeof(HSQUIRRELVM v) { sq_pushstring(v,_SC( "EnumVARIANT" ),-1); return 1; } struct ScriptClassMemberDecl __EnumVARIANT_members[] = { {_SC("_typeof"),__EnumVARIANT__typeof,1,NULL}, {_SC("__value__"), NULL, 0, NULL }, //_PURE_SLOT(__value__) {NULL,NULL,0,NULL} }; struct SquirrelClassDecl __EnumVARIANT_decl = { _SC( "EnumVARIANT" ), NULL, __EnumVARIANT_members }; // end _PURE_SLOT(__value__) // Modified by Lusp DECLARE_INSTANCE_TYPE( CSmartVariant ); _IMPL_SCALAR_CONSTRUCTION(CSmartVariant,CSmartVariant); _MEMBER_FUNCTION_IMPL(CSmartVariant,constructor) { StackHandler sa(v); CSmartVariant val; if (sa.GetParamCount() > 1) { // Modified by Lusp // ScriptObject o = sa.GetObject(2); ScriptObject o = ScriptObject( sa.GetObjectHandle(2) ); VARIANTARG va; SQ2OLE(va,o); val.Set(va); } // Modified by Lusp // return construct_ByRef(val); return construct_CSmartVariant( &val ); } _MEMBER_FUNCTION_IMPL(CSmartVariant,value) { StackHandler sa(v); _CHECK_SELF(CSmartVariant,CSmartVariant); return sa.Return(OLE2SQ(*self)); } _BEGIN_CLASS(CSmartVariant) _MEMBER_FUNCTION(CSmartVariant,constructor,-1,_T("x.")) _MEMBER_FUNCTION(CSmartVariant,value,0,NULL) _END_CLASS(CSmartVariant) // end EnumVARIANT // // // CAutomationObject CAutomationObject::CAutomationObject(void) {// empty } CAutomationObject::~CAutomationObject(void) { if (tinfo != NULL) { FuncDescMap::iterator itr; for( itr = methods.begin (); itr != methods.end (); itr ++ ) { tinfo->ReleaseFuncDesc(itr->second); } methods.clear(); for( itr = get.begin (); itr != get.end (); itr ++ ) { tinfo->ReleaseFuncDesc(itr->second); } get.clear(); for( itr = put.begin (); itr != put.end (); itr ++ ) { tinfo->ReleaseFuncDesc(itr->second); } put.clear(); } } BOOL CAutomationObject::Init(IDispatch *pdisp) { HRESULT hr; disp = pdisp; ITypeInfo *ti; TYPEATTR *ta = NULL; FUNCDESC *fd = NULL; BOOL result = FALSE; hr = pdisp->GetTypeInfo(0,LOCALE_SYSTEM_DEFAULT,&ti); if(FAILED(hr)) { goto __error; } tinfo.Attach(ti); hr = ti->GetTypeAttr(&ta); if(FAILED(hr)) { goto __error; } unsigned int nfuncs = ta->cFuncs; unsigned int n; for (n = 0; n < nfuncs; n++) { hr = ti->GetFuncDesc(n,&fd); if(FAILED(hr)) { goto __error; } BSTR funcname; unsigned int count; hr = ti->GetNames(fd->memid,&funcname,1,&count); if(FAILED(hr)) { goto __error; } switch (fd->invkind) { case INVOKE_PROPERTYGET: get.insert(FuncDescMap::value_type(CString( funcname ),fd)); break; case INVOKE_PROPERTYPUT: put.insert(FuncDescMap::value_type(CString( funcname ),fd)); break; case INVOKE_PROPERTYPUTREF: TRACE1( "WARNING %s is INVOKE_PROPERTYPUTREF(not supported yet)",funcname); break; case INVOKE_FUNC: methods.insert(FuncDescMap::value_type(CString( funcname ),fd)); break; default: _ASSERT(0); break; } ::SysFreeString(funcname); } result = TRUE; __error: if (ta) { ti->ReleaseTypeAttr(ta); } return result; }// end CAutomationObject::Init(IDispatch *pdisp) FUNCDESC* CAutomationObject::GetFuncOrPropGet(const NCHAR *s) { FuncDescMap::iterator itr = methods.find(s); if (itr == methods.end()) { itr = get.find(s); if (itr == get.end()) { return NULL; } } return itr->second; } FUNCDESC* CAutomationObject::GetPropPut(const NCHAR *s) { FuncDescMap::iterator itr = put.find(s); if (itr == put.end()) { return NULL; } return itr->second; } DECLARE_INSTANCE_TYPE( CAutomationObject ); _IMPL_EXTERNAL_CONSTRUCTION(CAutomationObject,CAutomationObject); //_IMPL_NATIVE_CONSTRUCTION(CAutomationObject,CAutomationObject); ScriptObject OLE2SQ(VARIANT &v) { ScriptObject ret; switch (v.vt) { case VT_EMPTY: case VT_NULL: break; case VT_I2: ret = (SQInteger)(v.iVal); break; case VT_I4: ret = (SQInteger)(v.intVal); break; case VT_R4: ret = (SQFloat)(v.fltVal); break; case VT_R8: ret = (SQFloat)(v.dblVal); break; case VT_CY: ret = (SQInteger)( v.cyVal.int64 ); break; case VT_DATE: ASSERT(0); break; case VT_BSTR: ret = ( LPCTSTR )CString( v.bstrVal ); break; case VT_DISPATCH: { CAutomationObject *ao = new CAutomationObject(); ao->Init(v.pdispVal); ret = new_CAutomationObject(ao); } break; case VT_ERROR: ASSERT(0); break; case VT_BOOL: ret = v.boolVal?true:false; break; case VT_VARIANT: ret = OLE2SQ(*v.pvarVal); break; case VT_UNKNOWN: ASSERT(0); break; case VT_DECIMAL: ASSERT(0); break; case VT_RECORD: ASSERT(0); break; case VT_I1: ret = (SQInteger)(v.cVal); break; case VT_UI1: ret = (SQInteger)(v.bVal); break; case VT_UI2: ret = (SQInteger)(v.uiVal); break; case VT_UI4: ret = (SQInteger)(v.uintVal); break; case VT_I8: ret = (SQInteger)(v.llVal); break; case VT_UI8: ret = (SQInteger)(v.ullVal); break; case VT_INT: ret = (SQInteger)(v.intVal); break; case VT_UINT: ret = (SQInteger)(v.uintVal); break; case VT_ARRAY: ASSERT(0); case VT_I2|VT_BYREF: ret = (SQInteger)(*v.piVal); break; case VT_I4|VT_BYREF: ret = (SQInteger)(*v.pintVal); break; case VT_R4|VT_BYREF: ret = (SQFloat)(*v.pfltVal); break; case VT_R8|VT_BYREF: ret = (SQFloat)(*v.pdblVal); break; case VT_CY|VT_BYREF: ASSERT(0); break; case VT_DATE|VT_BYREF: ASSERT(0); break; case VT_BSTR|VT_BYREF: ret = v.bstrVal; break; case VT_DISPATCH|VT_BYREF: { CAutomationObject *ao = new CAutomationObject(); ao->Init(*v.ppdispVal); ret = new_CAutomationObject(ao); } break; case VT_ERROR|VT_BYREF: ASSERT(0); break; case VT_BOOL|VT_BYREF: ret = *v.pboolVal?true:false; break; case VT_VARIANT|VT_BYREF: ret = OLE2SQ(*v.pvarVal); break; case VT_UNKNOWN|VT_BYREF: ASSERT(0); break; case VT_DECIMAL|VT_BYREF: ASSERT(0); break; case VT_RECORD|VT_BYREF: ASSERT(0); break; case VT_I1|VT_BYREF: ret = (SQInteger)(*v.pcVal); break; case VT_UI1|VT_BYREF: ret = (SQInteger)(*v.pbVal); break; case VT_UI2|VT_BYREF: ret = (SQInteger)(*v.puiVal); break; case VT_UI4|VT_BYREF: ret = (SQInteger)(*v.puintVal); break; case VT_I8|VT_BYREF: ret = (SQInteger)(*v.pllVal); break; case VT_UI8|VT_BYREF: ret = (SQInteger)(*v.pullVal); break; case VT_INT|VT_BYREF: ret = (SQInteger)(*v.pintVal); break; case VT_UINT|VT_BYREF: ret = (SQInteger)(*v.puintVal); break; default: ASSERT(0); break; } return ret; }// end OLE2SQ // #define _INTPARAM(type,slot) case type: / // if(sqo.GetType() != OT_INTEGER) { return FALSE; } / // arg.slot = sqo.ToInteger(); / // break; // // #define _FLOATPARAM(type,slot) case type: / // if(sqo.GetType() != OT_FLOAT) { return FALSE; } / // arg.slot = sqo.ToFloat(); / // break; BOOL SQ2OLE(VARIANTARG &arg,ScriptObject &sqo) { switch (sqo.GetType()) { case OT_INTEGER: arg.vt = VT_INT; arg.intVal = sqo.ToInteger(); break; case OT_FLOAT: arg.vt = VT_R4; arg.fltVal = sqo.ToFloat(); break; case OT_STRING: arg.vt = VT_BSTR; arg.bstrVal = ::SysAllocString( _bstr_t( sqo.ToString() ) ); break; case OT_BOOL: arg.vt = VT_BOOL; arg.intVal = sqo.ToBool()?VARIANT_TRUE:VARIANT_FALSE; break; case OT_INSTANCE: { // Modified by Lusp CAutomationObject *ao = (CAutomationObject *)sqo.GetInstanceUP(SQUserPointer(_CLASS_TAG(CAutomationObject))); if (ao) { arg.vt = VT_DISPATCH; //ao->disp->AddRef(); // Modified by Lusp arg.pdispVal = ao->disp; break; } // Modified by Lusp // CSmartVariant *sv = (CSmartVariant *)sqo.GetInstanceUP(_CLASS_TAG(CSmartVariant)); CSmartVariant *sv = (CSmartVariant *)sqo.GetInstanceUP( SQUserPointer( _CLASS_TAG(CSmartVariant) ) ); if (sv) { arg.vt = VT_VARIANT |VT_BYREF; arg.pvarVal = sv; break; } return FALSE; } break; default: return FALSE; } return TRUE; /*if(!(expected & VT_BYREF)) { switch(expected) { case VT_EMPTY: case VT_NULL: break; //INT ------------------------------------------------------- _INTPARAM(VT_I2,iVal) _INTPARAM(VT_I4,intVal) _INTPARAM(VT_I1,cVal) _INTPARAM(VT_UI1,bVal) _INTPARAM(VT_UI2,uiVal) _INTPARAM(VT_UI4,uintVal) _INTPARAM(VT_I8,llVal) _INTPARAM(VT_UI8,ullVal) _INTPARAM(VT_INT,intVal) _INTPARAM(VT_UINT,uintVal) _FLOATPARAM(VT_R4,fltVal) _FLOATPARAM(VT_R8,dblVal) //DATE ------------------------------------------------------ case VT_USERDEFINED; _INTPARAM(VT_I8,llVal); break; case VT_DATE: _ASSERT(0); break; //STRING ---------------------------------------------------- case VT_BSTR: if(sqo.GetType() != OT_STRING) return FALSE; arg.bstrVal = ::SysAllocString(sqo.ToString()); break; //BOOL ------------------------------------------------------ case VT_BOOL: if(sqo.GetType() != OT_BOOL) return FALSE; arg.boolVal = sqo.ToBool()?VARIANT_TRUE:VARIANT_FALSE; break; case VT_DISPATCH:{ if(sqo.GetType() != OT_INSTANCE) return FALSE; CAutomationObject *ao = (CAutomationObject *)sqo.GetInstanceUP(_CLASS_TAG(CAutomationObject)); if(!ao) return FALSE; arg.vt = VT_DISPATCH; ao->disp->AddRef(); arg.pdispVal = ao->disp; } break; case VT_VARIANT: switch(sqo.GetType()) { case OT_INTEGER: arg.vt = VT_I4; arg.intVal = sqo.ToInteger(); return TRUE; break; case OT_FLOAT: arg.vt = VT_R4; arg.fltVal = sqo.ToFloat(); return TRUE; break; case OT_STRING: arg.vt = VT_BSTR; arg.bstrVal = ::SysAllocString(sqo.ToString()); return TRUE; break; case OT_BOOL: arg.vt = VT_BOOL; arg.boolVal = sqo.ToBool()?VARIANT_TRUE:VARIANT_FALSE; return TRUE; break; case OT_NULL: arg.vt = VT_NULL; arg.llVal = 0; return TRUE; break; default: return FALSE; } break; //INT BY REF------------------------------------------------------- default: return FALSE; } } //BY REF-------------------------------------------------------------------- else { CSmartVariant *byref = (CSmartVariant*)sqo.GetInstanceUP(_CLASS_TAG(CSmartVariant)); if(!byref) return FALSE; arg.vt = VT_EMPTY; arg.pvarVal = byref; } arg.vt = expected;*/ return TRUE; }// end SQ2OLE() int ThrowAutomationError(StackHandler &sa,HRESULT hr,EXCEPINFO &pexinfo,IDispatch *obj) { String err = "Unknown Automatio Error(no IErrorInfo)"; ISupportErrorInfo *sei; bool errorfound = false; if (SUCCEEDED(obj->QueryInterface(IID_ISupportErrorInfo,(void**)&sei))) { if (SUCCEEDED(sei->InterfaceSupportsErrorInfo(IID_IErrorInfo))) { IErrorInfo *ei; if (SUCCEEDED(::GetErrorInfo(NULL,&ei))) { BSTR desc; if (ei) { if (SUCCEEDED(ei->GetDescription(&desc))) { errorfound = true; err = desc; ::SysFreeString(desc); } ei->Release(); } else { errorfound = false; } } } sei->Release(); } if (!errorfound) { switch (hr) { case DISP_E_BADPARAMCOUNT: err = "The number of elements provided in DISPPARAMS is different from the number of arguments accepted by the method or property."; break; case DISP_E_BADVARTYPE: err = "One of the arguments in DISPPARAMS is not a valid variant type. "; break; case DISP_E_MEMBERNOTFOUND: err = "The requested member does not exist."; break; case DISP_E_NONAMEDARGS: err = "This implementation of IDispatch does not support named arguments."; break; case DISP_E_OVERFLOW: err = "One of the arguments in DISPPARAMS could not be coerced to the specified type."; break; case DISP_E_PARAMNOTFOUND: err = "One of the parameter IDs does not correspond to a parameter on the method. In this case, puArgErr is set to the first argument that contains the error."; break; case DISP_E_PARAMNOTOPTIONAL: err = "A required parameter was omitted."; break; case DISP_E_TYPEMISMATCH: err = "One or more of the arguments could not be coerced. The index of the first parameter with the incorrect type within rgvarg is returned in puArgErr."; break; case E_INVALIDARG: err = "One of the arguments is invalid."; break; case E_OUTOFMEMORY: err = "Insufficient memory to complete the operation."; break; case DISP_E_EXCEPTION: err = pexinfo.bstrDescription; break; case 0x800706ba/*RPC_S_SERVER_UNAVAILABLE*/: err = "The RPC server is unavailable."; break; default: break; } } return sa.ThrowError(err/*.c_str()*/); // Modified by Lusp }// end ThrowAutomationError() _MEMBER_FUNCTION_IMPL(CAutomationObject,constructor) { StackHandler sa(v); CLSID clsid; HRESULT hr; IDispatch *pdisp; BSTR bstrProgid = _bstr_t( sa.GetString(2) ); if (FAILED(hr = CLSIDFromProgID( bstrProgid, &clsid ) )) { return sa.ThrowError("Invalid Prog ID"); } if (FAILED(hr = CoCreateInstance(clsid,NULL,CLSCTX_ALL , __uuidof(IDispatch),(void**)&pdisp))) { return sa.ThrowError("CoCreateInstance failed"); } CAutomationObject *ao = new CAutomationObject(); if (!ao->Init(pdisp)) { pdisp->Release(); delete ao; return sa.ThrowError("Error enumerating IDispatch"); } pdisp->Release(); sq_setinstanceup(v,1,(SQUserPointer)ao); sq_setreleasehook(v,1,CAutomationObject_release_hook); // construct_CAutomationObject( ao ); return 1; } _MEMBER_FUNCTION_IMPL(CAutomationObject,_set) { StackHandler sa(v); _CHECK_SELF(CAutomationObject,CAutomationObject); const NCHAR *index = sa.GetString(2); FUNCDESC *desc = self->GetPropPut(index); if (!desc) { return sa.ThrowError("invalid property"); } DISPPARAMS dp; VARIANT res; DISPID mydispid = DISPID_PROPERTYPUT; unsigned int errarg; VARIANTARG inarg; memset(&dp,0,sizeof(dp)); dp.cArgs = 1; dp.cNamedArgs = 1; dp.rgdispidNamedArgs = &mydispid; dp.rgvarg = &inarg; // Modified by Lusp // ScriptObject o = sa.GetObject(3); ScriptObject o = ScriptObject( sa.GetObjectHandle(3) ); ELEMDESC &ed = desc->lprgelemdescParam[0]; VariantInit(&inarg); if (!SQ2OLE(inarg,o)) { return sa.ThrowError("invalid param types(or not supported yet"); } EXCEPINFO exinfo; /*HRESULT hr = DispInvoke(self->disp,self->tinfo,desc->memid, DISPATCH_PROPERTYPUT,&dp,&res,&exinfo,&errarg);*/ HRESULT hr = self->disp->Invoke(desc->memid, IID_NULL,LOCALE_SYSTEM_DEFAULT, DISPATCH_PROPERTYPUT, &dp,&res, &exinfo,&errarg); if (FAILED(hr)) { return ThrowAutomationError(sa,hr,exinfo,self->disp); } ScriptObject ret = OLE2SQ(res); VariantClear(&res); return sa.Return(ret); } int _CAutomation_Invoke(HSQUIRRELVM v) { StackHandler sa(v); _CHECK_SELF(CAutomationObject,CAutomationObject); const NCHAR *funcname; if (SQ_FAILED(sq_getstring(v,-1,&funcname))) return sa.ThrowError("internal Invoke error method name expected"); FUNCDESC *desc = self->GetFuncOrPropGet(funcname); if (!desc) { return sa.ThrowError("invalid method"); } #define MAX_ARGS 20 VARIANTARG vargs[MAX_ARGS]; DISPPARAMS params; unsigned int nparams = desc->cParams; unsigned int reqparams = desc->cParams - desc->cParamsOpt; unsigned int npassedparams = sa.GetParamCount() - 2; // -2 is 'this' and 1 'free variable' int retvalparam = -1; //checks if any of the params is a retval so that can be //considered 'not required' for (unsigned int j = 0; j < nparams; j++) { ELEMDESC &ed = desc->lprgelemdescParam[j]; if (ed.paramdesc.wParamFlags & PARAMFLAG_FRETVAL) { retvalparam = j; reqparams--; break; } } if (npassedparams < reqparams || npassedparams > reqparams) { return sa.ThrowError("wrong number of parameters"); } memset(¶ms,0,sizeof(params)); params.rgvarg = vargs; params.cArgs = npassedparams; for (unsigned int j = 0; j < npassedparams; j++) { ELEMDESC &ed = desc->lprgelemdescParam[j]; VARTYPE ptype; if (ed.paramdesc.wParamFlags & PARAMFLAG_FRETVAL) continue; switch (ed.tdesc.vt) { case VT_SAFEARRAY: //VT_ARRAY return sa.ThrowError("safearray not supported"); break; case VT_PTR: //VT_BYREF ptype = ed.tdesc.lptdesc->vt | VT_BYREF; break; default: //BY VAL ptype = ed.tdesc.vt; break; } // Modified by Lusp // ScriptObject sqval = sa.GetObject(j + 2); ScriptObject sqval = ScriptObject( sa.GetObjectHandle(j + 2) ); if (!SQ2OLE(vargs[npassedparams-j-1],sqval)) { return sa.ThrowError("invalid param types(or not supported yet"); } } VARIANT res; ::VariantInit(&res); unsigned int errarg = 0; EXCEPINFO exinfo; memset(&exinfo,0,sizeof(exinfo)); /* DISPID dispid; OLECHAR * szMember = (NCHAR*)funcname; HRESULT hrx = self->disp->GetIDsOfNames(IID_NULL, &szMember, 1, LOCALE_SYSTEM_DEFAULT, &dispid);*/ HRESULT hr = self->disp->Invoke( desc->memid, IID_NULL, LOCALE_SYSTEM_DEFAULT, DISPATCH_METHOD, ¶ms, &res, &exinfo, &errarg); /*HRESULT hr = DispInvoke(self->disp.GetPtr(),self->tinfo,desc->memid, DISPATCH_METHOD,¶ms,&res,&exinfo,&errarg);*/ if (FAILED(hr)) { return ThrowAutomationError(sa,hr,exinfo,self->disp); } ScriptObject ret = OLE2SQ(res); VariantClear(&res); return sa.Return(ret); } _MEMBER_FUNCTION_IMPL(CAutomationObject,_get) { StackHandler sa(v); _CHECK_SELF(CAutomationObject,CAutomationObject); if (sa.GetType(2) == OT_INSTANCE) { // Modified by Lusp // ScriptObject _this = sa.GetObject(2); ScriptObject _this = ScriptObject( sa.GetObjectHandle(2) ); return sa.Return(_this.GetValue("__value__")); } const NCHAR *index = sa.GetString(2); FUNCDESC *desc = self->GetFuncOrPropGet(index); if (!desc) return sa.ThrowError("index not found"); if (desc->invkind == INVOKE_PROPERTYGET) { //get! DISPPARAMS dp; VARIANT res; unsigned int errarg; EXCEPINFO exinfo; memset(&dp,0,sizeof(dp)); HRESULT hr = self->disp->Invoke(desc->memid,IID_NULL,LOCALE_SYSTEM_DEFAULT, DISPATCH_PROPERTYGET,&dp,&res,&exinfo,&errarg); if (FAILED(hr)) { return ThrowAutomationError(sa,hr,exinfo,self->disp); } ScriptObject ret = OLE2SQ(res); VariantClear(&res); return sa.Return(ret); } else //INVOKE_FUNC { sq_pushstring(v,index,-1); sq_newclosure(ScriptVM::GetVMPtr(),_CAutomation_Invoke,1); return 1; } return 0; } _MEMBER_FUNCTION_IMPL(CAutomationObject,_nexti) { StackHandler sa(v); _CHECK_SELF(CAutomationObject,CAutomationObject); ScriptObject enumVARIANT; IEnumVARIANT *ev = NULL; if (sa.GetType(2) == OT_NULL) { /*if(SUCCEEDED(self->disp->QueryInterface(__uuidof(IEnumVARIANT),(void**)&ev))) { enumVARIANT = new_EnumVARIANT(ev); } else { return sa.ThrowError("iteration not supported"); }*/ FUNCDESC *desc = self->GetFuncOrPropGet("_NewEnum"); if (!desc || desc->invkind != INVOKE_PROPERTYGET) { return sa.ThrowError("not iterateble"); } DISPPARAMS dp; VARIANT res; unsigned int errarg; memset(&dp,0,sizeof(dp)); HRESULT hr = self->disp->Invoke(desc->memid,IID_NULL,LOCALE_SYSTEM_DEFAULT, DISPATCH_PROPERTYGET,&dp,&res,NULL,&errarg); if (FAILED(hr)) { return sa.ThrowError("_NewEnum failed"); } if (res.vt != VT_UNKNOWN) { return sa.ThrowError("unexpected type"); } hr = res.punkVal->QueryInterface(__uuidof(IEnumVARIANT),(void**)&ev); VariantClear(&res); if (SUCCEEDED(hr)) { enumVARIANT = new_EnumVARIANT(ev); } else { return sa.ThrowError("iteration not supported"); } } else { _CHECK_INST_PARAM(evar,2,IEnumVARIANT,EnumVARIANT); ev = evar; // Modified by Lusp // enumVARIANT = sa.GetObject(2); enumVARIANT = ScriptObject( sa.GetObjectHandle(2) ); } VARIANT ret; unsigned long fetched = 0; VariantInit(&ret); if (SUCCEEDED(ev->Next(1,&ret,&fetched)) && fetched > 0) { enumVARIANT.SetValue("__value__",OLE2SQ(ret)); VariantClear(&ret); return sa.Return(enumVARIANT); } return sa.Return(); //end iteration } _BEGIN_CLASS(CAutomationObject) _MEMBER_FUNCTION(CAutomationObject,constructor,2,_T("xs")) _MEMBER_FUNCTION(CAutomationObject,_get,2,_T("xs|x")) _MEMBER_FUNCTION(CAutomationObject,_set,3,_T("xs.")) _MEMBER_FUNCTION(CAutomationObject,_nexti,0,NULL) _END_CLASS(CAutomationObject) // end CAutomationObject //