21#include <sys/socket.h>
31class ProcessImpl :
public Process
34 int spawn(
const std::string& new_exe_name,
const fs::path& argv0_path,
int& pid)
override
37 fs::path path = argv0_path;
38 path.remove_filename();
44 bool checkSpawned(
int argc,
char* argv[],
int& fd)
override
49 if (argc != 3 || strcmp(argv[1],
"-ipcfd") != 0) {
58 const auto maybe_fd{ToIntegral<int32_t>(argv[2])};
60 throw std::runtime_error(
strprintf(
"Invalid -ipcfd number '%s'", argv[2]));
65 int connect(
const fs::path& data_dir,
66 const std::string& dest_exe_name,
67 std::string& address)
override;
68 int bind(
const fs::path& data_dir,
const std::string& exe_name, std::string& address)
override;
71static bool ParseAddress(std::string& address,
72 const fs::path& data_dir,
73 const std::string& dest_exe_name,
74 struct sockaddr_un& addr,
77 if (address ==
"unix" || address.starts_with(
"unix:")) {
79 if (address.size() <= 5) {
86 if (path_str.size() >=
sizeof(addr.sun_path)) {
87 error =
strprintf(
"Unix address path %s exceeded maximum socket path length", fs::quoted(fs::PathToString(path)));
90 memset(&addr, 0,
sizeof(addr));
91 addr.sun_family = AF_UNIX;
92 strncpy(addr.sun_path, path_str.c_str(),
sizeof(addr.sun_path)-1);
96 error =
strprintf(
"Unrecognized address '%s'", address);
100int ProcessImpl::connect(
const fs::path& data_dir,
101 const std::string& dest_exe_name,
102 std::string& address)
104 struct sockaddr_un addr;
106 if (!ParseAddress(address, data_dir, dest_exe_name, addr, error)) {
107 throw std::invalid_argument(error);
111 if ((fd = ::socket(addr.sun_family, SOCK_STREAM, 0)) == -1) {
112 throw std::system_error(errno, std::system_category());
114 if (::connect(fd, (
struct sockaddr*)&addr,
sizeof(addr)) == 0) {
117 int connect_error = errno;
118 if (::close(fd) != 0) {
121 throw std::system_error(connect_error, std::system_category());
124int ProcessImpl::bind(
const fs::path& data_dir,
const std::string& exe_name, std::string& address)
126 struct sockaddr_un addr;
128 if (!ParseAddress(address, data_dir, exe_name, addr, error)) {
129 throw std::invalid_argument(error);
132 if (addr.sun_family == AF_UNIX) {
133 fs::path path = addr.sun_path;
135 if (fs::symlink_status(path).type() == fs::file_type::socket) {
141 if ((fd = ::socket(addr.sun_family, SOCK_STREAM, 0)) == -1) {
142 throw std::system_error(errno, std::system_category());
145 if (::bind(fd, (
struct sockaddr*)&addr,
sizeof(addr)) == 0) {
148 int bind_error = errno;
149 if (::close(fd) != 0) {
152 throw std::system_error(bind_error, std::system_category());
156std::unique_ptr<Process>
MakeProcess() {
return std::make_unique<ProcessImpl>(); }
static bool create_directories(const std::filesystem::path &p)
Create directory (and if necessary its parents), unless the leaf directory already exists or is a sym...
static std::string PathToString(const path &path)
Convert path object to a byte string.
static path PathFromString(const std::string &string)
Convert byte string to path object.
std::unique_ptr< Process > MakeProcess()
Constructor for Process interface.
int WaitProcess(int pid)
Wait for a process to exit and return its exit code.
int SpawnProcess(int &pid, FdToArgsFn &&fd_to_args)
Spawn a new process that communicates with the current process over a socket pair.
std::string_view RemovePrefixView(std::string_view str, std::string_view prefix)
std::string SysErrorString(int err)
Return system error string from errno value.