net/rxrpc/skbuff.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /* ar-skbuff.c: socket buffer destruction handling
   3  *
   4  * Copyright (C) 2007 Red Hat, Inc. All Rights Reserved.
   5  * Written by David Howells (dhowells@redhat.com)
   6  */
   7
   8 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
   9
  10 #include <linux/module.h>
  11 #include <linux/net.h>
  12 #include <linux/skbuff.h>
  13 #include <net/sock.h>
  14 #include <net/af_rxrpc.h>
  15 #include "ar-internal.h"
  16
  17 #define is_tx_skb(skb) (rxrpc_skb(skb)->rx_flags & RXRPC_SKB_TX_BUFFER)
  18 #define select_skb_count(skb) (is_tx_skb(skb) ? &rxrpc_n_tx_skbs : &rxrpc_n_rx_skbs)
  19
  20 /*
  21  * Note the allocation or reception of a socket buffer.
  22  */
  23 void rxrpc_new_skb(struct sk_buff *skb, enum rxrpc_skb_trace op)
  24 {
  25         const void *here = __builtin_return_address(0);
  26         int n = atomic_inc_return(select_skb_count(skb));
  27         trace_rxrpc_skb(skb, op, refcount_read(&skb->users), n, here);
  28 }
  29
  30 /*
  31  * Note the re-emergence of a socket buffer from a queue or buffer.
  32  */
  33 void rxrpc_see_skb(struct sk_buff *skb, enum rxrpc_skb_trace op)
  34 {
  35         const void *here = __builtin_return_address(0);
  36         if (skb) {
  37                 int n = atomic_read(select_skb_count(skb));
  38                 trace_rxrpc_skb(skb, op, refcount_read(&skb->users), n, here);
  39         }
  40 }
  41
  42 /*
  43  * Note the addition of a ref on a socket buffer.
  44  */
  45 void rxrpc_get_skb(struct sk_buff *skb, enum rxrpc_skb_trace op)
  46 {
  47         const void *here = __builtin_return_address(0);
  48         int n = atomic_inc_return(select_skb_count(skb));
  49         trace_rxrpc_skb(skb, op, refcount_read(&skb->users), n, here);
  50         skb_get(skb);
  51 }
  52
  53 /*
  54  * Note the destruction of a socket buffer.
  55  */
  56 void rxrpc_free_skb(struct sk_buff *skb, enum rxrpc_skb_trace op)
  57 {
  58         const void *here = __builtin_return_address(0);
  59         if (skb) {
  60                 int n;
  61                 CHECK_SLAB_OKAY(&skb->users);
  62                 n = atomic_dec_return(select_skb_count(skb));
  63                 trace_rxrpc_skb(skb, op, refcount_read(&skb->users), n, here);
  64                 kfree_skb(skb);
  65         }
  66 }
  67
  68 /*
  69  * Clear a queue of socket buffers.
  70  */
  71 void rxrpc_purge_queue(struct sk_buff_head *list)
  72 {
  73         const void *here = __builtin_return_address(0);
  74         struct sk_buff *skb;
  75         while ((skb = skb_dequeue((list))) != NULL) {
  76                 int n = atomic_dec_return(select_skb_count(skb));
  77                 trace_rxrpc_skb(skb, rxrpc_skb_purged,
  78                                 refcount_read(&skb->users), n, here);
  79                 kfree_skb(skb);
  80         }
  81 }