blob: 617f609515a91e2c69ea93aa4d56db4da1fda71c (
plain)
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
|
// Copyright 2013 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 CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_WRITE_FROM_URL_OPERATION_H_
#define CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_WRITE_FROM_URL_OPERATION_H_
#include <stdint.h>
#include "chrome/browser/extensions/api/image_writer_private/operation.h"
#include "services/network/public/mojom/url_loader_factory.mojom.h"
#include "url/gurl.h"
namespace network {
struct ResourceResponseHead;
class SimpleURLLoader;
} // namespace network
namespace extensions {
namespace image_writer {
class OperationManager;
// Encapsulates a write of an image accessed via URL.
class WriteFromUrlOperation : public Operation {
public:
WriteFromUrlOperation(base::WeakPtr<OperationManager> manager,
const ExtensionId& extension_id,
network::mojom::URLLoaderFactoryPtrInfo factory_info,
GURL url,
const std::string& hash,
const std::string& storage_unit_id,
const base::FilePath& download_folder);
void StartImpl() override;
protected:
friend class OperationForTest;
friend class WriteFromUrlOperationForTest;
~WriteFromUrlOperation() override;
// Sets the image_path to the correct location to download to.
void GetDownloadTarget(base::OnceClosure continuation);
// Downloads the |url| to the currently configured |image_path|. Should not
// be called without calling |GetDownloadTarget| first.
void Download(base::OnceClosure continuation);
// Verifies the download matches |hash|. If the hash is empty, this stage is
// skipped.
void VerifyDownload(base::OnceClosure continuation);
private:
void DestroySimpleURLLoader();
void OnResponseStarted(const GURL& final_url,
const network::ResourceResponseHead& response_head);
void OnDataDownloaded(uint64_t current);
void OnSimpleLoaderComplete(base::FilePath file_path);
void VerifyDownloadCompare(base::OnceClosure continuation,
const std::string& download_hash);
void VerifyDownloadComplete(base::OnceClosure continuation);
// Arguments
network::mojom::URLLoaderFactoryPtrInfo url_loader_factory_ptr_info_;
GURL url_;
const std::string hash_;
// Local state
std::unique_ptr<network::SimpleURLLoader> simple_url_loader_;
base::OnceClosure download_continuation_;
int total_response_bytes_ = -1;
};
} // namespace image_writer
} // namespace extensions
#endif // CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_WRITE_FROM_URL_OPERATION_H_
|