git.oblomov.eu Git - wine/blob - dlls/jscript/jsutils.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 "config.h"
  20 #include "wine/port.h"
  21
  22 #include <math.h>
  23 #include <assert.h>
  24
  25 #include "jscript.h"
  26 #include "engine.h"
  27
  28 #include "wine/debug.h"
  29
  30 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
  31 WINE_DECLARE_DEBUG_CHANNEL(heap);
  32
  33 const char *debugstr_variant(const VARIANT *v)
  34 {
  35     if(!v)
  36         return "(null)";
  37
  38     switch(V_VT(v)) {
  39     case VT_EMPTY:
  40         return "{VT_EMPTY}";
  41     case VT_NULL:
  42         return "{VT_NULL}";
  43     case VT_I4:
  44         return wine_dbg_sprintf("{VT_I4: %d}", V_I4(v));
  45     case VT_UI4:
  46         return wine_dbg_sprintf("{VT_UI4: %u}", V_UI4(v));
  47     case VT_R8:
  48         return wine_dbg_sprintf("{VT_R8: %lf}", V_R8(v));
  49     case VT_BSTR:
  50         return wine_dbg_sprintf("{VT_BSTR: %s}", debugstr_w(V_BSTR(v)));
  51     case VT_DISPATCH:
  52         return wine_dbg_sprintf("{VT_DISPATCH: %p}", V_DISPATCH(v));
  53     case VT_BOOL:
  54         return wine_dbg_sprintf("{VT_BOOL: %x}", V_BOOL(v));
  55     default:
  56         return wine_dbg_sprintf("{vt %d}", V_VT(v));
  57     }
  58 }
  59
  60 #define MIN_BLOCK_SIZE  128
  61 #define ARENA_FREE_FILLER  0xaa
  62
  63 static inline DWORD block_size(DWORD block)
  64 {
  65     return MIN_BLOCK_SIZE << block;
  66 }
  67
  68 void jsheap_init(jsheap_t *heap)
  69 {
  70     memset(heap, 0, sizeof(*heap));
  71     list_init(&heap->custom_blocks);
  72 }
  73
  74 void *jsheap_alloc(jsheap_t *heap, DWORD size)
  75 {
  76     struct list *list;
  77     void *tmp;
  78
  79     if(!heap->block_cnt) {
  80         if(!heap->blocks) {
  81             heap->blocks = heap_alloc(sizeof(void*));
  82             if(!heap->blocks)
  83                 return NULL;
  84         }
  85
  86         tmp = heap_alloc(block_size(0));
  87         if(!tmp)
  88             return NULL;
  89
  90         heap->blocks[0] = tmp;
  91         heap->block_cnt = 1;
  92     }
  93
  94     if(heap->offset + size <= block_size(heap->last_block)) {
  95         tmp = ((BYTE*)heap->blocks[heap->last_block])+heap->offset;
  96         heap->offset += size;
  97         return tmp;
  98     }
  99
 100     if(size <= block_size(heap->last_block+1)) {
 101         if(heap->last_block+1 == heap->block_cnt) {
 102             tmp = heap_realloc(heap->blocks, (heap->block_cnt+1)*sizeof(void*));
 103             if(!tmp)
 104                 return NULL;
 105
 106             heap->blocks = tmp;
 107             heap->blocks[heap->block_cnt] = heap_alloc(block_size(heap->block_cnt));
 108             if(!heap->blocks[heap->block_cnt])
 109                 return NULL;
 110
 111             heap->block_cnt++;
 112         }
 113
 114         heap->last_block++;
 115         heap->offset = size;
 116         return heap->blocks[heap->last_block];
 117     }
 118
 119     list = heap_alloc(size + sizeof(struct list));
 120     if(!list)
 121         return NULL;
 122
 123     list_add_head(&heap->custom_blocks, list);
 124     return list+1;
 125 }
 126
 127 void *jsheap_grow(jsheap_t *heap, void *mem, DWORD size, DWORD inc)
 128 {
 129     if(mem == (BYTE*)heap->blocks[heap->last_block] + heap->offset-size
 130        && heap->offset+inc < block_size(heap->last_block)) {
 131         heap->offset += inc;
 132         return mem;
 133     }
 134
 135     return jsheap_alloc(heap, size+inc);
 136 }
 137
 138 void jsheap_clear(jsheap_t *heap)
 139 {
 140     struct list *tmp;
 141
 142     if(!heap)
 143         return;
 144
 145     while((tmp = list_next(&heap->custom_blocks, &heap->custom_blocks))) {
 146         list_remove(tmp);
 147         heap_free(tmp);
 148     }
 149
 150     if(WARN_ON(heap)) {
 151         DWORD i;
 152
 153         for(i=0; i < heap->block_cnt; i++)
 154             memset(heap->blocks[i], ARENA_FREE_FILLER, block_size(i));
 155     }
 156
 157     heap->last_block = heap->offset = 0;
 158     heap->mark = FALSE;
 159 }
 160
 161 void jsheap_free(jsheap_t *heap)
 162 {
 163     DWORD i;
 164
 165     jsheap_clear(heap);
 166
 167     for(i=0; i < heap->block_cnt; i++)
 168         heap_free(heap->blocks[i]);
 169     heap_free(heap->blocks);
 170
 171     jsheap_init(heap);
 172 }
 173
 174 jsheap_t *jsheap_mark(jsheap_t *heap)
 175 {
 176     if(heap->mark)
 177         return NULL;
 178
 179     heap->mark = TRUE;
 180     return heap;
 181 }
 182
 183 /* ECMA-262 3rd Edition    9.1 */
 184 HRESULT to_primitive(script_ctx_t *ctx, VARIANT *v, jsexcept_t *ei, VARIANT *ret, hint_t hint)
 185 {
 186     switch(V_VT(v)) {
 187     case VT_EMPTY:
 188     case VT_NULL:
 189     case VT_BOOL:
 190     case VT_I4:
 191     case VT_R8:
 192         *ret = *v;
 193         break;
 194     case VT_BSTR:
 195         V_VT(ret) = VT_BSTR;
 196         V_BSTR(ret) = SysAllocString(V_BSTR(v));
 197         break;
 198     case VT_DISPATCH: {
 199         jsdisp_t *jsdisp;
 200         DISPID id;
 201         DISPPARAMS dp = {NULL, NULL, 0, 0};
 202         HRESULT hres;
 203
 204         static const WCHAR toStringW[] = {'t','o','S','t','r','i','n','g',0};
 205         static const WCHAR valueOfW[] = {'v','a','l','u','e','O','f',0};
 206
 207         if(!V_DISPATCH(v)) {
 208             V_VT(ret) = VT_NULL;
 209             break;
 210         }
 211
 212         jsdisp = iface_to_jsdisp((IUnknown*)V_DISPATCH(v));
 213         if(!jsdisp) {
 214             V_VT(ret) = VT_EMPTY;
 215             return disp_propget(ctx, V_DISPATCH(v), DISPID_VALUE, ret, ei);
 216         }
 217
 218         if(hint == NO_HINT)
 219             hint = is_class(jsdisp, JSCLASS_DATE) ? HINT_STRING : HINT_NUMBER;
 220
 221         /* Native implementation doesn't throw TypeErrors, returns strange values */
 222
 223         hres = jsdisp_get_id(jsdisp, hint == HINT_STRING ? toStringW : valueOfW, 0, &id);
 224         if(SUCCEEDED(hres)) {
 225             hres = jsdisp_call(jsdisp, id, DISPATCH_METHOD, &dp, ret, ei);
 226             if(FAILED(hres)) {
 227                 WARN("call error - forwarding exception\n");
 228                 jsdisp_release(jsdisp);
 229                 return hres;
 230             }
 231             else if(V_VT(ret) != VT_DISPATCH) {
 232                 jsdisp_release(jsdisp);
 233                 return S_OK;
 234             }
 235             else
 236                 IDispatch_Release(V_DISPATCH(ret));
 237         }
 238
 239         hres = jsdisp_get_id(jsdisp, hint == HINT_STRING ? valueOfW : toStringW, 0, &id);
 240         if(SUCCEEDED(hres)) {
 241             hres = jsdisp_call(jsdisp, id, DISPATCH_METHOD, &dp, ret, ei);
 242             if(FAILED(hres)) {
 243                 WARN("call error - forwarding exception\n");
 244                 jsdisp_release(jsdisp);
 245                 return hres;
 246             }
 247             else if(V_VT(ret) != VT_DISPATCH) {
 248                 jsdisp_release(jsdisp);
 249                 return S_OK;
 250             }
 251             else
 252                 IDispatch_Release(V_DISPATCH(ret));
 253         }
 254
 255         jsdisp_release(jsdisp);
 256
 257         WARN("failed\n");
 258         return throw_type_error(ctx, ei, JS_E_TO_PRIMITIVE, NULL);
 259     }
 260     default:
 261         FIXME("Unimplemented for vt %d\n", V_VT(v));
 262         return E_NOTIMPL;
 263     }
 264
 265     return S_OK;
 266 }
 267
 268 /* ECMA-262 3rd Edition    9.2 */
 269 HRESULT to_boolean(VARIANT *v, VARIANT_BOOL *b)
 270 {
 271     switch(V_VT(v)) {
 272     case VT_EMPTY:
 273     case VT_NULL:
 274         *b = VARIANT_FALSE;
 275         break;
 276     case VT_I4:
 277         *b = V_I4(v) ? VARIANT_TRUE : VARIANT_FALSE;
 278         break;
 279     case VT_R8:
 280         if(isnan(V_R8(v))) *b = VARIANT_FALSE;
 281         else *b = V_R8(v) ? VARIANT_TRUE : VARIANT_FALSE;
 282         break;
 283     case VT_BSTR:
 284         *b = V_BSTR(v) && *V_BSTR(v) ? VARIANT_TRUE : VARIANT_FALSE;
 285         break;
 286     case VT_DISPATCH:
 287         *b = V_DISPATCH(v) ? VARIANT_TRUE : VARIANT_FALSE;
 288         break;
 289     case VT_BOOL:
 290         *b = V_BOOL(v);
 291         break;
 292     default:
 293         FIXME("unimplemented for vt %d\n", V_VT(v));
 294         return E_NOTIMPL;
 295     }
 296
 297     return S_OK;
 298 }
 299
 300 static int hex_to_int(WCHAR c)
 301 {
 302     if('0' <= c && c <= '9')
 303         return c-'0';
 304
 305     if('a' <= c && c <= 'f')
 306         return c-'a'+10;
 307
 308     if('A' <= c && c <= 'F')
 309         return c-'A'+10;
 310
 311     return -1;
 312 }
 313
 314 /* ECMA-262 3rd Edition    9.3.1 */
 315 static HRESULT str_to_number(BSTR str, double *ret)
 316 {
 317     const WCHAR *ptr = str;
 318     BOOL neg = FALSE;
 319     DOUBLE d = 0.0;
 320
 321     static const WCHAR infinityW[] = {'I','n','f','i','n','i','t','y'};
 322
 323     while(isspaceW(*ptr))
 324         ptr++;
 325
 326     if(*ptr == '-') {
 327         neg = TRUE;
 328         ptr++;
 329     }else if(*ptr == '+') {
 330         ptr++;
 331     }
 332
 333     if(!strncmpW(ptr, infinityW, sizeof(infinityW)/sizeof(WCHAR))) {
 334         ptr += sizeof(infinityW)/sizeof(WCHAR);
 335         while(*ptr && isspaceW(*ptr))
 336             ptr++;
 337
 338         if(*ptr)
 339             *ret = ret_nan();
 340         else
 341             *ret = neg ? -ret_inf() : ret_inf();
 342         return S_OK;
 343     }
 344
 345     if(*ptr == '0' && ptr[1] == 'x') {
 346         DWORD l = 0;
 347
 348         ptr += 2;
 349         while((l = hex_to_int(*ptr)) != -1) {
 350             d = d*16 + l;
 351             ptr++;
 352         }
 353
 354         *ret = d;
 355         return S_OK;
 356     }
 357
 358     while(isdigitW(*ptr))
 359         d = d*10 + (*ptr++ - '0');
 360
 361     if(*ptr == 'e' || *ptr == 'E') {
 362         BOOL eneg = FALSE;
 363         LONG l = 0;
 364
 365         ptr++;
 366         if(*ptr == '-') {
 367             ptr++;
 368             eneg = TRUE;
 369         }else if(*ptr == '+') {
 370             ptr++;
 371         }
 372
 373         while(isdigitW(*ptr))
 374             l = l*10 + (*ptr++ - '0');
 375         if(eneg)
 376             l = -l;
 377
 378         d *= pow(10, l);
 379     }else if(*ptr == '.') {
 380         DOUBLE dec = 0.1;
 381
 382         ptr++;
 383         while(isdigitW(*ptr)) {
 384             d += dec * (*ptr++ - '0');
 385             dec *= 0.1;
 386         }
 387     }
 388
 389     while(isspaceW(*ptr))
 390         ptr++;
 391
 392     if(*ptr) {
 393         *ret = ret_nan();
 394         return S_OK;
 395     }
 396
 397     if(neg)
 398         d = -d;
 399
 400     *ret = d;
 401     return S_OK;
 402 }
 403
 404 /* ECMA-262 3rd Edition    9.3 */
 405 HRESULT to_number(script_ctx_t *ctx, VARIANT *v, jsexcept_t *ei, double *ret)
 406 {
 407     switch(V_VT(v)) {
 408     case VT_EMPTY:
 409         *ret = ret_nan();
 410         break;
 411     case VT_NULL:
 412         *ret = 0;
 413         break;
 414     case VT_I4:
 415         *ret = V_I4(v);
 416         break;
 417     case VT_R8:
 418         *ret = V_R8(v);
 419         break;
 420     case VT_BSTR:
 421         return str_to_number(V_BSTR(v), ret);
 422     case VT_DISPATCH: {
 423         VARIANT prim;
 424         HRESULT hres;
 425
 426         hres = to_primitive(ctx, v, ei, &prim, HINT_NUMBER);
 427         if(FAILED(hres))
 428             return hres;
 429
 430         hres = to_number(ctx, &prim, ei, ret);
 431         VariantClear(&prim);
 432         return hres;
 433     }
 434     case VT_BOOL:
 435         *ret = V_BOOL(v) ? 1 : 0;
 436         break;
 437     default:
 438         FIXME("unimplemented for vt %d\n", V_VT(v));
 439         return E_NOTIMPL;
 440     }
 441
 442     return S_OK;
 443 }
 444
 445 /* ECMA-262 3rd Edition    9.4 */
 446 HRESULT to_integer(script_ctx_t *ctx, VARIANT *v, jsexcept_t *ei, VARIANT *ret)
 447 {
 448     double n;
 449     HRESULT hres;
 450
 451     if(V_VT(v) == VT_I4) {
 452         *ret = *v;
 453         return S_OK;
 454     }
 455
 456     hres = to_number(ctx, v, ei, &n);
 457     if(FAILED(hres))
 458         return hres;
 459
 460     if(isnan(n)) {
 461         V_VT(ret) = VT_I4;
 462         V_I4(ret) = 0;
 463     }else {
 464         num_set_val(ret, n >= 0.0 ? floor(n) : -floor(-n));
 465     }
 466
 467     return S_OK;
 468 }
 469
 470 /* ECMA-262 3rd Edition    9.5 */
 471 HRESULT to_int32(script_ctx_t *ctx, VARIANT *v, jsexcept_t *ei, INT *ret)
 472 {
 473     double n;
 474     HRESULT hres;
 475
 476     if(V_VT(v) == VT_I4) {
 477         *ret = V_I4(v);
 478         return S_OK;
 479     }
 480
 481     hres = to_number(ctx, v, ei, &n);
 482     if(FAILED(hres))
 483         return hres;
 484
 485     *ret = isnan(n) || isinf(n) ? 0 : n;
 486     return S_OK;
 487 }
 488
 489 /* ECMA-262 3rd Edition    9.6 */
 490 HRESULT to_uint32(script_ctx_t *ctx, VARIANT *v, jsexcept_t *ei, DWORD *ret)
 491 {
 492     double n;
 493     HRESULT hres;
 494
 495     if(V_VT(v) == VT_I4) {
 496         *ret = V_I4(v);
 497         return S_OK;
 498     }
 499
 500     hres = to_number(ctx, v, ei, &n);
 501     if(FAILED(hres))
 502         return hres;
 503
 504     *ret = isnan(n) || isinf(n) ? 0 : n;
 505     return S_OK;
 506 }
 507
 508 BSTR int_to_bstr(int i)
 509 {
 510     WCHAR buf[12], *p;
 511     BOOL neg = FALSE;
 512
 513     if(!i) {
 514         static const WCHAR zeroW[] = {'0',0};
 515         return SysAllocString(zeroW);
 516     }
 517
 518     if(i < 0) {
 519         neg = TRUE;
 520         i = -i;
 521     }
 522
 523     p = buf + sizeof(buf)/sizeof(*buf)-1;
 524     *p-- = 0;
 525     while(i) {
 526         *p-- = i%10 + '0';
 527         i /= 10;
 528     }
 529
 530     if(neg)
 531         *p = '-';
 532     else
 533         p++;
 534
 535     return SysAllocString(p);
 536 }
 537
 538 HRESULT double_to_bstr(double n, BSTR *str)
 539 {
 540     const WCHAR NaNW[] = {'N','a','N',0};
 541     const WCHAR InfinityW[] = {'-','I','n','f','i','n','i','t','y',0};
 542
 543     if(isnan(n)) {
 544        *str = SysAllocString(NaNW);
 545     }else if(isinf(n)) {
 546         *str = SysAllocString(n<0 ? InfinityW : InfinityW+1);
 547     }else {
 548         VARIANT strv, v;
 549         HRESULT hres;
 550
 551         V_VT(&v) = VT_R8;
 552         V_R8(&v) = n;
 553         V_VT(&strv) = VT_EMPTY;
 554         hres = VariantChangeTypeEx(&strv, &v, MAKELCID(MAKELANGID(LANG_ENGLISH,SUBLANG_ENGLISH_US),SORT_DEFAULT), 0, VT_BSTR);
 555         if(FAILED(hres))
 556             return hres;
 557
 558         *str = V_BSTR(&strv);
 559     }
 560
 561     return *str ? S_OK : E_OUTOFMEMORY;
 562 }
 563
 564 /* ECMA-262 3rd Edition    9.8 */
 565 HRESULT to_string(script_ctx_t *ctx, VARIANT *v, jsexcept_t *ei, BSTR *str)
 566 {
 567     const WCHAR undefinedW[] = {'u','n','d','e','f','i','n','e','d',0};
 568     const WCHAR nullW[] = {'n','u','l','l',0};
 569     const WCHAR trueW[] = {'t','r','u','e',0};
 570     const WCHAR falseW[] = {'f','a','l','s','e',0};
 571
 572     switch(V_VT(v)) {
 573     case VT_EMPTY:
 574         *str = SysAllocString(undefinedW);
 575         break;
 576     case VT_NULL:
 577         *str = SysAllocString(nullW);
 578         break;
 579     case VT_I4:
 580         *str = int_to_bstr(V_I4(v));
 581         break;
 582     case VT_R8:
 583         return double_to_bstr(V_R8(v), str);
 584     case VT_BSTR:
 585         *str = SysAllocString(V_BSTR(v));
 586         break;
 587     case VT_DISPATCH: {
 588         VARIANT prim;
 589         HRESULT hres;
 590
 591         hres = to_primitive(ctx, v, ei, &prim, HINT_STRING);
 592         if(FAILED(hres))
 593             return hres;
 594
 595         hres = to_string(ctx, &prim, ei, str);
 596         VariantClear(&prim);
 597         return hres;
 598     }
 599     case VT_BOOL:
 600         *str = SysAllocString(V_BOOL(v) ? trueW : falseW);
 601         break;
 602     default:
 603         FIXME("unsupported vt %d\n", V_VT(v));
 604         return E_NOTIMPL;
 605     }
 606
 607     return *str ? S_OK : E_OUTOFMEMORY;
 608 }
 609
 610 /* ECMA-262 3rd Edition    9.9 */
 611 HRESULT to_object(script_ctx_t *ctx, VARIANT *v, IDispatch **disp)
 612 {
 613     jsdisp_t *dispex;
 614     HRESULT hres;
 615
 616     switch(V_VT(v)) {
 617     case VT_BSTR:
 618         hres = create_string(ctx, V_BSTR(v), SysStringLen(V_BSTR(v)), &dispex);
 619         if(FAILED(hres))
 620             return hres;
 621
 622         *disp = to_disp(dispex);
 623         break;
 624     case VT_I4:
 625     case VT_R8:
 626         hres = create_number(ctx, v, &dispex);
 627         if(FAILED(hres))
 628             return hres;
 629
 630         *disp = to_disp(dispex);
 631         break;
 632     case VT_DISPATCH:
 633         if(V_DISPATCH(v)) {
 634             IDispatch_AddRef(V_DISPATCH(v));
 635             *disp = V_DISPATCH(v);
 636         }else {
 637             jsdisp_t *obj;
 638
 639             hres = create_object(ctx, NULL, &obj);
 640             if(FAILED(hres))
 641                 return hres;
 642
 643             *disp = to_disp(obj);
 644         }
 645         break;
 646     case VT_BOOL:
 647         hres = create_bool(ctx, V_BOOL(v), &dispex);
 648         if(FAILED(hres))
 649             return hres;
 650
 651         *disp = to_disp(dispex);
 652         break;
 653     case VT_ARRAY|VT_VARIANT:
 654         hres = create_vbarray(ctx, V_ARRAY(v), &dispex);
 655         if(FAILED(hres))
 656             return hres;
 657
 658         *disp = to_disp(dispex);
 659         break;
 660     default:
 661         FIXME("unsupported vt %d\n", V_VT(v));
 662         return E_NOTIMPL;
 663     }
 664
 665     return S_OK;
 666 }
 667
 668 HRESULT variant_change_type(script_ctx_t *ctx, VARIANT *dst, VARIANT *src, VARTYPE vt)
 669 {
 670     jsexcept_t ei;
 671     HRESULT hres;
 672
 673     memset(&ei, 0, sizeof(ei));
 674
 675     switch(vt) {
 676     case VT_I2:
 677     case VT_I4: {
 678         INT i;
 679
 680         hres = to_int32(ctx, src, &ei, &i);
 681         if(SUCCEEDED(hres)) {
 682             if(vt == VT_I4)
 683                 V_I4(dst) = i;
 684             else
 685                 V_I2(dst) = i;
 686         }
 687         break;
 688     }
 689     case VT_R8: {
 690         double n;
 691         hres = to_number(ctx, src, &ei, &n);
 692         if(SUCCEEDED(hres))
 693             V_R8(dst) = n;
 694         break;
 695     }
 696     case VT_R4: {
 697         double n;
 698
 699         hres = to_number(ctx, src, &ei, &n);
 700         if(SUCCEEDED(hres))
 701             V_R4(dst) = n;
 702         break;
 703     }
 704     case VT_BOOL: {
 705         VARIANT_BOOL b;
 706
 707         hres = to_boolean(src, &b);
 708         if(SUCCEEDED(hres))
 709             V_BOOL(dst) = b;
 710         break;
 711     }
 712     case VT_BSTR: {
 713         BSTR str;
 714
 715         hres = to_string(ctx, src, &ei, &str);
 716         if(SUCCEEDED(hres))
 717             V_BSTR(dst) = str;
 718         break;
 719     }
 720     case VT_EMPTY:
 721         hres = V_VT(src) == VT_EMPTY ? S_OK : E_NOTIMPL;
 722         break;
 723     case VT_NULL:
 724         hres = V_VT(src) == VT_NULL ? S_OK : E_NOTIMPL;
 725         break;
 726     default:
 727         FIXME("vt %d not implemented\n", vt);
 728         hres = E_NOTIMPL;
 729     }
 730
 731     if(FAILED(hres)) {
 732         VariantClear(&ei.var);
 733         return hres;
 734     }
 735
 736     V_VT(dst) = vt;
 737     return S_OK;
 738 }
 739
 740 static inline JSCaller *impl_from_IServiceProvider(IServiceProvider *iface)
 741 {
 742     return CONTAINING_RECORD(iface, JSCaller, IServiceProvider_iface);
 743 }
 744
 745 static HRESULT WINAPI JSCaller_QueryInterface(IServiceProvider *iface, REFIID riid, void **ppv)
 746 {
 747     JSCaller *This = impl_from_IServiceProvider(iface);
 748
 749     if(IsEqualGUID(&IID_IUnknown, riid)) {
 750         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
 751         *ppv = &This->IServiceProvider_iface;
 752     }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
 753         TRACE("(%p)->(IID_IServiceProvider %p)\n", This, ppv);
 754         *ppv = &This->IServiceProvider_iface;
 755     }else {
 756         WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
 757         *ppv = NULL;
 758         return E_NOINTERFACE;
 759     }
 760
 761     IUnknown_AddRef((IUnknown*)*ppv);
 762     return S_OK;
 763 }
 764
 765 static ULONG WINAPI JSCaller_AddRef(IServiceProvider *iface)
 766 {
 767     JSCaller *This = impl_from_IServiceProvider(iface);
 768     LONG ref = InterlockedIncrement(&This->ref);
 769
 770     TRACE("(%p) ref=%d\n", This, ref);
 771
 772     return ref;
 773 }
 774
 775 static ULONG WINAPI JSCaller_Release(IServiceProvider *iface)
 776 {
 777     JSCaller *This = impl_from_IServiceProvider(iface);
 778     LONG ref = InterlockedIncrement(&This->ref);
 779
 780     TRACE("(%p) ref=%d\n", This, ref);
 781
 782     if(!ref) {
 783         assert(!This->ctx);
 784         heap_free(This);
 785     }
 786
 787     return ref;
 788 }
 789
 790 static HRESULT WINAPI JSCaller_QueryService(IServiceProvider *iface, REFGUID guidService,
 791         REFIID riid, void **ppv)
 792 {
 793     JSCaller *This = impl_from_IServiceProvider(iface);
 794
 795     if(IsEqualGUID(guidService, &SID_VariantConversion) && This->ctx && This->ctx->active_script) {
 796         TRACE("(%p)->(SID_VariantConversion)\n", This);
 797         return IActiveScript_QueryInterface(This->ctx->active_script, riid, ppv);
 798     }
 799
 800     FIXME("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
 801
 802     *ppv = NULL;
 803     return E_NOINTERFACE;
 804 }
 805
 806 static const IServiceProviderVtbl ServiceProviderVtbl = {
 807     JSCaller_QueryInterface,
 808     JSCaller_AddRef,
 809     JSCaller_Release,
 810     JSCaller_QueryService
 811 };
 812
 813 HRESULT create_jscaller(script_ctx_t *ctx)
 814 {
 815     JSCaller *ret;
 816
 817     ret = heap_alloc(sizeof(*ret));
 818     if(!ret)
 819         return E_OUTOFMEMORY;
 820
 821     ret->IServiceProvider_iface.lpVtbl = &ServiceProviderVtbl;
 822     ret->ref = 1;
 823     ret->ctx = ctx;
 824
 825     ctx->jscaller = ret;
 826     return S_OK;
 827 }