Commit b3231d35 authored by Benjamin Berg's avatar Benjamin Berg Committed by Johannes Berg
Browse files

kunit: add a convenience allocation wrapper for SKBs 



Add a simple convenience helper to allocate and zero fill an SKB for the
use by a kunit test. Also provide a way to free it again in case that
may be desirable.

This simply mirrors the kunit_kmalloc API.

Signed-off-by: default avatarBenjamin Berg <benjamin.berg@intel.com>
Reviewed-by: default avatarDavid Gow <davidgow@google.com>
Link: https://msgid.link/20231220151952.415232-3-benjamin@sipsolutions.net


[adjust file description as discussed]
Signed-off-by: default avatarJohannes Berg <johannes.berg@intel.com>
parent 292010ee

include/kunit/skbuff.h

0 → 100644
+56 −0
Original line number Diff line number Diff line 
/* SPDX-License-Identifier: GPL-2.0 */

/*

 * KUnit resource management helpers for SKBs (skbuff).

 *

 * Copyright (C) 2023 Intel Corporation

 */



#ifndef _KUNIT_SKBUFF_H

#define _KUNIT_SKBUFF_H



#include <kunit/resource.h>

#include <linux/skbuff.h>



static void kunit_action_kfree_skb(void *p)

{

	kfree_skb((struct sk_buff *)p);

}



/**

 * kunit_zalloc_skb() - Allocate and initialize a resource managed skb.

 * @test: The test case to which the skb belongs

 * @len: size to allocate

 *

 * Allocate a new struct sk_buff with GFP_KERNEL, zero fill the give length

 * and add it as a resource to the kunit test for automatic cleanup.

 *

 * Returns: newly allocated SKB, or %NULL on error

 */

static inline struct sk_buff *kunit_zalloc_skb(struct kunit *test, int len,

					       gfp_t gfp)

{

	struct sk_buff *res = alloc_skb(len, GFP_KERNEL);



	if (!res || skb_pad(res, len))

		return NULL;



	if (kunit_add_action_or_reset(test, kunit_action_kfree_skb, res))

		return NULL;



	return res;

}



/**

 * kunit_kfree_skb() - Like kfree_skb except for allocations managed by KUnit.

 * @test: The test case to which the resource belongs.

 * @skb: The SKB to free.

 */

static inline void kunit_kfree_skb(struct kunit *test, struct sk_buff *skb)

{

	if (!skb)

		return;



	kunit_release_action(test, kunit_action_kfree_skb, (void *)skb);

}



#endif /* _KUNIT_SKBUFF_H */