| 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> | 
| Tom Cherry | e7656b7 | 2017-05-01 17:10:09 -0700 | [diff] [blame] | 29 | #include <selinux/label.h> | 
| James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 30 |  | 
| Andreas Gampe | a016c42 | 2014-11-24 19:52:41 -0800 | [diff] [blame] | 31 | #define COLDBOOT_DONE "/dev/.coldboot_done" | 
| Colin Cross | f83d0b9 | 2010-04-21 12:04:20 -0700 | [diff] [blame] | 32 |  | 
| Bowgo Tsai | d262017 | 2017-04-17 22:17:09 +0800 | [diff] [blame] | 33 | const std::string kAndroidDtDir("/proc/device-tree/firmware/android/"); | 
|  | 34 |  | 
| James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 35 | using android::base::boot_clock; | 
| Elliott Hughes | 9605a94 | 2016-11-10 17:43:47 -0800 | [diff] [blame] | 36 | using namespace std::chrono_literals; | 
|  | 37 |  | 
| Mark Salyzyn | b066fcc | 2017-05-05 14:44:35 -0700 | [diff] [blame] | 38 | int CreateSocket(const char* name, int type, bool passcred, mode_t perm, uid_t uid, gid_t gid, | 
|  | 39 | const char* socketcon, selabel_handle* sehandle); | 
| Elliott Hughes | f682b47 | 2015-02-06 12:19:48 -0800 | [diff] [blame] | 40 |  | 
| Tom Cherry | 2cbbe9f | 2017-05-04 18:17:33 -0700 | [diff] [blame] | 41 | bool ReadFile(const std::string& path, std::string* content, std::string* err); | 
|  | 42 | bool WriteFile(const std::string& path, const std::string& content, std::string* err); | 
| Elliott Hughes | f682b47 | 2015-02-06 12:19:48 -0800 | [diff] [blame] | 43 |  | 
| Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 44 | class Timer { | 
| James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 45 | public: | 
|  | 46 | Timer() : start_(boot_clock::now()) {} | 
| Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 47 |  | 
| James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 48 | double duration_s() const { | 
|  | 49 | typedef std::chrono::duration<double> double_duration; | 
|  | 50 | return std::chrono::duration_cast<double_duration>(boot_clock::now() - start_).count(); | 
|  | 51 | } | 
| Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 52 |  | 
| James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 53 | int64_t duration_ms() const { | 
|  | 54 | return std::chrono::duration_cast<std::chrono::milliseconds>(boot_clock::now() - start_) | 
|  | 55 | .count(); | 
|  | 56 | } | 
| Elliott Hughes | 331cf2f | 2016-11-29 19:20:58 +0000 | [diff] [blame] | 57 |  | 
| James Hawkins | e78ea77 | 2017-03-24 11:43:02 -0700 | [diff] [blame] | 58 | private: | 
|  | 59 | android::base::boot_clock::time_point start_; | 
| Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 60 | }; | 
|  | 61 |  | 
| Elliott Hughes | 331cf2f | 2016-11-29 19:20:58 +0000 | [diff] [blame] | 62 | std::ostream& operator<<(std::ostream& os, const Timer& t); | 
|  | 63 |  | 
| Tom Cherry | 517e1f1 | 2017-05-04 17:40:33 -0700 | [diff] [blame] | 64 | bool DecodeUid(const std::string& name, uid_t* uid, std::string* err); | 
| Colin Cross | 3899e9f | 2010-04-13 20:35:46 -0700 | [diff] [blame] | 65 |  | 
| Tom Cherry | e7656b7 | 2017-05-01 17:10:09 -0700 | [diff] [blame] | 66 | int mkdir_recursive(const std::string& pathname, mode_t mode, selabel_handle* sehandle); | 
| Elliott Hughes | 9605a94 | 2016-11-10 17:43:47 -0800 | [diff] [blame] | 67 | int wait_for_file(const char *filename, std::chrono::nanoseconds timeout); | 
| Elliott Hughes | e5ce30f | 2015-05-06 19:19:24 -0700 | [diff] [blame] | 68 | void import_kernel_cmdline(bool in_qemu, | 
| Chih-Hung Hsieh | 8f7b9e3 | 2016-07-27 16:25:51 -0700 | [diff] [blame] | 69 | const std::function<void(const std::string&, const std::string&, bool)>&); | 
| Tom Cherry | e7656b7 | 2017-05-01 17:10:09 -0700 | [diff] [blame] | 70 | int make_dir(const char* path, mode_t mode, selabel_handle* sehandle); | 
| 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 |