androidmanager.cc

#include <boost/algorithm/string/find.hpp>
#include <boost/filesystem.hpp>
#include <forward_list>

#include "androidmanager.h"
#include "packagemanagerfactory.h"

#include "utilities/utils.h"

#include <boost/phoenix/stl/algorithm/transformation.hpp>
#include <boost/spirit/include/phoenix_container.hpp>
#include <boost/spirit/include/phoenix_core.hpp>
#include <boost/spirit/include/qi.hpp>

namespace qi = boost::spirit::qi;
namespace fs = boost::filesystem;

const std::string AndroidManager::data_ota_package_dir_ = "/data/ota_package";

AUTO_REGISTER_PACKAGE_MANAGER(PACKAGE_MANAGER_ANDROID, AndroidManager);

Json::Value AndroidManager::getInstalledPackages() const {
  using boost::phoenix::copy;
  using qi::_1;
  using qi::char_;

  std::string pm_output;
  Json::Value packages(Json::arrayValue);
  if (0 != Utils::shell("pm list packages --show-versioncode", &pm_output)) {
    return packages;
  }

  qi::rule<std::string::iterator, std::vector<char>()> char_seq = qi::lexeme[*(char_ - ' ')];

  std::istringstream pv_lines(pm_output);
  for (std::string line; std::getline(pv_lines, line);) {
    std::string p;
    std::string v;
    if (qi::parse(line.begin(), line.end(),
                  ("package:" >> char_seq[copy(_1, std::back_inserter(p))] >> ' ' >> "versionCode:" >>
                   char_seq[copy(_1, std::back_inserter(v))]))) {
      Json::Value package;
      package["name"] = p;
      package["version"] = v;
      packages.append(package);
    }
  }
  return packages;
}

Uptane::Target AndroidManager::getCurrent() const {
  using boost::phoenix::push_front;
  using boost::spirit::ascii::xdigit;
  using qi::_1;

  std::string getprop_output;
  if (0 == Utils::shell("getprop ota.last_installed_package_file", &getprop_output)) {
    std::forward_list<char> hash;
    qi::phrase_parse(getprop_output.crbegin(), getprop_output.crend(),
                     *(xdigit[push_front(boost::phoenix::ref(hash), _1)]), boost::spirit::ascii::cntrl);
    std::vector<Uptane::Target> installed_versions;
    storage_->loadPrimaryInstallationLog(&installed_versions, false);
    for (const auto& target : installed_versions) {
      if (std::equal(hash.cbegin(), hash.cend(), target.sha256Hash().cbegin())) {
        return target;
      }
    }
  }
  return Uptane::Target::Unknown();
}

data::InstallationResult AndroidManager::install(const Uptane::Target& target) const {
  LOG_INFO << "Begin Android package installation";
  auto package_filename = (fs::path(data_ota_package_dir_) / target.filename()).string() + "." + target.sha256Hash();
  std::ofstream package_file(package_filename.c_str());
  if (!package_file.good()) {
    throw std::runtime_error(std::string("Error opening file ") + package_filename);
  }
  package_file << *storage_->openTargetFile(target);

  if (bootloader_ != nullptr) {
    bootloader_->rebootFlagSet();
  }
  LOG_INFO << "Performing sync()";
  sync();
  return data::InstallationResult(data::ResultCode::Numeric::kNeedCompletion, "need reboot");
}

data::InstallationResult AndroidManager::finalizeInstall(const Uptane::Target& target) {
  std::string ota_package_file_path = GetOTAPackageFilePath(target.sha256Hash());
  if (!ota_package_file_path.empty()) {
    fs::remove(ota_package_file_path);
  }
  std::string errorMessage{"n/a"};
  if (installationAborted(&errorMessage)) {
    return data::InstallationResult(data::ResultCode::Numeric::kInstallFailed, errorMessage);
  }
  return data::InstallationResult(data::ResultCode::Numeric::kOk, "package installation successfully finalized");
}

std::string AndroidManager::GetOTAPackageFilePath(const std::string& hash) {
  fs::directory_iterator entryItEnd;
  fs::directory_iterator entryIt{fs::path(data_ota_package_dir_)};
  for (; entryIt != entryItEnd; ++entryIt) {
    auto& entry_path = entryIt->path();
    if (boost::filesystem::is_directory(*entryIt)) {
      continue;
    }
    auto ext = entry_path.extension().string();
    ext = ext.substr(1);
    if (ext == hash) {
      return entry_path.string();
    }
  }
  return std::string{};
}

bool AndroidManager::installationAborted(std::string* errorMessage) const {
  std::string installation_log_file{"/cache/recovery/last_install"};
  std::ifstream log(installation_log_file.c_str());
  if (!log.good()) {
    throw std::runtime_error(std::string("Error opening file ") + installation_log_file);
  }
  for (std::string line; std::getline(log, line);) {
    if (!boost::algorithm::find_first(line, "error:").empty()) {
      int error_code = std::stoi(line.substr(6));
      if (error_code != 0) {
        *errorMessage = std::string("Error code: ") + std::to_string(error_code);
        return true;
      }
    }
  }
  return false;
}