1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
// Copyright 2014 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 COMPONENTS_SYNC_ENGINE_IMPL_ATTACHMENTS_ATTACHMENT_DOWNLOADER_IMPL_H_
#define COMPONENTS_SYNC_ENGINE_IMPL_ATTACHMENTS_ATTACHMENT_DOWNLOADER_IMPL_H_
#include <stdint.h>
#include <memory>
#include <string>
#include <unordered_map>
#include <vector>
#include "base/gtest_prod_util.h"
#include "base/macros.h"
#include "base/threading/non_thread_safe.h"
#include "components/sync/base/model_type.h"
#include "components/sync/engine/attachments/attachment_downloader.h"
#include "google_apis/gaia/oauth2_token_service_request.h"
#include "net/url_request/url_fetcher_delegate.h"
#include "net/url_request/url_request_context_getter.h"
#include "url/gurl.h"
namespace net {
class HttpResponseHeaders;
} // namespace net
namespace syncer {
// An implementation of AttachmentDownloader.
class AttachmentDownloaderImpl : public AttachmentDownloader,
public OAuth2TokenService::Consumer,
public net::URLFetcherDelegate,
public base::NonThreadSafe {
public:
// |sync_service_url| is the URL of the sync service.
//
// |url_request_context_getter| provides a URLRequestContext.
//
// |account_id| is the account id to use for downloads.
//
// |scopes| is the set of scopes to use for downloads.
//
// |token_service_provider| provides an OAuth2 token service.
//
// |store_birthday| is the raw, sync store birthday.
//
// |model_type| is the model type this downloader is used with.
AttachmentDownloaderImpl(
const GURL& sync_service_url,
const scoped_refptr<net::URLRequestContextGetter>&
url_request_context_getter,
const std::string& account_id,
const OAuth2TokenService::ScopeSet& scopes,
const scoped_refptr<OAuth2TokenServiceRequest::TokenServiceProvider>&
token_service_provider,
const std::string& store_birthday,
ModelType model_type);
~AttachmentDownloaderImpl() override;
// AttachmentDownloader implementation.
void DownloadAttachment(const AttachmentId& attachment_id,
const DownloadCallback& callback) override;
// OAuth2TokenService::Consumer implementation.
void OnGetTokenSuccess(const OAuth2TokenService::Request* request,
const std::string& access_token,
const base::Time& expiration_time) override;
void OnGetTokenFailure(const OAuth2TokenService::Request* request,
const GoogleServiceAuthError& error) override;
// net::URLFetcherDelegate implementation.
void OnURLFetchComplete(const net::URLFetcher* source) override;
private:
FRIEND_TEST_ALL_PREFIXES(AttachmentDownloaderImplTest,
ExtractCrc32c_NoHeaders);
FRIEND_TEST_ALL_PREFIXES(AttachmentDownloaderImplTest, ExtractCrc32c_First);
FRIEND_TEST_ALL_PREFIXES(AttachmentDownloaderImplTest, ExtractCrc32c_TooLong);
FRIEND_TEST_ALL_PREFIXES(AttachmentDownloaderImplTest, ExtractCrc32c_None);
FRIEND_TEST_ALL_PREFIXES(AttachmentDownloaderImplTest, ExtractCrc32c_Empty);
struct DownloadState;
typedef std::string AttachmentUrl;
typedef std::unordered_map<AttachmentUrl, std::unique_ptr<DownloadState>>
StateMap;
typedef std::vector<DownloadState*> StateList;
std::unique_ptr<net::URLFetcher> CreateFetcher(
const AttachmentUrl& url,
const std::string& access_token);
void RequestAccessToken(DownloadState* download_state);
void ReportResult(
const DownloadState& download_state,
const DownloadResult& result,
const scoped_refptr<base::RefCountedString>& attachment_data);
// Extract the crc32c from an X-Goog-Hash header in |headers|.
//
// Return true if a crc32c was found and useable for checking data integrity.
// "Usable" means headers are present, there is "x-goog-hash" header with
// "crc32c" hash in it, this hash is correctly base64 encoded 32 bit integer.
static bool ExtractCrc32c(const net::HttpResponseHeaders* headers,
uint32_t* crc32c);
GURL sync_service_url_;
scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_;
std::string account_id_;
OAuth2TokenService::ScopeSet oauth2_scopes_;
scoped_refptr<OAuth2TokenServiceRequest::TokenServiceProvider>
token_service_provider_;
std::unique_ptr<OAuth2TokenService::Request> access_token_request_;
std::string raw_store_birthday_;
StateMap state_map_;
// |requests_waiting_for_access_token_| only keeps references to DownloadState
// objects while access token request is pending. It doesn't control objects'
// lifetime.
StateList requests_waiting_for_access_token_;
ModelType model_type_;
DISALLOW_COPY_AND_ASSIGN(AttachmentDownloaderImpl);
};
} // namespace syncer
#endif // COMPONENTS_SYNC_ENGINE_IMPL_ATTACHMENTS_ATTACHMENT_DOWNLOADER_IMPL_H_
|
