/*
* Copyright (C) Igor Sysoev
* Copyright (C) NGINX, Inc.
*/
#include <nxt_main.h>
/* The arguments passed to main(). */
char **nxt_process_argv;
/*
* MacOSX environ(7):
*
* Shared libraries and bundles don't have direct access to environ,
* which is only available to the loader ld(1) when a complete program
* is being linked.
*
* So nxt_process_environ contains an address of environ to allow
* change environ[] placement.
*/
char ***nxt_process_environ;
#if (NXT_SETPROCTITLE_ARGV)
/*
* A process title on Linux, Solaris, and MacOSX can be changed by
* copying a new title to a place where the program argument argv[0]
* points originally to. However, the argv[0] may be too small to hold
* the new title. Fortunately, these OSes place the program argument
* argv[] strings and the environment environ[] strings contiguously
* and their space can be used for the long new process title.
*
* Solaris "ps" command shows the new title only if it is run in
* UCB mode: either "/usr/ucb/ps -axwww" or "/usr/bin/ps axwww".
*/
static u_char *nxt_process_title_start;
static u_char *nxt_process_title_end;
void
nxt_process_arguments(nxt_task_t *task, char **orig_argv, char ***orig_envp)
{
u_char *p, *end, *argv_end, **argv, **env;
size_t size, argv_size, environ_size, strings_size;
nxt_uint_t i;
nxt_process_argv = orig_argv;
nxt_process_environ = orig_envp;
if (orig_envp == NULL) {
return;
}
/*
* Set a conservative title space for a case if program argument
* strings and environment strings are not contiguous.
*/
argv = (u_char **) orig_argv;
nxt_process_title_start = argv[0];
nxt_process_title_end = argv[0] + nxt_strlen(argv[0]);
end = argv[0];
strings_size = 0;
argv_size = sizeof(void *);
for (i = 0; argv[i] != NULL; i++) {
argv_size += sizeof(void *);
if (argv[i] == end) {
/* Argument strings are contiguous. */
size = nxt_strlen(argv[i]) + 1;
strings_size += size;
end = argv[i] + size;
}
}
argv = nxt_malloc(argv_size);
if (argv == NULL) {
return;
}
/*
* Copy the entire original argv[] array. The elements of this array
* can point to copied strings or if original argument strings are not
* contiguous, to the original argument strings.
*/
nxt_memcpy(argv, orig_argv, argv_size);
/*
* The argv[1] must be set to NULL on Solaris otherwise the "ps"
* command outputs strings pointed by original argv[] elements.
* The original argv[] array has always at least two elements so
* it is safe to set argv[1].
*/
orig_argv[1] = NULL;
nxt_process_argv = (char **) argv;
argv_end = end;
env = (u_char **) *orig_envp;
environ_size = sizeof(void *);
for (i = 0; env[i] != NULL; i++) {
environ_size += sizeof(void *);
if (env[i] == end) {
/* Environment strings are contiguous. */
size = nxt_strlen(env[i]) + 1;
strings_size += size;
end = env[i] + size;
}
}
p = nxt_malloc(strings_size);
if (p == NULL) {
return;
}
if (argv_end == end) {
/*
* There is no reason to modify environ if arguments
* and environment are not contiguous.
*/
nxt_debug(task, "arguments and environment are not contiguous");
goto done;
}
end = argv[0];
for (i = 0; argv[i] != NULL; i++) {
if (argv[i] != end) {
/* Argument strings are not contiguous. */
goto done;
}
size = nxt_strlen(argv[i]) + 1;
nxt_memcpy(p, argv[i], size);
end = argv[i] + size;
argv[i] = p;
p += size;
}
env = nxt_malloc(environ_size);
if (env == NULL) {
return;
}
/*
* Copy the entire original environ[] array. The elements of
* this array can point to copied strings or if original environ
* strings are not contiguous, to the original environ strings.
*/
nxt_memcpy(env, *orig_envp, environ_size);
/* Set the global environ variable to the new array. */
*orig_envp = (char **) env;
for (i = 0; env[i] != NULL; i++) {
if (env[i] != end) {
/* Environment strings are not contiguous. */
goto done;
}
size = nxt_strlen(env[i]) + 1;
nxt_memcpy(p, env[i], size);
end = env[i] + size;
env[i] = p;
p += size;
}
done:
/* Preserve space for the trailing zero. */
end--;
nxt_process_title_end = end;
}
void
nxt_process_title(nxt_task_t *task, const char *fmt, ...)
{
u_char *p, *start, *end;
va_list args;
start = nxt_process_title_start;
if (start == NULL) {
return;
}
end = nxt_process_title_end;
va_start(args, fmt);
p = nxt_vsprintf(start, end, fmt, args);
va_end(args);
#if (NXT_SOLARIS)
/*
* Solaris "ps" command shows a new process title only if it is
* longer than original command line. A simple workaround is just
* to append the original command line in parenthesis to the title.
*/
{
size_t size;
nxt_uint_t i;
size = 0;
for (i = 0; nxt_process_argv[i] != NULL; i++) {
size += nxt_strlen(nxt_process_argv[i]);
}
if (size > (size_t) (p - start)) {
p = nxt_sprintf(p, end, " (");
for (i = 0; nxt_process_argv[i] != NULL; i++) {
p = nxt_sprintf(p, end, "%s ", nxt_process_argv[i]);
}
if (*(p - 1) == ' ') {
*(p - 1) = ')';
}
}
}
#endif
/*
* A process title must be padded with zeros on MacOSX. Otherwise
* the "ps" command may output parts of environment strings.
*/
nxt_memset(p, '\0', end - p);
nxt_debug(task, "setproctitle: \"%s\"", start);
}
#else /* !(NXT_SETPROCTITLE_ARGV) */
void
nxt_process_arguments(nxt_task_t *task, char **orig_argv, char ***orig_envp)
{
nxt_process_argv = orig_argv;
nxt_process_environ = orig_envp;
}
#endif
