git.oblomov.eu Git - wine/blob - dlls/vbscript/vbscript.c

   1 /*
   2  * Copyright 2011 Jacek Caban for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include "config.h"
  20
  21 #include <stdarg.h>
  22
  23 #define COBJMACROS
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "ole2.h"
  28
  29 #include "vbscript.h"
  30
  31 #include "wine/debug.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(vbscript);
  34
  35 #ifdef _WIN64
  36
  37 #define CTXARG_T DWORDLONG
  38 #define IActiveScriptParseVtbl IActiveScriptParse64Vtbl
  39
  40 #else
  41
  42 #define CTXARG_T DWORD
  43 #define IActiveScriptParseVtbl IActiveScriptParse32Vtbl
  44
  45 #endif
  46
  47 struct VBScript {
  48     IActiveScript IActiveScript_iface;
  49     IActiveScriptParse IActiveScriptParse_iface;
  50
  51     LONG ref;
  52
  53     SCRIPTSTATE state;
  54     IActiveScriptSite *site;
  55     script_ctx_t *ctx;
  56     LONG thread_id;
  57     LCID lcid;
  58 };
  59
  60 static void change_state(VBScript *This, SCRIPTSTATE state)
  61 {
  62     if(This->state == state)
  63         return;
  64
  65     This->state = state;
  66     if(This->site)
  67         IActiveScriptSite_OnStateChange(This->site, state);
  68 }
  69
  70 static HRESULT set_ctx_site(VBScript *This)
  71 {
  72     This->ctx->lcid = This->lcid;
  73
  74     IActiveScriptSite_AddRef(This->site);
  75     This->ctx->site = This->site;
  76
  77     change_state(This, SCRIPTSTATE_INITIALIZED);
  78     return S_OK;
  79 }
  80
  81 static void destroy_script(script_ctx_t *ctx)
  82 {
  83     if(ctx->site)
  84         IActiveScriptSite_Release(ctx->site);
  85     heap_free(ctx);
  86 }
  87
  88 static inline VBScript *impl_from_IActiveScript(IActiveScript *iface)
  89 {
  90     return CONTAINING_RECORD(iface, VBScript, IActiveScript_iface);
  91 }
  92
  93 static HRESULT WINAPI VBScript_QueryInterface(IActiveScript *iface, REFIID riid, void **ppv)
  94 {
  95     VBScript *This = impl_from_IActiveScript(iface);
  96
  97     if(IsEqualGUID(riid, &IID_IUnknown)) {
  98         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
  99         *ppv = &This->IActiveScript_iface;
 100     }else if(IsEqualGUID(riid, &IID_IActiveScript)) {
 101         TRACE("(%p)->(IID_IActiveScript %p)\n", This, ppv);
 102         *ppv = &This->IActiveScript_iface;
 103     }else if(IsEqualGUID(riid, &IID_IActiveScriptParse)) {
 104         TRACE("(%p)->(IID_IActiveScriptParse %p)\n", This, ppv);
 105         *ppv = &This->IActiveScriptParse_iface;
 106     }else {
 107         FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
 108         *ppv = NULL;
 109         return E_NOINTERFACE;
 110     }
 111
 112     IUnknown_AddRef((IUnknown*)*ppv);
 113     return S_OK;
 114 }
 115
 116 static ULONG WINAPI VBScript_AddRef(IActiveScript *iface)
 117 {
 118     VBScript *This = impl_from_IActiveScript(iface);
 119     LONG ref = InterlockedIncrement(&This->ref);
 120
 121     TRACE("(%p) ref=%d\n", This, ref);
 122
 123     return ref;
 124 }
 125
 126 static ULONG WINAPI VBScript_Release(IActiveScript *iface)
 127 {
 128     VBScript *This = impl_from_IActiveScript(iface);
 129     LONG ref = InterlockedDecrement(&This->ref);
 130
 131     TRACE("(%p) ref=%d\n", iface, ref);
 132
 133     if(!ref) {
 134         if(This->site)
 135             IActiveScriptSite_Release(This->site);
 136         heap_free(This);
 137     }
 138
 139     return ref;
 140 }
 141
 142 static HRESULT WINAPI VBScript_SetScriptSite(IActiveScript *iface, IActiveScriptSite *pass)
 143 {
 144     VBScript *This = impl_from_IActiveScript(iface);
 145     LCID lcid;
 146     HRESULT hres;
 147
 148     TRACE("(%p)->(%p)\n", This, pass);
 149
 150     if(!pass)
 151         return E_POINTER;
 152
 153     if(This->site)
 154         return E_UNEXPECTED;
 155
 156     if(InterlockedCompareExchange(&This->thread_id, GetCurrentThreadId(), 0))
 157         return E_UNEXPECTED;
 158
 159     This->site = pass;
 160     IActiveScriptSite_AddRef(This->site);
 161
 162     hres = IActiveScriptSite_GetLCID(This->site, &lcid);
 163     if(hres == S_OK)
 164         This->lcid = lcid;
 165
 166     return This->ctx ? set_ctx_site(This) : S_OK;
 167 }
 168
 169 static HRESULT WINAPI VBScript_GetScriptSite(IActiveScript *iface, REFIID riid,
 170                                             void **ppvObject)
 171 {
 172     VBScript *This = impl_from_IActiveScript(iface);
 173     FIXME("(%p)->()\n", This);
 174     return E_NOTIMPL;
 175 }
 176
 177 static HRESULT WINAPI VBScript_SetScriptState(IActiveScript *iface, SCRIPTSTATE ss)
 178 {
 179     VBScript *This = impl_from_IActiveScript(iface);
 180     FIXME("(%p)->(%d)\n", This, ss);
 181     return S_OK;
 182 }
 183
 184 static HRESULT WINAPI VBScript_GetScriptState(IActiveScript *iface, SCRIPTSTATE *pssState)
 185 {
 186     VBScript *This = impl_from_IActiveScript(iface);
 187     FIXME("(%p)->(%p)\n", This, pssState);
 188     return E_NOTIMPL;
 189 }
 190
 191 static HRESULT WINAPI VBScript_Close(IActiveScript *iface)
 192 {
 193     VBScript *This = impl_from_IActiveScript(iface);
 194     FIXME("(%p)->()\n", This);
 195     return E_NOTIMPL;
 196 }
 197
 198 static HRESULT WINAPI VBScript_AddNamedItem(IActiveScript *iface, LPCOLESTR pstrName, DWORD dwFlags)
 199 {
 200     VBScript *This = impl_from_IActiveScript(iface);
 201     FIXME("(%p)->(%s %x)\n", This, debugstr_w(pstrName), dwFlags);
 202     return S_OK;
 203 }
 204
 205 static HRESULT WINAPI VBScript_AddTypeLib(IActiveScript *iface, REFGUID rguidTypeLib,
 206         DWORD dwMajor, DWORD dwMinor, DWORD dwFlags)
 207 {
 208     VBScript *This = impl_from_IActiveScript(iface);
 209     FIXME("(%p)->()\n", This);
 210     return E_NOTIMPL;
 211 }
 212
 213 static HRESULT WINAPI VBScript_GetScriptDispatch(IActiveScript *iface, LPCOLESTR pstrItemName, IDispatch **ppdisp)
 214 {
 215     VBScript *This = impl_from_IActiveScript(iface);
 216     FIXME("(%p)->(%p)\n", This, ppdisp);
 217     return E_NOTIMPL;
 218 }
 219
 220 static HRESULT WINAPI VBScript_GetCurrentScriptThreadID(IActiveScript *iface,
 221                                                        SCRIPTTHREADID *pstridThread)
 222 {
 223     VBScript *This = impl_from_IActiveScript(iface);
 224     FIXME("(%p)->()\n", This);
 225     return E_NOTIMPL;
 226 }
 227
 228 static HRESULT WINAPI VBScript_GetScriptThreadID(IActiveScript *iface,
 229                                                 DWORD dwWin32ThreadId, SCRIPTTHREADID *pstidThread)
 230 {
 231     VBScript *This = impl_from_IActiveScript(iface);
 232     FIXME("(%p)->()\n", This);
 233     return E_NOTIMPL;
 234 }
 235
 236 static HRESULT WINAPI VBScript_GetScriptThreadState(IActiveScript *iface,
 237         SCRIPTTHREADID stidThread, SCRIPTTHREADSTATE *pstsState)
 238 {
 239     VBScript *This = impl_from_IActiveScript(iface);
 240     FIXME("(%p)->()\n", This);
 241     return E_NOTIMPL;
 242 }
 243
 244 static HRESULT WINAPI VBScript_InterruptScriptThread(IActiveScript *iface,
 245         SCRIPTTHREADID stidThread, const EXCEPINFO *pexcepinfo, DWORD dwFlags)
 246 {
 247     VBScript *This = impl_from_IActiveScript(iface);
 248     FIXME("(%p)->()\n", This);
 249     return E_NOTIMPL;
 250 }
 251
 252 static HRESULT WINAPI VBScript_Clone(IActiveScript *iface, IActiveScript **ppscript)
 253 {
 254     VBScript *This = impl_from_IActiveScript(iface);
 255     FIXME("(%p)->()\n", This);
 256     return E_NOTIMPL;
 257 }
 258
 259 static const IActiveScriptVtbl VBScriptVtbl = {
 260     VBScript_QueryInterface,
 261     VBScript_AddRef,
 262     VBScript_Release,
 263     VBScript_SetScriptSite,
 264     VBScript_GetScriptSite,
 265     VBScript_SetScriptState,
 266     VBScript_GetScriptState,
 267     VBScript_Close,
 268     VBScript_AddNamedItem,
 269     VBScript_AddTypeLib,
 270     VBScript_GetScriptDispatch,
 271     VBScript_GetCurrentScriptThreadID,
 272     VBScript_GetScriptThreadID,
 273     VBScript_GetScriptThreadState,
 274     VBScript_InterruptScriptThread,
 275     VBScript_Clone
 276 };
 277
 278 static inline VBScript *impl_from_IActiveScriptParse(IActiveScriptParse *iface)
 279 {
 280     return CONTAINING_RECORD(iface, VBScript, IActiveScriptParse_iface);
 281 }
 282
 283 static HRESULT WINAPI VBScriptParse_QueryInterface(IActiveScriptParse *iface, REFIID riid, void **ppv)
 284 {
 285     VBScript *This = impl_from_IActiveScriptParse(iface);
 286     return IActiveScript_QueryInterface(&This->IActiveScript_iface, riid, ppv);
 287 }
 288
 289 static ULONG WINAPI VBScriptParse_AddRef(IActiveScriptParse *iface)
 290 {
 291     VBScript *This = impl_from_IActiveScriptParse(iface);
 292     return IActiveScript_AddRef(&This->IActiveScript_iface);
 293 }
 294
 295 static ULONG WINAPI VBScriptParse_Release(IActiveScriptParse *iface)
 296 {
 297     VBScript *This = impl_from_IActiveScriptParse(iface);
 298     return IActiveScript_Release(&This->IActiveScript_iface);
 299 }
 300
 301 static HRESULT WINAPI VBScriptParse_InitNew(IActiveScriptParse *iface)
 302 {
 303     VBScript *This = impl_from_IActiveScriptParse(iface);
 304     script_ctx_t *ctx;
 305
 306     TRACE("(%p)\n", This);
 307
 308     if(This->ctx)
 309         return E_UNEXPECTED;
 310
 311     ctx = heap_alloc_zero(sizeof(script_ctx_t));
 312     if(!ctx)
 313         return E_OUTOFMEMORY;
 314
 315     ctx = InterlockedCompareExchangePointer((void**)&This->ctx, ctx, NULL);
 316     if(ctx) {
 317         destroy_script(ctx);
 318         return E_UNEXPECTED;
 319     }
 320
 321     return This->site ? set_ctx_site(This) : S_OK;
 322 }
 323
 324 static HRESULT WINAPI VBScriptParse_AddScriptlet(IActiveScriptParse *iface,
 325         LPCOLESTR pstrDefaultName, LPCOLESTR pstrCode, LPCOLESTR pstrItemName,
 326         LPCOLESTR pstrSubItemName, LPCOLESTR pstrEventName, LPCOLESTR pstrDelimiter,
 327         CTXARG_T dwSourceContextCookie, ULONG ulStartingLineNumber, DWORD dwFlags,
 328         BSTR *pbstrName, EXCEPINFO *pexcepinfo)
 329 {
 330     VBScript *This = impl_from_IActiveScriptParse(iface);
 331     FIXME("(%p)->(%s %s %s %s %s %s %s %u %x %p %p)\n", This, debugstr_w(pstrDefaultName),
 332           debugstr_w(pstrCode), debugstr_w(pstrItemName), debugstr_w(pstrSubItemName),
 333           debugstr_w(pstrEventName), debugstr_w(pstrDelimiter), wine_dbgstr_longlong(dwSourceContextCookie),
 334           ulStartingLineNumber, dwFlags, pbstrName, pexcepinfo);
 335     return E_NOTIMPL;
 336 }
 337
 338 static HRESULT WINAPI VBScriptParse_ParseScriptText(IActiveScriptParse *iface,
 339         LPCOLESTR pstrCode, LPCOLESTR pstrItemName, IUnknown *punkContext,
 340         LPCOLESTR pstrDelimiter, CTXARG_T dwSourceContextCookie, ULONG ulStartingLine,
 341         DWORD dwFlags, VARIANT *pvarResult, EXCEPINFO *pexcepinfo)
 342 {
 343     VBScript *This = impl_from_IActiveScriptParse(iface);
 344     FIXME("(%p)->(%s %s %p %s %s %u %x %p %p)\n", This, debugstr_w(pstrCode),
 345           debugstr_w(pstrItemName), punkContext, debugstr_w(pstrDelimiter),
 346           wine_dbgstr_longlong(dwSourceContextCookie), ulStartingLine, dwFlags, pvarResult, pexcepinfo);
 347     return E_NOTIMPL;
 348 }
 349
 350 static const IActiveScriptParseVtbl VBScriptParseVtbl = {
 351     VBScriptParse_QueryInterface,
 352     VBScriptParse_AddRef,
 353     VBScriptParse_Release,
 354     VBScriptParse_InitNew,
 355     VBScriptParse_AddScriptlet,
 356     VBScriptParse_ParseScriptText
 357 };
 358
 359 HRESULT WINAPI VBScriptFactory_CreateInstance(IClassFactory *iface, IUnknown *pUnkOuter, REFIID riid, void **ppv)
 360 {
 361     VBScript *ret;
 362     HRESULT hres;
 363
 364     TRACE("(%p %s %p)\n", pUnkOuter, debugstr_guid(riid), ppv);
 365
 366     ret = heap_alloc_zero(sizeof(*ret));
 367     if(!ret)
 368         return E_OUTOFMEMORY;
 369
 370     ret->IActiveScript_iface.lpVtbl = &VBScriptVtbl;
 371     ret->IActiveScriptParse_iface.lpVtbl = &VBScriptParseVtbl;
 372
 373     ret->ref = 1;
 374     ret->state = SCRIPTSTATE_UNINITIALIZED;
 375
 376     hres = IActiveScript_QueryInterface(&ret->IActiveScript_iface, riid, ppv);
 377     IActiveScript_Release(&ret->IActiveScript_iface);
 378     return hres;
 379 }