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