Colin Cross | 3899e9f | 2010-04-13 20:35:46 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2010 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | #ifndef _INIT_UTIL_H_ |
| 18 | #define _INIT_UTIL_H_ |
| 19 | |
Colin Cross | f83d0b9 | 2010-04-21 12:04:20 -0700 | [diff] [blame] | 20 | #include <sys/stat.h> |
| 21 | #include <sys/types.h> |
| 22 | |
Elliott Hughes | 9605a94 | 2016-11-10 17:43:47 -0800 | [diff] [blame] | 23 | #include <chrono> |
Nick Kralevich | f667a32 | 2015-04-25 17:42:52 -0700 | [diff] [blame] | 24 | #include <functional> |
Elliott Hughes | 331cf2f | 2016-11-29 19:20:58 +0000 | [diff] [blame] | 25 | #include <ostream> |
| 26 | #include <string> |
Elliott Hughes | f682b47 | 2015-02-06 12:19:48 -0800 | [diff] [blame] | 27 | |
James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 28 | #include <android-base/chrono_utils.h> |
| 29 | |
Andreas Gampe | a016c42 | 2014-11-24 19:52:41 -0800 | [diff] [blame] | 30 | #define COLDBOOT_DONE "/dev/.coldboot_done" |
Colin Cross | f83d0b9 | 2010-04-21 12:04:20 -0700 | [diff] [blame] | 31 | |
Bowgo Tsai | d262017 | 2017-04-17 22:17:09 +0800 | [diff] [blame^] | 32 | const std::string kAndroidDtDir("/proc/device-tree/firmware/android/"); |
| 33 | |
James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 34 | using android::base::boot_clock; |
Elliott Hughes | 9605a94 | 2016-11-10 17:43:47 -0800 | [diff] [blame] | 35 | using namespace std::chrono_literals; |
| 36 | |
Colin Cross | 3899e9f | 2010-04-13 20:35:46 -0700 | [diff] [blame] | 37 | int create_socket(const char *name, int type, mode_t perm, |
Stephen Smalley | 8348d27 | 2013-05-13 12:37:04 -0400 | [diff] [blame] | 38 | uid_t uid, gid_t gid, const char *socketcon); |
Elliott Hughes | f682b47 | 2015-02-06 12:19:48 -0800 | [diff] [blame] | 39 | |
Tom Cherry | 53089aa | 2017-03-31 15:47:33 -0700 | [diff] [blame] | 40 | bool read_file(const std::string& path, std::string* content); |
| 41 | bool write_file(const std::string& path, const std::string& content); |
Elliott Hughes | f682b47 | 2015-02-06 12:19:48 -0800 | [diff] [blame] | 42 | |
Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 43 | class Timer { |
James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 44 | public: |
| 45 | Timer() : start_(boot_clock::now()) {} |
Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 46 | |
James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 47 | double duration_s() const { |
| 48 | typedef std::chrono::duration<double> double_duration; |
| 49 | return std::chrono::duration_cast<double_duration>(boot_clock::now() - start_).count(); |
| 50 | } |
Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 51 | |
James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 52 | int64_t duration_ms() const { |
| 53 | return std::chrono::duration_cast<std::chrono::milliseconds>(boot_clock::now() - start_) |
| 54 | .count(); |
| 55 | } |
Elliott Hughes | 331cf2f | 2016-11-29 19:20:58 +0000 | [diff] [blame] | 56 | |
James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 57 | private: |
| 58 | android::base::boot_clock::time_point start_; |
Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 59 | }; |
| 60 | |
Elliott Hughes | 331cf2f | 2016-11-29 19:20:58 +0000 | [diff] [blame] | 61 | std::ostream& operator<<(std::ostream& os, const Timer& t); |
| 62 | |
Colin Cross | 3899e9f | 2010-04-13 20:35:46 -0700 | [diff] [blame] | 63 | unsigned int decode_uid(const char *s); |
| 64 | |
Tom Cherry | 060b74b | 2017-04-12 14:27:51 -0700 | [diff] [blame] | 65 | int mkdir_recursive(const std::string& pathname, mode_t mode); |
Elliott Hughes | 9605a94 | 2016-11-10 17:43:47 -0800 | [diff] [blame] | 66 | int wait_for_file(const char *filename, std::chrono::nanoseconds timeout); |
Elliott Hughes | e5ce30f | 2015-05-06 19:19:24 -0700 | [diff] [blame] | 67 | void import_kernel_cmdline(bool in_qemu, |
Chih-Hung Hsieh | 8f7b9e3 | 2016-07-27 16:25:51 -0700 | [diff] [blame] | 68 | const std::function<void(const std::string&, const std::string&, bool)>&); |
Stephen Smalley | e096e36 | 2012-06-11 13:37:39 -0400 | [diff] [blame] | 69 | int make_dir(const char *path, mode_t mode); |
Paul Lawrence | a8d8434 | 2016-11-14 15:40:18 -0800 | [diff] [blame] | 70 | int restorecon(const char *pathname, int flags = 0); |
Andres Morales | db5f5d4 | 2015-05-08 08:30:33 -0700 | [diff] [blame] | 71 | std::string bytes_to_hex(const uint8_t *bytes, size_t bytes_len); |
Lee Campbell | f13b1b3 | 2015-07-24 16:57:14 -0700 | [diff] [blame] | 72 | bool is_dir(const char* pathname); |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 73 | bool expand_props(const std::string& src, std::string* dst); |
Elliott Hughes | 331cf2f | 2016-11-29 19:20:58 +0000 | [diff] [blame] | 74 | |
Elliott Hughes | 331cf2f | 2016-11-29 19:20:58 +0000 | [diff] [blame] | 75 | void panic() __attribute__((__noreturn__)); |
| 76 | |
Bowgo Tsai | d262017 | 2017-04-17 22:17:09 +0800 | [diff] [blame^] | 77 | // Reads or compares the content of device tree file under kAndroidDtDir directory. |
| 78 | bool read_android_dt_file(const std::string& sub_path, std::string* dt_content); |
| 79 | bool is_android_dt_value_expected(const std::string& sub_path, const std::string& expected_content); |
| 80 | |
Colin Cross | 3899e9f | 2010-04-13 20:35:46 -0700 | [diff] [blame] | 81 | #endif |