Pipe the devTools FrameId from blink into the browser for headless

headless/public/util/generic_url_request_job.h

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef HEADLESS_PUBLIC_UTIL_GENERIC_URL_REQUEST_JOB_H_
 #define HEADLESS_PUBLIC_UTIL_GENERIC_URL_REQUEST_JOB_H_
 
 #include <stddef.h>
 #include <functional>
 #include <memory>
@@ skipping 13 matching lines @@
 class HttpResponseHeaders;
 class IOBuffer;
 }  // namespace net
 
 namespace content {
 class ResourceRequestInfo;
 }  // namespace content
 
 namespace headless {
 
+class HeadlessBrowserContext;
 class URLRequestDispatcher;
 
 // Wrapper around net::URLRequest with helpers to access select metadata.
 class Request {
  public:
   virtual uint64_t GetRequestId() const = 0;
 
   virtual const net::URLRequest* GetURLRequest() const = 0;
 
   // The frame from which the request came from.
@@ skipping 103 matching lines @@
         scoped_refptr<net::HttpResponseHeaders> response_headers,
         const char* body,
         size_t body_size) = 0;
 
    protected:
     virtual ~Delegate() {}
   };
 
   // NOTE |url_request_dispatcher| and |delegate| must outlive the
   // GenericURLRequestJob.
-  GenericURLRequestJob(net::URLRequest* request,
-                       net::NetworkDelegate* network_delegate,
-                       URLRequestDispatcher* url_request_dispatcher,
-                       std::unique_ptr<URLFetcher> url_fetcher,
-                       Delegate* delegate);
+  // TODO(alexclarke): Remove the default parameter.
+  GenericURLRequestJob(
+      net::URLRequest* request,
+      net::NetworkDelegate* network_delegate,
+      URLRequestDispatcher* url_request_dispatcher,
+      std::unique_ptr<URLFetcher> url_fetcher,
+      Delegate* delegate,
+      HeadlessBrowserContext* headless_browser_context = nullptr);
   ~GenericURLRequestJob() override;
 
   // net::URLRequestJob implementation:
   void SetExtraRequestHeaders(const net::HttpRequestHeaders& headers) override;
   void Start() override;
   int ReadRawData(net::IOBuffer* buf, int buf_size) override;
   void GetResponseInfo(net::HttpResponseInfo* info) override;
   bool GetMimeType(std::string* mime_type) const override;
   bool GetCharset(std::string* charset) override;
   void GetLoadTimingInfo(net::LoadTimingInfo* load_timing_info) const override;
@@ skipping 33 matching lines @@
                           const std::string& method,
                           const base::Closure& done_callback,
                           const net::CookieList& cookie_list);
 
   std::unique_ptr<URLFetcher> url_fetcher_;
   net::HttpRequestHeaders extra_request_headers_;
   scoped_refptr<net::HttpResponseHeaders> response_headers_;
   scoped_refptr<base::SingleThreadTaskRunner> origin_task_runner_;
   std::unique_ptr<MockResponseData> mock_response_;
   Delegate* delegate_;          // Not owned.
+  HeadlessBrowserContext* headless_browser_context_;           // Not owned.
   const content::ResourceRequestInfo* request_resource_info_;  // Not owned.
   const char* body_ = nullptr;  // Not owned.
   size_t body_size_ = 0;
   size_t read_offset_ = 0;
   base::TimeTicks response_time_;
   const uint64_t request_id_;
   static uint64_t next_request_id_;
 
   base::WeakPtrFactory<GenericURLRequestJob> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(GenericURLRequestJob);
 };
 
 }  // namespace headless
 
 #endif  // HEADLESS_PUBLIC_UTIL_GENERIC_URL_REQUEST_JOB_H_