1/*-
 2 * SPDX-License-Identifier: BSD-2-Clause
 3 *
 4 * Copyright (c) 2023 Alexander V. Chernikov <melifaro@FreeBSD.org>
 5 *
 6 * Redistribution and use in source and binary forms, with or without
 7 * modification, are permitted provided that the following conditions
 8 * are met:
 9 * 1. Redistributions of source code must retain the above copyright
10 *    notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 *    notice, this list of conditions and the following disclaimer in the
13 *    documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 * SUCH DAMAGE.
26 */
27
28#ifndef _NETLINK_KTEST_NETLINK_MESSAGE_WRITER_H_
29#define _NETLINK_KTEST_NETLINK_MESSAGE_WRITER_H_
30
31#if defined(_KERNEL) && defined(INVARIANTS)
32
33bool nlmsg_get_buf_type_wrapper(struct nl_writer *nw, int size, int type, bool waitok);
34void nlmsg_set_callback_wrapper(struct nl_writer *nw);
35struct mbuf *nl_get_mbuf_chain_wrapper(int len, int malloc_flags);
36
37#ifndef KTEST_CALLER
38
39bool
40nlmsg_get_buf_type_wrapper(struct nl_writer *nw, int size, int type, bool waitok)
41{
42	return (nlmsg_get_buf_type(nw, size, type, waitok));
43}
44
45void
46nlmsg_set_callback_wrapper(struct nl_writer *nw)
47{
48	nlmsg_set_callback(nw);
49}
50
51struct mbuf *
52nl_get_mbuf_chain_wrapper(int len, int malloc_flags)
53{
54	return (nl_get_mbuf_chain(len, malloc_flags));
55}
56#endif
57
58#endif
59
60#endif