• 注册
当前位置：1313e > 默认分类 >正文

linux下修改进程名称

在编写网络服务器程序时，为了响应客户端的请求，我们经常需要新建进程来处理业务流程；而且又是为了关闭某个非法请求或者关闭长连接的客户端，这时就需要杀死进程 killall  proc_name。 但是在新建进程时，子进程名与父进程名相同。因此需要由进程名及参数来区分客户端连接。
   在linux中prctl可以满足这个要求，下满是man手册：
       PR_SET_NAME  (since Linux 2.6.9)
              Set the process name for the calling process, using the value in
              the location pointed to by (char *) arg2.   The name can be up to
              16  bytes  long,  and  should  be null terminated if it contains
              fewer bytes .
但是prctl修改的进程名，只能是16个字节(包括'\0')。下面是修改的代码（changetitle.c）：

点击(此处)折叠或打开

  1. #include <stdio.h>
  2. #include <sys/prctl.h>

  4. int main(int argc, char *argv[], char *envp[])
  5. {
  6.     char *new_name = "abcdefghijklmnopqrstuvwxyz";
  7.     
  8.     getchar();
  9.     prctl(PR_SET_NAME, new_name);
  10.     getchar();

  12.     return 0;
  13. }
    当新名称长度大于16时就会截断，上面的新名字截断后是abcdefghijklmno。这对于我们来说是有缺陷的。而且通过ps -aux 查看，进程名称并没有改变，改变的只是/prco/$(PID)/stat和
/prco/$(PID)/status的值，而/prco/$(PID)/cmdline并没有改变。这种方式使用起来也是不方便的。
   下面介绍另一种方式，可以与上面的方式互补。
    首先看一下main函数的原型:int main(int argc, char *argv[]);
        argv[0]存放的是终端执行的程序名称也就是进程名。argv[1...argc-1]存放的是命令行参数。
        linux中main()还有一个隐藏参数就是环境变量信息，存放了运行时所需要的环境变量。
        我们可以通过以下来访问这个变量

点击(此处)折叠或打开

  1. extern char **environ;
      argv与environ是连续存放在栈区的。下面代码可以查看参数信息：

点击(此处)折叠或打开

  1. #include <stdio.h>
  2. #include <string.h>

  4. extern char **environ;
  5. int main(int argc , char *argv[])
  6. {
  7.     int i;

  9.     printf("argc:%d\n" , argc);

  11.     for (i = 0; i < argc; ++i)
  12.     {
  13.         printf("argv[%d](0x%x):%s\n" , i , (unsigned int)argv[i], argv[i]);
  14.     }

  16.     printf("evriron=0x%x\n" , (unsigned int)environ[0]);

  18.     return 0;
  19. }
        通过上面可以看出，我们只需要修改argv[0]所指向的内存空间的内容，就可以修改进程名。但是如果新名称比argv[0]的长度小，我们可以直接修改，并把多余的部分请0，如果新名称
比argv[0]长我们需要两步：
            1、申请新内存保存环境变量信息和argv[1...argc-1]参数信息
            2、修改argv[0]，将新名称往后到environ的最后一项清0
        以下是参考代码：

点击(此处)折叠或打开

  1. #include <unistd.h>
  2. #include <stdio.h>
  3. #include <stdarg.h>
  4. #include <string.h>
  5. #include <stdlib.h>
  6. #include <sys/prctl.h>

  8. # define MAXLINE 2048

  10. extern char **environ;

  12. static char **g_main_Argv = NULL; /* pointer to argument vector */
  13. static char *g_main_LastArgv = NULL; /* end of argv */

  15. void setproctitle_init(int argc, char **argv, char **envp)
  16. {
  17.     int i;

  19.     for (i = 0; envp[i] != NULL; i++) // calc envp num
  20.         continue;
  21.     environ = (char **) malloc(sizeof (char *) * (i + 1)); // malloc envp pointer
  22.     
  23.     for (i = 0; envp[i] != NULL; i++)
  24.     {
  25.         environ[i] = malloc(sizeof(char) * strlen(envp[i]));
  26.         strcpy(environ[i], envp[i]);
  27.     }
  28.     environ[i] = NULL;

  30.     g_main_Argv = argv;
  31.     if (i > 0)
  32.       g_main_LastArgv = envp[i - 1] + strlen(envp[i - 1]);
  33.     else
  34.       g_main_LastArgv = argv[argc - 1] + strlen(argv[argc - 1]);
  35. }

  37. void setproctitle(const char *fmt, ...)
  38. {
  39.     char *p;
  40.     int i;
  41.     char buf[MAXLINE];

  43.     extern char **g_main_Argv;
  44.     extern char *g_main_LastArgv;
  45.     va_list ap;
  46.     p = buf;

  48.     va_start(ap, fmt);
  49.     vsprintf(p, fmt, ap);
  50.     va_end(ap);

  52.     i = strlen(buf);

  54.     if (i > g_main_LastArgv - g_main_Argv[0] - 2)
  55.     {
  56.         i = g_main_LastArgv - g_main_Argv[0] - 2;
  57.         buf[i] = '\0';
  58.     }
  59.     (void) strcpy(g_main_Argv[0], buf);

  61.     p = &g_main_Argv[0][i];
  62.     while (p < g_main_LastArgv)
  63.         *p++ = '\0';
  64.     g_main_Argv[1] = NULL;

  66.     prctl(PR_SET_NAME,buf);
  67. }

  69. int main(int argc, char *argv[])
  70. {
  71.     char argv_buf[MAXLINE] = {0}; // save argv paramters

  73.     for(int i = 1; i < argc; i++)
  74.     {
  75.         strcat(argv_buf, argv[i]);
  76.         strcat(argv_buf, " ");
  77.     }

  79.     setproctitle_init(argc, argv, environ);

  81.     setproctitle("%s@%s %s", "new_name", "ip", argv_buf);

  83.     for (int i = 0; environ[i] != NULL; i++)
  84.         free(environ[i]);
  85.     getchar();

  87.     return 0;
  88. }
      上面的代码使用了prctl和修改argv[0]两种修改方法的结合，通过ps -a 、 ps -ef  、ps -aux、 top 等等命令都只能查询到新进程名，/proc/$PID/ 下的文件也显示了新进程名的信息。
      应用场景：
         1、标识父子进程名称，防止被误杀
         2、构造假的进程名及参数，引导非法进入人员到蜜罐系统，取证
第二种办法： 
#include 
#include 
#include 
#include 
#ifndef PR_SET_NAME
#define PR_SET_NAME 15
#endif
#ifndef PR_GET_NAME
#define PR_GET_NAME 16
#endif
static char *arg_start;
static char *arg_end;
static char *env_start;
void init_proc_title(int argc, char **argv) {int i;arg_start = argv[0];arg_end = argv[argc-1] + strlen(argv[argc-1])+1;env_start = environ[0];for(i=0; i

本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 162202241@qq.com 举报，一经查实，本站将立刻删除。

最新评论

欢迎您发表评论:

请登录之后再进行评论

登录
相关推荐
Copyright © 2023 - 2024 1313e All Rights Reserved