Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349/* * Copyright 2019 Andreas Maier * * 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 <assert.h>
#include "jscript.h"
#include "wine/debug.h"
WINE_DEFAULT_DEBUG_CHANNEL(jscript);
typedef struct { jsdisp_t dispex; /* IEnumVARIANT returned by _NewEnum */ IEnumVARIANT *enumvar; /* current item */ jsval_t item; BOOL atend;} EnumeratorInstance;
static const WCHAR atEndW[] = {'a','t','E','n','d',0};static const WCHAR itemW[] = {'i','t','e','m',0};static const WCHAR moveFirstW[] = {'m','o','v','e','F','i','r','s','t',0};static const WCHAR moveNextW[] = {'m','o','v','e','N','e','x','t',0};
static inline EnumeratorInstance *enumerator_from_jsdisp(jsdisp_t *jsdisp){ return CONTAINING_RECORD(jsdisp, EnumeratorInstance, dispex);}
static inline EnumeratorInstance *enumerator_from_vdisp(vdisp_t *vdisp){ return enumerator_from_jsdisp(vdisp->u.jsdisp);}
static inline EnumeratorInstance *enumerator_this(vdisp_t *jsthis){ return is_vclass(jsthis, JSCLASS_ENUMERATOR) ? enumerator_from_vdisp(jsthis) : NULL;}
static inline HRESULT enumvar_get_next_item(EnumeratorInstance *This){ HRESULT hres; VARIANT nextitem;
if (This->atend) return S_OK;
/* don't leak previous value */ jsval_release(This->item);
/* not at end ... get next item */ VariantInit(&nextitem); hres = IEnumVARIANT_Next(This->enumvar, 1, &nextitem, NULL); if (hres == S_OK) { hres = variant_to_jsval(&nextitem, &This->item); VariantClear(&nextitem); if (FAILED(hres)) { WARN("failed to convert jsval to variant!\n"); This->item = jsval_undefined(); return hres; } } else { This->item = jsval_undefined(); This->atend = TRUE; }
return S_OK;}
static void Enumerator_destructor(jsdisp_t *dispex){ EnumeratorInstance *This = enumerator_from_jsdisp(dispex);
TRACE("\n");
jsval_release(This->item); heap_free(dispex);}
static HRESULT Enumerator_atEnd(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv, jsval_t *r){ EnumeratorInstance *This;
if (!(This = enumerator_this(jsthis))) return throw_type_error(ctx, JS_E_ENUMERATOR_EXPECTED, NULL);
TRACE("%d\n", This->atend);
if (r) *r = jsval_bool(This->atend); return S_OK;}
static HRESULT Enumerator_item(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv, jsval_t *r){ EnumeratorInstance *This;
TRACE("\n");
if (!(This = enumerator_this(jsthis))) return throw_type_error(ctx, JS_E_ENUMERATOR_EXPECTED, NULL);
return r ? jsval_copy(This->item, r) : S_OK;}
static HRESULT Enumerator_moveFirst(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv, jsval_t *r){ EnumeratorInstance *This; HRESULT hres = S_OK;
TRACE("\n");
if (!(This = enumerator_this(jsthis))) return throw_type_error(ctx, JS_E_ENUMERATOR_EXPECTED, NULL);
if (This->enumvar) { hres = IEnumVARIANT_Reset(This->enumvar); if (FAILED(hres)) return hres;
This->atend = FALSE; hres = enumvar_get_next_item(This); if(FAILED(hres)) return hres; }
if (r) *r = jsval_undefined(); return S_OK;}
static HRESULT Enumerator_moveNext(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv, jsval_t *r){ EnumeratorInstance *This; HRESULT hres = S_OK;
TRACE("\n");
if (!(This = enumerator_this(jsthis))) return throw_type_error(ctx, JS_E_ENUMERATOR_EXPECTED, NULL);
if (This->enumvar) { hres = enumvar_get_next_item(This); if (FAILED(hres)) return hres; }
if (r) *r = jsval_undefined(); return S_OK;}
static const builtin_prop_t Enumerator_props[] = { {atEndW, Enumerator_atEnd, PROPF_METHOD}, {itemW, Enumerator_item, PROPF_METHOD}, {moveFirstW, Enumerator_moveFirst, PROPF_METHOD}, {moveNextW, Enumerator_moveNext, PROPF_METHOD},};
static const builtin_info_t Enumerator_info = { JSCLASS_ENUMERATOR, {NULL, NULL, 0}, ARRAY_SIZE(Enumerator_props), Enumerator_props, NULL, NULL};
static const builtin_info_t EnumeratorInst_info = { JSCLASS_ENUMERATOR, {NULL, NULL, 0, NULL}, 0, NULL, Enumerator_destructor, NULL};
static HRESULT alloc_enumerator(script_ctx_t *ctx, jsdisp_t *object_prototype, EnumeratorInstance **ret){ EnumeratorInstance *enumerator; HRESULT hres;
enumerator = heap_alloc_zero(sizeof(EnumeratorInstance)); if(!enumerator) return E_OUTOFMEMORY;
if(object_prototype) hres = init_dispex(&enumerator->dispex, ctx, &Enumerator_info, object_prototype); else hres = init_dispex_from_constr(&enumerator->dispex, ctx, &EnumeratorInst_info, ctx->enumerator_constr);
if(FAILED(hres)) { heap_free(enumerator); return hres; }
*ret = enumerator; return S_OK;}
static HRESULT create_enumerator(script_ctx_t *ctx, jsval_t *argv, jsdisp_t **ret){ EnumeratorInstance *enumerator; HRESULT hres; IDispatch *obj; DISPPARAMS dispparams = {NULL, NULL, 0, 0}; IEnumVARIANT *enumvar = NULL;
if (argv) { VARIANT varresult;
if (!is_object_instance(*argv)) { FIXME("I don't know how to handle this type!\n"); return E_NOTIMPL; }
obj = get_object(*argv);
/* Try to get a IEnumVARIANT by _NewEnum */ VariantInit(&varresult); hres = IDispatch_Invoke(obj, DISPID_NEWENUM, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, NULL, NULL); if (FAILED(hres)) { WARN("Enumerator: no DISPID_NEWENUM.\n"); return E_INVALIDARG; }
if ((V_VT(&varresult) == VT_DISPATCH) || (V_VT(&varresult) == VT_UNKNOWN)) { hres = IUnknown_QueryInterface(V_UNKNOWN(&varresult), &IID_IEnumVARIANT, (void**)&enumvar); } else { FIXME("Enumerator: NewEnum unexpected type of varresult (%d).\n", V_VT(&varresult)); hres = E_INVALIDARG; } VariantClear(&varresult); if (FAILED(hres)) return hres; }
hres = alloc_enumerator(ctx, NULL, &enumerator); if (FAILED(hres)) { if (enumvar) IEnumVARIANT_Release(enumvar); return hres; }
enumerator->enumvar = enumvar; enumerator->atend = !enumvar; hres = enumvar_get_next_item(enumerator); if (FAILED(hres)) { jsdisp_release(&enumerator->dispex); return hres; }
*ret = &enumerator->dispex; return S_OK;}
static HRESULT EnumeratorConstr_value(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv, jsval_t *r){ jsdisp_t *obj; HRESULT hres;
TRACE("\n");
switch(flags) { case DISPATCH_CONSTRUCT: { if (argc > 1) return throw_syntax_error(ctx, JS_E_INVALIDARG, NULL);
hres = create_enumerator(ctx, (argc == 1) ? &argv[0] : 0, &obj); if(FAILED(hres)) return hres;
*r = jsval_obj(obj); break; } default: FIXME("unimplemented flags: %x\n", flags); return E_NOTIMPL; }
return S_OK;}
static const builtin_info_t EnumeratorConstr_info = { JSCLASS_FUNCTION, DEFAULT_FUNCTION_VALUE, 0, NULL, NULL, NULL};
HRESULT create_enumerator_constr(script_ctx_t *ctx, jsdisp_t *object_prototype, jsdisp_t **ret){ EnumeratorInstance *enumerator; HRESULT hres; static const WCHAR EnumeratorW[] = {'E','n','u','m','e','r','a','t','o','r',0};
hres = alloc_enumerator(ctx, object_prototype, &enumerator); if(FAILED(hres)) return hres;
hres = create_builtin_constructor(ctx, EnumeratorConstr_value, EnumeratorW, &EnumeratorConstr_info, PROPF_CONSTR|7, &enumerator->dispex, ret); jsdisp_release(&enumerator->dispex);
return hres;}