SAFE public projects git trees. - safe/jmp/linux-2.6/blob - net/phonet/datagram.c

   1 /*
   2  * File: datagram.c
   3  *
   4  * Datagram (ISI) Phonet sockets
   5  *
   6  * Copyright (C) 2008 Nokia Corporation.
   7  *
   8  * Contact: Remi Denis-Courmont <remi.denis-courmont@nokia.com>
   9  * Original author: Sakari Ailus <sakari.ailus@nokia.com>
  10  *
  11  * This program is free software; you can redistribute it and/or
  12  * modify it under the terms of the GNU General Public License
  13  * version 2 as published by the Free Software Foundation.
  14  *
  15  * This program is distributed in the hope that it will be useful, but
  16  * WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  * General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
  23  * 02110-1301 USA
  24  */
  25
  26 #include <linux/kernel.h>
  27 #include <linux/socket.h>
  28 #include <asm/ioctls.h>
  29 #include <net/sock.h>
  30
  31 #include <linux/phonet.h>
  32 #include <net/phonet/phonet.h>
  33
  34 static int pn_backlog_rcv(struct sock *sk, struct sk_buff *skb);
  35
  36 /* associated socket ceases to exist */
  37 static void pn_sock_close(struct sock *sk, long timeout)
  38 {
  39         sk_common_release(sk);
  40 }
  41
  42 static int pn_ioctl(struct sock *sk, int cmd, unsigned long arg)
  43 {
  44         struct sk_buff *skb;
  45         int answ;
  46
  47         switch (cmd) {
  48         case SIOCINQ:
  49                 lock_sock(sk);
  50                 skb = skb_peek(&sk->sk_receive_queue);
  51                 answ = skb ? skb->len : 0;
  52                 release_sock(sk);
  53                 return put_user(answ, (int __user *)arg);
  54         }
  55
  56         return -ENOIOCTLCMD;
  57 }
  58
  59 /* Destroy socket. All references are gone. */
  60 static void pn_destruct(struct sock *sk)
  61 {
  62         skb_queue_purge(&sk->sk_receive_queue);
  63 }
  64
  65 static int pn_init(struct sock *sk)
  66 {
  67         sk->sk_destruct = pn_destruct;
  68         return 0;
  69 }
  70
  71 static int pn_sendmsg(struct kiocb *iocb, struct sock *sk,
  72                         struct msghdr *msg, size_t len)
  73 {
  74         struct sockaddr_pn *target;
  75         struct sk_buff *skb;
  76         int err;
  77
  78         if (msg->msg_flags & ~(MSG_DONTWAIT|MSG_EOR|MSG_NOSIGNAL|
  79                                 MSG_CMSG_COMPAT))
  80                 return -EOPNOTSUPP;
  81
  82         if (msg->msg_name == NULL)
  83                 return -EDESTADDRREQ;
  84
  85         if (msg->msg_namelen < sizeof(struct sockaddr_pn))
  86                 return -EINVAL;
  87
  88         target = (struct sockaddr_pn *)msg->msg_name;
  89         if (target->spn_family != AF_PHONET)
  90                 return -EAFNOSUPPORT;
  91
  92         skb = sock_alloc_send_skb(sk, MAX_PHONET_HEADER + len,
  93                                         msg->msg_flags & MSG_DONTWAIT, &err);
  94         if (skb == NULL)
  95                 return err;
  96         skb_reserve(skb, MAX_PHONET_HEADER);
  97
  98         err = memcpy_fromiovec((void *)skb_put(skb, len), msg->msg_iov, len);
  99         if (err < 0) {
 100                 kfree_skb(skb);
 101                 return err;
 102         }
 103
 104         /*
 105          * Fill in the Phonet header and
 106          * finally pass the packet forwards.
 107          */
 108         err = pn_skb_send(sk, skb, target);
 109
 110         /* If ok, return len. */
 111         return (err >= 0) ? len : err;
 112 }
 113
 114 static int pn_recvmsg(struct kiocb *iocb, struct sock *sk,
 115                         struct msghdr *msg, size_t len, int noblock,
 116                         int flags, int *addr_len)
 117 {
 118         struct sk_buff *skb = NULL;
 119         struct sockaddr_pn sa;
 120         int rval = -EOPNOTSUPP;
 121         int copylen;
 122
 123         if (flags & ~(MSG_PEEK|MSG_TRUNC|MSG_DONTWAIT|MSG_NOSIGNAL|
 124                         MSG_CMSG_COMPAT))
 125                 goto out_nofree;
 126
 127         if (addr_len)
 128                 *addr_len = sizeof(sa);
 129
 130         skb = skb_recv_datagram(sk, flags, noblock, &rval);
 131         if (skb == NULL)
 132                 goto out_nofree;
 133
 134         pn_skb_get_src_sockaddr(skb, &sa);
 135
 136         copylen = skb->len;
 137         if (len < copylen) {
 138                 msg->msg_flags |= MSG_TRUNC;
 139                 copylen = len;
 140         }
 141
 142         rval = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copylen);
 143         if (rval) {
 144                 rval = -EFAULT;
 145                 goto out;
 146         }
 147
 148         rval = (flags & MSG_TRUNC) ? skb->len : copylen;
 149
 150         if (msg->msg_name != NULL)
 151                 memcpy(msg->msg_name, &sa, sizeof(struct sockaddr_pn));
 152
 153 out:
 154         skb_free_datagram(sk, skb);
 155
 156 out_nofree:
 157         return rval;
 158 }
 159
 160 /* Queue an skb for a sock. */
 161 static int pn_backlog_rcv(struct sock *sk, struct sk_buff *skb)
 162 {
 163         int err = sock_queue_rcv_skb(sk, skb);
 164
 165         if (err < 0)
 166                 kfree_skb(skb);
 167         return err ? NET_RX_DROP : NET_RX_SUCCESS;
 168 }
 169
 170 /* Module registration */
 171 static struct proto pn_proto = {
 172         .close          = pn_sock_close,
 173         .ioctl          = pn_ioctl,
 174         .init           = pn_init,
 175         .sendmsg        = pn_sendmsg,
 176         .recvmsg        = pn_recvmsg,
 177         .backlog_rcv    = pn_backlog_rcv,
 178         .hash           = pn_sock_hash,
 179         .unhash         = pn_sock_unhash,
 180         .get_port       = pn_sock_get_port,
 181         .obj_size       = sizeof(struct pn_sock),
 182         .owner          = THIS_MODULE,
 183         .name           = "PHONET",
 184 };
 185
 186 static struct phonet_protocol pn_dgram_proto = {
 187         .ops            = &phonet_dgram_ops,
 188         .prot           = &pn_proto,
 189         .sock_type      = SOCK_DGRAM,
 190 };
 191
 192 int __init isi_register(void)
 193 {
 194         return phonet_proto_register(PN_PROTO_PHONET, &pn_dgram_proto);
 195 }
 196
 197 void __exit isi_unregister(void)
 198 {
 199         phonet_proto_unregister(PN_PROTO_PHONET, &pn_dgram_proto);
 200 }