include/kunit/skbuff.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 /*
   3  * KUnit resource management helpers for SKBs (skbuff).
   4  *
   5  * Copyright (C) 2023 Intel Corporation
   6  */
   7
   8 #ifndef _KUNIT_SKBUFF_H
   9 #define _KUNIT_SKBUFF_H
  10
  11 #include <kunit/resource.h>
  12 #include <linux/skbuff.h>
  13
  14 static void kunit_action_kfree_skb(void *p)
  15 {
  16         kfree_skb((struct sk_buff *)p);
  17 }
  18
  19 /**
  20  * kunit_zalloc_skb() - Allocate and initialize a resource managed skb.
  21  * @test: The test case to which the skb belongs
  22  * @len: size to allocate
  23  * @gfp: allocation flags
  24  *
  25  * Allocate a new struct sk_buff with gfp flags, zero fill the given length
  26  * and add it as a resource to the kunit test for automatic cleanup.
  27  *
  28  * Returns: newly allocated SKB, or %NULL on error
  29  */
  30 static inline struct sk_buff *kunit_zalloc_skb(struct kunit *test, int len,
  31                                                gfp_t gfp)
  32 {
  33         struct sk_buff *res = alloc_skb(len, gfp);
  34
  35         if (!res || skb_pad(res, len))
  36                 return NULL;
  37
  38         if (kunit_add_action_or_reset(test, kunit_action_kfree_skb, res))
  39                 return NULL;
  40
  41         return res;
  42 }
  43
  44 /**
  45  * kunit_kfree_skb() - Like kfree_skb except for allocations managed by KUnit.
  46  * @test: The test case to which the resource belongs.
  47  * @skb: The SKB to free.
  48  */
  49 static inline void kunit_kfree_skb(struct kunit *test, struct sk_buff *skb)
  50 {
  51         if (!skb)
  52                 return;
  53
  54         kunit_release_action(test, kunit_action_kfree_skb, (void *)skb);
  55 }
  56
  57 #endif /* _KUNIT_SKBUFF_H */