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> |
Elliott Hughes | f682b47 | 2015-02-06 12:19:48 -0800 | [diff] [blame] | 24 | #include <string> |
Nick Kralevich | f667a32 | 2015-04-25 17:42:52 -0700 | [diff] [blame] | 25 | #include <functional> |
Elliott Hughes | f682b47 | 2015-02-06 12:19:48 -0800 | [diff] [blame] | 26 | |
Andreas Gampe | a016c42 | 2014-11-24 19:52:41 -0800 | [diff] [blame] | 27 | #define COLDBOOT_DONE "/dev/.coldboot_done" |
Colin Cross | f83d0b9 | 2010-04-21 12:04:20 -0700 | [diff] [blame] | 28 | |
Elliott Hughes | 9605a94 | 2016-11-10 17:43:47 -0800 | [diff] [blame] | 29 | using namespace std::chrono_literals; |
| 30 | |
Colin Cross | 3899e9f | 2010-04-13 20:35:46 -0700 | [diff] [blame] | 31 | int create_socket(const char *name, int type, mode_t perm, |
Stephen Smalley | 8348d27 | 2013-05-13 12:37:04 -0400 | [diff] [blame] | 32 | uid_t uid, gid_t gid, const char *socketcon); |
Mark Salyzyn | 62767fe | 2016-10-27 07:45:34 -0700 | [diff] [blame] | 33 | int create_file(const char *path, int mode, mode_t perm, |
| 34 | uid_t uid, gid_t gid, const char *filecon); |
Elliott Hughes | f682b47 | 2015-02-06 12:19:48 -0800 | [diff] [blame] | 35 | |
| 36 | bool read_file(const char* path, std::string* content); |
| 37 | int write_file(const char* path, const char* content); |
| 38 | |
Elliott Hughes | 9605a94 | 2016-11-10 17:43:47 -0800 | [diff] [blame] | 39 | // A std::chrono clock based on CLOCK_BOOTTIME. |
| 40 | class boot_clock { |
| 41 | public: |
| 42 | typedef std::chrono::nanoseconds duration; |
| 43 | typedef std::chrono::time_point<boot_clock, duration> time_point; |
| 44 | static constexpr bool is_steady = true; |
| 45 | |
| 46 | static time_point now(); |
| 47 | }; |
Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 48 | |
| 49 | class Timer { |
| 50 | public: |
Elliott Hughes | 9605a94 | 2016-11-10 17:43:47 -0800 | [diff] [blame] | 51 | Timer() : start_(boot_clock::now()) { |
Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 52 | } |
| 53 | |
| 54 | double duration() { |
Elliott Hughes | 9605a94 | 2016-11-10 17:43:47 -0800 | [diff] [blame] | 55 | typedef std::chrono::duration<double> double_duration; |
| 56 | return std::chrono::duration_cast<double_duration>(boot_clock::now() - start_).count(); |
Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 57 | } |
| 58 | |
| 59 | private: |
Elliott Hughes | 9605a94 | 2016-11-10 17:43:47 -0800 | [diff] [blame] | 60 | boot_clock::time_point start_; |
Elliott Hughes | da40c00 | 2015-03-27 23:20:44 -0700 | [diff] [blame] | 61 | }; |
| 62 | |
Colin Cross | 3899e9f | 2010-04-13 20:35:46 -0700 | [diff] [blame] | 63 | unsigned int decode_uid(const char *s); |
| 64 | |
Colin Cross | b0ab94b | 2010-04-08 16:16:20 -0700 | [diff] [blame] | 65 | int mkdir_recursive(const char *pathname, mode_t mode); |
| 66 | void sanitize(char *p); |
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)>&); |
Stephen Smalley | e096e36 | 2012-06-11 13:37:39 -0400 | [diff] [blame] | 70 | int make_dir(const char *path, mode_t mode); |
Paul Lawrence | a8d8434 | 2016-11-14 15:40:18 -0800 | [diff] [blame] | 71 | int restorecon(const char *pathname, int flags = 0); |
Andres Morales | db5f5d4 | 2015-05-08 08:30:33 -0700 | [diff] [blame] | 72 | 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] | 73 | bool is_dir(const char* pathname); |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 74 | bool expand_props(const std::string& src, std::string* dst); |
Colin Cross | 3899e9f | 2010-04-13 20:35:46 -0700 | [diff] [blame] | 75 | #endif |