Reviewed by NOBODY (OOPS!).
[WebKit-https.git] / 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 }