• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (C) 2008 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 #include "util.h"
18 
19 #include <ctype.h>
20 #include <errno.h>
21 #include <fcntl.h>
22 #include <pwd.h>
23 #include <signal.h>
24 #include <stdarg.h>
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <sys/socket.h>
29 #include <sys/un.h>
30 #include <time.h>
31 #include <unistd.h>
32 
33 #include <thread>
34 
35 #include <android-base/file.h>
36 #include <android-base/logging.h>
37 #include <android-base/properties.h>
38 #include <android-base/scopeguard.h>
39 #include <android-base/strings.h>
40 #include <android-base/unique_fd.h>
41 #include <cutils/sockets.h>
42 #include <selinux/android.h>
43 
44 #ifdef INIT_FULL_SOURCES
45 #include <android/api-level.h>
46 #include <sys/system_properties.h>
47 
48 #include "reboot_utils.h"
49 #include "selabel.h"
50 #include "selinux.h"
51 #else
52 #include "host_init_stubs.h"
53 #endif
54 
55 using android::base::boot_clock;
56 using android::base::StartsWith;
57 using namespace std::literals::string_literals;
58 
59 namespace android {
60 namespace init {
61 
62 const std::string kDefaultAndroidDtDir("/proc/device-tree/firmware/android/");
63 
64 void (*trigger_shutdown)(const std::string& command) = nullptr;
65 
66 // DecodeUid() - decodes and returns the given string, which can be either the
67 // numeric or name representation, into the integer uid or gid.
DecodeUid(const std::string & name)68 Result<uid_t> DecodeUid(const std::string& name) {
69     if (isalpha(name[0])) {
70         passwd* pwd = getpwnam(name.c_str());
71         if (!pwd) return ErrnoError() << "getpwnam failed";
72 
73         return pwd->pw_uid;
74     }
75 
76     errno = 0;
77     uid_t result = static_cast<uid_t>(strtoul(name.c_str(), 0, 0));
78     if (errno) return ErrnoError() << "strtoul failed";
79 
80     return result;
81 }
82 
83 /*
84  * CreateSocket - creates a Unix domain socket in ANDROID_SOCKET_DIR
85  * ("/dev/socket") as dictated in init.rc. This socket is inherited by the
86  * daemon. We communicate the file descriptor's value via the environment
87  * variable ANDROID_SOCKET_ENV_PREFIX<name> ("ANDROID_SOCKET_foo").
88  */
CreateSocket(const std::string & name,int type,bool passcred,bool should_listen,mode_t perm,uid_t uid,gid_t gid,const std::string & socketcon)89 Result<int> CreateSocket(const std::string& name, int type, bool passcred, bool should_listen,
90                          mode_t perm, uid_t uid, gid_t gid, const std::string& socketcon) {
91     if (!socketcon.empty()) {
92         if (setsockcreatecon(socketcon.c_str()) == -1) {
93             return ErrnoError() << "setsockcreatecon(\"" << socketcon << "\") failed";
94         }
95     }
96 
97     android::base::unique_fd fd(socket(PF_UNIX, type, 0));
98     if (fd < 0) {
99         return ErrnoError() << "Failed to open socket '" << name << "'";
100     }
101 
102     if (!socketcon.empty()) setsockcreatecon(nullptr);
103 
104     struct sockaddr_un addr;
105     memset(&addr, 0 , sizeof(addr));
106     addr.sun_family = AF_UNIX;
107     snprintf(addr.sun_path, sizeof(addr.sun_path), ANDROID_SOCKET_DIR "/%s", name.c_str());
108 
109     if ((unlink(addr.sun_path) != 0) && (errno != ENOENT)) {
110         return ErrnoError() << "Failed to unlink old socket '" << name << "'";
111     }
112 
113     std::string secontext;
114     if (SelabelLookupFileContext(addr.sun_path, S_IFSOCK, &secontext) && !secontext.empty()) {
115         setfscreatecon(secontext.c_str());
116     }
117 
118     if (passcred) {
119         int on = 1;
120         if (setsockopt(fd, SOL_SOCKET, SO_PASSCRED, &on, sizeof(on))) {
121             return ErrnoError() << "Failed to set SO_PASSCRED '" << name << "'";
122         }
123     }
124 
125     int ret = bind(fd, (struct sockaddr *) &addr, sizeof (addr));
126     int savederrno = errno;
127 
128     if (!secontext.empty()) {
129         setfscreatecon(nullptr);
130     }
131 
132     auto guard = android::base::make_scope_guard([&addr] { unlink(addr.sun_path); });
133 
134     if (ret) {
135         errno = savederrno;
136         return ErrnoError() << "Failed to bind socket '" << name << "'";
137     }
138 
139     if (lchown(addr.sun_path, uid, gid)) {
140         return ErrnoError() << "Failed to lchown socket '" << addr.sun_path << "'";
141     }
142     if (fchmodat(AT_FDCWD, addr.sun_path, perm, AT_SYMLINK_NOFOLLOW)) {
143         return ErrnoError() << "Failed to fchmodat socket '" << addr.sun_path << "'";
144     }
145     if (should_listen && listen(fd, /* use OS maximum */ 1 << 30)) {
146         return ErrnoError() << "Failed to listen on socket '" << addr.sun_path << "'";
147     }
148 
149     LOG(INFO) << "Created socket '" << addr.sun_path << "'"
150               << ", mode " << std::oct << perm << std::dec
151               << ", user " << uid
152               << ", group " << gid;
153 
154     guard.Disable();
155     return fd.release();
156 }
157 
ReadFile(const std::string & path)158 Result<std::string> ReadFile(const std::string& path) {
159     android::base::unique_fd fd(
160         TEMP_FAILURE_RETRY(open(path.c_str(), O_RDONLY | O_NOFOLLOW | O_CLOEXEC)));
161     if (fd == -1) {
162         return ErrnoError() << "open() failed";
163     }
164 
165     // For security reasons, disallow world-writable
166     // or group-writable files.
167     struct stat sb;
168     if (fstat(fd, &sb) == -1) {
169         return ErrnoError() << "fstat failed()";
170     }
171     if ((sb.st_mode & (S_IWGRP | S_IWOTH)) != 0) {
172         return Error() << "Skipping insecure file";
173     }
174 
175     std::string content;
176     if (!android::base::ReadFdToString(fd, &content)) {
177         return ErrnoError() << "Unable to read file contents";
178     }
179     return content;
180 }
181 
OpenFile(const std::string & path,int flags,mode_t mode)182 static int OpenFile(const std::string& path, int flags, mode_t mode) {
183     std::string secontext;
184     if (SelabelLookupFileContext(path, mode, &secontext) && !secontext.empty()) {
185         setfscreatecon(secontext.c_str());
186     }
187 
188     int rc = open(path.c_str(), flags, mode);
189 
190     if (!secontext.empty()) {
191         int save_errno = errno;
192         setfscreatecon(nullptr);
193         errno = save_errno;
194     }
195 
196     return rc;
197 }
198 
WriteFile(const std::string & path,const std::string & content)199 Result<void> WriteFile(const std::string& path, const std::string& content) {
200     android::base::unique_fd fd(TEMP_FAILURE_RETRY(
201         OpenFile(path, O_WRONLY | O_CREAT | O_NOFOLLOW | O_TRUNC | O_CLOEXEC, 0600)));
202     if (fd == -1) {
203         return ErrnoError() << "open() failed";
204     }
205     if (!android::base::WriteStringToFd(content, fd)) {
206         return ErrnoError() << "Unable to write file contents";
207     }
208     return {};
209 }
210 
mkdir_recursive(const std::string & path,mode_t mode)211 bool mkdir_recursive(const std::string& path, mode_t mode) {
212     std::string::size_type slash = 0;
213     while ((slash = path.find('/', slash + 1)) != std::string::npos) {
214         auto directory = path.substr(0, slash);
215         struct stat info;
216         if (stat(directory.c_str(), &info) != 0) {
217             auto ret = make_dir(directory, mode);
218             if (!ret && errno != EEXIST) return false;
219         }
220     }
221     auto ret = make_dir(path, mode);
222     if (!ret && errno != EEXIST) return false;
223     return true;
224 }
225 
wait_for_file(const char * filename,std::chrono::nanoseconds timeout)226 int wait_for_file(const char* filename, std::chrono::nanoseconds timeout) {
227     android::base::Timer t;
228     while (t.duration() < timeout) {
229         struct stat sb;
230         if (stat(filename, &sb) != -1) {
231             LOG(INFO) << "wait for '" << filename << "' took " << t;
232             return 0;
233         }
234         std::this_thread::sleep_for(10ms);
235     }
236     LOG(WARNING) << "wait for '" << filename << "' timed out and took " << t;
237     return -1;
238 }
239 
ImportKernelCmdline(const std::function<void (const std::string &,const std::string &)> & fn)240 void ImportKernelCmdline(const std::function<void(const std::string&, const std::string&)>& fn) {
241     std::string cmdline;
242     android::base::ReadFileToString("/proc/cmdline", &cmdline);
243 
244     for (const auto& entry : android::base::Split(android::base::Trim(cmdline), " ")) {
245         std::vector<std::string> pieces = android::base::Split(entry, "=");
246         if (pieces.size() == 2) {
247             fn(pieces[0], pieces[1]);
248         }
249     }
250 }
251 
ImportBootconfig(const std::function<void (const std::string &,const std::string &)> & fn)252 void ImportBootconfig(const std::function<void(const std::string&, const std::string&)>& fn) {
253     std::string bootconfig;
254     android::base::ReadFileToString("/proc/bootconfig", &bootconfig);
255 
256     for (const auto& entry : android::base::Split(bootconfig, "\n")) {
257         std::vector<std::string> pieces = android::base::Split(entry, "=");
258         if (pieces.size() == 2) {
259             // get rid of the extra space between a list of values and remove the quotes.
260             std::string value = android::base::StringReplace(pieces[1], "\", \"", ",", true);
261             value.erase(std::remove(value.begin(), value.end(), '"'), value.end());
262             fn(android::base::Trim(pieces[0]), android::base::Trim(value));
263         }
264     }
265 }
266 
make_dir(const std::string & path,mode_t mode)267 bool make_dir(const std::string& path, mode_t mode) {
268     std::string secontext;
269     if (SelabelLookupFileContext(path, mode, &secontext) && !secontext.empty()) {
270         setfscreatecon(secontext.c_str());
271     }
272 
273     int rc = mkdir(path.c_str(), mode);
274 
275     if (!secontext.empty()) {
276         int save_errno = errno;
277         setfscreatecon(nullptr);
278         errno = save_errno;
279     }
280 
281     return rc == 0;
282 }
283 
284 /*
285  * Returns true is pathname is a directory
286  */
is_dir(const char * pathname)287 bool is_dir(const char* pathname) {
288     struct stat info;
289     if (stat(pathname, &info) == -1) {
290         return false;
291     }
292     return S_ISDIR(info.st_mode);
293 }
294 
ExpandProps(const std::string & src)295 Result<std::string> ExpandProps(const std::string& src) {
296     const char* src_ptr = src.c_str();
297 
298     std::string dst;
299 
300     /* - variables can either be $x.y or ${x.y}, in case they are only part
301      *   of the string.
302      * - will accept $$ as a literal $.
303      * - no nested property expansion, i.e. ${foo.${bar}} is not supported,
304      *   bad things will happen
305      * - ${x.y:-default} will return default value if property empty.
306      */
307     while (*src_ptr) {
308         const char* c;
309 
310         c = strchr(src_ptr, '$');
311         if (!c) {
312             dst.append(src_ptr);
313             return dst;
314         }
315 
316         dst.append(src_ptr, c);
317         c++;
318 
319         if (*c == '$') {
320             dst.push_back(*(c++));
321             src_ptr = c;
322             continue;
323         } else if (*c == '\0') {
324             return dst;
325         }
326 
327         std::string prop_name;
328         std::string def_val;
329         if (*c == '{') {
330             c++;
331             const char* end = strchr(c, '}');
332             if (!end) {
333                 // failed to find closing brace, abort.
334                 return Error() << "unexpected end of string in '" << src << "', looking for }";
335             }
336             prop_name = std::string(c, end);
337             c = end + 1;
338             size_t def = prop_name.find(":-");
339             if (def < prop_name.size()) {
340                 def_val = prop_name.substr(def + 2);
341                 prop_name = prop_name.substr(0, def);
342             }
343         } else {
344             prop_name = c;
345             if (SelinuxGetVendorAndroidVersion() >= __ANDROID_API_R__) {
346                 return Error() << "using deprecated syntax for specifying property '" << c
347                                << "', use ${name} instead";
348             } else {
349                 LOG(ERROR) << "using deprecated syntax for specifying property '" << c
350                            << "', use ${name} instead";
351             }
352             c += prop_name.size();
353         }
354 
355         if (prop_name.empty()) {
356             return Error() << "invalid zero-length property name in '" << src << "'";
357         }
358 
359         std::string prop_val = android::base::GetProperty(prop_name, "");
360         if (prop_val.empty()) {
361             if (def_val.empty()) {
362                 return Error() << "property '" << prop_name << "' doesn't exist while expanding '"
363                                << src << "'";
364             }
365             prop_val = def_val;
366         }
367 
368         dst.append(prop_val);
369         src_ptr = c;
370     }
371 
372     return dst;
373 }
374 
init_android_dt_dir()375 static std::string init_android_dt_dir() {
376     // Use the standard procfs-based path by default
377     std::string android_dt_dir = kDefaultAndroidDtDir;
378     // The platform may specify a custom Android DT path in kernel cmdline
379     ImportKernelCmdline([&](const std::string& key, const std::string& value) {
380         if (key == "androidboot.android_dt_dir") {
381             android_dt_dir = value;
382         }
383     });
384     // ..Or bootconfig
385     if (android_dt_dir == kDefaultAndroidDtDir) {
386         ImportBootconfig([&](const std::string& key, const std::string& value) {
387             if (key == "androidboot.android_dt_dir") {
388                 android_dt_dir = value;
389             }
390         });
391     }
392 
393     LOG(INFO) << "Using Android DT directory " << android_dt_dir;
394     return android_dt_dir;
395 }
396 
397 // FIXME: The same logic is duplicated in system/core/fs_mgr/
get_android_dt_dir()398 const std::string& get_android_dt_dir() {
399     // Set once and saves time for subsequent calls to this function
400     static const std::string kAndroidDtDir = init_android_dt_dir();
401     return kAndroidDtDir;
402 }
403 
404 // Reads the content of device tree file under the platform's Android DT directory.
405 // Returns true if the read is success, false otherwise.
read_android_dt_file(const std::string & sub_path,std::string * dt_content)406 bool read_android_dt_file(const std::string& sub_path, std::string* dt_content) {
407     const std::string file_name = get_android_dt_dir() + sub_path;
408     if (android::base::ReadFileToString(file_name, dt_content)) {
409         if (!dt_content->empty()) {
410             dt_content->pop_back();  // Trims the trailing '\0' out.
411             return true;
412         }
413     }
414     return false;
415 }
416 
is_android_dt_value_expected(const std::string & sub_path,const std::string & expected_content)417 bool is_android_dt_value_expected(const std::string& sub_path, const std::string& expected_content) {
418     std::string dt_content;
419     if (read_android_dt_file(sub_path, &dt_content)) {
420         if (dt_content == expected_content) {
421             return true;
422         }
423     }
424     return false;
425 }
426 
IsLegalPropertyName(const std::string & name)427 bool IsLegalPropertyName(const std::string& name) {
428     size_t namelen = name.size();
429 
430     if (namelen < 1) return false;
431     if (name[0] == '.') return false;
432     if (name[namelen - 1] == '.') return false;
433 
434     /* Only allow alphanumeric, plus '.', '-', '@', ':', or '_' */
435     /* Don't allow ".." to appear in a property name */
436     for (size_t i = 0; i < namelen; i++) {
437         if (name[i] == '.') {
438             // i=0 is guaranteed to never have a dot. See above.
439             if (name[i - 1] == '.') return false;
440             continue;
441         }
442         if (name[i] == '_' || name[i] == '-' || name[i] == '@' || name[i] == ':') continue;
443         if (name[i] >= 'a' && name[i] <= 'z') continue;
444         if (name[i] >= 'A' && name[i] <= 'Z') continue;
445         if (name[i] >= '0' && name[i] <= '9') continue;
446         return false;
447     }
448 
449     return true;
450 }
451 
IsLegalPropertyValue(const std::string & name,const std::string & value)452 Result<void> IsLegalPropertyValue(const std::string& name, const std::string& value) {
453     if (value.size() >= PROP_VALUE_MAX && !StartsWith(name, "ro.")) {
454         return Error() << "Property value too long";
455     }
456 
457     if (mbstowcs(nullptr, value.data(), 0) == static_cast<std::size_t>(-1)) {
458         return Error() << "Value is not a UTF8 encoded string";
459     }
460 
461     return {};
462 }
463 
FscryptInferAction(const std::string & dir)464 static FscryptAction FscryptInferAction(const std::string& dir) {
465     const std::string prefix = "/data/";
466 
467     if (!android::base::StartsWith(dir, prefix)) {
468         return FscryptAction::kNone;
469     }
470 
471     // Special-case /data/media/obb per b/64566063
472     if (dir == "/data/media/obb") {
473         // Try to set policy on this directory, but if it is non-empty this may fail.
474         return FscryptAction::kAttempt;
475     }
476 
477     // Only set policy on first level /data directories
478     // To make this less restrictive, consider using a policy file.
479     // However this is overkill for as long as the policy is simply
480     // to apply a global policy to all /data folders created via makedir
481     if (dir.find_first_of('/', prefix.size()) != std::string::npos) {
482         return FscryptAction::kNone;
483     }
484 
485     // Special case various directories that must not be encrypted,
486     // often because their subdirectories must be encrypted.
487     // This isn't a nice way to do this, see b/26641735
488     std::vector<std::string> directories_to_exclude = {
489             "lost+found", "system_ce", "system_de", "misc_ce",     "misc_de",
490             "vendor_ce",  "vendor_de", "media",     "data",        "user",
491             "user_de",    "apex",      "preloads",  "app-staging", "gsi",
492     };
493     for (const auto& d : directories_to_exclude) {
494         if ((prefix + d) == dir) {
495             return FscryptAction::kNone;
496         }
497     }
498     // Empty these directories if policy setting fails.
499     std::vector<std::string> wipe_on_failure = {
500             "rollback", "rollback-observer",  // b/139193659
501     };
502     for (const auto& d : wipe_on_failure) {
503         if ((prefix + d) == dir) {
504             return FscryptAction::kDeleteIfNecessary;
505         }
506     }
507     return FscryptAction::kRequire;
508 }
509 
ParseMkdir(const std::vector<std::string> & args)510 Result<MkdirOptions> ParseMkdir(const std::vector<std::string>& args) {
511     mode_t mode = 0755;
512     Result<uid_t> uid = -1;
513     Result<gid_t> gid = -1;
514     FscryptAction fscrypt_inferred_action = FscryptInferAction(args[1]);
515     FscryptAction fscrypt_action = fscrypt_inferred_action;
516     std::string ref_option = "ref";
517     bool set_option_encryption = false;
518     bool set_option_key = false;
519 
520     for (size_t i = 2; i < args.size(); i++) {
521         switch (i) {
522             case 2:
523                 mode = std::strtoul(args[2].c_str(), 0, 8);
524                 break;
525             case 3:
526                 uid = DecodeUid(args[3]);
527                 if (!uid.ok()) {
528                     return Error()
529                            << "Unable to decode UID for '" << args[3] << "': " << uid.error();
530                 }
531                 break;
532             case 4:
533                 gid = DecodeUid(args[4]);
534                 if (!gid.ok()) {
535                     return Error()
536                            << "Unable to decode GID for '" << args[4] << "': " << gid.error();
537                 }
538                 break;
539             default:
540                 auto parts = android::base::Split(args[i], "=");
541                 if (parts.size() != 2) {
542                     return Error() << "Can't parse option: '" << args[i] << "'";
543                 }
544                 auto optname = parts[0];
545                 auto optval = parts[1];
546                 if (optname == "encryption") {
547                     if (set_option_encryption) {
548                         return Error() << "Duplicated option: '" << optname << "'";
549                     }
550                     if (optval == "Require") {
551                         fscrypt_action = FscryptAction::kRequire;
552                     } else if (optval == "None") {
553                         fscrypt_action = FscryptAction::kNone;
554                     } else if (optval == "Attempt") {
555                         fscrypt_action = FscryptAction::kAttempt;
556                     } else if (optval == "DeleteIfNecessary") {
557                         fscrypt_action = FscryptAction::kDeleteIfNecessary;
558                     } else {
559                         return Error() << "Unknown encryption option: '" << optval << "'";
560                     }
561                     set_option_encryption = true;
562                 } else if (optname == "key") {
563                     if (set_option_key) {
564                         return Error() << "Duplicated option: '" << optname << "'";
565                     }
566                     if (optval == "ref" || optval == "per_boot_ref") {
567                         ref_option = optval;
568                     } else {
569                         return Error() << "Unknown key option: '" << optval << "'";
570                     }
571                     set_option_key = true;
572                 } else {
573                     return Error() << "Unknown option: '" << args[i] << "'";
574                 }
575         }
576     }
577     if (set_option_key && fscrypt_action == FscryptAction::kNone) {
578         return Error() << "Key option set but encryption action is none";
579     }
580     const std::string prefix = "/data/";
581     if (StartsWith(args[1], prefix) &&
582         args[1].find_first_of('/', prefix.size()) == std::string::npos) {
583         if (!set_option_encryption) {
584             LOG(WARNING) << "Top-level directory needs encryption action, eg mkdir " << args[1]
585                          << " <mode> <uid> <gid> encryption=Require";
586         }
587         if (fscrypt_action == FscryptAction::kNone) {
588             LOG(INFO) << "Not setting encryption policy on: " << args[1];
589         }
590     }
591     if (fscrypt_action != fscrypt_inferred_action) {
592         LOG(WARNING) << "Inferred action different from explicit one, expected "
593                      << static_cast<int>(fscrypt_inferred_action) << " but got "
594                      << static_cast<int>(fscrypt_action);
595     }
596 
597     return MkdirOptions{args[1], mode, *uid, *gid, fscrypt_action, ref_option};
598 }
599 
ParseMountAll(const std::vector<std::string> & args)600 Result<MountAllOptions> ParseMountAll(const std::vector<std::string>& args) {
601     bool compat_mode = false;
602     bool import_rc = false;
603     if (SelinuxGetVendorAndroidVersion() <= __ANDROID_API_Q__) {
604         if (args.size() <= 1) {
605             return Error() << "mount_all requires at least 1 argument";
606         }
607         compat_mode = true;
608         import_rc = true;
609     }
610 
611     std::size_t first_option_arg = args.size();
612     enum mount_mode mode = MOUNT_MODE_DEFAULT;
613 
614     // If we are <= Q, then stop looking for non-fstab arguments at slot 2.
615     // Otherwise, stop looking at slot 1 (as the fstab path argument is optional >= R).
616     for (std::size_t na = args.size() - 1; na > (compat_mode ? 1 : 0); --na) {
617         if (args[na] == "--early") {
618             first_option_arg = na;
619             mode = MOUNT_MODE_EARLY;
620         } else if (args[na] == "--late") {
621             first_option_arg = na;
622             mode = MOUNT_MODE_LATE;
623             import_rc = false;
624         }
625     }
626 
627     std::string fstab_path;
628     if (first_option_arg > 1) {
629         fstab_path = args[1];
630     } else if (compat_mode) {
631         return Error() << "mount_all argument 1 must be the fstab path";
632     }
633 
634     std::vector<std::string> rc_paths;
635     for (std::size_t na = 2; na < first_option_arg; ++na) {
636         rc_paths.push_back(args[na]);
637     }
638 
639     return MountAllOptions{rc_paths, fstab_path, mode, import_rc};
640 }
641 
ParseRestorecon(const std::vector<std::string> & args)642 Result<std::pair<int, std::vector<std::string>>> ParseRestorecon(
643         const std::vector<std::string>& args) {
644     struct flag_type {
645         const char* name;
646         int value;
647     };
648     static const flag_type flags[] = {
649             {"--recursive", SELINUX_ANDROID_RESTORECON_RECURSE},
650             {"--skip-ce", SELINUX_ANDROID_RESTORECON_SKIPCE},
651             {"--cross-filesystems", SELINUX_ANDROID_RESTORECON_CROSS_FILESYSTEMS},
652             {0, 0}};
653 
654     int flag = 0;
655     std::vector<std::string> paths;
656 
657     bool in_flags = true;
658     for (size_t i = 1; i < args.size(); ++i) {
659         if (android::base::StartsWith(args[i], "--")) {
660             if (!in_flags) {
661                 return Error() << "flags must precede paths";
662             }
663             bool found = false;
664             for (size_t j = 0; flags[j].name; ++j) {
665                 if (args[i] == flags[j].name) {
666                     flag |= flags[j].value;
667                     found = true;
668                     break;
669                 }
670             }
671             if (!found) {
672                 return Error() << "bad flag " << args[i];
673             }
674         } else {
675             in_flags = false;
676             paths.emplace_back(args[i]);
677         }
678     }
679     return std::pair(flag, paths);
680 }
681 
ParseSwaponAll(const std::vector<std::string> & args)682 Result<std::string> ParseSwaponAll(const std::vector<std::string>& args) {
683     if (args.size() <= 1) {
684         if (SelinuxGetVendorAndroidVersion() <= __ANDROID_API_Q__) {
685             return Error() << "swapon_all requires at least 1 argument";
686         }
687         return {};
688     }
689     return args[1];
690 }
691 
ParseUmountAll(const std::vector<std::string> & args)692 Result<std::string> ParseUmountAll(const std::vector<std::string>& args) {
693     if (args.size() <= 1) {
694         if (SelinuxGetVendorAndroidVersion() <= __ANDROID_API_Q__) {
695             return Error() << "umount_all requires at least 1 argument";
696         }
697         return {};
698     }
699     return args[1];
700 }
701 
InitAborter(const char * abort_message)702 static void InitAborter(const char* abort_message) {
703     // When init forks, it continues to use this aborter for LOG(FATAL), but we want children to
704     // simply abort instead of trying to reboot the system.
705     if (getpid() != 1) {
706         android::base::DefaultAborter(abort_message);
707         return;
708     }
709 
710     InitFatalReboot(SIGABRT);
711 }
712 
713 // The kernel opens /dev/console and uses that fd for stdin/stdout/stderr if there is a serial
714 // console enabled and no initramfs, otherwise it does not provide any fds for stdin/stdout/stderr.
715 // SetStdioToDevNull() is used to close these existing fds if they exist and replace them with
716 // /dev/null regardless.
717 //
718 // In the case that these fds are provided by the kernel, the exec of second stage init causes an
719 // SELinux denial as it does not have access to /dev/console.  In the case that they are not
720 // provided, exec of any further process is potentially dangerous as the first fd's opened by that
721 // process will take the stdin/stdout/stderr fileno's, which can cause issues if printf(), etc is
722 // then used by that process.
723 //
724 // Lastly, simply calling SetStdioToDevNull() in first stage init is not enough, since first
725 // stage init still runs in kernel context, future child processes will not have permissions to
726 // access any fds that it opens, including the one opened below for /dev/null.  Therefore,
727 // SetStdioToDevNull() must be called again in second stage init.
SetStdioToDevNull(char ** argv)728 void SetStdioToDevNull(char** argv) {
729     // Make stdin/stdout/stderr all point to /dev/null.
730     int fd = open("/dev/null", O_RDWR);  // NOLINT(android-cloexec-open)
731     if (fd == -1) {
732         int saved_errno = errno;
733         android::base::InitLogging(argv, &android::base::KernelLogger, InitAborter);
734         errno = saved_errno;
735         PLOG(FATAL) << "Couldn't open /dev/null";
736     }
737     dup2(fd, STDIN_FILENO);
738     dup2(fd, STDOUT_FILENO);
739     dup2(fd, STDERR_FILENO);
740     if (fd > STDERR_FILENO) close(fd);
741 }
742 
InitKernelLogging(char ** argv)743 void InitKernelLogging(char** argv) {
744     SetFatalRebootTarget();
745     android::base::InitLogging(argv, &android::base::KernelLogger, InitAborter);
746 }
747 
IsRecoveryMode()748 bool IsRecoveryMode() {
749     return access("/system/bin/recovery", F_OK) == 0;
750 }
751 
752 // Check if default mount namespace is ready to be used with APEX modules
753 static bool is_default_mount_namespace_ready = false;
754 
IsDefaultMountNamespaceReady()755 bool IsDefaultMountNamespaceReady() {
756     return is_default_mount_namespace_ready;
757 }
758 
SetDefaultMountNamespaceReady()759 void SetDefaultMountNamespaceReady() {
760     is_default_mount_namespace_ready = true;
761 }
762 
IsMicrodroid()763 bool IsMicrodroid() {
764     static bool is_microdroid = android::base::GetProperty("ro.hardware", "") == "microdroid";
765     return is_microdroid;
766 }
767 
768 }  // namespace init
769 }  // namespace android
770