2
|
1
|
|
2 /*
|
|
3 * Copyright (C) Igor Sysoev
|
|
4 */
|
|
5
|
|
6
|
|
7 #include <ngx_config.h>
|
|
8 #include <ngx_core.h>
|
|
9
|
|
10
|
|
11 #if (NGX_SETPROCTITLE_USES_ENV)
|
|
12
|
|
13 /*
|
|
14 * To change the process title in Linux and Solaris we have to set argv[1]
|
|
15 * to NULL and to copy the title to the same place where the argv[0] points to.
|
|
16 * However, argv[0] may be too small to hold a new title. Fortunately, Linux
|
|
17 * and Solaris store argv[] and environ[] one after another. So we should
|
|
18 * ensure that is the continuous memory and then we allocate the new memory
|
|
19 * for environ[] and copy it. After this we could use the memory starting
|
|
20 * from argv[0] for our process title.
|
|
21 *
|
|
22 * The Solaris's standard /bin/ps does not show the changed process title.
|
|
23 * You have to use "/usr/ucb/ps -w" instead. Besides, the UCB ps dos not
|
|
24 * show a new title if its length less than the origin command line length.
|
|
25 * To avoid it we append to a new title the origin command line in the
|
|
26 * parenthesis.
|
|
27 */
|
|
28
|
|
29 extern char **environ;
|
|
30
|
|
31 static char *ngx_os_argv_last;
|
|
32
|
42
|
33 ngx_int_t
|
|
34 ngx_init_setproctitle(ngx_log_t *log)
|
2
|
35 {
|
42
|
36 u_char *p;
|
2
|
37 size_t size;
|
|
38 ngx_uint_t i;
|
|
39
|
|
40 size = 0;
|
|
41
|
|
42 for (i = 0; environ[i]; i++) {
|
|
43 size += ngx_strlen(environ[i]) + 1;
|
|
44 }
|
|
45
|
|
46 if (!(p = ngx_alloc(size, log))) {
|
|
47 return NGX_ERROR;
|
|
48 }
|
|
49
|
|
50 ngx_os_argv_last = ngx_os_argv[0];
|
|
51
|
|
52 for (i = 0; ngx_os_argv[i]; i++) {
|
|
53 if (ngx_os_argv_last == ngx_os_argv[i]) {
|
|
54 ngx_os_argv_last = ngx_os_argv[i] + ngx_strlen(ngx_os_argv[i]) + 1;
|
|
55 }
|
|
56 }
|
|
57
|
|
58 for (i = 0; environ[i]; i++) {
|
|
59 if (ngx_os_argv_last == environ[i]) {
|
|
60
|
|
61 size = ngx_strlen(environ[i]) + 1;
|
|
62 ngx_os_argv_last = environ[i] + size;
|
|
63
|
42
|
64 ngx_cpystrn(p, (u_char *) environ[i], size);
|
|
65 environ[i] = (char *) p;
|
2
|
66 p += size;
|
|
67 }
|
|
68 }
|
|
69
|
|
70 ngx_os_argv_last--;
|
|
71
|
|
72 return NGX_OK;
|
|
73 }
|
|
74
|
|
75
|
42
|
76 void
|
|
77 ngx_setproctitle(char *title)
|
2
|
78 {
|
|
79 u_char *p;
|
|
80
|
4
|
81 #if (NGX_SOLARIS)
|
2
|
82
|
|
83 ngx_int_t i;
|
|
84 size_t size;
|
|
85
|
|
86 #endif
|
|
87
|
|
88 ngx_os_argv[1] = NULL;
|
|
89
|
|
90 p = ngx_cpystrn((u_char *) ngx_os_argv[0], "nginx: ",
|
|
91 ngx_os_argv_last - ngx_os_argv[0]);
|
|
92
|
|
93 p = ngx_cpystrn(p, (u_char *) title, ngx_os_argv_last - (char *) p);
|
|
94
|
4
|
95 #if (NGX_SOLARIS)
|
2
|
96
|
|
97 size = 0;
|
|
98
|
|
99 for (i = 0; i < ngx_argc; i++) {
|
|
100 size += ngx_strlen(ngx_argv[i]) + 1;
|
|
101 }
|
|
102
|
|
103 if (size > (size_t) ((char *) p - ngx_os_argv[0])) {
|
|
104
|
|
105 /*
|
|
106 * ngx_setproctitle() is too rare operation so we use
|
|
107 * the non-optimized copies
|
|
108 */
|
|
109
|
|
110 p = ngx_cpystrn(p, (u_char *) " (", ngx_os_argv_last - (char *) p);
|
|
111
|
|
112 for (i = 0; i < ngx_argc; i++) {
|
|
113 p = ngx_cpystrn(p, (u_char *) ngx_argv[i],
|
|
114 ngx_os_argv_last - (char *) p);
|
|
115 p = ngx_cpystrn(p, (u_char *) " ", ngx_os_argv_last - (char *) p);
|
|
116 }
|
|
117
|
|
118 if (*(p - 1) == ' ') {
|
|
119 *(p - 1) = ')';
|
|
120 }
|
|
121 }
|
|
122
|
|
123 #endif
|
|
124
|
|
125 if (ngx_os_argv_last - (char *) p) {
|
|
126 ngx_memset(p, NGX_SETPROCTITLE_PAD, ngx_os_argv_last - (char *) p);
|
|
127 }
|
|
128
|
|
129 ngx_log_debug1(NGX_LOG_DEBUG_CORE, ngx_cycle->log, 0,
|
|
130 "setproctitle: \"%s\"", ngx_os_argv[0]);
|
|
131 }
|
|
132
|
|
133
|
|
134 #elif !defined(ngx_setproctitle)
|
|
135
|
42
|
136 ngx_int_t
|
|
137 ngx_init_setproctitle(ngx_log_t *log)
|
2
|
138 {
|
|
139 return NGX_OK;
|
|
140 }
|
|
141
|
42
|
142 void
|
|
143 ngx_setproctitle(char *title)
|
2
|
144 {
|
42
|
145 return;
|
2
|
146 }
|
|
147
|
|
148 #endif
|