mirror of
https://github.com/SerenityOS/serenity
synced 2024-07-21 01:55:07 +00:00
00b897af8f
`Process::get_name()` and `Process::set_name()` are basically the same as `get_process_name()` and `set_process_name()`, except making use of convenient Serenity standard types and returning ErrorOr, instead of char* and errno shenanigans. `Process::set_name()` has an optional `SetThreadName` parameter, for when you also want to set the thread's name to the same thing. That's true for the two places that use `set_process_name()`.
138 lines
4 KiB
C++
138 lines
4 KiB
C++
/*
|
|
* Copyright (c) 2021, Andreas Kling <kling@serenityos.org>
|
|
* Copyright (c) 2022, MacDue <macdue@dueutil.tech>
|
|
* Copyright (c) 2023, Sam Atkins <atkinssj@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <AK/DeprecatedString.h>
|
|
#include <AK/String.h>
|
|
#include <AK/Vector.h>
|
|
#include <LibCore/Process.h>
|
|
#include <LibCore/System.h>
|
|
#include <errno.h>
|
|
#include <spawn.h>
|
|
#include <unistd.h>
|
|
|
|
#ifdef AK_OS_SERENITY
|
|
# include <serenity.h>
|
|
# include <syscall.h>
|
|
#endif
|
|
|
|
extern char** environ;
|
|
|
|
namespace Core {
|
|
|
|
struct ArgvList {
|
|
DeprecatedString m_path;
|
|
DeprecatedString m_working_directory;
|
|
Vector<char const*, 10> m_argv;
|
|
|
|
ArgvList(DeprecatedString 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(DeprecatedString 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<DeprecatedString const> arguments, DeprecatedString 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, DeprecatedString working_directory)
|
|
{
|
|
Vector<DeprecatedString> 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, DeprecatedString working_directory)
|
|
{
|
|
ArgvList argv { path, arguments.size() };
|
|
for (auto arg : arguments)
|
|
argv.append(arg);
|
|
argv.set_working_directory(working_directory);
|
|
return argv.spawn();
|
|
}
|
|
|
|
ErrorOr<String> Process::get_name()
|
|
{
|
|
#if defined(AK_OS_SERENITY)
|
|
char buffer[BUFSIZ];
|
|
int rc = get_process_name(buffer, BUFSIZ);
|
|
if (rc != 0)
|
|
return Error::from_syscall("get_process_name"sv, -rc);
|
|
return String::from_utf8(StringView { buffer, strlen(buffer) });
|
|
#else
|
|
// FIXME: Implement Process::get_name() for other platforms.
|
|
return String::from_utf8_short_string("???"sv);
|
|
#endif
|
|
}
|
|
|
|
ErrorOr<void> Process::set_name([[maybe_unused]] StringView name, [[maybe_unused]] SetThreadName set_thread_name)
|
|
{
|
|
#if defined(AK_OS_SERENITY)
|
|
int rc = set_process_name(name.characters_without_null_termination(), name.length());
|
|
if (rc != 0)
|
|
return Error::from_syscall("set_process_name"sv, -rc);
|
|
if (set_thread_name == SetThreadName::No)
|
|
return {};
|
|
|
|
rc = syscall(SC_set_thread_name, gettid(), name.characters_without_null_termination(), name.length());
|
|
if (rc != 0)
|
|
return Error::from_syscall("set_thread_name"sv, -rc);
|
|
return {};
|
|
#else
|
|
// FIXME: Implement Process::set_name() for other platforms.
|
|
return {};
|
|
#endif
|
|
}
|
|
|
|
}
|