packet: add a generic drop reason for receive

[linux-2.6-microblaze.git] / net / packet / af_packet.c

diff --git a/net/packet/af_packet.c b/net/packet/af_packet.c
index a84e00b..f92edba 100644 (file)
--- a/net/packet/af_packet.c
+++ b/net/packet/af_packet.c
@@ -2121,13 +2121,13 @@ static int packet_rcv_vnet(struct msghdr *msg, const struct sk_buff *skb,
 static int packet_rcv(struct sk_buff *skb, struct net_device *dev,
                      struct packet_type *pt, struct net_device *orig_dev)
 {
+       enum skb_drop_reason drop_reason = SKB_CONSUMED;
        struct sock *sk;
        struct sockaddr_ll *sll;
        struct packet_sock *po;
        u8 *skb_head = skb->data;
        int skb_len = skb->len;
        unsigned int snaplen, res;
-       bool is_drop_n_account = false;
 
        if (skb->pkt_type == PACKET_LOOPBACK)
                goto drop;
@@ -2217,9 +2217,9 @@ static int packet_rcv(struct sk_buff *skb, struct net_device *dev,
        return 0;
 
 drop_n_acct:
-       is_drop_n_account = true;
        atomic_inc(&po->tp_drops);
        atomic_inc(&sk->sk_drops);
+       drop_reason = SKB_DROP_REASON_PACKET_SOCK_ERROR;
 
 drop_n_restore:
        if (skb_head != skb->data && skb_shared(skb)) {
@@ -2227,16 +2227,14 @@ drop_n_restore:
                skb->len = skb_len;
        }
 drop:
-       if (!is_drop_n_account)
-               consume_skb(skb);
-       else
-               kfree_skb(skb);
+       kfree_skb_reason(skb, drop_reason);
        return 0;
 }
 
 static int tpacket_rcv(struct sk_buff *skb, struct net_device *dev,
                       struct packet_type *pt, struct net_device *orig_dev)
 {
+       enum skb_drop_reason drop_reason = SKB_CONSUMED;
        struct sock *sk;
        struct packet_sock *po;
        struct sockaddr_ll *sll;
@@ -2250,7 +2248,6 @@ static int tpacket_rcv(struct sk_buff *skb, struct net_device *dev,
        struct sk_buff *copy_skb = NULL;
        struct timespec64 ts;
        __u32 ts_status;
-       bool is_drop_n_account = false;
        unsigned int slot_id = 0;
        int vnet_hdr_sz = 0;
 
@@ -2498,19 +2495,16 @@ drop_n_restore:
                skb->len = skb_len;
        }
 drop:
-       if (!is_drop_n_account)
-               consume_skb(skb);
-       else
-               kfree_skb(skb);
+       kfree_skb_reason(skb, drop_reason);
        return 0;
 
 drop_n_account:
        spin_unlock(&sk->sk_receive_queue.lock);
        atomic_inc(&po->tp_drops);
-       is_drop_n_account = true;
+       drop_reason = SKB_DROP_REASON_PACKET_SOCK_ERROR;
 
        sk->sk_data_ready(sk);
-       kfree_skb(copy_skb);
+       kfree_skb_reason(copy_skb, drop_reason);
        goto drop_n_restore;
 }