[Devtools] Fixed websocket initiators

third_party/WebKit/Source/core/inspector/InspectorNetworkAgent.cpp

 /*
  * Copyright (C) 2011 Google 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:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 801 matching lines @@
         return initiatorObject;
     }
 
     if (m_isRecalculatingStyle && m_styleRecalculationInitiator)
         return m_styleRecalculationInitiator->clone();
 
     return protocol::Network::Initiator::create()
         .setType(protocol::Network::Initiator::TypeEnum::Other).build();
 }
 
-void InspectorNetworkAgent::didCreateWebSocket(Document*, unsigned long identifier, const KURL& requestURL, const String&)
+void InspectorNetworkAgent::didCreateWebSocket(Document* document, unsigned long identifier, const KURL& requestURL, const String&)
 {
-    frontend()->webSocketCreated(IdentifiersFactory::requestId(identifier), urlWithoutFragment(requestURL).getString());
+    std::unique_ptr<protocol::Runtime::StackTrace> currentStackTrace = SourceLocation::capture(document)->buildInspectorObject();
+    DCHECK(currentStackTrace);

[dgozman, 2016/06/24 20:21:03]

This DCHECK looks suspicious. What if websocket was initiated from native? Can
we replay it, for example?
I think we should make initiator optional.

+    std::unique_ptr<protocol::Network::Initiator> initiatorObject = protocol::Network::Initiator::create()
+        .setType(protocol::Network::Initiator::TypeEnum::Script).build();
+    initiatorObject->setStack(std::move(currentStackTrace));
+
+    frontend()->webSocketCreated(IdentifiersFactory::requestId(identifier), urlWithoutFragment(requestURL).getString(), std::move(initiatorObject));
 }
 
 void InspectorNetworkAgent::willSendWebSocketHandshakeRequest(Document*, unsigned long identifier, const WebSocketHandshakeRequest* request)
 {
     DCHECK(request);
     std::unique_ptr<protocol::Network::WebSocketRequest> requestObject = protocol::Network::WebSocketRequest::create()
         .setHeaders(buildObjectForHeaders(request->headerFields())).build();
     frontend()->webSocketWillSendHandshakeRequest(IdentifiersFactory::requestId(identifier), monotonicallyIncreasingTime(), currentTime(), std::move(requestObject));
 }
 
@@ skipping 288 matching lines @@
     , m_removeFinishedReplayXHRTimer(this, &InspectorNetworkAgent::removeFinishedReplayXHRFired)
 {
 }
 
 bool InspectorNetworkAgent::shouldForceCORSPreflight()
 {
     return m_state->booleanProperty(NetworkAgentState::cacheDisabled, false);
 }
 
 } // namespace blink