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
|
// 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.
#include "chrome/browser/ui/webui/chromeos/image_source.h"
#include <vector>
#include "base/bind.h"
#include "base/files/file_util.h"
#include "base/location.h"
#include "base/memory/ref_counted_memory.h"
#include "base/sequenced_task_runner.h"
#include "base/single_thread_task_runner.h"
#include "base/thread_task_runner_handle.h"
#include "chrome/browser/chromeos/login/users/avatar/user_image_loader.h"
#include "chrome/common/url_constants.h"
#include "components/user_manager/user_image/user_image.h"
#include "content/public/browser/browser_thread.h"
#include "net/base/mime_util.h"
using content::BrowserThread;
namespace chromeos {
namespace {
const char* kWhitelistedDirectories[] = {
"regulatory_labels"
};
// Callback for user_manager::UserImageLoader.
void ImageLoaded(
const content::URLDataSource::GotDataCallback& got_data_callback,
const user_manager::UserImage& user_image) {
DCHECK_CURRENTLY_ON(BrowserThread::UI);
if (user_image.has_raw_image())
got_data_callback.Run(new base::RefCountedBytes(user_image.raw_image()));
else
got_data_callback.Run(NULL);
}
// Looks for the image at |path| under the shared assets directory.
void StartOnBlockingPool(
const std::string& path,
scoped_refptr<UserImageLoader> image_loader,
const content::URLDataSource::GotDataCallback& got_data_callback,
scoped_refptr<base::SingleThreadTaskRunner> thread_task_runner) {
DCHECK(BrowserThread::GetBlockingPool()->RunsTasksOnCurrentThread());
const base::FilePath asset_dir(FILE_PATH_LITERAL(chrome::kChromeOSAssetPath));
const base::FilePath image_path = asset_dir.AppendASCII(path);
if (base::PathExists(image_path)) {
image_loader->Start(image_path.value(), 0,
base::Bind(&ImageLoaded, got_data_callback));
} else {
thread_task_runner->PostTask(FROM_HERE,
base::Bind(got_data_callback, nullptr));
}
}
} // namespace
ImageSource::ImageSource() : weak_factory_(this) {
base::SequencedWorkerPool* blocking_pool =
BrowserThread::GetBlockingPool();
task_runner_ = blocking_pool->GetSequencedTaskRunnerWithShutdownBehavior(
blocking_pool->GetSequenceToken(),
base::SequencedWorkerPool::CONTINUE_ON_SHUTDOWN);
}
ImageSource::~ImageSource() {
}
std::string ImageSource::GetSource() const {
return chrome::kChromeOSAssetHost;
}
void ImageSource::StartDataRequest(
const std::string& path,
int render_process_id,
int render_frame_id,
const content::URLDataSource::GotDataCallback& got_data_callback) {
if (!IsWhitelisted(path)) {
got_data_callback.Run(NULL);
return;
}
if (!image_loader_) {
image_loader_ = new UserImageLoader(ImageDecoder::DEFAULT_CODEC,
task_runner_);
}
content::BrowserThread::GetBlockingPool()->PostTask(
FROM_HERE,
base::Bind(&StartOnBlockingPool,
path,
image_loader_,
got_data_callback,
base::ThreadTaskRunnerHandle::Get()));
}
std::string ImageSource::GetMimeType(const std::string& path) const {
std::string mime_type;
std::string ext = base::FilePath(path).Extension();
if (!ext.empty())
net::GetWellKnownMimeTypeFromExtension(ext.substr(1), &mime_type);
return mime_type;
}
bool ImageSource::IsWhitelisted(const std::string& path) const {
base::FilePath file_path(path);
if (file_path.ReferencesParent())
return false;
// Check if the path starts with a whitelisted directory.
std::vector<std::string> components;
file_path.GetComponents(&components);
if (components.empty())
return false;
for (size_t i = 0; i < arraysize(kWhitelistedDirectories); i++) {
if (components[0] == kWhitelistedDirectories[i])
return true;
}
return false;
}
} // namespace chromeos
|
