| 1 | #include<StructureConstants.au3>
|
|---|
| 2 | #include<Memory.au3>
|
|---|
| 3 |
|
|---|
| 4 | ;/* Flags for IDispatch::Invoke */
|
|---|
| 5 |
|
|---|
| 6 | Global Const $__Au3Obj_E_NOTIMPL = 0x80004001;
|
|---|
| 7 | Global Const $__Au3Obj_S_OK = 0;
|
|---|
| 8 | Global Const $__Au3Obj_DISP_E_BADINDEX = 0x8002000B
|
|---|
| 9 | Global Const $__Au3Obj_E_POINTER = 0x80004003
|
|---|
| 10 | Global Const $__Au3Obj_E_NOINTERFACE = 0x80004002
|
|---|
| 11 | Global Const $__AU3Obj_DISP_E_UNKNOWNNAME = 0x80020006
|
|---|
| 12 | Global Const $__AU3Obj_DISP_E_EXCEPTION = 0x80020009
|
|---|
| 13 | Global Const $__AU3Obj_DISPID_UNKNOWN = -1
|
|---|
| 14 |
|
|---|
| 15 | Global Const $__Au3Obj_vTable = DllStructCreate("ptr QueryInterface; ptr AddRef; ptr Release; ptr GetTypeInfoCount; ptr GetTypeInfo; ptr GetIDsOfNames; ptr Invoke;")
|
|---|
| 16 | ; set DllCallbacks
|
|---|
| 17 | Dim $aCallbacks[8]
|
|---|
| 18 | $aCallbacks[1] = DllCallbackRegister("__Au3Obj_QueryInterface", "long", "ptr;dword;dword") ;QureyInterface
|
|---|
| 19 | $aCallbacks[2] = DllCallbackRegister("__Au3Obj_AddRef", "long", "ptr") ;Addref
|
|---|
| 20 | $aCallbacks[3] = DllCallbackRegister("__Au3Obj_Release", "long", "ptr") ;Release
|
|---|
| 21 | $aCallbacks[4] = DllCallbackRegister("__Au3Obj_GetTypeInfoCount", "long", "ptr;ptr")
|
|---|
| 22 | $aCallbacks[5] = DllCallbackRegister("__Au3Obj_GetTypeInfo", "long", "ptr;uint;int;ptr")
|
|---|
| 23 | $aCallbacks[6] = DllCallbackRegister("__Au3Obj_GetIDsOfNames", "long", "ptr;ptr;ptr;uint;int;ptr")
|
|---|
| 24 | $aCallbacks[7] = DllCallbackRegister("__Au3Obj_Invoke", "long", "ptr;int;ptr;int;ushort;ptr;ptr;ptr;ptr")
|
|---|
| 25 | For $i = 1 To 7
|
|---|
| 26 | DllStructSetData($__Au3Obj_vTable, $i, DllCallbackGetPtr($aCallbacks[$i]))
|
|---|
| 27 | Next
|
|---|
| 28 |
|
|---|
| 29 |
|
|---|
| 30 | Global Const $__Au3Obj_DISPATCH_METHOD = 0x1
|
|---|
| 31 | Global Const $__Au3Obj_DISPATCH_PROPERTYGET = 0x2
|
|---|
| 32 | Global Const $__Au3Obj_DISPATCH_PROPERTYPUT = 0x4
|
|---|
| 33 | Global Const $__Au3Obj_DISPATCH_PROPERTYPUTREF = 0x8
|
|---|
| 34 |
|
|---|
| 35 | ; In the end we still want the autoit object. This function converts a raw pointer to an autoit object
|
|---|
| 36 | Func ConvertPtrToIDispatch($pIDispatch)
|
|---|
| 37 | ; Author: monoceres
|
|---|
| 38 | ; This would have been 10000x easier if autoit had supported the idispatch* type in dllstructs...
|
|---|
| 39 | ; Fortunetely memcpy can copy the pointer into a idispatch*, lucky us.
|
|---|
| 40 | Local $aCall = DllCall("kernel32.dll", "none", "RtlMoveMemory", "idispatch*", 0, "ptr*", $pIDispatch, "dword", DllStructGetSize(DllStructCreate('ptr',1)))
|
|---|
| 41 | If @error Then Return SetError(1, 0, 0)
|
|---|
| 42 | Return $aCall[1]
|
|---|
| 43 | EndFunc ;==>ConvertPtrToIDispatch
|
|---|
| 44 |
|
|---|
| 45 |
|
|---|
| 46 | Func __Au3Obj_QueryInterface($pObject, $iid, $ppvObject)
|
|---|
| 47 | ; Actually Autoit is not calling this function, so leave it empty for testing ;)
|
|---|
| 48 | Return $__Au3Obj_E_NOINTERFACE
|
|---|
| 49 | EndFunc ;==>__Au3Obj_QueryInterface
|
|---|
| 50 |
|
|---|
| 51 | Func __Au3Obj_AddRef($pObject)
|
|---|
| 52 | ; Author: Prog@ndy
|
|---|
| 53 | Local $st = DllStructCreate("ptr;dword", $pObject)
|
|---|
| 54 | Local $iCount = DllStructGetData($st, 2) + 1
|
|---|
| 55 | DllStructSetData($st, 2, $iCount)
|
|---|
| 56 | Return $iCount
|
|---|
| 57 | EndFunc ;==>__Au3Obj_AddRef
|
|---|
| 58 |
|
|---|
| 59 | Func __Au3Obj_Release($pObject)
|
|---|
| 60 | ; Author: Prog@ndy
|
|---|
| 61 | Local $st = DllStructCreate("ptr;dword", $pObject)
|
|---|
| 62 | Local $iCount = DllStructGetData($st, 2) - 1
|
|---|
| 63 | If $iCount < 0 Then Return 0
|
|---|
| 64 | DllStructSetData($st, 2, $iCount)
|
|---|
| 65 | If $iCount = 0 Then ; delete Object
|
|---|
| 66 | _MemGlobalFree($pObject)
|
|---|
| 67 | EndIf
|
|---|
| 68 | Return $iCount
|
|---|
| 69 | EndFunc ;==>__Au3Obj_Release
|
|---|
| 70 |
|
|---|
| 71 | Func __Au3Obj_GetTypeInfoCount($pObject, $pctinfo)
|
|---|
| 72 | ; Author: Prog@ndy
|
|---|
| 73 | Return $__Au3Obj_E_NOTIMPL
|
|---|
| 74 | EndFunc ;==>__Au3Obj_GetTypeInfoCount
|
|---|
| 75 | Func __Au3Obj_GetTypeInfo($pObject, $iTInfo, $lcid, $ppTInfo)
|
|---|
| 76 | ; Author: Prog@ndy
|
|---|
| 77 | Return $__Au3Obj_E_NOTIMPL
|
|---|
| 78 | EndFunc ;==>__Au3Obj_GetTypeInfo
|
|---|
| 79 | Func __Au3Obj_GetIDsOfNames($pObject, $riid, $rgszNames, $cNames, $lcid, $rgDispId)
|
|---|
| 80 | ; Author: Prog@ndy
|
|---|
| 81 | If $cNames = 0 Then Return $__AU3Obj_DISP_E_UNKNOWNNAME
|
|---|
| 82 |
|
|---|
| 83 | Local $sName = PtrStringRead(DllStructGetData(DllStructCreate('ptr', $rgszNames), 1))
|
|---|
| 84 | If $sName = "example" Then
|
|---|
| 85 | DllStructSetData(DllStructCreate("int", $rgDispId),1, 1) ; give $sName the DispID 1
|
|---|
| 86 | Return $__Au3Obj_S_OK
|
|---|
| 87 | EndIf
|
|---|
| 88 |
|
|---|
| 89 | Return $__AU3Obj_DISP_E_UNKNOWNNAME
|
|---|
| 90 | EndFunc ;==>__Au3Obj_GetIDsOfNames
|
|---|
| 91 | Func PtrStringRead($pStr)
|
|---|
| 92 | ; Author: Prog@ndy
|
|---|
| 93 | Local $aResult = DllCall("kernel32.dll", 'int', 'lstrlenW', 'ptr', $pStr)
|
|---|
| 94 | If @error Or $aResult<1 Then Return SetError(1, 0, 0)
|
|---|
| 95 | Return DllStructGetData(DllStructCreate("wchar[" & $aResult[0] & "]", $pStr), 1)
|
|---|
| 96 | EndFunc ;==>PtrStringRead
|
|---|
| 97 |
|
|---|
| 98 |
|
|---|
| 99 | Func __Au3Obj_Invoke($pObject, $dispIdMember, $riid, $lcid, $wFlags, $pDispParams, $pVarResult, $pExcepInfo, $puArgErr)
|
|---|
| 100 | ; Author: monoceres, Prog@ndy
|
|---|
| 101 | If $dispIdMember <> 1 Then Return $__AU3Obj_DISP_E_EXCEPTION ; we only support the call with ID 1
|
|---|
| 102 | Select
|
|---|
| 103 | Case BitAND($wFlags, $__Au3Obj_DISPATCH_PROPERTYPUTREF)
|
|---|
| 104 | ;$tExcepInfo = DllStructCreate("ushort wCode; ushort wReserved; ptr bstrSource; ptr bstrDescription; ptr bstrHelpFile; ulong dwHelpContext; ptr pvReserved; ptr pfnDeferredFillIn: ushort scode;", $pExcepInfo)
|
|---|
| 105 | ;DllStructSetData($tExcepInfo, 1, 1001)
|
|---|
| 106 | Return $__AU3Obj_DISP_E_EXCEPTION
|
|---|
| 107 | Case $wFlags = BitOR($__Au3Obj_DISPATCH_METHOD, $__Au3Obj_DISPATCH_PROPERTYGET)
|
|---|
| 108 | MsgBox(0, '', "Method call or GET property with dispId " & $dispIdMember)
|
|---|
| 109 | Return $__Au3Obj_S_OK
|
|---|
| 110 | Case BitAND($wFlags, $__Au3Obj_DISPATCH_PROPERTYPUT)
|
|---|
| 111 | MsgBox(0, '', "PUT property with DispId" & $dispIdMember)
|
|---|
| 112 | Return $__Au3Obj_S_OK
|
|---|
| 113 | Case Else
|
|---|
| 114 | Return $__AU3Obj_DISP_E_EXCEPTION
|
|---|
| 115 | EndSelect
|
|---|
| 116 | Return $__AU3Obj_DISP_E_EXCEPTION
|
|---|
| 117 | EndFunc ;==>__Au3Obj_Invoke
|
|---|
| 118 |
|
|---|
| 119 |
|
|---|
| 120 | ; construct object
|
|---|
| 121 | Local $iSize = DllStructGetSize(DllStructCreate("ptr vTable; dword refcount;", 1))
|
|---|
| 122 | Local $pMem = _MemGlobalAlloc($iSize, $GPTR) ; will be freed in Objectz::release
|
|---|
| 123 | Local $tStruct = DllStructCreate("ptr vTable; dword refcount;", $pMem)
|
|---|
| 124 | DllStructSetData($tStruct, 1, DllStructGetPtr($__Au3Obj_vTable))
|
|---|
| 125 | $object = ConvertPtrToIDispatch($pMem)
|
|---|
| 126 | OnAutoItExitRegister("__ReleaseObject")
|
|---|
| 127 |
|
|---|
| 128 | ; calling existing method/property
|
|---|
| 129 | $object.example ;defined in __Au3Obj_GetIDsOfNames
|
|---|
| 130 | $object.example = "bla"
|
|---|
| 131 |
|
|---|
| 132 | ; Test:
|
|---|
| 133 | ; Exit without releasing the object
|
|---|
| 134 | ;~ OnAutoItExitUnregister("__ReleaseObject")
|
|---|
| 135 |
|
|---|
| 136 | Func __ReleaseObject()
|
|---|
| 137 | $object=0
|
|---|
| 138 | EndFunc
|
|---|
