Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309/* * Copyright 2011 Jacek Caban for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */
#include "initguid.h"
#include "vbscript.h"#include "objsafe.h"#include "mshtmhst.h"#include "rpcproxy.h"#include "vbscript_classes.h"#include "vbsglobal.h"#include "vbsregexp55.h"
#include "wine/debug.h"
WINE_DEFAULT_DEBUG_CHANNEL(vbscript);WINE_DECLARE_DEBUG_CHANNEL(heap);
DEFINE_GUID(GUID_NULL,0,0,0,0,0,0,0,0,0,0,0);
static HINSTANCE vbscript_hinstance;
BSTR get_vbscript_string(int id){ WCHAR buf[512]; if(!LoadStringW(vbscript_hinstance, id, buf, ARRAY_SIZE(buf))) return NULL; return SysAllocString(buf);}
BSTR get_vbscript_error_string(HRESULT error){ BSTR ret; if(HRESULT_FACILITY(error) != FACILITY_VBS || !(ret = get_vbscript_string(HRESULT_CODE(error)))) ret = get_vbscript_string(VBS_UNKNOWN_RUNTIME_ERROR); return ret;}
#define MIN_BLOCK_SIZE 128#define ARENA_FREE_FILLER 0xaa
static inline DWORD block_size(DWORD block){ return MIN_BLOCK_SIZE << block;}
void heap_pool_init(heap_pool_t *heap){ memset(heap, 0, sizeof(*heap)); list_init(&heap->custom_blocks);}
void *heap_pool_alloc(heap_pool_t *heap, size_t size){ struct list *list; void *tmp;
size = (size+3)&~3;
if(!heap->block_cnt) { if(!heap->blocks) { heap->blocks = heap_alloc(sizeof(void*)); if(!heap->blocks) return NULL; }
tmp = heap_alloc(block_size(0)); if(!tmp) return NULL;
heap->blocks[0] = tmp; heap->block_cnt = 1; }
if(heap->offset + size <= block_size(heap->last_block)) { tmp = ((BYTE*)heap->blocks[heap->last_block])+heap->offset; heap->offset += size; return tmp; }
if(size <= block_size(heap->last_block+1)) { if(heap->last_block+1 == heap->block_cnt) { tmp = heap_realloc(heap->blocks, (heap->block_cnt+1)*sizeof(void*)); if(!tmp) return NULL;
heap->blocks = tmp; heap->blocks[heap->block_cnt] = heap_alloc(block_size(heap->block_cnt)); if(!heap->blocks[heap->block_cnt]) return NULL;
heap->block_cnt++; }
heap->last_block++; heap->offset = size; return heap->blocks[heap->last_block]; }
list = heap_alloc(size + sizeof(struct list)); if(!list) return NULL;
list_add_head(&heap->custom_blocks, list); return list+1;}
void *heap_pool_grow(heap_pool_t *heap, void *mem, DWORD size, DWORD inc){ void *ret;
if(mem == (BYTE*)heap->blocks[heap->last_block] + heap->offset-size && heap->offset+inc < block_size(heap->last_block)) { heap->offset += inc; return mem; }
ret = heap_pool_alloc(heap, size+inc); if(ret) /* FIXME: avoid copying for custom blocks */ memcpy(ret, mem, size); return ret;}
void heap_pool_clear(heap_pool_t *heap){ struct list *tmp;
if(!heap) return;
while((tmp = list_next(&heap->custom_blocks, &heap->custom_blocks))) { list_remove(tmp); heap_free(tmp); }
if(WARN_ON(heap)) { DWORD i;
for(i=0; i < heap->block_cnt; i++) memset(heap->blocks[i], ARENA_FREE_FILLER, block_size(i)); }
heap->last_block = heap->offset = 0; heap->mark = FALSE;}
void heap_pool_free(heap_pool_t *heap){ DWORD i;
heap_pool_clear(heap);
for(i=0; i < heap->block_cnt; i++) heap_free(heap->blocks[i]); heap_free(heap->blocks);
heap_pool_init(heap);}
heap_pool_t *heap_pool_mark(heap_pool_t *heap){ if(heap->mark) return NULL;
heap->mark = TRUE; return heap;}
static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv){ *ppv = NULL;
if(IsEqualGUID(&IID_IUnknown, riid)) { TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv); *ppv = iface; }else if(IsEqualGUID(&IID_IClassFactory, riid)) { TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv); *ppv = iface; }
if(*ppv) { IUnknown_AddRef((IUnknown*)*ppv); return S_OK; }
FIXME("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv); return E_NOINTERFACE;}
static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface){ TRACE("(%p)\n", iface); return 2;}
static ULONG WINAPI ClassFactory_Release(IClassFactory *iface){ TRACE("(%p)\n", iface); return 1;}
static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL fLock){ TRACE("(%p)->(%x)\n", iface, fLock); return S_OK;}
static const IClassFactoryVtbl VBScriptFactoryVtbl = { ClassFactory_QueryInterface, ClassFactory_AddRef, ClassFactory_Release, VBScriptFactory_CreateInstance, ClassFactory_LockServer};
static IClassFactory VBScriptFactory = { &VBScriptFactoryVtbl };
static const IClassFactoryVtbl VBScriptRegExpFactoryVtbl = { ClassFactory_QueryInterface, ClassFactory_AddRef, ClassFactory_Release, VBScriptRegExpFactory_CreateInstance, ClassFactory_LockServer};
static IClassFactory VBScriptRegExpFactory = { &VBScriptRegExpFactoryVtbl };
/****************************************************************** * DllMain (vbscript.@) */BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpv){ TRACE("(%p %d %p)\n", hInstDLL, fdwReason, lpv);
switch(fdwReason) {#ifndef __REACTOS__ case DLL_WINE_PREATTACH: return FALSE; /* prefer native version */#endif case DLL_PROCESS_ATTACH: DisableThreadLibraryCalls(hInstDLL); vbscript_hinstance = hInstDLL; break; case DLL_PROCESS_DETACH: if (lpv) break; release_regexp_typelib(); }
return TRUE;}
/*********************************************************************** * DllGetClassObject (vbscript.@) */HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv){ if(IsEqualGUID(&CLSID_VBScript, rclsid)) { TRACE("(CLSID_VBScript %s %p)\n", debugstr_guid(riid), ppv); return IClassFactory_QueryInterface(&VBScriptFactory, riid, ppv); }else if(IsEqualGUID(&CLSID_VBScriptRegExp, rclsid)) { TRACE("(CLSID_VBScriptRegExp %s %p)\n", debugstr_guid(riid), ppv); return IClassFactory_QueryInterface(&VBScriptRegExpFactory, riid, ppv); }
FIXME("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv); return CLASS_E_CLASSNOTAVAILABLE;}
/*********************************************************************** * DllCanUnloadNow (vbscript.@) */HRESULT WINAPI DllCanUnloadNow(void){ return S_FALSE;}
/*********************************************************************** * DllRegisterServer (vbscript.@) */HRESULT WINAPI DllRegisterServer(void){ TRACE("()\n"); return __wine_register_resources(vbscript_hinstance);}
/*********************************************************************** * DllUnregisterServer (vbscript.@) */HRESULT WINAPI DllUnregisterServer(void){ TRACE("()\n"); return __wine_unregister_resources(vbscript_hinstance);}