1// Copyright (c) 2012 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#include "chrome/browser/component_updater/swiftshader_component_installer.h"
6
7#include <stdint.h>
8#include <string>
9#include <vector>
10
11#include "base/base_paths.h"
12#include "base/bind.h"
13#include "base/compiler_specific.h"
14#include "base/files/file_enumerator.h"
15#include "base/files/file_path.h"
16#include "base/files/file_util.h"
17#include "base/logging.h"
18#include "base/path_service.h"
19#include "base/strings/string_util.h"
20#include "base/values.h"
21#include "components/component_updater/component_updater_paths.h"
22#include "components/component_updater/component_updater_service.h"
23#include "content/public/browser/browser_thread.h"
24#include "content/public/browser/gpu_data_manager.h"
25#include "content/public/browser/gpu_data_manager_observer.h"
26#include "gpu/config/gpu_feature_type.h"
27
28using content::BrowserThread;
29using content::GpuDataManager;
30
31namespace component_updater {
32
33namespace {
34
35// CRX hash. The extension id is: nhfgdggnnopgbfdlpeoalgcjdgfafocg.
36const uint8_t kSha2Hash[] = {0xd7, 0x56, 0x36, 0x6d, 0xde, 0xf6, 0x15, 0x3b,
37                             0xf4, 0xe0, 0xb6, 0x29, 0x36, 0x50, 0x5e, 0x26,
38                             0xbd, 0x77, 0x8b, 0x8e, 0x35, 0xc2, 0x7e, 0x43,
39                             0x52, 0x47, 0x62, 0xed, 0x12, 0xca, 0xcc, 0x6a};
40
41// File name of the internal SwiftShader plugin on different platforms.
42const base::FilePath::CharType kSwiftShaderEglName[] =
43    FILE_PATH_LITERAL("libegl.dll");
44const base::FilePath::CharType kSwiftShaderGlesName[] =
45    FILE_PATH_LITERAL("libglesv2.dll");
46
47const char kSwiftShaderManifestName[] = "SwiftShader";
48
49// If we don't have a SwiftShader component, this is the version we claim.
50const char kNullVersion[] = "0.0.0.0";
51
52// The base directory on windows looks like:
53// <profile>\AppData\Local\Google\Chrome\User Data\SwiftShader\.
54base::FilePath GetSwiftShaderBaseDirectory() {
55  base::FilePath result;
56  if (!PathService::Get(DIR_SWIFT_SHADER, &result))
57    NOTREACHED() << "Couldn't get SwiftShader directory.";
58  return result;
59}
60
61// SwiftShader has version encoded in the path itself
62// so we need to enumerate the directories to find the full path.
63// On success it returns something like:
64// <profile>\AppData\Local\Google\Chrome\User Data\SwiftShader\10.3.44.555\.
65bool GetLatestSwiftShaderDirectory(base::FilePath* result,
66                                   Version* latest,
67                                   std::vector<base::FilePath>* older_dirs) {
68  base::FilePath base_dir = GetSwiftShaderBaseDirectory();
69  bool found = false;
70  base::FileEnumerator file_enumerator(
71      base_dir, false, base::FileEnumerator::DIRECTORIES);
72  for (base::FilePath path = file_enumerator.Next(); !path.value().empty();
73       path = file_enumerator.Next()) {
74    Version version(path.BaseName().MaybeAsASCII());
75    if (!version.IsValid())
76      continue;
77    if (version.CompareTo(*latest) > 0 &&
78        base::PathExists(path.Append(kSwiftShaderEglName)) &&
79        base::PathExists(path.Append(kSwiftShaderGlesName))) {
80      if (found && older_dirs)
81        older_dirs->push_back(*result);
82      *latest = version;
83      *result = path;
84      found = true;
85    } else {
86      if (older_dirs)
87        older_dirs->push_back(path);
88    }
89  }
90  return found;
91}
92
93void RegisterSwiftShaderWithChrome(const base::FilePath& path) {
94  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
95  GpuDataManager::GetInstance()->RegisterSwiftShaderPath(path);
96}
97
98class SwiftShaderComponentInstaller : public ComponentInstaller {
99 public:
100  explicit SwiftShaderComponentInstaller(const Version& version);
101
102  virtual ~SwiftShaderComponentInstaller() {}
103
104  virtual void OnUpdateError(int error) OVERRIDE;
105
106  virtual bool Install(const base::DictionaryValue& manifest,
107                       const base::FilePath& unpack_path) OVERRIDE;
108
109  virtual bool GetInstalledFile(const std::string& file,
110                                base::FilePath* installed_file) OVERRIDE;
111
112 private:
113  Version current_version_;
114};
115
116SwiftShaderComponentInstaller::SwiftShaderComponentInstaller(
117    const Version& version)
118    : current_version_(version) {
119  DCHECK(version.IsValid());
120}
121
122void SwiftShaderComponentInstaller::OnUpdateError(int error) {
123  NOTREACHED() << "SwiftShader update error: " << error;
124}
125
126bool SwiftShaderComponentInstaller::Install(
127    const base::DictionaryValue& manifest,
128    const base::FilePath& unpack_path) {
129  std::string name;
130  manifest.GetStringASCII("name", &name);
131  if (name != kSwiftShaderManifestName)
132    return false;
133  std::string proposed_version;
134  manifest.GetStringASCII("version", &proposed_version);
135  Version version(proposed_version.c_str());
136  if (!version.IsValid())
137    return false;
138  if (current_version_.CompareTo(version) >= 0)
139    return false;
140  if (!base::PathExists(unpack_path.Append(kSwiftShaderEglName)) ||
141      !base::PathExists(unpack_path.Append(kSwiftShaderGlesName)))
142    return false;
143  // Passed the basic tests. Time to install it.
144  base::FilePath path =
145      GetSwiftShaderBaseDirectory().AppendASCII(version.GetString());
146  if (base::PathExists(path))
147    return false;
148  if (!base::Move(unpack_path, path))
149    return false;
150  // Installation is done. Now tell the rest of chrome.
151  current_version_ = version;
152  BrowserThread::PostTask(BrowserThread::UI,
153                          FROM_HERE,
154                          base::Bind(&RegisterSwiftShaderWithChrome, path));
155  return true;
156}
157
158bool SwiftShaderComponentInstaller::GetInstalledFile(
159    const std::string& file,
160    base::FilePath* installed_file) {
161  return false;
162}
163
164void FinishSwiftShaderUpdateRegistration(ComponentUpdateService* cus,
165                                         const Version& version) {
166  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
167
168  CrxComponent swiftshader;
169  swiftshader.name = "Swift Shader";
170  swiftshader.installer = new SwiftShaderComponentInstaller(version);
171  swiftshader.version = version;
172  swiftshader.pk_hash.assign(kSha2Hash, &kSha2Hash[sizeof(kSha2Hash)]);
173  if (cus->RegisterComponent(swiftshader) != ComponentUpdateService::kOk) {
174    NOTREACHED() << "SwiftShader component registration fail";
175  }
176}
177
178class UpdateChecker : public content::GpuDataManagerObserver {
179 public:
180  explicit UpdateChecker(ComponentUpdateService* cus);
181
182  virtual void OnGpuInfoUpdate() OVERRIDE;
183
184 private:
185  ComponentUpdateService* cus_;
186};
187
188UpdateChecker::UpdateChecker(ComponentUpdateService* cus) : cus_(cus) {
189}
190
191void UpdateChecker::OnGpuInfoUpdate() {
192  GpuDataManager* gpu_data_manager = GpuDataManager::GetInstance();
193
194  if (!gpu_data_manager->GpuAccessAllowed(NULL) ||
195      gpu_data_manager->IsFeatureBlacklisted(gpu::GPU_FEATURE_TYPE_WEBGL) ||
196      gpu_data_manager->ShouldUseSwiftShader()) {
197    gpu_data_manager->RemoveObserver(this);
198    DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
199    base::FilePath path = GetSwiftShaderBaseDirectory();
200
201    Version version(kNullVersion);
202    GetLatestSwiftShaderDirectory(&path, &version, NULL);
203
204    BrowserThread::PostTask(
205        BrowserThread::UI,
206        FROM_HERE,
207        base::Bind(&FinishSwiftShaderUpdateRegistration, cus_, version));
208  }
209}
210
211#if defined(ENABLE_SWIFTSHADER)
212
213// Check if there already is a version of swiftshader installed,
214// and if so register it.
215void RegisterSwiftShaderPath(ComponentUpdateService* cus) {
216  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
217  base::FilePath path = GetSwiftShaderBaseDirectory();
218  if (!base::PathExists(path)) {
219    if (!base::CreateDirectory(path)) {
220      NOTREACHED() << "Could not create SwiftShader directory.";
221      return;
222    }
223  }
224
225  Version version(kNullVersion);
226  std::vector<base::FilePath> older_dirs;
227  if (GetLatestSwiftShaderDirectory(&path, &version, &older_dirs))
228    BrowserThread::PostTask(BrowserThread::UI,
229                            FROM_HERE,
230                            base::Bind(&RegisterSwiftShaderWithChrome, path));
231
232  UpdateChecker* update_checker = new UpdateChecker(cus);
233  GpuDataManager::GetInstance()->AddObserver(update_checker);
234  update_checker->OnGpuInfoUpdate();
235  // We leak update_checker here, because it has to stick around for the life
236  // of the GpuDataManager.
237
238  // Remove older versions of SwiftShader.
239  for (std::vector<base::FilePath>::iterator iter = older_dirs.begin();
240       iter != older_dirs.end();
241       ++iter) {
242    base::DeleteFile(*iter, true);
243  }
244}
245
246#endif  // ENABLE_SWIFTSHADER
247
248}  // namespace
249
250void RegisterSwiftShaderComponent(ComponentUpdateService* cus) {
251#if defined(ENABLE_SWIFTSHADER)
252  BrowserThread::PostTask(BrowserThread::FILE,
253                          FROM_HERE,
254                          base::Bind(&RegisterSwiftShaderPath, cus));
255#endif
256}
257
258}  // namespace component_updater
259