blob: c5d3e052b0bf3e4fa3d8ff343f22df3b5d644b2e [file] [log] [blame]
// Copyright (c) 2012 The Chromium OS 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/utils.h"
#include <attr/xattr.h>
#include <sys/mount.h>
#include <sys/resource.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <algorithm>
#include <vector>
#include <base/file_path.h>
#include <base/file_util.h>
#include <base/logging.h>
#include <base/posix/eintr_wrapper.h>
#include <base/rand_util.h>
#include <base/string_number_conversions.h>
#include <base/string_split.h>
#include <base/string_util.h>
#include <base/stringprintf.h>
#include <glib.h>
#include <google/protobuf/stubs/common.h>
#include "update_engine/constants.h"
#include "update_engine/file_writer.h"
#include "update_engine/omaha_request_params.h"
#include "update_engine/subprocess.h"
#include "update_engine/system_state.h"
#include "update_engine/update_attempter.h"
using base::Time;
using base::TimeDelta;
using std::min;
using std::pair;
using std::string;
using std::vector;
namespace chromeos_update_engine {
namespace {
// The following constants control how UnmountFilesystem should retry if
// umount() fails with an errno EBUSY, i.e. retry 5 times over the course of
// one second.
const int kUnmountMaxNumOfRetries = 5;
const int kUnmountRetryIntervalInMicroseconds = 200 * 1000; // 200 ms
} // namespace
namespace utils {
static const char kDevImageMarker[] = "/root/.dev_mode";
// Cgroup container is created in update-engine's upstart script located at
// /etc/init/update-engine.conf.
static const char kCGroupDir[] = "/sys/fs/cgroup/cpu/update-engine";
bool IsOfficialBuild() {
return !file_util::PathExists(FilePath(kDevImageMarker));
}
bool IsNormalBootMode() {
// TODO(petkov): Convert to a library call once a crossystem library is
// available (crosbug.com/13291).
int exit_code = 0;
vector<string> cmd(1, "/usr/bin/crossystem");
cmd.push_back("devsw_boot?1");
// Assume dev mode if the dev switch is set to 1 and there was no error
// executing crossystem. Assume normal mode otherwise.
bool success = Subprocess::SynchronousExec(cmd, &exit_code, NULL);
bool dev_mode = success && exit_code == 0;
LOG_IF(INFO, dev_mode) << "Booted in dev mode.";
return !dev_mode;
}
string ReadValueFromCrosSystem(const string& key){
int exit_code = 0;
vector<string> cmd(1, "/usr/bin/crossystem");
cmd.push_back(key);
string return_value;
bool success = Subprocess::SynchronousExec(cmd, &exit_code, &return_value);
if (success && !exit_code) {
TrimWhitespaceASCII(return_value, TRIM_ALL, &return_value);
return return_value;
}
LOG(ERROR) << "Unable to read " << key << " (" << exit_code << ") "
<< return_value;
return "";
}
string GetHardwareClass() {
return ReadValueFromCrosSystem("hwid");
}
string GetFirmwareVersion() {
return ReadValueFromCrosSystem("fwid");
}
string GetECVersion(const char* input_line) {
string line;
if(input_line == NULL) {
int exit_code = 0;
vector<string> cmd(1, "/usr/sbin/mosys");
cmd.push_back("-k");
cmd.push_back("ec");
cmd.push_back("info");
bool success = Subprocess::SynchronousExec(cmd, &exit_code, &line);
if (!success || exit_code) {
LOG(ERROR) << "Unable to read ec info from mosys (" << exit_code << ")";
return "";
}
} else {
line = input_line;
}
TrimWhitespaceASCII(line, TRIM_ALL, &line);
// At this point we want to conver the format key=value pair from mosys to
// a vector of key value pairs.
vector<pair<string, string> > kv_pairs;
if (base::SplitStringIntoKeyValuePairs(line, '=', ' ', &kv_pairs)) {
for (vector<pair<string, string> >::iterator it = kv_pairs.begin();
it != kv_pairs.end(); ++it) {
// Finally match against the fw_verion which may have quotes.
if (it->first == "fw_version") {
string output;
// Trim any quotes.
TrimString(it->second, "\"", &output);
return output;
}
}
}
LOG(ERROR) << "Unable to parse fwid from ec info.";
return "";
}
bool WriteFile(const char* path, const char* data, int data_len) {
DirectFileWriter writer;
TEST_AND_RETURN_FALSE_ERRNO(0 == writer.Open(path,
O_WRONLY | O_CREAT | O_TRUNC,
0600));
ScopedFileWriterCloser closer(&writer);
TEST_AND_RETURN_FALSE_ERRNO(writer.Write(data, data_len));
return true;
}
bool WriteAll(int fd, const void* buf, size_t count) {
const char* c_buf = static_cast<const char*>(buf);
ssize_t bytes_written = 0;
while (bytes_written < static_cast<ssize_t>(count)) {
ssize_t rc = write(fd, c_buf + bytes_written, count - bytes_written);
TEST_AND_RETURN_FALSE_ERRNO(rc >= 0);
bytes_written += rc;
}
return true;
}
bool PWriteAll(int fd, const void* buf, size_t count, off_t offset) {
const char* c_buf = static_cast<const char*>(buf);
size_t bytes_written = 0;
int num_attempts = 0;
while (bytes_written < count) {
num_attempts++;
ssize_t rc = pwrite(fd, c_buf + bytes_written, count - bytes_written,
offset + bytes_written);
// TODO(garnold) for debugging failure in chromium-os:31077; to be removed.
if (rc < 0) {
PLOG(ERROR) << "pwrite error; num_attempts=" << num_attempts
<< " bytes_written=" << bytes_written
<< " count=" << count << " offset=" << offset;
}
TEST_AND_RETURN_FALSE_ERRNO(rc >= 0);
bytes_written += rc;
}
return true;
}
bool PReadAll(int fd, void* buf, size_t count, off_t offset,
ssize_t* out_bytes_read) {
char* c_buf = static_cast<char*>(buf);
ssize_t bytes_read = 0;
while (bytes_read < static_cast<ssize_t>(count)) {
ssize_t rc = pread(fd, c_buf + bytes_read, count - bytes_read,
offset + bytes_read);
TEST_AND_RETURN_FALSE_ERRNO(rc >= 0);
if (rc == 0) {
break;
}
bytes_read += rc;
}
*out_bytes_read = bytes_read;
return true;
}
// Append |nbytes| of content from |buf| to the vector pointed to by either
// |vec_p| or |str_p|.
static void AppendBytes(const char* buf, size_t nbytes,
std::vector<char>* vec_p) {
CHECK(buf);
CHECK(vec_p);
vec_p->insert(vec_p->end(), buf, buf + nbytes);
}
static void AppendBytes(const char* buf, size_t nbytes,
std::string* str_p) {
CHECK(buf);
CHECK(str_p);
str_p->append(buf, nbytes);
}
// Reads from an open file |fp|, appending the read content to the container
// pointer to by |out_p|. Returns true upon successful reading all of the
// file's content, false otherwise. If |size| is not -1, reads up to |size|
// bytes.
template <class T>
static bool Read(FILE* fp, off_t size, T* out_p) {
CHECK(fp);
CHECK(size == -1 || size >= 0);
char buf[1024];
while (size == -1 || size > 0) {
off_t bytes_to_read = sizeof(buf);
if (size > 0 && bytes_to_read > size) {
bytes_to_read = size;
}
size_t nbytes = fread(buf, 1, bytes_to_read, fp);
if (!nbytes) {
break;
}
AppendBytes(buf, nbytes, out_p);
if (size != -1) {
CHECK(size >= static_cast<off_t>(nbytes));
size -= nbytes;
}
}
if (ferror(fp)) {
return false;
}
return size == 0 || feof(fp);
}
// Opens a file |path| for reading and appends its the contents to a container
// |out_p|. Starts reading the file from |offset|. If |offset| is beyond the end
// of the file, returns success. If |size| is not -1, reads up to |size| bytes.
template <class T>
static bool ReadFileChunkAndAppend(const std::string& path,
off_t offset,
off_t size,
T* out_p) {
CHECK_GE(offset, 0);
CHECK(size == -1 || size >= 0);
file_util::ScopedFILE fp(fopen(path.c_str(), "r"));
if (!fp.get())
return false;
if (offset) {
// Return success without appending any data if a chunk beyond the end of
// the file is requested.
if (offset >= FileSize(path)) {
return true;
}
TEST_AND_RETURN_FALSE_ERRNO(fseek(fp.get(), offset, SEEK_SET) == 0);
}
return Read(fp.get(), size, out_p);
}
// Invokes a pipe |cmd|, then uses |append_func| to append its stdout to a
// container |out_p|.
template <class T>
static bool ReadPipeAndAppend(const std::string& cmd, T* out_p) {
FILE* fp = popen(cmd.c_str(), "r");
if (!fp)
return false;
bool success = Read(fp, -1, out_p);
return (success && pclose(fp) >= 0);
}
bool ReadFile(const string& path, vector<char>* out_p) {
return ReadFileChunkAndAppend(path, 0, -1, out_p);
}
bool ReadFile(const string& path, string* out_p) {
return ReadFileChunkAndAppend(path, 0, -1, out_p);
}
bool ReadFileChunk(const string& path, off_t offset, off_t size,
vector<char>* out_p) {
return ReadFileChunkAndAppend(path, offset, size, out_p);
}
bool ReadPipe(const string& cmd, vector<char>* out_p) {
return ReadPipeAndAppend(cmd, out_p);
}
bool ReadPipe(const string& cmd, string* out_p) {
return ReadPipeAndAppend(cmd, out_p);
}
off_t FileSize(const string& path) {
struct stat stbuf;
int rc = stat(path.c_str(), &stbuf);
CHECK_EQ(rc, 0);
if (rc < 0)
return rc;
return stbuf.st_size;
}
void HexDumpArray(const unsigned char* const arr, const size_t length) {
const unsigned char* const char_arr =
reinterpret_cast<const unsigned char* const>(arr);
LOG(INFO) << "Logging array of length: " << length;
const unsigned int bytes_per_line = 16;
for (uint32_t i = 0; i < length; i += bytes_per_line) {
const unsigned int bytes_remaining = length - i;
const unsigned int bytes_per_this_line = min(bytes_per_line,
bytes_remaining);
char header[100];
int r = snprintf(header, sizeof(header), "0x%08x : ", i);
TEST_AND_RETURN(r == 13);
string line = header;
for (unsigned int j = 0; j < bytes_per_this_line; j++) {
char buf[20];
unsigned char c = char_arr[i + j];
r = snprintf(buf, sizeof(buf), "%02x ", static_cast<unsigned int>(c));
TEST_AND_RETURN(r == 3);
line += buf;
}
LOG(INFO) << line;
}
}
namespace {
class ScopedDirCloser {
public:
explicit ScopedDirCloser(DIR** dir) : dir_(dir) {}
~ScopedDirCloser() {
if (dir_ && *dir_) {
int r = closedir(*dir_);
TEST_AND_RETURN_ERRNO(r == 0);
*dir_ = NULL;
dir_ = NULL;
}
}
private:
DIR** dir_;
};
} // namespace {}
bool RecursiveUnlinkDir(const std::string& path) {
struct stat stbuf;
int r = lstat(path.c_str(), &stbuf);
TEST_AND_RETURN_FALSE_ERRNO((r == 0) || (errno == ENOENT));
if ((r < 0) && (errno == ENOENT))
// path request is missing. that's fine.
return true;
if (!S_ISDIR(stbuf.st_mode)) {
TEST_AND_RETURN_FALSE_ERRNO((unlink(path.c_str()) == 0) ||
(errno == ENOENT));
// success or path disappeared before we could unlink.
return true;
}
{
// We have a dir, unlink all children, then delete dir
DIR *dir = opendir(path.c_str());
TEST_AND_RETURN_FALSE_ERRNO(dir);
ScopedDirCloser dir_closer(&dir);
struct dirent dir_entry;
struct dirent *dir_entry_p;
int err = 0;
while ((err = readdir_r(dir, &dir_entry, &dir_entry_p)) == 0) {
if (dir_entry_p == NULL) {
// end of stream reached
break;
}
// Skip . and ..
if (!strcmp(dir_entry_p->d_name, ".") ||
!strcmp(dir_entry_p->d_name, ".."))
continue;
TEST_AND_RETURN_FALSE(RecursiveUnlinkDir(path + "/" +
dir_entry_p->d_name));
}
TEST_AND_RETURN_FALSE(err == 0);
}
// unlink dir
TEST_AND_RETURN_FALSE_ERRNO((rmdir(path.c_str()) == 0) || (errno == ENOENT));
return true;
}
string RootDevice(const string& partition_device) {
FilePath device_path(partition_device);
if (device_path.DirName().value() != "/dev") {
return "";
}
string::const_iterator it = --partition_device.end();
for (; it >= partition_device.begin(); --it) {
if (!isdigit(*it))
break;
}
// Some devices contain a p before the partitions. For example:
// /dev/mmc0p4 should be shortened to /dev/mmc0.
if (*it == 'p')
--it;
return string(partition_device.begin(), it + 1);
}
string PartitionNumber(const string& partition_device) {
CHECK(!partition_device.empty());
string::const_iterator it = --partition_device.end();
for (; it >= partition_device.begin(); --it) {
if (!isdigit(*it))
break;
}
return string(it + 1, partition_device.end());
}
string SysfsBlockDevice(const string& device) {
FilePath device_path(device);
if (device_path.DirName().value() != "/dev") {
return "";
}
return FilePath("/sys/block").Append(device_path.BaseName()).value();
}
bool IsRemovableDevice(const std::string& device) {
string sysfs_block = SysfsBlockDevice(device);
string removable;
if (sysfs_block.empty() ||
!file_util::ReadFileToString(FilePath(sysfs_block).Append("removable"),
&removable)) {
return false;
}
TrimWhitespaceASCII(removable, TRIM_ALL, &removable);
return removable == "1";
}
std::string ErrnoNumberAsString(int err) {
char buf[100];
buf[0] = '\0';
return strerror_r(err, buf, sizeof(buf));
}
std::string NormalizePath(const std::string& path, bool strip_trailing_slash) {
string ret;
bool last_insert_was_slash = false;
for (string::const_iterator it = path.begin(); it != path.end(); ++it) {
if (*it == '/') {
if (last_insert_was_slash)
continue;
last_insert_was_slash = true;
} else {
last_insert_was_slash = false;
}
ret.push_back(*it);
}
if (strip_trailing_slash && last_insert_was_slash) {
string::size_type last_non_slash = ret.find_last_not_of('/');
if (last_non_slash != string::npos) {
ret.resize(last_non_slash + 1);
} else {
ret = "";
}
}
return ret;
}
bool FileExists(const char* path) {
struct stat stbuf;
return 0 == lstat(path, &stbuf);
}
bool IsSymlink(const char* path) {
struct stat stbuf;
return lstat(path, &stbuf) == 0 && S_ISLNK(stbuf.st_mode) != 0;
}
std::string TempFilename(string path) {
static const string suffix("XXXXXX");
CHECK(StringHasSuffix(path, suffix));
do {
string new_suffix;
for (unsigned int i = 0; i < suffix.size(); i++) {
int r = rand() % (26 * 2 + 10); // [a-zA-Z0-9]
if (r < 26)
new_suffix.append(1, 'a' + r);
else if (r < (26 * 2))
new_suffix.append(1, 'A' + r - 26);
else
new_suffix.append(1, '0' + r - (26 * 2));
}
CHECK_EQ(new_suffix.size(), suffix.size());
path.resize(path.size() - new_suffix.size());
path.append(new_suffix);
} while (FileExists(path.c_str()));
return path;
}
bool MakeTempFile(const std::string& filename_template,
std::string* filename,
int* fd) {
DCHECK(filename || fd);
vector<char> buf(filename_template.size() + 1);
memcpy(&buf[0], filename_template.data(), filename_template.size());
buf[filename_template.size()] = '\0';
int mkstemp_fd = mkstemp(&buf[0]);
TEST_AND_RETURN_FALSE_ERRNO(mkstemp_fd >= 0);
if (filename) {
*filename = &buf[0];
}
if (fd) {
*fd = mkstemp_fd;
} else {
close(mkstemp_fd);
}
return true;
}
bool MakeTempDirectory(const std::string& dirname_template,
std::string* dirname) {
DCHECK(dirname);
vector<char> buf(dirname_template.size() + 1);
memcpy(&buf[0], dirname_template.data(), dirname_template.size());
buf[dirname_template.size()] = '\0';
char* return_code = mkdtemp(&buf[0]);
TEST_AND_RETURN_FALSE_ERRNO(return_code != NULL);
*dirname = &buf[0];
return true;
}
bool StringHasSuffix(const std::string& str, const std::string& suffix) {
if (suffix.size() > str.size())
return false;
return 0 == str.compare(str.size() - suffix.size(), suffix.size(), suffix);
}
bool StringHasPrefix(const std::string& str, const std::string& prefix) {
if (prefix.size() > str.size())
return false;
return 0 == str.compare(0, prefix.size(), prefix);
}
bool MountFilesystem(const string& device,
const string& mountpoint,
unsigned long mountflags) {
int rc = mount(device.c_str(), mountpoint.c_str(), "ext3", mountflags, NULL);
if (rc < 0) {
string msg = ErrnoNumberAsString(errno);
LOG(ERROR) << "Unable to mount destination device: " << msg << ". "
<< device << " on " << mountpoint;
return false;
}
return true;
}
bool UnmountFilesystem(const string& mountpoint) {
for (int num_retries = 0; ; ++num_retries) {
if (umount(mountpoint.c_str()) == 0)
break;
TEST_AND_RETURN_FALSE_ERRNO(errno == EBUSY &&
num_retries < kUnmountMaxNumOfRetries);
g_usleep(kUnmountRetryIntervalInMicroseconds);
}
return true;
}
bool GetFilesystemSize(const std::string& device,
int* out_block_count,
int* out_block_size) {
int fd = HANDLE_EINTR(open(device.c_str(), O_RDONLY));
TEST_AND_RETURN_FALSE(fd >= 0);
ScopedFdCloser fd_closer(&fd);
return GetFilesystemSizeFromFD(fd, out_block_count, out_block_size);
}
bool GetFilesystemSizeFromFD(int fd,
int* out_block_count,
int* out_block_size) {
TEST_AND_RETURN_FALSE(fd >= 0);
// Determine the ext3 filesystem size by directly reading the block count and
// block size information from the superblock. See include/linux/ext3_fs.h for
// more details on the structure.
ssize_t kBufferSize = 16 * sizeof(uint32_t);
char buffer[kBufferSize];
const int kSuperblockOffset = 1024;
if (HANDLE_EINTR(pread(fd, buffer, kBufferSize, kSuperblockOffset)) !=
kBufferSize) {
PLOG(ERROR) << "Unable to determine file system size:";
return false;
}
uint32_t block_count; // ext3_fs.h: ext3_super_block.s_blocks_count
uint32_t log_block_size; // ext3_fs.h: ext3_super_block.s_log_block_size
uint16_t magic; // ext3_fs.h: ext3_super_block.s_magic
memcpy(&block_count, &buffer[1 * sizeof(int32_t)], sizeof(block_count));
memcpy(&log_block_size, &buffer[6 * sizeof(int32_t)], sizeof(log_block_size));
memcpy(&magic, &buffer[14 * sizeof(int32_t)], sizeof(magic));
block_count = le32toh(block_count);
const int kExt3MinBlockLogSize = 10; // ext3_fs.h: EXT3_MIN_BLOCK_LOG_SIZE
log_block_size = le32toh(log_block_size) + kExt3MinBlockLogSize;
magic = le16toh(magic);
// Sanity check the parameters.
const uint16_t kExt3SuperMagic = 0xef53; // ext3_fs.h: EXT3_SUPER_MAGIC
TEST_AND_RETURN_FALSE(magic == kExt3SuperMagic);
const int kExt3MinBlockSize = 1024; // ext3_fs.h: EXT3_MIN_BLOCK_SIZE
const int kExt3MaxBlockSize = 4096; // ext3_fs.h: EXT3_MAX_BLOCK_SIZE
int block_size = 1 << log_block_size;
TEST_AND_RETURN_FALSE(block_size >= kExt3MinBlockSize &&
block_size <= kExt3MaxBlockSize);
TEST_AND_RETURN_FALSE(block_count > 0);
if (out_block_count) {
*out_block_count = block_count;
}
if (out_block_size) {
*out_block_size = block_size;
}
return true;
}
bool GetBootloader(BootLoader* out_bootloader) {
// For now, hardcode to syslinux.
*out_bootloader = BootLoader_SYSLINUX;
return true;
}
string GetAndFreeGError(GError** error) {
if (!*error) {
return "Unknown GLib error.";
}
string message =
base::StringPrintf("GError(%d): %s",
(*error)->code,
(*error)->message ? (*error)->message : "(unknown)");
g_error_free(*error);
*error = NULL;
return message;
}
bool Reboot() {
vector<string> command;
command.push_back("/sbin/shutdown");
command.push_back("-r");
command.push_back("now");
int rc = 0;
Subprocess::SynchronousExec(command, &rc, NULL);
TEST_AND_RETURN_FALSE(rc == 0);
return true;
}
namespace {
// Do the actual trigger. We do it as a main-loop callback to (try to) get a
// consistent stack trace.
gboolean TriggerCrashReporterUpload(void* unused) {
pid_t pid = fork();
CHECK(pid >= 0) << "fork failed"; // fork() failed. Something is very wrong.
if (pid == 0) {
// We are the child. Crash.
abort(); // never returns
}
// We are the parent. Wait for child to terminate.
pid_t result = waitpid(pid, NULL, 0);
LOG_IF(ERROR, result < 0) << "waitpid() failed";
return FALSE; // Don't call this callback again
}
} // namespace {}
void ScheduleCrashReporterUpload() {
g_idle_add(&TriggerCrashReporterUpload, NULL);
}
bool SetCpuShares(CpuShares shares) {
string string_shares = base::IntToString(static_cast<int>(shares));
string cpu_shares_file = string(utils::kCGroupDir) + "/cpu.shares";
LOG(INFO) << "Setting cgroup cpu shares to " << string_shares;
if(utils::WriteFile(cpu_shares_file.c_str(), string_shares.c_str(),
string_shares.size())){
return true;
} else {
LOG(ERROR) << "Failed to change cgroup cpu shares to "<< string_shares
<< " using " << cpu_shares_file;
return false;
}
}
int CompareCpuShares(CpuShares shares_lhs,
CpuShares shares_rhs) {
return static_cast<int>(shares_lhs) - static_cast<int>(shares_rhs);
}
int FuzzInt(int value, unsigned int range) {
int min = value - range / 2;
int max = value + range - range / 2;
return base::RandInt(min, max);
}
gboolean GlibRunClosure(gpointer data) {
google::protobuf::Closure* callback =
reinterpret_cast<google::protobuf::Closure*>(data);
callback->Run();
return FALSE;
}
string FormatSecs(unsigned secs) {
return FormatTimeDelta(TimeDelta::FromSeconds(secs));
}
string FormatTimeDelta(TimeDelta delta) {
// Canonicalize into days, hours, minutes, seconds and microseconds.
unsigned days = delta.InDays();
delta -= TimeDelta::FromDays(days);
unsigned hours = delta.InHours();
delta -= TimeDelta::FromHours(hours);
unsigned mins = delta.InMinutes();
delta -= TimeDelta::FromMinutes(mins);
unsigned secs = delta.InSeconds();
delta -= TimeDelta::FromSeconds(secs);
unsigned usecs = delta.InMicroseconds();
// Construct and return string.
string str;
if (days)
base::StringAppendF(&str, "%ud", days);
if (days || hours)
base::StringAppendF(&str, "%uh", hours);
if (days || hours || mins)
base::StringAppendF(&str, "%um", mins);
base::StringAppendF(&str, "%u", secs);
if (usecs) {
int width = 6;
while ((usecs / 10) * 10 == usecs) {
usecs /= 10;
width--;
}
base::StringAppendF(&str, ".%0*u", width, usecs);
}
base::StringAppendF(&str, "s");
return str;
}
string ToString(const Time utc_time) {
Time::Exploded exp_time;
utc_time.UTCExplode(&exp_time);
return StringPrintf("%d/%d/%d %d:%02d:%02d GMT",
exp_time.month,
exp_time.day_of_month,
exp_time.year,
exp_time.hour,
exp_time.minute,
exp_time.second);
}
string ToString(bool b) {
return (b ? "true" : "false");
}
string ToString(DownloadSource source) {
switch (source) {
case kDownloadSourceHttpsServer: return "HttpsServer";
case kDownloadSourceHttpServer: return "HttpServer";
case kDownloadSourceHttpPeer: return "HttpPeer";
case kNumDownloadSources: return "Unknown";
// Don't add a default case to let the compiler warn about newly added
// download sources which should be added here.
}
return "Unknown";
}
string ToString(PayloadType payload_type) {
switch (payload_type) {
case kPayloadTypeDelta: return "Delta";
case kPayloadTypeFull: return "Full";
case kPayloadTypeForcedFull: return "ForcedFull";
case kNumPayloadTypes: return "Unknown";
// Don't add a default case to let the compiler warn about newly added
// payload types which should be added here.
}
return "Unknown";
}
ErrorCode GetBaseErrorCode(ErrorCode code) {
// Ignore the higher order bits in the code by applying the mask as
// we want the enumerations to be in the small contiguous range
// with values less than kErrorCodeUmaReportedMax.
ErrorCode base_code = static_cast<ErrorCode>(code & ~kErrorCodeSpecialFlags);
// Make additional adjustments required for UMA and error classification.
// TODO(jaysri): Move this logic to UeErrorCode.cc when we fix
// chromium-os:34369.
if (base_code >= kErrorCodeOmahaRequestHTTPResponseBase) {
// Since we want to keep the enums to a small value, aggregate all HTTP
// errors into this one bucket for UMA and error classification purposes.
LOG(INFO) << "Converting error code " << base_code
<< " to kErrorCodeOmahaErrorInHTTPResponse";
base_code = kErrorCodeOmahaErrorInHTTPResponse;
}
return base_code;
}
// Returns a printable version of the various flags denoted in the higher order
// bits of the given code. Returns an empty string if none of those bits are
// set.
string GetFlagNames(uint32_t code) {
uint32_t flags = code & kErrorCodeSpecialFlags;
string flag_names;
string separator = "";
for(size_t i = 0; i < sizeof(flags) * 8; i++) {
uint32_t flag = flags & (1 << i);
if (flag) {
flag_names += separator + CodeToString(static_cast<ErrorCode>(flag));
separator = ", ";
}
}
return flag_names;
}
void SendErrorCodeToUma(SystemState* system_state, ErrorCode code) {
if (!system_state)
return;
ErrorCode uma_error_code = GetBaseErrorCode(code);
// If the code doesn't have flags computed already, compute them now based on
// the state of the current update attempt.
uint32_t flags = code & kErrorCodeSpecialFlags;
if (!flags)
flags = system_state->update_attempter()->GetErrorCodeFlags();
// Determine the UMA bucket depending on the flags. But, ignore the resumed
// flag, as it's perfectly normal for production devices to resume their
// downloads and so we want to record those cases also in NormalErrorCodes
// bucket.
string metric = (flags & ~kErrorCodeResumedFlag) ?
"Installer.DevModeErrorCodes" : "Installer.NormalErrorCodes";
LOG(INFO) << "Sending error code " << uma_error_code
<< " (" << CodeToString(uma_error_code) << ")"
<< " to UMA metric: " << metric
<< ". Flags = " << (flags ? GetFlagNames(flags) : "None");
system_state->metrics_lib()->SendEnumToUMA(metric,
uma_error_code,
kErrorCodeUmaReportedMax);
}
string CodeToString(ErrorCode code) {
// If the given code has both parts (i.e. the error code part and the flags
// part) then strip off the flags part since the switch statement below
// has case statements only for the base error code or a single flag but
// doesn't support any combinations of those.
if ((code & kErrorCodeSpecialFlags) && (code & ~kErrorCodeSpecialFlags))
code = static_cast<ErrorCode>(code & ~kErrorCodeSpecialFlags);
switch (code) {
case kErrorCodeSuccess: return "kErrorCodeSuccess";
case kErrorCodeError: return "kErrorCodeError";
case kErrorCodeOmahaRequestError: return "kErrorCodeOmahaRequestError";
case kErrorCodeOmahaResponseHandlerError:
return "kErrorCodeOmahaResponseHandlerError";
case kErrorCodeFilesystemCopierError:
return "kErrorCodeFilesystemCopierError";
case kErrorCodePostinstallRunnerError:
return "kErrorCodePostinstallRunnerError";
case kErrorCodePayloadMismatchedType:
return "kErrorCodePayloadMismatchedType";
case kErrorCodeInstallDeviceOpenError:
return "kErrorCodeInstallDeviceOpenError";
case kErrorCodeKernelDeviceOpenError:
return "kErrorCodeKernelDeviceOpenError";
case kErrorCodeDownloadTransferError:
return "kErrorCodeDownloadTransferError";
case kErrorCodePayloadHashMismatchError:
return "kErrorCodePayloadHashMismatchError";
case kErrorCodePayloadSizeMismatchError:
return "kErrorCodePayloadSizeMismatchError";
case kErrorCodeDownloadPayloadVerificationError:
return "kErrorCodeDownloadPayloadVerificationError";
case kErrorCodeDownloadNewPartitionInfoError:
return "kErrorCodeDownloadNewPartitionInfoError";
case kErrorCodeDownloadWriteError:
return "kErrorCodeDownloadWriteError";
case kErrorCodeNewRootfsVerificationError:
return "kErrorCodeNewRootfsVerificationError";
case kErrorCodeNewKernelVerificationError:
return "kErrorCodeNewKernelVerificationError";
case kErrorCodeSignedDeltaPayloadExpectedError:
return "kErrorCodeSignedDeltaPayloadExpectedError";
case kErrorCodeDownloadPayloadPubKeyVerificationError:
return "kErrorCodeDownloadPayloadPubKeyVerificationError";
case kErrorCodePostinstallBootedFromFirmwareB:
return "kErrorCodePostinstallBootedFromFirmwareB";
case kErrorCodeDownloadStateInitializationError:
return "kErrorCodeDownloadStateInitializationError";
case kErrorCodeDownloadInvalidMetadataMagicString:
return "kErrorCodeDownloadInvalidMetadataMagicString";
case kErrorCodeDownloadSignatureMissingInManifest:
return "kErrorCodeDownloadSignatureMissingInManifest";
case kErrorCodeDownloadManifestParseError:
return "kErrorCodeDownloadManifestParseError";
case kErrorCodeDownloadMetadataSignatureError:
return "kErrorCodeDownloadMetadataSignatureError";
case kErrorCodeDownloadMetadataSignatureVerificationError:
return "kErrorCodeDownloadMetadataSignatureVerificationError";
case kErrorCodeDownloadMetadataSignatureMismatch:
return "kErrorCodeDownloadMetadataSignatureMismatch";
case kErrorCodeDownloadOperationHashVerificationError:
return "kErrorCodeDownloadOperationHashVerificationError";
case kErrorCodeDownloadOperationExecutionError:
return "kErrorCodeDownloadOperationExecutionError";
case kErrorCodeDownloadOperationHashMismatch:
return "kErrorCodeDownloadOperationHashMismatch";
case kErrorCodeOmahaRequestEmptyResponseError:
return "kErrorCodeOmahaRequestEmptyResponseError";
case kErrorCodeOmahaRequestXMLParseError:
return "kErrorCodeOmahaRequestXMLParseError";
case kErrorCodeDownloadInvalidMetadataSize:
return "kErrorCodeDownloadInvalidMetadataSize";
case kErrorCodeDownloadInvalidMetadataSignature:
return "kErrorCodeDownloadInvalidMetadataSignature";
case kErrorCodeOmahaResponseInvalid:
return "kErrorCodeOmahaResponseInvalid";
case kErrorCodeOmahaUpdateIgnoredPerPolicy:
return "kErrorCodeOmahaUpdateIgnoredPerPolicy";
case kErrorCodeOmahaUpdateDeferredPerPolicy:
return "kErrorCodeOmahaUpdateDeferredPerPolicy";
case kErrorCodeOmahaErrorInHTTPResponse:
return "kErrorCodeOmahaErrorInHTTPResponse";
case kErrorCodeDownloadOperationHashMissingError:
return "kErrorCodeDownloadOperationHashMissingError";
case kErrorCodeDownloadMetadataSignatureMissingError:
return "kErrorCodeDownloadMetadataSignatureMissingError";
case kErrorCodeOmahaUpdateDeferredForBackoff:
return "kErrorCodeOmahaUpdateDeferredForBackoff";
case kErrorCodePostinstallPowerwashError:
return "kErrorCodePostinstallPowerwashError";
case kErrorCodeUpdateCanceledByChannelChange:
return "kErrorCodeUpdateCanceledByChannelChange";
case kErrorCodeUmaReportedMax:
return "kErrorCodeUmaReportedMax";
case kErrorCodeOmahaRequestHTTPResponseBase:
return "kErrorCodeOmahaRequestHTTPResponseBase";
case kErrorCodeResumedFlag:
return "Resumed";
case kErrorCodeDevModeFlag:
return "DevMode";
case kErrorCodeTestImageFlag:
return "TestImage";
case kErrorCodeTestOmahaUrlFlag:
return "TestOmahaUrl";
case kErrorCodeSpecialFlags:
return "kErrorCodeSpecialFlags";
case kErrorCodePostinstallFirmwareRONotUpdatable:
return "kErrorCodePostinstallFirmwareRONotUpdatable";
// Don't add a default case to let the compiler warn about newly added
// error codes which should be added here.
}
return "Unknown error: " + base::UintToString(static_cast<unsigned>(code));
}
bool CreatePowerwashMarkerFile(const char* file_path) {
const char* marker_file = file_path ? file_path : kPowerwashMarkerFile;
bool result = utils::WriteFile(marker_file,
kPowerwashCommand,
strlen(kPowerwashCommand));
if (result) {
LOG(INFO) << "Created " << marker_file << " to powerwash on next reboot";
} else {
PLOG(ERROR) << "Error in creating powerwash marker file: " << marker_file;
}
return result;
}
bool DeletePowerwashMarkerFile(const char* file_path) {
const char* marker_file = file_path ? file_path : kPowerwashMarkerFile;
const FilePath kPowerwashMarkerPath(marker_file);
bool result = file_util::Delete(kPowerwashMarkerPath, false);
if (result)
LOG(INFO) << "Successfully deleted the powerwash marker file : "
<< marker_file;
else
PLOG(ERROR) << "Could not delete the powerwash marker file : "
<< marker_file;
return result;
}
bool GetInstallDev(const std::string& boot_dev, std::string* install_dev) {
TEST_AND_RETURN_FALSE(StringHasPrefix(boot_dev, "/dev/"));
string::iterator it;
string ubiblock_prefix("/dev/ubiblock");
install_dev->assign(boot_dev);
if(StringHasPrefix(boot_dev, ubiblock_prefix)) {
// UBI-based device
it = install_dev->begin() + ubiblock_prefix.length();
} else {
// non-UBI device
it = install_dev->end() - 1; // last character in string
}
// Right now, we just switch '3' and '5' partition numbers.
TEST_AND_RETURN_FALSE(*it == '3' || *it == '5');
*it = (*it == '3' ? '5' : '3');
return true;
}
Time TimeFromStructTimespec(struct timespec *ts) {
int64 us = static_cast<int64>(ts->tv_sec) * Time::kMicrosecondsPerSecond +
static_cast<int64>(ts->tv_nsec) / Time::kNanosecondsPerMicrosecond;
return Time::UnixEpoch() + TimeDelta::FromMicroseconds(us);
}
gchar** StringVectorToGStrv(const vector<string> &vector) {
GPtrArray *p = g_ptr_array_new();
for (std::vector<string>::const_iterator i = vector.begin();
i != vector.end(); ++i) {
g_ptr_array_add(p, g_strdup(i->c_str()));
}
g_ptr_array_add(p, NULL);
return reinterpret_cast<gchar**>(g_ptr_array_free(p, FALSE));
}
string StringVectorToString(const vector<string> &vector) {
string str = "[";
for (std::vector<string>::const_iterator i = vector.begin();
i != vector.end(); ++i) {
if (i != vector.begin())
str += ", ";
str += '"';
str += *i;
str += '"';
}
str += "]";
return str;
}
string CalculateP2PFileId(const string& payload_hash, size_t payload_size) {
string encoded_hash;
OmahaHashCalculator::Base64Encode(payload_hash.c_str(),
payload_hash.size(),
&encoded_hash);
return StringPrintf("cros_update_size_%zu_hash_%s",
payload_size,
encoded_hash.c_str());
}
bool IsXAttrSupported(const base::FilePath& dir_path) {
char *path = strdup(dir_path.Append("xattr_test_XXXXXX").value().c_str());
int fd = mkstemp(path);
if (fd == -1) {
PLOG(ERROR) << "Error creating temporary file in " << dir_path.value();
free(path);
return false;
}
if (unlink(path) != 0) {
PLOG(ERROR) << "Error unlinking temporary file " << path;
close(fd);
free(path);
return false;
}
int xattr_res = fsetxattr(fd, "user.xattr-test", "value", strlen("value"), 0);
if (xattr_res != 0) {
if (errno == ENOTSUP) {
// Leave it to call-sites to warn about non-support.
} else {
PLOG(ERROR) << "Error setting xattr on " << path;
}
}
close(fd);
free(path);
return xattr_res == 0;
}
} // namespace utils
} // namespace chromeos_update_engine