service.cpp revision a72436067d5f88885e4b32fa7c5d22f1ea8c0756
1bac3299720623f4226bca103b26260052732ad30Tom Cherry/* 2bac3299720623f4226bca103b26260052732ad30Tom Cherry * Copyright (C) 2015 The Android Open Source Project 3bac3299720623f4226bca103b26260052732ad30Tom Cherry * 4bac3299720623f4226bca103b26260052732ad30Tom Cherry * Licensed under the Apache License, Version 2.0 (the "License"); 5bac3299720623f4226bca103b26260052732ad30Tom Cherry * you may not use this file except in compliance with the License. 6bac3299720623f4226bca103b26260052732ad30Tom Cherry * You may obtain a copy of the License at 7bac3299720623f4226bca103b26260052732ad30Tom Cherry * 8bac3299720623f4226bca103b26260052732ad30Tom Cherry * http://www.apache.org/licenses/LICENSE-2.0 9bac3299720623f4226bca103b26260052732ad30Tom Cherry * 10bac3299720623f4226bca103b26260052732ad30Tom Cherry * Unless required by applicable law or agreed to in writing, software 11bac3299720623f4226bca103b26260052732ad30Tom Cherry * distributed under the License is distributed on an "AS IS" BASIS, 12bac3299720623f4226bca103b26260052732ad30Tom Cherry * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13bac3299720623f4226bca103b26260052732ad30Tom Cherry * See the License for the specific language governing permissions and 14bac3299720623f4226bca103b26260052732ad30Tom Cherry * limitations under the License. 15bac3299720623f4226bca103b26260052732ad30Tom Cherry */ 16bac3299720623f4226bca103b26260052732ad30Tom Cherry 17bac3299720623f4226bca103b26260052732ad30Tom Cherry#include "service.h" 18bac3299720623f4226bca103b26260052732ad30Tom Cherry 19bac3299720623f4226bca103b26260052732ad30Tom Cherry#include <fcntl.h> 209605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes#include <inttypes.h> 2124b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes#include <linux/securebits.h> 221b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes#include <sched.h> 231b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes#include <sys/mount.h> 241b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes#include <sys/prctl.h> 25081705c258efbe938d71c2022528d809fa6d42c5Vitalii Tomkiv#include <sys/resource.h> 26bac3299720623f4226bca103b26260052732ad30Tom Cherry#include <sys/stat.h> 27f57c0bfab20b4fae10fc2e385d5e32fc93895670Tom Cherry#include <sys/system_properties.h> 28081705c258efbe938d71c2022528d809fa6d42c5Vitalii Tomkiv#include <sys/time.h> 29b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET#include <sys/wait.h> 30bac3299720623f4226bca103b26260052732ad30Tom Cherry#include <termios.h> 31af9ba4dc6ca98e136e887b0baa59b72fb8302dd3Dan Albert#include <unistd.h> 32bac3299720623f4226bca103b26260052732ad30Tom Cherry 334f71319df011d796a60a43fc1bc68e16fbf7d321Elliott Hughes#include <android-base/file.h> 343f5eaae526413a29de899270714469c76dc91ec8Tom Cherry#include <android-base/logging.h> 35da46b392f10d2809b5696632f67485f272ec5698Elliott Hughes#include <android-base/parseint.h> 36ccf23537eeacfa47e5f18dd3b75089886d177c1bTom Cherry#include <android-base/properties.h> 378d13d808a5634701c55276796683857ac2d5c99aTom Cherry#include <android-base/scopeguard.h> 384f71319df011d796a60a43fc1bc68e16fbf7d321Elliott Hughes#include <android-base/stringprintf.h> 39f86b5a6b90619e02d1d034ef7b0adc3b439f4abbElliott Hughes#include <android-base/strings.h> 40f7e79b99c1e9e3128dd9921871f7740bebb755e6Collin Mulliner#include <processgroup/processgroup.h> 413f5eaae526413a29de899270714469c76dc91ec8Tom Cherry#include <selinux/selinux.h> 423f5eaae526413a29de899270714469c76dc91ec8Tom Cherry#include <system/thread_defs.h> 43f7e79b99c1e9e3128dd9921871f7740bebb755e6Collin Mulliner 44bac3299720623f4226bca103b26260052732ad30Tom Cherry#include "init.h" 45bac3299720623f4226bca103b26260052732ad30Tom Cherry#include "property_service.h" 46bac3299720623f4226bca103b26260052732ad30Tom Cherry#include "util.h" 47bac3299720623f4226bca103b26260052732ad30Tom Cherry 48e78ea77f6992013b16fd57c07548b53129cdd12fJames Hawkinsusing android::base::boot_clock; 4981f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherryusing android::base::GetProperty; 5081f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherryusing android::base::Join; 518d13d808a5634701c55276796683857ac2d5c99aTom Cherryusing android::base::make_scope_guard; 52da46b392f10d2809b5696632f67485f272ec5698Elliott Hughesusing android::base::ParseInt; 5381f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherryusing android::base::StartsWith; 54b7349902a945903f9e36a569051f5131beb0bc24Tom Cherryusing android::base::StringPrintf; 55b7349902a945903f9e36a569051f5131beb0bc24Tom Cherryusing android::base::WriteStringToFile; 56b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 5781f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherrynamespace android { 5881f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherrynamespace init { 5981f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry 60344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obesstatic std::string ComputeContextFromExecutable(std::string& service_name, 61344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes const std::string& service_path) { 62344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes std::string computed_context; 63344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes 64344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes char* raw_con = nullptr; 65344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes char* raw_filecon = nullptr; 66344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes 67344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (getcon(&raw_con) == -1) { 68344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes LOG(ERROR) << "could not get context while starting '" << service_name << "'"; 69344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes return ""; 70344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 71344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes std::unique_ptr<char> mycon(raw_con); 72344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes 73344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (getfilecon(service_path.c_str(), &raw_filecon) == -1) { 74344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes LOG(ERROR) << "could not get file context while starting '" << service_name << "'"; 75344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes return ""; 76344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 77344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes std::unique_ptr<char> filecon(raw_filecon); 78344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes 79344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes char* new_con = nullptr; 80344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes int rc = security_compute_create(mycon.get(), filecon.get(), 81344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes string_to_security_class("process"), &new_con); 82344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (rc == 0) { 83344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes computed_context = new_con; 84344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes free(new_con); 85344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 86344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (rc == 0 && computed_context == mycon.get()) { 87344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes LOG(ERROR) << "service " << service_name << " does not have a SELinux domain defined"; 88344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes return ""; 89344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 90344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (rc < 0) { 91344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes LOG(ERROR) << "could not get context while starting '" << service_name << "'"; 92344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes return ""; 93344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 94344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes return computed_context; 95344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes} 96344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes 971b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obesstatic void SetUpPidNamespace(const std::string& service_name) { 981b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes constexpr unsigned int kSafeFlags = MS_NODEV | MS_NOEXEC | MS_NOSUID; 991b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes 1001b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes // It's OK to LOG(FATAL) in this function since it's running in the first 1011b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes // child process. 1021b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes if (mount("", "/proc", "proc", kSafeFlags | MS_REMOUNT, "") == -1) { 103e18e7e5c43c39677c6d49f8f76893d720f90b4fbElliott Hughes PLOG(FATAL) << "couldn't remount(/proc) for " << service_name; 1041b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } 1051b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes 1061b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes if (prctl(PR_SET_NAME, service_name.c_str()) == -1) { 107e18e7e5c43c39677c6d49f8f76893d720f90b4fbElliott Hughes PLOG(FATAL) << "couldn't set name for " << service_name; 1081b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } 1091b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes 1101b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes pid_t child_pid = fork(); 1111b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes if (child_pid == -1) { 112e18e7e5c43c39677c6d49f8f76893d720f90b4fbElliott Hughes PLOG(FATAL) << "couldn't fork init inside the PID namespace for " << service_name; 1131b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } 1141b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes 1151b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes if (child_pid > 0) { 1161b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes // So that we exit with the right status. 1171b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes static int init_exitstatus = 0; 1181b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes signal(SIGTERM, [](int) { _exit(init_exitstatus); }); 1191b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes 1201b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes pid_t waited_pid; 1211b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes int status; 1221b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes while ((waited_pid = wait(&status)) > 0) { 1231b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes // This loop will end when there are no processes left inside the 1241b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes // PID namespace or when the init process inside the PID namespace 1251b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes // gets a signal. 1261b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes if (waited_pid == child_pid) { 1271b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes init_exitstatus = status; 1281b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } 1291b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } 1301b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes if (!WIFEXITED(init_exitstatus)) { 1311b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes _exit(EXIT_FAILURE); 1321b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } 1331b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes _exit(WEXITSTATUS(init_exitstatus)); 1341b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } 1351b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes} 1361b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes 137344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obesstatic void ExpandArgs(const std::vector<std::string>& args, std::vector<char*>* strs) { 138344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes std::vector<std::string> expanded_args; 139344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes expanded_args.resize(args.size()); 140344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes strs->push_back(const_cast<char*>(args[0].c_str())); 141344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes for (std::size_t i = 1; i < args.size(); ++i) { 142344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (!expand_props(args[i], &expanded_args[i])) { 143344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes LOG(FATAL) << args[0] << ": cannot expand '" << args[i] << "'"; 144344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 145344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes strs->push_back(const_cast<char*>(expanded_args[i].c_str())); 146344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 147344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes strs->push_back(nullptr); 148344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes} 149344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes 150bac3299720623f4226bca103b26260052732ad30Tom CherryServiceEnvironmentInfo::ServiceEnvironmentInfo() { 151bac3299720623f4226bca103b26260052732ad30Tom Cherry} 152bac3299720623f4226bca103b26260052732ad30Tom Cherry 153bac3299720623f4226bca103b26260052732ad30Tom CherryServiceEnvironmentInfo::ServiceEnvironmentInfo(const std::string& name, 154bac3299720623f4226bca103b26260052732ad30Tom Cherry const std::string& value) 155bac3299720623f4226bca103b26260052732ad30Tom Cherry : name(name), value(value) { 156bac3299720623f4226bca103b26260052732ad30Tom Cherry} 157bac3299720623f4226bca103b26260052732ad30Tom Cherry 158641ff0a4d8d05349d9c22f726a035a2936a2a94dWei WangService::Service(const std::string& name, const std::vector<std::string>& args) 1595d17d044c757d2419194f057bd3191f145da4aa5Tom Cherry : Service(name, 0, 0, 0, {}, 0, 0, "", args) {} 160bac3299720623f4226bca103b26260052732ad30Tom Cherry 161641ff0a4d8d05349d9c22f726a035a2936a2a94dWei WangService::Service(const std::string& name, unsigned flags, uid_t uid, gid_t gid, 162641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang const std::vector<gid_t>& supp_gids, const CapSet& capabilities, 163641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang unsigned namespace_flags, const std::string& seclabel, 1641b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes const std::vector<std::string>& args) 165641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang : name_(name), 166641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang classnames_({"default"}), 167641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang flags_(flags), 168641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang pid_(0), 169641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang crash_count_(0), 170641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang uid_(uid), 171641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang gid_(gid), 172641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang supp_gids_(supp_gids), 173641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang capabilities_(capabilities), 174641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang namespace_flags_(namespace_flags), 175641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang seclabel_(seclabel), 176012c573e267b8dd70de14237cb470bd7301ee8eaTom Cherry onrestart_(false, "<Service '" + name + "' onrestart>", 0), 1777da548578c25683fe0082283303e16961df312daTom Cherry keychord_id_(0), 178641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang ioprio_class_(IoSchedClass_NONE), 179641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang ioprio_pri_(0), 180641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang priority_(0), 181641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang oom_score_adjust_(-1000), 182d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea swappiness_(-1), 183d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea soft_limit_in_bytes_(-1), 184d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea limit_in_bytes_(-1), 185641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang args_(args) { 186bac3299720623f4226bca103b26260052732ad30Tom Cherry onrestart_.InitSingleTrigger("onrestart"); 187bac3299720623f4226bca103b26260052732ad30Tom Cherry} 188bac3299720623f4226bca103b26260052732ad30Tom Cherry 189bac3299720623f4226bca103b26260052732ad30Tom Cherryvoid Service::NotifyStateChange(const std::string& new_state) const { 190b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry if ((flags_ & SVC_TEMPORARY) != 0) { 191b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry // Services created by 'exec' are temporary and don't have properties tracking their state. 192bac3299720623f4226bca103b26260052732ad30Tom Cherry return; 193bac3299720623f4226bca103b26260052732ad30Tom Cherry } 194bac3299720623f4226bca103b26260052732ad30Tom Cherry 1951c3a53f03ca3c2c647f83cd8b8ae7e18c5c7bc69Tom Cherry std::string prop_name = "init.svc." + name_; 1961c3a53f03ca3c2c647f83cd8b8ae7e18c5c7bc69Tom Cherry property_set(prop_name, new_state); 1979605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes 1989605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes if (new_state == "running") { 1999605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes uint64_t start_ns = time_started_.time_since_epoch().count(); 2001c3a53f03ca3c2c647f83cd8b8ae7e18c5c7bc69Tom Cherry property_set("ro.boottime." + name_, std::to_string(start_ns)); 2019605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes } 202bac3299720623f4226bca103b26260052732ad30Tom Cherry} 203bac3299720623f4226bca103b26260052732ad30Tom Cherry 204ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughesvoid Service::KillProcessGroup(int signal) { 20533838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry // If we've already seen a successful result from killProcessGroup*(), then we have removed 20633838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry // the cgroup already and calling these functions a second time will simply result in an error. 20733838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry // This is true regardless of which signal was sent. 20833838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry // These functions handle their own logging, so no additional logging is needed. 20933838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry if (!process_cgroup_empty_) { 21033838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry LOG(INFO) << "Sending signal " << signal << " to service '" << name_ << "' (pid " << pid_ 21133838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry << ") process group..."; 21233838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry int r; 21333838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry if (signal == SIGTERM) { 21433838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry r = killProcessGroupOnce(uid_, pid_, signal); 21533838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry } else { 21633838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry r = killProcessGroup(uid_, pid_, signal); 21733838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry } 21833838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry 21933838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry if (r == 0) process_cgroup_empty_ = true; 22033838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry } 221ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes} 222ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes 223344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obesvoid Service::SetProcessAttributes() { 22424b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes // Keep capabilites on uid change. 22524b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes if (capabilities_.any() && uid_) { 226f5965519d183ff188273689d151bfc2cb20ece2aLuis Hector Chavez // If Android is running in a container, some securebits might already 227f5965519d183ff188273689d151bfc2cb20ece2aLuis Hector Chavez // be locked, so don't change those. 228a72436067d5f88885e4b32fa7c5d22f1ea8c0756Ben Fennema unsigned long securebits = prctl(PR_GET_SECUREBITS); 229a72436067d5f88885e4b32fa7c5d22f1ea8c0756Ben Fennema if (securebits == -1UL) { 230f5965519d183ff188273689d151bfc2cb20ece2aLuis Hector Chavez PLOG(FATAL) << "prctl(PR_GET_SECUREBITS) failed for " << name_; 231f5965519d183ff188273689d151bfc2cb20ece2aLuis Hector Chavez } 232f5965519d183ff188273689d151bfc2cb20ece2aLuis Hector Chavez securebits |= SECBIT_KEEP_CAPS | SECBIT_KEEP_CAPS_LOCKED; 233f5965519d183ff188273689d151bfc2cb20ece2aLuis Hector Chavez if (prctl(PR_SET_SECUREBITS, securebits) != 0) { 234f5965519d183ff188273689d151bfc2cb20ece2aLuis Hector Chavez PLOG(FATAL) << "prctl(PR_SET_SECUREBITS) failed for " << name_; 23524b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes } 23624b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes } 23724b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes 238e18e7e5c43c39677c6d49f8f76893d720f90b4fbElliott Hughes // TODO: work out why this fails for `console` then upgrade to FATAL. 239e18e7e5c43c39677c6d49f8f76893d720f90b4fbElliott Hughes if (setpgid(0, getpid()) == -1) PLOG(ERROR) << "setpgid failed for " << name_; 240344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes 241344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (gid_) { 242344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (setgid(gid_) != 0) { 243e18e7e5c43c39677c6d49f8f76893d720f90b4fbElliott Hughes PLOG(FATAL) << "setgid failed for " << name_; 244344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 245344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 24680960d2a9a6e15931d946cc826dcb3d5bf68ca4fNick Kralevich if (setgroups(supp_gids_.size(), &supp_gids_[0]) != 0) { 24780960d2a9a6e15931d946cc826dcb3d5bf68ca4fNick Kralevich PLOG(FATAL) << "setgroups failed for " << name_; 248344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 249344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (uid_) { 250344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (setuid(uid_) != 0) { 251e18e7e5c43c39677c6d49f8f76893d720f90b4fbElliott Hughes PLOG(FATAL) << "setuid failed for " << name_; 252344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 253344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 254344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (!seclabel_.empty()) { 255344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (setexeccon(seclabel_.c_str()) < 0) { 256e18e7e5c43c39677c6d49f8f76893d720f90b4fbElliott Hughes PLOG(FATAL) << "cannot setexeccon('" << seclabel_ << "') for " << name_; 257344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 258344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 259344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (priority_ != 0) { 260344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (setpriority(PRIO_PROCESS, 0, priority_) != 0) { 261e18e7e5c43c39677c6d49f8f76893d720f90b4fbElliott Hughes PLOG(FATAL) << "setpriority failed for " << name_; 262344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 263344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes } 26424b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes if (capabilities_.any()) { 26524b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes if (!SetCapsForExec(capabilities_)) { 26624b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes LOG(FATAL) << "cannot set capabilities for " << name_; 26724b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes } 26824b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes } 269344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes} 270344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes 271b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherryvoid Service::Reap() { 272bac3299720623f4226bca103b26260052732ad30Tom Cherry if (!(flags_ & SVC_ONESHOT) || (flags_ & SVC_RESTART)) { 273ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes KillProcessGroup(SIGKILL); 274bac3299720623f4226bca103b26260052732ad30Tom Cherry } 275bac3299720623f4226bca103b26260052732ad30Tom Cherry 27662767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn // Remove any descriptor resources we may have created. 27762767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn std::for_each(descriptors_.begin(), descriptors_.end(), 27862767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn std::bind(&DescriptorInfo::Clean, std::placeholders::_1)); 279bac3299720623f4226bca103b26260052732ad30Tom Cherry 280b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry if (flags_ & SVC_TEMPORARY) { 281b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry return; 282bac3299720623f4226bca103b26260052732ad30Tom Cherry } 283bac3299720623f4226bca103b26260052732ad30Tom Cherry 284bac3299720623f4226bca103b26260052732ad30Tom Cherry pid_ = 0; 285bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ &= (~SVC_RUNNING); 286bac3299720623f4226bca103b26260052732ad30Tom Cherry 287bac3299720623f4226bca103b26260052732ad30Tom Cherry // Oneshot processes go into the disabled state on exit, 288bac3299720623f4226bca103b26260052732ad30Tom Cherry // except when manually restarted. 289bac3299720623f4226bca103b26260052732ad30Tom Cherry if ((flags_ & SVC_ONESHOT) && !(flags_ & SVC_RESTART)) { 290bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ |= SVC_DISABLED; 291bac3299720623f4226bca103b26260052732ad30Tom Cherry } 292bac3299720623f4226bca103b26260052732ad30Tom Cherry 293bac3299720623f4226bca103b26260052732ad30Tom Cherry // Disabled and reset processes do not get restarted automatically. 294bac3299720623f4226bca103b26260052732ad30Tom Cherry if (flags_ & (SVC_DISABLED | SVC_RESET)) { 295bac3299720623f4226bca103b26260052732ad30Tom Cherry NotifyStateChange("stopped"); 296b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry return; 297bac3299720623f4226bca103b26260052732ad30Tom Cherry } 298bac3299720623f4226bca103b26260052732ad30Tom Cherry 2999605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes // If we crash > 4 times in 4 minutes, reboot into recovery. 3009605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes boot_clock::time_point now = boot_clock::now(); 301bac3299720623f4226bca103b26260052732ad30Tom Cherry if ((flags_ & SVC_CRITICAL) && !(flags_ & SVC_RESTART)) { 3029605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes if (now < time_crashed_ + 4min) { 3039605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes if (++crash_count_ > 4) { 304331cf2fb7c16b5b25064f8d2f00284105a9b413fElliott Hughes LOG(ERROR) << "critical process '" << name_ << "' exited 4 times in 4 minutes"; 305331cf2fb7c16b5b25064f8d2f00284105a9b413fElliott Hughes panic(); 306bac3299720623f4226bca103b26260052732ad30Tom Cherry } 307bac3299720623f4226bca103b26260052732ad30Tom Cherry } else { 308bac3299720623f4226bca103b26260052732ad30Tom Cherry time_crashed_ = now; 3099605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes crash_count_ = 1; 310bac3299720623f4226bca103b26260052732ad30Tom Cherry } 311bac3299720623f4226bca103b26260052732ad30Tom Cherry } 312bac3299720623f4226bca103b26260052732ad30Tom Cherry 313bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ &= (~SVC_RESTART); 314bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ |= SVC_RESTARTING; 315bac3299720623f4226bca103b26260052732ad30Tom Cherry 316bac3299720623f4226bca103b26260052732ad30Tom Cherry // Execute all onrestart commands for this service. 317bac3299720623f4226bca103b26260052732ad30Tom Cherry onrestart_.ExecuteAllCommands(); 318bac3299720623f4226bca103b26260052732ad30Tom Cherry 319bac3299720623f4226bca103b26260052732ad30Tom Cherry NotifyStateChange("restarting"); 320b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry return; 321bac3299720623f4226bca103b26260052732ad30Tom Cherry} 322bac3299720623f4226bca103b26260052732ad30Tom Cherry 323bac3299720623f4226bca103b26260052732ad30Tom Cherryvoid Service::DumpState() const { 324f86b5a6b90619e02d1d034ef7b0adc3b439f4abbElliott Hughes LOG(INFO) << "service " << name_; 32581f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry LOG(INFO) << " class '" << Join(classnames_, " ") << "'"; 32681f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry LOG(INFO) << " exec " << Join(args_, " "); 32762767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn std::for_each(descriptors_.begin(), descriptors_.end(), 32862767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn [] (const auto& info) { LOG(INFO) << *info; }); 329bac3299720623f4226bca103b26260052732ad30Tom Cherry} 330bac3299720623f4226bca103b26260052732ad30Tom Cherry 33124b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obesbool Service::ParseCapabilities(const std::vector<std::string>& args, std::string* err) { 33224b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes capabilities_ = 0; 33324b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes 334f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes if (!CapAmbientSupported()) { 335f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes *err = "capabilities requested but the kernel does not support ambient capabilities"; 336f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes return false; 337f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes } 338f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes 339f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes unsigned int last_valid_cap = GetLastValidCap(); 340f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes if (last_valid_cap >= capabilities_.size()) { 341f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes LOG(WARNING) << "last valid run-time capability is larger than CAP_LAST_CAP"; 342f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes } 343f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes 34424b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes for (size_t i = 1; i < args.size(); i++) { 34524b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes const std::string& arg = args[i]; 346f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes int res = LookupCap(arg); 347f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes if (res < 0) { 34824b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes *err = StringPrintf("invalid capability '%s'", arg.c_str()); 34924b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes return false; 35024b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes } 351f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes unsigned int cap = static_cast<unsigned int>(res); // |res| is >= 0. 352f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes if (cap > last_valid_cap) { 353f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes *err = StringPrintf("capability '%s' not supported by the kernel", arg.c_str()); 354f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes return false; 355f3f824ee42892fb69cb0d9b0557cd9c5aed357d2Jorge Lucangeli Obes } 35624b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes capabilities_[cap] = true; 35724b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes } 35824b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes return true; 35924b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes} 36024b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes 361177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseClass(const std::vector<std::string>& args, std::string* err) { 362641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang classnames_ = std::set<std::string>(args.begin() + 1, args.end()); 363b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 364b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 365bac3299720623f4226bca103b26260052732ad30Tom Cherry 366177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseConsole(const std::vector<std::string>& args, std::string* err) { 367b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry flags_ |= SVC_CONSOLE; 36870daa67062c016eea1a30be2e1de0dcba1d23a13Viorel Suman console_ = args.size() > 1 ? "/dev/" + args[1] : ""; 369b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 370b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 371bac3299720623f4226bca103b26260052732ad30Tom Cherry 372177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseCritical(const std::vector<std::string>& args, std::string* err) { 373b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry flags_ |= SVC_CRITICAL; 374b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 375b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 376bac3299720623f4226bca103b26260052732ad30Tom Cherry 377177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseDisabled(const std::vector<std::string>& args, std::string* err) { 378b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry flags_ |= SVC_DISABLED; 379b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry flags_ |= SVC_RC_DISABLED; 380b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 381b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 382bac3299720623f4226bca103b26260052732ad30Tom Cherry 383177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseGroup(const std::vector<std::string>& args, std::string* err) { 384517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry std::string decode_uid_err; 385517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry if (!DecodeUid(args[1], &gid_, &decode_uid_err)) { 386517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry *err = "Unable to find GID for '" + args[1] + "': " + decode_uid_err; 387517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry return false; 388517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry } 389b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry for (std::size_t n = 2; n < args.size(); n++) { 390517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry gid_t gid; 391517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry if (!DecodeUid(args[n], &gid, &decode_uid_err)) { 392517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry *err = "Unable to find GID for '" + args[n] + "': " + decode_uid_err; 393517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry return false; 394517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry } 395517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry supp_gids_.emplace_back(gid); 396b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry } 397b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 398b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 399bac3299720623f4226bca103b26260052732ad30Tom Cherry 400177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParsePriority(const std::vector<std::string>& args, std::string* err) { 401da46b392f10d2809b5696632f67485f272ec5698Elliott Hughes priority_ = 0; 402da46b392f10d2809b5696632f67485f272ec5698Elliott Hughes if (!ParseInt(args[1], &priority_, 403dd34ca45ea80c519a579578562e73d5e7f86703bKeun-young Park static_cast<int>(ANDROID_PRIORITY_HIGHEST), // highest is negative 404dd34ca45ea80c519a579578562e73d5e7f86703bKeun-young Park static_cast<int>(ANDROID_PRIORITY_LOWEST))) { 405081705c258efbe938d71c2022528d809fa6d42c5Vitalii Tomkiv *err = StringPrintf("process priority value must be range %d - %d", 406081705c258efbe938d71c2022528d809fa6d42c5Vitalii Tomkiv ANDROID_PRIORITY_HIGHEST, ANDROID_PRIORITY_LOWEST); 407081705c258efbe938d71c2022528d809fa6d42c5Vitalii Tomkiv return false; 408081705c258efbe938d71c2022528d809fa6d42c5Vitalii Tomkiv } 409081705c258efbe938d71c2022528d809fa6d42c5Vitalii Tomkiv return true; 410081705c258efbe938d71c2022528d809fa6d42c5Vitalii Tomkiv} 411081705c258efbe938d71c2022528d809fa6d42c5Vitalii Tomkiv 412177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseIoprio(const std::vector<std::string>& args, std::string* err) { 413da46b392f10d2809b5696632f67485f272ec5698Elliott Hughes if (!ParseInt(args[2], &ioprio_pri_, 0, 7)) { 414b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry *err = "priority value must be range 0 - 7"; 415b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return false; 416b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry } 417b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 418b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry if (args[1] == "rt") { 419b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry ioprio_class_ = IoSchedClass_RT; 420b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry } else if (args[1] == "be") { 421b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry ioprio_class_ = IoSchedClass_BE; 422b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry } else if (args[1] == "idle") { 423b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry ioprio_class_ = IoSchedClass_IDLE; 424b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry } else { 425b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry *err = "ioprio option usage: ioprio <rt|be|idle> <0-7>"; 426b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return false; 427bac3299720623f4226bca103b26260052732ad30Tom Cherry } 428bac3299720623f4226bca103b26260052732ad30Tom Cherry 429b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 430b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 431bac3299720623f4226bca103b26260052732ad30Tom Cherry 432177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseKeycodes(const std::vector<std::string>& args, std::string* err) { 433b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry for (std::size_t i = 1; i < args.size(); i++) { 434da46b392f10d2809b5696632f67485f272ec5698Elliott Hughes int code; 435da46b392f10d2809b5696632f67485f272ec5698Elliott Hughes if (ParseInt(args[i], &code)) { 436da46b392f10d2809b5696632f67485f272ec5698Elliott Hughes keycodes_.emplace_back(code); 437da46b392f10d2809b5696632f67485f272ec5698Elliott Hughes } else { 438da46b392f10d2809b5696632f67485f272ec5698Elliott Hughes LOG(WARNING) << "ignoring invalid keycode: " << args[i]; 439da46b392f10d2809b5696632f67485f272ec5698Elliott Hughes } 440bac3299720623f4226bca103b26260052732ad30Tom Cherry } 441b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 442b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 443b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 444177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseOneshot(const std::vector<std::string>& args, std::string* err) { 445b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry flags_ |= SVC_ONESHOT; 446b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 447b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 448b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 449177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseOnrestart(const std::vector<std::string>& args, std::string* err) { 450b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry std::vector<std::string> str_args(args.begin() + 1, args.end()); 451012c573e267b8dd70de14237cb470bd7301ee8eaTom Cherry int line = onrestart_.NumCommands() + 1; 452012c573e267b8dd70de14237cb470bd7301ee8eaTom Cherry onrestart_.AddCommand(str_args, line, err); 453b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 454b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 455bac3299720623f4226bca103b26260052732ad30Tom Cherry 456177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseNamespace(const std::vector<std::string>& args, std::string* err) { 4571b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes for (size_t i = 1; i < args.size(); i++) { 4581b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes if (args[i] == "pid") { 4591b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes namespace_flags_ |= CLONE_NEWPID; 4601b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes // PID namespaces require mount namespaces. 4611b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes namespace_flags_ |= CLONE_NEWNS; 4621b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } else if (args[i] == "mnt") { 4631b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes namespace_flags_ |= CLONE_NEWNS; 4641b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } else { 4651b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes *err = "namespace must be 'pid' or 'mnt'"; 4661b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes return false; 4671b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } 4681b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } 4691b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes return true; 4701b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes} 4711b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes 472310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissenbool Service::ParseOomScoreAdjust(const std::vector<std::string>& args, std::string* err) { 473da46b392f10d2809b5696632f67485f272ec5698Elliott Hughes if (!ParseInt(args[1], &oom_score_adjust_, -1000, 1000)) { 474310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen *err = "oom_score_adjust value must be in range -1000 - +1000"; 475310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen return false; 476310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen } 477310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen return true; 478310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen} 479310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen 480d485226951013247dbb502fef7edbc6cf83acb0fRobert Beneabool Service::ParseMemcgSwappiness(const std::vector<std::string>& args, std::string* err) { 481d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea if (!ParseInt(args[1], &swappiness_, 0)) { 482d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea *err = "swappiness value must be equal or greater than 0"; 483d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea return false; 484d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea } 485d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea return true; 486d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea} 487d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea 488d485226951013247dbb502fef7edbc6cf83acb0fRobert Beneabool Service::ParseMemcgLimitInBytes(const std::vector<std::string>& args, std::string* err) { 489d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea if (!ParseInt(args[1], &limit_in_bytes_, 0)) { 490d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea *err = "limit_in_bytes value must be equal or greater than 0"; 491d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea return false; 492d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea } 493d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea return true; 494d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea} 495d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea 496d485226951013247dbb502fef7edbc6cf83acb0fRobert Beneabool Service::ParseMemcgSoftLimitInBytes(const std::vector<std::string>& args, std::string* err) { 497d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea if (!ParseInt(args[1], &soft_limit_in_bytes_, 0)) { 498d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea *err = "soft_limit_in_bytes value must be equal or greater than 0"; 499d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea return false; 500d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea } 501d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea return true; 502d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea} 503d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea 504177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseSeclabel(const std::vector<std::string>& args, std::string* err) { 505b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry seclabel_ = args[1]; 506b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 507b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 508b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 509177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseSetenv(const std::vector<std::string>& args, std::string* err) { 510b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry envvars_.emplace_back(args[1], args[2]); 511b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 512b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 513b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 514cccb34fce8798f2a359c6eb0e394ef9460673629Keun-young Parkbool Service::ParseShutdown(const std::vector<std::string>& args, std::string* err) { 515cccb34fce8798f2a359c6eb0e394ef9460673629Keun-young Park if (args[1] == "critical") { 516cccb34fce8798f2a359c6eb0e394ef9460673629Keun-young Park flags_ |= SVC_SHUTDOWN_CRITICAL; 517cccb34fce8798f2a359c6eb0e394ef9460673629Keun-young Park return true; 518cccb34fce8798f2a359c6eb0e394ef9460673629Keun-young Park } 519cccb34fce8798f2a359c6eb0e394ef9460673629Keun-young Park return false; 520cccb34fce8798f2a359c6eb0e394ef9460673629Keun-young Park} 521cccb34fce8798f2a359c6eb0e394ef9460673629Keun-young Park 52262767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyntemplate <typename T> 52362767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzynbool Service::AddDescriptor(const std::vector<std::string>& args, std::string* err) { 52462767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn int perm = args.size() > 3 ? std::strtoul(args[3].c_str(), 0, 8) : -1; 525517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry uid_t uid = 0; 526517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry gid_t gid = 0; 52762767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn std::string context = args.size() > 6 ? args[6] : ""; 52862767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn 529517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry std::string decode_uid_err; 530517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry if (args.size() > 4) { 531517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry if (!DecodeUid(args[4], &uid, &decode_uid_err)) { 532517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry *err = "Unable to find UID for '" + args[4] + "': " + decode_uid_err; 533517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry return false; 534517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry } 535517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry } 536517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry 537517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry if (args.size() > 5) { 538517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry if (!DecodeUid(args[5], &gid, &decode_uid_err)) { 539517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry *err = "Unable to find GID for '" + args[5] + "': " + decode_uid_err; 540517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry return false; 541517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry } 542517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry } 543517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry 54462767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn auto descriptor = std::make_unique<T>(args[1], args[2], uid, gid, perm, context); 54562767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn 54662767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn auto old = 54762767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn std::find_if(descriptors_.begin(), descriptors_.end(), 54862767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn [&descriptor] (const auto& other) { return descriptor.get() == other.get(); }); 54962767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn 55062767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn if (old != descriptors_.end()) { 55162767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn *err = "duplicate descriptor " + args[1] + " " + args[2]; 55262767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn return false; 55362767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn } 55462767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn 55562767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn descriptors_.emplace_back(std::move(descriptor)); 55662767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn return true; 55762767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn} 55862767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn 55962767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn// name type perm [ uid gid context ] 560177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseSocket(const std::vector<std::string>& args, std::string* err) { 56181f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry if (!StartsWith(args[2], "dgram") && !StartsWith(args[2], "stream") && 56281f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry !StartsWith(args[2], "seqpacket")) { 563b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry *err = "socket type must be 'dgram', 'stream' or 'seqpacket'"; 564bac3299720623f4226bca103b26260052732ad30Tom Cherry return false; 565bac3299720623f4226bca103b26260052732ad30Tom Cherry } 56662767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn return AddDescriptor<SocketInfo>(args, err); 56762767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn} 568b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 56962767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn// name type perm [ uid gid context ] 57062767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzynbool Service::ParseFile(const std::vector<std::string>& args, std::string* err) { 57162767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn if (args[2] != "r" && args[2] != "w" && args[2] != "rw") { 57262767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn *err = "file type must be 'r', 'w' or 'rw'"; 57362767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn return false; 57462767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn } 57562767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn if ((args[1][0] != '/') || (args[1].find("../") != std::string::npos)) { 57662767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn *err = "file name must not be relative"; 57762767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn return false; 57862767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn } 57962767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn return AddDescriptor<FileInfo>(args, err); 580bac3299720623f4226bca103b26260052732ad30Tom Cherry} 581bac3299720623f4226bca103b26260052732ad30Tom Cherry 582177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseUser(const std::vector<std::string>& args, std::string* err) { 583517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry std::string decode_uid_err; 584517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry if (!DecodeUid(args[1], &uid_, &decode_uid_err)) { 585517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry *err = "Unable to find UID for '" + args[1] + "': " + decode_uid_err; 586517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry return false; 587517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry } 588b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 589b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 590b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 591177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseWritepid(const std::vector<std::string>& args, std::string* err) { 592b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry writepid_files_.assign(args.begin() + 1, args.end()); 593b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 594b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 595b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 596177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesclass Service::OptionParserMap : public KeywordMap<OptionParser> { 597ad54d094cde838d19034b8325705b22dca0e9af0Tom Cherry public: 598ad54d094cde838d19034b8325705b22dca0e9af0Tom Cherry OptionParserMap() {} 599ad54d094cde838d19034b8325705b22dca0e9af0Tom Cherry 600ad54d094cde838d19034b8325705b22dca0e9af0Tom Cherry private: 601ad54d094cde838d19034b8325705b22dca0e9af0Tom Cherry const Map& map() const override; 602b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry}; 603b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 604ad54d094cde838d19034b8325705b22dca0e9af0Tom Cherryconst Service::OptionParserMap::Map& Service::OptionParserMap::map() const { 605b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry constexpr std::size_t kMax = std::numeric_limits<std::size_t>::max(); 606641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang // clang-format off 607177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes static const Map option_parsers = { 60824b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes {"capabilities", 60924b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes {1, kMax, &Service::ParseCapabilities}}, 610641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang {"class", {1, kMax, &Service::ParseClass}}, 611177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"console", {0, 1, &Service::ParseConsole}}, 612177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"critical", {0, 0, &Service::ParseCritical}}, 613177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"disabled", {0, 0, &Service::ParseDisabled}}, 614177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"group", {1, NR_SVC_SUPP_GIDS + 1, &Service::ParseGroup}}, 615177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"ioprio", {2, 2, &Service::ParseIoprio}}, 616177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"priority", {1, 1, &Service::ParsePriority}}, 617177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"keycodes", {1, kMax, &Service::ParseKeycodes}}, 618177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"oneshot", {0, 0, &Service::ParseOneshot}}, 619177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"onrestart", {1, kMax, &Service::ParseOnrestart}}, 620310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen {"oom_score_adjust", 621310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen {1, 1, &Service::ParseOomScoreAdjust}}, 622d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea {"memcg.swappiness", 623d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea {1, 1, &Service::ParseMemcgSwappiness}}, 624d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea {"memcg.soft_limit_in_bytes", 625d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea {1, 1, &Service::ParseMemcgSoftLimitInBytes}}, 626d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea {"memcg.limit_in_bytes", 627d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea {1, 1, &Service::ParseMemcgLimitInBytes}}, 628177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"namespace", {1, 2, &Service::ParseNamespace}}, 629177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"seclabel", {1, 1, &Service::ParseSeclabel}}, 630177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"setenv", {2, 2, &Service::ParseSetenv}}, 631cccb34fce8798f2a359c6eb0e394ef9460673629Keun-young Park {"shutdown", {1, 1, &Service::ParseShutdown}}, 632177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"socket", {3, 6, &Service::ParseSocket}}, 633978fd0ea254f11f84e38b41a74bbe70c81edc197Mark Salyzyn {"file", {2, 2, &Service::ParseFile}}, 634177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"user", {1, 1, &Service::ParseUser}}, 635177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes {"writepid", {1, kMax, &Service::ParseWritepid}}, 636b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry }; 637641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang // clang-format on 638177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes return option_parsers; 639b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 640b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 641177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obesbool Service::ParseLine(const std::vector<std::string>& args, std::string* err) { 642177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes static const OptionParserMap parser_map; 643fe062055cb11fcb1a6178b046173fc0361ad5b96Tom Cherry auto parser = parser_map.FindFunction(args, err); 644b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 645177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes if (!parser) { 646b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return false; 647b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry } 648b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 649177b27d4f5bfa498cc46aad24d9375d65630bea0Jorge Lucangeli Obes return (this->*parser)(args, err); 650b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 651b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 652ede0d538501dfc78c741fb3b0645406636d1d1fdTom Cherrybool Service::ExecStart(std::unique_ptr<android::base::Timer>* exec_waiter) { 653b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry flags_ |= SVC_EXEC | SVC_ONESHOT; 654b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry 655ede0d538501dfc78c741fb3b0645406636d1d1fdTom Cherry exec_waiter->reset(new android::base::Timer); 656b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry 657b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry if (!Start()) { 658b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry exec_waiter->reset(); 659b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry return false; 660b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry } 661b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry return true; 662b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry} 663b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry 664bdeac39a42c1b9c7195ada1c30fe12f94314490fElliott Hughesbool Service::Start() { 665bac3299720623f4226bca103b26260052732ad30Tom Cherry // Starting a service removes it from the disabled or reset state and 666bac3299720623f4226bca103b26260052732ad30Tom Cherry // immediately takes it out of the restarting state if it was in there. 667bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ &= (~(SVC_DISABLED|SVC_RESTARTING|SVC_RESET|SVC_RESTART|SVC_DISABLED_START)); 668bac3299720623f4226bca103b26260052732ad30Tom Cherry 669bac3299720623f4226bca103b26260052732ad30Tom Cherry // Running processes require no additional work --- if they're in the 670bac3299720623f4226bca103b26260052732ad30Tom Cherry // process of exiting, we've ensured that they will immediately restart 671bac3299720623f4226bca103b26260052732ad30Tom Cherry // on exit, unless they are ONESHOT. 672bac3299720623f4226bca103b26260052732ad30Tom Cherry if (flags_ & SVC_RUNNING) { 673bac3299720623f4226bca103b26260052732ad30Tom Cherry return false; 674bac3299720623f4226bca103b26260052732ad30Tom Cherry } 675bac3299720623f4226bca103b26260052732ad30Tom Cherry 676bac3299720623f4226bca103b26260052732ad30Tom Cherry bool needs_console = (flags_ & SVC_CONSOLE); 67770daa67062c016eea1a30be2e1de0dcba1d23a13Viorel Suman if (needs_console) { 67870daa67062c016eea1a30be2e1de0dcba1d23a13Viorel Suman if (console_.empty()) { 67970daa67062c016eea1a30be2e1de0dcba1d23a13Viorel Suman console_ = default_console; 68070daa67062c016eea1a30be2e1de0dcba1d23a13Viorel Suman } 68170daa67062c016eea1a30be2e1de0dcba1d23a13Viorel Suman 68224ef8601c203a28b85a5e9168e5bada713e6a8f1Adrian Salido // Make sure that open call succeeds to ensure a console driver is 68324ef8601c203a28b85a5e9168e5bada713e6a8f1Adrian Salido // properly registered for the device node 68424ef8601c203a28b85a5e9168e5bada713e6a8f1Adrian Salido int console_fd = open(console_.c_str(), O_RDWR | O_CLOEXEC); 68524ef8601c203a28b85a5e9168e5bada713e6a8f1Adrian Salido if (console_fd < 0) { 68624ef8601c203a28b85a5e9168e5bada713e6a8f1Adrian Salido PLOG(ERROR) << "service '" << name_ << "' couldn't open console '" << console_ << "'"; 68770daa67062c016eea1a30be2e1de0dcba1d23a13Viorel Suman flags_ |= SVC_DISABLED; 68870daa67062c016eea1a30be2e1de0dcba1d23a13Viorel Suman return false; 68970daa67062c016eea1a30be2e1de0dcba1d23a13Viorel Suman } 69024ef8601c203a28b85a5e9168e5bada713e6a8f1Adrian Salido close(console_fd); 691bac3299720623f4226bca103b26260052732ad30Tom Cherry } 692bac3299720623f4226bca103b26260052732ad30Tom Cherry 693bac3299720623f4226bca103b26260052732ad30Tom Cherry struct stat sb; 694bac3299720623f4226bca103b26260052732ad30Tom Cherry if (stat(args_[0].c_str(), &sb) == -1) { 695f86b5a6b90619e02d1d034ef7b0adc3b439f4abbElliott Hughes PLOG(ERROR) << "cannot find '" << args_[0] << "', disabling '" << name_ << "'"; 696bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ |= SVC_DISABLED; 697bac3299720623f4226bca103b26260052732ad30Tom Cherry return false; 698bac3299720623f4226bca103b26260052732ad30Tom Cherry } 699bac3299720623f4226bca103b26260052732ad30Tom Cherry 700bac3299720623f4226bca103b26260052732ad30Tom Cherry std::string scon; 701bac3299720623f4226bca103b26260052732ad30Tom Cherry if (!seclabel_.empty()) { 702bac3299720623f4226bca103b26260052732ad30Tom Cherry scon = seclabel_; 703bac3299720623f4226bca103b26260052732ad30Tom Cherry } else { 704344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes scon = ComputeContextFromExecutable(name_, args_[0]); 705344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes if (scon == "") { 706bac3299720623f4226bca103b26260052732ad30Tom Cherry return false; 707bac3299720623f4226bca103b26260052732ad30Tom Cherry } 708bac3299720623f4226bca103b26260052732ad30Tom Cherry } 709bac3299720623f4226bca103b26260052732ad30Tom Cherry 710a285dac9c0b95a0351dfa98dde60c74a1445dc57Wei Wang LOG(INFO) << "starting service '" << name_ << "'..."; 711bac3299720623f4226bca103b26260052732ad30Tom Cherry 7121b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes pid_t pid = -1; 7131b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes if (namespace_flags_) { 714344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes pid = clone(nullptr, nullptr, namespace_flags_ | SIGCHLD, nullptr); 7151b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } else { 7161b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes pid = fork(); 7171b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } 7181b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes 719bac3299720623f4226bca103b26260052732ad30Tom Cherry if (pid == 0) { 720bac3299720623f4226bca103b26260052732ad30Tom Cherry umask(077); 721bac3299720623f4226bca103b26260052732ad30Tom Cherry 7221b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes if (namespace_flags_ & CLONE_NEWPID) { 7231b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes // This will fork again to run an init process inside the PID 7241b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes // namespace. 7251b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes SetUpPidNamespace(name_); 7261b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes } 7271b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes 728bac3299720623f4226bca103b26260052732ad30Tom Cherry for (const auto& ei : envvars_) { 729bac3299720623f4226bca103b26260052732ad30Tom Cherry add_environment(ei.name.c_str(), ei.value.c_str()); 730bac3299720623f4226bca103b26260052732ad30Tom Cherry } 731bac3299720623f4226bca103b26260052732ad30Tom Cherry 73262767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn std::for_each(descriptors_.begin(), descriptors_.end(), 73362767fe29f8aaf62470781a3cf419ba11187d178Mark Salyzyn std::bind(&DescriptorInfo::CreateAndPublish, std::placeholders::_1, scon)); 734bac3299720623f4226bca103b26260052732ad30Tom Cherry 7350828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko // See if there were "writepid" instructions to write to files under /dev/cpuset/. 7360828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko auto cpuset_predicate = [](const std::string& path) { 73781f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry return StartsWith(path, "/dev/cpuset/"); 7380828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko }; 7390828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko auto iter = std::find_if(writepid_files_.begin(), writepid_files_.end(), cpuset_predicate); 7400828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko if (iter == writepid_files_.end()) { 7410828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko // There were no "writepid" instructions for cpusets, check if the system default 7420828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko // cpuset is specified to be used for the process. 74381f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry std::string default_cpuset = GetProperty("ro.cpuset.default", ""); 7440828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko if (!default_cpuset.empty()) { 7450828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko // Make sure the cpuset name starts and ends with '/'. 7460828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko // A single '/' means the 'root' cpuset. 7470828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko if (default_cpuset.front() != '/') { 7480828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko default_cpuset.insert(0, 1, '/'); 7490828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko } 7500828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko if (default_cpuset.back() != '/') { 7510828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko default_cpuset.push_back('/'); 7520828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko } 7530828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko writepid_files_.push_back( 7540828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko StringPrintf("/dev/cpuset%stasks", default_cpuset.c_str())); 7550828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko } 7560828676dff618c8ecc9852cb319543570b3a2e72Alex Vakulenko } 7571c3a53f03ca3c2c647f83cd8b8ae7e18c5c7bc69Tom Cherry std::string pid_str = std::to_string(getpid()); 758bac3299720623f4226bca103b26260052732ad30Tom Cherry for (const auto& file : writepid_files_) { 759b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry if (!WriteStringToFile(pid_str, file)) { 760f86b5a6b90619e02d1d034ef7b0adc3b439f4abbElliott Hughes PLOG(ERROR) << "couldn't write " << pid_str << " to " << file; 761bac3299720623f4226bca103b26260052732ad30Tom Cherry } 762bac3299720623f4226bca103b26260052732ad30Tom Cherry } 763bac3299720623f4226bca103b26260052732ad30Tom Cherry 764bac3299720623f4226bca103b26260052732ad30Tom Cherry if (ioprio_class_ != IoSchedClass_NONE) { 765bac3299720623f4226bca103b26260052732ad30Tom Cherry if (android_set_ioprio(getpid(), ioprio_class_, ioprio_pri_)) { 766344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes PLOG(ERROR) << "failed to set pid " << getpid() 767f86b5a6b90619e02d1d034ef7b0adc3b439f4abbElliott Hughes << " ioprio=" << ioprio_class_ << "," << ioprio_pri_; 768bac3299720623f4226bca103b26260052732ad30Tom Cherry } 769bac3299720623f4226bca103b26260052732ad30Tom Cherry } 770bac3299720623f4226bca103b26260052732ad30Tom Cherry 771bac3299720623f4226bca103b26260052732ad30Tom Cherry if (needs_console) { 772bac3299720623f4226bca103b26260052732ad30Tom Cherry setsid(); 773bac3299720623f4226bca103b26260052732ad30Tom Cherry OpenConsole(); 774bac3299720623f4226bca103b26260052732ad30Tom Cherry } else { 775bac3299720623f4226bca103b26260052732ad30Tom Cherry ZapStdio(); 776bac3299720623f4226bca103b26260052732ad30Tom Cherry } 777bac3299720623f4226bca103b26260052732ad30Tom Cherry 778344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes // As requested, set our gid, supplemental gids, uid, context, and 779344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes // priority. Aborts on failure. 780344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes SetProcessAttributes(); 781bac3299720623f4226bca103b26260052732ad30Tom Cherry 782bac3299720623f4226bca103b26260052732ad30Tom Cherry std::vector<char*> strs; 783344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes ExpandArgs(args_, &strs); 784bac3536cc949b3af0185e7b8f9b18318b37ac8b0Tom Cherry if (execve(strs[0], (char**) &strs[0], (char**) ENV) < 0) { 785f86b5a6b90619e02d1d034ef7b0adc3b439f4abbElliott Hughes PLOG(ERROR) << "cannot execve('" << strs[0] << "')"; 786bac3299720623f4226bca103b26260052732ad30Tom Cherry } 787bac3299720623f4226bca103b26260052732ad30Tom Cherry 788bac3299720623f4226bca103b26260052732ad30Tom Cherry _exit(127); 789bac3299720623f4226bca103b26260052732ad30Tom Cherry } 790bac3299720623f4226bca103b26260052732ad30Tom Cherry 791bac3299720623f4226bca103b26260052732ad30Tom Cherry if (pid < 0) { 792f86b5a6b90619e02d1d034ef7b0adc3b439f4abbElliott Hughes PLOG(ERROR) << "failed to fork for '" << name_ << "'"; 793bac3299720623f4226bca103b26260052732ad30Tom Cherry pid_ = 0; 794bac3299720623f4226bca103b26260052732ad30Tom Cherry return false; 795bac3299720623f4226bca103b26260052732ad30Tom Cherry } 796bac3299720623f4226bca103b26260052732ad30Tom Cherry 797310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen if (oom_score_adjust_ != -1000) { 7981c3a53f03ca3c2c647f83cd8b8ae7e18c5c7bc69Tom Cherry std::string oom_str = std::to_string(oom_score_adjust_); 799310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen std::string oom_file = StringPrintf("/proc/%d/oom_score_adj", pid); 800310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen if (!WriteStringToFile(oom_str, oom_file)) { 801310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen PLOG(ERROR) << "couldn't write oom_score_adj: " << strerror(errno); 802310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen } 803310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen } 804310f6704d0dfe9ca4cfb153ce8e4212cc7596190Marco Nelissen 8059605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes time_started_ = boot_clock::now(); 806bac3299720623f4226bca103b26260052732ad30Tom Cherry pid_ = pid; 807bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ |= SVC_RUNNING; 80833838b1156cd1a5f8eec94756bee9e030a3dd2ebTom Cherry process_cgroup_empty_ = false; 809ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes 810ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes errno = -createProcessGroup(uid_, pid_); 811ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes if (errno != 0) { 812344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes PLOG(ERROR) << "createProcessGroup(" << uid_ << ", " << pid_ << ") failed for service '" 813344d01f99f6049565e4342b4c4202bd9ab96340bJorge Lucangeli Obes << name_ << "'"; 814d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea } else { 815d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea if (swappiness_ != -1) { 816d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea if (!setProcessGroupSwappiness(uid_, pid_, swappiness_)) { 817d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea PLOG(ERROR) << "setProcessGroupSwappiness failed"; 818d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea } 819d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea } 820d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea 821d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea if (soft_limit_in_bytes_ != -1) { 822d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea if (!setProcessGroupSoftLimit(uid_, pid_, soft_limit_in_bytes_)) { 823d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea PLOG(ERROR) << "setProcessGroupSoftLimit failed"; 824d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea } 825d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea } 826d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea 827d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea if (limit_in_bytes_ != -1) { 828d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea if (!setProcessGroupLimit(uid_, pid_, limit_in_bytes_)) { 829d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea PLOG(ERROR) << "setProcessGroupLimit failed"; 830d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea } 831d485226951013247dbb502fef7edbc6cf83acb0fRobert Benea } 832ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes } 833bac3299720623f4226bca103b26260052732ad30Tom Cherry 834bac3299720623f4226bca103b26260052732ad30Tom Cherry if ((flags_ & SVC_EXEC) != 0) { 8351c3a53f03ca3c2c647f83cd8b8ae7e18c5c7bc69Tom Cherry LOG(INFO) << "SVC_EXEC pid " << pid_ << " (uid " << uid_ << " gid " << gid_ << "+" 8361c3a53f03ca3c2c647f83cd8b8ae7e18c5c7bc69Tom Cherry << supp_gids_.size() << " context " 8371c3a53f03ca3c2c647f83cd8b8ae7e18c5c7bc69Tom Cherry << (!seclabel_.empty() ? seclabel_ : "default") << ") started; waiting..."; 838bac3299720623f4226bca103b26260052732ad30Tom Cherry } 839bac3299720623f4226bca103b26260052732ad30Tom Cherry 840bac3299720623f4226bca103b26260052732ad30Tom Cherry NotifyStateChange("running"); 841bac3299720623f4226bca103b26260052732ad30Tom Cherry return true; 842bac3299720623f4226bca103b26260052732ad30Tom Cherry} 843bac3299720623f4226bca103b26260052732ad30Tom Cherry 844bac3299720623f4226bca103b26260052732ad30Tom Cherrybool Service::StartIfNotDisabled() { 845bac3299720623f4226bca103b26260052732ad30Tom Cherry if (!(flags_ & SVC_DISABLED)) { 846bac3299720623f4226bca103b26260052732ad30Tom Cherry return Start(); 847bac3299720623f4226bca103b26260052732ad30Tom Cherry } else { 848bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ |= SVC_DISABLED_START; 849bac3299720623f4226bca103b26260052732ad30Tom Cherry } 850bac3299720623f4226bca103b26260052732ad30Tom Cherry return true; 851bac3299720623f4226bca103b26260052732ad30Tom Cherry} 852bac3299720623f4226bca103b26260052732ad30Tom Cherry 853bac3299720623f4226bca103b26260052732ad30Tom Cherrybool Service::Enable() { 854bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ &= ~(SVC_DISABLED | SVC_RC_DISABLED); 855bac3299720623f4226bca103b26260052732ad30Tom Cherry if (flags_ & SVC_DISABLED_START) { 856bac3299720623f4226bca103b26260052732ad30Tom Cherry return Start(); 857bac3299720623f4226bca103b26260052732ad30Tom Cherry } 858bac3299720623f4226bca103b26260052732ad30Tom Cherry return true; 859bac3299720623f4226bca103b26260052732ad30Tom Cherry} 860bac3299720623f4226bca103b26260052732ad30Tom Cherry 861bac3299720623f4226bca103b26260052732ad30Tom Cherryvoid Service::Reset() { 862bac3299720623f4226bca103b26260052732ad30Tom Cherry StopOrReset(SVC_RESET); 863bac3299720623f4226bca103b26260052732ad30Tom Cherry} 864bac3299720623f4226bca103b26260052732ad30Tom Cherry 865bac3299720623f4226bca103b26260052732ad30Tom Cherryvoid Service::Stop() { 866bac3299720623f4226bca103b26260052732ad30Tom Cherry StopOrReset(SVC_DISABLED); 867bac3299720623f4226bca103b26260052732ad30Tom Cherry} 868bac3299720623f4226bca103b26260052732ad30Tom Cherry 869b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNETvoid Service::Terminate() { 870b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET flags_ &= ~(SVC_RESTARTING | SVC_DISABLED_START); 871b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET flags_ |= SVC_DISABLED; 872b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET if (pid_) { 873ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes KillProcessGroup(SIGTERM); 874b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET NotifyStateChange("stopping"); 875b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET } 876b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET} 877b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET 878bac3299720623f4226bca103b26260052732ad30Tom Cherryvoid Service::Restart() { 879bac3299720623f4226bca103b26260052732ad30Tom Cherry if (flags_ & SVC_RUNNING) { 880bac3299720623f4226bca103b26260052732ad30Tom Cherry /* Stop, wait, then start the service. */ 881bac3299720623f4226bca103b26260052732ad30Tom Cherry StopOrReset(SVC_RESTART); 882bac3299720623f4226bca103b26260052732ad30Tom Cherry } else if (!(flags_ & SVC_RESTARTING)) { 883bac3299720623f4226bca103b26260052732ad30Tom Cherry /* Just start the service since it's not running. */ 884bac3299720623f4226bca103b26260052732ad30Tom Cherry Start(); 885bac3299720623f4226bca103b26260052732ad30Tom Cherry } /* else: Service is restarting anyways. */ 886bac3299720623f4226bca103b26260052732ad30Tom Cherry} 887bac3299720623f4226bca103b26260052732ad30Tom Cherry 8889605a945f7a497c0307b512b9cd762f2d23973caElliott Hughesvoid Service::RestartIfNeeded(time_t* process_needs_restart_at) { 8899605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes boot_clock::time_point now = boot_clock::now(); 8909605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes boot_clock::time_point next_start = time_started_ + 5s; 8919605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes if (now > next_start) { 892bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ &= (~SVC_RESTARTING); 893bac3299720623f4226bca103b26260052732ad30Tom Cherry Start(); 894bac3299720623f4226bca103b26260052732ad30Tom Cherry return; 895bac3299720623f4226bca103b26260052732ad30Tom Cherry } 896bac3299720623f4226bca103b26260052732ad30Tom Cherry 8979605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes time_t next_start_time_t = time(nullptr) + 8989605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes time_t(std::chrono::duration_cast<std::chrono::seconds>(next_start - now).count()); 8999605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes if (next_start_time_t < *process_needs_restart_at || *process_needs_restart_at == 0) { 9009605a945f7a497c0307b512b9cd762f2d23973caElliott Hughes *process_needs_restart_at = next_start_time_t; 901bac3299720623f4226bca103b26260052732ad30Tom Cherry } 902bac3299720623f4226bca103b26260052732ad30Tom Cherry} 903bac3299720623f4226bca103b26260052732ad30Tom Cherry 904ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes// The how field should be either SVC_DISABLED, SVC_RESET, or SVC_RESTART. 905bac3299720623f4226bca103b26260052732ad30Tom Cherryvoid Service::StopOrReset(int how) { 906ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes // The service is still SVC_RUNNING until its process exits, but if it has 907ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes // already exited it shoudn't attempt a restart yet. 908bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ &= ~(SVC_RESTARTING | SVC_DISABLED_START); 909bac3299720623f4226bca103b26260052732ad30Tom Cherry 910bac3299720623f4226bca103b26260052732ad30Tom Cherry if ((how != SVC_DISABLED) && (how != SVC_RESET) && (how != SVC_RESTART)) { 911ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes // An illegal flag: default to SVC_DISABLED. 912bac3299720623f4226bca103b26260052732ad30Tom Cherry how = SVC_DISABLED; 913bac3299720623f4226bca103b26260052732ad30Tom Cherry } 914ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes 915ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes // If the service has not yet started, prevent it from auto-starting with its class. 916bac3299720623f4226bca103b26260052732ad30Tom Cherry if (how == SVC_RESET) { 917bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ |= (flags_ & SVC_RC_DISABLED) ? SVC_DISABLED : SVC_RESET; 918bac3299720623f4226bca103b26260052732ad30Tom Cherry } else { 919bac3299720623f4226bca103b26260052732ad30Tom Cherry flags_ |= how; 920bac3299720623f4226bca103b26260052732ad30Tom Cherry } 921bac3299720623f4226bca103b26260052732ad30Tom Cherry 922bac3299720623f4226bca103b26260052732ad30Tom Cherry if (pid_) { 923ad8e94e017173471e90c704eb7d8de5a14712aa7Elliott Hughes KillProcessGroup(SIGKILL); 924bac3299720623f4226bca103b26260052732ad30Tom Cherry NotifyStateChange("stopping"); 925bac3299720623f4226bca103b26260052732ad30Tom Cherry } else { 926bac3299720623f4226bca103b26260052732ad30Tom Cherry NotifyStateChange("stopped"); 927bac3299720623f4226bca103b26260052732ad30Tom Cherry } 928bac3299720623f4226bca103b26260052732ad30Tom Cherry} 929bac3299720623f4226bca103b26260052732ad30Tom Cherry 930bac3299720623f4226bca103b26260052732ad30Tom Cherryvoid Service::ZapStdio() const { 931bac3299720623f4226bca103b26260052732ad30Tom Cherry int fd; 932bac3299720623f4226bca103b26260052732ad30Tom Cherry fd = open("/dev/null", O_RDWR); 933bac3299720623f4226bca103b26260052732ad30Tom Cherry dup2(fd, 0); 934bac3299720623f4226bca103b26260052732ad30Tom Cherry dup2(fd, 1); 935bac3299720623f4226bca103b26260052732ad30Tom Cherry dup2(fd, 2); 936bac3299720623f4226bca103b26260052732ad30Tom Cherry close(fd); 937bac3299720623f4226bca103b26260052732ad30Tom Cherry} 938bac3299720623f4226bca103b26260052732ad30Tom Cherry 939bac3299720623f4226bca103b26260052732ad30Tom Cherryvoid Service::OpenConsole() const { 94070daa67062c016eea1a30be2e1de0dcba1d23a13Viorel Suman int fd = open(console_.c_str(), O_RDWR); 94170daa67062c016eea1a30be2e1de0dcba1d23a13Viorel Suman if (fd == -1) fd = open("/dev/null", O_RDWR); 942bac3299720623f4226bca103b26260052732ad30Tom Cherry ioctl(fd, TIOCSCTTY, 0); 943bac3299720623f4226bca103b26260052732ad30Tom Cherry dup2(fd, 0); 944bac3299720623f4226bca103b26260052732ad30Tom Cherry dup2(fd, 1); 945bac3299720623f4226bca103b26260052732ad30Tom Cherry dup2(fd, 2); 946bac3299720623f4226bca103b26260052732ad30Tom Cherry close(fd); 947bac3299720623f4226bca103b26260052732ad30Tom Cherry} 948bac3299720623f4226bca103b26260052732ad30Tom Cherry 949bac3299720623f4226bca103b26260052732ad30Tom Cherryint ServiceManager::exec_count_ = 0; 950bac3299720623f4226bca103b26260052732ad30Tom Cherry 951bac3299720623f4226bca103b26260052732ad30Tom CherryServiceManager::ServiceManager() { 952bac3299720623f4226bca103b26260052732ad30Tom Cherry} 953bac3299720623f4226bca103b26260052732ad30Tom Cherry 954bac3299720623f4226bca103b26260052732ad30Tom CherryServiceManager& ServiceManager::GetInstance() { 955bac3299720623f4226bca103b26260052732ad30Tom Cherry static ServiceManager instance; 956bac3299720623f4226bca103b26260052732ad30Tom Cherry return instance; 957bac3299720623f4226bca103b26260052732ad30Tom Cherry} 958bac3299720623f4226bca103b26260052732ad30Tom Cherry 959b7349902a945903f9e36a569051f5131beb0bc24Tom Cherryvoid ServiceManager::AddService(std::unique_ptr<Service> service) { 960b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry services_.emplace_back(std::move(service)); 961bac3299720623f4226bca103b26260052732ad30Tom Cherry} 962bac3299720623f4226bca103b26260052732ad30Tom Cherry 963b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherrybool ServiceManager::Exec(const std::vector<std::string>& args) { 964b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry Service* svc = MakeExecOneshotService(args); 965b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry if (!svc) { 966b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry LOG(ERROR) << "Could not create exec service"; 967b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry return false; 968b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry } 969b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry if (!svc->ExecStart(&exec_waiter_)) { 970b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry LOG(ERROR) << "Could not start exec service"; 971b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry ServiceManager::GetInstance().RemoveService(*svc); 972b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry return false; 973b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry } 974b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry return true; 975b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry} 976b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry 977b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherrybool ServiceManager::ExecStart(const std::string& name) { 978b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry Service* svc = FindServiceByName(name); 979b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry if (!svc) { 980b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry LOG(ERROR) << "ExecStart(" << name << "): Service not found"; 981b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry return false; 982b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry } 983b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry if (!svc->ExecStart(&exec_waiter_)) { 984b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry LOG(ERROR) << "ExecStart(" << name << "): Could not start Service"; 985b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry return false; 986b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry } 987b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry return true; 988b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry} 989b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry 990b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherrybool ServiceManager::IsWaitingForExec() const { return exec_waiter_ != nullptr; } 991b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry 992bac3299720623f4226bca103b26260052732ad30Tom CherryService* ServiceManager::MakeExecOneshotService(const std::vector<std::string>& args) { 993bac3299720623f4226bca103b26260052732ad30Tom Cherry // Parse the arguments: exec [SECLABEL [UID [GID]*] --] COMMAND ARGS... 994bac3299720623f4226bca103b26260052732ad30Tom Cherry // SECLABEL can be a - to denote default 995bac3299720623f4226bca103b26260052732ad30Tom Cherry std::size_t command_arg = 1; 996bac3299720623f4226bca103b26260052732ad30Tom Cherry for (std::size_t i = 1; i < args.size(); ++i) { 997bac3299720623f4226bca103b26260052732ad30Tom Cherry if (args[i] == "--") { 998bac3299720623f4226bca103b26260052732ad30Tom Cherry command_arg = i + 1; 999bac3299720623f4226bca103b26260052732ad30Tom Cherry break; 1000bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1001bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1002bac3299720623f4226bca103b26260052732ad30Tom Cherry if (command_arg > 4 + NR_SVC_SUPP_GIDS) { 1003f86b5a6b90619e02d1d034ef7b0adc3b439f4abbElliott Hughes LOG(ERROR) << "exec called with too many supplementary group ids"; 1004bac3299720623f4226bca103b26260052732ad30Tom Cherry return nullptr; 1005bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1006bac3299720623f4226bca103b26260052732ad30Tom Cherry 1007bac3299720623f4226bca103b26260052732ad30Tom Cherry if (command_arg >= args.size()) { 1008f86b5a6b90619e02d1d034ef7b0adc3b439f4abbElliott Hughes LOG(ERROR) << "exec called without command"; 1009bac3299720623f4226bca103b26260052732ad30Tom Cherry return nullptr; 1010bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1011bac3299720623f4226bca103b26260052732ad30Tom Cherry std::vector<std::string> str_args(args.begin() + command_arg, args.end()); 1012bac3299720623f4226bca103b26260052732ad30Tom Cherry 1013bac3299720623f4226bca103b26260052732ad30Tom Cherry exec_count_++; 101481f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry std::string name = "exec " + std::to_string(exec_count_) + " (" + Join(str_args, " ") + ")"; 101586e31a81238fb30d113947e7689a38daf49097f3Tom Cherry 1016b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry unsigned flags = SVC_EXEC | SVC_ONESHOT | SVC_TEMPORARY; 101724b29132a017f7fbfd009c3e6aec499d1b815dbfJorge Lucangeli Obes CapSet no_capabilities; 10181b3fa3d6506d04570aab60147b0ec743e38c8796Jorge Lucangeli Obes unsigned namespace_flags = 0; 1019bac3299720623f4226bca103b26260052732ad30Tom Cherry 1020bac3299720623f4226bca103b26260052732ad30Tom Cherry std::string seclabel = ""; 1021bac3299720623f4226bca103b26260052732ad30Tom Cherry if (command_arg > 2 && args[1] != "-") { 1022bac3299720623f4226bca103b26260052732ad30Tom Cherry seclabel = args[1]; 1023bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1024bac3299720623f4226bca103b26260052732ad30Tom Cherry uid_t uid = 0; 1025bac3299720623f4226bca103b26260052732ad30Tom Cherry if (command_arg > 3) { 1026517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry std::string decode_uid_err; 1027517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry if (!DecodeUid(args[2], &uid, &decode_uid_err)) { 1028517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry LOG(ERROR) << "Unable to find UID for '" << args[2] << "': " << decode_uid_err; 1029517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry return nullptr; 1030517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry } 1031bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1032bac3299720623f4226bca103b26260052732ad30Tom Cherry gid_t gid = 0; 1033bac3299720623f4226bca103b26260052732ad30Tom Cherry std::vector<gid_t> supp_gids; 1034bac3299720623f4226bca103b26260052732ad30Tom Cherry if (command_arg > 4) { 1035517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry std::string decode_uid_err; 1036517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry if (!DecodeUid(args[3], &gid, &decode_uid_err)) { 1037517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry LOG(ERROR) << "Unable to find GID for '" << args[3] << "': " << decode_uid_err; 1038517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry return nullptr; 1039517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry } 1040bac3299720623f4226bca103b26260052732ad30Tom Cherry std::size_t nr_supp_gids = command_arg - 1 /* -- */ - 4 /* exec SECLABEL UID GID */; 1041bac3299720623f4226bca103b26260052732ad30Tom Cherry for (size_t i = 0; i < nr_supp_gids; ++i) { 1042517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry gid_t supp_gid; 1043517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry if (!DecodeUid(args[4 + i], &supp_gid, &decode_uid_err)) { 1044517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry LOG(ERROR) << "Unable to find UID for '" << args[4 + i] << "': " << decode_uid_err; 1045517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry return nullptr; 1046517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry } 1047517e1f17cfec2143d4d10a64b1496a550acf3ea2Tom Cherry supp_gids.push_back(supp_gid); 1048bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1049bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1050bac3299720623f4226bca103b26260052732ad30Tom Cherry 1051641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang auto svc_p = std::make_unique<Service>(name, flags, uid, gid, supp_gids, no_capabilities, 1052641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang namespace_flags, seclabel, str_args); 1053bac3299720623f4226bca103b26260052732ad30Tom Cherry Service* svc = svc_p.get(); 10549bdddd713af925a4be161f65e5fd66eccc8bd1ebTom Cherry services_.emplace_back(std::move(svc_p)); 1055bac3299720623f4226bca103b26260052732ad30Tom Cherry 1056bac3299720623f4226bca103b26260052732ad30Tom Cherry return svc; 1057bac3299720623f4226bca103b26260052732ad30Tom Cherry} 1058bac3299720623f4226bca103b26260052732ad30Tom Cherry 1059bac3299720623f4226bca103b26260052732ad30Tom CherryService* ServiceManager::FindServiceByName(const std::string& name) const { 1060bac3299720623f4226bca103b26260052732ad30Tom Cherry auto svc = std::find_if(services_.begin(), services_.end(), 1061bac3299720623f4226bca103b26260052732ad30Tom Cherry [&name] (const std::unique_ptr<Service>& s) { 1062bac3299720623f4226bca103b26260052732ad30Tom Cherry return name == s->name(); 1063bac3299720623f4226bca103b26260052732ad30Tom Cherry }); 1064bac3299720623f4226bca103b26260052732ad30Tom Cherry if (svc != services_.end()) { 1065bac3299720623f4226bca103b26260052732ad30Tom Cherry return svc->get(); 1066bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1067bac3299720623f4226bca103b26260052732ad30Tom Cherry return nullptr; 1068bac3299720623f4226bca103b26260052732ad30Tom Cherry} 1069bac3299720623f4226bca103b26260052732ad30Tom Cherry 1070bac3299720623f4226bca103b26260052732ad30Tom CherryService* ServiceManager::FindServiceByPid(pid_t pid) const { 1071bac3299720623f4226bca103b26260052732ad30Tom Cherry auto svc = std::find_if(services_.begin(), services_.end(), 1072bac3299720623f4226bca103b26260052732ad30Tom Cherry [&pid] (const std::unique_ptr<Service>& s) { 1073bac3299720623f4226bca103b26260052732ad30Tom Cherry return s->pid() == pid; 1074bac3299720623f4226bca103b26260052732ad30Tom Cherry }); 1075bac3299720623f4226bca103b26260052732ad30Tom Cherry if (svc != services_.end()) { 1076bac3299720623f4226bca103b26260052732ad30Tom Cherry return svc->get(); 1077bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1078bac3299720623f4226bca103b26260052732ad30Tom Cherry return nullptr; 1079bac3299720623f4226bca103b26260052732ad30Tom Cherry} 1080bac3299720623f4226bca103b26260052732ad30Tom Cherry 1081bac3299720623f4226bca103b26260052732ad30Tom CherryService* ServiceManager::FindServiceByKeychord(int keychord_id) const { 1082bac3299720623f4226bca103b26260052732ad30Tom Cherry auto svc = std::find_if(services_.begin(), services_.end(), 1083bac3299720623f4226bca103b26260052732ad30Tom Cherry [&keychord_id] (const std::unique_ptr<Service>& s) { 1084bac3299720623f4226bca103b26260052732ad30Tom Cherry return s->keychord_id() == keychord_id; 1085bac3299720623f4226bca103b26260052732ad30Tom Cherry }); 1086bac3299720623f4226bca103b26260052732ad30Tom Cherry 1087bac3299720623f4226bca103b26260052732ad30Tom Cherry if (svc != services_.end()) { 1088bac3299720623f4226bca103b26260052732ad30Tom Cherry return svc->get(); 1089bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1090bac3299720623f4226bca103b26260052732ad30Tom Cherry return nullptr; 1091bac3299720623f4226bca103b26260052732ad30Tom Cherry} 1092bac3299720623f4226bca103b26260052732ad30Tom Cherry 10938f7b9e3d39cdae8363816aa7bcbab0d79fd70ceaChih-Hung Hsiehvoid ServiceManager::ForEachService(const std::function<void(Service*)>& callback) const { 1094bac3299720623f4226bca103b26260052732ad30Tom Cherry for (const auto& s : services_) { 1095b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET callback(s.get()); 1096bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1097bac3299720623f4226bca103b26260052732ad30Tom Cherry} 1098bac3299720623f4226bca103b26260052732ad30Tom Cherry 1099bac3299720623f4226bca103b26260052732ad30Tom Cherryvoid ServiceManager::ForEachServiceInClass(const std::string& classname, 1100bac3299720623f4226bca103b26260052732ad30Tom Cherry void (*func)(Service* svc)) const { 1101bac3299720623f4226bca103b26260052732ad30Tom Cherry for (const auto& s : services_) { 1102641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang if (s->classnames().find(classname) != s->classnames().end()) { 1103bac3299720623f4226bca103b26260052732ad30Tom Cherry func(s.get()); 1104bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1105bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1106bac3299720623f4226bca103b26260052732ad30Tom Cherry} 1107bac3299720623f4226bca103b26260052732ad30Tom Cherry 1108bac3299720623f4226bca103b26260052732ad30Tom Cherryvoid ServiceManager::ForEachServiceWithFlags(unsigned matchflags, 1109bac3299720623f4226bca103b26260052732ad30Tom Cherry void (*func)(Service* svc)) const { 1110bac3299720623f4226bca103b26260052732ad30Tom Cherry for (const auto& s : services_) { 1111bac3299720623f4226bca103b26260052732ad30Tom Cherry if (s->flags() & matchflags) { 1112bac3299720623f4226bca103b26260052732ad30Tom Cherry func(s.get()); 1113bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1114bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1115bac3299720623f4226bca103b26260052732ad30Tom Cherry} 1116bac3299720623f4226bca103b26260052732ad30Tom Cherry 1117b7349902a945903f9e36a569051f5131beb0bc24Tom Cherryvoid ServiceManager::RemoveService(const Service& svc) { 1118bac3299720623f4226bca103b26260052732ad30Tom Cherry auto svc_it = std::find_if(services_.begin(), services_.end(), 1119bac3299720623f4226bca103b26260052732ad30Tom Cherry [&svc] (const std::unique_ptr<Service>& s) { 1120bac3299720623f4226bca103b26260052732ad30Tom Cherry return svc.name() == s->name(); 1121bac3299720623f4226bca103b26260052732ad30Tom Cherry }); 1122bac3299720623f4226bca103b26260052732ad30Tom Cherry if (svc_it == services_.end()) { 1123bac3299720623f4226bca103b26260052732ad30Tom Cherry return; 1124bac3299720623f4226bca103b26260052732ad30Tom Cherry } 1125bac3299720623f4226bca103b26260052732ad30Tom Cherry 1126bac3299720623f4226bca103b26260052732ad30Tom Cherry services_.erase(svc_it); 1127bac3299720623f4226bca103b26260052732ad30Tom Cherry} 1128bac3299720623f4226bca103b26260052732ad30Tom Cherry 1129b7349902a945903f9e36a569051f5131beb0bc24Tom Cherryvoid ServiceManager::DumpState() const { 1130b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry for (const auto& s : services_) { 1131b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry s->DumpState(); 1132b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry } 1133b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 1134b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 1135b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNETbool ServiceManager::ReapOneProcess() { 11368d13d808a5634701c55276796683857ac2d5c99aTom Cherry siginfo_t siginfo = {}; 11378d13d808a5634701c55276796683857ac2d5c99aTom Cherry // This returns a zombie pid or informs us that there are no zombies left to be reaped. 11388d13d808a5634701c55276796683857ac2d5c99aTom Cherry // It does NOT reap the pid; that is done below. 11398d13d808a5634701c55276796683857ac2d5c99aTom Cherry if (TEMP_FAILURE_RETRY(waitid(P_ALL, 0, &siginfo, WEXITED | WNOHANG | WNOWAIT)) != 0) { 11408d13d808a5634701c55276796683857ac2d5c99aTom Cherry PLOG(ERROR) << "waitid failed"; 1141b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET return false; 11428d13d808a5634701c55276796683857ac2d5c99aTom Cherry } 11438d13d808a5634701c55276796683857ac2d5c99aTom Cherry 11448d13d808a5634701c55276796683857ac2d5c99aTom Cherry auto pid = siginfo.si_pid; 11458d13d808a5634701c55276796683857ac2d5c99aTom Cherry if (pid == 0) return false; 11468d13d808a5634701c55276796683857ac2d5c99aTom Cherry 11478d13d808a5634701c55276796683857ac2d5c99aTom Cherry // At this point we know we have a zombie pid, so we use this scopeguard to reap the pid 11488d13d808a5634701c55276796683857ac2d5c99aTom Cherry // whenever the function returns from this point forward. 11498d13d808a5634701c55276796683857ac2d5c99aTom Cherry // We do NOT want to reap the zombie earlier as in Service::Reap(), we kill(-pid, ...) and we 11508d13d808a5634701c55276796683857ac2d5c99aTom Cherry // want the pid to remain valid throughout that (and potentially future) usages. 11518d13d808a5634701c55276796683857ac2d5c99aTom Cherry auto reaper = make_scope_guard([pid] { TEMP_FAILURE_RETRY(waitpid(pid, nullptr, WNOHANG)); }); 11528d13d808a5634701c55276796683857ac2d5c99aTom Cherry 11538d13d808a5634701c55276796683857ac2d5c99aTom Cherry if (PropertyChildReap(pid)) { 115462696908ff840aeb3663b7510c0ae8ff208604f5Tom Marshall return true; 1155b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET } 1156b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET 1157b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET Service* svc = FindServiceByPid(pid); 1158b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET 1159b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET std::string name; 11604de31e1481dbbe6f08d85f341fa4884443433f35Keun-young Park std::string wait_string; 1161b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET if (svc) { 116281f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry name = StringPrintf("Service '%s' (pid %d)", svc->name().c_str(), pid); 11634de31e1481dbbe6f08d85f341fa4884443433f35Keun-young Park if (svc->flags() & SVC_EXEC) { 1164ede0d538501dfc78c741fb3b0645406636d1d1fdTom Cherry wait_string = StringPrintf(" waiting took %f seconds", 1165ede0d538501dfc78c741fb3b0645406636d1d1fdTom Cherry exec_waiter_->duration().count() / 1000.0f); 11664de31e1481dbbe6f08d85f341fa4884443433f35Keun-young Park } 1167b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET } else { 116881f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry name = StringPrintf("Untracked pid %d", pid); 1169b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET } 1170b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET 11718d13d808a5634701c55276796683857ac2d5c99aTom Cherry auto status = siginfo.si_status; 1172b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET if (WIFEXITED(status)) { 11734de31e1481dbbe6f08d85f341fa4884443433f35Keun-young Park LOG(INFO) << name << " exited with status " << WEXITSTATUS(status) << wait_string; 1174b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET } else if (WIFSIGNALED(status)) { 11754de31e1481dbbe6f08d85f341fa4884443433f35Keun-young Park LOG(INFO) << name << " killed by signal " << WTERMSIG(status) << wait_string; 1176b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET } 1177b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET 1178b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET if (!svc) { 1179b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET return true; 1180b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET } 1181b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET 1182b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry svc->Reap(); 1183b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry 1184b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry if (svc->flags() & SVC_EXEC) { 1185b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry exec_waiter_.reset(); 1186b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry } 1187b27004aa05039b5196f1e878169dca41b68aadd6Tom Cherry if (svc->flags() & SVC_TEMPORARY) { 1188b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET RemoveService(*svc); 1189b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET } 1190b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET 1191b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET return true; 1192b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET} 1193b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET 1194b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNETvoid ServiceManager::ReapAnyOutstandingChildren() { 1195b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET while (ReapOneProcess()) { 1196b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET } 1197b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET} 1198b7e03e82b89a30b09fea88eaf2a5638df1017cf6Bertrand SIMONNET 1199eeab491efd8f456324f88e444f228b1016712e45Wei Wangvoid ServiceManager::ClearExecWait() { 1200eeab491efd8f456324f88e444f228b1016712e45Wei Wang // Clear EXEC flag if there is one pending 1201eeab491efd8f456324f88e444f228b1016712e45Wei Wang // And clear the wait flag 1202eeab491efd8f456324f88e444f228b1016712e45Wei Wang for (const auto& s : services_) { 1203eeab491efd8f456324f88e444f228b1016712e45Wei Wang s->UnSetExec(); 1204eeab491efd8f456324f88e444f228b1016712e45Wei Wang } 1205eeab491efd8f456324f88e444f228b1016712e45Wei Wang exec_waiter_.reset(); 1206eeab491efd8f456324f88e444f228b1016712e45Wei Wang} 1207eeab491efd8f456324f88e444f228b1016712e45Wei Wang 120830a6f276fd8850b0a78689d7bff3cb06a18cb286Tom Cherrybool ServiceParser::ParseSection(std::vector<std::string>&& args, const std::string& filename, 1209012c573e267b8dd70de14237cb470bd7301ee8eaTom Cherry int line, std::string* err) { 1210b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry if (args.size() < 3) { 1211b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry *err = "services must have a name and a program"; 1212b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return false; 1213b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry } 1214b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 1215b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry const std::string& name = args[1]; 1216b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry if (!IsValidName(name)) { 1217b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry *err = StringPrintf("invalid service name '%s'", name.c_str()); 1218b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return false; 1219b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry } 1220b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 122130a6f276fd8850b0a78689d7bff3cb06a18cb286Tom Cherry Service* old_service = service_manager_->FindServiceByName(name); 122230a6f276fd8850b0a78689d7bff3cb06a18cb286Tom Cherry if (old_service) { 122330a6f276fd8850b0a78689d7bff3cb06a18cb286Tom Cherry *err = "ignored duplicate definition of service '" + name + "'"; 122430a6f276fd8850b0a78689d7bff3cb06a18cb286Tom Cherry return false; 122530a6f276fd8850b0a78689d7bff3cb06a18cb286Tom Cherry } 122630a6f276fd8850b0a78689d7bff3cb06a18cb286Tom Cherry 1227b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry std::vector<std::string> str_args(args.begin() + 2, args.end()); 1228641ff0a4d8d05349d9c22f726a035a2936a2a94dWei Wang service_ = std::make_unique<Service>(name, str_args); 1229b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry return true; 1230b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 1231b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 123230a6f276fd8850b0a78689d7bff3cb06a18cb286Tom Cherrybool ServiceParser::ParseLineSection(std::vector<std::string>&& args, int line, std::string* err) { 123330a6f276fd8850b0a78689d7bff3cb06a18cb286Tom Cherry return service_ ? service_->ParseLine(std::move(args), err) : false; 1234b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 1235b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 1236b7349902a945903f9e36a569051f5131beb0bc24Tom Cherryvoid ServiceParser::EndSection() { 1237b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry if (service_) { 123830a6f276fd8850b0a78689d7bff3cb06a18cb286Tom Cherry service_manager_->AddService(std::move(service_)); 1239b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry } 1240b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry} 1241b7349902a945903f9e36a569051f5131beb0bc24Tom Cherry 1242b7349902a945903f9e36a569051f5131beb0bc24Tom Cherrybool ServiceParser::IsValidName(const std::string& name) const { 1243b7788fd454da2201c6e48dededa9b11873ef621eElliott Hughes // Property names can be any length, but may only contain certain characters. 1244b7788fd454da2201c6e48dededa9b11873ef621eElliott Hughes // Property values can contain any characters, but may only be a certain length. 1245b7788fd454da2201c6e48dededa9b11873ef621eElliott Hughes // (The latter restriction is needed because `start` and `stop` work by writing 1246b7788fd454da2201c6e48dededa9b11873ef621eElliott Hughes // the service name to the "ctl.start" and "ctl.stop" properties.) 1247b7788fd454da2201c6e48dededa9b11873ef621eElliott Hughes return is_legal_property_name("init.svc." + name) && name.size() <= PROP_VALUE_MAX; 1248bac3299720623f4226bca103b26260052732ad30Tom Cherry} 124981f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry 125081f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry} // namespace init 125181f5d3ebef2c3789737bf718fc2a2cdd7b9e8b33Tom Cherry} // namespace android 1252