ieee802154: split away MAC commands implementation
[safe/jmp/linux-2.6] / net / ieee802154 / dgram.c
index 53dd912..9aac5ae 100644 (file)
@@ -40,9 +40,11 @@ static DEFINE_RWLOCK(dgram_lock);
 struct dgram_sock {
        struct sock sk;
 
-       int bound;
        struct ieee802154_addr src_addr;
        struct ieee802154_addr dst_addr;
+
+       unsigned bound:1;
+       unsigned want_ack:1;
 };
 
 static inline struct dgram_sock *dgram_sk(const struct sock *sk)
@@ -50,7 +52,6 @@ static inline struct dgram_sock *dgram_sk(const struct sock *sk)
        return container_of(sk, struct dgram_sock, sk);
 }
 
-
 static void dgram_hash(struct sock *sk)
 {
        write_lock_bh(&dgram_lock);
@@ -73,6 +74,7 @@ static int dgram_init(struct sock *sk)
 
        ro->dst_addr.addr_type = IEEE802154_ADDR_LONG;
        ro->dst_addr.pan_id = 0xffff;
+       ro->want_ack = 1;
        memset(&ro->dst_addr.hwaddr, 0xff, sizeof(ro->dst_addr.hwaddr));
        return 0;
 }
@@ -86,18 +88,18 @@ static int dgram_bind(struct sock *sk, struct sockaddr *uaddr, int len)
 {
        struct sockaddr_ieee802154 *addr = (struct sockaddr_ieee802154 *)uaddr;
        struct dgram_sock *ro = dgram_sk(sk);
-       int err = 0;
+       int err = -EINVAL;
        struct net_device *dev;
 
+       lock_sock(sk);
+
        ro->bound = 0;
 
        if (len < sizeof(*addr))
-               return -EINVAL;
+               goto out;
 
        if (addr->family != AF_IEEE802154)
-               return -EINVAL;
-
-       lock_sock(sk);
+               goto out;
 
        dev = ieee802154_get_dev(sock_net(sk), &addr->addr);
        if (!dev) {
@@ -113,6 +115,7 @@ static int dgram_bind(struct sock *sk, struct sockaddr *uaddr, int len)
        memcpy(&ro->src_addr, &addr->addr, sizeof(struct ieee802154_addr));
 
        ro->bound = 1;
+       err = 0;
 out_put:
        dev_put(dev);
 out:
@@ -235,7 +238,10 @@ static int dgram_sendmsg(struct kiocb *iocb, struct sock *sk,
 
        skb_reset_network_header(skb);
 
-       mac_cb(skb)->flags = IEEE802154_FC_TYPE_DATA | MAC_CB_FLAG_ACKREQ;
+       mac_cb(skb)->flags = IEEE802154_FC_TYPE_DATA;
+       if (ro->want_ack)
+               mac_cb(skb)->flags |= MAC_CB_FLAG_ACKREQ;
+
        mac_cb(skb)->seq = ieee802154_mlme_ops(dev)->get_dsn(dev);
        err = dev_hard_header(skb, dev, ETH_P_IEEE802154, &ro->dst_addr,
                        ro->bound ? &ro->src_addr : NULL, size);
@@ -297,7 +303,7 @@ static int dgram_recvmsg(struct kiocb *iocb, struct sock *sk,
        if (err)
                goto done;
 
-       sock_recv_timestamp(msg, sk, skb);
+       sock_recv_ts_and_drops(msg, sk, skb);
 
        if (flags & MSG_TRUNC)
                copied = skb->len;
@@ -312,7 +318,6 @@ out:
 static int dgram_rcv_skb(struct sock *sk, struct sk_buff *skb)
 {
        if (sock_queue_rcv_skb(sk, skb) < 0) {
-               atomic_inc(&sk->sk_drops);
                kfree_skb(skb);
                return NET_RX_DROP;
        }
@@ -377,6 +382,64 @@ int ieee802154_dgram_deliver(struct net_device *dev, struct sk_buff *skb)
        return ret;
 }
 
+static int dgram_getsockopt(struct sock *sk, int level, int optname,
+                   char __user *optval, int __user *optlen)
+{
+       struct dgram_sock *ro = dgram_sk(sk);
+
+       int val, len;
+
+       if (level != SOL_IEEE802154)
+               return -EOPNOTSUPP;
+
+       if (get_user(len, optlen))
+               return -EFAULT;
+
+       len = min_t(unsigned int, len, sizeof(int));
+
+       switch (optname) {
+       case WPAN_WANTACK:
+               val = ro->want_ack;
+               break;
+       default:
+               return -ENOPROTOOPT;
+       }
+
+       if (put_user(len, optlen))
+               return -EFAULT;
+       if (copy_to_user(optval, &val, len))
+               return -EFAULT;
+       return 0;
+}
+
+static int dgram_setsockopt(struct sock *sk, int level, int optname,
+                   char __user *optval, unsigned int optlen)
+{
+       struct dgram_sock *ro = dgram_sk(sk);
+       int val;
+       int err = 0;
+
+       if (optlen < sizeof(int))
+               return -EINVAL;
+
+       if (get_user(val, (int __user *)optval))
+               return -EFAULT;
+
+       lock_sock(sk);
+
+       switch (optname) {
+       case WPAN_WANTACK:
+               ro->want_ack = !!val;
+               break;
+       default:
+               err = -ENOPROTOOPT;
+               break;
+       }
+
+       release_sock(sk);
+       return err;
+}
+
 struct proto ieee802154_dgram_prot = {
        .name           = "IEEE-802.15.4-MAC",
        .owner          = THIS_MODULE,
@@ -391,5 +454,7 @@ struct proto ieee802154_dgram_prot = {
        .connect        = dgram_connect,
        .disconnect     = dgram_disconnect,
        .ioctl          = dgram_ioctl,
+       .getsockopt     = dgram_getsockopt,
+       .setsockopt     = dgram_setsockopt,
 };