1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
/* $Id$ */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include "osdep.h"
#ifdef NEED_SETPROCTITLE
static char **Argv = NULL; /* pointer to argument vector */
static int argv_size; /* end of argv */
extern char **environ;
void
initsetproctitle(int argc, char **argv, char **envp)
{
int i, len=0, nenv=0;
/*
* Move the environment so setproctitle can use the space at the top of
* memory.
*/
for (i = 0; envp[i]; i++)
len += strlen(envp[i])+1;
nenv = i + 1;
len += sizeof(char*)*nenv;
environ = malloc(len);
len = 0;
for (i = 0; envp[i]; i++) {
environ[i] = (char*)environ + nenv * sizeof(char*) + len;
strcpy(environ[i], envp[i]);
len += strlen(envp[i])+1;
}
environ[i] = NULL;
/* Save start and extent of argv for setproctitle. */
Argv = argv;
if (i > 0)
argv_size = envp[i - 1] + strlen(envp[i - 1]) - Argv[0];
else
argv_size = argv[argc - 1] + strlen(argv[argc - 1]) - Argv[0];
}
static void
do_setproctitle(const char *cmdline)
{
int len;
len = strlen(cmdline) + 1; // +1 for '\0'
if (len > argv_size - 2) // 2 ??
len = argv_size - 2;
memset(Argv[0], 0, argv_size);
strlcpy(Argv[0], cmdline, len);
Argv[1] = NULL;
}
void
setproctitle(const char *format,...)
{
char buf[256];
va_list args;
va_start(args, format);
vsnprintf(buf, sizeof(buf), format, args);
do_setproctitle(buf);
va_end(args);
}
#endif
|
