dlls/vbscript/compile.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 <assert.h>
  20
  21 #include "vbscript.h"
  22 #include "parse.h"
  23 #include "parser.tab.h"
  24
  25 #include "wine/debug.h"
  26
  27 WINE_DEFAULT_DEBUG_CHANNEL(vbscript);
  28
  29 typedef struct {
  30     parser_ctx_t parser;
  31
  32     unsigned instr_cnt;
  33     unsigned instr_size;
  34     vbscode_t *code;
  35 } compile_ctx_t;
  36
  37 static HRESULT compile_expression(compile_ctx_t*,expression_t*);
  38
  39 static inline void *compiler_alloc(vbscode_t *vbscode, size_t size)
  40 {
  41     return vbsheap_alloc(&vbscode->heap, size);
  42 }
  43
  44 static WCHAR *compiler_alloc_string(vbscode_t *vbscode, const WCHAR *str)
  45 {
  46     size_t size;
  47     WCHAR *ret;
  48
  49     size = (strlenW(str)+1)*sizeof(WCHAR);
  50     ret = compiler_alloc(vbscode, size);
  51     if(ret)
  52         memcpy(ret, str, size);
  53     return ret;
  54 }
  55
  56 static inline instr_t *instr_ptr(compile_ctx_t *ctx, unsigned id)
  57 {
  58     assert(id < ctx->instr_cnt);
  59     return ctx->code->instrs + id;
  60 }
  61
  62 static unsigned push_instr(compile_ctx_t *ctx, vbsop_t op)
  63 {
  64     assert(ctx->instr_size && ctx->instr_size >= ctx->instr_cnt);
  65
  66     if(ctx->instr_size == ctx->instr_cnt) {
  67         instr_t *new_instr;
  68
  69         new_instr = heap_realloc(ctx->code->instrs, ctx->instr_size*2*sizeof(instr_t));
  70         if(!new_instr)
  71             return -1;
  72
  73         ctx->code->instrs = new_instr;
  74         ctx->instr_size *= 2;
  75     }
  76
  77     ctx->code->instrs[ctx->instr_cnt].op = op;
  78     return ctx->instr_cnt++;
  79 }
  80
  81 static HRESULT push_instr_int(compile_ctx_t *ctx, vbsop_t op, LONG arg)
  82 {
  83     unsigned ret;
  84
  85     ret = push_instr(ctx, op);
  86     if(ret == -1)
  87         return E_OUTOFMEMORY;
  88
  89     instr_ptr(ctx, ret)->arg1.lng = arg;
  90     return S_OK;
  91 }
  92
  93 static HRESULT push_instr_str(compile_ctx_t *ctx, vbsop_t op, const WCHAR *arg)
  94 {
  95     unsigned instr;
  96     WCHAR *str;
  97
  98     str = compiler_alloc_string(ctx->code, arg);
  99     if(!str)
 100         return E_OUTOFMEMORY;
 101
 102     instr = push_instr(ctx, op);
 103     if(instr == -1)
 104         return E_OUTOFMEMORY;
 105
 106     instr_ptr(ctx, instr)->arg1.str = str;
 107     return S_OK;
 108 }
 109
 110 static HRESULT push_instr_double(compile_ctx_t *ctx, vbsop_t op, double arg)
 111 {
 112     unsigned instr;
 113     double *d;
 114
 115     d = compiler_alloc(ctx->code, sizeof(double));
 116     if(!d)
 117         return E_OUTOFMEMORY;
 118
 119     instr = push_instr(ctx, op);
 120     if(instr == -1)
 121         return E_OUTOFMEMORY;
 122
 123     *d = arg;
 124     instr_ptr(ctx, instr)->arg1.dbl = d;
 125     return S_OK;
 126 }
 127
 128 static BSTR alloc_bstr_arg(compile_ctx_t *ctx, const WCHAR *str)
 129 {
 130     if(!ctx->code->bstr_pool_size) {
 131         ctx->code->bstr_pool = heap_alloc(8 * sizeof(BSTR));
 132         if(!ctx->code->bstr_pool)
 133             return NULL;
 134         ctx->code->bstr_pool_size = 8;
 135     }else if(ctx->code->bstr_pool_size == ctx->code->bstr_cnt) {
 136        BSTR *new_pool;
 137
 138         new_pool = heap_realloc(ctx->code->bstr_pool, ctx->code->bstr_pool_size*2*sizeof(BSTR));
 139         if(!new_pool)
 140             return NULL;
 141
 142         ctx->code->bstr_pool = new_pool;
 143         ctx->code->bstr_pool_size *= 2;
 144     }
 145
 146     ctx->code->bstr_pool[ctx->code->bstr_cnt] = SysAllocString(str);
 147     if(!ctx->code->bstr_pool[ctx->code->bstr_cnt])
 148         return NULL;
 149
 150     return ctx->code->bstr_pool[ctx->code->bstr_cnt++];
 151 }
 152
 153 static HRESULT push_instr_bstr_uint(compile_ctx_t *ctx, vbsop_t op, const WCHAR *arg1, unsigned arg2)
 154 {
 155     unsigned instr;
 156     BSTR bstr;
 157
 158     bstr = alloc_bstr_arg(ctx, arg1);
 159     if(!bstr)
 160         return E_OUTOFMEMORY;
 161
 162     instr = push_instr(ctx, op);
 163     if(instr == -1)
 164         return E_OUTOFMEMORY;
 165
 166     instr_ptr(ctx, instr)->arg1.bstr = bstr;
 167     instr_ptr(ctx, instr)->arg2.uint = arg2;
 168     return S_OK;
 169 }
 170
 171 static HRESULT compile_args(compile_ctx_t *ctx, expression_t *args, unsigned *ret)
 172 {
 173     unsigned arg_cnt = 0;
 174     HRESULT hres;
 175
 176     while(args) {
 177         hres = compile_expression(ctx, args);
 178         if(FAILED(hres))
 179             return hres;
 180
 181         arg_cnt++;
 182         args = args->next;
 183     }
 184
 185     *ret = arg_cnt;
 186     return S_OK;
 187 }
 188
 189 static HRESULT compile_member_expression(compile_ctx_t *ctx, member_expression_t *expr, BOOL ret_val)
 190 {
 191     unsigned arg_cnt = 0;
 192     HRESULT hres;
 193
 194     hres = compile_args(ctx, expr->args, &arg_cnt);
 195     if(FAILED(hres))
 196         return hres;
 197
 198     if(expr->obj_expr) {
 199         FIXME("obj_expr not implemented\n");
 200         hres = E_NOTIMPL;
 201     }else {
 202         hres = push_instr_bstr_uint(ctx, ret_val ? OP_icall : OP_icallv, expr->identifier, arg_cnt);
 203     }
 204
 205     return hres;
 206 }
 207
 208 static HRESULT compile_unary_expression(compile_ctx_t *ctx, unary_expression_t *expr, vbsop_t op)
 209 {
 210     HRESULT hres;
 211
 212     hres = compile_expression(ctx, expr->subexpr);
 213     if(FAILED(hres))
 214         return hres;
 215
 216     return push_instr(ctx, op) == -1 ? E_OUTOFMEMORY : S_OK;
 217 }
 218
 219 static HRESULT compile_binary_expression(compile_ctx_t *ctx, binary_expression_t *expr, vbsop_t op)
 220 {
 221     HRESULT hres;
 222
 223     hres = compile_expression(ctx, expr->left);
 224     if(FAILED(hres))
 225         return hres;
 226
 227     hres = compile_expression(ctx, expr->right);
 228     if(FAILED(hres))
 229         return hres;
 230
 231     return push_instr(ctx, op) == -1 ? E_OUTOFMEMORY : S_OK;
 232 }
 233
 234 static HRESULT compile_expression(compile_ctx_t *ctx, expression_t *expr)
 235 {
 236     switch(expr->type) {
 237     case EXPR_BOOL:
 238         return push_instr_int(ctx, OP_bool, ((bool_expression_t*)expr)->value);
 239     case EXPR_DOUBLE:
 240         return push_instr_double(ctx, OP_double, ((double_expression_t*)expr)->value);
 241     case EXPR_EMPTY:
 242         return push_instr(ctx, OP_empty) != -1 ? S_OK : E_OUTOFMEMORY;
 243     case EXPR_EQUAL:
 244         return compile_binary_expression(ctx, (binary_expression_t*)expr, OP_equal);
 245     case EXPR_MEMBER:
 246         return compile_member_expression(ctx, (member_expression_t*)expr, TRUE);
 247     case EXPR_NOT:
 248         return compile_unary_expression(ctx, (unary_expression_t*)expr, OP_not);
 249     case EXPR_NULL:
 250         return push_instr(ctx, OP_null) != -1 ? S_OK : E_OUTOFMEMORY;
 251     case EXPR_STRING:
 252         return push_instr_str(ctx, OP_string, ((string_expression_t*)expr)->value);
 253     case EXPR_USHORT:
 254         return push_instr_int(ctx, OP_short, ((int_expression_t*)expr)->value);
 255     case EXPR_ULONG:
 256         return push_instr_int(ctx, OP_long, ((int_expression_t*)expr)->value);
 257     default:
 258         FIXME("Unimplemented expression type %d\n", expr->type);
 259         return E_NOTIMPL;
 260     }
 261
 262     return S_OK;
 263 }
 264
 265 static HRESULT compile_statement(compile_ctx_t *ctx, statement_t *stat)
 266 {
 267     HRESULT hres;
 268
 269     while(stat) {
 270         switch(stat->type) {
 271         case STAT_CALL:
 272             hres = compile_member_expression(ctx, ((call_statement_t*)stat)->expr, FALSE);
 273             break;
 274         default:
 275             FIXME("Unimplemented statement type %d\n", stat->type);
 276             hres = E_NOTIMPL;
 277         }
 278
 279         if(FAILED(hres))
 280             return hres;
 281         stat = stat->next;
 282     }
 283
 284     return S_OK;
 285 }
 286
 287 static HRESULT compile_func(compile_ctx_t *ctx, statement_t *stat, function_t *func)
 288 {
 289     HRESULT hres;
 290
 291     func->code_off = ctx->instr_cnt;
 292
 293     hres = compile_statement(ctx, stat);
 294     if(FAILED(hres))
 295         return hres;
 296
 297     if(push_instr(ctx, OP_ret) == -1)
 298         return E_OUTOFMEMORY;
 299
 300     return S_OK;
 301 }
 302
 303 void release_vbscode(vbscode_t *code)
 304 {
 305     unsigned i;
 306
 307     list_remove(&code->entry);
 308
 309     for(i=0; i < code->bstr_cnt; i++)
 310         SysFreeString(code->bstr_pool[i]);
 311
 312     vbsheap_free(&code->heap);
 313
 314     heap_free(code->bstr_pool);
 315     heap_free(code->source);
 316     heap_free(code->instrs);
 317     heap_free(code);
 318 }
 319
 320 static vbscode_t *alloc_vbscode(compile_ctx_t *ctx, const WCHAR *source)
 321 {
 322     vbscode_t *ret;
 323
 324     ret = heap_alloc(sizeof(*ret));
 325     if(!ret)
 326         return NULL;
 327
 328     ret->source = heap_strdupW(source);
 329     if(!ret->source) {
 330         heap_free(ret);
 331         return NULL;
 332     }
 333
 334     ret->instrs = heap_alloc(32*sizeof(instr_t));
 335     if(!ret->instrs) {
 336         release_vbscode(ret);
 337         return NULL;
 338     }
 339
 340     ctx->instr_cnt = 0;
 341     ctx->instr_size = 32;
 342     vbsheap_init(&ret->heap);
 343
 344     ret->option_explicit = ctx->parser.option_explicit;
 345
 346     ret->bstr_pool = NULL;
 347     ret->bstr_pool_size = 0;
 348     ret->bstr_cnt = 0;
 349
 350     ret->global_code.code_ctx = ret;
 351
 352     list_init(&ret->entry);
 353     return ret;
 354 }
 355
 356 HRESULT compile_script(script_ctx_t *script, const WCHAR *src, vbscode_t **ret)
 357 {
 358     compile_ctx_t ctx;
 359     HRESULT hres;
 360
 361     hres = parse_script(&ctx.parser, src);
 362     if(FAILED(hres))
 363         return hres;
 364
 365     ctx.code = alloc_vbscode(&ctx, src);
 366     if(!ctx.code)
 367         return E_OUTOFMEMORY;
 368
 369     hres = compile_func(&ctx, ctx.parser.stats, &ctx.code->global_code);
 370     if(FAILED(hres)) {
 371         release_vbscode(ctx.code);
 372         return hres;
 373     }
 374
 375     list_add_tail(&script->code_list, &ctx.code->entry);
 376     *ret = ctx.code;
 377     return S_OK;
 378 }