120 lines
4.6 KiB
C++
120 lines
4.6 KiB
C++
// Copyright (c) 2016 GitHub, Inc.
|
|
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#ifndef ATOM_BROWSER_NET_ATOM_URL_REQUEST_H_
|
|
#define ATOM_BROWSER_NET_ATOM_URL_REQUEST_H_
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include "atom/browser/api/atom_api_url_request.h"
|
|
#include "atom/browser/atom_browser_context.h"
|
|
#include "base/memory/ref_counted.h"
|
|
#include "base/memory/weak_ptr.h"
|
|
#include "net/base/auth.h"
|
|
#include "net/base/chunked_upload_data_stream.h"
|
|
#include "net/base/io_buffer.h"
|
|
#include "net/base/upload_element_reader.h"
|
|
#include "net/http/http_response_headers.h"
|
|
#include "net/url_request/url_request.h"
|
|
#include "net/url_request/url_request_context_getter_observer.h"
|
|
|
|
namespace atom {
|
|
|
|
class AtomURLRequest : public base::RefCountedThreadSafe<AtomURLRequest>,
|
|
public net::URLRequest::Delegate,
|
|
public net::URLRequestContextGetterObserver {
|
|
public:
|
|
static scoped_refptr<AtomURLRequest> Create(
|
|
AtomBrowserContext* browser_context,
|
|
const std::string& method,
|
|
const std::string& url,
|
|
const std::string& redirect_policy,
|
|
api::URLRequest* delegate);
|
|
void Terminate();
|
|
|
|
bool Write(scoped_refptr<const net::IOBufferWithSize> buffer, bool is_last);
|
|
void SetChunkedUpload(bool is_chunked_upload);
|
|
void Cancel();
|
|
void FollowRedirect();
|
|
void SetExtraHeader(const std::string& name, const std::string& value) const;
|
|
void RemoveExtraHeader(const std::string& name) const;
|
|
void PassLoginInformation(const base::string16& username,
|
|
const base::string16& password) const;
|
|
void SetLoadFlags(int flags) const;
|
|
void GetUploadProgress(mate::Dictionary* progress) const;
|
|
|
|
protected:
|
|
// Overrides of net::URLRequest::Delegate
|
|
void OnReceivedRedirect(net::URLRequest* request,
|
|
const net::RedirectInfo& info,
|
|
bool* defer_redirect) override;
|
|
void OnAuthRequired(net::URLRequest* request,
|
|
net::AuthChallengeInfo* auth_info) override;
|
|
void OnResponseStarted(net::URLRequest* request, int net_error) override;
|
|
void OnReadCompleted(net::URLRequest* request, int bytes_read) override;
|
|
|
|
// Overrides of net::URLRequestContextGetterObserver
|
|
void OnContextShuttingDown() override;
|
|
|
|
private:
|
|
friend class base::RefCountedThreadSafe<AtomURLRequest>;
|
|
|
|
explicit AtomURLRequest(api::URLRequest* delegate);
|
|
~AtomURLRequest() override;
|
|
|
|
void DoInitialize(scoped_refptr<net::URLRequestContextGetter>,
|
|
const std::string& method,
|
|
const std::string& url,
|
|
const std::string& redirect_policy);
|
|
void DoTerminate();
|
|
void DoWriteBuffer(scoped_refptr<const net::IOBufferWithSize> buffer,
|
|
bool is_last);
|
|
void DoCancel();
|
|
void DoFollowRedirect();
|
|
void DoSetExtraHeader(const std::string& name,
|
|
const std::string& value) const;
|
|
void DoRemoveExtraHeader(const std::string& name) const;
|
|
void DoSetAuth(const base::string16& username,
|
|
const base::string16& password) const;
|
|
void DoCancelAuth() const;
|
|
void DoCancelWithError(const std::string& error, bool isRequestError);
|
|
void DoSetLoadFlags(int flags) const;
|
|
|
|
void ReadResponse();
|
|
bool CopyAndPostBuffer(int bytes_read);
|
|
|
|
void InformDelegateReceivedRedirect(
|
|
int status_code,
|
|
const std::string& method,
|
|
const GURL& url,
|
|
scoped_refptr<net::HttpResponseHeaders> response_headers) const;
|
|
void InformDelegateAuthenticationRequired(
|
|
scoped_refptr<net::AuthChallengeInfo> auth_info) const;
|
|
void InformDelegateResponseStarted(
|
|
scoped_refptr<net::HttpResponseHeaders>) const;
|
|
void InformDelegateResponseData(
|
|
scoped_refptr<net::IOBufferWithSize> data) const;
|
|
void InformDelegateResponseCompleted() const;
|
|
void InformDelegateErrorOccured(const std::string& error,
|
|
bool isRequestError) const;
|
|
|
|
api::URLRequest* delegate_;
|
|
std::unique_ptr<net::URLRequest> request_;
|
|
scoped_refptr<net::URLRequestContextGetter> request_context_getter_;
|
|
|
|
bool is_chunked_upload_ = false;
|
|
std::string redirect_policy_;
|
|
std::unique_ptr<net::ChunkedUploadDataStream> chunked_stream_;
|
|
std::unique_ptr<net::ChunkedUploadDataStream::Writer> chunked_stream_writer_;
|
|
std::vector<std::unique_ptr<net::UploadElementReader>>
|
|
upload_element_readers_;
|
|
scoped_refptr<net::IOBuffer> response_read_buffer_;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(AtomURLRequest);
|
|
};
|
|
|
|
} // namespace atom
|
|
|
|
#endif // ATOM_BROWSER_NET_ATOM_URL_REQUEST_H_
|