omaha_request_prep_action.cc

// Copyright (c) 2009 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 "update_engine/omaha_request_prep_action.h"
#include <sys/utsname.h>
#include <errno.h>
#include <map>
#include <string>
#include "base/string_util.h"
#include "update_engine/simple_key_value_store.h"
#include "update_engine/utils.h"

using std::map;
using std::string;

// This gathers local system information and prepares info used by the
// update check action.

namespace {
const string OmahaIdPath() {
  return string(chromeos_update_engine::utils::kStatefulPartition) +
      "/etc/omaha_id";
}
}  // namespace {}

namespace chromeos_update_engine {

void OmahaRequestPrepAction::PerformAction() {
  // TODO(adlr): honor force_full_update_
  ScopedActionCompleter completer(processor_, this);
  string machine_id;
  TEST_AND_RETURN(GetMachineId(&machine_id));
  const string version(GetLsbValue("CHROMEOS_RELEASE_VERSION", ""));
  const string sp(version + "_" + GetMachineType());
  const string track(GetLsbValue("CHROMEOS_RELEASE_TRACK", ""));
  const string update_url(GetLsbValue("CHROMEOS_AUSERVER",
                                      UpdateCheckParams::kUpdateUrl));
  const string board(GetLsbValue("CHROMEOS_RELEASE_BOARD", ""));

  UpdateCheckParams out(machine_id,  // machine_id
                        machine_id,  // user_id (use machine_id)
                        UpdateCheckParams::kOsPlatform,
                        UpdateCheckParams::kOsVersion,
                        sp,  // e.g. 0.2.3.3_i686
                        board,  // e.g. x86-generic
                        UpdateCheckParams::kAppId,
                        version,  // app version (from lsb-release)
                        "en-US",  // lang
                        track,  // track
                        update_url);

  CHECK(HasOutputPipe());
  SetOutputObject(out);
  completer.set_success(true);
}

namespace {
const size_t kGuidDataByteLength = 128 / 8;
const string::size_type kGuidStringLength = 38;
// Formats 16 bytes (128 bits) of data as a GUID:
// "{XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX}" where X is a hex digit
string GuidFromData(const unsigned char data[kGuidDataByteLength]) {
  return StringPrintf(
      "{%02X%02X%02X%02X-%02X%02X-%02X%02X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
      data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7],
      data[8], data[9], data[10], data[11], data[12], data[13], data[14],
      data[15]);
}
}

// Returns true on success.
bool OmahaRequestPrepAction::GetMachineId(std::string* out_id) const {
  // See if we have an existing Machine ID
  const string omaha_id_path = root_ + OmahaIdPath();
  
  if (utils::ReadFileToString(omaha_id_path, out_id) &&
      out_id->size() == kGuidStringLength) {
    return true;
  }
  
  // Create a new ID
  int rand_fd = open("/dev/urandom", O_RDONLY, 0);
  TEST_AND_RETURN_FALSE_ERRNO(rand_fd >= 0);
  ScopedFdCloser rand_fd_closer(&rand_fd);
  unsigned char buf[kGuidDataByteLength];
  size_t bytes_read = 0;
  while (bytes_read < sizeof(buf)) {
    ssize_t rc = read(rand_fd, buf + bytes_read, sizeof(buf) - bytes_read);
    TEST_AND_RETURN_FALSE_ERRNO(rc > 0);
    bytes_read += rc;
  }
  string guid = GuidFromData(buf);
  TEST_AND_RETURN_FALSE(
      utils::WriteFile(omaha_id_path.c_str(), guid.data(), guid.size()));
  *out_id = guid;
  return true;
}

string OmahaRequestPrepAction::GetLsbValue(
    const string& key, const string& default_value) const {
  string files[] = {string(utils::kStatefulPartition) + "/etc/lsb-release",
                    "/etc/lsb-release"};
  for (unsigned int i = 0; i < arraysize(files); i++) {
    // TODO(adlr): make sure files checked are owned as root (and all
    // their parents are recursively, too).
    string file_data;
    if (!utils::ReadFileToString(root_ + files[i], &file_data))
      continue;
    
    map<string, string> data = simple_key_value_store::ParseString(file_data);
    if (utils::MapContainsKey(data, key))
      return data[key];
  }
  // not found
  return default_value;
}

string OmahaRequestPrepAction::GetMachineType() const {
  struct utsname buf;
  string ret;
  if (uname(&buf) == 0)
    ret = buf.machine;
  return ret;
}

}  // namespace chromeos_update_engine