Git conversion: Make reactos the root directory, move rosapps, rostests, wallpapers into modules, and delete rossubsys.

dll/win32/mshtml/htmlelemcol.c

@@ -0,0 +1,827 @@
+/*
+ * Copyright 2006-2008 Jacek Caban for CodeWeavers
+ *
+ * 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 "mshtml_private.h"
+
+typedef struct {
+    DispatchEx dispex;
+    IHTMLElementCollection IHTMLElementCollection_iface;
+
+    HTMLElement **elems;
+    DWORD len;
+
+    LONG ref;
+} HTMLElementCollection;
+
+typedef struct {
+    IEnumVARIANT IEnumVARIANT_iface;
+
+    LONG ref;
+
+    ULONG iter;
+    HTMLElementCollection *col;
+} HTMLElementCollectionEnum;
+
+typedef struct {
+    HTMLElement **buf;
+    DWORD len;
+    DWORD size;
+} elem_vector_t;
+
+/* FIXME: Handle it better way */
+static inline HTMLElement *elem_from_HTMLDOMNode(HTMLDOMNode *iface)
+{
+    return CONTAINING_RECORD(iface, HTMLElement, node);
+}
+
+static IHTMLElementCollection *HTMLElementCollection_Create(HTMLElement **elems, DWORD len);
+
+static void elem_vector_add(elem_vector_t *buf, HTMLElement *elem)
+{
+    if(buf->len == buf->size) {
+        buf->size <<= 1;
+        buf->buf = heap_realloc(buf->buf, buf->size*sizeof(HTMLElement*));
+    }
+
+    buf->buf[buf->len++] = elem;
+}
+
+static void elem_vector_normalize(elem_vector_t *buf)
+{
+    if(!buf->len) {
+        heap_free(buf->buf);
+        buf->buf = NULL;
+    }else if(buf->size > buf->len) {
+        buf->buf = heap_realloc(buf->buf, buf->len*sizeof(HTMLElement*));
+    }
+
+    buf->size = buf->len;
+}
+
+static inline BOOL is_elem_node(nsIDOMNode *node)
+{
+    UINT16 type=0;
+
+    nsIDOMNode_GetNodeType(node, &type);
+
+    return type == ELEMENT_NODE || type == COMMENT_NODE;
+}
+
+static inline HTMLElementCollectionEnum *impl_from_IEnumVARIANT(IEnumVARIANT *iface)
+{
+    return CONTAINING_RECORD(iface, HTMLElementCollectionEnum, IEnumVARIANT_iface);
+}
+
+static HRESULT WINAPI HTMLElementCollectionEnum_QueryInterface(IEnumVARIANT *iface, REFIID riid, void **ppv)
+{
+    HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
+
+    TRACE("(%p)->(%s %p)\n", This, debugstr_mshtml_guid(riid), ppv);
+
+    if(IsEqualGUID(riid, &IID_IUnknown)) {
+        *ppv = &This->IEnumVARIANT_iface;
+    }else if(IsEqualGUID(riid, &IID_IEnumVARIANT)) {
+        *ppv = &This->IEnumVARIANT_iface;
+    }else {
+        FIXME("Unsupported iface %s\n", debugstr_mshtml_guid(riid));
+        *ppv = NULL;
+        return E_NOINTERFACE;
+    }
+
+    IUnknown_AddRef((IUnknown*)*ppv);
+    return S_OK;
+}
+
+static ULONG WINAPI HTMLElementCollectionEnum_AddRef(IEnumVARIANT *iface)
+{
+    HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
+    LONG ref = InterlockedIncrement(&This->ref);
+
+    TRACE("(%p) ref=%d\n", This, ref);
+
+    return ref;
+}
+
+static ULONG WINAPI HTMLElementCollectionEnum_Release(IEnumVARIANT *iface)
+{
+    HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
+    LONG ref = InterlockedDecrement(&This->ref);
+
+    TRACE("(%p) ref=%d\n", This, ref);
+
+    if(!ref) {
+        IHTMLElementCollection_Release(&This->col->IHTMLElementCollection_iface);
+        heap_free(This);
+    }
+
+    return ref;
+}
+
+static HRESULT WINAPI HTMLElementCollectionEnum_Next(IEnumVARIANT *iface, ULONG celt, VARIANT *rgVar, ULONG *pCeltFetched)
+{
+    HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
+    ULONG fetched = 0;
+
+    TRACE("(%p)->(%d %p %p)\n", This, celt, rgVar, pCeltFetched);
+
+    while(This->iter+fetched < This->col->len && fetched < celt) {
+        V_VT(rgVar+fetched) = VT_DISPATCH;
+        V_DISPATCH(rgVar+fetched) = (IDispatch*)&This->col->elems[This->iter+fetched]->IHTMLElement_iface;
+        IDispatch_AddRef(V_DISPATCH(rgVar+fetched));
+        fetched++;
+    }
+
+    This->iter += fetched;
+    if(pCeltFetched)
+        *pCeltFetched = fetched;
+    return fetched == celt ? S_OK : S_FALSE;
+}
+
+static HRESULT WINAPI HTMLElementCollectionEnum_Skip(IEnumVARIANT *iface, ULONG celt)
+{
+    HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
+
+    TRACE("(%p)->(%d)\n", This, celt);
+
+    if(This->iter + celt > This->col->len) {
+        This->iter = This->col->len;
+        return S_FALSE;
+    }
+
+    This->iter += celt;
+    return S_OK;
+}
+
+static HRESULT WINAPI HTMLElementCollectionEnum_Reset(IEnumVARIANT *iface)
+{
+    HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
+
+    TRACE("(%p)->()\n", This);
+
+    This->iter = 0;
+    return S_OK;
+}
+
+static HRESULT WINAPI HTMLElementCollectionEnum_Clone(IEnumVARIANT *iface, IEnumVARIANT **ppEnum)
+{
+    HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
+    FIXME("(%p)->(%p)\n", This, ppEnum);
+    return E_NOTIMPL;
+}
+
+static const IEnumVARIANTVtbl HTMLElementCollectionEnumVtbl = {
+    HTMLElementCollectionEnum_QueryInterface,
+    HTMLElementCollectionEnum_AddRef,
+    HTMLElementCollectionEnum_Release,
+    HTMLElementCollectionEnum_Next,
+    HTMLElementCollectionEnum_Skip,
+    HTMLElementCollectionEnum_Reset,
+    HTMLElementCollectionEnum_Clone
+};
+
+static inline HTMLElementCollection *impl_from_IHTMLElementCollection(IHTMLElementCollection *iface)
+{
+    return CONTAINING_RECORD(iface, HTMLElementCollection, IHTMLElementCollection_iface);
+}
+
+static HRESULT WINAPI HTMLElementCollection_QueryInterface(IHTMLElementCollection *iface,
+                                                           REFIID riid, void **ppv)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+
+    TRACE("(%p)->(%s %p)\n", This, debugstr_mshtml_guid(riid), ppv);
+
+    if(IsEqualGUID(&IID_IUnknown, riid)) {
+        *ppv = &This->IHTMLElementCollection_iface;
+    }else if(IsEqualGUID(&IID_IHTMLElementCollection, riid)) {
+        *ppv = &This->IHTMLElementCollection_iface;
+    }else if(dispex_query_interface(&This->dispex, riid, ppv)) {
+        return *ppv ? S_OK : E_NOINTERFACE;
+    }else {
+        *ppv = NULL;
+        FIXME("Unsupported iface %s\n", debugstr_mshtml_guid(riid));
+        return E_NOINTERFACE;
+    }
+
+    IHTMLElementCollection_AddRef(&This->IHTMLElementCollection_iface);
+    return S_OK;
+}
+
+static ULONG WINAPI HTMLElementCollection_AddRef(IHTMLElementCollection *iface)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+    LONG ref = InterlockedIncrement(&This->ref);
+
+    TRACE("(%p) ref=%d\n", This, ref);
+
+    return ref;
+}
+
+static ULONG WINAPI HTMLElementCollection_Release(IHTMLElementCollection *iface)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+    LONG ref = InterlockedDecrement(&This->ref);
+
+    TRACE("(%p) ref=%d\n", This, ref);
+
+    if(!ref) {
+        unsigned i;
+
+        for(i=0; i < This->len; i++)
+            node_release(&This->elems[i]->node);
+        heap_free(This->elems);
+
+        release_dispex(&This->dispex);
+        heap_free(This);
+    }
+
+    return ref;
+}
+
+static HRESULT WINAPI HTMLElementCollection_GetTypeInfoCount(IHTMLElementCollection *iface,
+                                                             UINT *pctinfo)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+    return IDispatchEx_GetTypeInfoCount(&This->dispex.IDispatchEx_iface, pctinfo);
+}
+
+static HRESULT WINAPI HTMLElementCollection_GetTypeInfo(IHTMLElementCollection *iface,
+        UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+    return IDispatchEx_GetTypeInfo(&This->dispex.IDispatchEx_iface, iTInfo, lcid, ppTInfo);
+}
+
+static HRESULT WINAPI HTMLElementCollection_GetIDsOfNames(IHTMLElementCollection *iface,
+        REFIID riid, LPOLESTR *rgszNames, UINT cNames, LCID lcid, DISPID *rgDispId)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+    return IDispatchEx_GetIDsOfNames(&This->dispex.IDispatchEx_iface, riid, rgszNames, cNames,
+            lcid, rgDispId);
+}
+
+static HRESULT WINAPI HTMLElementCollection_Invoke(IHTMLElementCollection *iface,
+        DISPID dispIdMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
+        VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+    return IDispatchEx_Invoke(&This->dispex.IDispatchEx_iface, dispIdMember, riid, lcid,
+            wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
+}
+
+static HRESULT WINAPI HTMLElementCollection_toString(IHTMLElementCollection *iface,
+                                                     BSTR *String)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+    FIXME("(%p)->(%p)\n", This, String);
+    return E_NOTIMPL;
+}
+
+static HRESULT WINAPI HTMLElementCollection_put_length(IHTMLElementCollection *iface,
+                                                       LONG v)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+    FIXME("(%p)->(%d)\n", This, v);
+    return E_NOTIMPL;
+}
+
+static HRESULT WINAPI HTMLElementCollection_get_length(IHTMLElementCollection *iface,
+                                                       LONG *p)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+
+    TRACE("(%p)->(%p)\n", This, p);
+
+    *p = This->len;
+    return S_OK;
+}
+
+static HRESULT WINAPI HTMLElementCollection_get__newEnum(IHTMLElementCollection *iface,
+                                                         IUnknown **p)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+    HTMLElementCollectionEnum *ret;
+
+    TRACE("(%p)->(%p)\n", This, p);
+
+    ret = heap_alloc(sizeof(*ret));
+    if(!ret)
+        return E_OUTOFMEMORY;
+
+    ret->IEnumVARIANT_iface.lpVtbl = &HTMLElementCollectionEnumVtbl;
+    ret->ref = 1;
+    ret->iter = 0;
+
+    IHTMLElementCollection_AddRef(&This->IHTMLElementCollection_iface);
+    ret->col = This;
+
+    *p = (IUnknown*)&ret->IEnumVARIANT_iface;
+    return S_OK;
+}
+
+static BOOL is_elem_id(HTMLElement *elem, LPCWSTR name)
+{
+    BSTR elem_id;
+    HRESULT hres;
+
+    hres = IHTMLElement_get_id(&elem->IHTMLElement_iface, &elem_id);
+    if(FAILED(hres)){
+        WARN("IHTMLElement_get_id failed: 0x%08x\n", hres);
+        return FALSE;
+    }
+
+    if(elem_id && !strcmpW(elem_id, name)) {
+        SysFreeString(elem_id);
+        return TRUE;
+    }
+
+    SysFreeString(elem_id);
+    return FALSE;
+}
+
+static BOOL is_elem_name(HTMLElement *elem, LPCWSTR name)
+{
+    const PRUnichar *str;
+    nsAString nsstr;
+    BOOL ret = FALSE;
+    nsresult nsres;
+
+    static const PRUnichar nameW[] = {'n','a','m','e',0};
+
+    if(!elem->nselem)
+        return FALSE;
+
+    nsAString_Init(&nsstr, NULL);
+    nsIDOMHTMLElement_GetId(elem->nselem, &nsstr);
+    nsAString_GetData(&nsstr, &str);
+    if(!strcmpiW(str, name)) {
+        nsAString_Finish(&nsstr);
+        return TRUE;
+    }
+
+    nsres = get_elem_attr_value(elem->nselem, nameW, &nsstr, &str);
+    if(NS_SUCCEEDED(nsres)) {
+        ret = !strcmpiW(str, name);
+        nsAString_Finish(&nsstr);
+    }
+
+    return ret;
+}
+
+static HRESULT get_item_idx(HTMLElementCollection *This, UINT idx, IDispatch **ret)
+{
+    if(idx < This->len) {
+        *ret = (IDispatch*)This->elems[idx];
+        IDispatch_AddRef(*ret);
+    }
+
+    return S_OK;
+}
+
+static HRESULT WINAPI HTMLElementCollection_item(IHTMLElementCollection *iface,
+        VARIANT name, VARIANT index, IDispatch **pdisp)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+    HRESULT hres = S_OK;
+
+    TRACE("(%p)->(%s %s %p)\n", This, debugstr_variant(&name), debugstr_variant(&index), pdisp);
+
+    *pdisp = NULL;
+
+    switch(V_VT(&name)) {
+    case VT_I4:
+    case VT_INT:
+        if(V_I4(&name) < 0)
+            return E_INVALIDARG;
+        hres = get_item_idx(This, V_I4(&name), pdisp);
+        break;
+
+    case VT_UI4:
+    case VT_UINT:
+        hres = get_item_idx(This, V_UINT(&name), pdisp);
+        break;
+
+    case VT_BSTR: {
+        DWORD i;
+
+        if(V_VT(&index) == VT_I4) {
+            LONG idx = V_I4(&index);
+
+            if(idx < 0)
+                return E_INVALIDARG;
+
+            for(i=0; i<This->len; i++) {
+                if(is_elem_name(This->elems[i], V_BSTR(&name)) && !idx--)
+                    break;
+            }
+
+            if(i != This->len) {
+                *pdisp = (IDispatch*)&This->elems[i]->IHTMLElement_iface;
+                IDispatch_AddRef(*pdisp);
+            }
+        }else {
+            elem_vector_t buf = {NULL, 0, 8};
+
+            buf.buf = heap_alloc(buf.size*sizeof(HTMLElement*));
+
+            for(i=0; i<This->len; i++) {
+                if(is_elem_name(This->elems[i], V_BSTR(&name))) {
+                    node_addref(&This->elems[i]->node);
+                    elem_vector_add(&buf, This->elems[i]);
+                }
+            }
+
+            if(buf.len > 1) {
+                elem_vector_normalize(&buf);
+                *pdisp = (IDispatch*)HTMLElementCollection_Create(buf.buf, buf.len);
+            }else {
+                if(buf.len == 1) {
+                    /* Already AddRef-ed */
+                    *pdisp = (IDispatch*)&buf.buf[0]->IHTMLElement_iface;
+                }
+
+                heap_free(buf.buf);
+            }
+        }
+        break;
+    }
+
+    default:
+        FIXME("Unsupported name %s\n", debugstr_variant(&name));
+        hres = E_NOTIMPL;
+    }
+
+    if(SUCCEEDED(hres))
+        TRACE("returning %p\n", *pdisp);
+    return hres;
+}
+
+static HRESULT WINAPI HTMLElementCollection_tags(IHTMLElementCollection *iface,
+                                                 VARIANT tagName, IDispatch **pdisp)
+{
+    HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
+    DWORD i;
+    nsAString tag_str;
+    const PRUnichar *tag;
+    elem_vector_t buf = {NULL, 0, 8};
+
+    if(V_VT(&tagName) != VT_BSTR) {
+        WARN("Invalid arg\n");
+        return DISP_E_MEMBERNOTFOUND;
+    }
+
+    TRACE("(%p)->(%s %p)\n", This, debugstr_w(V_BSTR(&tagName)), pdisp);
+
+    buf.buf = heap_alloc(buf.size*sizeof(HTMLElement*));
+
+    nsAString_Init(&tag_str, NULL);
+
+    for(i=0; i<This->len; i++) {
+        if(!This->elems[i]->nselem)
+            continue;
+
+        nsIDOMHTMLElement_GetTagName(This->elems[i]->nselem, &tag_str);
+        nsAString_GetData(&tag_str, &tag);
+
+        if(CompareStringW(LOCALE_SYSTEM_DEFAULT, NORM_IGNORECASE, tag, -1,
+                          V_BSTR(&tagName), -1) == CSTR_EQUAL) {
+            node_addref(&This->elems[i]->node);
+            elem_vector_add(&buf, This->elems[i]);
+        }
+    }
+
+    nsAString_Finish(&tag_str);
+    elem_vector_normalize(&buf);
+
+    TRACE("fount %d tags\n", buf.len);
+
+    *pdisp = (IDispatch*)HTMLElementCollection_Create(buf.buf, buf.len);
+    return S_OK;
+}
+
+static const IHTMLElementCollectionVtbl HTMLElementCollectionVtbl = {
+    HTMLElementCollection_QueryInterface,
+    HTMLElementCollection_AddRef,
+    HTMLElementCollection_Release,
+    HTMLElementCollection_GetTypeInfoCount,
+    HTMLElementCollection_GetTypeInfo,
+    HTMLElementCollection_GetIDsOfNames,
+    HTMLElementCollection_Invoke,
+    HTMLElementCollection_toString,
+    HTMLElementCollection_put_length,
+    HTMLElementCollection_get_length,
+    HTMLElementCollection_get__newEnum,
+    HTMLElementCollection_item,
+    HTMLElementCollection_tags
+};
+
+static inline HTMLElementCollection *impl_from_DispatchEx(DispatchEx *iface)
+{
+    return CONTAINING_RECORD(iface, HTMLElementCollection, dispex);
+}
+
+#define DISPID_ELEMCOL_0 MSHTML_DISPID_CUSTOM_MIN
+
+static HRESULT HTMLElementCollection_get_dispid(DispatchEx *dispex, BSTR name, DWORD flags, DISPID *dispid)
+{
+    HTMLElementCollection *This = impl_from_DispatchEx(dispex);
+    WCHAR *ptr;
+    DWORD idx=0;
+
+    if(!*name)
+        return DISP_E_UNKNOWNNAME;
+
+    for(ptr = name; *ptr && isdigitW(*ptr); ptr++)
+        idx = idx*10 + (*ptr-'0');
+
+    if(*ptr) {
+        /* the name contains alpha characters, so search by name & id */
+        for(idx = 0; idx < This->len; ++idx) {
+            if(is_elem_id(This->elems[idx], name) ||
+                    is_elem_name(This->elems[idx], name))
+                break;
+        }
+    }
+
+    if(idx >= This->len)
+        return DISP_E_UNKNOWNNAME;
+
+    *dispid = DISPID_ELEMCOL_0 + idx;
+    TRACE("ret %x\n", *dispid);
+    return S_OK;
+}
+
+static HRESULT HTMLElementCollection_invoke(DispatchEx *dispex, DISPID id, LCID lcid, WORD flags, DISPPARAMS *params,
+        VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
+{
+    HTMLElementCollection *This = impl_from_DispatchEx(dispex);
+    DWORD idx;
+
+    TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, flags, params, res, ei, caller);
+
+    idx = id - DISPID_ELEMCOL_0;
+    if(idx >= This->len)
+        return DISP_E_UNKNOWNNAME;
+
+    switch(flags) {
+    case DISPATCH_PROPERTYGET:
+        V_VT(res) = VT_DISPATCH;
+        V_DISPATCH(res) = (IDispatch*)&This->elems[idx]->IHTMLElement_iface;
+        IHTMLElement_AddRef(&This->elems[idx]->IHTMLElement_iface);
+        break;
+    default:
+        FIXME("unimplemented flags %x\n", flags);
+        return E_NOTIMPL;
+    }
+
+    return S_OK;
+}
+
+static const dispex_static_data_vtbl_t HTMLElementColection_dispex_vtbl = {
+    NULL,
+    HTMLElementCollection_get_dispid,
+    HTMLElementCollection_invoke,
+    NULL
+};
+
+static const tid_t HTMLElementCollection_iface_tids[] = {
+    IHTMLElementCollection_tid,
+    0
+};
+
+static dispex_static_data_t HTMLElementCollection_dispex = {
+    &HTMLElementColection_dispex_vtbl,
+    DispHTMLElementCollection_tid,
+    NULL,
+    HTMLElementCollection_iface_tids
+};
+
+static void create_all_list(HTMLDocumentNode *doc, HTMLDOMNode *elem, elem_vector_t *buf)
+{
+    nsIDOMNodeList *nsnode_list;
+    nsIDOMNode *iter;
+    UINT32 list_len = 0, i;
+    nsresult nsres;
+    HRESULT hres;
+
+    nsres = nsIDOMNode_GetChildNodes(elem->nsnode, &nsnode_list);
+    if(NS_FAILED(nsres)) {
+        ERR("GetChildNodes failed: %08x\n", nsres);
+        return;
+    }
+
+    nsIDOMNodeList_GetLength(nsnode_list, &list_len);
+    if(!list_len)
+        return;
+
+    for(i=0; i<list_len; i++) {
+        nsres = nsIDOMNodeList_Item(nsnode_list, i, &iter);
+        if(NS_FAILED(nsres)) {
+            ERR("Item failed: %08x\n", nsres);
+            continue;
+        }
+
+        if(is_elem_node(iter)) {
+            HTMLDOMNode *node;
+
+            hres = get_node(doc, iter, TRUE, &node);
+            if(FAILED(hres)) {
+                FIXME("get_node failed: %08x\n", hres);
+                continue;
+            }
+
+            elem_vector_add(buf, elem_from_HTMLDOMNode(node));
+            create_all_list(doc, node, buf);
+        }
+    }
+}
+
+IHTMLElementCollection *create_all_collection(HTMLDOMNode *node, BOOL include_root)
+{
+    elem_vector_t buf = {NULL, 0, 8};
+
+    buf.buf = heap_alloc(buf.size*sizeof(HTMLElement*));
+
+    if(include_root) {
+        node_addref(node);
+        elem_vector_add(&buf, elem_from_HTMLDOMNode(node));
+    }
+    create_all_list(node->doc, node, &buf);
+    elem_vector_normalize(&buf);
+
+    return HTMLElementCollection_Create(buf.buf, buf.len);
+}
+
+IHTMLElementCollection *create_collection_from_nodelist(HTMLDocumentNode *doc, nsIDOMNodeList *nslist)
+{
+    UINT32 length = 0, i;
+    HTMLDOMNode *node;
+    elem_vector_t buf;
+    HRESULT hres;
+
+    nsIDOMNodeList_GetLength(nslist, &length);
+
+    buf.len = 0;
+    buf.size = length;
+    if(length) {
+        nsIDOMNode *nsnode;
+
+        buf.buf = heap_alloc(buf.size*sizeof(HTMLElement*));
+
+        for(i=0; i<length; i++) {
+            nsIDOMNodeList_Item(nslist, i, &nsnode);
+            if(is_elem_node(nsnode)) {
+                hres = get_node(doc, nsnode, TRUE, &node);
+                if(FAILED(hres))
+                    continue;
+                buf.buf[buf.len++] = elem_from_HTMLDOMNode(node);
+            }
+            nsIDOMNode_Release(nsnode);
+        }
+
+        elem_vector_normalize(&buf);
+    }else {
+        buf.buf = NULL;
+    }
+
+    return HTMLElementCollection_Create(buf.buf, buf.len);
+}
+
+IHTMLElementCollection *create_collection_from_htmlcol(HTMLDocumentNode *doc, nsIDOMHTMLCollection *nscol)
+{
+    UINT32 length = 0, i;
+    elem_vector_t buf;
+    HTMLDOMNode *node;
+    HRESULT hres = S_OK;
+
+    nsIDOMHTMLCollection_GetLength(nscol, &length);
+
+    buf.len = buf.size = length;
+    if(buf.len) {
+        nsIDOMNode *nsnode;
+
+        buf.buf = heap_alloc(buf.size*sizeof(HTMLElement*));
+
+        for(i=0; i<length; i++) {
+            nsIDOMHTMLCollection_Item(nscol, i, &nsnode);
+            hres = get_node(doc, nsnode, TRUE, &node);
+            nsIDOMNode_Release(nsnode);
+            if(FAILED(hres))
+                break;
+            buf.buf[i] = elem_from_HTMLDOMNode(node);
+        }
+    }else {
+        buf.buf = NULL;
+    }
+
+    if(FAILED(hres)) {
+        heap_free(buf.buf);
+        return NULL;
+    }
+
+    return HTMLElementCollection_Create(buf.buf, buf.len);
+}
+
+HRESULT get_elem_source_index(HTMLElement *elem, LONG *ret)
+{
+    elem_vector_t buf = {NULL, 0, 8};
+    nsIDOMNode *parent_node, *iter;
+    UINT16 parent_type;
+    HTMLDOMNode *node;
+    int i;
+    nsresult nsres;
+    HRESULT hres;
+
+    iter = elem->node.nsnode;
+    nsIDOMNode_AddRef(iter);
+
+    /* Find document or document fragment parent. */
+    while(1) {
+        nsres = nsIDOMNode_GetParentNode(iter, &parent_node);
+        nsIDOMNode_Release(iter);
+        assert(nsres == NS_OK);
+        if(!parent_node)
+            break;
+
+        nsres = nsIDOMNode_GetNodeType(parent_node, &parent_type);
+        assert(nsres == NS_OK);
+
+        if(parent_type != ELEMENT_NODE) {
+            if(parent_type != DOCUMENT_NODE && parent_type != DOCUMENT_FRAGMENT_NODE)
+                FIXME("Unexpected parent_type %d\n", parent_type);
+            break;
+        }
+
+        iter = parent_node;
+    }
+
+    if(!parent_node) {
+        *ret = -1;
+        return S_OK;
+    }
+
+    hres = get_node(elem->node.doc, parent_node, TRUE, &node);
+    nsIDOMNode_Release(parent_node);
+    if(FAILED(hres))
+        return hres;
+
+
+    /* Create all children collection and find the element in it.
+     * This could be optimized if we ever find the reason. */
+    buf.buf = heap_alloc(buf.size*sizeof(*buf.buf));
+    if(!buf.buf) {
+        IHTMLDOMNode_Release(&node->IHTMLDOMNode_iface);
+        return E_OUTOFMEMORY;
+    }
+
+    create_all_list(elem->node.doc, node, &buf);
+
+    for(i=0; i < buf.len; i++) {
+        if(buf.buf[i] == elem)
+            break;
+    }
+    IHTMLDOMNode_Release(&node->IHTMLDOMNode_iface);
+    heap_free(buf.buf);
+    if(i == buf.len) {
+        FIXME("The element is not in parent's child list?\n");
+        return E_UNEXPECTED;
+    }
+
+    *ret = i;
+    return S_OK;
+}
+
+static IHTMLElementCollection *HTMLElementCollection_Create(HTMLElement **elems, DWORD len)
+{
+    HTMLElementCollection *ret = heap_alloc_zero(sizeof(HTMLElementCollection));
+
+    if (!ret)
+        return NULL;
+
+    ret->IHTMLElementCollection_iface.lpVtbl = &HTMLElementCollectionVtbl;
+    ret->ref = 1;
+    ret->elems = elems;
+    ret->len = len;
+
+    init_dispex(&ret->dispex, (IUnknown*)&ret->IHTMLElementCollection_iface,
+            &HTMLElementCollection_dispex);
+
+    TRACE("ret=%p len=%d\n", ret, len);
+
+    return &ret->IHTMLElementCollection_iface;
+}