]> rtime.felk.cvut.cz Git - can-eth-gw-linux.git/blobdiff - net/can/canethgw.c
canethgw: Premature release of job when garbage is received on UDP
[can-eth-gw-linux.git] / net / can / canethgw.c
index 1c99facc53e4cfd8d2056bbca8b56768ae7ee52d..532ce9540a9db78ea896835bf487c215de428df0 100644 (file)
 
 MODULE_LICENSE("GPL");
 
+enum msg_types {
+       CAN_FRAME,
+};
+
+struct cegw_job
+{
+       struct kref refcount;
+       struct socket* can_sock;
+       struct socket* udp_sock;
+       u32  udp_dstcnt;
+       u32  udp_addrlen;
+       u8   udp_dst[0];
+};
+
 static int cegw_udp2can(void *data);
 static int cegw_udp_send(struct socket *udp_sock, struct can_frame *cf,
-               struct sockaddr_in* addr);
+               struct sockaddr* addr, int addrlen);
 static int cegw_can2udp(void *data);
 static int cegw_can_send(struct socket *can_sock, struct can_frame *cf);
 static int cegw_thread_start(void *data);
 static int cegw_thread_stop(struct cegw_job *job);
 static void cegw_job_release(struct kref *ref);
 
-static int cegw_udp_send(struct socket *udp_sock, struct can_frame *cf, struct sockaddr_in* addr)
+static int cegw_udp_send(struct socket *udp_sock, struct can_frame *cf, struct sockaddr *addr,
+               int addrlen)
 {
        struct msghdr mh;
-       struct kvec vec;
+       struct kvec vec[2];
        int err;
+       __u16 type = CAN_FRAME;
 
        mh.msg_name = addr;
-       mh.msg_namelen = sizeof(*addr);
+       mh.msg_namelen = addrlen;
        mh.msg_control = NULL;
        mh.msg_controllen = 0;
        mh.msg_flags = 0;
 
-       vec.iov_base = cf;
-       vec.iov_len = sizeof(*cf);
+       type = cpu_to_be16(type);
+       vec[0].iov_base = &type;
+       vec[0].iov_len = sizeof(type);
+       vec[1].iov_base = cf;
+       vec[1].iov_len = sizeof(*cf);
 
-       err = kernel_sendmsg(udp_sock, &mh, &vec, 1, sizeof(*cf));
+       err = kernel_sendmsg(udp_sock, &mh, vec, 2, sizeof(type)+sizeof(*cf));
 
        return err;
 }
@@ -81,26 +100,34 @@ static int cegw_can_send(struct socket* can_sock, struct can_frame* cf)
 static int cegw_udp2can(void *data)
 {
        struct can_frame cf;
-       struct kvec vec;
+       struct kvec vec[2];
        struct msghdr mh;
        struct cegw_job *job = (struct cegw_job *)data;
        struct socket *udp_sock = NULL, *can_sock = NULL;
        int ret = 0;
+       __u16 type;
 
        memset(&mh, 0, sizeof(mh));
        udp_sock = job->udp_sock;
        can_sock = job->can_sock;
 
        while (1) {
-               vec.iov_base = &cf;
-               vec.iov_len = sizeof(cf);
-               ret = kernel_recvmsg(udp_sock, &mh, &vec, 1,
-                               sizeof(cf), 0);
-               if (ret < 1)
+               vec[0].iov_base = &type;
+               vec[0].iov_len = sizeof(type);
+               vec[1].iov_base = &cf;
+               vec[1].iov_len = sizeof(cf);
+               ret = kernel_recvmsg(udp_sock, &mh, vec, 2,
+                                    sizeof(type) + sizeof(cf), 0);
+               if (ret != sizeof(type) + sizeof(cf))
                        break;
 
-               cf.can_id = be32_to_cpu(cf.can_id);
-               cegw_can_send(can_sock, &cf);
+               type = be16_to_cpu(type);
+               switch (type) {
+               case CAN_FRAME:
+                       cf.can_id = be32_to_cpu(cf.can_id);
+                       cegw_can_send(can_sock, &cf);
+               default:;
+               }
        }
 
        cegw_thread_stop(job);
@@ -117,6 +144,7 @@ static int cegw_can2udp(void* data)
 {
        struct msghdr mh;
        struct kvec vec;
+       struct sockaddr *udst;
        struct can_frame cf;
        struct cegw_job* job = (struct cegw_job*)data;
        struct socket* udp_sock = job->udp_sock;
@@ -131,13 +159,14 @@ static int cegw_can2udp(void* data)
                vec.iov_len = sizeof(cf);
 
                ret = kernel_recvmsg(can_sock, &mh, &vec, 1,
-                                          sizeof(cf), 0);
-               if (ret < 1)
+                                    sizeof(cf), 0);
+               if (ret != sizeof(cf))
                        break;
 
                cf.can_id = cpu_to_be32(cf.can_id);
                for (i=0; i<job->udp_dstcnt; i++) {
-                       cegw_udp_send(udp_sock, &cf, &job->udp_dst[i]);
+                       udst = (struct sockaddr *)(job->udp_dst + i*job->udp_addrlen);
+                       cegw_udp_send(udp_sock, &cf, udst, job->udp_addrlen);
                }
        }
 
@@ -167,15 +196,14 @@ static int cegw_thread_start(void *data)
        struct task_struct *task = NULL;
        struct cegw_job *job = (struct cegw_job *)data;
 
-       kref_init(&job->refcount);
        kref_get(&job->refcount);
-
        task = kthread_run(cegw_udp2can, data, "canethgw_udp2can");
        if (IS_ERR(task)) {
-               kref_sub(&job->refcount, 2, cegw_job_release);
+               kref_put(&job->refcount, cegw_job_release);
                return -ENOMEM;
        }
 
+       kref_get(&job->refcount);
        task = kthread_run(cegw_can2udp, data, "canethgw_can2udp");
        if (IS_ERR(task)) {
                cegw_thread_stop(job);
@@ -196,7 +224,8 @@ static int cegw_thread_stop(struct cegw_job *job)
        struct socket *udp_sock = job->udp_sock;
        struct socket *can_sock = job->can_sock;
 
-       kernel_sock_shutdown(udp_sock, SHUT_RDWR);
+       if (udp_sock)
+               kernel_sock_shutdown(udp_sock, SHUT_RDWR);
 
        /* PF_CAN sockets do not implement shutdown - do it manualy */
        sk = can_sock->sk;
@@ -226,6 +255,7 @@ static int cegw_release(struct inode *inode, struct file *file)
        if (job) {
                cegw_thread_stop(job);
        }
+       kref_put(&job->refcount, cegw_job_release);
 
        module_put(THIS_MODULE);
        return 0;
@@ -242,9 +272,11 @@ static int cegw_release(struct inode *inode, struct file *file)
 static long cegw_ioctl_start(struct file *file, unsigned long arg)
 {
        int i;
+       int chckfam;
        int err = 0;
        __u32 dstcnt = 0;
        __u32 addrlen = 0;
+       struct sockaddr *sa;
        struct cegw_ioctl gwctl;
        struct cegw_job *job = NULL;
 
@@ -255,52 +287,71 @@ static long cegw_ioctl_start(struct file *file, unsigned long arg)
        dstcnt = gwctl.udp_dstcnt;
        addrlen = gwctl.udp_addrlen;
 
-       if (addrlen != sizeof(struct sockaddr_in))
+       if (addrlen != sizeof(struct sockaddr_in) && addrlen != sizeof(struct sockaddr_in6))
                return -EAFNOSUPPORT;
 
-       job = kmalloc(GFP_KERNEL, sizeof(*job) + dstcnt*addrlen );
+       /* ToDo: consider dstcnt maximum */
+       job = kmalloc(GFP_KERNEL, sizeof(*job) + dstcnt*addrlen);
        if (job == NULL)
                return -ENOMEM;
 
        err = copy_from_user(&job->udp_dst, (void __user *)(arg + sizeof(struct cegw_ioctl)), dstcnt*addrlen);
        if (err != 0) {
-               kfree(job);
-               return -EFAULT;
+               err = -EFAULT;
+               goto err_free;
        }
 
-       for (i=0; i<dstcnt; i++) {
-               if (job->udp_dst[i].sin_family != AF_INET) {
-                       kfree(job);
-                       return -EAFNOSUPPORT;
+       /* */
+       if (dstcnt > 0)
+               sa = (struct sockaddr *)job->udp_dst;
+               chckfam = sa->sa_family;
+
+       for (i=1; i<dstcnt; i++) {
+               sa = (struct sockaddr *)(job->udp_dst + i*addrlen);
+               if (sa->sa_family != chckfam) {
+                       err = -EAFNOSUPPORT;
+                       goto err_free;
                }
        }
 
        job->udp_sock = sockfd_lookup(gwctl.udp_sock, &err);
-       if (job->udp_sock == NULL) {
-               kfree(job);
-               return err;
-       }
+       if (job->udp_sock == NULL)
+               goto err_free;
 
        job->can_sock = sockfd_lookup(gwctl.can_sock, &err);
-       if (job->can_sock == NULL) {
-               fput(job->udp_sock->file);
-               kfree(job);
-               return err;
+       if (job->can_sock == NULL)
+               goto err_put_udp;
+
+       if (job->can_sock->ops->family != AF_CAN ||
+           job->can_sock->type != SOCK_RAW) {
+               err = -EBADF;
+               goto err_put_all;
        }
 
+       kref_init(&job->refcount);
+
        job->udp_dstcnt = dstcnt;
+       job->udp_addrlen = addrlen;
 
        err = cegw_thread_start(job);
        if (err != 0)
-               return err;
+               return err; /* cegw_thread_start performs cleaup for us.  */
 
        file->private_data = job;
        return 0;
+
+err_put_all:
+       fput(job->can_sock->file);
+err_put_udp:
+       fput(job->udp_sock->file);
+err_free:
+       kfree(job);
+       return err;
 }
 
 static long cegw_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
 {
-       int err;
+       long err;
 
        switch (cmd) {
                case CEGW_IOCTL_START:
@@ -318,7 +369,8 @@ static const struct file_operations cegw_fops = {
        .owner = THIS_MODULE,
        .open = cegw_open,
        .release = cegw_release,
-       .unlocked_ioctl = cegw_ioctl
+       .unlocked_ioctl = cegw_ioctl,
+       .compat_ioctl = cegw_ioctl,
 };
 
 static struct miscdevice cegw_device = {
@@ -329,6 +381,7 @@ static struct miscdevice cegw_device = {
 
 static int __init cegw_init(void)
 {
+       pr_info("can: can-eth gateway\n");
        return misc_register(&cegw_device);
 }