如何使用 C 从 PID 获取进程名称

Question

我想从task_struct获取进程名，但是 我收到一个错误，取消引用指向不完整类型（任务-> comm）的指针。 我必须使用pid_task 函数。 我不知道为什么它不起作用。

ssize_t simple_read(struct file *filp, char __user *user_buf, size_t count, loff_t *f_pos) {

    int len=0;
    pid_struct = find_get_pid(pid);
    task = pid_task(pid_struct,PIDTYPE_PID);

    len = sprintf(user_buf,"\nname %s\n ",task->comm);
    return len;
}

Answer 1

要找到进程的task_struct，我们可以使用在 kernel/pid.c 中定义的函数 pid_task 。

struct task_struct *pid_task(struct pid *pid, enum pid_type type)

参数：

pid : Pointer to the struct pid of the process. 
pid_type:  PIDTYPE_PID,
       PIDTYPE_PGID,
          PIDTYPE_SID,
          PIDTYPE_MAX

如果我们有进程的 pid，要找到 pid 结构，我们可以使用在 kernel/pid.c 中定义的函数find_get_pid

struct pid *find_get_pid(pid_t nr)

在下面的模块中，我们创建了一个名为 task_by_pid 的读/写 proc 条目。我们想找到哪个进程使用它的 pid 的 task_struct，我们可以将编号写入 proc 条目。

当我们读取 proc 条目时，它会显示与我们写入其中的 pid 对应的进程名称。

proc_task_pid：

#include <linux/module.h>
#include <linux/kernel.h>
#include <asm/uaccess.h>
#include <linux/fs.h> 
#include <linux/cdev.h>
#include <linux/proc_fs.h>
#include <linux/pid.h>

#include <linux/pid_namespace.h>
int p_id;
struct pid *pid_struct;
struct task_struct *task;
static struct proc_dir_entry *proc_write_entry;
char *proc_name="task_by_pid";

int read_proc(char *buf,char **start,off_t offset,int count,int *eof,void *data )
{
int len=0;
pid_struct = find_get_pid(p_id);
task = pid_task(pid_struct,PIDTYPE_PID);

 len = sprintf(buf,"\nname %s\n ",task->comm);

return len;
}

int write_proc(struct file *file,const char *buf,int count,void *data )
{
int ret;
char *id;
id = (char *)kmalloc(1000*sizeof(char),GFP_KERNEL);
printk(KERN_INFO "buf passed %s",buf);
if(copy_from_user(id,buf,count))
    return -EFAULT;
printk(KERN_INFO "id passed %s",id);
p_id = simple_strtoul(id,NULL,0);
printk(KERN_INFO "pid %d ret %d",p_id,ret);
return sizeof(buf);
}

void create_new_proc_entry()
{
proc_write_entry = create_proc_entry(proc_name,0666,NULL);
if(!proc_write_entry)
      {
    printk(KERN_INFO "Error creating proc entry");
    return -ENOMEM;
    }
proc_write_entry->read_proc = read_proc ;
proc_write_entry->write_proc = write_proc;
printk(KERN_INFO "proc initialized");

}



int proc_init (void) {
    create_new_proc_entry();
    return 0;
}

void proc_cleanup(void) {
    printk(KERN_INFO " Inside cleanup_module\n");
    remove_proc_entry(proc_name,NULL);
}
MODULE_LICENSE("GPL");   
module_init(proc_init);
module_exit(proc_cleanup);

使用以下make文件进行编译：

ifneq ($(KERNELRELEASE),) 
   obj-m := proc_task_pid.o
else 

KERNELDIR ?= /lib/modules/$(shell uname -r)/build 

PWD := $(shell pwd)

default: 
 $(MAKE) -C $(KERNELDIR) M=$(PWD) modules
endif 
clean:
 $(MAKE) -C $(KERNELDIR) M=$(PWD) clean

Compile it using 

$make 

将其插入内核：

$ insmod proc_task_pid.ko 

现在让我们尝试查找 pid 为“1”的进程的名称，它始终是 init。

$ printf "1" > /proc/task_by_pid 
$ cat /proc/task_by_pid
name init

正如预期的那样，输出是“init”。因此，我们可以使用它的 pid 找到任何进程的 task_struct。

源代码来自here。