blob: 1fcd7a974f43dc7c3536db2249f549f1c94db5eb [file] [log] [blame]
// Copyright 2018 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef SHILL_EXTERNAL_TASK_H_
#define SHILL_EXTERNAL_TASK_H_
#include <sys/types.h>
#include <map>
#include <memory>
#include <string>
#include <vector>
#include <base/callback.h>
#include <base/files/file_path.h>
#include <base/memory/weak_ptr.h>
#include <gtest/gtest_prod.h> // for FRIEND_TEST
#include "shill/rpc_task.h"
namespace shill {
class ControlInterface;
class Error;
class ProcessManager;
class ExternalTask : public RpcTaskDelegate {
public:
// |death_callback| will be called when the corresponding task is
// terminated. By the time it is called, this ExternalTask will already be
// Stop()'d. Clients are free to destruct the ExternalTask within the death
// callback.
ExternalTask(ControlInterface* control,
ProcessManager* process_manager,
const base::WeakPtr<RpcTaskDelegate>& task_delegate,
const base::Callback<void(pid_t, int)>& death_callback);
~ExternalTask() override;
// Forks off a process to run |program|, with the command-line
// arguments |arguments|, and the environment variables specified in
// |environment|.
//
// If |terminate_with_parent| is true, the child process will be
// configured to terminate itself if this process dies. Otherwise,
// the child process will retain its default behavior.
//
// On success, returns true, and leaves |error| unmodified.
// On failure, returns false, and sets |error|.
//
// |environment| SHOULD NOT contain kRpcTaskServiceVariable or
// kRpcTaskPathVariable, as that may prevent the child process
// from communicating back to the ExternalTask.
virtual bool Start(const base::FilePath& program,
const std::vector<std::string>& arguments,
const std::map<std::string, std::string>& environment,
bool terminate_with_parent,
Error* error);
// Forks off a process to run |program|, with the command-line arguments
// |arguments|. Takes RPC identifiers that would be passed as environment
// variables and passes them on the command line instead, since minijail does
// not support the setting of custom environment variables for a spawned
// program. |inherit_supplementary_groups| indicates whether the child
// child program should be spawned with the programatic equivalent of the
// minijail -G flag. |close_nonstd_fds| indicates that non-standard file
// descriptors should be closed so they cannot be inherited by the child
// process.
//
// On success, returns true, and leaves |error| unmodified.
// On failure, returns false, and sets |error|.
virtual bool StartInMinijail(const base::FilePath& program,
std::vector<std::string>* arguments,
const std::string user,
const std::string group,
uint64_t mask,
bool inherit_supplementary_groups,
bool close_nonstd_fds,
Error* error);
virtual void Stop();
private:
friend class ExternalTaskTest;
FRIEND_TEST(ExternalTaskTest, Destructor);
FRIEND_TEST(ExternalTaskTest, GetLogin);
FRIEND_TEST(ExternalTaskTest, Notify);
FRIEND_TEST(ExternalTaskTest, OnTaskDied);
FRIEND_TEST(ExternalTaskTest, Start);
FRIEND_TEST(ExternalTaskTest, Stop);
FRIEND_TEST(ExternalTaskTest, StopNotStarted);
// Implements RpcTaskDelegate.
void GetLogin(std::string* user, std::string* password) override;
void Notify(const std::string& event,
const std::map<std::string, std::string>& details) override;
// Called when the external process exits.
void OnTaskDied(int exit_status);
ControlInterface* control_;
ProcessManager* process_manager_;
std::unique_ptr<RpcTask> rpc_task_;
base::WeakPtr<RpcTaskDelegate> task_delegate_;
base::Callback<void(pid_t, int)> death_callback_;
// The PID of the spawned process. May be 0 if no process has been
// spawned yet or the process has died.
pid_t pid_;
DISALLOW_COPY_AND_ASSIGN(ExternalTask);
};
} // namespace shill
#endif // SHILL_EXTERNAL_TASK_H_