diff -r 000000000000 -r dd21522fd290 webengine/osswebengine/WebKitTools/DumpRenderTree/win/FrameLoaderDelegate.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/webengine/osswebengine/WebKitTools/DumpRenderTree/win/FrameLoaderDelegate.cpp Mon Mar 30 12:54:55 2009 +0300 @@ -0,0 +1,198 @@ +/* + * Copyright (C) 2005, 2006, 2007 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 "DumpRenderTree.h" +#include "FrameLoaderDelegate.h" + +#include "EventSender.h" +#include "GCController.h" +#include "LayoutTestController.h" +#include "WorkQueueItem.h" +#include "WorkQueue.h" +#include +#include +#include +#include +#include +#include + +static FrameLoadDelegate* g_delegateWaitingOnTimer; + +FrameLoadDelegate::FrameLoadDelegate() + : m_refCount(1) + , m_gcController(new GCController) +{ +} + +FrameLoadDelegate::~FrameLoadDelegate() +{ +} + +HRESULT STDMETHODCALLTYPE FrameLoadDelegate::QueryInterface(REFIID riid, void** ppvObject) +{ + *ppvObject = 0; + if (IsEqualGUID(riid, IID_IUnknown)) + *ppvObject = static_cast(this); + else if (IsEqualGUID(riid, IID_IWebFrameLoadDelegate)) + *ppvObject = static_cast(this); + else + return E_NOINTERFACE; + + AddRef(); + return S_OK; +} + +ULONG STDMETHODCALLTYPE FrameLoadDelegate::AddRef(void) +{ + return ++m_refCount; +} + +ULONG STDMETHODCALLTYPE FrameLoadDelegate::Release(void) +{ + ULONG newRef = --m_refCount; + if (!newRef) + delete(this); + + return newRef; +} + + +HRESULT STDMETHODCALLTYPE FrameLoadDelegate::didStartProvisionalLoadForFrame( + /* [in] */ IWebView* webView, + /* [in] */ IWebFrame* frame) +{ + // Make sure we only set this once per test. If it gets cleared, and then set again, we might + // end up doing two dumps for one test. + if (!topLoadingFrame && !done) + topLoadingFrame = frame; + + return S_OK; +} + +HRESULT STDMETHODCALLTYPE FrameLoadDelegate::didCommitLoadForFrame( + /* [in] */ IWebView *webView, + /* [in] */ IWebFrame *frame) +{ + COMPtr webViewPrivate; + HRESULT hr = webView->QueryInterface(&webViewPrivate); + if (FAILED(hr)) + return hr; + webViewPrivate->updateActiveState(); + return S_OK; +} + +HRESULT STDMETHODCALLTYPE FrameLoadDelegate::didReceiveTitle( + /* [in] */ IWebView *webView, + /* [in] */ BSTR title, + /* [in] */ IWebFrame *frame) +{ + if (::layoutTestController->dumpTitleChanges() && !done) + printf("TITLE CHANGED: %S\n", title ? title : L""); + return S_OK; +} + +void FrameLoadDelegate::processWork() +{ + // quit doing work once a load is in progress + while (!topLoadingFrame && WorkQueue::shared()->count()) { + WorkQueueItem* item = WorkQueue::shared()->dequeue(); + ASSERT(item); + item->invoke(); + } + + // if we didn't start a new load, then we finished all the commands, so we're ready to dump state + if (!topLoadingFrame && !::layoutTestController->waitToDump()) + dump(); +} + +static void CALLBACK processWorkTimer(HWND, UINT, UINT_PTR id, DWORD) +{ + ::KillTimer(0, id); + FrameLoadDelegate* d = g_delegateWaitingOnTimer; + g_delegateWaitingOnTimer = 0; + d->processWork(); +} + +void FrameLoadDelegate::locationChangeDone(IWebError*, IWebFrame* frame) +{ + if (frame != topLoadingFrame) + return; + + topLoadingFrame = 0; + WorkQueue::shared()->setFrozen(true); + + if (::layoutTestController->waitToDump()) + return; + + if (WorkQueue::shared()->count()) { + ASSERT(!g_delegateWaitingOnTimer); + g_delegateWaitingOnTimer = this; + ::SetTimer(0, 0, 0, processWorkTimer); + return; + } + + dump(); +} + +HRESULT STDMETHODCALLTYPE FrameLoadDelegate::didFinishLoadForFrame( + /* [in] */ IWebView* webView, + /* [in] */ IWebFrame* frame) +{ + locationChangeDone(0, frame); + return S_OK; +} + +HRESULT STDMETHODCALLTYPE FrameLoadDelegate::didFailLoadWithError( + /* [in] */ IWebView* webView, + /* [in] */ IWebError* error, + /* [in] */ IWebFrame* forFrame) +{ + locationChangeDone(error, forFrame); + return S_OK; +} + +HRESULT STDMETHODCALLTYPE FrameLoadDelegate::windowScriptObjectAvailable( + /* [in] */ IWebView *sender, + /* [in] */ JSContextRef context, + /* [in] */ JSObjectRef windowObject) +{ + JSValueRef exception = 0; + + ::layoutTestController->makeWindowObject(context, windowObject, &exception); + ASSERT(!exception); + + m_gcController->makeWindowObject(context, windowObject, &exception); + ASSERT(!exception); + + JSStringRef eventSenderStr = JSStringCreateWithUTF8CString("eventSender"); + JSValueRef eventSender = makeEventSender(context); + JSObjectSetProperty(context, windowObject, eventSenderStr, eventSender, kJSPropertyAttributeReadOnly | kJSPropertyAttributeDontDelete, 0); + JSStringRelease(eventSenderStr); + + return S_OK; +}