init/subcontext.cpp - platform/system/core - Gitiles

 /*
  * Copyright (C) 2017 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 "subcontext.h"

 #include <fcntl.h>
 #include <poll.h>
 #include <sys/socket.h>
 #include <unistd.h>

 #include <android-base/file.h>
 #include <android-base/logging.h>
 #include <android-base/strings.h>
 #include <selinux/android.h>

 #include "action.h"
 #include "util.h"

 #if defined(__ANDROID__)
 #include "property_service.h"
 #include "selinux.h"
 #else
 #include "host_init_stubs.h"
 #endif

 using android::base::GetExecutablePath;
 using android::base::Join;
 using android::base::Socketpair;
 using android::base::Split;
 using android::base::StartsWith;
 using android::base::unique_fd;

 namespace android {
 namespace init {

 const std::string kInitContext = "u:r:init:s0";
 const std::string kVendorContext = "u:r:vendor_init:s0";

 const char* const paths_and_secontexts[2][2] = {
     {"/vendor", kVendorContext.c_str()},
     {"/odm", kVendorContext.c_str()},
 };

 namespace {

 constexpr size_t kBufferSize = 4096;

 Result<std::string> ReadMessage(int socket) {
     char buffer[kBufferSize] = {};
     auto result = TEMP_FAILURE_RETRY(recv(socket, buffer, sizeof(buffer), 0));
     if (result <= 0) {
         return ErrnoError();
     }
     return std::string(buffer, result);
 }

 template <typename T>
 Result<Success> SendMessage(int socket, const T& message) {
     std::string message_string;
     if (!message.SerializeToString(&message_string)) {
         return Error() << "Unable to serialize message";
     }

     if (message_string.size() > kBufferSize) {
         return Error() << "Serialized message too long to send";
     }

     if (auto result =
             TEMP_FAILURE_RETRY(send(socket, message_string.c_str(), message_string.size(), 0));
         result != static_cast<long>(message_string.size())) {
         return ErrnoError() << "send() failed to send message contents";
     }
     return Success();
 }

 std::vector<std::pair<std::string, std::string>> properties_to_set;

 uint32_t SubcontextPropertySet(const std::string& name, const std::string& value) {
     properties_to_set.emplace_back(name, value);
     return 0;
 }

 class SubcontextProcess {
   public:
     SubcontextProcess(const KeywordFunctionMap* function_map, std::string context, int init_fd)
         : function_map_(function_map), context_(std::move(context)), init_fd_(init_fd){};
     void MainLoop();

   private:
     void RunCommand(const SubcontextCommand::ExecuteCommand& execute_command,
                     SubcontextReply* reply) const;
     void ExpandArgs(const SubcontextCommand::ExpandArgsCommand& expand_args_command,
                     SubcontextReply* reply) const;

     const KeywordFunctionMap* function_map_;
     const std::string context_;
     const int init_fd_;
 };

 void SubcontextProcess::RunCommand(const SubcontextCommand::ExecuteCommand& execute_command,
                                    SubcontextReply* reply) const {
     // Need to use ArraySplice instead of this code.
     auto args = std::vector<std::string>();
     for (const auto& string : execute_command.args()) {
         args.emplace_back(string);
     }

     auto map_result = function_map_->FindFunction(args);
     Result<Success> result;
     if (!map_result) {
         result = Error() << "Cannot find command: " << map_result.error();
     } else {
         result = RunBuiltinFunction(map_result->second, args, context_);
     }

     for (const auto& [name, value] : properties_to_set) {
         auto property = reply->add_properties_to_set();
         property->set_name(name);
         property->set_value(value);
     }

     properties_to_set.clear();

     if (result) {
         reply->set_success(true);
     } else {
         auto* failure = reply->mutable_failure();
         failure->set_error_string(result.error_string());
         failure->set_error_errno(result.error_errno());
     }
 }

 void SubcontextProcess::ExpandArgs(const SubcontextCommand::ExpandArgsCommand& expand_args_command,
                                    SubcontextReply* reply) const {
     for (const auto& arg : expand_args_command.args()) {
         auto expanded_prop = std::string{};
         if (!expand_props(arg, &expanded_prop)) {
             auto* failure = reply->mutable_failure();
             failure->set_error_string("Failed to expand '" + arg + "'");
             failure->set_error_errno(0);
             return;
         } else {
             auto* expand_args_reply = reply->mutable_expand_args_reply();
             expand_args_reply->add_expanded_args(expanded_prop);
         }
     }
 }

 void SubcontextProcess::MainLoop() {
     pollfd ufd[1];
     ufd[0].events = POLLIN;
     ufd[0].fd = init_fd_;

     while (true) {
         ufd[0].revents = 0;
         int nr = TEMP_FAILURE_RETRY(poll(ufd, arraysize(ufd), -1));
         if (nr == 0) continue;
         if (nr < 0) {
             PLOG(FATAL) << "poll() of subcontext socket failed, continuing";
         }

         auto init_message = ReadMessage(init_fd_);
         if (!init_message) {
             LOG(FATAL) << "Could not read message from init: " << init_message.error();
         }

         auto subcontext_command = SubcontextCommand();
         if (!subcontext_command.ParseFromString(*init_message)) {
             LOG(FATAL) << "Unable to parse message from init";
         }

         auto reply = SubcontextReply();
         switch (subcontext_command.command_case()) {
             case SubcontextCommand::kExecuteCommand: {
                 RunCommand(subcontext_command.execute_command(), &reply);
                 break;
             }
             case SubcontextCommand::kExpandArgsCommand: {
                 ExpandArgs(subcontext_command.expand_args_command(), &reply);
                 break;
             }
             default:
                 LOG(FATAL) << "Unknown message type from init: "
                            << subcontext_command.command_case();
         }

         if (auto result = SendMessage(init_fd_, reply); !result) {
             LOG(FATAL) << "Failed to send message to init: " << result.error();
         }
     }
 }

 }  // namespace

 int SubcontextMain(int argc, char** argv, const KeywordFunctionMap* function_map) {
     if (argc < 4) LOG(FATAL) << "Fewer than 4 args specified to subcontext (" << argc << ")";

     auto context = std::string(argv[2]);
     auto init_fd = std::atoi(argv[3]);

     SelabelInitialize();

     property_set = SubcontextPropertySet;

     auto subcontext_process = SubcontextProcess(function_map, context, init_fd);
     subcontext_process.MainLoop();
     return 0;
 }

 void Subcontext::Fork() {
     unique_fd subcontext_socket;
     if (!Socketpair(AF_UNIX, SOCK_SEQPACKET | SOCK_CLOEXEC, 0, &socket_, &subcontext_socket)) {
         LOG(FATAL) << "Could not create socket pair to communicate to subcontext";
         return;
     }

     auto result = fork();

     if (result == -1) {
         LOG(FATAL) << "Could not fork subcontext";
     } else if (result == 0) {
         socket_.reset();

         // We explicitly do not use O_CLOEXEC here, such that we can reference this FD by number
         // in the subcontext process after we exec.
         int child_fd = dup(subcontext_socket);
         if (child_fd < 0) {
             PLOG(FATAL) << "Could not dup child_fd";
         }

         if (setexeccon(context_.c_str()) < 0) {
             PLOG(FATAL) << "Could not set execcon for '" << context_ << "'";
         }

         auto init_path = GetExecutablePath();
         auto child_fd_string = std::to_string(child_fd);
         const char* args[] = {init_path.c_str(), "subcontext", context_.c_str(),
                               child_fd_string.c_str(), nullptr};
         execv(init_path.data(), const_cast<char**>(args));

         PLOG(FATAL) << "Could not execv subcontext init";
     } else {
         subcontext_socket.reset();
         pid_ = result;
         LOG(INFO) << "Forked subcontext for '" << context_ << "' with pid " << pid_;
     }
 }

 void Subcontext::Restart() {
     LOG(ERROR) << "Restarting subcontext '" << context_ << "'";
     if (pid_) {
         kill(pid_, SIGKILL);
     }
     pid_ = 0;
     socket_.reset();
     Fork();
 }

 Result<SubcontextReply> Subcontext::TransmitMessage(const SubcontextCommand& subcontext_command) {
     if (auto result = SendMessage(socket_, subcontext_command); !result) {
         Restart();
         return ErrnoError() << "Failed to send message to subcontext";
     }

     auto subcontext_message = ReadMessage(socket_);
     if (!subcontext_message) {
         Restart();
         return Error() << "Failed to receive result from subcontext: " << subcontext_message.error();
     }

     auto subcontext_reply = SubcontextReply{};
     if (!subcontext_reply.ParseFromString(*subcontext_message)) {
         Restart();
         return Error() << "Unable to parse message from subcontext";
     }
     return subcontext_reply;
 }

 Result<Success> Subcontext::Execute(const std::vector<std::string>& args) {
     auto subcontext_command = SubcontextCommand();
     std::copy(
         args.begin(), args.end(),
         RepeatedPtrFieldBackInserter(subcontext_command.mutable_execute_command()->mutable_args()));

     auto subcontext_reply = TransmitMessage(subcontext_command);
     if (!subcontext_reply) {
         return subcontext_reply.error();
     }

     for (const auto& property : subcontext_reply->properties_to_set()) {
         ucred cr = {.pid = pid_, .uid = 0, .gid = 0};
         std::string error;
         if (HandlePropertySet(property.name(), property.value(), context_, cr, &error) != 0) {
             LOG(ERROR) << "Subcontext init could not set '" << property.name() << "' to '"
                        << property.value() << "': " << error;
         }
     }

     if (subcontext_reply->reply_case() == SubcontextReply::kFailure) {
         auto& failure = subcontext_reply->failure();
         return ResultError(failure.error_string(), failure.error_errno());
     }

     if (subcontext_reply->reply_case() != SubcontextReply::kSuccess) {
         return Error() << "Unexpected message type from subcontext: "
                        << subcontext_reply->reply_case();
     }

     return Success();
 }

 Result<std::vector<std::string>> Subcontext::ExpandArgs(const std::vector<std::string>& args) {
     auto subcontext_command = SubcontextCommand{};
     std::copy(args.begin(), args.end(),
               RepeatedPtrFieldBackInserter(
                   subcontext_command.mutable_expand_args_command()->mutable_args()));

     auto subcontext_reply = TransmitMessage(subcontext_command);
     if (!subcontext_reply) {
         return subcontext_reply.error();
     }

     if (subcontext_reply->reply_case() == SubcontextReply::kFailure) {
         auto& failure = subcontext_reply->failure();
         return ResultError(failure.error_string(), failure.error_errno());
     }

     if (subcontext_reply->reply_case() != SubcontextReply::kExpandArgsReply) {
         return Error() << "Unexpected message type from subcontext: "
                        << subcontext_reply->reply_case();
     }

     auto& reply = subcontext_reply->expand_args_reply();
     auto expanded_args = std::vector<std::string>{};
     for (const auto& string : reply.expanded_args()) {
         expanded_args.emplace_back(string);
     }
     return expanded_args;
 }

 static std::vector<Subcontext> subcontexts;

 std::vector<Subcontext>* InitializeSubcontexts() {
     for (const auto& [path_prefix, secontext] : paths_and_secontexts) {
         subcontexts.emplace_back(path_prefix, secontext);
     }
     return &subcontexts;
 }

 bool SubcontextChildReap(pid_t pid) {
     for (auto& subcontext : subcontexts) {
         if (subcontext.pid() == pid) {
             subcontext.Restart();
             return true;
         }
     }
     return false;
 }

 }  // namespace init
 }  // namespace android
	/*
	* Copyright (C) 2017 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 "subcontext.h"

	#include <fcntl.h>
	#include <poll.h>
	#include <sys/socket.h>
	#include <unistd.h>

	#include <android-base/file.h>
	#include <android-base/logging.h>
	#include <android-base/strings.h>
	#include <selinux/android.h>

	#include "action.h"
	#include "util.h"

	#if defined(__ANDROID__)
	#include "property_service.h"
	#include "selinux.h"
	#else
	#include "host_init_stubs.h"
	#endif

	using android::base::GetExecutablePath;
	using android::base::Join;
	using android::base::Socketpair;
	using android::base::Split;
	using android::base::StartsWith;
	using android::base::unique_fd;

	namespace android {
	namespace init {

	const std::string kInitContext = "u:r:init:s0";
	const std::string kVendorContext = "u:r:vendor_init:s0";

	const char* const paths_and_secontexts[2][2] = {
	{"/vendor", kVendorContext.c_str()},
	{"/odm", kVendorContext.c_str()},
	};

	namespace {

	constexpr size_t kBufferSize = 4096;

	Result<std::string> ReadMessage(int socket) {
	char buffer[kBufferSize] = {};
	auto result = TEMP_FAILURE_RETRY(recv(socket, buffer, sizeof(buffer), 0));
	if (result <= 0) {
	return ErrnoError();
	}
	return std::string(buffer, result);
	}

	template <typename T>
	Result<Success> SendMessage(int socket, const T& message) {
	std::string message_string;
	if (!message.SerializeToString(&message_string)) {
	return Error() << "Unable to serialize message";
	}

	if (message_string.size() > kBufferSize) {
	return Error() << "Serialized message too long to send";
	}

	if (auto result =
	TEMP_FAILURE_RETRY(send(socket, message_string.c_str(), message_string.size(), 0));
	result != static_cast<long>(message_string.size())) {
	return ErrnoError() << "send() failed to send message contents";
	}
	return Success();
	}

	std::vector<std::pair<std::string, std::string>> properties_to_set;

	uint32_t SubcontextPropertySet(const std::string& name, const std::string& value) {
	properties_to_set.emplace_back(name, value);
	return 0;
	}

	class SubcontextProcess {
	public:
	SubcontextProcess(const KeywordFunctionMap* function_map, std::string context, int init_fd)
	: function_map_(function_map), context_(std::move(context)), init_fd_(init_fd){};
	void MainLoop();

	private:
	void RunCommand(const SubcontextCommand::ExecuteCommand& execute_command,
	SubcontextReply* reply) const;
	void ExpandArgs(const SubcontextCommand::ExpandArgsCommand& expand_args_command,
	SubcontextReply* reply) const;

	const KeywordFunctionMap* function_map_;
	const std::string context_;
	const int init_fd_;
	};

	void SubcontextProcess::RunCommand(const SubcontextCommand::ExecuteCommand& execute_command,
	SubcontextReply* reply) const {
	// Need to use ArraySplice instead of this code.
	auto args = std::vector<std::string>();
	for (const auto& string : execute_command.args()) {
	args.emplace_back(string);
	}

	auto map_result = function_map_->FindFunction(args);
	Result<Success> result;
	if (!map_result) {
	result = Error() << "Cannot find command: " << map_result.error();
	} else {
	result = RunBuiltinFunction(map_result->second, args, context_);
	}

	for (const auto& [name, value] : properties_to_set) {
	auto property = reply->add_properties_to_set();
	property->set_name(name);
	property->set_value(value);
	}

	properties_to_set.clear();

	if (result) {
	reply->set_success(true);
	} else {
	auto* failure = reply->mutable_failure();
	failure->set_error_string(result.error_string());
	failure->set_error_errno(result.error_errno());
	}
	}

	void SubcontextProcess::ExpandArgs(const SubcontextCommand::ExpandArgsCommand& expand_args_command,
	SubcontextReply* reply) const {
	for (const auto& arg : expand_args_command.args()) {
	auto expanded_prop = std::string{};
	if (!expand_props(arg, &expanded_prop)) {
	auto* failure = reply->mutable_failure();
	failure->set_error_string("Failed to expand '" + arg + "'");
	failure->set_error_errno(0);
	return;
	} else {
	auto* expand_args_reply = reply->mutable_expand_args_reply();
	expand_args_reply->add_expanded_args(expanded_prop);
	}
	}
	}

	void SubcontextProcess::MainLoop() {
	pollfd ufd[1];
	ufd[0].events = POLLIN;
	ufd[0].fd = init_fd_;

	while (true) {
	ufd[0].revents = 0;
	int nr = TEMP_FAILURE_RETRY(poll(ufd, arraysize(ufd), -1));
	if (nr == 0) continue;
	if (nr < 0) {
	PLOG(FATAL) << "poll() of subcontext socket failed, continuing";
	}

	auto init_message = ReadMessage(init_fd_);
	if (!init_message) {
	LOG(FATAL) << "Could not read message from init: " << init_message.error();
	}

	auto subcontext_command = SubcontextCommand();
	if (!subcontext_command.ParseFromString(*init_message)) {
	LOG(FATAL) << "Unable to parse message from init";
	}

	auto reply = SubcontextReply();
	switch (subcontext_command.command_case()) {
	case SubcontextCommand::kExecuteCommand: {
	RunCommand(subcontext_command.execute_command(), &reply);
	break;
	}
	case SubcontextCommand::kExpandArgsCommand: {
	ExpandArgs(subcontext_command.expand_args_command(), &reply);
	break;
	}
	default:
	LOG(FATAL) << "Unknown message type from init: "
	<< subcontext_command.command_case();
	}

	if (auto result = SendMessage(init_fd_, reply); !result) {
	LOG(FATAL) << "Failed to send message to init: " << result.error();
	}
	}
	}

	} // namespace

	int SubcontextMain(int argc, char** argv, const KeywordFunctionMap* function_map) {
	if (argc < 4) LOG(FATAL) << "Fewer than 4 args specified to subcontext (" << argc << ")";

	auto context = std::string(argv[2]);
	auto init_fd = std::atoi(argv[3]);

	SelabelInitialize();

	property_set = SubcontextPropertySet;

	auto subcontext_process = SubcontextProcess(function_map, context, init_fd);
	subcontext_process.MainLoop();
	return 0;
	}

	void Subcontext::Fork() {
	unique_fd subcontext_socket;
	if (!Socketpair(AF_UNIX, SOCK_SEQPACKET \| SOCK_CLOEXEC, 0, &socket_, &subcontext_socket)) {
	LOG(FATAL) << "Could not create socket pair to communicate to subcontext";
	return;
	}

	auto result = fork();

	if (result == -1) {
	LOG(FATAL) << "Could not fork subcontext";
	} else if (result == 0) {
	socket_.reset();

	// We explicitly do not use O_CLOEXEC here, such that we can reference this FD by number
	// in the subcontext process after we exec.
	int child_fd = dup(subcontext_socket);
	if (child_fd < 0) {
	PLOG(FATAL) << "Could not dup child_fd";
	}

	if (setexeccon(context_.c_str()) < 0) {
	PLOG(FATAL) << "Could not set execcon for '" << context_ << "'";
	}

	auto init_path = GetExecutablePath();
	auto child_fd_string = std::to_string(child_fd);
	const char* args[] = {init_path.c_str(), "subcontext", context_.c_str(),
	child_fd_string.c_str(), nullptr};
	execv(init_path.data(), const_cast<char**>(args));

	PLOG(FATAL) << "Could not execv subcontext init";
	} else {
	subcontext_socket.reset();
	pid_ = result;
	LOG(INFO) << "Forked subcontext for '" << context_ << "' with pid " << pid_;
	}
	}

	void Subcontext::Restart() {
	LOG(ERROR) << "Restarting subcontext '" << context_ << "'";
	if (pid_) {
	kill(pid_, SIGKILL);
	}
	pid_ = 0;
	socket_.reset();
	Fork();
	}

	Result<SubcontextReply> Subcontext::TransmitMessage(const SubcontextCommand& subcontext_command) {
	if (auto result = SendMessage(socket_, subcontext_command); !result) {
	Restart();
	return ErrnoError() << "Failed to send message to subcontext";
	}

	auto subcontext_message = ReadMessage(socket_);
	if (!subcontext_message) {
	Restart();
	return Error() << "Failed to receive result from subcontext: " << subcontext_message.error();
	}

	auto subcontext_reply = SubcontextReply{};
	if (!subcontext_reply.ParseFromString(*subcontext_message)) {
	Restart();
	return Error() << "Unable to parse message from subcontext";
	}
	return subcontext_reply;
	}

	Result<Success> Subcontext::Execute(const std::vector<std::string>& args) {
	auto subcontext_command = SubcontextCommand();
	std::copy(
	args.begin(), args.end(),
	RepeatedPtrFieldBackInserter(subcontext_command.mutable_execute_command()->mutable_args()));

	auto subcontext_reply = TransmitMessage(subcontext_command);
	if (!subcontext_reply) {
	return subcontext_reply.error();
	}

	for (const auto& property : subcontext_reply->properties_to_set()) {
	ucred cr = {.pid = pid_, .uid = 0, .gid = 0};
	std::string error;
	if (HandlePropertySet(property.name(), property.value(), context_, cr, &error) != 0) {
	LOG(ERROR) << "Subcontext init could not set '" << property.name() << "' to '"
	<< property.value() << "': " << error;
	}
	}

	if (subcontext_reply->reply_case() == SubcontextReply::kFailure) {
	auto& failure = subcontext_reply->failure();
	return ResultError(failure.error_string(), failure.error_errno());
	}

	if (subcontext_reply->reply_case() != SubcontextReply::kSuccess) {
	return Error() << "Unexpected message type from subcontext: "
	<< subcontext_reply->reply_case();
	}

	return Success();
	}

	Result<std::vector<std::string>> Subcontext::ExpandArgs(const std::vector<std::string>& args) {
	auto subcontext_command = SubcontextCommand{};
	std::copy(args.begin(), args.end(),
	RepeatedPtrFieldBackInserter(
	subcontext_command.mutable_expand_args_command()->mutable_args()));

	auto subcontext_reply = TransmitMessage(subcontext_command);
	if (!subcontext_reply) {
	return subcontext_reply.error();
	}

	if (subcontext_reply->reply_case() == SubcontextReply::kFailure) {
	auto& failure = subcontext_reply->failure();
	return ResultError(failure.error_string(), failure.error_errno());
	}

	if (subcontext_reply->reply_case() != SubcontextReply::kExpandArgsReply) {
	return Error() << "Unexpected message type from subcontext: "
	<< subcontext_reply->reply_case();
	}

	auto& reply = subcontext_reply->expand_args_reply();
	auto expanded_args = std::vector<std::string>{};
	for (const auto& string : reply.expanded_args()) {
	expanded_args.emplace_back(string);
	}
	return expanded_args;
	}

	static std::vector<Subcontext> subcontexts;

	std::vector<Subcontext>* InitializeSubcontexts() {
	for (const auto& [path_prefix, secontext] : paths_and_secontexts) {
	subcontexts.emplace_back(path_prefix, secontext);
	}
	return &subcontexts;
	}

	bool SubcontextChildReap(pid_t pid) {
	for (auto& subcontext : subcontexts) {
	if (subcontext.pid() == pid) {
	subcontext.Restart();
	return true;
	}
	}
	return false;
	}

	} // namespace init
	} // namespace android