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 #pragma once
20
21 #include <stdarg.h>
22 #include <stdint.h>
23
24 #define COBJMACROS
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "ole2.h"
29 #include "dispex.h"
30 #include "activscp.h"
31 #include "activdbg.h"
32
33 #ifdef __REACTOS__
34 #include <initguid.h>
35 #endif
36 #include "vbscript_classes.h"
37 #include "vbscript_defs.h"
38
39 #include "wine/heap.h"
40 #include "wine/list.h"
41
42 typedef struct {
43 void **blocks;
44 DWORD block_cnt;
45 DWORD last_block;
46 DWORD offset;
47 BOOL mark;
48 struct list custom_blocks;
49 } heap_pool_t;
50
51 void heap_pool_init(heap_pool_t*) DECLSPEC_HIDDEN;
52 void *heap_pool_alloc(heap_pool_t*,size_t) __WINE_ALLOC_SIZE(2) DECLSPEC_HIDDEN;
53 void *heap_pool_grow(heap_pool_t*,void*,DWORD,DWORD) DECLSPEC_HIDDEN;
54 void heap_pool_clear(heap_pool_t*) DECLSPEC_HIDDEN;
55 void heap_pool_free(heap_pool_t*) DECLSPEC_HIDDEN;
56 heap_pool_t *heap_pool_mark(heap_pool_t*) DECLSPEC_HIDDEN;
57
58 typedef struct _function_t function_t;
59 typedef struct _vbscode_t vbscode_t;
60 typedef struct _script_ctx_t script_ctx_t;
61 typedef struct _vbdisp_t vbdisp_t;
62
63 typedef struct named_item_t {
64 IDispatch *disp;
65 DWORD flags;
66 LPWSTR name;
67
68 struct list entry;
69 } named_item_t;
70
71 typedef enum {
72 VBDISP_CALLGET,
73 VBDISP_LET,
74 VBDISP_SET,
75 VBDISP_ANY
76 } vbdisp_invoke_type_t;
77
78 typedef struct {
79 unsigned dim_cnt;
80 SAFEARRAYBOUND *bounds;
81 } array_desc_t;
82
83 typedef struct {
84 BOOL is_public;
85 BOOL is_array;
86 const WCHAR *name;
87 } vbdisp_prop_desc_t;
88
89 typedef struct {
90 const WCHAR *name;
91 BOOL is_public;
92 BOOL is_array;
93 function_t *entries[VBDISP_ANY];
94 } vbdisp_funcprop_desc_t;
95
96 typedef struct _class_desc_t {
97 const WCHAR *name;
98 script_ctx_t *ctx;
99
100 unsigned class_initialize_id;
101 unsigned class_terminate_id;
102 unsigned func_cnt;
103 vbdisp_funcprop_desc_t *funcs;
104
105 unsigned prop_cnt;
106 vbdisp_prop_desc_t *props;
107
108 unsigned array_cnt;
109 array_desc_t *array_descs;
110
111 function_t *value_func;
112
113 struct _class_desc_t *next;
114 } class_desc_t;
115
116 struct _vbdisp_t {
117 IDispatchEx IDispatchEx_iface;
118
119 LONG ref;
120 BOOL terminator_ran;
121 struct list entry;
122
123 const class_desc_t *desc;
124 SAFEARRAY **arrays;
125 VARIANT props[1];
126 };
127
128 typedef struct _ident_map_t ident_map_t;
129
130 typedef struct {
131 IDispatchEx IDispatchEx_iface;
132 LONG ref;
133
134 ident_map_t *ident_map;
135 unsigned ident_map_cnt;
136 unsigned ident_map_size;
137
138 script_ctx_t *ctx;
139 } ScriptDisp;
140
141 typedef struct _builtin_prop_t builtin_prop_t;
142
143 typedef struct {
144 IDispatch IDispatch_iface;
145 LONG ref;
146 size_t member_cnt;
147 const builtin_prop_t *members;
148 script_ctx_t *ctx;
149 } BuiltinDisp;
150
151 HRESULT create_vbdisp(const class_desc_t*,vbdisp_t**) DECLSPEC_HIDDEN;
152 HRESULT disp_get_id(IDispatch*,BSTR,vbdisp_invoke_type_t,BOOL,DISPID*) DECLSPEC_HIDDEN;
153 HRESULT vbdisp_get_id(vbdisp_t*,BSTR,vbdisp_invoke_type_t,BOOL,DISPID*) DECLSPEC_HIDDEN;
154 HRESULT disp_call(script_ctx_t*,IDispatch*,DISPID,DISPPARAMS*,VARIANT*) DECLSPEC_HIDDEN;
155 HRESULT disp_propput(script_ctx_t*,IDispatch*,DISPID,WORD,DISPPARAMS*) DECLSPEC_HIDDEN;
156 HRESULT get_disp_value(script_ctx_t*,IDispatch*,VARIANT*) DECLSPEC_HIDDEN;
157 void collect_objects(script_ctx_t*) DECLSPEC_HIDDEN;
158 HRESULT create_script_disp(script_ctx_t*,ScriptDisp**) DECLSPEC_HIDDEN;
159
160 HRESULT to_int(VARIANT*,int*) DECLSPEC_HIDDEN;
161
arg_cnt(const DISPPARAMS * dp)162 static inline unsigned arg_cnt(const DISPPARAMS *dp)
163 {
164 return dp->cArgs - dp->cNamedArgs;
165 }
166
get_arg(DISPPARAMS * dp,DWORD i)167 static inline VARIANT *get_arg(DISPPARAMS *dp, DWORD i)
168 {
169 return dp->rgvarg + dp->cArgs-i-1;
170 }
171
172 typedef struct _dynamic_var_t {
173 struct _dynamic_var_t *next;
174 VARIANT v;
175 const WCHAR *name;
176 BOOL is_const;
177 } dynamic_var_t;
178
179 struct _script_ctx_t {
180 IActiveScriptSite *site;
181 LCID lcid;
182
183 IInternetHostSecurityManager *secmgr;
184 DWORD safeopt;
185
186 IDispatch *host_global;
187
188 ScriptDisp *script_obj;
189
190 BuiltinDisp *global_obj;
191 BuiltinDisp *err_obj;
192
193 EXCEPINFO ei;
194
195 dynamic_var_t *global_vars;
196 function_t *global_funcs;
197 class_desc_t *classes;
198 class_desc_t *procs;
199
200 heap_pool_t heap;
201
202 struct list objects;
203 struct list code_list;
204 struct list named_items;
205 };
206
207 HRESULT init_global(script_ctx_t*) DECLSPEC_HIDDEN;
208 HRESULT init_err(script_ctx_t*) DECLSPEC_HIDDEN;
209
210 IUnknown *create_ax_site(script_ctx_t*) DECLSPEC_HIDDEN;
211
212 typedef enum {
213 ARG_NONE = 0,
214 ARG_STR,
215 ARG_BSTR,
216 ARG_INT,
217 ARG_UINT,
218 ARG_ADDR,
219 ARG_DOUBLE
220 } instr_arg_type_t;
221
222 #define OP_LIST \
223 X(add, 1, 0, 0) \
224 X(and, 1, 0, 0) \
225 X(assign_ident, 1, ARG_BSTR, ARG_UINT) \
226 X(assign_member, 1, ARG_BSTR, ARG_UINT) \
227 X(bool, 1, ARG_INT, 0) \
228 X(catch, 1, ARG_ADDR, ARG_UINT) \
229 X(case, 0, ARG_ADDR, 0) \
230 X(concat, 1, 0, 0) \
231 X(const, 1, ARG_BSTR, 0) \
232 X(dim, 1, ARG_BSTR, ARG_UINT) \
233 X(div, 1, 0, 0) \
234 X(double, 1, ARG_DOUBLE, 0) \
235 X(empty, 1, 0, 0) \
236 X(enumnext, 0, ARG_ADDR, ARG_BSTR) \
237 X(equal, 1, 0, 0) \
238 X(hres, 1, ARG_UINT, 0) \
239 X(errmode, 1, ARG_INT, 0) \
240 X(eqv, 1, 0, 0) \
241 X(exp, 1, 0, 0) \
242 X(gt, 1, 0, 0) \
243 X(gteq, 1, 0, 0) \
244 X(icall, 1, ARG_BSTR, ARG_UINT) \
245 X(icallv, 1, ARG_BSTR, ARG_UINT) \
246 X(idiv, 1, 0, 0) \
247 X(imp, 1, 0, 0) \
248 X(incc, 1, ARG_BSTR, 0) \
249 X(int, 1, ARG_INT, 0) \
250 X(is, 1, 0, 0) \
251 X(jmp, 0, ARG_ADDR, 0) \
252 X(jmp_false, 0, ARG_ADDR, 0) \
253 X(jmp_true, 0, ARG_ADDR, 0) \
254 X(lt, 1, 0, 0) \
255 X(lteq, 1, 0, 0) \
256 X(mcall, 1, ARG_BSTR, ARG_UINT) \
257 X(mcallv, 1, ARG_BSTR, ARG_UINT) \
258 X(me, 1, 0, 0) \
259 X(mod, 1, 0, 0) \
260 X(mul, 1, 0, 0) \
261 X(neg, 1, 0, 0) \
262 X(nequal, 1, 0, 0) \
263 X(new, 1, ARG_STR, 0) \
264 X(newenum, 1, 0, 0) \
265 X(not, 1, 0, 0) \
266 X(nothing, 1, 0, 0) \
267 X(null, 1, 0, 0) \
268 X(or, 1, 0, 0) \
269 X(pop, 1, ARG_UINT, 0) \
270 X(ret, 0, 0, 0) \
271 X(retval, 1, 0, 0) \
272 X(set_ident, 1, ARG_BSTR, ARG_UINT) \
273 X(set_member, 1, ARG_BSTR, ARG_UINT) \
274 X(step, 0, ARG_ADDR, ARG_BSTR) \
275 X(stop, 1, 0, 0) \
276 X(string, 1, ARG_STR, 0) \
277 X(sub, 1, 0, 0) \
278 X(val, 1, 0, 0) \
279 X(xor, 1, 0, 0)
280
281 typedef enum {
282 #define X(x,n,a,b) OP_##x,
283 OP_LIST
284 #undef X
285 OP_LAST
286 } vbsop_t;
287
288 typedef union {
289 const WCHAR *str;
290 BSTR bstr;
291 unsigned uint;
292 LONG lng;
293 double *dbl;
294 } instr_arg_t;
295
296 typedef struct {
297 vbsop_t op;
298 instr_arg_t arg1;
299 instr_arg_t arg2;
300 } instr_t;
301
302 typedef struct {
303 const WCHAR *name;
304 BOOL by_ref;
305 } arg_desc_t;
306
307 typedef enum {
308 FUNC_GLOBAL,
309 FUNC_FUNCTION,
310 FUNC_SUB,
311 FUNC_PROPGET,
312 FUNC_PROPLET,
313 FUNC_PROPSET,
314 FUNC_DEFGET
315 } function_type_t;
316
317 typedef struct {
318 const WCHAR *name;
319 } var_desc_t;
320
321 struct _function_t {
322 function_type_t type;
323 const WCHAR *name;
324 BOOL is_public;
325 arg_desc_t *args;
326 unsigned arg_cnt;
327 var_desc_t *vars;
328 unsigned var_cnt;
329 array_desc_t *array_descs;
330 unsigned array_cnt;
331 unsigned code_off;
332 vbscode_t *code_ctx;
333 function_t *next;
334 };
335
336 struct _vbscode_t {
337 instr_t *instrs;
338 WCHAR *source;
339
340 BOOL option_explicit;
341
342 BOOL pending_exec;
343 function_t main_code;
344 IDispatch *context;
345
346 BSTR *bstr_pool;
347 unsigned bstr_pool_size;
348 unsigned bstr_cnt;
349 heap_pool_t heap;
350
351 struct list entry;
352 };
353
354 void release_vbscode(vbscode_t*) DECLSPEC_HIDDEN;
355 HRESULT compile_script(script_ctx_t*,const WCHAR*,const WCHAR*,DWORD,vbscode_t**) DECLSPEC_HIDDEN;
356 HRESULT compile_procedure(script_ctx_t*,const WCHAR*,const WCHAR*,DWORD,class_desc_t**) DECLSPEC_HIDDEN;
357 HRESULT exec_script(script_ctx_t*,BOOL,function_t*,vbdisp_t*,DISPPARAMS*,VARIANT*) DECLSPEC_HIDDEN;
358 void release_dynamic_vars(dynamic_var_t*) DECLSPEC_HIDDEN;
359 IDispatch *lookup_named_item(script_ctx_t*,const WCHAR*,unsigned) DECLSPEC_HIDDEN;
360 void clear_ei(EXCEPINFO*) DECLSPEC_HIDDEN;
361 HRESULT report_script_error(script_ctx_t*) DECLSPEC_HIDDEN;
362 void detach_global_objects(script_ctx_t*) DECLSPEC_HIDDEN;
363 HRESULT get_builtin_id(BuiltinDisp*,const WCHAR*,DISPID*) DECLSPEC_HIDDEN;
364
365 void release_regexp_typelib(void) DECLSPEC_HIDDEN;
366
is_int32(double d)367 static inline BOOL is_int32(double d)
368 {
369 return INT32_MIN <= d && d <= INT32_MAX && (double)(int)d == d;
370 }
371
372 HRESULT create_regexp(IDispatch**) DECLSPEC_HIDDEN;
373
374 HRESULT map_hres(HRESULT) DECLSPEC_HIDDEN;
375
376 HRESULT create_safearray_iter(SAFEARRAY *sa, IEnumVARIANT **ev) DECLSPEC_HIDDEN;
377
378 #define FACILITY_VBS 0xa
379 #define MAKE_VBSERROR(code) MAKE_HRESULT(SEVERITY_ERROR, FACILITY_VBS, code)
380
381 HRESULT WINAPI VBScriptFactory_CreateInstance(IClassFactory*,IUnknown*,REFIID,void**) DECLSPEC_HIDDEN;
382 HRESULT WINAPI VBScriptRegExpFactory_CreateInstance(IClassFactory*,IUnknown*,REFIID,void**) DECLSPEC_HIDDEN;
383
384 BSTR get_vbscript_string(int) DECLSPEC_HIDDEN;
385 BSTR get_vbscript_error_string(HRESULT) DECLSPEC_HIDDEN;
386
heap_strdupW(LPCWSTR str)387 static inline LPWSTR heap_strdupW(LPCWSTR str)
388 {
389 LPWSTR ret = NULL;
390
391 if(str) {
392 DWORD size;
393
394 size = (lstrlenW(str)+1)*sizeof(WCHAR);
395 ret = heap_alloc(size);
396 if(ret)
397 memcpy(ret, str, size);
398 }
399
400 return ret;
401 }
402
403 #define VBSCRIPT_BUILD_VERSION 16978
404 #define VBSCRIPT_MAJOR_VERSION 5
405 #define VBSCRIPT_MINOR_VERSION 8
406