wine/dlls/jscript/enumerator.c
Gabriel Ivăncescu 0a4b70a5f6 jscript: Fix EnumVARIANT leak in enumerators.
Signed-off-by: Gabriel Ivăncescu <gabrielopcode@gmail.com>
2023-11-03 21:46:07 +01:00

349 lines
8.6 KiB
C

/*
* 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_this(jsval_t vthis)
{
jsdisp_t *jsdisp = is_object_instance(vthis) ? to_jsdisp(get_object(vthis)) : NULL;
return (jsdisp && is_class(jsdisp, JSCLASS_ENUMERATOR)) ? enumerator_from_jsdisp(jsdisp) : NULL;
}
static inline HRESULT enumvar_get_next_item(EnumeratorInstance *This, script_ctx_t *ctx)
{
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(ctx, &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");
if(This->enumvar)
IEnumVARIANT_Release(This->enumvar);
jsval_release(This->item);
free(dispex);
}
static HRESULT Enumerator_gc_traverse(struct gc_ctx *gc_ctx, enum gc_traverse_op op, jsdisp_t *dispex)
{
return gc_process_linked_val(gc_ctx, op, dispex, &enumerator_from_jsdisp(dispex)->item);
}
static HRESULT Enumerator_atEnd(script_ctx_t *ctx, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv,
jsval_t *r)
{
EnumeratorInstance *This;
if (!(This = enumerator_this(vthis)))
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, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv,
jsval_t *r)
{
EnumeratorInstance *This;
TRACE("\n");
if (!(This = enumerator_this(vthis)))
return JS_E_ENUMERATOR_EXPECTED;
return r ? jsval_copy(This->item, r) : S_OK;
}
static HRESULT Enumerator_moveFirst(script_ctx_t *ctx, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv,
jsval_t *r)
{
EnumeratorInstance *This;
HRESULT hres = S_OK;
TRACE("\n");
if (!(This = enumerator_this(vthis)))
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, ctx);
if(FAILED(hres))
return hres;
}
if (r)
*r = jsval_undefined();
return S_OK;
}
static HRESULT Enumerator_moveNext(script_ctx_t *ctx, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv,
jsval_t *r)
{
EnumeratorInstance *This;
HRESULT hres = S_OK;
TRACE("\n");
if (!(This = enumerator_this(vthis)))
return JS_E_ENUMERATOR_EXPECTED;
if (This->enumvar)
{
hres = enumvar_get_next_item(This, ctx);
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,
ARRAY_SIZE(Enumerator_props),
Enumerator_props,
NULL,
NULL
};
static const builtin_info_t EnumeratorInst_info = {
JSCLASS_ENUMERATOR,
NULL,
0,
NULL,
Enumerator_destructor,
NULL,
NULL,
NULL,
NULL,
Enumerator_gc_traverse
};
static HRESULT alloc_enumerator(script_ctx_t *ctx, jsdisp_t *object_prototype, EnumeratorInstance **ret)
{
EnumeratorInstance *enumerator;
HRESULT hres;
enumerator = calloc(1, 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))
{
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, ctx);
if (FAILED(hres))
{
jsdisp_release(&enumerator->dispex);
return hres;
}
*ret = &enumerator->dispex;
return S_OK;
}
static HRESULT EnumeratorConstr_value(script_ctx_t *ctx, jsval_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;
if(r) *r = jsval_obj(obj);
else jsdisp_release(obj);
break;
}
default:
FIXME("unimplemented flags: %x\n", flags);
return E_NOTIMPL;
}
return S_OK;
}
static const builtin_info_t EnumeratorConstr_info = {
JSCLASS_FUNCTION,
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;
}