nginx/src/os/unix/ngx_setproctitle.c

   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
  33 ngx_int_t
  34 ngx_init_setproctitle(ngx_log_t *log)
  35 {
  36     u_char      *p;
  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     p = ngx_alloc(size, log);
  47     if (p == NULL) {
  48         return NGX_ERROR;
  49     }
  50
  51     ngx_os_argv_last = ngx_os_argv[0];
  52
  53     for (i = 0; ngx_os_argv[i]; i++) {
  54         if (ngx_os_argv_last == ngx_os_argv[i]) {
  55             ngx_os_argv_last = ngx_os_argv[i] + ngx_strlen(ngx_os_argv[i]) + 1;
  56         }
  57     }
  58
  59     for (i = 0; environ[i]; i++) {
  60         if (ngx_os_argv_last == environ[i]) {
  61
  62             size = ngx_strlen(environ[i]) + 1;
  63             ngx_os_argv_last = environ[i] + size;
  64
  65             ngx_cpystrn(p, (u_char *) environ[i], size);
  66             environ[i] = (char *) p;
  67             p += size;
  68         }
  69     }
  70
  71     ngx_os_argv_last--;
  72
  73     return NGX_OK;
  74 }
  75
  76
  77 void
  78 ngx_setproctitle(char *title)
  79 {
  80     u_char     *p;
  81
  82 #if (NGX_SOLARIS)
  83
  84     ngx_int_t   i;
  85     size_t      size;
  86
  87 #endif
  88
  89     ngx_os_argv[1] = NULL;
  90
  91     p = ngx_cpystrn((u_char *) ngx_os_argv[0], (u_char *) "nginx: ",
  92                     ngx_os_argv_last - ngx_os_argv[0]);
  93
  94     p = ngx_cpystrn(p, (u_char *) title, ngx_os_argv_last - (char *) p);
  95
  96 #if (NGX_SOLARIS)
  97
  98     size = 0;
  99
 100     for (i = 0; i < ngx_argc; i++) {
 101         size += ngx_strlen(ngx_argv[i]) + 1;
 102     }
 103
 104     if (size > (size_t) ((char *) p - ngx_os_argv[0])) {
 105
 106         /*
 107          * ngx_setproctitle() is too rare operation so we use
 108          * the non-optimized copies
 109          */
 110
 111         p = ngx_cpystrn(p, (u_char *) " (", ngx_os_argv_last - (char *) p);
 112
 113         for (i = 0; i < ngx_argc; i++) {
 114             p = ngx_cpystrn(p, (u_char *) ngx_argv[i],
 115                             ngx_os_argv_last - (char *) p);
 116             p = ngx_cpystrn(p, (u_char *) " ", ngx_os_argv_last - (char *) p);
 117         }
 118
 119         if (*(p - 1) == ' ') {
 120             *(p - 1) = ')';
 121         }
 122     }
 123
 124 #endif
 125
 126     if (ngx_os_argv_last - (char *) p) {
 127         ngx_memset(p, NGX_SETPROCTITLE_PAD, ngx_os_argv_last - (char *) p);
 128     }
 129
 130     ngx_log_debug1(NGX_LOG_DEBUG_CORE, ngx_cycle->log, 0,
 131                    "setproctitle: \"%s\"", ngx_os_argv[0]);
 132 }
 133
 134 #endif /* NGX_SETPROCTITLE_USES_ENV */