/* * Copyright 2007 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 */ #define COBJMACROS #define CONST_VTABLE #include #include #include #include "windef.h" #include "winbase.h" #include "ole2.h" #include "mshtml.h" #include "docobj.h" static const char doc_blank[] = ""; static const char doc_str1[] = "test"; static const char doc_str2[] = "test abc 123
it's text
"; static const WCHAR noneW[] = {'N','o','n','e',0}; static WCHAR characterW[] = {'c','h','a','r','a','c','t','e','r',0}; static WCHAR wordW[] = {'w','o','r','d',0}; static const char *dbgstr_w(LPCWSTR str) { static char buf[512]; if(!str) return "(null)"; WideCharToMultiByte(CP_ACP, 0, str, -1, buf, sizeof(buf), NULL, NULL); return buf; } static int strcmp_wa(LPCWSTR strw, const char *stra) { WCHAR buf[512]; MultiByteToWideChar(CP_ACP, 0, stra, -1, buf, sizeof(buf)/sizeof(WCHAR)); return lstrcmpW(strw, buf); } static IHTMLDocument2 *create_document(void) { IHTMLDocument2 *doc; HRESULT hres; hres = CoCreateInstance(&CLSID_HTMLDocument, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER, &IID_IHTMLDocument2, (void**)&doc); ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres); return doc; } #define test_node_name(u,n) _test_node_name(__LINE__,u,n) static void _test_node_name(unsigned line, IUnknown *unk, const char *exname) { IHTMLDOMNode *node; BSTR name; HRESULT hres; hres = IUnknown_QueryInterface(unk, &IID_IHTMLDOMNode, (void**)&node); ok_(__FILE__, line) (hres == S_OK, "QueryInterface(IID_IHTMLNode) failed: %08x\n", hres); hres = IHTMLDOMNode_get_nodeName(node, &name); IHTMLDOMNode_Release(node); ok_(__FILE__, line) (hres == S_OK, "get_nodeName failed: %08x\n", hres); ok_(__FILE__, line) (!strcmp_wa(name, exname), "got name: %s, expected %s\n", dbgstr_w(name), exname); SysFreeString(name); } #define test_elem_tag(u,n) _test_elem_tag(__LINE__,u,n) static void _test_elem_tag(unsigned line, IUnknown *unk, const char *extag) { IHTMLElement *elem; BSTR tag; HRESULT hres; hres = IUnknown_QueryInterface(unk, &IID_IHTMLElement, (void**)&elem); ok_(__FILE__, line) (hres == S_OK, "QueryInterface(IID_IHTMLElement) failed: %08x\n", hres); hres = IHTMLElement_get_tagName(elem, &tag); IHTMLElement_Release(elem); ok_(__FILE__, line) (hres == S_OK, "get_tagName failed: %08x\n", hres); ok_(__FILE__, line) (!strcmp_wa(tag, extag), "got tag: %s, expected %s\n", dbgstr_w(tag), extag); SysFreeString(tag); } static void test_doc_elem(IHTMLDocument2 *doc) { IHTMLElement *elem; IHTMLDocument3 *doc3; HRESULT hres; hres = IHTMLDocument2_QueryInterface(doc, &IID_IHTMLDocument3, (void**)&doc3); ok(hres == S_OK, "QueryInterface(IID_IHTMLDocument3) failed: %08x\n", hres); hres = IHTMLDocument3_get_documentElement(doc3, &elem); IHTMLDocument3_Release(doc3); ok(hres == S_OK, "get_documentElement failed: %08x\n", hres); test_node_name((IUnknown*)elem, "HTML"); test_elem_tag((IUnknown*)elem, "HTML"); IHTMLElement_Release(elem); } #define test_range_text(r,t) _test_range_text(__LINE__,r,t) static void _test_range_text(unsigned line, IHTMLTxtRange *range, const char *extext) { BSTR text; HRESULT hres; hres = IHTMLTxtRange_get_text(range, &text); ok_(__FILE__, line) (hres == S_OK, "get_text failed: %08x\n", hres); if(extext) { ok_(__FILE__, line) (text != NULL, "text == NULL\n"); ok_(__FILE__, line) (!strcmp_wa(text, extext), "text=\"%s\", expected \"%s\"\n", dbgstr_w(text), extext); }else { ok_(__FILE__, line) (text == NULL, "text=\"%s\", expected NULL\n", dbgstr_w(text)); } SysFreeString(text); } #define test_range_collapse(r,b) _test_range_collapse(__LINE__,r,b) static void _test_range_collapse(unsigned line, IHTMLTxtRange *range, BOOL b) { HRESULT hres; hres = IHTMLTxtRange_collapse(range, b); ok_(__FILE__, line) (hres == S_OK, "collapse failed: %08x\n", hres); _test_range_text(line, range, NULL); } #define test_range_expand(r,u,b,t) _test_range_expand(__LINE__,r,u,b,t) static void _test_range_expand(unsigned line, IHTMLTxtRange *range, LPWSTR unit, VARIANT_BOOL exb, const char *extext) { VARIANT_BOOL b = 0xe0e0; HRESULT hres; hres = IHTMLTxtRange_expand(range, unit, &b); ok_(__FILE__,line) (hres == S_OK, "expand failed: %08x\n", hres); ok_(__FILE__,line) (b == exb, "b=%x, expected %x\n", b, exb); _test_range_text(line, range, extext); } #define test_range_move(r,u,c,e) _test_range_move(__LINE__,r,u,c,e) static void _test_range_move(unsigned line, IHTMLTxtRange *range, LPWSTR unit, long cnt, long excnt) { long c = 0xdeadbeef; HRESULT hres; hres = IHTMLTxtRange_move(range, unit, cnt, &c); ok_(__FILE__,line) (hres == S_OK, "move failed: %08x\n", hres); ok_(__FILE__,line) (c == excnt, "count=%ld, expected %ld\n", c, excnt); _test_range_text(line, range, NULL); } #define test_range_moveend(r,u,c,e) _test_range_moveend(__LINE__,r,u,c,e) static void _test_range_moveend(unsigned line, IHTMLTxtRange *range, LPWSTR unit, long cnt, long excnt) { long c = 0xdeadbeef; HRESULT hres; hres = IHTMLTxtRange_moveEnd(range, unit, cnt, &c); ok_(__FILE__,line) (hres == S_OK, "move failed: %08x\n", hres); ok_(__FILE__,line) (c == excnt, "count=%ld, expected %ld\n", c, excnt); } #define test_range_put_text(r,t) _test_range_put_text(__LINE__,r,t) static void _test_range_put_text(unsigned line, IHTMLTxtRange *range, LPCWSTR text) { HRESULT hres; hres = IHTMLTxtRange_put_text(range, (BSTR)text); ok_(__FILE__,line) (hres == S_OK, "put_text failed: %08x\n", hres); _test_range_text(line, range, NULL); } #define test_range_inrange(r1,r2,b) _test_range_inrange(__LINE__,r1,r2,b) static void _test_range_inrange(unsigned line, IHTMLTxtRange *range1, IHTMLTxtRange *range2, VARIANT_BOOL exb) { VARIANT_BOOL b; HRESULT hres; b = 0xe0e0; hres = IHTMLTxtRange_inRange(range1, range2, &b); ok_(__FILE__,line) (hres == S_OK, "(1->2) isEqual failed: %08x\n", hres); ok_(__FILE__,line) (b == exb, "(1->2) b=%x, expected %x\n", b, exb); } #define test_range_isequal(r1,r2,b) _test_range_isequal(__LINE__,r1,r2,b) static void _test_range_isequal(unsigned line, IHTMLTxtRange *range1, IHTMLTxtRange *range2, VARIANT_BOOL exb) { VARIANT_BOOL b; HRESULT hres; b = 0xe0e0; hres = IHTMLTxtRange_isEqual(range1, range2, &b); ok_(__FILE__,line) (hres == S_OK, "(1->2) isEqual failed: %08x\n", hres); ok_(__FILE__,line) (b == exb, "(1->2) b=%x, expected %x\n", b, exb); b = 0xe0e0; hres = IHTMLTxtRange_isEqual(range2, range1, &b); ok_(__FILE__,line) (hres == S_OK, "(2->1) isEqual failed: %08x\n", hres); ok_(__FILE__,line) (b == exb, "(2->1) b=%x, expected %x\n", b, exb); if(exb) { test_range_inrange(range1, range2, VARIANT_TRUE); test_range_inrange(range2, range1, VARIANT_TRUE); } } static void test_txtrange(IHTMLDocument2 *doc) { IHTMLElement *elem; IHTMLBodyElement *body; IHTMLTxtRange *body_range, *range, *range2; HRESULT hres; hres = IHTMLDocument2_get_body(doc, &elem); ok(hres == S_OK, "get_body failed: %08x\n", hres); hres = IHTMLElement_QueryInterface(elem, &IID_IHTMLBodyElement, (void**)&body); IHTMLElement_Release(elem); hres = IHTMLBodyElement_createTextRange(body, &body_range); IHTMLBodyElement_Release(body); ok(hres == S_OK, "createTextRange failed: %08x\n", hres); test_range_text(body_range, "test abc 123\r\nit's text"); hres = IHTMLTxtRange_duplicate(body_range, &range); ok(hres == S_OK, "duplicate failed: %08x\n", hres); hres = IHTMLTxtRange_duplicate(body_range, &range2); ok(hres == S_OK, "duplicate failed: %08x\n", hres); test_range_isequal(range, range2, VARIANT_TRUE); test_range_text(range, "test abc 123\r\nit's text"); test_range_text(body_range, "test abc 123\r\nit's text"); test_range_collapse(range, TRUE); test_range_isequal(range, range2, VARIANT_FALSE); test_range_inrange(range, range2, VARIANT_FALSE); test_range_inrange(range2, range, VARIANT_TRUE); IHTMLTxtRange_Release(range2); test_range_expand(range, wordW, VARIANT_TRUE, "test "); test_range_expand(range, wordW, VARIANT_FALSE, "test "); test_range_move(range, characterW, 2, 2); test_range_expand(range, wordW, VARIANT_TRUE, "test "); test_range_collapse(range, FALSE); test_range_expand(range, wordW, VARIANT_TRUE, "abc "); test_range_collapse(range, FALSE); test_range_expand(range, wordW, VARIANT_TRUE, "123"); test_range_expand(range, wordW, VARIANT_FALSE, "123"); test_range_move(range, characterW, 2, 2); test_range_expand(range, wordW, VARIANT_TRUE, "123"); IHTMLTxtRange_Release(range); hres = IHTMLTxtRange_duplicate(body_range, &range); ok(hres == S_OK, "duplicate failed: %08x\n", hres); test_range_text(range, "test abc 123\r\nit's text"); test_range_move(range, characterW, 3, 3); test_range_moveend(range, characterW, 1, 1); test_range_text(range, "t"); test_range_moveend(range, characterW, 3, 3); test_range_text(range, "t ab"); test_range_moveend(range, characterW, -2, -2); test_range_text(range, "t "); test_range_move(range, characterW, 6, 6); test_range_moveend(range, characterW, 3, 3); test_range_text(range, "123"); test_range_moveend(range, characterW, 2, 2); test_range_text(range, "123\r\ni"); IHTMLTxtRange_Release(range); hres = IHTMLTxtRange_duplicate(body_range, &range); ok(hres == S_OK, "duplicate failed: %08x\n", hres); test_range_move(range, wordW, 1, 1); test_range_moveend(range, characterW, 2, 2); test_range_text(range, "ab"); test_range_move(range, characterW, -2, -2); test_range_moveend(range, characterW, 2, 2); test_range_text(range, "t "); test_range_move(range, wordW, 3, 3); test_range_move(range, wordW, -2, -2); test_range_moveend(range, characterW, 2, 2); test_range_text(range, "ab"); test_range_move(range, characterW, -6, -5); test_range_moveend(range, characterW, -1, 0); test_range_moveend(range, characterW, -6, 0); test_range_move(range, characterW, 2, 2); test_range_moveend(range, characterW, 2, 2); test_range_text(range, "st"); test_range_moveend(range, characterW, -6, -4); test_range_moveend(range, characterW, 2, 2); IHTMLTxtRange_Release(range); hres = IHTMLTxtRange_duplicate(body_range, &range); ok(hres == S_OK, "duplicate failed: %08x\n", hres); test_range_move(range, wordW, 2, 2); test_range_moveend(range, characterW, 2, 2); test_range_text(range, "12"); test_range_move(range, characterW, 15, 14); test_range_move(range, characterW, -2, -2); test_range_moveend(range, characterW, 3, 2); test_range_text(range, "t"); test_range_moveend(range, characterW, -1, -1); test_range_text(range, "t"); test_range_expand(range, wordW, VARIANT_TRUE, "text"); test_range_move(range, characterW, -2, -2); test_range_moveend(range, characterW, 2, 2); test_range_text(range, "s "); test_range_move(range, characterW, 100, 7); test_range_move(range, wordW, 1, 0); test_range_move(range, characterW, -2, -2); test_range_moveend(range, characterW, 3, 2); test_range_text(range, "t"); IHTMLTxtRange_Release(range); hres = IHTMLTxtRange_duplicate(body_range, &range); ok(hres == S_OK, "duplicate failed: %08x\n", hres); test_range_collapse(range, TRUE); test_range_expand(range, wordW, VARIANT_TRUE, "test "); test_range_put_text(range, wordW); test_range_text(body_range, "wordabc 123\r\nit's text"); IHTMLTxtRange_Release(range); IHTMLTxtRange_Release(body_range); } static void test_compatmode(IHTMLDocument2 *doc) { IHTMLDocument5 *doc5; BSTR mode; HRESULT hres; hres = IHTMLDocument2_QueryInterface(doc, &IID_IHTMLDocument5, (void**)&doc5); ok(hres == S_OK, "Could not get IHTMLDocument5 interface: %08x\n", hres); if(FAILED(hres)) return; hres = IHTMLDocument5_get_compatMode(doc5, &mode); IHTMLDocument5_Release(doc5); ok(hres == S_OK, "get_compatMode failed: %08x\n", hres); ok(!strcmp_wa(mode, "BackCompat"), "compatMode=%s\n", dbgstr_w(mode)); SysFreeString(mode); } static void test_default_style(IHTMLStyle *style) { VARIANT_BOOL b; VARIANT v; BSTR str; HRESULT hres; str = (void*)0xdeadbeef; hres = IHTMLStyle_get_fontFamily(style, &str); ok(hres == S_OK, "get_fontFamily failed: %08x\n", hres); ok(!str, "fontFamily = %s\n", dbgstr_w(str)); str = (void*)0xdeadbeef; hres = IHTMLStyle_get_fontWeight(style, &str); ok(hres == S_OK, "get_fontWeight failed: %08x\n", hres); ok(!str, "fontWeight = %s\n", dbgstr_w(str)); V_VT(&v) = VT_NULL; hres = IHTMLStyle_get_fontSize(style, &v); ok(hres == S_OK, "get_fontSize failed: %08x\n", hres); ok(V_VT(&v) == VT_BSTR, "V_VT(fontSize) = %d\n", V_VT(&v)); ok(!V_BSTR(&v), "V_BSTR(fontSize) = %s\n", dbgstr_w(V_BSTR(&v))); V_VT(&v) = VT_NULL; hres = IHTMLStyle_get_color(style, &v); ok(hres == S_OK, "get_color failed: %08x\n", hres); ok(V_VT(&v) == VT_BSTR, "V_VT(color) = %d\n", V_VT(&v)); ok(!V_BSTR(&v), "V_BSTR(color) = %s\n", dbgstr_w(V_BSTR(&v))); b = 0xfefe; hres = IHTMLStyle_get_textDecorationUnderline(style, &b); ok(hres == S_OK, "get_textDecorationUnderline failed: %08x\n", hres); ok(b == VARIANT_FALSE, "textDecorationUnderline = %x\n", b); b = 0xfefe; hres = IHTMLStyle_get_textDecorationLineThrough(style, &b); ok(hres == S_OK, "get_textDecorationLineThrough failed: %08x\n", hres); ok(b == VARIANT_FALSE, "textDecorationLineThrough = %x\n", b); } static void test_default_selection(IHTMLDocument2 *doc) { IHTMLSelectionObject *selection; IHTMLTxtRange *range; IDispatch *disp; BSTR str; HRESULT hres; hres = IHTMLDocument2_get_selection(doc, &selection); ok(hres == S_OK, "get_selection failed: %08x\n", hres); hres = IHTMLSelectionObject_get_type(selection, &str); ok(hres == S_OK, "get_type failed: %08x\n", hres); ok(!lstrcmpW(str, noneW), "type = %s\n", dbgstr_w(str)); SysFreeString(str); hres = IHTMLSelectionObject_createRange(selection, &disp); IHTMLSelectionObject_Release(selection); ok(hres == S_OK, "createRange failed: %08x\n", hres); hres = IDispatch_QueryInterface(disp, &IID_IHTMLTxtRange, (void**)&range); IDispatch_Release(disp); ok(hres == S_OK, "Could not get IHTMLTxtRange interface: %08x\n", hres); test_range_text(range, NULL); IHTMLTxtRange_Release(range); } static void test_defaults(IHTMLDocument2 *doc) { IHTMLStyleSheetsCollection *stylesheetcol; IHTMLElement *elem; IHTMLStyle *style; long l; HRESULT hres; hres = IHTMLDocument2_get_body(doc, &elem); ok(hres == S_OK, "get_body failed: %08x\n", hres); hres = IHTMLElement_get_style(elem, &style); IHTMLElement_Release(elem); ok(hres == S_OK, "get_style failed: %08x\n", hres); test_default_style(style); test_compatmode(doc); IHTMLStyle_Release(style); hres = IHTMLDocument2_get_styleSheets(doc, &stylesheetcol); ok(hres == S_OK, "get_styleSheets failed: %08x\n", hres); l = 0xdeadbeef; hres = IHTMLStyleSheetsCollection_get_length(stylesheetcol, &l); ok(hres == S_OK, "get_length failed: %08x\n", hres); ok(l == 0, "length = %ld\n", l); IHTMLStyleSheetsCollection_Release(stylesheetcol); test_default_selection(doc); } static IHTMLDocument2 *notif_doc; static BOOL doc_complete; static HRESULT WINAPI PropertyNotifySink_QueryInterface(IPropertyNotifySink *iface, REFIID riid, void**ppv) { if(IsEqualGUID(&IID_IPropertyNotifySink, riid)) { *ppv = iface; return S_OK; } ok(0, "unexpected call\n"); return E_NOINTERFACE; } static ULONG WINAPI PropertyNotifySink_AddRef(IPropertyNotifySink *iface) { return 2; } static ULONG WINAPI PropertyNotifySink_Release(IPropertyNotifySink *iface) { return 1; } static HRESULT WINAPI PropertyNotifySink_OnChanged(IPropertyNotifySink *iface, DISPID dispID) { if(dispID == DISPID_READYSTATE){ BSTR state; HRESULT hres; static const WCHAR completeW[] = {'c','o','m','p','l','e','t','e',0}; hres = IHTMLDocument2_get_readyState(notif_doc, &state); ok(hres == S_OK, "get_readyState failed: %08x\n", hres); if(!lstrcmpW(state, completeW)) doc_complete = TRUE; SysFreeString(state); } return S_OK; } static HRESULT WINAPI PropertyNotifySink_OnRequestEdit(IPropertyNotifySink *iface, DISPID dispID) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static IPropertyNotifySinkVtbl PropertyNotifySinkVtbl = { PropertyNotifySink_QueryInterface, PropertyNotifySink_AddRef, PropertyNotifySink_Release, PropertyNotifySink_OnChanged, PropertyNotifySink_OnRequestEdit }; static IPropertyNotifySink PropertyNotifySink = { &PropertyNotifySinkVtbl }; static IHTMLDocument2 *create_doc_with_string(const char *str) { IPersistStreamInit *init; IStream *stream; IHTMLDocument2 *doc; HGLOBAL mem; SIZE_T len; notif_doc = doc = create_document(); if(!doc) return NULL; doc_complete = FALSE; len = strlen(str); mem = GlobalAlloc(0, len); memcpy(mem, str, len); CreateStreamOnHGlobal(mem, TRUE, &stream); IHTMLDocument2_QueryInterface(doc, &IID_IPersistStreamInit, (void**)&init); IPersistStreamInit_Load(init, stream); IPersistStreamInit_Release(init); IStream_Release(stream); return doc; } static void do_advise(IUnknown *unk, REFIID riid, IUnknown *unk_advise) { IConnectionPointContainer *container; IConnectionPoint *cp; DWORD cookie; HRESULT hres; hres = IUnknown_QueryInterface(unk, &IID_IConnectionPointContainer, (void**)&container); ok(hres == S_OK, "QueryInterface(IID_IConnectionPointContainer) failed: %08x\n", hres); hres = IConnectionPointContainer_FindConnectionPoint(container, riid, &cp); IConnectionPointContainer_Release(container); ok(hres == S_OK, "FindConnectionPoint failed: %08x\n", hres); hres = IConnectionPoint_Advise(cp, unk_advise, &cookie); IConnectionPoint_Release(cp); ok(hres == S_OK, "Advise failed: %08x\n", hres); } typedef void (*domtest_t)(IHTMLDocument2*); static void run_domtest(const char *str, domtest_t test) { IHTMLDocument2 *doc; IHTMLElement *body = NULL; ULONG ref; MSG msg; HRESULT hres; doc = create_doc_with_string(str); do_advise((IUnknown*)doc, &IID_IPropertyNotifySink, (IUnknown*)&PropertyNotifySink); while(!doc_complete && GetMessage(&msg, NULL, 0, 0)) { TranslateMessage(&msg); DispatchMessage(&msg); } hres = IHTMLDocument2_get_body(doc, &body); ok(hres == S_OK, "get_body failed: %08x\n", hres); if(body) { IHTMLElement_Release(body); test(doc); }else { skip("Could not get document body. Assuming no Gecko installed.\n"); } ref = IHTMLDocument2_Release(doc); ok(!ref, "ref = %d\n", ref); } static void gecko_installer_workaround(BOOL disable) { HKEY hkey; DWORD res; static BOOL has_url = FALSE; static char url[2048]; if(!disable && !has_url) return; res = RegOpenKey(HKEY_CURRENT_USER, "Software\\Wine\\MSHTML", &hkey); if(res != ERROR_SUCCESS) return; if(disable) { DWORD type, size = sizeof(url); res = RegQueryValueEx(hkey, "GeckoUrl", NULL, &type, (PVOID)url, &size); if(res == ERROR_SUCCESS && type == REG_SZ) has_url = TRUE; RegDeleteValue(hkey, "GeckoUrl"); }else { RegSetValueEx(hkey, "GeckoUrl", 0, REG_SZ, (PVOID)url, lstrlenA(url)+1); } RegCloseKey(hkey); } START_TEST(dom) { gecko_installer_workaround(TRUE); CoInitialize(NULL); run_domtest(doc_str1, test_doc_elem); run_domtest(doc_str2, test_txtrange); run_domtest(doc_blank, test_defaults); CoUninitialize(); gecko_installer_workaround(FALSE); }