prctl()为线程指定名字

Llinu中使用prctl函数为线程指定名字

一、前言
    有时候我们通过fork()、pthread_create()创建出来的新任务其名字和创建任务的parent是一样的，如果有需要，该如何修改这些新创建任务的名字呢？

    有办法，用proctl()函数来实现。这个函数可用来对任务进行控制，其具体情况如下所示：
#include <sys/prctl.h>
 
int prctl(int option, unsigned long arg2, unsigned long arg3, unsigned long arg4, unsigned long arg5);
    其中第一个参数“option”用来指明prctl()对线程(也可以称为任务)实施的具体的控制行为，其可选的值在<linux/protl.h>文件中定义；而"option"后面参数的值意义需要根据"option"来定。
    对于"option"的选项有很多，我们这里只关注的是"PR_SET_NAME"这个选项，其他的选项这里就不一一给出了，感兴趣的小哥哥小姐姐们可以参考： prctl（2）。
    当入参"option"值为"PR_SET_NAME"时，prctl()会为 调用此函数的 线程设置新名字，新名字设置为参数(char *)arg2。其中，线程名字的参数arg2字符串的最大长度为16 bytes(包括字符串终止符)。如果arg2长度超过16 bytes，它将会被截断为16 bytes。
    除了使用“PR_SET_NAME”参数可以设置线程名字外，另外一个函数pthread_setname_np()也可以实现此功能。设置完成后我们可以通过线程的tid来查看它的新名字：
cat /proc/pid/task/[tid]/comm
    上面的"pid"表示这个线程所属线程组leader的"pid"，"tid"则是调用prctl()函数的线程id。
二、代码
    了解了proctl()的使用后我们看看具体的例子：

#include <stdlib.h>
#include <unistd.h>
#include <stdio.h>
#include <pthread.h>
#include <stdarg.h>        /* asked by va_star and va_end */
#include <sys/prctl.h>  /* asked by PR_SET_NAME */
 
void die(const char *fmt, ...);
int * thread1(void *arg);
int * thread2(void *arg);
int * thread3(void *arg);
int thread_create(void);
void process_create(int num_proc);
 
int main(int argc,char *argv[]){
    printf("will create threads\n");
    thread_create();
    while(1){sleep(1);}
}
 
void die(const char *fmt, ...)
{
    va_list args;
    va_start(args, fmt);
    vfprintf(stderr, fmt, args);
    va_end(args);
    fflush(stdout);
    fflush(stderr);
    exit(1);
}
 
void process_create(int num_proc) {
    pid_t pid;
    int i = 0;
    char childname[][10] = {
        "process-1",
        "process-2",
        "process-3",
        "process-4",
        "process-5",
        "process-6",
    };
    if (num_proc < 1)
        die("num of proc must > 1\n");
    while(i < num_proc) {
        if ((pid = fork()) < 0) 
            die("fork failed\n");
        if (0 == pid) {
            prctl(PR_SET_NAME, childname[i], NULL, NULL, NULL); 
            while(1);
            exit(0);
        }
        i++;
    }
}
 
int * thread1(void *arg)
{
    prctl(PR_SET_NAME, "THREAD1");
    printf("thread id is %lu.\n", (unsigned long)pthread_self());
        while(1){sleep(5);};
        return NULL;
}
 
int * thread2(void *arg)
{
    prctl(PR_SET_NAME,"THREAD2");
        printf("thread id is %lu.\n", (unsigned long)pthread_self());
        while(1){sleep(5);};
        return NULL;
}
 
int * thread3(void *arg)
{    
    prctl(PR_SET_NAME, "THREAD3");
    char command[128]= {0};
    process_create(5);        /* create 5 processes */
    printf("thread id is %lu.\n",(unsigned long)pthread_self());
    while(1){sleep(5);};
    return NULL;
}
 
int thread_create(void)
{
    pthread_t id1, id2, id3;
    printf("Main thread id is %lu \n", (unsigned long)pthread_self());
    if(!pthread_create(&id1, NULL, (void *)thread1, NULL))    {
        printf("thread 1 succed!\n");
    }
    if(!pthread_create(&id2, NULL, (void *)thread2, NULL))    {
        printf("thread 2 succed!\n");
    }
    if(!pthread_create(&id3, NULL, (void *)thread3, NULL))    {
        printf("thread 3 succed!\n");
    }
}
三、代码分析

    上面的例子中创建来三个线程，并通过prctl(PR_SET_NAME，....）分别将名字设置为THREAD1/THREAD2/THREAD3；接着THREAD3又创建来5个子进程，同样，通过prctl(PR_SET_NAME，....）函数这些子进程将自己的名字分别设置为"process-1" ～ "process-5"。

    可以通过如下命令进行编译：

gcc -pthread -o prctl_name prctl_name.c
    编译完成后生成可执行文件prctl_name ，然后运行 ./prctl_name &,   并在终端中通过 "ps -L"来查看结果。