Question

我正在添加一个内核函数来修改指定的进程pid的nice和prio值。

以下是代码：

SYSCALL_DEFINE5(mysetnice, pid_t, pid, int, flag, int, nicevalue, void __user *, prio, void __user *, nice)
{
    if (pid < 0 || (flag != 1 && flag != 0)) {
        return EFAULT;
    }
    if (nice == NULL || prio == NULL) {
        return EFAULT;
    }
    struct task_struct *p;
    for_each_process(p) {
        if (p->pid == pid) {
            if (flag == 1) {
                set_user_nice(p, nicevalue);
                printk("change nice=%d of pid=%d\n", nicevalue, pid);
            }
            printk("pid:%d,current nice:%d, prio:%d\n",pid,task_nice(p),task_prio(p));
            copy_to_user(prio,(const void*)task_prio(p),sizeof(int));
            copy_to_user(nice,(const void*)task_nice(p),sizeof(int));
            return 0;
        }
    }
    return EFAULT;

}

在用户范围内，我写了一个这样的测试函数：

int main()
{
    int pid = 0;
    int flag = 0;
    int nicevalue = 0;
    int prio = 0;
    int nice = 0;
    int ret;

    printf("please input the params: pid, flag, nicevalue\n");
    while(~scanf("%d %d %d", &pid, &flag, &nicevalue))
    {
        ret = syscall(__NR_mysyscall, pid, flag, nicevalue, (void *)&prio, (void *)&nice);
        if (ret != 0)
        {
            printf("syscall error! error code:%d\n", ret);
            return 0;
        }
        printf("pid:%d, flag:%d, nicevalue:%d, prio:%d, nice:%d\n", pid, flag, nicevalue, prio + 100, nice);
    }

    return 0;
}

但我发现我无法获得nice值，而prio始终为100。那么如何将数据从内核范围复制到用户范围呢？

Answer 1

一般来说，代码质量很差。

SYSCALL_DEFINE5(mysetnice, pid_t, pid, int, flag, int, nicevalue, void __user *, prio, void __user *, nice)
{
    if (pid < 0 || (flag != 1 && flag != 0)) {
        return EFAULT;

错误代码否定。即这应该是-EFAULT，可以在其他系统调用中看到。对于这种特殊情况，EFAULT本身是一个极差的选择。通常无效的参数会导致EINVAL。

    }
    if (nice == NULL || prio == NULL) {
        return EFAULT;
    }
    struct task_struct *p;
    for_each_process(p) {

你从哪里拿到这个？

走过程列表有0个理由。找到一种翻译pid的方法 - ＆gt; task_struct *留作练习。提示：有一些系统调用以pid作为参数。

由于锁定不足（或缺乏），遍历本身是不正确的。

        if (p->pid == pid) {
            if (flag == 1) {
                set_user_nice(p, nicevalue);
                printk("change nice=%d of pid=%d\n", nicevalue, pid);
            }
            printk("pid:%d,current nice:%d, prio:%d\n",pid,task_nice(p),task_prio(p));

printk中缺少日志级别。检查任何现有的printk，看看我的意思。

            copy_to_user(prio,(const void*)task_prio(p),sizeof(int));

不仅没有错误检查，而且第二个参数显然是错误的转换。

            copy_to_user(nice,(const void*)task_nice(p),sizeof(int));
            return 0;
        }
    }
    return EFAULT;

这个应该是ESRCH。

Answer 2

问题是，您必须提供源值的地址，而不是值本身。尝试类似：

int val = task_prio(p);
copy_to_user(prio,(const void*)&val,sizeof(int));

nice相同，那么它应该按预期工作。

将数据从Linux内核空间复制到用户空间

2 个答案: