blob: 9e4a4d8ea373d3e5656f98f8c77ce3640743a543 [file] [log] [blame]
Mike Frysinger8155d082012-04-06 15:23:18 -04001// Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
adlr@google.com3defe6a2009-12-04 20:57:17 +00002// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "update_engine/utils.h"
Darin Petkovf74eb652010-08-04 12:08:38 -07006
David Zeuthen910ec5b2013-09-26 12:10:58 -07007#include <attr/xattr.h>
adlr@google.com3defe6a2009-12-04 20:57:17 +00008#include <dirent.h>
9#include <errno.h>
Andrew de los Reyes970bb282009-12-09 16:34:04 -080010#include <fcntl.h>
adlr@google.com3defe6a2009-12-04 20:57:17 +000011#include <stdio.h>
12#include <stdlib.h>
13#include <string.h>
Alex Deymoc4acdf42014-05-28 21:07:10 -070014#include <sys/mount.h>
15#include <sys/resource.h>
16#include <sys/stat.h>
17#include <sys/types.h>
18#include <sys/wait.h>
adlr@google.com3defe6a2009-12-04 20:57:17 +000019#include <unistd.h>
Darin Petkovf74eb652010-08-04 12:08:38 -070020
adlr@google.com3defe6a2009-12-04 20:57:17 +000021#include <algorithm>
Chris Sosac1972482013-04-30 22:31:10 -070022#include <vector>
Darin Petkovf74eb652010-08-04 12:08:38 -070023
Will Drewry8f71da82010-08-30 14:07:11 -050024#include <base/file_util.h>
Ben Chan736fcb52014-05-21 18:28:22 -070025#include <base/files/file_path.h>
26#include <base/files/scoped_file.h>
Mike Frysinger8155d082012-04-06 15:23:18 -040027#include <base/logging.h>
Chris Sosafc661a12013-02-26 14:43:21 -080028#include <base/posix/eintr_wrapper.h>
Will Drewry8f71da82010-08-30 14:07:11 -050029#include <base/rand_util.h>
Alex Vakulenko75039d72014-03-25 12:36:28 -070030#include <base/strings/string_number_conversions.h>
31#include <base/strings/string_split.h>
32#include <base/strings/string_util.h>
33#include <base/strings/stringprintf.h>
Gilad Arnold8e3f1262013-01-08 14:59:54 -080034#include <glib.h>
Andrew de los Reyesf3ed8e72011-02-16 10:35:46 -080035#include <google/protobuf/stubs/common.h>
Will Drewry8f71da82010-08-30 14:07:11 -050036
David Zeuthen33bae492014-02-25 16:16:18 -080037#include "update_engine/clock_interface.h"
Jay Srinivasan1c0fe792013-03-28 16:45:25 -070038#include "update_engine/constants.h"
Andrew de los Reyes970bb282009-12-09 16:34:04 -080039#include "update_engine/file_writer.h"
Darin Petkov33d30642010-08-04 10:18:57 -070040#include "update_engine/omaha_request_params.h"
David Zeuthen33bae492014-02-25 16:16:18 -080041#include "update_engine/prefs_interface.h"
Darin Petkov296889c2010-07-23 16:20:54 -070042#include "update_engine/subprocess.h"
Jay Srinivasan55f50c22013-01-10 19:24:35 -080043#include "update_engine/system_state.h"
44#include "update_engine/update_attempter.h"
adlr@google.com3defe6a2009-12-04 20:57:17 +000045
Jay Srinivasan480ddfa2012-06-01 19:15:26 -070046using base::Time;
Gilad Arnold8e3f1262013-01-08 14:59:54 -080047using base::TimeDelta;
adlr@google.com3defe6a2009-12-04 20:57:17 +000048using std::min;
Chris Sosac1972482013-04-30 22:31:10 -070049using std::pair;
adlr@google.com3defe6a2009-12-04 20:57:17 +000050using std::string;
51using std::vector;
52
53namespace chromeos_update_engine {
54
Ben Chan77a1eba2012-10-07 22:54:55 -070055namespace {
56
57// The following constants control how UnmountFilesystem should retry if
58// umount() fails with an errno EBUSY, i.e. retry 5 times over the course of
59// one second.
60const int kUnmountMaxNumOfRetries = 5;
61const int kUnmountRetryIntervalInMicroseconds = 200 * 1000; // 200 ms
Alex Deymo032e7722014-03-25 17:53:56 -070062
63// Number of bytes to read from a file to attempt to detect its contents. Used
64// in GetFileFormat.
65const int kGetFileFormatMaxHeaderSize = 32;
66
Ben Chan77a1eba2012-10-07 22:54:55 -070067} // namespace
68
adlr@google.com3defe6a2009-12-04 20:57:17 +000069namespace utils {
70
Chris Sosa4f8ee272012-11-30 13:01:54 -080071// Cgroup container is created in update-engine's upstart script located at
72// /etc/init/update-engine.conf.
73static const char kCGroupDir[] = "/sys/fs/cgroup/cpu/update-engine";
74
J. Richard Barnette63137e52013-10-28 10:57:29 -070075string ParseECVersion(string input_line) {
Ben Chan736fcb52014-05-21 18:28:22 -070076 base::TrimWhitespaceASCII(input_line, base::TRIM_ALL, &input_line);
Chris Sosac1972482013-04-30 22:31:10 -070077
Alex Vakulenko75039d72014-03-25 12:36:28 -070078 // At this point we want to convert the format key=value pair from mosys to
Chris Sosac1972482013-04-30 22:31:10 -070079 // a vector of key value pairs.
80 vector<pair<string, string> > kv_pairs;
J. Richard Barnette63137e52013-10-28 10:57:29 -070081 if (base::SplitStringIntoKeyValuePairs(input_line, '=', ' ', &kv_pairs)) {
Chris Sosac1972482013-04-30 22:31:10 -070082 for (vector<pair<string, string> >::iterator it = kv_pairs.begin();
83 it != kv_pairs.end(); ++it) {
84 // Finally match against the fw_verion which may have quotes.
85 if (it->first == "fw_version") {
86 string output;
87 // Trim any quotes.
Alex Vakulenko75039d72014-03-25 12:36:28 -070088 base::TrimString(it->second, "\"", &output);
Chris Sosac1972482013-04-30 22:31:10 -070089 return output;
90 }
91 }
92 }
93 LOG(ERROR) << "Unable to parse fwid from ec info.";
94 return "";
95}
96
97
Alex Vakulenkof3f85bb2014-03-26 16:36:35 -070098const string KernelDeviceOfBootDevice(const string& boot_device) {
99 string kernel_partition_name;
J. Richard Barnette30842932013-10-28 15:04:23 -0700100
Alex Vakulenkof3f85bb2014-03-26 16:36:35 -0700101 string disk_name;
102 int partition_num;
103 if (SplitPartitionName(boot_device, &disk_name, &partition_num)) {
104 if (disk_name == "/dev/ubiblock") {
105 // Special case for NAND devices.
106 // eg: /dev/ubiblock3_0 becomes /dev/mtdblock2
107 disk_name = "/dev/mtdblock";
108 }
109 // Currently this assumes the partition number of the boot device is
110 // 3, 5, or 7, and changes it to 2, 4, or 6, respectively, to
111 // get the kernel device.
112 if (partition_num == 3 || partition_num == 5 || partition_num == 7) {
113 kernel_partition_name = MakePartitionName(disk_name, partition_num - 1);
114 }
J. Richard Barnette30842932013-10-28 15:04:23 -0700115 }
116
Alex Vakulenkof3f85bb2014-03-26 16:36:35 -0700117 return kernel_partition_name;
J. Richard Barnette30842932013-10-28 15:04:23 -0700118}
119
120
Andrew de los Reyes970bb282009-12-09 16:34:04 -0800121bool WriteFile(const char* path, const char* data, int data_len) {
122 DirectFileWriter writer;
123 TEST_AND_RETURN_FALSE_ERRNO(0 == writer.Open(path,
124 O_WRONLY | O_CREAT | O_TRUNC,
Chris Masone4dc2ada2010-09-23 12:43:03 -0700125 0600));
Andrew de los Reyes970bb282009-12-09 16:34:04 -0800126 ScopedFileWriterCloser closer(&writer);
Don Garrette410e0f2011-11-10 15:39:01 -0800127 TEST_AND_RETURN_FALSE_ERRNO(writer.Write(data, data_len));
Andrew de los Reyes970bb282009-12-09 16:34:04 -0800128 return true;
129}
130
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700131bool WriteAll(int fd, const void* buf, size_t count) {
Andrew de los Reyesb10320d2010-03-31 16:44:44 -0700132 const char* c_buf = static_cast<const char*>(buf);
133 ssize_t bytes_written = 0;
134 while (bytes_written < static_cast<ssize_t>(count)) {
135 ssize_t rc = write(fd, c_buf + bytes_written, count - bytes_written);
136 TEST_AND_RETURN_FALSE_ERRNO(rc >= 0);
137 bytes_written += rc;
138 }
139 return true;
140}
141
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700142bool PWriteAll(int fd, const void* buf, size_t count, off_t offset) {
143 const char* c_buf = static_cast<const char*>(buf);
Gilad Arnold780db212012-07-11 13:12:49 -0700144 size_t bytes_written = 0;
145 int num_attempts = 0;
146 while (bytes_written < count) {
147 num_attempts++;
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700148 ssize_t rc = pwrite(fd, c_buf + bytes_written, count - bytes_written,
149 offset + bytes_written);
Gilad Arnold780db212012-07-11 13:12:49 -0700150 // TODO(garnold) for debugging failure in chromium-os:31077; to be removed.
151 if (rc < 0) {
152 PLOG(ERROR) << "pwrite error; num_attempts=" << num_attempts
153 << " bytes_written=" << bytes_written
154 << " count=" << count << " offset=" << offset;
155 }
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700156 TEST_AND_RETURN_FALSE_ERRNO(rc >= 0);
157 bytes_written += rc;
158 }
159 return true;
160}
161
162bool PReadAll(int fd, void* buf, size_t count, off_t offset,
163 ssize_t* out_bytes_read) {
164 char* c_buf = static_cast<char*>(buf);
165 ssize_t bytes_read = 0;
166 while (bytes_read < static_cast<ssize_t>(count)) {
167 ssize_t rc = pread(fd, c_buf + bytes_read, count - bytes_read,
168 offset + bytes_read);
169 TEST_AND_RETURN_FALSE_ERRNO(rc >= 0);
170 if (rc == 0) {
171 break;
172 }
173 bytes_read += rc;
174 }
175 *out_bytes_read = bytes_read;
176 return true;
Darin Petkov296889c2010-07-23 16:20:54 -0700177
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700178}
179
Gilad Arnold19a45f02012-07-19 12:36:10 -0700180// Append |nbytes| of content from |buf| to the vector pointed to by either
181// |vec_p| or |str_p|.
182static void AppendBytes(const char* buf, size_t nbytes,
183 std::vector<char>* vec_p) {
184 CHECK(buf);
185 CHECK(vec_p);
186 vec_p->insert(vec_p->end(), buf, buf + nbytes);
187}
188static void AppendBytes(const char* buf, size_t nbytes,
189 std::string* str_p) {
190 CHECK(buf);
191 CHECK(str_p);
192 str_p->append(buf, nbytes);
193}
194
195// Reads from an open file |fp|, appending the read content to the container
196// pointer to by |out_p|. Returns true upon successful reading all of the
Darin Petkov8e447e02013-04-16 16:23:50 +0200197// file's content, false otherwise. If |size| is not -1, reads up to |size|
198// bytes.
Gilad Arnold19a45f02012-07-19 12:36:10 -0700199template <class T>
Darin Petkov8e447e02013-04-16 16:23:50 +0200200static bool Read(FILE* fp, off_t size, T* out_p) {
Gilad Arnold19a45f02012-07-19 12:36:10 -0700201 CHECK(fp);
Darin Petkov8e447e02013-04-16 16:23:50 +0200202 CHECK(size == -1 || size >= 0);
Gilad Arnold19a45f02012-07-19 12:36:10 -0700203 char buf[1024];
Darin Petkov8e447e02013-04-16 16:23:50 +0200204 while (size == -1 || size > 0) {
205 off_t bytes_to_read = sizeof(buf);
206 if (size > 0 && bytes_to_read > size) {
207 bytes_to_read = size;
208 }
209 size_t nbytes = fread(buf, 1, bytes_to_read, fp);
210 if (!nbytes) {
211 break;
212 }
Gilad Arnold19a45f02012-07-19 12:36:10 -0700213 AppendBytes(buf, nbytes, out_p);
Darin Petkov8e447e02013-04-16 16:23:50 +0200214 if (size != -1) {
215 CHECK(size >= static_cast<off_t>(nbytes));
216 size -= nbytes;
217 }
218 }
219 if (ferror(fp)) {
220 return false;
221 }
222 return size == 0 || feof(fp);
Gilad Arnold19a45f02012-07-19 12:36:10 -0700223}
224
Darin Petkov8e447e02013-04-16 16:23:50 +0200225// Opens a file |path| for reading and appends its the contents to a container
226// |out_p|. Starts reading the file from |offset|. If |offset| is beyond the end
227// of the file, returns success. If |size| is not -1, reads up to |size| bytes.
Gilad Arnold19a45f02012-07-19 12:36:10 -0700228template <class T>
Darin Petkov8e447e02013-04-16 16:23:50 +0200229static bool ReadFileChunkAndAppend(const std::string& path,
230 off_t offset,
231 off_t size,
232 T* out_p) {
233 CHECK_GE(offset, 0);
234 CHECK(size == -1 || size >= 0);
Ben Chan736fcb52014-05-21 18:28:22 -0700235 base::ScopedFILE fp(fopen(path.c_str(), "r"));
Darin Petkov8e447e02013-04-16 16:23:50 +0200236 if (!fp.get())
adlr@google.com3defe6a2009-12-04 20:57:17 +0000237 return false;
Darin Petkov8e447e02013-04-16 16:23:50 +0200238 if (offset) {
239 // Return success without appending any data if a chunk beyond the end of
240 // the file is requested.
241 if (offset >= FileSize(path)) {
242 return true;
243 }
244 TEST_AND_RETURN_FALSE_ERRNO(fseek(fp.get(), offset, SEEK_SET) == 0);
245 }
246 return Read(fp.get(), size, out_p);
adlr@google.com3defe6a2009-12-04 20:57:17 +0000247}
248
Gilad Arnold19a45f02012-07-19 12:36:10 -0700249// Invokes a pipe |cmd|, then uses |append_func| to append its stdout to a
250// container |out_p|.
251template <class T>
252static bool ReadPipeAndAppend(const std::string& cmd, T* out_p) {
253 FILE* fp = popen(cmd.c_str(), "r");
254 if (!fp)
adlr@google.com3defe6a2009-12-04 20:57:17 +0000255 return false;
Darin Petkov8e447e02013-04-16 16:23:50 +0200256 bool success = Read(fp, -1, out_p);
Gilad Arnold19a45f02012-07-19 12:36:10 -0700257 return (success && pclose(fp) >= 0);
258}
259
260
Darin Petkov8e447e02013-04-16 16:23:50 +0200261bool ReadFile(const string& path, vector<char>* out_p) {
262 return ReadFileChunkAndAppend(path, 0, -1, out_p);
Gilad Arnold19a45f02012-07-19 12:36:10 -0700263}
264
Darin Petkov8e447e02013-04-16 16:23:50 +0200265bool ReadFile(const string& path, string* out_p) {
266 return ReadFileChunkAndAppend(path, 0, -1, out_p);
Gilad Arnold19a45f02012-07-19 12:36:10 -0700267}
268
Darin Petkov8e447e02013-04-16 16:23:50 +0200269bool ReadFileChunk(const string& path, off_t offset, off_t size,
270 vector<char>* out_p) {
271 return ReadFileChunkAndAppend(path, offset, size, out_p);
272}
273
274bool ReadPipe(const string& cmd, vector<char>* out_p) {
Gilad Arnold19a45f02012-07-19 12:36:10 -0700275 return ReadPipeAndAppend(cmd, out_p);
276}
277
Darin Petkov8e447e02013-04-16 16:23:50 +0200278bool ReadPipe(const string& cmd, string* out_p) {
Gilad Arnold19a45f02012-07-19 12:36:10 -0700279 return ReadPipeAndAppend(cmd, out_p);
adlr@google.com3defe6a2009-12-04 20:57:17 +0000280}
281
Andrew de los Reyesf4c7ef12010-04-30 10:37:00 -0700282off_t FileSize(const string& path) {
283 struct stat stbuf;
284 int rc = stat(path.c_str(), &stbuf);
285 CHECK_EQ(rc, 0);
286 if (rc < 0)
287 return rc;
288 return stbuf.st_size;
289}
290
adlr@google.com3defe6a2009-12-04 20:57:17 +0000291void HexDumpArray(const unsigned char* const arr, const size_t length) {
292 const unsigned char* const char_arr =
293 reinterpret_cast<const unsigned char* const>(arr);
294 LOG(INFO) << "Logging array of length: " << length;
295 const unsigned int bytes_per_line = 16;
Andrew de los Reyes08c4e272010-04-15 14:02:17 -0700296 for (uint32_t i = 0; i < length; i += bytes_per_line) {
adlr@google.com3defe6a2009-12-04 20:57:17 +0000297 const unsigned int bytes_remaining = length - i;
298 const unsigned int bytes_per_this_line = min(bytes_per_line,
299 bytes_remaining);
300 char header[100];
301 int r = snprintf(header, sizeof(header), "0x%08x : ", i);
302 TEST_AND_RETURN(r == 13);
303 string line = header;
304 for (unsigned int j = 0; j < bytes_per_this_line; j++) {
305 char buf[20];
306 unsigned char c = char_arr[i + j];
307 r = snprintf(buf, sizeof(buf), "%02x ", static_cast<unsigned int>(c));
308 TEST_AND_RETURN(r == 3);
309 line += buf;
310 }
311 LOG(INFO) << line;
312 }
313}
314
315namespace {
316class ScopedDirCloser {
317 public:
318 explicit ScopedDirCloser(DIR** dir) : dir_(dir) {}
319 ~ScopedDirCloser() {
320 if (dir_ && *dir_) {
321 int r = closedir(*dir_);
322 TEST_AND_RETURN_ERRNO(r == 0);
323 *dir_ = NULL;
324 dir_ = NULL;
325 }
326 }
327 private:
328 DIR** dir_;
329};
330} // namespace {}
331
332bool RecursiveUnlinkDir(const std::string& path) {
333 struct stat stbuf;
334 int r = lstat(path.c_str(), &stbuf);
335 TEST_AND_RETURN_FALSE_ERRNO((r == 0) || (errno == ENOENT));
336 if ((r < 0) && (errno == ENOENT))
337 // path request is missing. that's fine.
338 return true;
339 if (!S_ISDIR(stbuf.st_mode)) {
340 TEST_AND_RETURN_FALSE_ERRNO((unlink(path.c_str()) == 0) ||
Andrew de los Reyesb10320d2010-03-31 16:44:44 -0700341 (errno == ENOENT));
adlr@google.com3defe6a2009-12-04 20:57:17 +0000342 // success or path disappeared before we could unlink.
343 return true;
344 }
345 {
346 // We have a dir, unlink all children, then delete dir
347 DIR *dir = opendir(path.c_str());
348 TEST_AND_RETURN_FALSE_ERRNO(dir);
349 ScopedDirCloser dir_closer(&dir);
350 struct dirent dir_entry;
351 struct dirent *dir_entry_p;
352 int err = 0;
353 while ((err = readdir_r(dir, &dir_entry, &dir_entry_p)) == 0) {
354 if (dir_entry_p == NULL) {
355 // end of stream reached
356 break;
357 }
358 // Skip . and ..
359 if (!strcmp(dir_entry_p->d_name, ".") ||
360 !strcmp(dir_entry_p->d_name, ".."))
361 continue;
362 TEST_AND_RETURN_FALSE(RecursiveUnlinkDir(path + "/" +
Andrew de los Reyesb10320d2010-03-31 16:44:44 -0700363 dir_entry_p->d_name));
adlr@google.com3defe6a2009-12-04 20:57:17 +0000364 }
365 TEST_AND_RETURN_FALSE(err == 0);
366 }
367 // unlink dir
368 TEST_AND_RETURN_FALSE_ERRNO((rmdir(path.c_str()) == 0) || (errno == ENOENT));
369 return true;
370}
371
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800372std::string GetDiskName(const string& partition_name) {
373 std::string disk_name;
374 return SplitPartitionName(partition_name, &disk_name, nullptr) ?
375 disk_name : std::string();
Andrew de los Reyesf9714432010-05-04 10:21:23 -0700376}
377
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800378int GetPartitionNumber(const std::string& partition_name) {
379 int partition_num = 0;
380 return SplitPartitionName(partition_name, nullptr, &partition_num) ?
381 partition_num : 0;
382}
383
384bool SplitPartitionName(const std::string& partition_name,
385 std::string* out_disk_name,
386 int* out_partition_num) {
387 if (!StringHasPrefix(partition_name, "/dev/")) {
388 LOG(ERROR) << "Invalid partition device name: " << partition_name;
389 return false;
Andrew de los Reyesf9714432010-05-04 10:21:23 -0700390 }
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800391
Alex Vakulenkof3f85bb2014-03-26 16:36:35 -0700392 size_t last_nondigit_pos = partition_name.find_last_not_of("0123456789");
393 if (last_nondigit_pos == string::npos ||
394 (last_nondigit_pos + 1) == partition_name.size()) {
395 LOG(ERROR) << "Unable to parse partition device name: " << partition_name;
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800396 return false;
397 }
398
Alex Vakulenkof3f85bb2014-03-26 16:36:35 -0700399 size_t partition_name_len = std::string::npos;
400 if (partition_name[last_nondigit_pos] == '_') {
401 // NAND block devices have weird naming which could be something
402 // like "/dev/ubiblock2_0". We discard "_0" in such a case.
403 size_t prev_nondigit_pos =
404 partition_name.find_last_not_of("0123456789", last_nondigit_pos - 1);
405 if (prev_nondigit_pos == string::npos ||
406 (prev_nondigit_pos + 1) == last_nondigit_pos) {
407 LOG(ERROR) << "Unable to parse partition device name: " << partition_name;
408 return false;
409 }
410
411 partition_name_len = last_nondigit_pos - prev_nondigit_pos;
412 last_nondigit_pos = prev_nondigit_pos;
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800413 }
414
415 if (out_disk_name) {
416 // Special case for MMC devices which have the following naming scheme:
417 // mmcblk0p2
Alex Vakulenkof3f85bb2014-03-26 16:36:35 -0700418 size_t disk_name_len = last_nondigit_pos;
419 if (partition_name[last_nondigit_pos] != 'p' ||
420 last_nondigit_pos == 0 ||
421 !isdigit(partition_name[last_nondigit_pos - 1])) {
422 disk_name_len++;
423 }
424 *out_disk_name = partition_name.substr(0, disk_name_len);
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800425 }
426
427 if (out_partition_num) {
Alex Vakulenkof3f85bb2014-03-26 16:36:35 -0700428 std::string partition_str = partition_name.substr(last_nondigit_pos + 1,
429 partition_name_len);
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800430 *out_partition_num = atoi(partition_str.c_str());
431 }
432 return true;
433}
434
435std::string MakePartitionName(const std::string& disk_name,
436 int partition_num) {
437 if (!StringHasPrefix(disk_name, "/dev/")) {
438 LOG(ERROR) << "Invalid disk name: " << disk_name;
439 return std::string();
440 }
441
442 if (partition_num < 1) {
443 LOG(ERROR) << "Invalid partition number: " << partition_num;
444 return std::string();
445 }
446
447 std::string partition_name = disk_name;
Alex Vakulenkof3f85bb2014-03-26 16:36:35 -0700448 if (isdigit(partition_name.back())) {
449 // Special case for devices with names ending with a digit.
450 // Add "p" to separate the disk name from partition number,
451 // e.g. "/dev/loop0p2"
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800452 partition_name += 'p';
453 }
454
Alex Vakulenkof3f85bb2014-03-26 16:36:35 -0700455 partition_name += std::to_string(partition_num);
456
457 if (StringHasPrefix(partition_name, "/dev/ubiblock")) {
458 // Special case for UBI block devieces that have "_0" suffix.
459 partition_name += "_0";
460 }
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800461 return partition_name;
Andrew de los Reyesf9714432010-05-04 10:21:23 -0700462}
463
Darin Petkovf74eb652010-08-04 12:08:38 -0700464string SysfsBlockDevice(const string& device) {
Alex Vakulenko75039d72014-03-25 12:36:28 -0700465 base::FilePath device_path(device);
Darin Petkovf74eb652010-08-04 12:08:38 -0700466 if (device_path.DirName().value() != "/dev") {
467 return "";
468 }
Alex Vakulenko75039d72014-03-25 12:36:28 -0700469 return base::FilePath("/sys/block").Append(device_path.BaseName()).value();
Darin Petkovf74eb652010-08-04 12:08:38 -0700470}
471
472bool IsRemovableDevice(const std::string& device) {
473 string sysfs_block = SysfsBlockDevice(device);
474 string removable;
475 if (sysfs_block.empty() ||
Alex Vakulenko75039d72014-03-25 12:36:28 -0700476 !base::ReadFileToString(base::FilePath(sysfs_block).Append("removable"),
Ben Chan736fcb52014-05-21 18:28:22 -0700477 &removable)) {
Darin Petkovf74eb652010-08-04 12:08:38 -0700478 return false;
479 }
Ben Chan736fcb52014-05-21 18:28:22 -0700480 base::TrimWhitespaceASCII(removable, base::TRIM_ALL, &removable);
Darin Petkovf74eb652010-08-04 12:08:38 -0700481 return removable == "1";
482}
483
adlr@google.com3defe6a2009-12-04 20:57:17 +0000484std::string ErrnoNumberAsString(int err) {
485 char buf[100];
486 buf[0] = '\0';
487 return strerror_r(err, buf, sizeof(buf));
488}
489
490std::string NormalizePath(const std::string& path, bool strip_trailing_slash) {
491 string ret;
492 bool last_insert_was_slash = false;
493 for (string::const_iterator it = path.begin(); it != path.end(); ++it) {
494 if (*it == '/') {
495 if (last_insert_was_slash)
496 continue;
497 last_insert_was_slash = true;
498 } else {
499 last_insert_was_slash = false;
500 }
501 ret.push_back(*it);
502 }
503 if (strip_trailing_slash && last_insert_was_slash) {
504 string::size_type last_non_slash = ret.find_last_not_of('/');
505 if (last_non_slash != string::npos) {
506 ret.resize(last_non_slash + 1);
507 } else {
508 ret = "";
509 }
510 }
511 return ret;
512}
513
514bool FileExists(const char* path) {
515 struct stat stbuf;
516 return 0 == lstat(path, &stbuf);
517}
518
Darin Petkov30291ed2010-11-12 10:23:06 -0800519bool IsSymlink(const char* path) {
520 struct stat stbuf;
521 return lstat(path, &stbuf) == 0 && S_ISLNK(stbuf.st_mode) != 0;
522}
523
Alex Deymo7dc4c502014-05-20 20:09:58 -0700524bool IsDir(const char* path) {
525 struct stat stbuf;
526 TEST_AND_RETURN_FALSE_ERRNO(lstat(path, &stbuf) == 0);
527 return S_ISDIR(stbuf.st_mode);
528}
529
adlr@google.com3defe6a2009-12-04 20:57:17 +0000530std::string TempFilename(string path) {
531 static const string suffix("XXXXXX");
532 CHECK(StringHasSuffix(path, suffix));
533 do {
534 string new_suffix;
535 for (unsigned int i = 0; i < suffix.size(); i++) {
536 int r = rand() % (26 * 2 + 10); // [a-zA-Z0-9]
537 if (r < 26)
538 new_suffix.append(1, 'a' + r);
539 else if (r < (26 * 2))
540 new_suffix.append(1, 'A' + r - 26);
541 else
542 new_suffix.append(1, '0' + r - (26 * 2));
543 }
544 CHECK_EQ(new_suffix.size(), suffix.size());
545 path.resize(path.size() - new_suffix.size());
546 path.append(new_suffix);
547 } while (FileExists(path.c_str()));
548 return path;
549}
550
Gilad Arnoldd04f8e22014-01-09 13:13:40 -0800551// If |path| is absolute, or explicit relative to the current working directory,
552// leaves it as is. Otherwise, if TMPDIR is defined in the environment and is
553// non-empty, prepends it to |path|. Otherwise, prepends /tmp. Returns the
554// resulting path.
555static const string PrependTmpdir(const string& path) {
556 if (path[0] == '/' || StartsWithASCII(path, "./", true) ||
557 StartsWithASCII(path, "../", true))
558 return path;
559
560 const char *tmpdir = getenv("TMPDIR");
561 const string prefix = (tmpdir && *tmpdir ? tmpdir : "/tmp");
562 return prefix + "/" + path;
563}
564
565bool MakeTempFile(const std::string& base_filename_template,
Andrew de los Reyesb10320d2010-03-31 16:44:44 -0700566 std::string* filename,
567 int* fd) {
Gilad Arnoldd04f8e22014-01-09 13:13:40 -0800568 const string filename_template = PrependTmpdir(base_filename_template);
Andrew de los Reyesb10320d2010-03-31 16:44:44 -0700569 DCHECK(filename || fd);
570 vector<char> buf(filename_template.size() + 1);
571 memcpy(&buf[0], filename_template.data(), filename_template.size());
572 buf[filename_template.size()] = '\0';
Darin Petkov296889c2010-07-23 16:20:54 -0700573
Andrew de los Reyesb10320d2010-03-31 16:44:44 -0700574 int mkstemp_fd = mkstemp(&buf[0]);
575 TEST_AND_RETURN_FALSE_ERRNO(mkstemp_fd >= 0);
576 if (filename) {
577 *filename = &buf[0];
578 }
579 if (fd) {
580 *fd = mkstemp_fd;
581 } else {
582 close(mkstemp_fd);
583 }
584 return true;
585}
586
Gilad Arnoldd04f8e22014-01-09 13:13:40 -0800587bool MakeTempDirectory(const std::string& base_dirname_template,
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700588 std::string* dirname) {
Gilad Arnoldd04f8e22014-01-09 13:13:40 -0800589 const string dirname_template = PrependTmpdir(base_dirname_template);
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700590 DCHECK(dirname);
591 vector<char> buf(dirname_template.size() + 1);
592 memcpy(&buf[0], dirname_template.data(), dirname_template.size());
593 buf[dirname_template.size()] = '\0';
Darin Petkov296889c2010-07-23 16:20:54 -0700594
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700595 char* return_code = mkdtemp(&buf[0]);
596 TEST_AND_RETURN_FALSE_ERRNO(return_code != NULL);
597 *dirname = &buf[0];
598 return true;
599}
600
adlr@google.com3defe6a2009-12-04 20:57:17 +0000601bool StringHasSuffix(const std::string& str, const std::string& suffix) {
602 if (suffix.size() > str.size())
603 return false;
604 return 0 == str.compare(str.size() - suffix.size(), suffix.size(), suffix);
605}
606
607bool StringHasPrefix(const std::string& str, const std::string& prefix) {
608 if (prefix.size() > str.size())
609 return false;
610 return 0 == str.compare(0, prefix.size(), prefix);
611}
612
adlr@google.com3defe6a2009-12-04 20:57:17 +0000613bool MountFilesystem(const string& device,
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700614 const string& mountpoint,
615 unsigned long mountflags) {
616 int rc = mount(device.c_str(), mountpoint.c_str(), "ext3", mountflags, NULL);
adlr@google.com3defe6a2009-12-04 20:57:17 +0000617 if (rc < 0) {
618 string msg = ErrnoNumberAsString(errno);
619 LOG(ERROR) << "Unable to mount destination device: " << msg << ". "
620 << device << " on " << mountpoint;
621 return false;
622 }
623 return true;
624}
625
626bool UnmountFilesystem(const string& mountpoint) {
Ben Chan77a1eba2012-10-07 22:54:55 -0700627 for (int num_retries = 0; ; ++num_retries) {
628 if (umount(mountpoint.c_str()) == 0)
629 break;
630
631 TEST_AND_RETURN_FALSE_ERRNO(errno == EBUSY &&
632 num_retries < kUnmountMaxNumOfRetries);
Gilad Arnold8e3f1262013-01-08 14:59:54 -0800633 g_usleep(kUnmountRetryIntervalInMicroseconds);
Ben Chan77a1eba2012-10-07 22:54:55 -0700634 }
adlr@google.com3defe6a2009-12-04 20:57:17 +0000635 return true;
636}
637
Darin Petkovd3f8c892010-10-12 21:38:45 -0700638bool GetFilesystemSize(const std::string& device,
639 int* out_block_count,
640 int* out_block_size) {
641 int fd = HANDLE_EINTR(open(device.c_str(), O_RDONLY));
642 TEST_AND_RETURN_FALSE(fd >= 0);
643 ScopedFdCloser fd_closer(&fd);
644 return GetFilesystemSizeFromFD(fd, out_block_count, out_block_size);
645}
646
647bool GetFilesystemSizeFromFD(int fd,
648 int* out_block_count,
649 int* out_block_size) {
650 TEST_AND_RETURN_FALSE(fd >= 0);
651
652 // Determine the ext3 filesystem size by directly reading the block count and
653 // block size information from the superblock. See include/linux/ext3_fs.h for
654 // more details on the structure.
655 ssize_t kBufferSize = 16 * sizeof(uint32_t);
656 char buffer[kBufferSize];
657 const int kSuperblockOffset = 1024;
658 if (HANDLE_EINTR(pread(fd, buffer, kBufferSize, kSuperblockOffset)) !=
659 kBufferSize) {
660 PLOG(ERROR) << "Unable to determine file system size:";
661 return false;
662 }
663 uint32_t block_count; // ext3_fs.h: ext3_super_block.s_blocks_count
664 uint32_t log_block_size; // ext3_fs.h: ext3_super_block.s_log_block_size
665 uint16_t magic; // ext3_fs.h: ext3_super_block.s_magic
666 memcpy(&block_count, &buffer[1 * sizeof(int32_t)], sizeof(block_count));
667 memcpy(&log_block_size, &buffer[6 * sizeof(int32_t)], sizeof(log_block_size));
668 memcpy(&magic, &buffer[14 * sizeof(int32_t)], sizeof(magic));
669 block_count = le32toh(block_count);
670 const int kExt3MinBlockLogSize = 10; // ext3_fs.h: EXT3_MIN_BLOCK_LOG_SIZE
671 log_block_size = le32toh(log_block_size) + kExt3MinBlockLogSize;
672 magic = le16toh(magic);
673
674 // Sanity check the parameters.
675 const uint16_t kExt3SuperMagic = 0xef53; // ext3_fs.h: EXT3_SUPER_MAGIC
676 TEST_AND_RETURN_FALSE(magic == kExt3SuperMagic);
677 const int kExt3MinBlockSize = 1024; // ext3_fs.h: EXT3_MIN_BLOCK_SIZE
678 const int kExt3MaxBlockSize = 4096; // ext3_fs.h: EXT3_MAX_BLOCK_SIZE
679 int block_size = 1 << log_block_size;
680 TEST_AND_RETURN_FALSE(block_size >= kExt3MinBlockSize &&
681 block_size <= kExt3MaxBlockSize);
682 TEST_AND_RETURN_FALSE(block_count > 0);
683
684 if (out_block_count) {
685 *out_block_count = block_count;
686 }
687 if (out_block_size) {
688 *out_block_size = block_size;
689 }
690 return true;
691}
692
Alex Deymo032e7722014-03-25 17:53:56 -0700693// Tries to parse the header of an ELF file to obtain a human-readable
694// description of it on the |output| string.
695static bool GetFileFormatELF(const char* buffer, size_t size, string* output) {
696 // 0x00: EI_MAG - ELF magic header, 4 bytes.
697 if (size < 4 || memcmp(buffer, "\x7F""ELF", 4) != 0)
698 return false;
699 *output = "ELF";
700
701 // 0x04: EI_CLASS, 1 byte.
702 if (size < 0x04 + 1)
703 return true;
704 switch (buffer[4]) {
705 case 1:
706 *output += " 32-bit";
707 break;
708 case 2:
709 *output += " 64-bit";
710 break;
711 default:
712 *output += " ?-bit";
713 }
714
715 // 0x05: EI_DATA, endianness, 1 byte.
716 if (size < 0x05 + 1)
717 return true;
718 char ei_data = buffer[5];
719 switch (ei_data) {
720 case 1:
721 *output += " little-endian";
722 break;
723 case 2:
724 *output += " big-endian";
725 break;
726 default:
727 *output += " ?-endian";
728 // Don't parse anything after the 0x10 offset if endianness is unknown.
729 return true;
730 }
731
732 // 0x12: e_machine, 2 byte endianness based on ei_data
733 if (size < 0x12 + 2)
734 return true;
735 uint16 e_machine = *reinterpret_cast<const uint16*>(buffer+0x12);
736 // Fix endianess regardless of the host endianess.
737 if (ei_data == 1)
738 e_machine = le16toh(e_machine);
739 else
740 e_machine = be16toh(e_machine);
741
742 switch (e_machine) {
743 case 0x03:
744 *output += " x86";
745 break;
746 case 0x28:
747 *output += " arm";
748 break;
749 case 0x3E:
750 *output += " x86-64";
751 break;
752 default:
753 *output += " unknown-arch";
754 }
755 return true;
756}
757
758string GetFileFormat(const string& path) {
759 vector<char> buffer;
760 if (!ReadFileChunkAndAppend(path, 0, kGetFileFormatMaxHeaderSize, &buffer))
761 return "File not found.";
762
763 string result;
764 if (GetFileFormatELF(buffer.data(), buffer.size(), &result))
765 return result;
766
767 return "data";
768}
769
Andrew de los Reyesf9714432010-05-04 10:21:23 -0700770bool GetBootloader(BootLoader* out_bootloader) {
771 // For now, hardcode to syslinux.
772 *out_bootloader = BootLoader_SYSLINUX;
773 return true;
774}
775
Darin Petkova0b9e772011-10-06 05:05:56 -0700776string GetAndFreeGError(GError** error) {
777 if (!*error) {
778 return "Unknown GLib error.";
779 }
780 string message =
781 base::StringPrintf("GError(%d): %s",
782 (*error)->code,
783 (*error)->message ? (*error)->message : "(unknown)");
784 g_error_free(*error);
785 *error = NULL;
786 return message;
Andrew de los Reyesc7020782010-04-28 10:46:04 -0700787}
788
Darin Petkov296889c2010-07-23 16:20:54 -0700789bool Reboot() {
790 vector<string> command;
791 command.push_back("/sbin/shutdown");
792 command.push_back("-r");
793 command.push_back("now");
794 int rc = 0;
Darin Petkov85d02b72011-05-17 13:25:51 -0700795 Subprocess::SynchronousExec(command, &rc, NULL);
Darin Petkov296889c2010-07-23 16:20:54 -0700796 TEST_AND_RETURN_FALSE(rc == 0);
797 return true;
798}
799
Andrew de los Reyes712b3ac2011-01-07 13:47:52 -0800800namespace {
801// Do the actual trigger. We do it as a main-loop callback to (try to) get a
802// consistent stack trace.
803gboolean TriggerCrashReporterUpload(void* unused) {
804 pid_t pid = fork();
805 CHECK(pid >= 0) << "fork failed"; // fork() failed. Something is very wrong.
806 if (pid == 0) {
807 // We are the child. Crash.
808 abort(); // never returns
809 }
810 // We are the parent. Wait for child to terminate.
811 pid_t result = waitpid(pid, NULL, 0);
812 LOG_IF(ERROR, result < 0) << "waitpid() failed";
813 return FALSE; // Don't call this callback again
814}
815} // namespace {}
816
817void ScheduleCrashReporterUpload() {
818 g_idle_add(&TriggerCrashReporterUpload, NULL);
819}
820
Chris Sosa4f8ee272012-11-30 13:01:54 -0800821bool SetCpuShares(CpuShares shares) {
822 string string_shares = base::IntToString(static_cast<int>(shares));
823 string cpu_shares_file = string(utils::kCGroupDir) + "/cpu.shares";
824 LOG(INFO) << "Setting cgroup cpu shares to " << string_shares;
825 if(utils::WriteFile(cpu_shares_file.c_str(), string_shares.c_str(),
826 string_shares.size())){
827 return true;
828 } else {
829 LOG(ERROR) << "Failed to change cgroup cpu shares to "<< string_shares
830 << " using " << cpu_shares_file;
831 return false;
832 }
Darin Petkovc6c135c2010-08-11 13:36:18 -0700833}
834
Chris Sosa4f8ee272012-11-30 13:01:54 -0800835int CompareCpuShares(CpuShares shares_lhs,
836 CpuShares shares_rhs) {
837 return static_cast<int>(shares_lhs) - static_cast<int>(shares_rhs);
Darin Petkovc6c135c2010-08-11 13:36:18 -0700838}
839
Darin Petkov5c0a8af2010-08-24 13:39:13 -0700840int FuzzInt(int value, unsigned int range) {
841 int min = value - range / 2;
842 int max = value + range - range / 2;
843 return base::RandInt(min, max);
844}
845
Andrew de los Reyesf3ed8e72011-02-16 10:35:46 -0800846gboolean GlibRunClosure(gpointer data) {
847 google::protobuf::Closure* callback =
848 reinterpret_cast<google::protobuf::Closure*>(data);
849 callback->Run();
850 return FALSE;
851}
852
Alex Deymoc4acdf42014-05-28 21:07:10 -0700853void GlibDestroyClosure(gpointer data) {
854 delete reinterpret_cast<google::protobuf::Closure*>(data);
855}
856
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700857string FormatSecs(unsigned secs) {
Gilad Arnold8e3f1262013-01-08 14:59:54 -0800858 return FormatTimeDelta(TimeDelta::FromSeconds(secs));
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700859}
860
Gilad Arnold8e3f1262013-01-08 14:59:54 -0800861string FormatTimeDelta(TimeDelta delta) {
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700862 // Canonicalize into days, hours, minutes, seconds and microseconds.
863 unsigned days = delta.InDays();
Gilad Arnold8e3f1262013-01-08 14:59:54 -0800864 delta -= TimeDelta::FromDays(days);
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700865 unsigned hours = delta.InHours();
Gilad Arnold8e3f1262013-01-08 14:59:54 -0800866 delta -= TimeDelta::FromHours(hours);
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700867 unsigned mins = delta.InMinutes();
Gilad Arnold8e3f1262013-01-08 14:59:54 -0800868 delta -= TimeDelta::FromMinutes(mins);
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700869 unsigned secs = delta.InSeconds();
Gilad Arnold8e3f1262013-01-08 14:59:54 -0800870 delta -= TimeDelta::FromSeconds(secs);
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700871 unsigned usecs = delta.InMicroseconds();
Gilad Arnold1ebd8132012-03-05 10:19:29 -0800872
873 // Construct and return string.
874 string str;
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700875 if (days)
876 base::StringAppendF(&str, "%ud", days);
877 if (days || hours)
Gilad Arnold1ebd8132012-03-05 10:19:29 -0800878 base::StringAppendF(&str, "%uh", hours);
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700879 if (days || hours || mins)
Gilad Arnold1ebd8132012-03-05 10:19:29 -0800880 base::StringAppendF(&str, "%um", mins);
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700881 base::StringAppendF(&str, "%u", secs);
882 if (usecs) {
883 int width = 6;
884 while ((usecs / 10) * 10 == usecs) {
885 usecs /= 10;
886 width--;
887 }
888 base::StringAppendF(&str, ".%0*u", width, usecs);
889 }
890 base::StringAppendF(&str, "s");
Gilad Arnold1ebd8132012-03-05 10:19:29 -0800891 return str;
892}
893
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700894string ToString(const Time utc_time) {
895 Time::Exploded exp_time;
896 utc_time.UTCExplode(&exp_time);
Alex Vakulenko75039d72014-03-25 12:36:28 -0700897 return base::StringPrintf("%d/%d/%d %d:%02d:%02d GMT",
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700898 exp_time.month,
899 exp_time.day_of_month,
900 exp_time.year,
901 exp_time.hour,
902 exp_time.minute,
903 exp_time.second);
904}
adlr@google.com3defe6a2009-12-04 20:57:17 +0000905
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700906string ToString(bool b) {
907 return (b ? "true" : "false");
908}
909
Alex Deymo1c656c42013-06-28 11:02:14 -0700910string ToString(DownloadSource source) {
Jay Srinivasan19409b72013-04-12 19:23:36 -0700911 switch (source) {
912 case kDownloadSourceHttpsServer: return "HttpsServer";
913 case kDownloadSourceHttpServer: return "HttpServer";
David Zeuthenbb8bdc72013-09-03 13:43:48 -0700914 case kDownloadSourceHttpPeer: return "HttpPeer";
Jay Srinivasan19409b72013-04-12 19:23:36 -0700915 case kNumDownloadSources: return "Unknown";
916 // Don't add a default case to let the compiler warn about newly added
917 // download sources which should be added here.
918 }
919
920 return "Unknown";
921}
922
Alex Deymo1c656c42013-06-28 11:02:14 -0700923string ToString(PayloadType payload_type) {
924 switch (payload_type) {
925 case kPayloadTypeDelta: return "Delta";
926 case kPayloadTypeFull: return "Full";
927 case kPayloadTypeForcedFull: return "ForcedFull";
928 case kNumPayloadTypes: return "Unknown";
929 // Don't add a default case to let the compiler warn about newly added
930 // payload types which should be added here.
931 }
932
933 return "Unknown";
934}
935
David Zeuthena99981f2013-04-29 13:42:47 -0700936ErrorCode GetBaseErrorCode(ErrorCode code) {
Jay Srinivasanf0572052012-10-23 18:12:56 -0700937 // Ignore the higher order bits in the code by applying the mask as
938 // we want the enumerations to be in the small contiguous range
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700939 // with values less than ErrorCode::kUmaReportedMax.
940 ErrorCode base_code = static_cast<ErrorCode>(
941 static_cast<int>(code) & ~static_cast<int>(ErrorCode::kSpecialFlags));
Jay Srinivasanf0572052012-10-23 18:12:56 -0700942
Jay Srinivasan2b5a0f02012-12-19 17:25:56 -0800943 // Make additional adjustments required for UMA and error classification.
944 // TODO(jaysri): Move this logic to UeErrorCode.cc when we fix
945 // chromium-os:34369.
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700946 if (base_code >= ErrorCode::kOmahaRequestHTTPResponseBase) {
Jay Srinivasanf0572052012-10-23 18:12:56 -0700947 // Since we want to keep the enums to a small value, aggregate all HTTP
Jay Srinivasan2b5a0f02012-12-19 17:25:56 -0800948 // errors into this one bucket for UMA and error classification purposes.
Jay Srinivasan55f50c22013-01-10 19:24:35 -0800949 LOG(INFO) << "Converting error code " << base_code
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700950 << " to ErrorCode::kOmahaErrorInHTTPResponse";
951 base_code = ErrorCode::kOmahaErrorInHTTPResponse;
Jay Srinivasanf0572052012-10-23 18:12:56 -0700952 }
953
Jay Srinivasan2b5a0f02012-12-19 17:25:56 -0800954 return base_code;
955}
956
David Zeuthen33bae492014-02-25 16:16:18 -0800957metrics::AttemptResult GetAttemptResult(ErrorCode code) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700958 ErrorCode base_code = static_cast<ErrorCode>(
959 static_cast<int>(code) & ~static_cast<int>(ErrorCode::kSpecialFlags));
David Zeuthen33bae492014-02-25 16:16:18 -0800960
961 switch (base_code) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700962 case ErrorCode::kSuccess:
David Zeuthen33bae492014-02-25 16:16:18 -0800963 return metrics::AttemptResult::kUpdateSucceeded;
964
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700965 case ErrorCode::kDownloadTransferError:
David Zeuthen33bae492014-02-25 16:16:18 -0800966 return metrics::AttemptResult::kPayloadDownloadError;
967
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700968 case ErrorCode::kDownloadInvalidMetadataSize:
969 case ErrorCode::kDownloadInvalidMetadataMagicString:
970 case ErrorCode::kDownloadMetadataSignatureError:
971 case ErrorCode::kDownloadMetadataSignatureVerificationError:
972 case ErrorCode::kPayloadMismatchedType:
973 case ErrorCode::kUnsupportedMajorPayloadVersion:
974 case ErrorCode::kUnsupportedMinorPayloadVersion:
975 case ErrorCode::kDownloadNewPartitionInfoError:
976 case ErrorCode::kDownloadSignatureMissingInManifest:
977 case ErrorCode::kDownloadManifestParseError:
978 case ErrorCode::kDownloadOperationHashMissingError:
David Zeuthen33bae492014-02-25 16:16:18 -0800979 return metrics::AttemptResult::kMetadataMalformed;
980
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700981 case ErrorCode::kDownloadOperationHashMismatch:
982 case ErrorCode::kDownloadOperationHashVerificationError:
David Zeuthen33bae492014-02-25 16:16:18 -0800983 return metrics::AttemptResult::kOperationMalformed;
984
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700985 case ErrorCode::kDownloadOperationExecutionError:
986 case ErrorCode::kInstallDeviceOpenError:
987 case ErrorCode::kKernelDeviceOpenError:
988 case ErrorCode::kDownloadWriteError:
989 case ErrorCode::kFilesystemCopierError:
David Zeuthen33bae492014-02-25 16:16:18 -0800990 return metrics::AttemptResult::kOperationExecutionError;
991
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700992 case ErrorCode::kDownloadMetadataSignatureMismatch:
David Zeuthen33bae492014-02-25 16:16:18 -0800993 return metrics::AttemptResult::kMetadataVerificationFailed;
994
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700995 case ErrorCode::kPayloadSizeMismatchError:
996 case ErrorCode::kPayloadHashMismatchError:
997 case ErrorCode::kDownloadPayloadVerificationError:
998 case ErrorCode::kSignedDeltaPayloadExpectedError:
999 case ErrorCode::kDownloadPayloadPubKeyVerificationError:
David Zeuthen33bae492014-02-25 16:16:18 -08001000 return metrics::AttemptResult::kPayloadVerificationFailed;
1001
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001002 case ErrorCode::kNewRootfsVerificationError:
1003 case ErrorCode::kNewKernelVerificationError:
David Zeuthen33bae492014-02-25 16:16:18 -08001004 return metrics::AttemptResult::kVerificationFailed;
1005
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001006 case ErrorCode::kPostinstallRunnerError:
1007 case ErrorCode::kPostinstallBootedFromFirmwareB:
1008 case ErrorCode::kPostinstallFirmwareRONotUpdatable:
David Zeuthen33bae492014-02-25 16:16:18 -08001009 return metrics::AttemptResult::kPostInstallFailed;
1010
1011 // We should never get these errors in the update-attempt stage so
1012 // return internal error if this happens.
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001013 case ErrorCode::kError:
1014 case ErrorCode::kOmahaRequestXMLParseError:
1015 case ErrorCode::kOmahaRequestError:
1016 case ErrorCode::kOmahaResponseHandlerError:
1017 case ErrorCode::kDownloadStateInitializationError:
1018 case ErrorCode::kOmahaRequestEmptyResponseError:
1019 case ErrorCode::kDownloadInvalidMetadataSignature:
1020 case ErrorCode::kOmahaResponseInvalid:
1021 case ErrorCode::kOmahaUpdateIgnoredPerPolicy:
1022 case ErrorCode::kOmahaUpdateDeferredPerPolicy:
1023 case ErrorCode::kOmahaErrorInHTTPResponse:
1024 case ErrorCode::kDownloadMetadataSignatureMissingError:
1025 case ErrorCode::kOmahaUpdateDeferredForBackoff:
1026 case ErrorCode::kPostinstallPowerwashError:
1027 case ErrorCode::kUpdateCanceledByChannelChange:
David Zeuthen33bae492014-02-25 16:16:18 -08001028 return metrics::AttemptResult::kInternalError;
1029
1030 // Special flags. These can't happen (we mask them out above) but
1031 // the compiler doesn't know that. Just break out so we can warn and
1032 // return |kInternalError|.
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001033 case ErrorCode::kUmaReportedMax:
1034 case ErrorCode::kOmahaRequestHTTPResponseBase:
1035 case ErrorCode::kDevModeFlag:
1036 case ErrorCode::kResumedFlag:
1037 case ErrorCode::kTestImageFlag:
1038 case ErrorCode::kTestOmahaUrlFlag:
1039 case ErrorCode::kSpecialFlags:
David Zeuthen33bae492014-02-25 16:16:18 -08001040 break;
1041 }
1042
1043 LOG(ERROR) << "Unexpected error code " << base_code;
1044 return metrics::AttemptResult::kInternalError;
1045}
1046
1047
1048metrics::DownloadErrorCode GetDownloadErrorCode(ErrorCode code) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001049 ErrorCode base_code = static_cast<ErrorCode>(
1050 static_cast<int>(code) & ~static_cast<int>(ErrorCode::kSpecialFlags));
David Zeuthen33bae492014-02-25 16:16:18 -08001051
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001052 if (base_code >= ErrorCode::kOmahaRequestHTTPResponseBase) {
1053 int http_status =
1054 static_cast<int>(base_code) -
1055 static_cast<int>(ErrorCode::kOmahaRequestHTTPResponseBase);
David Zeuthen33bae492014-02-25 16:16:18 -08001056 if (http_status >= 200 && http_status <= 599) {
1057 return static_cast<metrics::DownloadErrorCode>(
1058 static_cast<int>(metrics::DownloadErrorCode::kHttpStatus200) +
1059 http_status - 200);
1060 } else if (http_status == 0) {
1061 // The code is using HTTP Status 0 for "Unable to get http
1062 // response code."
1063 return metrics::DownloadErrorCode::kDownloadError;
1064 }
1065 LOG(WARNING) << "Unexpected HTTP status code " << http_status;
1066 return metrics::DownloadErrorCode::kHttpStatusOther;
1067 }
1068
1069 switch (base_code) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001070 // Unfortunately, ErrorCode::kDownloadTransferError is returned for a wide
David Zeuthen33bae492014-02-25 16:16:18 -08001071 // variety of errors (proxy errors, host not reachable, timeouts etc.).
1072 //
1073 // For now just map that to kDownloading. See http://crbug.com/355745
1074 // for how we plan to add more detail in the future.
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001075 case ErrorCode::kDownloadTransferError:
David Zeuthen33bae492014-02-25 16:16:18 -08001076 return metrics::DownloadErrorCode::kDownloadError;
1077
1078 // All of these error codes are not related to downloading so break
1079 // out so we can warn and return InputMalformed.
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001080 case ErrorCode::kSuccess:
1081 case ErrorCode::kError:
1082 case ErrorCode::kOmahaRequestError:
1083 case ErrorCode::kOmahaResponseHandlerError:
1084 case ErrorCode::kFilesystemCopierError:
1085 case ErrorCode::kPostinstallRunnerError:
1086 case ErrorCode::kPayloadMismatchedType:
1087 case ErrorCode::kInstallDeviceOpenError:
1088 case ErrorCode::kKernelDeviceOpenError:
1089 case ErrorCode::kPayloadHashMismatchError:
1090 case ErrorCode::kPayloadSizeMismatchError:
1091 case ErrorCode::kDownloadPayloadVerificationError:
1092 case ErrorCode::kDownloadNewPartitionInfoError:
1093 case ErrorCode::kDownloadWriteError:
1094 case ErrorCode::kNewRootfsVerificationError:
1095 case ErrorCode::kNewKernelVerificationError:
1096 case ErrorCode::kSignedDeltaPayloadExpectedError:
1097 case ErrorCode::kDownloadPayloadPubKeyVerificationError:
1098 case ErrorCode::kPostinstallBootedFromFirmwareB:
1099 case ErrorCode::kDownloadStateInitializationError:
1100 case ErrorCode::kDownloadInvalidMetadataMagicString:
1101 case ErrorCode::kDownloadSignatureMissingInManifest:
1102 case ErrorCode::kDownloadManifestParseError:
1103 case ErrorCode::kDownloadMetadataSignatureError:
1104 case ErrorCode::kDownloadMetadataSignatureVerificationError:
1105 case ErrorCode::kDownloadMetadataSignatureMismatch:
1106 case ErrorCode::kDownloadOperationHashVerificationError:
1107 case ErrorCode::kDownloadOperationExecutionError:
1108 case ErrorCode::kDownloadOperationHashMismatch:
1109 case ErrorCode::kOmahaRequestEmptyResponseError:
1110 case ErrorCode::kOmahaRequestXMLParseError:
1111 case ErrorCode::kDownloadInvalidMetadataSize:
1112 case ErrorCode::kDownloadInvalidMetadataSignature:
1113 case ErrorCode::kOmahaResponseInvalid:
1114 case ErrorCode::kOmahaUpdateIgnoredPerPolicy:
1115 case ErrorCode::kOmahaUpdateDeferredPerPolicy:
1116 case ErrorCode::kOmahaErrorInHTTPResponse:
1117 case ErrorCode::kDownloadOperationHashMissingError:
1118 case ErrorCode::kDownloadMetadataSignatureMissingError:
1119 case ErrorCode::kOmahaUpdateDeferredForBackoff:
1120 case ErrorCode::kPostinstallPowerwashError:
1121 case ErrorCode::kUpdateCanceledByChannelChange:
1122 case ErrorCode::kPostinstallFirmwareRONotUpdatable:
1123 case ErrorCode::kUnsupportedMajorPayloadVersion:
1124 case ErrorCode::kUnsupportedMinorPayloadVersion:
David Zeuthen33bae492014-02-25 16:16:18 -08001125 break;
1126
1127 // Special flags. These can't happen (we mask them out above) but
1128 // the compiler doesn't know that. Just break out so we can warn and
1129 // return |kInputMalformed|.
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001130 case ErrorCode::kUmaReportedMax:
1131 case ErrorCode::kOmahaRequestHTTPResponseBase:
1132 case ErrorCode::kDevModeFlag:
1133 case ErrorCode::kResumedFlag:
1134 case ErrorCode::kTestImageFlag:
1135 case ErrorCode::kTestOmahaUrlFlag:
1136 case ErrorCode::kSpecialFlags:
David Zeuthen33bae492014-02-25 16:16:18 -08001137 LOG(ERROR) << "Unexpected error code " << base_code;
1138 break;
1139 }
1140
1141 return metrics::DownloadErrorCode::kInputMalformed;
1142}
1143
David Zeuthenb281f072014-04-02 10:20:19 -07001144metrics::ConnectionType GetConnectionType(
1145 NetworkConnectionType type,
1146 NetworkTethering tethering) {
1147 switch (type) {
1148 case kNetUnknown:
1149 return metrics::ConnectionType::kUnknown;
1150
1151 case kNetEthernet:
1152 if (tethering == NetworkTethering::kConfirmed)
1153 return metrics::ConnectionType::kTetheredEthernet;
1154 else
1155 return metrics::ConnectionType::kEthernet;
1156
1157 case kNetWifi:
1158 if (tethering == NetworkTethering::kConfirmed)
1159 return metrics::ConnectionType::kTetheredWifi;
1160 else
1161 return metrics::ConnectionType::kWifi;
1162
1163 case kNetWimax:
1164 return metrics::ConnectionType::kWimax;
1165
1166 case kNetBluetooth:
1167 return metrics::ConnectionType::kBluetooth;
1168
1169 case kNetCellular:
1170 return metrics::ConnectionType::kCellular;
1171 }
1172
1173 LOG(ERROR) << "Unexpected network connection type: type=" << type
1174 << ", tethering=" << static_cast<int>(tethering);
1175
1176 return metrics::ConnectionType::kUnknown;
1177}
1178
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001179// Returns a printable version of the various flags denoted in the higher order
1180// bits of the given code. Returns an empty string if none of those bits are
1181// set.
1182string GetFlagNames(uint32_t code) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001183 uint32_t flags = (static_cast<uint32_t>(code) &
1184 static_cast<uint32_t>(ErrorCode::kSpecialFlags));
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001185 string flag_names;
1186 string separator = "";
1187 for(size_t i = 0; i < sizeof(flags) * 8; i++) {
1188 uint32_t flag = flags & (1 << i);
1189 if (flag) {
David Zeuthena99981f2013-04-29 13:42:47 -07001190 flag_names += separator + CodeToString(static_cast<ErrorCode>(flag));
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001191 separator = ", ";
1192 }
1193 }
Jay Srinivasan2b5a0f02012-12-19 17:25:56 -08001194
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001195 return flag_names;
Jay Srinivasanf0572052012-10-23 18:12:56 -07001196}
1197
David Zeuthena99981f2013-04-29 13:42:47 -07001198void SendErrorCodeToUma(SystemState* system_state, ErrorCode code) {
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001199 if (!system_state)
1200 return;
1201
David Zeuthena99981f2013-04-29 13:42:47 -07001202 ErrorCode uma_error_code = GetBaseErrorCode(code);
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001203
1204 // If the code doesn't have flags computed already, compute them now based on
1205 // the state of the current update attempt.
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001206 uint32_t flags =
1207 static_cast<int>(code) & static_cast<int>(ErrorCode::kSpecialFlags);
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001208 if (!flags)
1209 flags = system_state->update_attempter()->GetErrorCodeFlags();
1210
1211 // Determine the UMA bucket depending on the flags. But, ignore the resumed
1212 // flag, as it's perfectly normal for production devices to resume their
1213 // downloads and so we want to record those cases also in NormalErrorCodes
1214 // bucket.
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001215 string metric =
1216 flags & ~static_cast<uint32_t>(ErrorCode::kResumedFlag) ?
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001217 "Installer.DevModeErrorCodes" : "Installer.NormalErrorCodes";
1218
1219 LOG(INFO) << "Sending error code " << uma_error_code
1220 << " (" << CodeToString(uma_error_code) << ")"
1221 << " to UMA metric: " << metric
1222 << ". Flags = " << (flags ? GetFlagNames(flags) : "None");
1223
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001224 system_state->metrics_lib()->SendEnumToUMA(
1225 metric, static_cast<int>(uma_error_code),
1226 static_cast<int>(ErrorCode::kUmaReportedMax));
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001227}
1228
David Zeuthena99981f2013-04-29 13:42:47 -07001229string CodeToString(ErrorCode code) {
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001230 // If the given code has both parts (i.e. the error code part and the flags
1231 // part) then strip off the flags part since the switch statement below
1232 // has case statements only for the base error code or a single flag but
1233 // doesn't support any combinations of those.
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001234 if ((static_cast<int>(code) & static_cast<int>(ErrorCode::kSpecialFlags)) &&
1235 (static_cast<int>(code) & ~static_cast<int>(ErrorCode::kSpecialFlags)))
1236 code = static_cast<ErrorCode>(
1237 static_cast<int>(code) & ~static_cast<int>(ErrorCode::kSpecialFlags));
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001238 switch (code) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001239 case ErrorCode::kSuccess: return "ErrorCode::kSuccess";
1240 case ErrorCode::kError: return "ErrorCode::kError";
1241 case ErrorCode::kOmahaRequestError: return "ErrorCode::kOmahaRequestError";
1242 case ErrorCode::kOmahaResponseHandlerError:
1243 return "ErrorCode::kOmahaResponseHandlerError";
1244 case ErrorCode::kFilesystemCopierError:
1245 return "ErrorCode::kFilesystemCopierError";
1246 case ErrorCode::kPostinstallRunnerError:
1247 return "ErrorCode::kPostinstallRunnerError";
1248 case ErrorCode::kPayloadMismatchedType:
1249 return "ErrorCode::kPayloadMismatchedType";
1250 case ErrorCode::kInstallDeviceOpenError:
1251 return "ErrorCode::kInstallDeviceOpenError";
1252 case ErrorCode::kKernelDeviceOpenError:
1253 return "ErrorCode::kKernelDeviceOpenError";
1254 case ErrorCode::kDownloadTransferError:
1255 return "ErrorCode::kDownloadTransferError";
1256 case ErrorCode::kPayloadHashMismatchError:
1257 return "ErrorCode::kPayloadHashMismatchError";
1258 case ErrorCode::kPayloadSizeMismatchError:
1259 return "ErrorCode::kPayloadSizeMismatchError";
1260 case ErrorCode::kDownloadPayloadVerificationError:
1261 return "ErrorCode::kDownloadPayloadVerificationError";
1262 case ErrorCode::kDownloadNewPartitionInfoError:
1263 return "ErrorCode::kDownloadNewPartitionInfoError";
1264 case ErrorCode::kDownloadWriteError:
1265 return "ErrorCode::kDownloadWriteError";
1266 case ErrorCode::kNewRootfsVerificationError:
1267 return "ErrorCode::kNewRootfsVerificationError";
1268 case ErrorCode::kNewKernelVerificationError:
1269 return "ErrorCode::kNewKernelVerificationError";
1270 case ErrorCode::kSignedDeltaPayloadExpectedError:
1271 return "ErrorCode::kSignedDeltaPayloadExpectedError";
1272 case ErrorCode::kDownloadPayloadPubKeyVerificationError:
1273 return "ErrorCode::kDownloadPayloadPubKeyVerificationError";
1274 case ErrorCode::kPostinstallBootedFromFirmwareB:
1275 return "ErrorCode::kPostinstallBootedFromFirmwareB";
1276 case ErrorCode::kDownloadStateInitializationError:
1277 return "ErrorCode::kDownloadStateInitializationError";
1278 case ErrorCode::kDownloadInvalidMetadataMagicString:
1279 return "ErrorCode::kDownloadInvalidMetadataMagicString";
1280 case ErrorCode::kDownloadSignatureMissingInManifest:
1281 return "ErrorCode::kDownloadSignatureMissingInManifest";
1282 case ErrorCode::kDownloadManifestParseError:
1283 return "ErrorCode::kDownloadManifestParseError";
1284 case ErrorCode::kDownloadMetadataSignatureError:
1285 return "ErrorCode::kDownloadMetadataSignatureError";
1286 case ErrorCode::kDownloadMetadataSignatureVerificationError:
1287 return "ErrorCode::kDownloadMetadataSignatureVerificationError";
1288 case ErrorCode::kDownloadMetadataSignatureMismatch:
1289 return "ErrorCode::kDownloadMetadataSignatureMismatch";
1290 case ErrorCode::kDownloadOperationHashVerificationError:
1291 return "ErrorCode::kDownloadOperationHashVerificationError";
1292 case ErrorCode::kDownloadOperationExecutionError:
1293 return "ErrorCode::kDownloadOperationExecutionError";
1294 case ErrorCode::kDownloadOperationHashMismatch:
1295 return "ErrorCode::kDownloadOperationHashMismatch";
1296 case ErrorCode::kOmahaRequestEmptyResponseError:
1297 return "ErrorCode::kOmahaRequestEmptyResponseError";
1298 case ErrorCode::kOmahaRequestXMLParseError:
1299 return "ErrorCode::kOmahaRequestXMLParseError";
1300 case ErrorCode::kDownloadInvalidMetadataSize:
1301 return "ErrorCode::kDownloadInvalidMetadataSize";
1302 case ErrorCode::kDownloadInvalidMetadataSignature:
1303 return "ErrorCode::kDownloadInvalidMetadataSignature";
1304 case ErrorCode::kOmahaResponseInvalid:
1305 return "ErrorCode::kOmahaResponseInvalid";
1306 case ErrorCode::kOmahaUpdateIgnoredPerPolicy:
1307 return "ErrorCode::kOmahaUpdateIgnoredPerPolicy";
1308 case ErrorCode::kOmahaUpdateDeferredPerPolicy:
1309 return "ErrorCode::kOmahaUpdateDeferredPerPolicy";
1310 case ErrorCode::kOmahaErrorInHTTPResponse:
1311 return "ErrorCode::kOmahaErrorInHTTPResponse";
1312 case ErrorCode::kDownloadOperationHashMissingError:
1313 return "ErrorCode::kDownloadOperationHashMissingError";
1314 case ErrorCode::kDownloadMetadataSignatureMissingError:
1315 return "ErrorCode::kDownloadMetadataSignatureMissingError";
1316 case ErrorCode::kOmahaUpdateDeferredForBackoff:
1317 return "ErrorCode::kOmahaUpdateDeferredForBackoff";
1318 case ErrorCode::kPostinstallPowerwashError:
1319 return "ErrorCode::kPostinstallPowerwashError";
1320 case ErrorCode::kUpdateCanceledByChannelChange:
1321 return "ErrorCode::kUpdateCanceledByChannelChange";
1322 case ErrorCode::kUmaReportedMax:
1323 return "ErrorCode::kUmaReportedMax";
1324 case ErrorCode::kOmahaRequestHTTPResponseBase:
1325 return "ErrorCode::kOmahaRequestHTTPResponseBase";
1326 case ErrorCode::kResumedFlag:
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001327 return "Resumed";
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001328 case ErrorCode::kDevModeFlag:
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001329 return "DevMode";
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001330 case ErrorCode::kTestImageFlag:
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001331 return "TestImage";
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001332 case ErrorCode::kTestOmahaUrlFlag:
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001333 return "TestOmahaUrl";
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001334 case ErrorCode::kSpecialFlags:
1335 return "ErrorCode::kSpecialFlags";
1336 case ErrorCode::kPostinstallFirmwareRONotUpdatable:
1337 return "ErrorCode::kPostinstallFirmwareRONotUpdatable";
1338 case ErrorCode::kUnsupportedMajorPayloadVersion:
1339 return "ErrorCode::kUnsupportedMajorPayloadVersion";
1340 case ErrorCode::kUnsupportedMinorPayloadVersion:
1341 return "ErrorCode::kUnsupportedMinorPayloadVersion";
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001342 // Don't add a default case to let the compiler warn about newly added
1343 // error codes which should be added here.
1344 }
1345
1346 return "Unknown error: " + base::UintToString(static_cast<unsigned>(code));
1347}
Jay Srinivasanf0572052012-10-23 18:12:56 -07001348
Gilad Arnold30dedd82013-07-03 06:19:09 -07001349bool CreatePowerwashMarkerFile(const char* file_path) {
1350 const char* marker_file = file_path ? file_path : kPowerwashMarkerFile;
1351 bool result = utils::WriteFile(marker_file,
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001352 kPowerwashCommand,
1353 strlen(kPowerwashCommand));
Gilad Arnold30dedd82013-07-03 06:19:09 -07001354 if (result) {
1355 LOG(INFO) << "Created " << marker_file << " to powerwash on next reboot";
1356 } else {
1357 PLOG(ERROR) << "Error in creating powerwash marker file: " << marker_file;
1358 }
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001359
1360 return result;
1361}
1362
Gilad Arnold30dedd82013-07-03 06:19:09 -07001363bool DeletePowerwashMarkerFile(const char* file_path) {
1364 const char* marker_file = file_path ? file_path : kPowerwashMarkerFile;
Alex Vakulenko75039d72014-03-25 12:36:28 -07001365 const base::FilePath kPowerwashMarkerPath(marker_file);
1366 bool result = base::DeleteFile(kPowerwashMarkerPath, false);
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001367
1368 if (result)
1369 LOG(INFO) << "Successfully deleted the powerwash marker file : "
Gilad Arnold30dedd82013-07-03 06:19:09 -07001370 << marker_file;
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001371 else
1372 PLOG(ERROR) << "Could not delete the powerwash marker file : "
Gilad Arnold30dedd82013-07-03 06:19:09 -07001373 << marker_file;
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001374
1375 return result;
1376}
1377
Chris Sosad317e402013-06-12 13:47:09 -07001378bool GetInstallDev(const std::string& boot_dev, std::string* install_dev) {
Alex Vakulenkof3f85bb2014-03-26 16:36:35 -07001379 std::string disk_name;
1380 int partition_num;
1381 if (!SplitPartitionName(boot_dev, &disk_name, &partition_num))
1382 return false;
Liam McLoughlin049d1652013-07-31 18:47:46 -07001383
Chris Sosad317e402013-06-12 13:47:09 -07001384 // Right now, we just switch '3' and '5' partition numbers.
Alex Vakulenkof3f85bb2014-03-26 16:36:35 -07001385 if (partition_num == 3) {
1386 partition_num = 5;
1387 } else if (partition_num == 5) {
1388 partition_num = 3;
1389 } else {
1390 return false;
1391 }
1392
1393 if (install_dev)
1394 *install_dev = MakePartitionName(disk_name, partition_num);
Liam McLoughlin049d1652013-07-31 18:47:46 -07001395
Chris Sosad317e402013-06-12 13:47:09 -07001396 return true;
1397}
1398
David Zeuthen27a48bc2013-08-06 12:06:29 -07001399Time TimeFromStructTimespec(struct timespec *ts) {
1400 int64 us = static_cast<int64>(ts->tv_sec) * Time::kMicrosecondsPerSecond +
1401 static_cast<int64>(ts->tv_nsec) / Time::kNanosecondsPerMicrosecond;
1402 return Time::UnixEpoch() + TimeDelta::FromMicroseconds(us);
1403}
1404
1405gchar** StringVectorToGStrv(const vector<string> &vector) {
1406 GPtrArray *p = g_ptr_array_new();
1407 for (std::vector<string>::const_iterator i = vector.begin();
1408 i != vector.end(); ++i) {
1409 g_ptr_array_add(p, g_strdup(i->c_str()));
1410 }
1411 g_ptr_array_add(p, NULL);
1412 return reinterpret_cast<gchar**>(g_ptr_array_free(p, FALSE));
1413}
1414
1415string StringVectorToString(const vector<string> &vector) {
1416 string str = "[";
1417 for (std::vector<string>::const_iterator i = vector.begin();
1418 i != vector.end(); ++i) {
1419 if (i != vector.begin())
1420 str += ", ";
1421 str += '"';
1422 str += *i;
1423 str += '"';
1424 }
1425 str += "]";
1426 return str;
1427}
1428
David Zeuthen8f191b22013-08-06 12:27:50 -07001429string CalculateP2PFileId(const string& payload_hash, size_t payload_size) {
1430 string encoded_hash;
1431 OmahaHashCalculator::Base64Encode(payload_hash.c_str(),
1432 payload_hash.size(),
1433 &encoded_hash);
Alex Vakulenko75039d72014-03-25 12:36:28 -07001434 return base::StringPrintf("cros_update_size_%zu_hash_%s",
David Zeuthen8f191b22013-08-06 12:27:50 -07001435 payload_size,
1436 encoded_hash.c_str());
1437}
1438
David Zeuthen910ec5b2013-09-26 12:10:58 -07001439bool IsXAttrSupported(const base::FilePath& dir_path) {
1440 char *path = strdup(dir_path.Append("xattr_test_XXXXXX").value().c_str());
1441
1442 int fd = mkstemp(path);
1443 if (fd == -1) {
1444 PLOG(ERROR) << "Error creating temporary file in " << dir_path.value();
1445 free(path);
1446 return false;
1447 }
1448
1449 if (unlink(path) != 0) {
1450 PLOG(ERROR) << "Error unlinking temporary file " << path;
1451 close(fd);
1452 free(path);
1453 return false;
1454 }
1455
1456 int xattr_res = fsetxattr(fd, "user.xattr-test", "value", strlen("value"), 0);
1457 if (xattr_res != 0) {
1458 if (errno == ENOTSUP) {
1459 // Leave it to call-sites to warn about non-support.
1460 } else {
1461 PLOG(ERROR) << "Error setting xattr on " << path;
1462 }
1463 }
1464 close(fd);
1465 free(path);
1466 return xattr_res == 0;
1467}
1468
David Zeuthene7f89172013-10-31 10:21:04 -07001469bool DecodeAndStoreBase64String(const std::string& base64_encoded,
1470 base::FilePath *out_path) {
1471 vector<char> contents;
1472
1473 out_path->clear();
1474
1475 if (base64_encoded.size() == 0) {
1476 LOG(ERROR) << "Can't decode empty string.";
1477 return false;
1478 }
1479
1480 if (!OmahaHashCalculator::Base64Decode(base64_encoded, &contents) ||
1481 contents.size() == 0) {
1482 LOG(ERROR) << "Error decoding base64.";
1483 return false;
1484 }
1485
Alex Vakulenko75039d72014-03-25 12:36:28 -07001486 FILE *file = base::CreateAndOpenTemporaryFile(out_path);
David Zeuthene7f89172013-10-31 10:21:04 -07001487 if (file == NULL) {
1488 LOG(ERROR) << "Error creating temporary file.";
1489 return false;
1490 }
1491
1492 if (fwrite(&contents[0], 1, contents.size(), file) != contents.size()) {
1493 PLOG(ERROR) << "Error writing to temporary file.";
1494 if (fclose(file) != 0)
1495 PLOG(ERROR) << "Error closing temporary file.";
1496 if (unlink(out_path->value().c_str()) != 0)
1497 PLOG(ERROR) << "Error unlinking temporary file.";
1498 out_path->clear();
1499 return false;
1500 }
1501
1502 if (fclose(file) != 0) {
1503 PLOG(ERROR) << "Error closing temporary file.";
1504 out_path->clear();
1505 return false;
1506 }
1507
1508 return true;
1509}
1510
David Zeuthen639aa362014-02-03 16:23:44 -08001511bool ConvertToOmahaInstallDate(base::Time time, int *out_num_days) {
1512 time_t unix_time = time.ToTimeT();
1513 // Output of: date +"%s" --date="Jan 1, 2007 0:00 PST".
1514 const time_t kOmahaEpoch = 1167638400;
1515 const int64_t kNumSecondsPerWeek = 7*24*3600;
1516 const int64_t kNumDaysPerWeek = 7;
1517
1518 time_t omaha_time = unix_time - kOmahaEpoch;
1519
1520 if (omaha_time < 0)
1521 return false;
1522
1523 // Note, as per the comment in utils.h we are deliberately not
1524 // handling DST correctly.
1525
1526 int64_t num_weeks_since_omaha_epoch = omaha_time / kNumSecondsPerWeek;
1527 *out_num_days = num_weeks_since_omaha_epoch * kNumDaysPerWeek;
1528
1529 return true;
1530}
1531
David Zeuthen33bae492014-02-25 16:16:18 -08001532bool WallclockDurationHelper(SystemState* system_state,
1533 const std::string& state_variable_key,
1534 base::TimeDelta* out_duration) {
1535 bool ret = false;
1536
1537 base::Time now = system_state->clock()->GetWallclockTime();
1538 int64_t stored_value;
1539 if (system_state->prefs()->GetInt64(state_variable_key, &stored_value)) {
1540 base::Time stored_time = base::Time::FromInternalValue(stored_value);
1541 if (stored_time > now) {
1542 LOG(ERROR) << "Stored time-stamp used for " << state_variable_key
1543 << " is in the future.";
1544 } else {
1545 *out_duration = now - stored_time;
1546 ret = true;
1547 }
1548 }
1549
1550 if (!system_state->prefs()->SetInt64(state_variable_key,
1551 now.ToInternalValue())) {
1552 LOG(ERROR) << "Error storing time-stamp in " << state_variable_key;
1553 }
1554
1555 return ret;
1556}
1557
1558bool MonotonicDurationHelper(SystemState* system_state,
1559 int64_t* storage,
1560 base::TimeDelta* out_duration) {
1561 bool ret = false;
1562
1563 base::Time now = system_state->clock()->GetMonotonicTime();
1564 if (*storage != 0) {
1565 base::Time stored_time = base::Time::FromInternalValue(*storage);
1566 *out_duration = now - stored_time;
1567 ret = true;
1568 }
1569 *storage = now.ToInternalValue();
1570
1571 return ret;
1572}
1573
adlr@google.com3defe6a2009-12-04 20:57:17 +00001574} // namespace utils
1575
1576} // namespace chromeos_update_engine