/* * 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 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 JS_E_ENUMERATOR_EXPECTED; 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 JS_E_ENUMERATOR_EXPECTED; 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 JS_E_ENUMERATOR_EXPECTED; 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 JS_E_ENUMERATOR_EXPECTED; 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[] = { {L"atEnd", Enumerator_atEnd, PROPF_METHOD}, {L"item", Enumerator_item, PROPF_METHOD}, {L"moveFirst", Enumerator_moveFirst, PROPF_METHOD}, {L"moveNext", 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 JS_E_INVALIDARG; 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; hres = alloc_enumerator(ctx, object_prototype, &enumerator); if(FAILED(hres)) return hres; hres = create_builtin_constructor(ctx, EnumeratorConstr_value, L"Enumerator", &EnumeratorConstr_info, PROPF_CONSTR|7, &enumerator->dispex, ret); jsdisp_release(&enumerator->dispex); return hres; }