blob: f242c39af70489d34d80612ceb3586c4f6d6b943 [file] [log] [blame]
Gilad Arnoldd7b513d2012-05-10 14:25:27 -07001// 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
Gilad Arnoldcf175a02014-07-10 16:48:47 -07005#ifndef UPDATE_ENGINE_UTILS_H_
6#define UPDATE_ENGINE_UTILS_H_
adlr@google.com3defe6a2009-12-04 20:57:17 +00007
Han Shen2643cb72012-06-26 14:45:33 -07008#include <errno.h>
Alex Vakulenkod2779df2014-06-16 13:19:00 -07009#include <unistd.h>
10
11#include <algorithm>
12#include <map>
Ben Chan02f7c1d2014-10-18 15:18:02 -070013#include <memory>
adlr@google.com3defe6a2009-12-04 20:57:17 +000014#include <set>
15#include <string>
16#include <vector>
Darin Petkovc6c135c2010-08-11 13:36:18 -070017
Alex Vakulenko75039d72014-03-25 12:36:28 -070018#include <base/files/file_path.h>
Chris Sosafc661a12013-02-26 14:43:21 -080019#include <base/posix/eintr_wrapper.h>
Alex Vakulenko75039d72014-03-25 12:36:28 -070020#include <base/time/time.h>
Alex Vakulenkof68bbbc2015-02-09 12:53:18 -080021#include <chromeos/secure_blob.h>
Andrew de los Reyesc7020782010-04-28 10:46:04 -070022#include <glib.h>
Jay Srinivasanf0572052012-10-23 18:12:56 -070023#include "metrics/metrics_library.h"
Darin Petkovc6c135c2010-08-11 13:36:18 -070024
adlr@google.com3defe6a2009-12-04 20:57:17 +000025#include "update_engine/action.h"
David Zeuthenb281f072014-04-02 10:20:19 -070026#include "update_engine/action_processor.h"
27#include "update_engine/connection_manager.h"
Jay Srinivasan19409b72013-04-12 19:23:36 -070028#include "update_engine/constants.h"
Nam T. Nguyenf1d582e2014-12-08 15:07:17 -080029#include "update_engine/file_descriptor.h"
David Zeuthen33bae492014-02-25 16:16:18 -080030#include "update_engine/metrics.h"
Allie Wood56873452015-03-27 17:48:40 -070031#include "update_engine/update_metadata.pb.h"
adlr@google.com3defe6a2009-12-04 20:57:17 +000032
33namespace chromeos_update_engine {
34
Jay Srinivasan55f50c22013-01-10 19:24:35 -080035class SystemState;
36
adlr@google.com3defe6a2009-12-04 20:57:17 +000037namespace utils {
38
David Zeuthen27a48bc2013-08-06 12:06:29 -070039// Converts a struct timespec representing a number of seconds since
40// the Unix epoch to a base::Time. Sub-microsecond time is rounded
41// down.
42base::Time TimeFromStructTimespec(struct timespec *ts);
43
Alex Vakulenko88b591f2014-08-28 16:48:57 -070044// Converts a vector of strings to a NUL-terminated array of
David Zeuthen27a48bc2013-08-06 12:06:29 -070045// strings. The resulting array should be freed with g_strfreev()
46// when are you done with it.
Alex Deymof329b932014-10-30 01:37:48 -070047gchar** StringVectorToGStrv(const std::vector<std::string> &vec_str);
David Zeuthen27a48bc2013-08-06 12:06:29 -070048
Alex Deymof329b932014-10-30 01:37:48 -070049// Formats |vec_str| as a string of the form ["<elem1>", "<elem2>"].
David Zeuthen27a48bc2013-08-06 12:06:29 -070050// Does no escaping, only use this for presentation in error messages.
Alex Deymof329b932014-10-30 01:37:48 -070051std::string StringVectorToString(const std::vector<std::string> &vec_str);
David Zeuthen27a48bc2013-08-06 12:06:29 -070052
David Zeuthen8f191b22013-08-06 12:27:50 -070053// Calculates the p2p file id from payload hash and size
54std::string CalculateP2PFileId(const std::string& payload_hash,
55 size_t payload_size);
56
J. Richard Barnette63137e52013-10-28 10:57:29 -070057// Parse the firmware version from one line of output from the
58// "mosys" command.
59std::string ParseECVersion(std::string input_line);
60
J. Richard Barnette30842932013-10-28 15:04:23 -070061// Given the name of the block device of a boot partition, return the
62// name of the associated kernel partition (e.g. given "/dev/sda3",
63// return "/dev/sda2").
64const std::string KernelDeviceOfBootDevice(const std::string& boot_device);
65
Andrew de los Reyes970bb282009-12-09 16:34:04 -080066// Writes the data passed to path. The file at path will be overwritten if it
67// exists. Returns true on success, false otherwise.
Alex Vakulenkof68bbbc2015-02-09 12:53:18 -080068bool WriteFile(const char* path, const void* data, int data_len);
Andrew de los Reyes970bb282009-12-09 16:34:04 -080069
Andrew de los Reyes09e56d62010-04-23 13:45:53 -070070// Calls write() or pwrite() repeatedly until all count bytes at buf are
71// written to fd or an error occurs. Returns true on success.
72bool WriteAll(int fd, const void* buf, size_t count);
73bool PWriteAll(int fd, const void* buf, size_t count, off_t offset);
74
Nam T. Nguyenf1d582e2014-12-08 15:07:17 -080075bool WriteAll(FileDescriptorPtr fd, const void* buf, size_t count);
76bool PWriteAll(FileDescriptorPtr fd,
77 const void* buf,
78 size_t count,
79 off_t offset);
80
Andrew de los Reyes09e56d62010-04-23 13:45:53 -070081// Calls pread() repeatedly until count bytes are read, or EOF is reached.
82// Returns number of bytes read in *bytes_read. Returns true on success.
83bool PReadAll(int fd, void* buf, size_t count, off_t offset,
84 ssize_t* out_bytes_read);
Andrew de los Reyesb10320d2010-03-31 16:44:44 -070085
Nam T. Nguyenf1d582e2014-12-08 15:07:17 -080086bool PReadAll(FileDescriptorPtr fd, void* buf, size_t count, off_t offset,
87 ssize_t* out_bytes_read);
88
Gilad Arnold19a45f02012-07-19 12:36:10 -070089// Opens |path| for reading and appends its entire content to the container
90// pointed to by |out_p|. Returns true upon successfully reading all of the
91// file's content, false otherwise, in which case the state of the output
Darin Petkov8e447e02013-04-16 16:23:50 +020092// container is unknown. ReadFileChunk starts reading the file from |offset|; if
93// |size| is not -1, only up to |size| bytes are read in.
Alex Vakulenkof68bbbc2015-02-09 12:53:18 -080094bool ReadFile(const std::string& path, chromeos::Blob* out_p);
Gilad Arnold19a45f02012-07-19 12:36:10 -070095bool ReadFile(const std::string& path, std::string* out_p);
Darin Petkov8e447e02013-04-16 16:23:50 +020096bool ReadFileChunk(const std::string& path, off_t offset, off_t size,
Alex Vakulenkof68bbbc2015-02-09 12:53:18 -080097 chromeos::Blob* out_p);
Gilad Arnold19a45f02012-07-19 12:36:10 -070098
99// Invokes |cmd| in a pipe and appends its stdout to the container pointed to by
100// |out_p|. Returns true upon successfully reading all of the output, false
101// otherwise, in which case the state of the output container is unknown.
Gilad Arnold19a45f02012-07-19 12:36:10 -0700102bool ReadPipe(const std::string& cmd, std::string* out_p);
adlr@google.com3defe6a2009-12-04 20:57:17 +0000103
Gabe Blackb92cd2e2014-09-08 02:47:41 -0700104// Returns the size of the block device at path, or the file descriptor fd. If
105// an error occurs, -1 is returned.
106off_t BlockDevSize(const std::string& path);
107off_t BlockDevSize(int fd);
108
109// Returns the size of the file at path, or the file desciptor fd. If the file
110// is actually a block device, this function will automatically call
111// BlockDevSize. If the file doesn't exist or some error occurrs, -1 is
112// returned.
Andrew de los Reyesf4c7ef12010-04-30 10:37:00 -0700113off_t FileSize(const std::string& path);
Gabe Blackb92cd2e2014-09-08 02:47:41 -0700114off_t FileSize(int fd);
Andrew de los Reyesf4c7ef12010-04-30 10:37:00 -0700115
adlr@google.com3defe6a2009-12-04 20:57:17 +0000116std::string ErrnoNumberAsString(int err);
117
118// Strips duplicate slashes, and optionally removes all trailing slashes.
119// Does not compact /./ or /../.
120std::string NormalizePath(const std::string& path, bool strip_trailing_slash);
121
122// Returns true if the file exists for sure. Returns false if it doesn't exist,
123// or an error occurs.
124bool FileExists(const char* path);
125
Darin Petkov30291ed2010-11-12 10:23:06 -0800126// Returns true if |path| exists and is a symbolic link.
127bool IsSymlink(const char* path);
128
Alex Deymo7dc4c502014-05-20 20:09:58 -0700129// Returns true if |path| exists and is a directory.
130bool IsDir(const char* path);
131
Nam T. Nguyena78b28c2015-03-06 22:30:12 -0800132// Try attaching UBI |volume_num|. If there is any error executing required
133// commands to attach the volume, this function returns false. This function
134// only returns true if "/dev/ubi%d_0" becomes available in |timeout| seconds.
135bool TryAttachingUbiVolume(int volume_num, int timeout);
136
Gilad Arnoldd04f8e22014-01-09 13:13:40 -0800137// If |base_filename_template| is neither absolute (starts with "/") nor
138// explicitly relative to the current working directory (starts with "./" or
139// "../"), then it is prepended the value of TMPDIR, which defaults to /tmp if
140// it isn't set or is empty. It then calls mkstemp(3) with the resulting
141// template. Writes the name of a new temporary file to |filename|. If |fd| is
Alex Vakulenko88b591f2014-08-28 16:48:57 -0700142// non-null, the file descriptor returned by mkstemp is written to it and
143// kept open; otherwise, it is closed. The template must end with "XXXXXX".
144// Returns true on success.
Gilad Arnoldd04f8e22014-01-09 13:13:40 -0800145bool MakeTempFile(const std::string& base_filename_template,
Andrew de los Reyesb10320d2010-03-31 16:44:44 -0700146 std::string* filename,
147 int* fd);
148
Gilad Arnoldd04f8e22014-01-09 13:13:40 -0800149// If |base_filename_template| is neither absolute (starts with "/") nor
150// explicitly relative to the current working directory (starts with "./" or
151// "../"), then it is prepended the value of TMPDIR, which defaults to /tmp if
152// it isn't set or is empty. It then calls mkdtemp() with the resulting
153// template. Writes the name of the new temporary directory to |dirname|.
154// The template must end with "XXXXXX". Returns true on success.
155bool MakeTempDirectory(const std::string& base_dirname_template,
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700156 std::string* dirname);
157
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800158// Returns the disk device name for a partition. For example,
159// GetDiskName("/dev/sda3") returns "/dev/sda". Returns an empty string
160// if the input device is not of the "/dev/xyz#" form.
161std::string GetDiskName(const std::string& partition_name);
Andrew de los Reyesf9714432010-05-04 10:21:23 -0700162
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800163// Returns the partition number, of partition device name. For example,
164// GetPartitionNumber("/dev/sda3") returns 3.
165// Returns 0 on failure
166int GetPartitionNumber(const std::string& partition_name);
167
168// Splits the partition device name into the block device name and partition
169// number. For example, "/dev/sda3" will be split into {"/dev/sda", 3} and
170// "/dev/mmcblk0p2" into {"/dev/mmcblk0", 2}
171// Returns false when malformed device name is passed in.
Alex Vakulenko88b591f2014-08-28 16:48:57 -0700172// If both output parameters are omitted (null), can be used
Alex Vakulenko4f5b1442014-02-21 12:19:44 -0800173// just to test the validity of the device name. Note that the function
174// simply checks if the device name looks like a valid device, no other
175// checks are performed (i.e. it doesn't check if the device actually exists).
176bool SplitPartitionName(const std::string& partition_name,
177 std::string* out_disk_name,
178 int* out_partition_num);
179
180// Builds a partition device name from the block device name and partition
181// number. For example:
182// {"/dev/sda", 1} => "/dev/sda1"
183// {"/dev/mmcblk2", 12} => "/dev/mmcblk2p12"
184// Returns empty string when invalid parameters are passed in
185std::string MakePartitionName(const std::string& disk_name,
186 int partition_num);
Andrew de los Reyesf9714432010-05-04 10:21:23 -0700187
Nam T. Nguyena78b28c2015-03-06 22:30:12 -0800188// Similar to "MakePartitionName" but returns a name that is suitable for
189// mounting. On NAND system we can write to "/dev/ubiX_0", which is what
190// MakePartitionName returns, but we cannot mount that device. To mount, we
191// have to use "/dev/ubiblockX_0" for rootfs. Stateful and OEM partitions are
192// mountable with "/dev/ubiX_0". The input is a partition device such as
193// /dev/sda3. Return empty string on error.
194std::string MakePartitionNameForMount(const std::string& part_name);
195
Darin Petkovf74eb652010-08-04 12:08:38 -0700196// Returns the sysfs block device for a root block device. For
197// example, SysfsBlockDevice("/dev/sda") returns
198// "/sys/block/sda". Returns an empty string if the input device is
199// not of the "/dev/xyz" form.
200std::string SysfsBlockDevice(const std::string& device);
201
202// Returns true if the root |device| (e.g., "/dev/sdb") is known to be
203// removable, false otherwise.
204bool IsRemovableDevice(const std::string& device);
205
adlr@google.com3defe6a2009-12-04 20:57:17 +0000206// Synchronously mount or unmount a filesystem. Return true on success.
Alex Deymo4c82df32014-11-10 22:25:57 -0800207// When mounting, it will attempt to mount the the device as "ext3", "ext2" and
208// "squashfs", with the passed |flags| options.
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700209bool MountFilesystem(const std::string& device, const std::string& mountpoint,
Alex Vakulenkod2779df2014-06-16 13:19:00 -0700210 unsigned long flags); // NOLINT(runtime/int)
Andrew de los Reyes4fe15d02009-12-10 19:01:36 -0800211bool UnmountFilesystem(const std::string& mountpoint);
adlr@google.com3defe6a2009-12-04 20:57:17 +0000212
Alex Deymo192393b2014-11-10 15:58:38 -0800213// Returns the block count and the block byte size of the file system on
Darin Petkovd3f8c892010-10-12 21:38:45 -0700214// |device| (which may be a real device or a path to a filesystem image) or on
215// an opened file descriptor |fd|. The actual file-system size is |block_count|
216// * |block_size| bytes. Returns true on success, false otherwise.
217bool GetFilesystemSize(const std::string& device,
218 int* out_block_count,
219 int* out_block_size);
220bool GetFilesystemSizeFromFD(int fd,
221 int* out_block_count,
222 int* out_block_size);
223
Alex Deymo192393b2014-11-10 15:58:38 -0800224// Determines the block count and block size of the ext3 fs. At least 2048 bytes
225// are required to parse the first superblock. Returns whether the buffer
226// contains a valid ext3 filesystem and the values were parsed.
227bool GetExt3Size(const uint8_t* buffer, size_t buffer_size,
228 int* out_block_count,
229 int* out_block_size);
230
231// Determines the block count and block size of the squashfs v4 fs. At least 96
232// bytes are required to parse the header of the filesystem. Since squashfs
233// doesn't define a physical block size, a value of 4096 is used for the block
234// size, which is the default padding when creating the filesystem.
235// Returns whether the buffer contains a valid squashfs v4 header and the size
236// was parsed. Only little endian squashfs is supported.
237bool GetSquashfs4Size(const uint8_t* buffer, size_t buffer_size,
238 int* out_block_count,
239 int* out_block_size);
240
Alex Deymode942f32015-03-13 11:57:15 -0700241// Returns whether the filesystem is an ext[234] filesystem. In case of failure,
242// such as if the file |device| doesn't exists or can't be read, it returns
243// false.
244bool IsExtFilesystem(const std::string& device);
245
246// Returns whether the filesystem is a squashfs filesystem. In case of failure,
247// such as if the file |device| doesn't exists or can't be read, it returns
248// false.
249bool IsSquashfsFilesystem(const std::string& device);
250
Alex Deymo719bfff2014-07-11 12:12:32 -0700251// Returns the path of the passed |command| on the board. This uses the
252// environment variable SYSROOT to determine the path to the command on the
253// board instead of the path on the running environment.
254std::string GetPathOnBoard(const std::string& command);
Alex Deymo032e7722014-03-25 17:53:56 -0700255
256// Returns a human-readable string with the file format based on magic constants
257// on the header of the file.
258std::string GetFileFormat(const std::string& path);
259
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700260// Returns the string representation of the given UTC time.
261// such as "11/14/2011 14:05:30 GMT".
262std::string ToString(const base::Time utc_time);
263
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700264// Returns true or false depending on the value of b.
265std::string ToString(bool b);
266
Jay Srinivasan19409b72013-04-12 19:23:36 -0700267// Returns a string representation of the given enum.
268std::string ToString(DownloadSource source);
269
Alex Deymo1c656c42013-06-28 11:02:14 -0700270// Returns a string representation of the given enum.
271std::string ToString(PayloadType payload_type);
272
Andrew de los Reyes712b3ac2011-01-07 13:47:52 -0800273// Schedules a Main Loop callback to trigger the crash reporter to perform an
274// upload as if this process had crashed.
275void ScheduleCrashReporterUpload();
276
Darin Petkov5c0a8af2010-08-24 13:39:13 -0700277// Fuzzes an integer |value| randomly in the range:
278// [value - range / 2, value + range - range / 2]
279int FuzzInt(int value, unsigned int range);
280
adlr@google.com3defe6a2009-12-04 20:57:17 +0000281// Log a string in hex to LOG(INFO). Useful for debugging.
Alex Vakulenkof68bbbc2015-02-09 12:53:18 -0800282void HexDumpArray(const uint8_t* const arr, const size_t length);
adlr@google.com3defe6a2009-12-04 20:57:17 +0000283inline void HexDumpString(const std::string& str) {
Alex Vakulenkof68bbbc2015-02-09 12:53:18 -0800284 HexDumpArray(reinterpret_cast<const uint8_t*>(str.data()), str.size());
adlr@google.com3defe6a2009-12-04 20:57:17 +0000285}
Alex Vakulenkof68bbbc2015-02-09 12:53:18 -0800286inline void HexDumpVector(const chromeos::Blob& vect) {
287 HexDumpArray(vect.data(), vect.size());
adlr@google.com3defe6a2009-12-04 20:57:17 +0000288}
289
adlr@google.com3defe6a2009-12-04 20:57:17 +0000290template<typename KeyType, typename ValueType>
291bool MapContainsKey(const std::map<KeyType, ValueType>& m, const KeyType& k) {
292 return m.find(k) != m.end();
293}
Andrew de los Reyes4fe15d02009-12-10 19:01:36 -0800294template<typename KeyType>
295bool SetContainsKey(const std::set<KeyType>& s, const KeyType& k) {
296 return s.find(k) != s.end();
297}
adlr@google.com3defe6a2009-12-04 20:57:17 +0000298
Andrew de los Reyes0ce161b2010-02-22 15:27:01 -0800299template<typename T>
300bool VectorContainsValue(const std::vector<T>& vect, const T& value) {
Darin Petkovc1a8b422010-07-19 11:34:49 -0700301 return std::find(vect.begin(), vect.end(), value) != vect.end();
Andrew de los Reyes0ce161b2010-02-22 15:27:01 -0800302}
303
Andrew de los Reyesb10320d2010-03-31 16:44:44 -0700304template<typename T>
305bool VectorIndexOf(const std::vector<T>& vect, const T& value,
306 typename std::vector<T>::size_type* out_index) {
307 typename std::vector<T>::const_iterator it = std::find(vect.begin(),
308 vect.end(),
309 value);
310 if (it == vect.end()) {
311 return false;
312 } else {
313 *out_index = it - vect.begin();
314 return true;
315 }
316}
317
Andrew de los Reyescc92cd32010-10-05 16:56:14 -0700318template<typename ValueType>
319void ApplyMap(std::vector<ValueType>* collection,
320 const std::map<ValueType, ValueType>& the_map) {
321 for (typename std::vector<ValueType>::iterator it = collection->begin();
322 it != collection->end(); ++it) {
323 typename std::map<ValueType, ValueType>::const_iterator map_it =
324 the_map.find(*it);
325 if (map_it != the_map.end()) {
326 *it = map_it->second;
327 }
328 }
329}
330
Chris Sosa4f8ee272012-11-30 13:01:54 -0800331// Cgroups cpu shares constants. 1024 is the default shares a standard process
332// gets and 2 is the minimum value. We set High as a value that gives the
333// update-engine 2x the cpu share of a standard process.
334enum CpuShares {
335 kCpuSharesHigh = 2048,
336 kCpuSharesNormal = 1024,
337 kCpuSharesLow = 2,
Darin Petkovc6c135c2010-08-11 13:36:18 -0700338};
339
Chris Sosa4f8ee272012-11-30 13:01:54 -0800340// Sets the current process shares to |shares|. Returns true on
Darin Petkovc6c135c2010-08-11 13:36:18 -0700341// success, false otherwise.
Chris Sosa4f8ee272012-11-30 13:01:54 -0800342bool SetCpuShares(CpuShares shares);
Andrew de los Reyesf9185172010-05-03 11:07:05 -0700343
Alex Deymoc4acdf42014-05-28 21:07:10 -0700344// Assumes |data| points to a Closure. Runs it and returns FALSE;
Andrew de los Reyesf3ed8e72011-02-16 10:35:46 -0800345gboolean GlibRunClosure(gpointer data);
346
Alex Deymoc4acdf42014-05-28 21:07:10 -0700347// Destroys the Closure pointed by |data|.
348void GlibDestroyClosure(gpointer data);
349
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700350// Converts seconds into human readable notation including days, hours, minutes
351// and seconds. For example, 185 will yield 3m5s, 4300 will yield 1h11m40s, and
352// 360000 will yield 4d4h0m0s. Zero padding not applied. Seconds are always
353// shown in the result.
354std::string FormatSecs(unsigned secs);
355
356// Converts a TimeDelta into human readable notation including days, hours,
357// minutes, seconds and fractions of a second down to microsecond granularity,
358// as necessary; for example, an output of 5d2h0m15.053s means that the input
359// time was precise to the milliseconds only. Zero padding not applied, except
360// for fractions. Seconds are always shown, but fractions thereof are only shown
David Zeuthen973449e2014-08-18 16:18:23 -0400361// when applicable. If |delta| is negative, the output will have a leading '-'
362// followed by the absolute duration.
Gilad Arnoldd7b513d2012-05-10 14:25:27 -0700363std::string FormatTimeDelta(base::TimeDelta delta);
Gilad Arnold1ebd8132012-03-05 10:19:29 -0800364
Jay Srinivasan2b5a0f02012-12-19 17:25:56 -0800365// This method transforms the given error code to be suitable for UMA and
366// for error classification purposes by removing the higher order bits and
367// aggregating error codes beyond the enum range, etc. This method is
368// idempotent, i.e. if called with a value previously returned by this method,
369// it'll return the same value again.
David Zeuthena99981f2013-04-29 13:42:47 -0700370ErrorCode GetBaseErrorCode(ErrorCode code);
Jay Srinivasan2b5a0f02012-12-19 17:25:56 -0800371
David Zeuthen33bae492014-02-25 16:16:18 -0800372// Transforms a ErrorCode value into a metrics::DownloadErrorCode.
373// This obviously only works for errors related to downloading so if |code|
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700374// is e.g. |ErrorCode::kFilesystemCopierError| then
David Zeuthen33bae492014-02-25 16:16:18 -0800375// |kDownloadErrorCodeInputMalformed| is returned.
376metrics::DownloadErrorCode GetDownloadErrorCode(ErrorCode code);
377
378// Transforms a ErrorCode value into a metrics::AttemptResult.
379//
380// If metrics::AttemptResult::kPayloadDownloadError is returned, you
381// can use utils::GetDownloadError() to get more detail.
382metrics::AttemptResult GetAttemptResult(ErrorCode code);
383
David Zeuthenb281f072014-04-02 10:20:19 -0700384// Calculates the internet connection type given |type| and |tethering|.
385metrics::ConnectionType GetConnectionType(NetworkConnectionType type,
386 NetworkTethering tethering);
387
Jay Srinivasan55f50c22013-01-10 19:24:35 -0800388// Sends the error code to UMA using the metrics interface object in the given
389// system state. It also uses the system state to determine the right UMA
390// bucket for the error code.
David Zeuthena99981f2013-04-29 13:42:47 -0700391void SendErrorCodeToUma(SystemState* system_state, ErrorCode code);
Jay Srinivasan55f50c22013-01-10 19:24:35 -0800392
David Zeuthena99981f2013-04-29 13:42:47 -0700393// Returns a string representation of the ErrorCodes (either the base
Jay Srinivasan55f50c22013-01-10 19:24:35 -0800394// error codes or the bit flags) for logging purposes.
David Zeuthena99981f2013-04-29 13:42:47 -0700395std::string CodeToString(ErrorCode code);
Jay Srinivasan55f50c22013-01-10 19:24:35 -0800396
Gilad Arnold30dedd82013-07-03 06:19:09 -0700397// Creates the powerwash marker file with the appropriate commands in it. Uses
Alex Vakulenko88b591f2014-08-28 16:48:57 -0700398// |file_path| as the path to the marker file if non-null, otherwise uses the
Gilad Arnold30dedd82013-07-03 06:19:09 -0700399// global default. Returns true if successfully created. False otherwise.
400bool CreatePowerwashMarkerFile(const char* file_path);
Jay Srinivasan1c0fe792013-03-28 16:45:25 -0700401
Gilad Arnold30dedd82013-07-03 06:19:09 -0700402// Deletes the marker file used to trigger Powerwash using clobber-state. Uses
Alex Vakulenko88b591f2014-08-28 16:48:57 -0700403// |file_path| as the path to the marker file if non-null, otherwise uses the
Gilad Arnold30dedd82013-07-03 06:19:09 -0700404// global default. Returns true if successfully deleted. False otherwise.
405bool DeletePowerwashMarkerFile(const char* file_path);
Jay Srinivasan1c0fe792013-03-28 16:45:25 -0700406
Chris Sosad317e402013-06-12 13:47:09 -0700407// Assumes you want to install on the "other" device, where the other
408// device is what you get if you swap 1 for 2 or 3 for 4 or vice versa
409// for the number at the end of the boot device. E.g., /dev/sda1 -> /dev/sda2
410// or /dev/sda4 -> /dev/sda3. See
411// http://www.chromium.org/chromium-os/chromiumos-design-docs/disk-format
412bool GetInstallDev(const std::string& boot_dev, std::string* install_dev);
413
David Zeuthene7f89172013-10-31 10:21:04 -0700414// Decodes the data in |base64_encoded| and stores it in a temporary
415// file. Returns false if the given data is empty, not well-formed
416// base64 or if an error occurred. If true is returned, the decoded
417// data is stored in the file returned in |out_path|. The file should
418// be deleted when no longer needed.
419bool DecodeAndStoreBase64String(const std::string& base64_encoded,
420 base::FilePath *out_path);
421
David Zeuthen639aa362014-02-03 16:23:44 -0800422// Converts |time| to an Omaha InstallDate which is defined as "the
423// number of PST8PDT calendar weeks since Jan 1st 2007 0:00 PST, times
424// seven" with PST8PDT defined as "Pacific Time" (e.g. UTC-07:00 if
425// daylight savings is observed and UTC-08:00 otherwise.)
426//
427// If the passed in |time| variable is before Monday January 1st 2007
428// 0:00 PST, False is returned and the value returned in
429// |out_num_days| is undefined. Otherwise the number of PST8PDT
430// calendar weeks since that date times seven is returned in
431// |out_num_days| and the function returns True.
432//
433// (NOTE: This function does not currently take daylight savings time
434// into account so the result may up to one hour off. This is because
435// the glibc date and timezone routines depend on the TZ environment
436// variable and changing environment variables is not thread-safe. An
437// alternative, thread-safe API to use would be GDateTime/GTimeZone
438// available in GLib 2.26 or later.)
439bool ConvertToOmahaInstallDate(base::Time time, int *out_num_days);
440
David Zeuthen33bae492014-02-25 16:16:18 -0800441// This function returns the duration on the wallclock since the last
442// time it was called for the same |state_variable_key| value.
443//
444// If the function returns |true|, the duration (always non-negative)
445// is returned in |out_duration|. If the function returns |false|
446// something went wrong or there was no previous measurement.
447bool WallclockDurationHelper(SystemState* system_state,
448 const std::string& state_variable_key,
449 base::TimeDelta* out_duration);
450
451// This function returns the duration on the monotonic clock since the
452// last time it was called for the same |storage| pointer.
453//
454// You should pass a pointer to a 64-bit integer in |storage| which
455// should be initialized to 0.
456//
457// If the function returns |true|, the duration (always non-negative)
458// is returned in |out_duration|. If the function returns |false|
459// something went wrong or there was no previous measurement.
460bool MonotonicDurationHelper(SystemState* system_state,
461 int64_t* storage,
462 base::TimeDelta* out_duration);
463
Allie Wood78750a42015-02-11 15:42:11 -0800464// This function looks for a configuration file at |path|. If it finds that
465// file, it will try get the PAYLOAD_MINOR_VERSION value from it and set
466// |minor_version| to that value.
467//
468// The function will return |true| if it succeeds at finding the file and
469// value and setting it, and |false| otherwise.
470bool GetMinorVersion(base::FilePath path, uint32_t* minor_version);
471
Allie Wood56873452015-03-27 17:48:40 -0700472// This function reads the specified data in |extents| into |out_data|. The
473// extents are read from the file at |path|. |out_data_size| is the size of
474// |out_data|. Returns false if the number of bytes to read given in
475// |extents| does not equal |out_data_size|.
476bool ReadExtents(const std::string& path, std::vector<Extent>* extents,
477 chromeos::Blob* out_data, ssize_t out_data_size,
478 size_t block_size);
479
adlr@google.com3defe6a2009-12-04 20:57:17 +0000480} // namespace utils
481
Jay Srinivasan08fce042012-06-07 16:31:01 -0700482
adlr@google.com3defe6a2009-12-04 20:57:17 +0000483// Utility class to close a file descriptor
484class ScopedFdCloser {
485 public:
486 explicit ScopedFdCloser(int* fd) : fd_(fd), should_close_(true) {}
adlr@google.com3defe6a2009-12-04 20:57:17 +0000487 ~ScopedFdCloser() {
Darin Petkov6f03a3b2010-11-10 14:27:14 -0800488 if (should_close_ && fd_ && (*fd_ >= 0)) {
Gilad Arnoldf9609112012-02-29 11:38:47 -0800489 if (!close(*fd_))
490 *fd_ = -1;
adlr@google.com3defe6a2009-12-04 20:57:17 +0000491 }
492 }
Darin Petkov6f03a3b2010-11-10 14:27:14 -0800493 void set_should_close(bool should_close) { should_close_ = should_close; }
adlr@google.com3defe6a2009-12-04 20:57:17 +0000494 private:
495 int* fd_;
496 bool should_close_;
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700497 DISALLOW_COPY_AND_ASSIGN(ScopedFdCloser);
498};
499
Gilad Arnoldf9609112012-02-29 11:38:47 -0800500// An EINTR-immune file descriptor closer.
501class ScopedEintrSafeFdCloser {
502 public:
503 explicit ScopedEintrSafeFdCloser(int* fd) : fd_(fd), should_close_(true) {}
504 ~ScopedEintrSafeFdCloser() {
505 if (should_close_ && fd_ && (*fd_ >= 0)) {
Mike Frysingerbcee2ca2014-05-14 16:28:23 -0400506 if (!IGNORE_EINTR(close(*fd_)))
Gilad Arnoldf9609112012-02-29 11:38:47 -0800507 *fd_ = -1;
508 }
509 }
510 void set_should_close(bool should_close) { should_close_ = should_close; }
511 private:
512 int* fd_;
513 bool should_close_;
514 DISALLOW_COPY_AND_ASSIGN(ScopedEintrSafeFdCloser);
515};
516
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700517// Utility class to delete a file when it goes out of scope.
518class ScopedPathUnlinker {
519 public:
Darin Petkov52dcaeb2011-01-14 15:33:06 -0800520 explicit ScopedPathUnlinker(const std::string& path)
521 : path_(path),
522 should_remove_(true) {}
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700523 ~ScopedPathUnlinker() {
Darin Petkov52dcaeb2011-01-14 15:33:06 -0800524 if (should_remove_ && unlink(path_.c_str()) < 0) {
Alex Deymo18f48aa2015-04-09 10:06:31 -0700525 PLOG(ERROR) << "Unable to unlink path " << path_;
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700526 }
527 }
Darin Petkov52dcaeb2011-01-14 15:33:06 -0800528 void set_should_remove(bool should_remove) { should_remove_ = should_remove; }
529
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700530 private:
531 const std::string path_;
Darin Petkov52dcaeb2011-01-14 15:33:06 -0800532 bool should_remove_;
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700533 DISALLOW_COPY_AND_ASSIGN(ScopedPathUnlinker);
534};
535
536// Utility class to delete an empty directory when it goes out of scope.
537class ScopedDirRemover {
538 public:
Darin Petkov6f03a3b2010-11-10 14:27:14 -0800539 explicit ScopedDirRemover(const std::string& path)
540 : path_(path),
541 should_remove_(true) {}
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700542 ~ScopedDirRemover() {
Darin Petkov6f03a3b2010-11-10 14:27:14 -0800543 if (should_remove_ && (rmdir(path_.c_str()) < 0)) {
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700544 PLOG(ERROR) << "Unable to remove dir " << path_;
Darin Petkov6f03a3b2010-11-10 14:27:14 -0800545 }
546 }
547 void set_should_remove(bool should_remove) { should_remove_ = should_remove; }
548
549 protected:
550 const std::string path_;
551
552 private:
553 bool should_remove_;
554 DISALLOW_COPY_AND_ASSIGN(ScopedDirRemover);
555};
556
557// Utility class to unmount a filesystem mounted on a temporary directory and
558// delete the temporary directory when it goes out of scope.
559class ScopedTempUnmounter : public ScopedDirRemover {
560 public:
561 explicit ScopedTempUnmounter(const std::string& path) :
562 ScopedDirRemover(path) {}
563 ~ScopedTempUnmounter() {
564 utils::UnmountFilesystem(path_);
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700565 }
566 private:
Darin Petkov6f03a3b2010-11-10 14:27:14 -0800567 DISALLOW_COPY_AND_ASSIGN(ScopedTempUnmounter);
adlr@google.com3defe6a2009-12-04 20:57:17 +0000568};
569
570// A little object to call ActionComplete on the ActionProcessor when
571// it's destructed.
572class ScopedActionCompleter {
573 public:
574 explicit ScopedActionCompleter(ActionProcessor* processor,
575 AbstractAction* action)
576 : processor_(processor),
577 action_(action),
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700578 code_(ErrorCode::kError),
adlr@google.com3defe6a2009-12-04 20:57:17 +0000579 should_complete_(true) {}
580 ~ScopedActionCompleter() {
581 if (should_complete_)
Darin Petkovc1a8b422010-07-19 11:34:49 -0700582 processor_->ActionComplete(action_, code_);
adlr@google.com3defe6a2009-12-04 20:57:17 +0000583 }
David Zeuthena99981f2013-04-29 13:42:47 -0700584 void set_code(ErrorCode code) { code_ = code; }
adlr@google.com3defe6a2009-12-04 20:57:17 +0000585 void set_should_complete(bool should_complete) {
586 should_complete_ = should_complete;
587 }
David Zeuthen33bae492014-02-25 16:16:18 -0800588 ErrorCode get_code() const { return code_; }
Darin Petkovc1a8b422010-07-19 11:34:49 -0700589
adlr@google.com3defe6a2009-12-04 20:57:17 +0000590 private:
591 ActionProcessor* processor_;
592 AbstractAction* action_;
David Zeuthena99981f2013-04-29 13:42:47 -0700593 ErrorCode code_;
adlr@google.com3defe6a2009-12-04 20:57:17 +0000594 bool should_complete_;
595 DISALLOW_COPY_AND_ASSIGN(ScopedActionCompleter);
596};
597
David Zeuthen27a48bc2013-08-06 12:06:29 -0700598// A base::FreeDeleter that frees memory using g_free(). Useful when
Ben Chan02f7c1d2014-10-18 15:18:02 -0700599// integrating with GLib since it can be used with std::unique_ptr to
David Zeuthen27a48bc2013-08-06 12:06:29 -0700600// automatically free memory when going out of scope.
601struct GLibFreeDeleter : public base::FreeDeleter {
602 inline void operator()(void *ptr) const {
603 g_free(reinterpret_cast<gpointer>(ptr));
604 }
605};
606
607// A base::FreeDeleter that frees memory using g_strfreev(). Useful
Ben Chan02f7c1d2014-10-18 15:18:02 -0700608// when integrating with GLib since it can be used with std::unique_ptr to
David Zeuthen27a48bc2013-08-06 12:06:29 -0700609// automatically free memory when going out of scope.
610struct GLibStrvFreeDeleter : public base::FreeDeleter {
611 inline void operator()(void *ptr) const {
612 g_strfreev(reinterpret_cast<gchar**>(ptr));
613 }
614};
615
adlr@google.com3defe6a2009-12-04 20:57:17 +0000616} // namespace chromeos_update_engine
617
618#define TEST_AND_RETURN_FALSE_ERRNO(_x) \
619 do { \
Nam T. Nguyen2b67a592014-12-03 14:56:00 -0800620 bool _success = static_cast<bool>(_x); \
adlr@google.com3defe6a2009-12-04 20:57:17 +0000621 if (!_success) { \
622 std::string _msg = \
623 chromeos_update_engine::utils::ErrnoNumberAsString(errno); \
624 LOG(ERROR) << #_x " failed: " << _msg; \
625 return false; \
626 } \
627 } while (0)
628
629#define TEST_AND_RETURN_FALSE(_x) \
630 do { \
Nam T. Nguyen2b67a592014-12-03 14:56:00 -0800631 bool _success = static_cast<bool>(_x); \
adlr@google.com3defe6a2009-12-04 20:57:17 +0000632 if (!_success) { \
633 LOG(ERROR) << #_x " failed."; \
634 return false; \
635 } \
636 } while (0)
637
638#define TEST_AND_RETURN_ERRNO(_x) \
639 do { \
Nam T. Nguyen2b67a592014-12-03 14:56:00 -0800640 bool _success = static_cast<bool>(_x); \
adlr@google.com3defe6a2009-12-04 20:57:17 +0000641 if (!_success) { \
642 std::string _msg = \
643 chromeos_update_engine::utils::ErrnoNumberAsString(errno); \
644 LOG(ERROR) << #_x " failed: " << _msg; \
645 return; \
646 } \
647 } while (0)
648
649#define TEST_AND_RETURN(_x) \
650 do { \
Nam T. Nguyen2b67a592014-12-03 14:56:00 -0800651 bool _success = static_cast<bool>(_x); \
adlr@google.com3defe6a2009-12-04 20:57:17 +0000652 if (!_success) { \
653 LOG(ERROR) << #_x " failed."; \
654 return; \
655 } \
656 } while (0)
657
Thieu Le5c7d9752010-12-15 16:09:28 -0800658#define TEST_AND_RETURN_FALSE_ERRCODE(_x) \
659 do { \
660 errcode_t _error = (_x); \
661 if (_error) { \
662 errno = _error; \
663 LOG(ERROR) << #_x " failed: " << _error; \
664 return false; \
665 } \
666 } while (0)
667
Gilad Arnoldcf175a02014-07-10 16:48:47 -0700668#endif // UPDATE_ENGINE_UTILS_H_