Merge "init: Add support for ambient capabilities." (043bc971) · Commits · e / os / android_system_core

init/Android.mk

+5 −3

Original line number	Diff line number	Diff line
		@@ -45,6 +45,7 @@ include $(CLEAR_VARS)
		LOCAL_CPPFLAGS := $(init_cflags)
		LOCAL_SRC_FILES:= \
		action.cpp \
		capabilities.cpp \
		import_parser.cpp \
		init_parser.cpp \
		log.cpp \
		@@ -53,6 +54,7 @@ LOCAL_SRC_FILES:= \
		util.cpp \

		LOCAL_STATIC_LIBRARIES := libbase libselinux liblog libprocessgroup
		LOCAL_WHOLE_STATIC_LIBRARIES := libcap
		LOCAL_MODULE := libinit
		LOCAL_SANITIZE := integer
		LOCAL_CLANG := true
		@@ -102,7 +104,7 @@ LOCAL_STATIC_LIBRARIES := \
		libz \
		libprocessgroup

		# Create symlinks
		# Create symlinks.
		LOCAL_POST_INSTALL_CMD := $(hide) mkdir -p $(TARGET_ROOT_OUT)/sbin; \
		ln -sf ../init $(TARGET_ROOT_OUT)/sbin/ueventd; \
		ln -sf ../init $(TARGET_ROOT_OUT)/sbin/watchdogd
		@@ -112,8 +114,8 @@ LOCAL_CLANG := true
		include $(BUILD_EXECUTABLE)




		# Unit tests.
		# =========================================================
		include $(CLEAR_VARS)
		LOCAL_MODULE := init_tests
		LOCAL_SRC_FILES := \

init/capabilities.cpp

0 → 100644

+166 −0

Original line number	Diff line number	Diff line
		// Copyright (C) 2016 The Android Open Source Project
		//
		// Licensed under the Apache License, Version 2.0 (the "License");
		// you may not use this file except in compliance with the License.
		// You may obtain a copy of the License at
		//
		// http://www.apache.org/licenses/LICENSE-2.0
		//
		// Unless required by applicable law or agreed to in writing, software
		// distributed under the License is distributed on an "AS IS" BASIS,
		// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
		// See the License for the specific language governing permissions and
		// limitations under the License.

		#include "capabilities.h"

		#include <sys/capability.h>
		#include <sys/prctl.h>

		#include <map>
		#include <memory>

		#include <android-base/logging.h>
		#include <android-base/macros.h>

		#define CAP_MAP_ENTRY(cap) { #cap, CAP_##cap }

		namespace {
		const std::map<std::string, int> cap_map = {
		CAP_MAP_ENTRY(CHOWN),
		CAP_MAP_ENTRY(DAC_OVERRIDE),
		CAP_MAP_ENTRY(DAC_READ_SEARCH),
		CAP_MAP_ENTRY(FOWNER),
		CAP_MAP_ENTRY(FSETID),
		CAP_MAP_ENTRY(KILL),
		CAP_MAP_ENTRY(SETGID),
		CAP_MAP_ENTRY(SETUID),
		CAP_MAP_ENTRY(SETPCAP),
		CAP_MAP_ENTRY(LINUX_IMMUTABLE),
		CAP_MAP_ENTRY(NET_BIND_SERVICE),
		CAP_MAP_ENTRY(NET_BROADCAST),
		CAP_MAP_ENTRY(NET_ADMIN),
		CAP_MAP_ENTRY(NET_RAW),
		CAP_MAP_ENTRY(IPC_LOCK),
		CAP_MAP_ENTRY(IPC_OWNER),
		CAP_MAP_ENTRY(SYS_MODULE),
		CAP_MAP_ENTRY(SYS_RAWIO),
		CAP_MAP_ENTRY(SYS_CHROOT),
		CAP_MAP_ENTRY(SYS_PTRACE),
		CAP_MAP_ENTRY(SYS_PACCT),
		CAP_MAP_ENTRY(SYS_ADMIN),
		CAP_MAP_ENTRY(SYS_BOOT),
		CAP_MAP_ENTRY(SYS_NICE),
		CAP_MAP_ENTRY(SYS_RESOURCE),
		CAP_MAP_ENTRY(SYS_TIME),
		CAP_MAP_ENTRY(SYS_TTY_CONFIG),
		CAP_MAP_ENTRY(MKNOD),
		CAP_MAP_ENTRY(LEASE),
		CAP_MAP_ENTRY(AUDIT_WRITE),
		CAP_MAP_ENTRY(AUDIT_CONTROL),
		CAP_MAP_ENTRY(SETFCAP),
		CAP_MAP_ENTRY(MAC_OVERRIDE),
		CAP_MAP_ENTRY(MAC_ADMIN),
		CAP_MAP_ENTRY(SYSLOG),
		CAP_MAP_ENTRY(WAKE_ALARM),
		CAP_MAP_ENTRY(BLOCK_SUSPEND),
		CAP_MAP_ENTRY(AUDIT_READ),
		};

		static_assert(CAP_LAST_CAP == CAP_AUDIT_READ, "CAP_LAST_CAP is not CAP_AUDIT_READ");

		bool DropBoundingSet(const CapSet& to_keep) {
		for (size_t cap = 0; cap < to_keep.size(); ++cap) {
		if (to_keep.test(cap)) {
		// No need to drop this capability.
		continue;
		}
		if (cap_drop_bound(cap) == -1) {
		PLOG(ERROR) << "cap_drop_bound(" << cap << ") failed";
		return false;
		}
		}
		return true;
		}

		bool SetProcCaps(const CapSet& to_keep, bool add_setpcap) {
		cap_t caps = cap_init();
		auto deleter = [](cap_t* p) { cap_free(*p); };
		std::unique_ptr<cap_t, decltype(deleter)> ptr_caps(&caps, deleter);

		cap_clear(caps);
		cap_value_t value[1];
		for (size_t cap = 0; cap <= to_keep.size(); ++cap) {
		if (to_keep.test(cap)) {
		value[0] = cap;
		if (cap_set_flag(caps, CAP_INHERITABLE, arraysize(value), value, CAP_SET) != 0 \|\|
		cap_set_flag(caps, CAP_PERMITTED, arraysize(value), value, CAP_SET) != 0) {
		PLOG(ERROR) << "cap_set_flag(INHERITABLE\|PERMITTED, " << cap << ") failed";
		return false;
		}
		}
		}

		if (add_setpcap) {
		value[0] = CAP_SETPCAP;
		if (cap_set_flag(caps, CAP_PERMITTED, arraysize(value), value, CAP_SET) != 0 \|\|
		cap_set_flag(caps, CAP_EFFECTIVE, arraysize(value), value, CAP_SET) != 0) {
		PLOG(ERROR) << "cap_set_flag(PERMITTED\|EFFECTIVE, " << CAP_SETPCAP << ") failed";
		return false;
		}
		}

		if (cap_set_proc(caps) != 0) {
		PLOG(ERROR) << "cap_set_proc(" << to_keep.to_ulong() << ") failed";
		return false;
		}
		return true;
		}

		bool SetAmbientCaps(const CapSet& to_raise) {
		for (size_t cap = 0; cap < to_raise.size(); ++cap) {
		if (to_raise.test(cap)) {
		if (prctl(PR_CAP_AMBIENT, PR_CAP_AMBIENT_RAISE, cap, 0, 0) != 0) {
		PLOG(ERROR) << "prctl(PR_CAP_AMBIENT, PR_CAP_AMBIENT_RAISE, " << cap << ") failed";
		return false;
		}
		}
		}
		return true;
		}

		} // namespace anonymous

		int LookupCap(const std::string& cap_name) {
		auto e = cap_map.find(cap_name);
		if (e != cap_map.end()) {
		return e->second;
		} else {
		return -1;
		}
		}

		bool SetCapsForExec(const CapSet& to_keep) {
		// Need to keep SETPCAP to drop bounding set below.
		bool add_setpcap = true;
		if (!SetProcCaps(to_keep, add_setpcap)) {
		LOG(ERROR) << "failed to apply initial capset";
		return false;
		}

		if (!DropBoundingSet(to_keep)) {
		return false;
		}

		// If SETPCAP wasn't specifically requested, drop it now.
		add_setpcap = false;
		if (!SetProcCaps(to_keep, add_setpcap)) {
		LOG(ERROR) << "failed to apply final capset";
		return false;
		}

		// Add the capabilities to the ambient set so that they are preserved across
		// execve(2).
		// See http://man7.org/linux/man-pages/man7/capabilities.7.html.
		return SetAmbientCaps(to_keep);
		}

init/capabilities.h

0 → 100644

+23 −0

Original line number	Diff line number	Diff line
		// Copyright (C) 2016 The Android Open Source Project
		//
		// Licensed under the Apache License, Version 2.0 (the "License");
		// you may not use this file except in compliance with the License.
		// You may obtain a copy of the License at
		//
		// http://www.apache.org/licenses/LICENSE-2.0
		//
		// Unless required by applicable law or agreed to in writing, software
		// distributed under the License is distributed on an "AS IS" BASIS,
		// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
		// See the License for the specific language governing permissions and
		// limitations under the License.

		#include <linux/capability.h>

		#include <bitset>
		#include <string>

		using CapSet = std::bitset<CAP_LAST_CAP + 1>;

		int LookupCap(const std::string& cap_name);
		bool SetCapsForExec(const CapSet& to_keep);

init/readme.txt

+12 −4

Original line number	Diff line number	Diff line
		@@ -149,25 +149,33 @@ socket <name> <type> <perm> [ <user> [ <group> [ <seclabel> ] ] ]
		computed based on the service executable file security context.

		user <username>
		Change to username before exec'ing this service.
		Change to 'username' before exec'ing this service.
		Currently defaults to root. (??? probably should default to nobody)
		As of Android M, processes should use this option even if they
		require linux capabilities. Previously, to acquire linux
		require Linux capabilities. Previously, to acquire Linux
		capabilities, a process would need to run as root, request the
		capabilities, then drop to its desired uid. There is a new
		mechanism through fs_config that allows device manufacturers to add
		linux capabilities to specific binaries on a file system that should
		Linux capabilities to specific binaries on a file system that should
		be used instead. This mechanism is described on
		http://source.android.com/devices/tech/config/filesystem.html. When
		using this new mechanism, processes can use the user option to
		select their desired uid without ever running as root.
		As of Android O, processes can also request capabilities directly in their .rc
		files. See the "capabilities" option below.

		group <groupname> [ <groupname> ]*
		Change to groupname before exec'ing this service. Additional
		Change to 'groupname' before exec'ing this service. Additional
		groupnames beyond the (required) first one are used to set the
		supplemental groups of the process (via setgroups()).
		Currently defaults to root. (??? probably should default to nobody)

		capabilities <capability> [ <capability> ]*
		Set capabilities when exec'ing this service. 'capability' should be a Linux
		capability without the "CAP_" prefix, like "NET_ADMIN" or "SETPCAP". See
		http://man7.org/linux/man-pages/man7/capabilities.7.html for a list of Linux
		capabilities.

		seclabel <seclabel>
		Change to 'seclabel' before exec'ing this service.
		Primarily for use by services run from the rootfs, e.g. ueventd, adbd.

init/service.cpp

+40 −7

Original line number	Diff line number	Diff line
		@@ -17,6 +17,7 @@
		#include "service.h"

		#include <fcntl.h>
		#include <linux/securebits.h>
		#include <sched.h>
		#include <sys/mount.h>
		#include <sys/prctl.h>
		@@ -171,14 +172,16 @@ Service::Service(const std::string& name, const std::string& classname,

		Service::Service(const std::string& name, const std::string& classname,
		unsigned flags, uid_t uid, gid_t gid,
		const std::vector<gid_t>& supp_gids, unsigned namespace_flags,
		const std::vector<gid_t>& supp_gids,
		const CapSet& capabilities, unsigned namespace_flags,
		const std::string& seclabel,
		const std::vector<std::string>& args)
		: name_(name), classname_(classname), flags_(flags), pid_(0),
		time_started_(0), time_crashed_(0), nr_crashed_(0), uid_(uid), gid_(gid),
		supp_gids_(supp_gids), namespace_flags_(namespace_flags),
		seclabel_(seclabel), ioprio_class_(IoSchedClass_NONE), ioprio_pri_(0),
		priority_(0), oom_score_adjust_(-1000), args_(args) {
		supp_gids_(supp_gids), capabilities_(capabilities),
		namespace_flags_(namespace_flags), seclabel_(seclabel),
		ioprio_class_(IoSchedClass_NONE), ioprio_pri_(0), priority_(0),
		oom_score_adjust_(-1000), args_(args) {
		onrestart_.InitSingleTrigger("onrestart");
		}

		@@ -225,6 +228,13 @@ void Service::CreateSockets(const std::string& context) {
		}

		void Service::SetProcessAttributes() {
		// Keep capabilites on uid change.
		if (capabilities_.any() && uid_) {
		if (prctl(PR_SET_SECUREBITS, SECBIT_KEEP_CAPS \| SECBIT_KEEP_CAPS_LOCKED) != 0) {
		PLOG(FATAL) << "prtcl(PR_SET_KEEPCAPS) failed for " << name_;
		}
		}

		// TODO: work out why this fails for `console` then upgrade to FATAL.
		if (setpgid(0, getpid()) == -1) PLOG(ERROR) << "setpgid failed for " << name_;

		@@ -251,6 +261,11 @@ void Service::SetProcessAttributes() {
		PLOG(FATAL) << "setpriority failed for " << name_;
		}
		}
		if (capabilities_.any()) {
		if (!SetCapsForExec(capabilities_)) {
		LOG(FATAL) << "cannot set capabilities for " << name_;
		}
		}
		}

		bool Service::Reap() {
		@@ -320,6 +335,21 @@ void Service::DumpState() const {
		}
		}

		bool Service::ParseCapabilities(const std::vector<std::string>& args, std::string* err) {
		capabilities_ = 0;

		for (size_t i = 1; i < args.size(); i++) {
		const std::string& arg = args[i];
		int cap = LookupCap(arg);
		if (cap == -1) {
		*err = StringPrintf("invalid capability '%s'", arg.c_str());
		return false;
		}
		capabilities_[cap] = true;
		}
		return true;
		}

		bool Service::ParseClass(const std::vector<std::string>& args, std::string* err) {
		classname_ = args[1];
		return true;
		@@ -476,6 +506,8 @@ private:
		Service::OptionParserMap::Map& Service::OptionParserMap::map() const {
		constexpr std::size_t kMax = std::numeric_limits<std::size_t>::max();
		static const Map option_parsers = {
		{"capabilities",
		{1, kMax, &Service::ParseCapabilities}},
		{"class", {1, 1, &Service::ParseClass}},
		{"console", {0, 1, &Service::ParseConsole}},
		{"critical", {0, 0, &Service::ParseCritical}},
		@@ -807,6 +839,7 @@ Service* ServiceManager::MakeExecOneshotService(const std::vector<std::string>&
		exec_count_++;
		std::string name = StringPrintf("exec %d (%s)", exec_count_, str_args[0].c_str());
		unsigned flags = SVC_EXEC \| SVC_ONESHOT;
		CapSet no_capabilities;
		unsigned namespace_flags = 0;

		std::string seclabel = "";
		@@ -827,9 +860,9 @@ Service* ServiceManager::MakeExecOneshotService(const std::vector<std::string>&
		}
		}

		std::unique_ptr<Service> svc_p(new Service(name, "default", flags, uid, gid,
		supp_gids, namespace_flags,
		seclabel, str_args));
		std::unique_ptr<Service> svc_p(new Service(name, "default", flags, uid, gid, supp_gids,
		no_capabilities, namespace_flags, seclabel,
		str_args));
		if (!svc_p) {
		LOG(ERROR) << "Couldn't allocate service for exec of '" << str_args[0] << "'";
		return nullptr;