内核模块中如何高效根据给定struct sock获取对应进程的task_struct？

这确实是内核网络编程里容易踩坑的点，我来给你梳理几个可行的方向，先帮你理清之前方法无效的核心原因：

你之前依赖sk->sk_socket->file->f_owner.pid的思路，问题出在对f_owner字段的用途理解偏差——这个字段是用来指定接收SIGIO/SIGURG信号的目标进程/进程组的，默认情况下只有当用户态主动调用fcntl(F_SETOWN)或ioctl(FIOSETOWN)时才会被赋值，套接字创建时内核并不会自动把current（当前创建进程）关联到这里，所以普通AF_INET套接字的f_owner.pid自然是空的。

接下来给你几个靠谱的实现方向：

方法1：通过struct file的凭证信息关联进程 #

当用户态创建套接字时，sk->sk_socket->file指向的struct file对象在初始化阶段，其f_cred会持有创建该文件的进程的凭证引用。你可以通过凭证中的PID来查找对应的struct task_struct：

#include <linux/sched.h>
#include <linux/cred.h>
#include <linux/file.h>
#include <net/sock.h>

struct task_struct *get_sock_creator_task(struct sock *sk) {
    struct task_struct *task = NULL;
    struct socket *sock;
    struct file *file;
    pid_t creator_pid;

    // 合法性检查
    if (!sk || !sk->sk_socket)
        return NULL;

    sock = sk->sk_socket;
    file = sock->file;
    if (!file)
        return NULL;

    // 从文件凭证中获取创建进程的PID
    creator_pid = file->f_cred->pid;
    // 通过PID查找对应的task_struct，注意引用计数管理
    task = get_pid_task(find_vpid(creator_pid), PIDTYPE_PID);

    return task;
}

使用这个函数时要注意：

• 拿到task_struct后必须确保引用计数正确（示例里get_pid_task已自动处理），使用完毕后一定要调用put_task_struct(task)释放，避免内存泄漏或悬空指针。
• 如果创建套接字的进程已经退出，find_vpid会返回NULL，要做好空值判断。
• 这个方法拿到的是创建套接字的原始进程，如果套接字被通过fork、sendmsg等方式传递给其他进程，返回的依然是原创建进程的task_struct。

方法2：挂钩套接字创建流程记录进程（精准追踪） #

如果需要更精准地追踪创建进程，或者要处理套接字传递后的场景，可以通过挂钩内核套接字创建的关键函数，把创建进程的task_struct直接关联到struct sock上：

• 对于AF_INET套接字，可以挂钩inet_create函数，在套接字初始化完成后，将current的引用存储到struct sock的sk_user_data字段（如果该字段未被其他模块占用）。
• 也可以通过内核的struct sock扩展机制，添加自定义字段来存储进程引用，但需要考虑不同内核版本的兼容性。

这种方法的优势是能直接拿到创建进程的task_struct，避免通过PID查找的开销和进程退出的失效问题，但需要对内核网络栈的初始化流程有一定了解。

方法3：遍历进程文件描述符表（仅调试用） #

如果上述方法都不适用，还可以遍历系统中所有进程的文件描述符表，查找持有该struct file的进程。但这种方法效率极低，会遍历所有进程，只适合调试场景，绝对不建议在生产环境的内核模块中使用：

// 仅作演示的伪代码，生产环境请勿使用
struct task_struct *find_task_by_sock_file(struct sock *sk) {
    struct task_struct *task = NULL;
    struct socket *sock;
    struct file *file;
    struct files_struct *files;
    struct fdtable *fdt;
    int i;

    if (!sk || !sk->sk_socket)
        return NULL;

    sock = sk->sk_socket;
    file = sock->file;
    if (!file)
        return NULL;

    for_each_process(task) {
        files = get_files_struct(task);
        if (!files)
            continue;
        fdt = files_fdtable(files);
        for (i = 0; i < fdt->max_fds; i++) {
            if (fdt->fd[i] == file) {
                put_files_struct(files);
                get_task_struct(task);
                return task;
            }
        }
        put_files_struct(files);
    }
    return NULL;
}

总结一下：优先推荐方法1，它利用内核已有的凭证信息，兼容性好且实现简单；如果有特殊的精准追踪需求，可以考虑方法2的挂钩方式。

内容的提问来源于stack exchange，提问作者ArndtF