android_webview/browser/net/android_stream_reader_url_request_job.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef ANDROID_WEBVIEW_NATIVE_ANDROID_STREAM_READER_URL_REQUEST_JOB_H_
   6 #define ANDROID_WEBVIEW_NATIVE_ANDROID_STREAM_READER_URL_REQUEST_JOB_H_
   7
   8 #include <string>
   9
  10 #include "base/android/scoped_java_ref.h"
  11 #include "base/location.h"
  12 #include "base/memory/ref_counted.h"
  13 #include "base/memory/scoped_ptr.h"
  14 #include "base/memory/weak_ptr.h"
  15 #include "base/threading/thread_checker.h"
  16 #include "net/http/http_byte_range.h"
  17 #include "net/url_request/url_request_job.h"
  18
  19 namespace android_webview {
  20 class InputStream;
  21 class InputStreamReader;
  22 }
  23
  24 namespace base {
  25 class TaskRunner;
  26 }
  27
  28 namespace net {
  29 class HttpResponseInfo;
  30 class URLRequest;
  31 }
  32
  33 class InputStreamReaderWrapper;
  34
  35 // A request job that reads data from a Java InputStream.
  36 class AndroidStreamReaderURLRequestJob : public net::URLRequestJob {
  37  public:
  38   /*
  39    * We use a delegate so that we can share code for this job in slightly
  40    * different contexts.
  41    */
  42   class Delegate {
  43    public:
  44     // This method is called from a worker thread, not from the IO thread.
  45     virtual scoped_ptr<android_webview::InputStream> OpenInputStream(
  46         JNIEnv* env,
  47         const GURL& url) = 0;
  48
  49     // This method is called on the Job's thread if the result of calling
  50     // OpenInputStream was null.
  51     // Setting the |restart| parameter to true will cause the request to be
  52     // restarted with a new job.
  53     virtual void OnInputStreamOpenFailed(
  54         net::URLRequest* request,
  55         bool* restart) = 0;
  56
  57     virtual bool GetMimeType(
  58         JNIEnv* env,
  59         net::URLRequest* request,
  60         android_webview::InputStream* stream,
  61         std::string* mime_type) = 0;
  62
  63     virtual bool GetCharset(
  64         JNIEnv* env,
  65         net::URLRequest* request,
  66         android_webview::InputStream* stream,
  67         std::string* charset) = 0;
  68
  69     virtual ~Delegate() {}
  70   };
  71
  72   AndroidStreamReaderURLRequestJob(
  73       net::URLRequest* request,
  74       net::NetworkDelegate* network_delegate,
  75       scoped_ptr<Delegate> delegate);
  76
  77   // URLRequestJob:
  78   virtual void Start() OVERRIDE;
  79   virtual void Kill() OVERRIDE;
  80   virtual bool ReadRawData(net::IOBuffer* buf,
  81                            int buf_size,
  82                            int* bytes_read) OVERRIDE;
  83   virtual void SetExtraRequestHeaders(
  84       const net::HttpRequestHeaders& headers) OVERRIDE;
  85   virtual bool GetMimeType(std::string* mime_type) const OVERRIDE;
  86   virtual bool GetCharset(std::string* charset) OVERRIDE;
  87   virtual int GetResponseCode() const OVERRIDE;
  88   virtual void GetResponseInfo(net::HttpResponseInfo* info) OVERRIDE;
  89
  90  protected:
  91   virtual ~AndroidStreamReaderURLRequestJob();
  92
  93   // Gets the TaskRunner for the worker thread.
  94   // Overridden in unittests.
  95   virtual base::TaskRunner* GetWorkerThreadRunner();
  96
  97   // Creates an InputStreamReader instance.
  98   // Overridden in unittests to return a mock.
  99   virtual scoped_ptr<android_webview::InputStreamReader>
 100       CreateStreamReader(android_webview::InputStream* stream);
 101
 102  private:
 103   void HeadersComplete(int status_code, const std::string& status_text);
 104
 105   void OnInputStreamOpened(
 106       scoped_ptr<Delegate> delegate,
 107       scoped_ptr<android_webview::InputStream> input_stream);
 108   void OnReaderSeekCompleted(int content_size);
 109   void OnReaderReadCompleted(int bytes_read);
 110
 111   net::HttpByteRange byte_range_;
 112   scoped_ptr<net::HttpResponseInfo> response_info_;
 113   scoped_ptr<Delegate> delegate_;
 114   scoped_refptr<InputStreamReaderWrapper> input_stream_reader_wrapper_;
 115   base::WeakPtrFactory<AndroidStreamReaderURLRequestJob> weak_factory_;
 116   base::ThreadChecker thread_checker_;
 117
 118   DISALLOW_COPY_AND_ASSIGN(AndroidStreamReaderURLRequestJob);
 119 };
 120
 121 #endif  // ANDROID_WEBVIEW_NATIVE_ANDROID_STREAM_READER_URL_REQUEST_JOB_H_