vbscript: Added interp_add implementation.
[wine] / dlls / vbscript / vbscript.h
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 <stdarg.h>
20
21 #define COBJMACROS
22
23 #include "windef.h"
24 #include "winbase.h"
25 #include "ole2.h"
26 #include "dispex.h"
27 #include "activscp.h"
28
29 #include "vbscript_classes.h"
30
31 #include "wine/list.h"
32 #include "wine/unicode.h"
33
34 typedef struct {
35     void **blocks;
36     DWORD block_cnt;
37     DWORD last_block;
38     DWORD offset;
39     struct list custom_blocks;
40 } vbsheap_t;
41
42 void vbsheap_init(vbsheap_t*) DECLSPEC_HIDDEN;
43 void *vbsheap_alloc(vbsheap_t*,size_t) __WINE_ALLOC_SIZE(2) DECLSPEC_HIDDEN;
44 void vbsheap_free(vbsheap_t*) DECLSPEC_HIDDEN;
45
46 typedef struct _function_t function_t;
47 typedef struct _vbscode_t vbscode_t;
48 typedef struct _script_ctx_t script_ctx_t;
49
50 typedef struct named_item_t {
51     IDispatch *disp;
52     DWORD flags;
53     LPWSTR name;
54
55     struct list entry;
56 } named_item_t;
57
58 typedef struct {
59     IDispatchEx IDispatchEx_iface;
60
61     LONG ref;
62 } vbdisp_t;
63
64 HRESULT disp_get_id(IDispatch*,BSTR,DISPID*);
65 HRESULT disp_call(script_ctx_t*,IDispatch*,DISPID,DISPPARAMS*,VARIANT*);
66
67 struct _script_ctx_t {
68     IActiveScriptSite *site;
69     LCID lcid;
70
71     IDispatch *host_global;
72
73     vbdisp_t *script_obj;
74
75     struct list code_list;
76     struct list named_items;
77 };
78
79 HRESULT init_global(script_ctx_t*);
80
81 typedef enum {
82     ARG_NONE = 0,
83     ARG_STR,
84     ARG_BSTR,
85     ARG_INT,
86     ARG_UINT,
87     ARG_DOUBLE
88 } instr_arg_type_t;
89
90 #define OP_LIST                                   \
91     X(add,            1, 0,           0)          \
92     X(bool,           1, ARG_INT,     0)          \
93     X(concat,         1, 0,           0)          \
94     X(double,         1, ARG_DOUBLE,  0)          \
95     X(empty,          1, 0,           0)          \
96     X(equal,          1, 0,           0)          \
97     X(icall,          1, ARG_BSTR,    ARG_UINT)   \
98     X(icallv,         1, ARG_BSTR,    ARG_UINT)   \
99     X(long,           1, ARG_INT,     0)          \
100     X(neg,            1, 0,           0)          \
101     X(not,            1, 0,           0)          \
102     X(null,           1, 0,           0)          \
103     X(ret,            0, 0,           0)          \
104     X(short,          1, ARG_INT,     0)          \
105     X(string,         1, ARG_STR,     0)          \
106     X(sub,            1, 0,           0)
107
108 typedef enum {
109 #define X(x,n,a,b) OP_##x,
110 OP_LIST
111 #undef X
112     OP_LAST
113 } vbsop_t;
114
115 typedef union {
116     const WCHAR *str;
117     BSTR bstr;
118     unsigned uint;
119     LONG lng;
120     double *dbl;
121 } instr_arg_t;
122
123 typedef struct {
124     vbsop_t op;
125     instr_arg_t arg1;
126     instr_arg_t arg2;
127 } instr_t;
128
129 struct _function_t {
130     unsigned code_off;
131     vbscode_t *code_ctx;
132 };
133
134 struct _vbscode_t {
135     instr_t *instrs;
136     WCHAR *source;
137
138     BOOL option_explicit;
139
140     BOOL global_executed;
141     function_t global_code;
142
143     BSTR *bstr_pool;
144     unsigned bstr_pool_size;
145     unsigned bstr_cnt;
146     vbsheap_t heap;
147
148     struct list entry;
149 };
150
151 void release_vbscode(vbscode_t*) DECLSPEC_HIDDEN;
152 HRESULT compile_script(script_ctx_t*,const WCHAR*,vbscode_t**) DECLSPEC_HIDDEN;
153 HRESULT exec_script(script_ctx_t*,function_t*) DECLSPEC_HIDDEN;
154
155 HRESULT WINAPI VBScriptFactory_CreateInstance(IClassFactory*,IUnknown*,REFIID,void**);
156
157 static inline void *heap_alloc(size_t len)
158 {
159     return HeapAlloc(GetProcessHeap(), 0, len);
160 }
161
162 static inline void *heap_alloc_zero(size_t len)
163 {
164     return HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len);
165 }
166
167 static inline void *heap_realloc(void *mem, size_t len)
168 {
169     return HeapReAlloc(GetProcessHeap(), 0, mem, len);
170 }
171
172 static inline BOOL heap_free(void *mem)
173 {
174     return HeapFree(GetProcessHeap(), 0, mem);
175 }
176
177 static inline LPWSTR heap_strdupW(LPCWSTR str)
178 {
179     LPWSTR ret = NULL;
180
181     if(str) {
182         DWORD size;
183
184         size = (strlenW(str)+1)*sizeof(WCHAR);
185         ret = heap_alloc(size);
186         if(ret)
187             memcpy(ret, str, size);
188     }
189
190     return ret;
191 }