1 files changed, 127 insertions, 0 deletions
diff --git a/chrome/browser/automation/url_request_automation_job.h b/chrome/browser/automation/url_request_automation_job.h
new file mode 100644
index 0000000..b8c7442
--- /dev/null
+++ b/chrome/browser/automation/url_request_automation_job.h
@@ -0,0 +1,127 @@
+// Copyright (c) 2006-2009 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.
+// This class simulates what wininet does when a dns lookup fails.
+
+#ifndef CHROME_BROWSER_AUTOMATION_URL_REQUEST_AUTOMATION_JOB_H_
+#define CHROME_BROWSER_AUTOMATION_URL_REQUEST_AUTOMATION_JOB_H_
+
+#include <vector>
+
+#include "chrome/browser/automation/automation_resource_message_filter.h"
+#include "chrome/common/ref_counted_util.h"
+#include "net/http/http_response_headers.h"
+#include "net/url_request/url_request.h"
+#include "net/url_request/url_request_job.h"
+
+class AutomationResourceMessageFilter;
+
+namespace IPC {
+class Message;
+struct AutomationURLResponse;
+};
+
+// URLRequestJob implementation that loads the resources using
+// automation.
+class URLRequestAutomationJob : public URLRequestJob {
+ public:
+  URLRequestAutomationJob(URLRequest* request, int tab, int request_id,
+                          AutomationResourceMessageFilter* filter,
+                          bool is_pending);
+
+  // Register our factory for HTTP/HTTPs requests.
+  static bool EnsureProtocolFactoryRegistered();
+
+  static URLRequest::ProtocolFactory Factory;
+
+  // URLRequestJob methods.
+  virtual void Start();
+  virtual void Kill();
+  virtual bool GetMimeType(std::string* mime_type) const;
+  virtual bool GetCharset(std::string* charset);
+  virtual void GetResponseInfo(net::HttpResponseInfo* info);
+  virtual int GetResponseCode() const;
+  virtual bool IsRedirectResponse(GURL* location, int* http_status_code);
+
+  // Peek and process automation messages for URL requests.
+  static bool MayFilterMessage(const IPC::Message& message, int* request_id);
+  void OnMessage(const IPC::Message& message);
+
+  int id() const {
+    return id_;
+  }
+
+  int request_id() const {
+    return request_id_;
+  }
+
+  bool is_pending() const {
+    return is_pending_;
+  }
+
+  AutomationResourceMessageFilter* message_filter() const {
+    return message_filter_;
+  }
+
+  // Resumes a job, which was waiting for the external host to connect to the
+  // automation channel. This is to ensure that this request gets routed to the
+  // external host.
+  void StartPendingJob(int new_tab_handle,
+                       AutomationResourceMessageFilter* new_filter);
+
+ protected:
+  // Protected URLRequestJob override.
+  virtual bool ReadRawData(net::IOBuffer* buf, int buf_size, int* bytes_read);
+
+  void StartAsync();
+  void Cleanup();
+  void DisconnectFromMessageFilter();
+
+  // IPC message handlers.
+  void OnRequestStarted(int tab, int id,
+      const IPC::AutomationURLResponse& response);
+  void OnDataAvailable(int tab, int id, const std::string& bytes);
+  void OnRequestEnd(int tab, int id, const URLRequestStatus& status);
+
+ private:
+  virtual ~URLRequestAutomationJob();
+
+  // Task which is scheduled in the URLRequestAutomationJob::ReadRawData
+  // function, which completes the job.
+  void NotifyJobCompletionTask();
+
+  int id_;
+  int tab_;
+  scoped_refptr<AutomationResourceMessageFilter> message_filter_;
+
+  scoped_refptr<net::IOBuffer> pending_buf_;
+  size_t pending_buf_size_;
+
+  std::string mime_type_;
+  scoped_refptr<net::HttpResponseHeaders> headers_;
+  std::string redirect_url_;
+  int redirect_status_;
+  int request_id_;
+
+  static int instance_count_;
+
+  static bool is_protocol_factory_registered_;
+  // The previous HTTP/HTTPs protocol factories. We pass unhandled
+  // requests off to these factories
+  static URLRequest::ProtocolFactory* old_http_factory_;
+  static URLRequest::ProtocolFactory* old_https_factory_;
+
+  // Set to true if the job is waiting for the external host to connect to the
+  // automation channel, which will be used for routing the network requests to
+  // the host.
+  bool is_pending_;
+
+  // Contains the request status code, which is eventually passed  to the http
+  // stack when we receive a Read request for a completed job.
+  URLRequestStatus request_status_;
+
+  DISALLOW_COPY_AND_ASSIGN(URLRequestAutomationJob);
+};
+
+#endif  // CHROME_BROWSER_AUTOMATION_URL_REQUEST_AUTOMATION_JOB_H_
+