WebKitTools/Drosera/win/ServerConnection.cpp

   1 /*
   2  * Copyright (C) 2007 Apple Inc.  All rights reserved.
   3  * Copyright (C) 2006, 2007 Vladimir Olexa (vladimir.olexa@gmail.com)
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  *
   9  * 1.  Redistributions of source code must retain the above copyright
  10  *     notice, this list of conditions and the following disclaimer.
  11  * 2.  Redistributions in binary form must reproduce the above copyright
  12  *     notice, this list of conditions and the following disclaimer in the
  13  *     documentation and/or other materials provided with the distribution.
  14  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
  15  *     its contributors may be used to endorse or promote products derived
  16  *     from this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
  19  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  20  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  21  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  22  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  23  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  24  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  25  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  */
  29
  30 #include "config.h"
  31 #include "ServerConnection.h"
  32
  33 #include "DebuggerDocument.h"
  34
  35 #include <JavaScriptCore/JSContextRef.h>
  36 #include <JavaScriptCore/JSRetainPtr.h>
  37 #include <JavaScriptCore/JSStringRefBSTR.h>
  38 #include <JavaScriptCore/RetainPtr.h>
  39 #include <WebKit/IWebScriptCallFrame.h>
  40 #include <WebKit/IWebScriptDebugServer.h>
  41 #include <WebKit/WebKit.h>
  42
  43 ServerConnection::ServerConnection()
  44     : m_globalContext(0)
  45 {
  46     HRESULT serverCreated = CoCreateInstance(CLSID_WebScriptDebugServer, 0, CLSCTX_LOCAL_SERVER, IID_IWebScriptDebugServer, (void**)&m_server);
  47     if (!FAILED(serverCreated))
  48         m_server->addListener(this);
  49 }
  50
  51 ServerConnection::~ServerConnection()
  52 {
  53     if (m_server)
  54         m_server->removeListener(this);
  55
  56     if (m_globalContext)
  57         JSGlobalContextRelease(m_globalContext);
  58 }
  59
  60 void ServerConnection::setGlobalContext(JSGlobalContextRef globalContextRef)
  61 {
  62     m_globalContext = JSGlobalContextRetain(globalContextRef);
  63 }
  64
  65 // Pause & Step
  66
  67 void ServerConnection::pause()
  68 {
  69     if (m_server)
  70         m_server->pause();
  71 }
  72
  73 void ServerConnection::resume()
  74 {
  75     if (m_server)
  76         m_server->resume();
  77 }
  78
  79 void ServerConnection::stepInto()
  80 {
  81     if (m_server)
  82         m_server->step();
  83 }
  84
  85 // Connection Handling
  86
  87 void ServerConnection::applicationTerminating()
  88 {
  89     if (m_server)
  90         m_server->removeListener(this);
  91 }
  92
  93 void ServerConnection::serverConnectionDidDie()
  94 {
  95     if (m_server)
  96         m_server->removeListener(this);
  97 }
  98
  99 // IUnknown --------------------------------------------------
 100 HRESULT STDMETHODCALLTYPE ServerConnection::QueryInterface(REFIID riid, void** ppvObject)
 101 {
 102     *ppvObject = 0;
 103     if (IsEqualGUID(riid, IID_IUnknown))
 104         *ppvObject = this;
 105     else if (IsEqualGUID(riid, IID_IWebScriptDebugListener))
 106         *ppvObject = static_cast<IWebScriptDebugListener*>(this);
 107     else
 108         return E_NOINTERFACE;
 109
 110     AddRef();
 111     return S_OK;
 112 }
 113
 114 ULONG STDMETHODCALLTYPE ServerConnection::AddRef(void)
 115 {
 116     // COM ref-counting isn't useful to us because we're in charge of the lifetime of the WebView.
 117     return 1;
 118 }
 119
 120 ULONG STDMETHODCALLTYPE ServerConnection::Release(void)
 121 {
 122     // COM ref-counting isn't useful to us because we're in charge of the lifetime of the WebView.
 123     return 1;
 124 }
 125 // IWebScriptDebugListener -----------------------------------
 126 HRESULT STDMETHODCALLTYPE ServerConnection::didLoadMainResourceForDataSource(
 127     /* [in] */ IWebView*,
 128     /* [in] */ IWebDataSource* dataSource)
 129 {
 130     // Get document source
 131     COMPtr<IWebDocumentRepresentation> rep;
 132     HRESULT ret = dataSource->representation(&rep);
 133     if (FAILED(ret))
 134         return ret;
 135
 136     BOOL canProvideDocumentSource = FALSE;
 137     ret = rep->canProvideDocumentSource(&canProvideDocumentSource);
 138     if (FAILED(ret))
 139         return ret;
 140
 141     BSTR documentSource = 0;
 142     if (canProvideDocumentSource)
 143         ret = rep->documentSource(&documentSource);
 144
 145     if (FAILED(ret) || !documentSource)
 146         return ret;
 147
 148     JSRetainPtr<JSStringRef> documentSourceJS(Adopt, JSStringCreateWithBSTR(documentSource));
 149     SysFreeString(documentSource);
 150
 151     // Get URL
 152     COMPtr<IWebURLResponse> response;
 153     ret = dataSource->response(&response);
 154     if (FAILED(ret))
 155         return ret;
 156
 157     BSTR url = 0;
 158     ret = response->URL(&url);
 159     if (FAILED(ret))
 160         return ret;
 161
 162     JSRetainPtr<JSStringRef> urlJS(Adopt, JSStringCreateWithBSTR(url));
 163     SysFreeString(url);
 164
 165     DebuggerDocument::updateFileSource(m_globalContext, documentSourceJS.get(), urlJS.get());
 166
 167     return S_OK;
 168 }
 169
 170 HRESULT STDMETHODCALLTYPE ServerConnection::didParseSource(
 171     /* [in] */ IWebView*,
 172     /* [in] */ BSTR sourceCode,
 173     /* [in] */ UINT baseLineNumber,
 174     /* [in] */ BSTR url,
 175     /* [in] */ int sourceID,
 176     /* [in] */ IWebFrame* webFrame)
 177 {
 178     HRESULT ret = S_OK;
 179     if (!m_globalContext || !sourceCode)
 180         return ret;
 181
 182     COMPtr<IWebDataSource> dataSource;
 183     ret = webFrame->dataSource(&dataSource);
 184     if (FAILED(ret))
 185         return ret;
 186
 187     COMPtr<IWebURLResponse> response;
 188     ret = dataSource->response(&response);
 189     if (FAILED(ret))
 190         return ret;
 191
 192     BSTR responseURL;
 193     ret = response->URL(&responseURL);
 194     if (FAILED(ret))
 195         return ret;
 196
 197     BSTR documentSource = 0;
 198     if (!url || !wcscmp(responseURL, url)) {
 199         COMPtr<IWebDocumentRepresentation> rep;
 200         ret = dataSource->representation(&rep);
 201         if (FAILED(ret))
 202             return ret;
 203
 204         BOOL canProvideDocumentSource;
 205         rep->canProvideDocumentSource(&canProvideDocumentSource);
 206         if (FAILED(ret))
 207             return ret;
 208
 209         if (canProvideDocumentSource) {
 210             ret = rep->documentSource(&documentSource);
 211             if (FAILED(ret))
 212                 return ret;
 213         }
 214
 215         if (!url) {
 216             ret = response->URL(&url);
 217             if (FAILED(ret))
 218                 return ret;
 219         }
 220     }
 221     SysFreeString(responseURL);
 222
 223     JSRetainPtr<JSStringRef> sourceJS(Adopt, JSStringCreateWithBSTR(sourceCode));
 224     JSRetainPtr<JSStringRef> documentSourceJS(Adopt, JSStringCreateWithBSTR(documentSource));
 225     SysFreeString(documentSource);
 226     JSRetainPtr<JSStringRef> urlJS(Adopt, JSStringCreateWithBSTR(url));
 227     JSValueRef sidJS = JSValueMakeNumber(m_globalContext, sourceID);
 228     JSValueRef baseLineJS = JSValueMakeNumber(m_globalContext, baseLineNumber);
 229
 230     DebuggerDocument::didParseScript(m_globalContext, sourceJS.get(), documentSourceJS.get(), urlJS.get(), sidJS, baseLineJS);
 231
 232     return S_OK;
 233 }
 234
 235 HRESULT STDMETHODCALLTYPE ServerConnection::failedToParseSource(
 236     /* [in] */ IWebView*,
 237     /* [in] */ BSTR,
 238     /* [in] */ UINT,
 239     /* [in] */ BSTR,
 240     /* [in] */ BSTR,
 241     /* [in] */ IWebFrame*)
 242 {
 243     return S_OK;
 244 }
 245
 246 HRESULT STDMETHODCALLTYPE ServerConnection::didEnterCallFrame(
 247     /* [in] */ IWebView*,
 248     /* [in] */ IWebScriptCallFrame* frame,
 249     /* [in] */ int sourceID,
 250     /* [in] */ int lineNumber,
 251     /* [in] */ IWebFrame*)
 252 {
 253     HRESULT ret = S_OK;
 254     if (!m_globalContext)
 255         return ret;
 256
 257     m_currentFrame = frame;
 258
 259     JSValueRef sidJS = JSValueMakeNumber(m_globalContext, sourceID);
 260     JSValueRef linenoJS = JSValueMakeNumber(m_globalContext, lineNumber);
 261
 262     DebuggerDocument::didEnterCallFrame(m_globalContext, sidJS, linenoJS);
 263
 264     return ret;
 265 }
 266
 267 HRESULT STDMETHODCALLTYPE ServerConnection::willExecuteStatement(
 268     /* [in] */ IWebView*,
 269     /* [in] */ IWebScriptCallFrame*,
 270     /* [in] */ int sourceID,
 271     /* [in] */ int lineNumber,
 272     /* [in] */ IWebFrame*)
 273 {
 274     HRESULT ret = S_OK;
 275     if (!m_globalContext)
 276         return ret;
 277
 278     JSValueRef sidJS = JSValueMakeNumber(m_globalContext, sourceID);
 279     JSValueRef linenoJS = JSValueMakeNumber(m_globalContext, lineNumber);
 280
 281     DebuggerDocument::willExecuteStatement(m_globalContext, sidJS, linenoJS);
 282     return ret;
 283 }
 284
 285 HRESULT STDMETHODCALLTYPE ServerConnection::willLeaveCallFrame(
 286     /* [in] */ IWebView*,
 287     /* [in] */ IWebScriptCallFrame* frame,
 288     /* [in] */ int sourceID,
 289     /* [in] */ int lineNumber,
 290     /* [in] */ IWebFrame*)
 291 {
 292     HRESULT ret = S_OK;
 293     if (!m_globalContext)
 294         return ret;
 295
 296     JSValueRef sidJS = JSValueMakeNumber(m_globalContext, sourceID);
 297     JSValueRef linenoJS = JSValueMakeNumber(m_globalContext, lineNumber);
 298
 299     DebuggerDocument::willLeaveCallFrame(m_globalContext, sidJS, linenoJS);
 300
 301     m_currentFrame = frame;
 302
 303     return S_OK;
 304 }
 305
 306 HRESULT STDMETHODCALLTYPE ServerConnection::exceptionWasRaised(
 307     /* [in] */ IWebView*,
 308     /* [in] */ IWebScriptCallFrame*,
 309     /* [in] */ int sourceID,
 310     /* [in] */ int lineNumber,
 311     /* [in] */ IWebFrame*)
 312 {
 313     HRESULT ret = S_OK;
 314     if (!m_globalContext)
 315         return ret;
 316
 317     JSValueRef sidJS = JSValueMakeNumber(m_globalContext, sourceID);
 318     JSValueRef linenoJS = JSValueMakeNumber(m_globalContext, lineNumber);
 319
 320     DebuggerDocument::exceptionWasRaised(m_globalContext, sidJS, linenoJS);
 321
 322     return ret;
 323 }
 324
 325 // Stack & Variables
 326
 327 IWebScriptCallFrame* ServerConnection::currentFrame() const
 328 {
 329     return m_currentFrame.get();
 330 }
 331
 332 IWebScriptCallFrame* ServerConnection::getCallerFrame(int callFrame) const
 333 {
 334     COMPtr<IWebScriptCallFrame> cframe = currentFrame();
 335     COMPtr<IWebScriptCallFrame> callerFrame;
 336     for (int count = 0; count < callFrame; count++) {
 337         if (FAILED(cframe->caller(&callerFrame)))
 338             return 0;
 339
 340         cframe = callerFrame;
 341     }
 342
 343     return cframe.get();
 344 }