Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2015 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | #include "service.h" |
| 18 | |
| 19 | #include <fcntl.h> |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 20 | #include <sched.h> |
| 21 | #include <sys/mount.h> |
| 22 | #include <sys/prctl.h> |
Vitalii Tomkiv | 081705c | 2016-05-18 17:36:30 -0700 | [diff] [blame] | 23 | #include <sys/resource.h> |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 24 | #include <sys/stat.h> |
Vitalii Tomkiv | 081705c | 2016-05-18 17:36:30 -0700 | [diff] [blame] | 25 | #include <sys/time.h> |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 26 | #include <sys/types.h> |
Bertrand SIMONNET | b7e03e8 | 2015-12-18 11:39:59 -0800 | [diff] [blame] | 27 | #include <sys/wait.h> |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 28 | #include <termios.h> |
Dan Albert | af9ba4d | 2015-08-11 16:37:04 -0700 | [diff] [blame] | 29 | #include <unistd.h> |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 30 | |
| 31 | #include <selinux/selinux.h> |
| 32 | |
Elliott Hughes | 4f71319 | 2015-12-04 22:00:26 -0800 | [diff] [blame] | 33 | #include <android-base/file.h> |
Elliott Hughes | da46b39 | 2016-10-11 17:09:00 -0700 | [diff] [blame] | 34 | #include <android-base/parseint.h> |
Elliott Hughes | 4f71319 | 2015-12-04 22:00:26 -0800 | [diff] [blame] | 35 | #include <android-base/stringprintf.h> |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 36 | #include <android-base/strings.h> |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 37 | #include <cutils/android_reboot.h> |
| 38 | #include <cutils/sockets.h> |
Vitalii Tomkiv | 081705c | 2016-05-18 17:36:30 -0700 | [diff] [blame] | 39 | #include <system/thread_defs.h> |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 40 | |
Collin Mulliner | f7e79b9 | 2016-06-01 21:03:55 +0000 | [diff] [blame] | 41 | #include <processgroup/processgroup.h> |
| 42 | |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 43 | #include "action.h" |
| 44 | #include "init.h" |
| 45 | #include "init_parser.h" |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 46 | #include "log.h" |
| 47 | #include "property_service.h" |
| 48 | #include "util.h" |
| 49 | |
Elliott Hughes | da46b39 | 2016-10-11 17:09:00 -0700 | [diff] [blame] | 50 | using android::base::ParseInt; |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 51 | using android::base::StringPrintf; |
| 52 | using android::base::WriteStringToFile; |
| 53 | |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 54 | #define CRITICAL_CRASH_THRESHOLD 4 // if we crash >4 times ... |
| 55 | #define CRITICAL_CRASH_WINDOW (4*60) // ... in 4 minutes, goto recovery |
| 56 | |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 57 | static std::string ComputeContextFromExecutable(std::string& service_name, |
| 58 | const std::string& service_path) { |
| 59 | std::string computed_context; |
| 60 | |
| 61 | char* raw_con = nullptr; |
| 62 | char* raw_filecon = nullptr; |
| 63 | |
| 64 | if (getcon(&raw_con) == -1) { |
| 65 | LOG(ERROR) << "could not get context while starting '" << service_name << "'"; |
| 66 | return ""; |
| 67 | } |
| 68 | std::unique_ptr<char> mycon(raw_con); |
| 69 | |
| 70 | if (getfilecon(service_path.c_str(), &raw_filecon) == -1) { |
| 71 | LOG(ERROR) << "could not get file context while starting '" << service_name << "'"; |
| 72 | return ""; |
| 73 | } |
| 74 | std::unique_ptr<char> filecon(raw_filecon); |
| 75 | |
| 76 | char* new_con = nullptr; |
| 77 | int rc = security_compute_create(mycon.get(), filecon.get(), |
| 78 | string_to_security_class("process"), &new_con); |
| 79 | if (rc == 0) { |
| 80 | computed_context = new_con; |
| 81 | free(new_con); |
| 82 | } |
| 83 | if (rc == 0 && computed_context == mycon.get()) { |
| 84 | LOG(ERROR) << "service " << service_name << " does not have a SELinux domain defined"; |
| 85 | return ""; |
| 86 | } |
| 87 | if (rc < 0) { |
| 88 | LOG(ERROR) << "could not get context while starting '" << service_name << "'"; |
| 89 | return ""; |
| 90 | } |
| 91 | return computed_context; |
| 92 | } |
| 93 | |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 94 | static void SetUpPidNamespace(const std::string& service_name) { |
| 95 | constexpr unsigned int kSafeFlags = MS_NODEV | MS_NOEXEC | MS_NOSUID; |
| 96 | |
| 97 | // It's OK to LOG(FATAL) in this function since it's running in the first |
| 98 | // child process. |
| 99 | if (mount("", "/proc", "proc", kSafeFlags | MS_REMOUNT, "") == -1) { |
Elliott Hughes | e18e7e5 | 2016-07-25 18:18:16 -0700 | [diff] [blame] | 100 | PLOG(FATAL) << "couldn't remount(/proc) for " << service_name; |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 101 | } |
| 102 | |
| 103 | if (prctl(PR_SET_NAME, service_name.c_str()) == -1) { |
Elliott Hughes | e18e7e5 | 2016-07-25 18:18:16 -0700 | [diff] [blame] | 104 | PLOG(FATAL) << "couldn't set name for " << service_name; |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 105 | } |
| 106 | |
| 107 | pid_t child_pid = fork(); |
| 108 | if (child_pid == -1) { |
Elliott Hughes | e18e7e5 | 2016-07-25 18:18:16 -0700 | [diff] [blame] | 109 | PLOG(FATAL) << "couldn't fork init inside the PID namespace for " << service_name; |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 110 | } |
| 111 | |
| 112 | if (child_pid > 0) { |
| 113 | // So that we exit with the right status. |
| 114 | static int init_exitstatus = 0; |
| 115 | signal(SIGTERM, [](int) { _exit(init_exitstatus); }); |
| 116 | |
| 117 | pid_t waited_pid; |
| 118 | int status; |
| 119 | while ((waited_pid = wait(&status)) > 0) { |
| 120 | // This loop will end when there are no processes left inside the |
| 121 | // PID namespace or when the init process inside the PID namespace |
| 122 | // gets a signal. |
| 123 | if (waited_pid == child_pid) { |
| 124 | init_exitstatus = status; |
| 125 | } |
| 126 | } |
| 127 | if (!WIFEXITED(init_exitstatus)) { |
| 128 | _exit(EXIT_FAILURE); |
| 129 | } |
| 130 | _exit(WEXITSTATUS(init_exitstatus)); |
| 131 | } |
| 132 | } |
| 133 | |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 134 | static void ExpandArgs(const std::vector<std::string>& args, std::vector<char*>* strs) { |
| 135 | std::vector<std::string> expanded_args; |
| 136 | expanded_args.resize(args.size()); |
| 137 | strs->push_back(const_cast<char*>(args[0].c_str())); |
| 138 | for (std::size_t i = 1; i < args.size(); ++i) { |
| 139 | if (!expand_props(args[i], &expanded_args[i])) { |
| 140 | LOG(FATAL) << args[0] << ": cannot expand '" << args[i] << "'"; |
| 141 | } |
| 142 | strs->push_back(const_cast<char*>(expanded_args[i].c_str())); |
| 143 | } |
| 144 | strs->push_back(nullptr); |
| 145 | } |
| 146 | |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 147 | SocketInfo::SocketInfo() : uid(0), gid(0), perm(0) { |
| 148 | } |
| 149 | |
| 150 | SocketInfo::SocketInfo(const std::string& name, const std::string& type, uid_t uid, |
| 151 | gid_t gid, int perm, const std::string& socketcon) |
| 152 | : name(name), type(type), uid(uid), gid(gid), perm(perm), socketcon(socketcon) { |
| 153 | } |
| 154 | |
| 155 | ServiceEnvironmentInfo::ServiceEnvironmentInfo() { |
| 156 | } |
| 157 | |
| 158 | ServiceEnvironmentInfo::ServiceEnvironmentInfo(const std::string& name, |
| 159 | const std::string& value) |
| 160 | : name(name), value(value) { |
| 161 | } |
| 162 | |
| 163 | Service::Service(const std::string& name, const std::string& classname, |
| 164 | const std::vector<std::string>& args) |
| 165 | : name_(name), classname_(classname), flags_(0), pid_(0), time_started_(0), |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 166 | time_crashed_(0), nr_crashed_(0), uid_(0), gid_(0), namespace_flags_(0), |
| 167 | seclabel_(""), ioprio_class_(IoSchedClass_NONE), ioprio_pri_(0), |
Marco Nelissen | 310f670 | 2016-07-22 12:07:06 -0700 | [diff] [blame] | 168 | priority_(0), oom_score_adjust_(-1000), args_(args) { |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 169 | onrestart_.InitSingleTrigger("onrestart"); |
| 170 | } |
| 171 | |
| 172 | Service::Service(const std::string& name, const std::string& classname, |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 173 | unsigned flags, uid_t uid, gid_t gid, |
| 174 | const std::vector<gid_t>& supp_gids, unsigned namespace_flags, |
| 175 | const std::string& seclabel, |
| 176 | const std::vector<std::string>& args) |
| 177 | : name_(name), classname_(classname), flags_(flags), pid_(0), |
| 178 | time_started_(0), time_crashed_(0), nr_crashed_(0), uid_(uid), gid_(gid), |
| 179 | supp_gids_(supp_gids), namespace_flags_(namespace_flags), |
| 180 | seclabel_(seclabel), ioprio_class_(IoSchedClass_NONE), ioprio_pri_(0), |
Marco Nelissen | 310f670 | 2016-07-22 12:07:06 -0700 | [diff] [blame] | 181 | priority_(0), oom_score_adjust_(-1000), args_(args) { |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 182 | onrestart_.InitSingleTrigger("onrestart"); |
| 183 | } |
| 184 | |
| 185 | void Service::NotifyStateChange(const std::string& new_state) const { |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 186 | if ((flags_ & SVC_EXEC) != 0) { |
| 187 | // 'exec' commands don't have properties tracking their state. |
| 188 | return; |
| 189 | } |
| 190 | |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 191 | std::string prop_name = StringPrintf("init.svc.%s", name_.c_str()); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 192 | if (prop_name.length() >= PROP_NAME_MAX) { |
| 193 | // If the property name would be too long, we can't set it. |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 194 | LOG(ERROR) << "Property name \"init.svc." << name_ << "\" too long; not setting to " << new_state; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 195 | return; |
| 196 | } |
| 197 | |
| 198 | property_set(prop_name.c_str(), new_state.c_str()); |
| 199 | } |
| 200 | |
Elliott Hughes | ad8e94e | 2016-06-15 14:49:57 -0700 | [diff] [blame] | 201 | void Service::KillProcessGroup(int signal) { |
Elliott Hughes | 1e73024 | 2016-08-02 14:20:40 -0700 | [diff] [blame] | 202 | LOG(INFO) << "Sending signal " << signal |
| 203 | << " to service '" << name_ |
| 204 | << "' (pid " << pid_ << ") process group..."; |
| 205 | if (killProcessGroup(uid_, pid_, signal) == -1) { |
| 206 | PLOG(ERROR) << "killProcessGroup(" << uid_ << ", " << pid_ << ", " << signal << ") failed"; |
| 207 | } |
| 208 | if (kill(-pid_, signal) == -1) { |
| 209 | PLOG(ERROR) << "kill(" << pid_ << ", " << signal << ") failed"; |
| 210 | } |
Elliott Hughes | ad8e94e | 2016-06-15 14:49:57 -0700 | [diff] [blame] | 211 | } |
| 212 | |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 213 | void Service::CreateSockets(const std::string& context) { |
| 214 | for (const auto& si : sockets_) { |
| 215 | int socket_type = ((si.type == "stream" ? SOCK_STREAM : |
| 216 | (si.type == "dgram" ? SOCK_DGRAM : |
| 217 | SOCK_SEQPACKET))); |
| 218 | const char* socketcon = !si.socketcon.empty() ? si.socketcon.c_str() : context.c_str(); |
| 219 | |
| 220 | int s = create_socket(si.name.c_str(), socket_type, si.perm, si.uid, si.gid, socketcon); |
| 221 | if (s >= 0) { |
| 222 | PublishSocket(si.name, s); |
| 223 | } |
| 224 | } |
| 225 | } |
| 226 | |
| 227 | void Service::SetProcessAttributes() { |
Elliott Hughes | e18e7e5 | 2016-07-25 18:18:16 -0700 | [diff] [blame] | 228 | // TODO: work out why this fails for `console` then upgrade to FATAL. |
| 229 | if (setpgid(0, getpid()) == -1) PLOG(ERROR) << "setpgid failed for " << name_; |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 230 | |
| 231 | if (gid_) { |
| 232 | if (setgid(gid_) != 0) { |
Elliott Hughes | e18e7e5 | 2016-07-25 18:18:16 -0700 | [diff] [blame] | 233 | PLOG(FATAL) << "setgid failed for " << name_; |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 234 | } |
| 235 | } |
Nick Kralevich | 80960d2 | 2016-10-29 12:20:00 -0700 | [diff] [blame^] | 236 | if (setgroups(supp_gids_.size(), &supp_gids_[0]) != 0) { |
| 237 | PLOG(FATAL) << "setgroups failed for " << name_; |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 238 | } |
| 239 | if (uid_) { |
| 240 | if (setuid(uid_) != 0) { |
Elliott Hughes | e18e7e5 | 2016-07-25 18:18:16 -0700 | [diff] [blame] | 241 | PLOG(FATAL) << "setuid failed for " << name_; |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 242 | } |
| 243 | } |
| 244 | if (!seclabel_.empty()) { |
| 245 | if (setexeccon(seclabel_.c_str()) < 0) { |
Elliott Hughes | e18e7e5 | 2016-07-25 18:18:16 -0700 | [diff] [blame] | 246 | PLOG(FATAL) << "cannot setexeccon('" << seclabel_ << "') for " << name_; |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 247 | } |
| 248 | } |
| 249 | if (priority_ != 0) { |
| 250 | if (setpriority(PRIO_PROCESS, 0, priority_) != 0) { |
Elliott Hughes | e18e7e5 | 2016-07-25 18:18:16 -0700 | [diff] [blame] | 251 | PLOG(FATAL) << "setpriority failed for " << name_; |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 252 | } |
| 253 | } |
| 254 | } |
| 255 | |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 256 | bool Service::Reap() { |
| 257 | if (!(flags_ & SVC_ONESHOT) || (flags_ & SVC_RESTART)) { |
Elliott Hughes | ad8e94e | 2016-06-15 14:49:57 -0700 | [diff] [blame] | 258 | KillProcessGroup(SIGKILL); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 259 | } |
| 260 | |
| 261 | // Remove any sockets we may have created. |
| 262 | for (const auto& si : sockets_) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 263 | std::string tmp = StringPrintf(ANDROID_SOCKET_DIR "/%s", si.name.c_str()); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 264 | unlink(tmp.c_str()); |
| 265 | } |
| 266 | |
| 267 | if (flags_ & SVC_EXEC) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 268 | LOG(INFO) << "SVC_EXEC pid " << pid_ << " finished..."; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 269 | return true; |
| 270 | } |
| 271 | |
| 272 | pid_ = 0; |
| 273 | flags_ &= (~SVC_RUNNING); |
| 274 | |
| 275 | // Oneshot processes go into the disabled state on exit, |
| 276 | // except when manually restarted. |
| 277 | if ((flags_ & SVC_ONESHOT) && !(flags_ & SVC_RESTART)) { |
| 278 | flags_ |= SVC_DISABLED; |
| 279 | } |
| 280 | |
| 281 | // Disabled and reset processes do not get restarted automatically. |
| 282 | if (flags_ & (SVC_DISABLED | SVC_RESET)) { |
| 283 | NotifyStateChange("stopped"); |
| 284 | return false; |
| 285 | } |
| 286 | |
| 287 | time_t now = gettime(); |
| 288 | if ((flags_ & SVC_CRITICAL) && !(flags_ & SVC_RESTART)) { |
| 289 | if (time_crashed_ + CRITICAL_CRASH_WINDOW >= now) { |
| 290 | if (++nr_crashed_ > CRITICAL_CRASH_THRESHOLD) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 291 | LOG(ERROR) << "critical process '" << name_ << "' exited " |
| 292 | << CRITICAL_CRASH_THRESHOLD << " times in " |
| 293 | << (CRITICAL_CRASH_WINDOW / 60) << " minutes; " |
| 294 | << "rebooting into recovery mode"; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 295 | android_reboot(ANDROID_RB_RESTART2, 0, "recovery"); |
| 296 | return false; |
| 297 | } |
| 298 | } else { |
| 299 | time_crashed_ = now; |
| 300 | nr_crashed_ = 1; |
| 301 | } |
| 302 | } |
| 303 | |
| 304 | flags_ &= (~SVC_RESTART); |
| 305 | flags_ |= SVC_RESTARTING; |
| 306 | |
| 307 | // Execute all onrestart commands for this service. |
| 308 | onrestart_.ExecuteAllCommands(); |
| 309 | |
| 310 | NotifyStateChange("restarting"); |
| 311 | return false; |
| 312 | } |
| 313 | |
| 314 | void Service::DumpState() const { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 315 | LOG(INFO) << "service " << name_; |
| 316 | LOG(INFO) << " class '" << classname_ << "'"; |
| 317 | LOG(INFO) << " exec "<< android::base::Join(args_, " "); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 318 | for (const auto& si : sockets_) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 319 | LOG(INFO) << " socket " << si.name << " " << si.type << " " << std::oct << si.perm; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 320 | } |
| 321 | } |
| 322 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 323 | bool Service::ParseClass(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 324 | classname_ = args[1]; |
| 325 | return true; |
| 326 | } |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 327 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 328 | bool Service::ParseConsole(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 329 | flags_ |= SVC_CONSOLE; |
Viorel Suman | 70daa67 | 2016-03-21 10:08:07 +0200 | [diff] [blame] | 330 | console_ = args.size() > 1 ? "/dev/" + args[1] : ""; |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 331 | return true; |
| 332 | } |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 333 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 334 | bool Service::ParseCritical(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 335 | flags_ |= SVC_CRITICAL; |
| 336 | return true; |
| 337 | } |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 338 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 339 | bool Service::ParseDisabled(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 340 | flags_ |= SVC_DISABLED; |
| 341 | flags_ |= SVC_RC_DISABLED; |
| 342 | return true; |
| 343 | } |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 344 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 345 | bool Service::ParseGroup(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 346 | gid_ = decode_uid(args[1].c_str()); |
| 347 | for (std::size_t n = 2; n < args.size(); n++) { |
| 348 | supp_gids_.emplace_back(decode_uid(args[n].c_str())); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 349 | } |
| 350 | return true; |
| 351 | } |
| 352 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 353 | bool Service::ParsePriority(const std::vector<std::string>& args, std::string* err) { |
Elliott Hughes | da46b39 | 2016-10-11 17:09:00 -0700 | [diff] [blame] | 354 | priority_ = 0; |
| 355 | if (!ParseInt(args[1], &priority_, |
| 356 | static_cast<int>(ANDROID_PRIORITY_LOWEST), |
| 357 | static_cast<int>(ANDROID_PRIORITY_HIGHEST))) { |
Vitalii Tomkiv | 081705c | 2016-05-18 17:36:30 -0700 | [diff] [blame] | 358 | *err = StringPrintf("process priority value must be range %d - %d", |
| 359 | ANDROID_PRIORITY_HIGHEST, ANDROID_PRIORITY_LOWEST); |
| 360 | return false; |
| 361 | } |
Vitalii Tomkiv | 081705c | 2016-05-18 17:36:30 -0700 | [diff] [blame] | 362 | return true; |
| 363 | } |
| 364 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 365 | bool Service::ParseIoprio(const std::vector<std::string>& args, std::string* err) { |
Elliott Hughes | da46b39 | 2016-10-11 17:09:00 -0700 | [diff] [blame] | 366 | if (!ParseInt(args[2], &ioprio_pri_, 0, 7)) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 367 | *err = "priority value must be range 0 - 7"; |
| 368 | return false; |
| 369 | } |
| 370 | |
| 371 | if (args[1] == "rt") { |
| 372 | ioprio_class_ = IoSchedClass_RT; |
| 373 | } else if (args[1] == "be") { |
| 374 | ioprio_class_ = IoSchedClass_BE; |
| 375 | } else if (args[1] == "idle") { |
| 376 | ioprio_class_ = IoSchedClass_IDLE; |
| 377 | } else { |
| 378 | *err = "ioprio option usage: ioprio <rt|be|idle> <0-7>"; |
| 379 | return false; |
| 380 | } |
| 381 | |
| 382 | return true; |
| 383 | } |
| 384 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 385 | bool Service::ParseKeycodes(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 386 | for (std::size_t i = 1; i < args.size(); i++) { |
Elliott Hughes | da46b39 | 2016-10-11 17:09:00 -0700 | [diff] [blame] | 387 | int code; |
| 388 | if (ParseInt(args[i], &code)) { |
| 389 | keycodes_.emplace_back(code); |
| 390 | } else { |
| 391 | LOG(WARNING) << "ignoring invalid keycode: " << args[i]; |
| 392 | } |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 393 | } |
| 394 | return true; |
| 395 | } |
| 396 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 397 | bool Service::ParseOneshot(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 398 | flags_ |= SVC_ONESHOT; |
| 399 | return true; |
| 400 | } |
| 401 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 402 | bool Service::ParseOnrestart(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 403 | std::vector<std::string> str_args(args.begin() + 1, args.end()); |
| 404 | onrestart_.AddCommand(str_args, "", 0, err); |
| 405 | return true; |
| 406 | } |
| 407 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 408 | bool Service::ParseNamespace(const std::vector<std::string>& args, std::string* err) { |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 409 | for (size_t i = 1; i < args.size(); i++) { |
| 410 | if (args[i] == "pid") { |
| 411 | namespace_flags_ |= CLONE_NEWPID; |
| 412 | // PID namespaces require mount namespaces. |
| 413 | namespace_flags_ |= CLONE_NEWNS; |
| 414 | } else if (args[i] == "mnt") { |
| 415 | namespace_flags_ |= CLONE_NEWNS; |
| 416 | } else { |
| 417 | *err = "namespace must be 'pid' or 'mnt'"; |
| 418 | return false; |
| 419 | } |
| 420 | } |
| 421 | return true; |
| 422 | } |
| 423 | |
Marco Nelissen | 310f670 | 2016-07-22 12:07:06 -0700 | [diff] [blame] | 424 | bool Service::ParseOomScoreAdjust(const std::vector<std::string>& args, std::string* err) { |
Elliott Hughes | da46b39 | 2016-10-11 17:09:00 -0700 | [diff] [blame] | 425 | if (!ParseInt(args[1], &oom_score_adjust_, -1000, 1000)) { |
Marco Nelissen | 310f670 | 2016-07-22 12:07:06 -0700 | [diff] [blame] | 426 | *err = "oom_score_adjust value must be in range -1000 - +1000"; |
| 427 | return false; |
| 428 | } |
Marco Nelissen | 310f670 | 2016-07-22 12:07:06 -0700 | [diff] [blame] | 429 | return true; |
| 430 | } |
| 431 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 432 | bool Service::ParseSeclabel(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 433 | seclabel_ = args[1]; |
| 434 | return true; |
| 435 | } |
| 436 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 437 | bool Service::ParseSetenv(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 438 | envvars_.emplace_back(args[1], args[2]); |
| 439 | return true; |
| 440 | } |
| 441 | |
| 442 | /* name type perm [ uid gid context ] */ |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 443 | bool Service::ParseSocket(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 444 | if (args[2] != "dgram" && args[2] != "stream" && args[2] != "seqpacket") { |
| 445 | *err = "socket type must be 'dgram', 'stream' or 'seqpacket'"; |
| 446 | return false; |
| 447 | } |
| 448 | |
Elliott Hughes | da46b39 | 2016-10-11 17:09:00 -0700 | [diff] [blame] | 449 | int perm = std::strtoul(args[3].c_str(), 0, 8); |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 450 | uid_t uid = args.size() > 4 ? decode_uid(args[4].c_str()) : 0; |
| 451 | gid_t gid = args.size() > 5 ? decode_uid(args[5].c_str()) : 0; |
| 452 | std::string socketcon = args.size() > 6 ? args[6] : ""; |
| 453 | |
| 454 | sockets_.emplace_back(args[1], args[2], uid, gid, perm, socketcon); |
| 455 | return true; |
| 456 | } |
| 457 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 458 | bool Service::ParseUser(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 459 | uid_ = decode_uid(args[1].c_str()); |
| 460 | return true; |
| 461 | } |
| 462 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 463 | bool Service::ParseWritepid(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 464 | writepid_files_.assign(args.begin() + 1, args.end()); |
| 465 | return true; |
| 466 | } |
| 467 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 468 | class Service::OptionParserMap : public KeywordMap<OptionParser> { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 469 | public: |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 470 | OptionParserMap() { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 471 | } |
| 472 | private: |
| 473 | Map& map() const override; |
| 474 | }; |
| 475 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 476 | Service::OptionParserMap::Map& Service::OptionParserMap::map() const { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 477 | constexpr std::size_t kMax = std::numeric_limits<std::size_t>::max(); |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 478 | static const Map option_parsers = { |
| 479 | {"class", {1, 1, &Service::ParseClass}}, |
| 480 | {"console", {0, 1, &Service::ParseConsole}}, |
| 481 | {"critical", {0, 0, &Service::ParseCritical}}, |
| 482 | {"disabled", {0, 0, &Service::ParseDisabled}}, |
| 483 | {"group", {1, NR_SVC_SUPP_GIDS + 1, &Service::ParseGroup}}, |
| 484 | {"ioprio", {2, 2, &Service::ParseIoprio}}, |
| 485 | {"priority", {1, 1, &Service::ParsePriority}}, |
| 486 | {"keycodes", {1, kMax, &Service::ParseKeycodes}}, |
| 487 | {"oneshot", {0, 0, &Service::ParseOneshot}}, |
| 488 | {"onrestart", {1, kMax, &Service::ParseOnrestart}}, |
Marco Nelissen | 310f670 | 2016-07-22 12:07:06 -0700 | [diff] [blame] | 489 | {"oom_score_adjust", |
| 490 | {1, 1, &Service::ParseOomScoreAdjust}}, |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 491 | {"namespace", {1, 2, &Service::ParseNamespace}}, |
| 492 | {"seclabel", {1, 1, &Service::ParseSeclabel}}, |
| 493 | {"setenv", {2, 2, &Service::ParseSetenv}}, |
| 494 | {"socket", {3, 6, &Service::ParseSocket}}, |
| 495 | {"user", {1, 1, &Service::ParseUser}}, |
| 496 | {"writepid", {1, kMax, &Service::ParseWritepid}}, |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 497 | }; |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 498 | return option_parsers; |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 499 | } |
| 500 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 501 | bool Service::ParseLine(const std::vector<std::string>& args, std::string* err) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 502 | if (args.empty()) { |
| 503 | *err = "option needed, but not provided"; |
| 504 | return false; |
| 505 | } |
| 506 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 507 | static const OptionParserMap parser_map; |
| 508 | auto parser = parser_map.FindFunction(args[0], args.size() - 1, err); |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 509 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 510 | if (!parser) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 511 | return false; |
| 512 | } |
| 513 | |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 514 | return (this->*parser)(args, err); |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 515 | } |
| 516 | |
Elliott Hughes | bdeac39 | 2016-04-12 15:38:27 -0700 | [diff] [blame] | 517 | bool Service::Start() { |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 518 | // Starting a service removes it from the disabled or reset state and |
| 519 | // immediately takes it out of the restarting state if it was in there. |
| 520 | flags_ &= (~(SVC_DISABLED|SVC_RESTARTING|SVC_RESET|SVC_RESTART|SVC_DISABLED_START)); |
| 521 | time_started_ = 0; |
| 522 | |
| 523 | // Running processes require no additional work --- if they're in the |
| 524 | // process of exiting, we've ensured that they will immediately restart |
| 525 | // on exit, unless they are ONESHOT. |
| 526 | if (flags_ & SVC_RUNNING) { |
| 527 | return false; |
| 528 | } |
| 529 | |
| 530 | bool needs_console = (flags_ & SVC_CONSOLE); |
Viorel Suman | 70daa67 | 2016-03-21 10:08:07 +0200 | [diff] [blame] | 531 | if (needs_console) { |
| 532 | if (console_.empty()) { |
| 533 | console_ = default_console; |
| 534 | } |
| 535 | |
| 536 | bool have_console = (open(console_.c_str(), O_RDWR | O_CLOEXEC) != -1); |
| 537 | if (!have_console) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 538 | PLOG(ERROR) << "service '" << name_ << "' couldn't open console '" << console_ << "'"; |
Viorel Suman | 70daa67 | 2016-03-21 10:08:07 +0200 | [diff] [blame] | 539 | flags_ |= SVC_DISABLED; |
| 540 | return false; |
| 541 | } |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 542 | } |
| 543 | |
| 544 | struct stat sb; |
| 545 | if (stat(args_[0].c_str(), &sb) == -1) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 546 | PLOG(ERROR) << "cannot find '" << args_[0] << "', disabling '" << name_ << "'"; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 547 | flags_ |= SVC_DISABLED; |
| 548 | return false; |
| 549 | } |
| 550 | |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 551 | std::string scon; |
| 552 | if (!seclabel_.empty()) { |
| 553 | scon = seclabel_; |
| 554 | } else { |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 555 | LOG(INFO) << "computing context for service '" << name_ << "'"; |
| 556 | scon = ComputeContextFromExecutable(name_, args_[0]); |
| 557 | if (scon == "") { |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 558 | return false; |
| 559 | } |
| 560 | } |
| 561 | |
Wei Wang | a285dac | 2016-10-04 14:05:39 -0700 | [diff] [blame] | 562 | LOG(INFO) << "starting service '" << name_ << "'..."; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 563 | |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 564 | pid_t pid = -1; |
| 565 | if (namespace_flags_) { |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 566 | pid = clone(nullptr, nullptr, namespace_flags_ | SIGCHLD, nullptr); |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 567 | } else { |
| 568 | pid = fork(); |
| 569 | } |
| 570 | |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 571 | if (pid == 0) { |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 572 | umask(077); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 573 | |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 574 | if (namespace_flags_ & CLONE_NEWPID) { |
| 575 | // This will fork again to run an init process inside the PID |
| 576 | // namespace. |
| 577 | SetUpPidNamespace(name_); |
| 578 | } |
| 579 | |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 580 | for (const auto& ei : envvars_) { |
| 581 | add_environment(ei.name.c_str(), ei.value.c_str()); |
| 582 | } |
| 583 | |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 584 | CreateSockets(scon); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 585 | |
Anestis Bechtsoudis | b702b46 | 2016-02-05 16:38:48 +0200 | [diff] [blame] | 586 | std::string pid_str = StringPrintf("%d", getpid()); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 587 | for (const auto& file : writepid_files_) { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 588 | if (!WriteStringToFile(pid_str, file)) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 589 | PLOG(ERROR) << "couldn't write " << pid_str << " to " << file; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 590 | } |
| 591 | } |
| 592 | |
| 593 | if (ioprio_class_ != IoSchedClass_NONE) { |
| 594 | if (android_set_ioprio(getpid(), ioprio_class_, ioprio_pri_)) { |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 595 | PLOG(ERROR) << "failed to set pid " << getpid() |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 596 | << " ioprio=" << ioprio_class_ << "," << ioprio_pri_; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 597 | } |
| 598 | } |
| 599 | |
| 600 | if (needs_console) { |
| 601 | setsid(); |
| 602 | OpenConsole(); |
| 603 | } else { |
| 604 | ZapStdio(); |
| 605 | } |
| 606 | |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 607 | // As requested, set our gid, supplemental gids, uid, context, and |
| 608 | // priority. Aborts on failure. |
| 609 | SetProcessAttributes(); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 610 | |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 611 | std::vector<char*> strs; |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 612 | ExpandArgs(args_, &strs); |
Tom Cherry | bac3536 | 2016-06-07 11:22:00 -0700 | [diff] [blame] | 613 | if (execve(strs[0], (char**) &strs[0], (char**) ENV) < 0) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 614 | PLOG(ERROR) << "cannot execve('" << strs[0] << "')"; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 615 | } |
| 616 | |
| 617 | _exit(127); |
| 618 | } |
| 619 | |
| 620 | if (pid < 0) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 621 | PLOG(ERROR) << "failed to fork for '" << name_ << "'"; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 622 | pid_ = 0; |
| 623 | return false; |
| 624 | } |
| 625 | |
Marco Nelissen | 310f670 | 2016-07-22 12:07:06 -0700 | [diff] [blame] | 626 | if (oom_score_adjust_ != -1000) { |
| 627 | std::string oom_str = StringPrintf("%d", oom_score_adjust_); |
| 628 | std::string oom_file = StringPrintf("/proc/%d/oom_score_adj", pid); |
| 629 | if (!WriteStringToFile(oom_str, oom_file)) { |
| 630 | PLOG(ERROR) << "couldn't write oom_score_adj: " << strerror(errno); |
| 631 | } |
| 632 | } |
| 633 | |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 634 | time_started_ = gettime(); |
| 635 | pid_ = pid; |
| 636 | flags_ |= SVC_RUNNING; |
Elliott Hughes | ad8e94e | 2016-06-15 14:49:57 -0700 | [diff] [blame] | 637 | |
| 638 | errno = -createProcessGroup(uid_, pid_); |
| 639 | if (errno != 0) { |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 640 | PLOG(ERROR) << "createProcessGroup(" << uid_ << ", " << pid_ << ") failed for service '" |
| 641 | << name_ << "'"; |
Elliott Hughes | ad8e94e | 2016-06-15 14:49:57 -0700 | [diff] [blame] | 642 | } |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 643 | |
| 644 | if ((flags_ & SVC_EXEC) != 0) { |
Jorge Lucangeli Obes | 344d01f | 2016-07-08 13:32:26 -0400 | [diff] [blame] | 645 | LOG(INFO) << android::base::StringPrintf( |
| 646 | "SVC_EXEC pid %d (uid %d gid %d+%zu context %s) started; waiting...", pid_, uid_, gid_, |
| 647 | supp_gids_.size(), !seclabel_.empty() ? seclabel_.c_str() : "default"); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 648 | } |
| 649 | |
| 650 | NotifyStateChange("running"); |
| 651 | return true; |
| 652 | } |
| 653 | |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 654 | bool Service::StartIfNotDisabled() { |
| 655 | if (!(flags_ & SVC_DISABLED)) { |
| 656 | return Start(); |
| 657 | } else { |
| 658 | flags_ |= SVC_DISABLED_START; |
| 659 | } |
| 660 | return true; |
| 661 | } |
| 662 | |
| 663 | bool Service::Enable() { |
| 664 | flags_ &= ~(SVC_DISABLED | SVC_RC_DISABLED); |
| 665 | if (flags_ & SVC_DISABLED_START) { |
| 666 | return Start(); |
| 667 | } |
| 668 | return true; |
| 669 | } |
| 670 | |
| 671 | void Service::Reset() { |
| 672 | StopOrReset(SVC_RESET); |
| 673 | } |
| 674 | |
| 675 | void Service::Stop() { |
| 676 | StopOrReset(SVC_DISABLED); |
| 677 | } |
| 678 | |
Bertrand SIMONNET | b7e03e8 | 2015-12-18 11:39:59 -0800 | [diff] [blame] | 679 | void Service::Terminate() { |
| 680 | flags_ &= ~(SVC_RESTARTING | SVC_DISABLED_START); |
| 681 | flags_ |= SVC_DISABLED; |
| 682 | if (pid_) { |
Elliott Hughes | ad8e94e | 2016-06-15 14:49:57 -0700 | [diff] [blame] | 683 | KillProcessGroup(SIGTERM); |
Bertrand SIMONNET | b7e03e8 | 2015-12-18 11:39:59 -0800 | [diff] [blame] | 684 | NotifyStateChange("stopping"); |
| 685 | } |
| 686 | } |
| 687 | |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 688 | void Service::Restart() { |
| 689 | if (flags_ & SVC_RUNNING) { |
| 690 | /* Stop, wait, then start the service. */ |
| 691 | StopOrReset(SVC_RESTART); |
| 692 | } else if (!(flags_ & SVC_RESTARTING)) { |
| 693 | /* Just start the service since it's not running. */ |
| 694 | Start(); |
| 695 | } /* else: Service is restarting anyways. */ |
| 696 | } |
| 697 | |
| 698 | void Service::RestartIfNeeded(time_t& process_needs_restart) { |
| 699 | time_t next_start_time = time_started_ + 5; |
| 700 | |
| 701 | if (next_start_time <= gettime()) { |
| 702 | flags_ &= (~SVC_RESTARTING); |
| 703 | Start(); |
| 704 | return; |
| 705 | } |
| 706 | |
| 707 | if ((next_start_time < process_needs_restart) || |
| 708 | (process_needs_restart == 0)) { |
| 709 | process_needs_restart = next_start_time; |
| 710 | } |
| 711 | } |
| 712 | |
Elliott Hughes | ad8e94e | 2016-06-15 14:49:57 -0700 | [diff] [blame] | 713 | // The how field should be either SVC_DISABLED, SVC_RESET, or SVC_RESTART. |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 714 | void Service::StopOrReset(int how) { |
Elliott Hughes | ad8e94e | 2016-06-15 14:49:57 -0700 | [diff] [blame] | 715 | // The service is still SVC_RUNNING until its process exits, but if it has |
| 716 | // already exited it shoudn't attempt a restart yet. |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 717 | flags_ &= ~(SVC_RESTARTING | SVC_DISABLED_START); |
| 718 | |
| 719 | if ((how != SVC_DISABLED) && (how != SVC_RESET) && (how != SVC_RESTART)) { |
Elliott Hughes | ad8e94e | 2016-06-15 14:49:57 -0700 | [diff] [blame] | 720 | // An illegal flag: default to SVC_DISABLED. |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 721 | how = SVC_DISABLED; |
| 722 | } |
Elliott Hughes | ad8e94e | 2016-06-15 14:49:57 -0700 | [diff] [blame] | 723 | |
| 724 | // If the service has not yet started, prevent it from auto-starting with its class. |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 725 | if (how == SVC_RESET) { |
| 726 | flags_ |= (flags_ & SVC_RC_DISABLED) ? SVC_DISABLED : SVC_RESET; |
| 727 | } else { |
| 728 | flags_ |= how; |
| 729 | } |
| 730 | |
| 731 | if (pid_) { |
Elliott Hughes | ad8e94e | 2016-06-15 14:49:57 -0700 | [diff] [blame] | 732 | KillProcessGroup(SIGKILL); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 733 | NotifyStateChange("stopping"); |
| 734 | } else { |
| 735 | NotifyStateChange("stopped"); |
| 736 | } |
| 737 | } |
| 738 | |
| 739 | void Service::ZapStdio() const { |
| 740 | int fd; |
| 741 | fd = open("/dev/null", O_RDWR); |
| 742 | dup2(fd, 0); |
| 743 | dup2(fd, 1); |
| 744 | dup2(fd, 2); |
| 745 | close(fd); |
| 746 | } |
| 747 | |
| 748 | void Service::OpenConsole() const { |
Viorel Suman | 70daa67 | 2016-03-21 10:08:07 +0200 | [diff] [blame] | 749 | int fd = open(console_.c_str(), O_RDWR); |
| 750 | if (fd == -1) fd = open("/dev/null", O_RDWR); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 751 | ioctl(fd, TIOCSCTTY, 0); |
| 752 | dup2(fd, 0); |
| 753 | dup2(fd, 1); |
| 754 | dup2(fd, 2); |
| 755 | close(fd); |
| 756 | } |
| 757 | |
| 758 | void Service::PublishSocket(const std::string& name, int fd) const { |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 759 | std::string key = StringPrintf(ANDROID_SOCKET_ENV_PREFIX "%s", name.c_str()); |
| 760 | std::string val = StringPrintf("%d", fd); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 761 | add_environment(key.c_str(), val.c_str()); |
| 762 | |
| 763 | /* make sure we don't close-on-exec */ |
| 764 | fcntl(fd, F_SETFD, 0); |
| 765 | } |
| 766 | |
| 767 | int ServiceManager::exec_count_ = 0; |
| 768 | |
| 769 | ServiceManager::ServiceManager() { |
| 770 | } |
| 771 | |
| 772 | ServiceManager& ServiceManager::GetInstance() { |
| 773 | static ServiceManager instance; |
| 774 | return instance; |
| 775 | } |
| 776 | |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 777 | void ServiceManager::AddService(std::unique_ptr<Service> service) { |
| 778 | Service* old_service = FindServiceByName(service->name()); |
| 779 | if (old_service) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 780 | LOG(ERROR) << "ignored duplicate definition of service '" << service->name() << "'"; |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 781 | return; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 782 | } |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 783 | services_.emplace_back(std::move(service)); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 784 | } |
| 785 | |
| 786 | Service* ServiceManager::MakeExecOneshotService(const std::vector<std::string>& args) { |
| 787 | // Parse the arguments: exec [SECLABEL [UID [GID]*] --] COMMAND ARGS... |
| 788 | // SECLABEL can be a - to denote default |
| 789 | std::size_t command_arg = 1; |
| 790 | for (std::size_t i = 1; i < args.size(); ++i) { |
| 791 | if (args[i] == "--") { |
| 792 | command_arg = i + 1; |
| 793 | break; |
| 794 | } |
| 795 | } |
| 796 | if (command_arg > 4 + NR_SVC_SUPP_GIDS) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 797 | LOG(ERROR) << "exec called with too many supplementary group ids"; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 798 | return nullptr; |
| 799 | } |
| 800 | |
| 801 | if (command_arg >= args.size()) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 802 | LOG(ERROR) << "exec called without command"; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 803 | return nullptr; |
| 804 | } |
| 805 | std::vector<std::string> str_args(args.begin() + command_arg, args.end()); |
| 806 | |
| 807 | exec_count_++; |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 808 | std::string name = StringPrintf("exec %d (%s)", exec_count_, str_args[0].c_str()); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 809 | unsigned flags = SVC_EXEC | SVC_ONESHOT; |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 810 | unsigned namespace_flags = 0; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 811 | |
| 812 | std::string seclabel = ""; |
| 813 | if (command_arg > 2 && args[1] != "-") { |
| 814 | seclabel = args[1]; |
| 815 | } |
| 816 | uid_t uid = 0; |
| 817 | if (command_arg > 3) { |
| 818 | uid = decode_uid(args[2].c_str()); |
| 819 | } |
| 820 | gid_t gid = 0; |
| 821 | std::vector<gid_t> supp_gids; |
| 822 | if (command_arg > 4) { |
| 823 | gid = decode_uid(args[3].c_str()); |
| 824 | std::size_t nr_supp_gids = command_arg - 1 /* -- */ - 4 /* exec SECLABEL UID GID */; |
| 825 | for (size_t i = 0; i < nr_supp_gids; ++i) { |
| 826 | supp_gids.push_back(decode_uid(args[4 + i].c_str())); |
| 827 | } |
| 828 | } |
| 829 | |
| 830 | std::unique_ptr<Service> svc_p(new Service(name, "default", flags, uid, gid, |
Jorge Lucangeli Obes | 1b3fa3d | 2016-04-21 15:35:09 -0700 | [diff] [blame] | 831 | supp_gids, namespace_flags, |
| 832 | seclabel, str_args)); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 833 | if (!svc_p) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 834 | LOG(ERROR) << "Couldn't allocate service for exec of '" << str_args[0] << "'"; |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 835 | return nullptr; |
| 836 | } |
| 837 | Service* svc = svc_p.get(); |
| 838 | services_.push_back(std::move(svc_p)); |
| 839 | |
| 840 | return svc; |
| 841 | } |
| 842 | |
| 843 | Service* ServiceManager::FindServiceByName(const std::string& name) const { |
| 844 | auto svc = std::find_if(services_.begin(), services_.end(), |
| 845 | [&name] (const std::unique_ptr<Service>& s) { |
| 846 | return name == s->name(); |
| 847 | }); |
| 848 | if (svc != services_.end()) { |
| 849 | return svc->get(); |
| 850 | } |
| 851 | return nullptr; |
| 852 | } |
| 853 | |
| 854 | Service* ServiceManager::FindServiceByPid(pid_t pid) const { |
| 855 | auto svc = std::find_if(services_.begin(), services_.end(), |
| 856 | [&pid] (const std::unique_ptr<Service>& s) { |
| 857 | return s->pid() == pid; |
| 858 | }); |
| 859 | if (svc != services_.end()) { |
| 860 | return svc->get(); |
| 861 | } |
| 862 | return nullptr; |
| 863 | } |
| 864 | |
| 865 | Service* ServiceManager::FindServiceByKeychord(int keychord_id) const { |
| 866 | auto svc = std::find_if(services_.begin(), services_.end(), |
| 867 | [&keychord_id] (const std::unique_ptr<Service>& s) { |
| 868 | return s->keychord_id() == keychord_id; |
| 869 | }); |
| 870 | |
| 871 | if (svc != services_.end()) { |
| 872 | return svc->get(); |
| 873 | } |
| 874 | return nullptr; |
| 875 | } |
| 876 | |
Chih-Hung Hsieh | 8f7b9e3 | 2016-07-27 16:25:51 -0700 | [diff] [blame] | 877 | void ServiceManager::ForEachService(const std::function<void(Service*)>& callback) const { |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 878 | for (const auto& s : services_) { |
Bertrand SIMONNET | b7e03e8 | 2015-12-18 11:39:59 -0800 | [diff] [blame] | 879 | callback(s.get()); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 880 | } |
| 881 | } |
| 882 | |
| 883 | void ServiceManager::ForEachServiceInClass(const std::string& classname, |
| 884 | void (*func)(Service* svc)) const { |
| 885 | for (const auto& s : services_) { |
| 886 | if (classname == s->classname()) { |
| 887 | func(s.get()); |
| 888 | } |
| 889 | } |
| 890 | } |
| 891 | |
| 892 | void ServiceManager::ForEachServiceWithFlags(unsigned matchflags, |
| 893 | void (*func)(Service* svc)) const { |
| 894 | for (const auto& s : services_) { |
| 895 | if (s->flags() & matchflags) { |
| 896 | func(s.get()); |
| 897 | } |
| 898 | } |
| 899 | } |
| 900 | |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 901 | void ServiceManager::RemoveService(const Service& svc) { |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 902 | auto svc_it = std::find_if(services_.begin(), services_.end(), |
| 903 | [&svc] (const std::unique_ptr<Service>& s) { |
| 904 | return svc.name() == s->name(); |
| 905 | }); |
| 906 | if (svc_it == services_.end()) { |
| 907 | return; |
| 908 | } |
| 909 | |
| 910 | services_.erase(svc_it); |
| 911 | } |
| 912 | |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 913 | void ServiceManager::DumpState() const { |
| 914 | for (const auto& s : services_) { |
| 915 | s->DumpState(); |
| 916 | } |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 917 | } |
| 918 | |
Bertrand SIMONNET | b7e03e8 | 2015-12-18 11:39:59 -0800 | [diff] [blame] | 919 | bool ServiceManager::ReapOneProcess() { |
| 920 | int status; |
| 921 | pid_t pid = TEMP_FAILURE_RETRY(waitpid(-1, &status, WNOHANG)); |
| 922 | if (pid == 0) { |
| 923 | return false; |
| 924 | } else if (pid == -1) { |
Elliott Hughes | f86b5a6 | 2016-06-24 15:12:21 -0700 | [diff] [blame] | 925 | PLOG(ERROR) << "waitpid failed"; |
Bertrand SIMONNET | b7e03e8 | 2015-12-18 11:39:59 -0800 | [diff] [blame] | 926 | return false; |
| 927 | } |
| 928 | |
| 929 | Service* svc = FindServiceByPid(pid); |
| 930 | |
| 931 | std::string name; |
| 932 | if (svc) { |
| 933 | name = android::base::StringPrintf("Service '%s' (pid %d)", |
| 934 | svc->name().c_str(), pid); |
| 935 | } else { |
| 936 | name = android::base::StringPrintf("Untracked pid %d", pid); |
| 937 | } |
| 938 | |
| 939 | if (WIFEXITED(status)) { |
Wei Wang | a285dac | 2016-10-04 14:05:39 -0700 | [diff] [blame] | 940 | LOG(INFO) << name << " exited with status " << WEXITSTATUS(status); |
Bertrand SIMONNET | b7e03e8 | 2015-12-18 11:39:59 -0800 | [diff] [blame] | 941 | } else if (WIFSIGNALED(status)) { |
Wei Wang | a285dac | 2016-10-04 14:05:39 -0700 | [diff] [blame] | 942 | LOG(INFO) << name << " killed by signal " << WTERMSIG(status); |
Bertrand SIMONNET | b7e03e8 | 2015-12-18 11:39:59 -0800 | [diff] [blame] | 943 | } else if (WIFSTOPPED(status)) { |
Wei Wang | a285dac | 2016-10-04 14:05:39 -0700 | [diff] [blame] | 944 | LOG(INFO) << name << " stopped by signal " << WSTOPSIG(status); |
Bertrand SIMONNET | b7e03e8 | 2015-12-18 11:39:59 -0800 | [diff] [blame] | 945 | } else { |
Wei Wang | a285dac | 2016-10-04 14:05:39 -0700 | [diff] [blame] | 946 | LOG(INFO) << name << " state changed"; |
Bertrand SIMONNET | b7e03e8 | 2015-12-18 11:39:59 -0800 | [diff] [blame] | 947 | } |
| 948 | |
| 949 | if (!svc) { |
| 950 | return true; |
| 951 | } |
| 952 | |
| 953 | if (svc->Reap()) { |
| 954 | waiting_for_exec = false; |
| 955 | RemoveService(*svc); |
| 956 | } |
| 957 | |
| 958 | return true; |
| 959 | } |
| 960 | |
| 961 | void ServiceManager::ReapAnyOutstandingChildren() { |
| 962 | while (ReapOneProcess()) { |
| 963 | } |
| 964 | } |
| 965 | |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 966 | bool ServiceParser::ParseSection(const std::vector<std::string>& args, |
| 967 | std::string* err) { |
| 968 | if (args.size() < 3) { |
| 969 | *err = "services must have a name and a program"; |
| 970 | return false; |
| 971 | } |
| 972 | |
| 973 | const std::string& name = args[1]; |
| 974 | if (!IsValidName(name)) { |
| 975 | *err = StringPrintf("invalid service name '%s'", name.c_str()); |
| 976 | return false; |
| 977 | } |
| 978 | |
| 979 | std::vector<std::string> str_args(args.begin() + 2, args.end()); |
| 980 | service_ = std::make_unique<Service>(name, "default", str_args); |
| 981 | return true; |
| 982 | } |
| 983 | |
| 984 | bool ServiceParser::ParseLineSection(const std::vector<std::string>& args, |
| 985 | const std::string& filename, int line, |
| 986 | std::string* err) const { |
Jorge Lucangeli Obes | 177b27d | 2016-06-29 14:32:49 -0400 | [diff] [blame] | 987 | return service_ ? service_->ParseLine(args, err) : false; |
Tom Cherry | b734990 | 2015-08-26 11:43:36 -0700 | [diff] [blame] | 988 | } |
| 989 | |
| 990 | void ServiceParser::EndSection() { |
| 991 | if (service_) { |
| 992 | ServiceManager::GetInstance().AddService(std::move(service_)); |
| 993 | } |
| 994 | } |
| 995 | |
| 996 | bool ServiceParser::IsValidName(const std::string& name) const { |
Iliyan Malchev | f655480 | 2016-09-19 20:58:20 -0700 | [diff] [blame] | 997 | return is_legal_property_name("init.svc." + name); |
Tom Cherry | bac3299 | 2015-07-31 12:45:25 -0700 | [diff] [blame] | 998 | } |