Universial component install for chrome os.

chrome/browser/component_updater/cros_component_installer.cc

+// Copyright 2017 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/component_updater/cros_component_installer.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/component_updater/component_installer_errors.h"
+#include "components/component_updater/component_updater_paths.h"
+#include "content/public/browser/browser_thread.h"
+
+#if defined(OS_CHROMEOS)

[Sorin Jianu, 2017/03/02 22:27:25]

#include "build/build_config.h"

[xiaochu, 2017/03/03 00:08:53]

Done.

+#include "chromeos/dbus/dbus_method_call_status.h"
+#include "chromeos/dbus/dbus_thread_manager.h"
+#include "chromeos/dbus/image_loader_client.h"
+#endif  // defined(OS_CHROMEOS)
+
+using content::BrowserThread;
+
+namespace component_updater {
+
+#if defined(OS_CHROMEOS)
+void LogRegistrationResult(chromeos::DBusMethodCallStatus call_status,
+                           bool result) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+  if (call_status != chromeos::DBUS_METHOD_CALL_SUCCESS) {
+    DVLOG(1) << "Call to imageloader service failed.";
+    return;
+  }
+  if (!result) {
+    DVLOG(1) << "Component registration failed";
+    return;
+  }
+}
+void ImageLoaderRegistration(const std::string& version,
+                             const base::FilePath& install_dir,
+                             const std::string& name) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+  chromeos::ImageLoaderClient* loader =
+      chromeos::DBusThreadManager::Get()->GetImageLoaderClient();
+
+  if (loader) {
+    loader->RegisterComponent(name, version, install_dir.value(),
+                              base::Bind(&LogRegistrationResult));
+  } else {
+    DVLOG(1) << "Failed to get ImageLoaderClient object.";
+  }
+}
+
+CrOSComponentInstallerTraits::CrOSComponentInstallerTraits(
+    std::string dir_name,
+    std::string name,
+    std::string sha2HashStr)
+    : dir_name(dir_name), name(name) {
+  if (sha2HashStr.length() != 64)
+    return;
+  for (unsigned int i = 0; i < sizeof(kSha2Hash_); i++) {

[Sorin Jianu, 2017/03/02 22:27:24]

use size_t for the loop variable. Also I wonder if you could use a for range
construct to iterate through the array. Not sure how this compiles kSha2Hash_[i]
= stoul() without truncation warnings.

[xiaochu, 2017/03/03 00:08:53]

Done.

+    kSha2Hash_[i] = stoul(sha2HashStr.substr(i * 2, 2), nullptr, 16);
+  }
+}
+
+bool CrOSComponentInstallerTraits::SupportsGroupPolicyEnabledComponentUpdates()
+    const {
+  return true;
+}
+
+bool CrOSComponentInstallerTraits::RequiresNetworkEncryption() const {
+  return true;
+}
+
+update_client::CrxInstaller::Result
+CrOSComponentInstallerTraits::OnCustomInstall(
+    const base::DictionaryValue& manifest,
+    const base::FilePath& install_dir) {
+  DVLOG(1) << "[CrOSComponentInstallerTraits::OnCustomInstall]";
+  std::string version;
+  if (!manifest.GetString("version", &version)) {
+    return ToInstallerResult(update_client::InstallError::GENERIC_ERROR);
+  }
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      base::Bind(&ImageLoaderRegistration, version, install_dir, name));
+  return update_client::CrxInstaller::Result(update_client::InstallError::NONE);
+}
+
+void CrOSComponentInstallerTraits::ComponentReady(
+    const base::Version& version,
+    const base::FilePath& path,
+    std::unique_ptr<base::DictionaryValue> manifest) {}
+
+bool CrOSComponentInstallerTraits::VerifyInstallation(
+    const base::DictionaryValue& manifest,
+    const base::FilePath& install_dir) const {
+  return true;
+}
+
+base::FilePath CrOSComponentInstallerTraits::GetRelativeInstallDir() const {
+  return base::FilePath(dir_name);
+}
+
+void CrOSComponentInstallerTraits::GetHash(std::vector<uint8_t>* hash) const {
+  hash->assign(kSha2Hash_, kSha2Hash_ + arraysize(kSha2Hash_));
+}
+
+std::string CrOSComponentInstallerTraits::GetName() const {
+  return name;
+}
+
+update_client::InstallerAttributes
+CrOSComponentInstallerTraits::GetInstallerAttributes() const {
+  return update_client::InstallerAttributes();
+}
+
+std::vector<std::string> CrOSComponentInstallerTraits::GetMimeTypes() const {
+  std::vector<std::string> mime_types;
+  return mime_types;
+}
+
+void RegisterCrOSComponentInternal(ComponentUpdateService* cus,
+                                   const ComponentConfig& config) {
+  std::unique_ptr<ComponentInstallerTraits> traits(
+      new CrOSComponentInstallerTraits(config.dir, config.name,
+                                       config.sha2hashstr));
+  // |cus| will take ownership of |installer| during
+  // installer->Register(cus).
+  DefaultComponentInstaller* installer =
+      new DefaultComponentInstaller(std::move(traits));
+  installer->Register(cus, base::Closure());
+}
+
+bool RegisterCrOSComponentInternal(ComponentUpdateService* cus,
+                                   const std::string& name) {
+  if (name.length() == 0) {

[Sorin Jianu, 2017/03/02 22:27:24]

use empty()

[xiaochu, 2017/03/03 00:08:52]

Done.

+    DVLOG(1) << "[RegisterCrOSComponents] name is empty.";
+    return false;
+  }
+  const std::map<std::string, std::map<std::string, std::string>> components = {
+      {"escpr",
+       {{"dir", "epson-inkjet-printer-escpr"},
+        {"sha2hashstr",
+         "1913a5e0a6cad30b6f03e176177e0d7ed62c5d6700a9c66da556d7c3f5d6a47e"}}}};
+  auto component = components.find(name);

[Sorin Jianu, 2017/03/02 22:27:25]

std::map has this odd api, where find returns an iterator to its value type,
which in this case is a pair.

The code reads better is we say:

const auto it = components.find(name);
if (it != components.end()) {
  ....

[xiaochu, 2017/03/03 00:08:52]

Done.

+  if (component != components.end()) {

[Sorin Jianu, 2017/03/02 22:27:25]

Some people prefer to handle the early exit from functions sooner. It makes the
code easier to read, arguably.

if (it == components.end()) {
  DVLOG...;
  return false;
}

....
return true;

[xiaochu, 2017/03/03 00:08:53]

Done.

+    ComponentConfig config;
+    config.name = component->first;
+    config.dir = component->second.find("dir")->second;
+    config.sha2hashstr = component->second.find("sha2hashstr")->second;
+    RegisterCrOSComponentInternal(cus, config);
+    return true;
+  }
+  DVLOG(1) << "[RegisterCrOSComponents] component " << name
+           << " is not in configuration.";
+  return false;
+}
+
+#endif  // defined(OS_CHROMEOS)
+
+bool RegisterCrOSComponent(ComponentUpdateService* cus,
+                           const std::string& name) {
+#if defined(OS_CHROMEOS)
+  return RegisterCrOSComponentInternal(cus, name);
+#else
+  return false;
+#endif  // defined(OS_CHROMEOS)
+}
+}  // namespace component_updater