fune/ipc/glue/SetProcessTitle.cpp
Jed Davis df9fa90f41 Bug 1851095 - Step 2: Connect setproctitle to the fork server. r=nika
This adds a small function that concatenates the args (because not
only does the BSD `setproctitle` take a single string, but also the
Linux kernel code that implements `/proc/{pid}/cmdline` won't allow the
"arguments" to extend past their original length unless it's a single
string), and connects it to the fork server.

Differential Revision: https://phabricator.services.mozilla.com/D187635
2023-09-22 03:41:39 +00:00

51 lines
1.2 KiB
C++

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "mozilla/ipc/SetProcessTitle.h"
#include "nsString.h"
#ifdef XP_LINUX
# include "base/set_process_title_linux.h"
# define HAVE_SETPROCTITLE
# define HAVE_SETPROCTITLE_INIT
#elif defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__) || \
defined(__DragonFly__)
# include <sys/types.h>
# include <unistd.h>
# define HAVE_SETPROCTITLE
#endif
namespace mozilla {
void SetProcessTitle(const std::vector<std::string>& aNewArgv) {
#ifdef HAVE_SETPROCTITLE
nsAutoCStringN<1024> buf;
bool firstArg = true;
for (const std::string& arg : aNewArgv) {
if (firstArg) {
firstArg = false;
} else {
buf.Append(' ');
}
buf.Append(arg.c_str());
}
setproctitle("-%s", buf.get());
#endif
}
void SetProcessTitleInit(char** aOrigArgv) {
#ifdef HAVE_SETPROCTITLE_INIT
setproctitle_init(aOrigArgv);
#endif
}
} // namespace mozilla