100 lines
2.7 KiB
C++
100 lines
2.7 KiB
C++
/*
|
|
* Copyright (c) 2021, Andreas Kling <kling@serenityos.org>
|
|
* Copyright (c) 2022, MacDue <macdue@dueutil.tech>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <AK/String.h>
|
|
#include <AK/Vector.h>
|
|
#include <LibCore/Process.h>
|
|
#include <LibCore/System.h>
|
|
#include <errno.h>
|
|
#include <spawn.h>
|
|
|
|
#ifdef AK_OS_SERENITY
|
|
# include <serenity.h>
|
|
#endif
|
|
|
|
extern char** environ;
|
|
|
|
namespace Core {
|
|
|
|
struct ArgvList {
|
|
String m_path;
|
|
String m_working_directory;
|
|
Vector<char const*, 10> m_argv;
|
|
|
|
ArgvList(String path, size_t size)
|
|
: m_path { path }
|
|
{
|
|
m_argv.ensure_capacity(size + 2);
|
|
m_argv.append(m_path.characters());
|
|
}
|
|
|
|
void append(char const* arg)
|
|
{
|
|
m_argv.append(arg);
|
|
}
|
|
|
|
Span<char const*> get()
|
|
{
|
|
if (m_argv.is_empty() || m_argv.last() != nullptr)
|
|
m_argv.append(nullptr);
|
|
return m_argv;
|
|
}
|
|
|
|
void set_working_directory(String const& working_directory)
|
|
{
|
|
m_working_directory = working_directory;
|
|
}
|
|
|
|
ErrorOr<pid_t> spawn()
|
|
{
|
|
#ifdef AK_OS_SERENITY
|
|
posix_spawn_file_actions_t spawn_actions;
|
|
posix_spawn_file_actions_init(&spawn_actions);
|
|
if (!m_working_directory.is_empty())
|
|
posix_spawn_file_actions_addchdir(&spawn_actions, m_working_directory.characters());
|
|
|
|
auto pid = TRY(System::posix_spawn(m_path.view(), &spawn_actions, nullptr, const_cast<char**>(get().data()), environ));
|
|
TRY(System::disown(pid));
|
|
#else
|
|
auto pid = TRY(System::posix_spawn(m_path.view(), nullptr, nullptr, const_cast<char**>(get().data()), environ));
|
|
#endif
|
|
return pid;
|
|
}
|
|
};
|
|
|
|
ErrorOr<pid_t> Process::spawn(StringView path, Span<String const> arguments, String working_directory)
|
|
{
|
|
ArgvList argv { path, arguments.size() };
|
|
for (auto const& arg : arguments)
|
|
argv.append(arg.characters());
|
|
argv.set_working_directory(working_directory);
|
|
return argv.spawn();
|
|
}
|
|
|
|
ErrorOr<pid_t> Process::spawn(StringView path, Span<StringView const> arguments, String working_directory)
|
|
{
|
|
Vector<String> backing_strings;
|
|
backing_strings.ensure_capacity(arguments.size());
|
|
ArgvList argv { path, arguments.size() };
|
|
for (auto const& arg : arguments) {
|
|
backing_strings.append(arg);
|
|
argv.append(backing_strings.last().characters());
|
|
}
|
|
argv.set_working_directory(working_directory);
|
|
return argv.spawn();
|
|
}
|
|
|
|
ErrorOr<pid_t> Process::spawn(StringView path, Span<char const* const> arguments, String working_directory)
|
|
{
|
|
ArgvList argv { path, arguments.size() };
|
|
for (auto arg : arguments)
|
|
argv.append(arg);
|
|
argv.set_working_directory(working_directory);
|
|
return argv.spawn();
|
|
}
|
|
|
|
}
|