设置进程的显示名称

有时候在LINUX下，fork子进程的时候，像nginx里的一样，想让子进程的名字可以自定义。

参考网上文章之后，可以通过修改argv[0]的值来改变子进程的名字，但是要注意新标题的长度。这时需要注意environ变量。

查看nginx源码，大致了解了这个过程。假设需要把a.out的ps名称修改为title，那么需要把title复制到argv[0]指向的内存，并且把argv[1]设为0，但是argv[0]指向的内存不一定容得下title。在linux下，argv和environ指向的内存是连续的，我们可以malloc能容得下environ的内存，把environ的内容复制到malloc出的内存里，让title占用environ现在的内存。environ的内存足够放的下title。

nginx说明如下：

https://github.com/phusion/nginx/blob/master/src/os/unix/ngx_setproctitle.c#L14

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
 
/* example
 * ./a.out start
 * memory layout, supporse ptr point to it
 * ./a.out\0start\0HOSTNAME=vbloger\0TERM=linux\0
 * argc = 2;
 * argv[0] = ptr;
 * argv[1] = ptr + strlen(argv[0])+1;
 * argv[2] = 0;
 * environ[0] = ptr + strlen(argv[0])+1 + strlen(argv[1])+1
 * environ[1] = ptr + strlen(argv[0])+1 + strlen(argv[1])+1 + strlen(environ[0])+1
 * environ[2] = 0;
 */
 
extern char **environ;
char *last;
 
void initProcTitle(int argc, char **argv)
{
    size_t size = 0;
    for (int i = 0; environ[i]; ++i) {
        size += strlen(environ[i])+1; 
    }   
 
    char *raw = new char[size];
    for (int i = 0; environ[i]; ++i) {
        memcpy(raw, environ[i], strlen(environ[i]) + 1); 
        environ[i] = raw;
        raw += strlen(environ[i]) + 1;
    }   
 
    last = argv[0];
    for (int i = 0; i < argc; ++i) {
        last += strlen(argv[i]) + 1;   
    }   
    for (int i = 0; environ[i]; ++i) {
        last += strlen(environ[i]) + 1;
    }   
}
 
void setProcTitle(int argc, char **argv, const char *title)
{
    argv[1] = 0;
    char *p = argv[0];
    memset(p, 0x00, last - p); 
    strncpy(p, title, last - p); 
}
 
int main(int argc, char *argv[])
{
    initProcTitle(argc, argv);
        setProctitle(argc, argv, "master so much argument can you believe");
    sleep(30);  // you can ps now
    return 0;
}

转载自：

https://www.oschina.net/code/snippet_54100_10833

https://www.oschina.net/code/snippet_54100_10833