| /* |
| * 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 <WebCore/COMPtr.h> |
| #include <JavaScriptCore/Assertions.h> |
| #include <JavaScriptCore/JavaScriptCore.h> |
| #include <WebKit/IWebFramePrivate.h> |
| #include <WebKit/IWebViewPrivate.h> |
| #include <stdio.h> |
| |
| 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<IWebFrameLoadDelegate*>(this); |
| else if (IsEqualGUID(riid, IID_IWebFrameLoadDelegate)) |
| *ppvObject = static_cast<IWebFrameLoadDelegate*>(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<IWebViewPrivate> 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; |
| } |