dlls/jscript/jscript_main.c

   1 /*
   2  * Copyright 2008 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 "initguid.h"
  20
  21 #include "jscript.h"
  22
  23 #include "winreg.h"
  24 #include "advpub.h"
  25 #include "activaut.h"
  26 #include "objsafe.h"
  27 #include "mshtmhst.h"
  28 #include "rpcproxy.h"
  29 #include "jscript_classes.h"
  30
  31 #include "wine/debug.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
  34
  35 LONG module_ref = 0;
  36
  37 DEFINE_GUID(GUID_NULL,0,0,0,0,0,0,0,0,0,0,0);
  38
  39 HINSTANCE jscript_hinstance;
  40 static DWORD jscript_tls;
  41 static ITypeInfo *dispatch_typeinfo;
  42
  43 static int weak_refs_compare(const void *key, const struct rb_entry *entry)
  44 {
  45     const struct weak_refs_entry *weak_refs_entry = RB_ENTRY_VALUE(entry, const struct weak_refs_entry, entry);
  46     ULONG_PTR a = (ULONG_PTR)key, b = (ULONG_PTR)LIST_ENTRY(weak_refs_entry->list.next, struct weakmap_entry, weak_refs_entry)->key;
  47     return (a > b) - (a < b);
  48 }
  49
  50 struct thread_data *get_thread_data(void)
  51 {
  52     struct thread_data *thread_data = TlsGetValue(jscript_tls);
  53
  54     if(!thread_data) {
  55         thread_data = calloc(1, sizeof(struct thread_data));
  56         if(!thread_data)
  57             return NULL;
  58         thread_data->thread_id = GetCurrentThreadId();
  59         list_init(&thread_data->objects);
  60         rb_init(&thread_data->weak_refs, weak_refs_compare);
  61         TlsSetValue(jscript_tls, thread_data);
  62     }
  63
  64     thread_data->ref++;
  65     return thread_data;
  66 }
  67
  68 void release_thread_data(struct thread_data *thread_data)
  69 {
  70     if(--thread_data->ref)
  71         return;
  72
  73     free(thread_data);
  74     TlsSetValue(jscript_tls, NULL);
  75 }
  76
  77 HRESULT get_dispatch_typeinfo(ITypeInfo **out)
  78 {
  79     ITypeInfo *typeinfo;
  80     ITypeLib *typelib;
  81     HRESULT hr;
  82
  83     if (!dispatch_typeinfo)
  84     {
  85         hr = LoadRegTypeLib(&IID_StdOle, STDOLE_MAJORVERNUM, STDOLE_MINORVERNUM, STDOLE_LCID, &typelib);
  86         if (FAILED(hr)) return hr;
  87
  88         hr = ITypeLib_GetTypeInfoOfGuid(typelib, &IID_IDispatch, &typeinfo);
  89         ITypeLib_Release(typelib);
  90         if (FAILED(hr)) return hr;
  91
  92         if (InterlockedCompareExchangePointer((void**)&dispatch_typeinfo, typeinfo, NULL))
  93             ITypeInfo_Release(typeinfo);
  94     }
  95
  96     *out = dispatch_typeinfo;
  97     return S_OK;
  98 }
  99
 100 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
 101 {
 102     *ppv = NULL;
 103
 104     if(IsEqualGUID(&IID_IUnknown, riid)) {
 105         TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
 106         *ppv = iface;
 107     }else if(IsEqualGUID(&IID_IClassFactory, riid)) {
 108         TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv);
 109         *ppv = iface;
 110     }
 111
 112     if(*ppv) {
 113         IUnknown_AddRef((IUnknown*)*ppv);
 114         return S_OK;
 115     }
 116
 117     FIXME("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
 118     return E_NOINTERFACE;
 119 }
 120
 121 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
 122 {
 123     TRACE("(%p)\n", iface);
 124     return 2;
 125 }
 126
 127 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
 128 {
 129     TRACE("(%p)\n", iface);
 130     return 1;
 131 }
 132
 133 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL fLock)
 134 {
 135     TRACE("(%p)->(%x)\n", iface, fLock);
 136
 137     if(fLock)
 138         lock_module();
 139     else
 140         unlock_module();
 141
 142     return S_OK;
 143 }
 144
 145 static HRESULT WINAPI JScriptFactory_CreateInstance(IClassFactory *iface, IUnknown *outer,
 146         REFIID riid, void **ppv)
 147 {
 148     TRACE("(%p %s %p)\n", outer, debugstr_guid(riid), ppv);
 149
 150     if(outer) {
 151         *ppv = NULL;
 152         return CLASS_E_NOAGGREGATION;
 153     }
 154
 155     return create_jscript_object(FALSE, riid, ppv);
 156 }
 157
 158 static const IClassFactoryVtbl JScriptFactoryVtbl = {
 159     ClassFactory_QueryInterface,
 160     ClassFactory_AddRef,
 161     ClassFactory_Release,
 162     JScriptFactory_CreateInstance,
 163     ClassFactory_LockServer
 164 };
 165
 166 static IClassFactory JScriptFactory = { &JScriptFactoryVtbl };
 167
 168 static HRESULT WINAPI JScriptEncodeFactory_CreateInstance(IClassFactory *iface, IUnknown *outer,
 169         REFIID riid, void **ppv)
 170 {
 171     TRACE("(%p %s %p)\n", outer, debugstr_guid(riid), ppv);
 172
 173     if(outer) {
 174         *ppv = NULL;
 175         return CLASS_E_NOAGGREGATION;
 176     }
 177
 178     return create_jscript_object(TRUE, riid, ppv);
 179 }
 180
 181 static const IClassFactoryVtbl JScriptEncodeFactoryVtbl = {
 182     ClassFactory_QueryInterface,
 183     ClassFactory_AddRef,
 184     ClassFactory_Release,
 185     JScriptEncodeFactory_CreateInstance,
 186     ClassFactory_LockServer
 187 };
 188
 189 static IClassFactory JScriptEncodeFactory = { &JScriptEncodeFactoryVtbl };
 190
 191 /******************************************************************
 192  *              DllMain (jscript.@)
 193  */
 194 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpv)
 195 {
 196     TRACE("(%p %ld %p)\n", hInstDLL, fdwReason, lpv);
 197
 198     switch(fdwReason) {
 199     case DLL_PROCESS_ATTACH:
 200         DisableThreadLibraryCalls(hInstDLL);
 201         jscript_hinstance = hInstDLL;
 202         jscript_tls = TlsAlloc();
 203         if(jscript_tls == TLS_OUT_OF_INDEXES || !init_strings())
 204             return FALSE;
 205         break;
 206     case DLL_PROCESS_DETACH:
 207         if (lpv) break;
 208         if (dispatch_typeinfo) ITypeInfo_Release(dispatch_typeinfo);
 209         if(jscript_tls != TLS_OUT_OF_INDEXES) TlsFree(jscript_tls);
 210         free_strings();
 211         break;
 212     }
 213
 214     return TRUE;
 215 }
 216
 217 /***********************************************************************
 218  *              DllGetClassObject       (jscript.@)
 219  */
 220 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
 221 {
 222     if(IsEqualGUID(&CLSID_JScript, rclsid)) {
 223         TRACE("(CLSID_JScript %s %p)\n", debugstr_guid(riid), ppv);
 224         return IClassFactory_QueryInterface(&JScriptFactory, riid, ppv);
 225     }
 226
 227     if(IsEqualGUID(&CLSID_JScriptEncode, rclsid)) {
 228         TRACE("(CLSID_JScriptEncode %s %p)\n", debugstr_guid(riid), ppv);
 229         return IClassFactory_QueryInterface(&JScriptEncodeFactory, riid, ppv);
 230     }
 231
 232     FIXME("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
 233     return CLASS_E_CLASSNOTAVAILABLE;
 234 }
 235
 236 /***********************************************************************
 237  *          DllCanUnloadNow (jscript.@)
 238  */
 239 HRESULT WINAPI DllCanUnloadNow(void)
 240 {
 241     TRACE("() ref=%ld\n", module_ref);
 242
 243     return module_ref ? S_FALSE : S_OK;
 244 }