diff -r 000000000000 -r 4f2f89ce4247 WebKitTools/DumpRenderTree/win/WorkQueueItemWin.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/WebKitTools/DumpRenderTree/win/WorkQueueItemWin.cpp Fri Sep 17 09:02:29 2010 +0300 @@ -0,0 +1,158 @@ +/* + * Copyright (C) 2007, 2009 Apple Inc. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of + * its contributors may be used to endorse or promote products derived + * from this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY + * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY + * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES + * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; + * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND + * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF + * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include "config.h" +#include "WorkQueueItem.h" + +#include "DumpRenderTree.h" +#include +#include +#include +#include +#include +#include +#include +#include + +using std::wstring; + +static wstring jsStringRefToWString(JSStringRef jsStr) +{ + size_t length = JSStringGetLength(jsStr); + Vector buffer(length + 1); + memcpy(buffer.data(), JSStringGetCharactersPtr(jsStr), length * sizeof(WCHAR)); + buffer[length] = '\0'; + + return buffer.data(); +} + +bool LoadItem::invoke() const +{ + wstring targetString = jsStringRefToWString(m_target.get()); + + COMPtr targetFrame; + if (targetString.empty()) + targetFrame = frame; + else { + BSTR targetBSTR = SysAllocString(targetString.c_str()); + bool failed = FAILED(frame->findFrameNamed(targetBSTR, &targetFrame)); + SysFreeString(targetBSTR); + if (failed) + return false; + } + + COMPtr request; + if (FAILED(WebKitCreateInstance(CLSID_WebURLRequest, 0, IID_IWebURLRequest, (void**)&request))) + return false; + + wstring urlString = jsStringRefToWString(m_url.get()); + BSTR urlBSTR = SysAllocString(urlString.c_str()); + bool failed = FAILED(request->initWithURL(urlBSTR, WebURLRequestUseProtocolCachePolicy, 60)); + SysFreeString(urlBSTR); + if (failed) + return false; + + targetFrame->loadRequest(request.get()); + return true; +} + +bool LoadHTMLStringItem::invoke() const +{ + wstring content = jsStringRefToWString(m_content.get()); + wstring baseURL = jsStringRefToWString(m_baseURL.get()); + + BSTR contentBSTR = SysAllocString(content.c_str()); + BSTR baseURLBSTR = SysAllocString(baseURL.c_str()); + + frame->loadHTMLString(contentBSTR, baseURLBSTR); + + SysFreeString(contentBSTR); + SysFreeString(baseURLBSTR); + + return true; +} + +bool ReloadItem::invoke() const +{ + COMPtr webView; + if (FAILED(frame->webView(&webView))) + return false; + + COMPtr webActions; + if (FAILED(webView->QueryInterface(&webActions))) + return false; + + webActions->reload(0); + return true; +} + +bool ScriptItem::invoke() const +{ + COMPtr webView; + if (FAILED(frame->webView(&webView))) + return false; + + wstring scriptString = jsStringRefToWString(m_script.get()); + + BSTR result; + BSTR scriptBSTR = SysAllocString(scriptString.c_str()); + webView->stringByEvaluatingJavaScriptFromString(scriptBSTR, &result); + SysFreeString(result); + SysFreeString(scriptBSTR); + + return true; +} + +bool BackForwardItem::invoke() const +{ + COMPtr webView; + if (FAILED(frame->webView(&webView))) + return false; + + BOOL result; + if (m_howFar == 1) { + webView->goForward(&result); + return true; + } + + if (m_howFar == -1) { + webView->goBack(&result); + return true; + } + + COMPtr bfList; + if (FAILED(webView->backForwardList(&bfList))) + return false; + + COMPtr item; + if (FAILED(bfList->itemAtIndex(m_howFar, &item))) + return false; + + webView->goToBackForwardItem(item.get(), &result); + return true; +}