mirror of
https://github.com/freebsd/freebsd-src
synced 2024-07-22 18:56:38 +00:00
4f8f43b06e
If a request ends up growing beyong the initially allocated space the
netlink functions (such as snl_add_msg_attr_u32()) will allocate a
new buffer. This invalidates the header pointer we can have received
from snl_create_msg_request(). Always use the hdr returned by
snl_finalize_msg().
Reviewed by: melifaro
MFC after: 1 week
Sponsored by: Rubicon Communications, LLC ("Netgate")
Differential Revision: https://reviews.freebsd.org/D42223
135 lines
4.4 KiB
C
135 lines
4.4 KiB
C
/*-
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*
|
|
* Copyright (c) 2022 Alexander V. Chernikov <melifaro@FreeBSD.org>
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
*/
|
|
#ifndef _NETLINK_NETLINK_SNL_GENERIC_H_
|
|
#define _NETLINK_NETLINK_SNL_GENERIC_H_
|
|
|
|
#include <netlink/netlink.h>
|
|
#include <netlink/netlink_generic.h>
|
|
#include <netlink/netlink_snl.h>
|
|
|
|
/* Genetlink helpers */
|
|
static inline struct nlmsghdr *
|
|
snl_create_genl_msg_request(struct snl_writer *nw, int genl_family, uint8_t genl_cmd)
|
|
{
|
|
assert(nw->hdr == NULL);
|
|
|
|
struct nlmsghdr *hdr = snl_reserve_msg_object(nw, struct nlmsghdr);
|
|
hdr->nlmsg_type = genl_family;
|
|
hdr->nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
|
|
nw->hdr = hdr;
|
|
struct genlmsghdr *ghdr = snl_reserve_msg_object(nw, struct genlmsghdr);
|
|
ghdr->cmd = genl_cmd;
|
|
|
|
return (hdr);
|
|
}
|
|
|
|
static struct snl_field_parser snl_fp_genl[] = {};
|
|
|
|
#define SNL_DECLARE_GENL_PARSER(_name, _np) SNL_DECLARE_PARSER(_name,\
|
|
struct genlmsghdr, snl_fp_genl, _np)
|
|
|
|
struct snl_genl_ctrl_mcast_group {
|
|
uint32_t mcast_grp_id;
|
|
char *mcast_grp_name;
|
|
};
|
|
|
|
struct snl_genl_ctrl_mcast_groups {
|
|
uint32_t num_groups;
|
|
struct snl_genl_ctrl_mcast_group **groups;
|
|
};
|
|
|
|
#define _OUT(_field) offsetof(struct snl_genl_ctrl_mcast_group, _field)
|
|
static struct snl_attr_parser _nla_p_getmc[] = {
|
|
{ .type = CTRL_ATTR_MCAST_GRP_NAME, .off = _OUT(mcast_grp_name), .cb = snl_attr_get_string },
|
|
{ .type = CTRL_ATTR_MCAST_GRP_ID, .off = _OUT(mcast_grp_id), .cb = snl_attr_get_uint32 },
|
|
};
|
|
#undef _OUT
|
|
SNL_DECLARE_ATTR_PARSER_EXT(_genl_ctrl_mc_parser,
|
|
sizeof(struct snl_genl_ctrl_mcast_group),
|
|
_nla_p_getmc, NULL);
|
|
|
|
struct _getfamily_attrs {
|
|
uint16_t family_id;
|
|
char *family_name;
|
|
struct snl_genl_ctrl_mcast_groups mcast_groups;
|
|
};
|
|
|
|
#define _IN(_field) offsetof(struct genlmsghdr, _field)
|
|
#define _OUT(_field) offsetof(struct _getfamily_attrs, _field)
|
|
static struct snl_attr_parser _nla_p_getfam[] = {
|
|
{ .type = CTRL_ATTR_FAMILY_ID , .off = _OUT(family_id), .cb = snl_attr_get_uint16 },
|
|
{ .type = CTRL_ATTR_FAMILY_NAME, .off = _OUT(family_name), .cb = snl_attr_get_string },
|
|
{
|
|
.type = CTRL_ATTR_MCAST_GROUPS,
|
|
.off = _OUT(mcast_groups),
|
|
.cb = snl_attr_get_parray,
|
|
.arg = &_genl_ctrl_mc_parser,
|
|
},
|
|
};
|
|
#undef _IN
|
|
#undef _OUT
|
|
SNL_DECLARE_GENL_PARSER(_genl_ctrl_getfam_parser, _nla_p_getfam);
|
|
|
|
static bool
|
|
snl_get_genl_family_info(struct snl_state *ss, const char *family_name,
|
|
struct _getfamily_attrs *attrs)
|
|
{
|
|
struct snl_writer nw;
|
|
struct nlmsghdr *hdr;
|
|
|
|
memset(attrs, 0, sizeof(*attrs));
|
|
|
|
snl_init_writer(ss, &nw);
|
|
hdr = snl_create_genl_msg_request(&nw, GENL_ID_CTRL, CTRL_CMD_GETFAMILY);
|
|
snl_add_msg_attr_string(&nw, CTRL_ATTR_FAMILY_NAME, family_name);
|
|
if ((hdr = snl_finalize_msg(&nw)) == NULL || !snl_send_message(ss, hdr))
|
|
return (false);
|
|
|
|
hdr = snl_read_reply(ss, hdr->nlmsg_seq);
|
|
if (hdr != NULL && hdr->nlmsg_type != NLMSG_ERROR) {
|
|
if (snl_parse_nlmsg(ss, hdr, &_genl_ctrl_getfam_parser, attrs))
|
|
return (true);
|
|
}
|
|
|
|
return (false);
|
|
}
|
|
|
|
static inline uint16_t
|
|
snl_get_genl_family(struct snl_state *ss, const char *family_name)
|
|
{
|
|
struct _getfamily_attrs attrs = {};
|
|
|
|
snl_get_genl_family_info(ss, family_name, &attrs);
|
|
return (attrs.family_id);
|
|
}
|
|
|
|
static const struct snl_hdr_parser *snl_all_genl_parsers[] = {
|
|
&_genl_ctrl_getfam_parser, &_genl_ctrl_mc_parser,
|
|
};
|
|
|
|
#endif
|