copy_from_user - 从用户空间复制双指针的困难

Question

涉及的结构如下所示。 结构节点{ 诠释一个; 诠释 b; }

struct ioctl_node {
    struct node **ptr;
    int         count; //Stores the no of pointers this ptr is pointing to.
};

在用户空间中，我使用count = 3 填充了一个ioctl_node 结构。 而ptr[0] 和ptr[2] 指向两个节点类型的结构，而ptr[1] 是NULL。我希望将其传递给内核。 我已发出 ioctl 调用以将此信息从用户空间传递到内核空间。

我在内核空间做了以下事情。

struct ioctl_node k_node;

copy_from_user(&k_node, arg, sizeof(struct ioctl_node)) //arg is the source

//k_node.count is showing the right value what I have set in the user space.

struct node *tmp = NULL;

然后我做了，copy_from_user(&tmp, k_node.ptr, sizeof(struct node *)，这个电话也返回了成功。 但是，我很难在内核空间中正确复制**ptr 的全部内容。 任何人都可以帮忙，我怎么能做到这一点。我应该如何执行下一个 copy_from_user 来复制所有内容。我试过了，但它给出了复制错误。

Answer 1

copy_from_user 将一个连续字节块从用户空间复制到内核空间。您需要先复制struct ioctl_node，然后将指针复制到各个节点，然后循环复制struct node。确保分配正确的内存量。由于事先知道节点的数量，因此可以分配它们的数组。

struct node_data {
    size_t node_count;
    struct nodes *nodes;
};
struct node_data *read_nodes(struct ioctl_node_user_pointer *arg)
{
    struct node_data *nd = NULL;
    struct ioctl_node kin = {0};
    struct node *node_user_pointers = NULL;
    int i;
    int err;
    /* Step 1: retrieve the root node */
    nd = kmalloc(sizeof(*nd), GFP_KERNEL);
    if (!nd) {
        err = -ENOMEM;
        goto error;
    }
    if (copy_from_user(ioctl_node_user_pointer, &kin, sizeof(kin))) {
        err = -EFAULT;
        goto error;
    }
    if (kin->count < 0 || kin->count > ((size_t)-1)/sizeof(*nodes)) {
        err = -EINVAL;
        goto error;
    }
    nd->node_count = kin.count;
    /* Step 2: retrieve the pointers to individual nodes */
    node_user_pointers = kmalloc(sizeof(*node_user_pointers) * nd->node_count, GFP_KERNEL);
    if (node_user_pointers) {
        err = -ENOMEM;
        goto error;
    }
    if (copy_from_user(kin->nodes, node_user_pointers, sizeof(*node_user_pointers) * nd->node_count)) {
        err = -EFAULT;
        goto error;
    }
    /* Step 3: retrieve the nodes themselves */
    nd->nodes = kmalloc(sizeof(*nodes) * nd->node_count, GFP_KERNEL);
    if (!nd->nodes) {
        err = -ENOMEM;
        goto error;
    }
    for (i = 0; i < head.count; i++) {
        if (copy_from_user(node_user_pointers[i], nd->nodes + i, sizeof(nd->nodes[0]))) {
            err = -EFAULT;
            goto error;
        }
    }
    kfree(node_user_pointers);
    return nd;
error:
    if (nd) {
        kfree(nd->nodes);
        kfree(nd);
    }
    kfree(node_user_pointers);
    return ERR_PTR(err);
}