src/components/cronet/android/url_request_peer.h

   1 // Copyright 2014 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 COMPONENTS_CRONET_ANDROID_URL_REQUEST_PEER_H_
   6 #define COMPONENTS_CRONET_ANDROID_URL_REQUEST_PEER_H_
   7
   8 #include "base/compiler_specific.h"
   9 #include "base/macros.h"
  10 #include "base/memory/ref_counted.h"
  11 #include "components/cronet/android/url_request_context_peer.h"
  12 #include "net/base/request_priority.h"
  13 #include "net/base/upload_data_stream.h"
  14 #include "net/http/http_request_headers.h"
  15 #include "net/url_request/url_request.h"
  16
  17 namespace cronet {
  18
  19 // An adapter from the JNI |UrlRequest| object and the Chromium |URLRequest|
  20 // object.
  21 class URLRequestPeer : public net::URLRequest::Delegate {
  22  public:
  23   // The delegate which is called when the request finishes.
  24   class URLRequestPeerDelegate
  25       : public base::RefCountedThreadSafe<URLRequestPeerDelegate> {
  26    public:
  27     virtual void OnAppendChunkCompleted(URLRequestPeer* request) = 0;
  28     virtual void OnResponseStarted(URLRequestPeer* request) = 0;
  29     virtual void OnBytesRead(URLRequestPeer* request) = 0;
  30     virtual void OnRequestFinished(URLRequestPeer* request) = 0;
  31
  32    protected:
  33     friend class base::RefCountedThreadSafe<URLRequestPeerDelegate>;
  34     virtual ~URLRequestPeerDelegate() {}
  35   };
  36
  37   URLRequestPeer(URLRequestContextPeer* context,
  38                  URLRequestPeerDelegate* delegate,
  39                  GURL url,
  40                  net::RequestPriority priority);
  41   virtual ~URLRequestPeer();
  42
  43   // Sets the request method GET, POST etc
  44   void SetMethod(const std::string& method);
  45
  46   // Adds a header to the request
  47   void AddHeader(const std::string& name, const std::string& value);
  48
  49   // Sets the contents of the POST request
  50   void SetPostContent(const char* bytes, int bytes_len);
  51
  52   // Indicates that the request body will be streamed by calling AppendChunk()
  53   // repeatedly. This must be called before Start().
  54   void EnableStreamingUpload();
  55
  56   // Appends a chunk to the POST body
  57   // This must be called after EnableChunkedUpload() and Start().
  58   void AppendChunk(const char* bytes, int bytes_len, bool is_last_chunk);
  59
  60   // Starts the request.
  61   void Start();
  62
  63   // Cancels the request.
  64   void Cancel();
  65
  66   // Releases all resources for the request and deletes the object itself.
  67   void Destroy();
  68
  69   // Returns the URL of the request.
  70   GURL url() const { return url_; }
  71
  72   // Returns the error code after the request is complete.
  73   // Negative codes indicate system errors.
  74   int error_code() const { return error_code_; }
  75
  76   // Returns the HTTP status code.
  77   int http_status_code() const {
  78     return http_status_code_;
  79   };
  80
  81   // Returns the value of the content-length response header.
  82   int64 content_length() const { return expected_size_; }
  83
  84   // Returns the value of the content-type response header.
  85   std::string content_type() const { return content_type_; }
  86
  87   // Returns the value of the specified response header.
  88   std::string GetHeader(const std::string& name) const;
  89
  90   // Returns the overall number of bytes read.
  91   size_t bytes_read() const { return bytes_read_; }
  92
  93   // Returns a pointer to the downloaded data.
  94   unsigned char* Data() const;
  95
  96   virtual void OnResponseStarted(net::URLRequest* request) OVERRIDE;
  97
  98   virtual void OnReadCompleted(net::URLRequest* request,
  99                                int bytes_read) OVERRIDE;
 100
 101  private:
 102   URLRequestContextPeer* context_;
 103   scoped_refptr<URLRequestPeerDelegate> delegate_;
 104   GURL url_;
 105   net::RequestPriority priority_;
 106   std::string method_;
 107   net::HttpRequestHeaders headers_;
 108   net::URLRequest* url_request_;
 109   scoped_ptr<net::UploadDataStream> upload_data_stream_;
 110   scoped_refptr<net::GrowableIOBuffer> read_buffer_;
 111   int bytes_read_;
 112   int total_bytes_read_;
 113   int error_code_;
 114   int http_status_code_;
 115   std::string content_type_;
 116   bool canceled_;
 117   int64 expected_size_;
 118   bool streaming_upload_;
 119
 120   static void OnDestroyRequest(URLRequestPeer* self);
 121
 122   void OnInitiateConnection();
 123   void OnCancelRequest();
 124   void OnRequestSucceeded();
 125   void OnRequestFailed();
 126   void OnRequestCompleted();
 127   void OnRequestCanceled();
 128   void OnBytesRead(int bytes_read);
 129   void OnAppendChunk(const char* bytes, int bytes_len, bool is_last_chunk);
 130
 131   void Read();
 132
 133   DISALLOW_COPY_AND_ASSIGN(URLRequestPeer);
 134 };
 135
 136 }  // namespace cronet
 137
 138 #endif  // COMPONENTS_CRONET_ANDROID_URL_REQUEST_PEER_H_