mirror of
https://mirrors.bfsu.edu.cn/git/linux.git
synced 2024-12-15 06:55:13 +08:00
8bbb77b7c7
This change introduces a few helpers to wrap open coded attribute
preparation in netlink.c. It also adds a libbpf_netlink_send_recv() that
is useful to wrap send + recv handling in a generic way. Subsequent patch
will also use this function for sending and receiving a netlink response.
The libbpf_nl_get_link() helper has been removed instead, moving socket
creation into the newly named libbpf_netlink_send_recv().
Every nested attribute's closure must happen using the helper
nlattr_end_nested(), which sets its length properly. NLA_F_NESTED is
enforced using nlattr_begin_nested() helper. Other simple attributes
can be added directly.
The maxsz parameter corresponds to the size of the request structure
which is being filled in, so for instance with req being:
struct {
struct nlmsghdr nh;
struct tcmsg t;
char buf[4096];
} req;
Then, maxsz should be sizeof(req).
This change also converts the open coded attribute preparation with these
helpers. Note that the only failure the internal call to nlattr_add()
could result in the nested helper would be -EMSGSIZE, hence that is what
we return to our caller.
The libbpf_netlink_send_recv() call takes care of opening the socket,
sending the netlink message, receiving the response, potentially invoking
callbacks, and return errors if any, and then finally close the socket.
This allows users to avoid identical socket setup code in different places.
The only user of libbpf_nl_get_link() has been converted to make use of it.
__bpf_set_link_xdp_fd_replace() has also been refactored to use it.
Signed-off-by: Kumar Kartikeya Dwivedi <memxor@gmail.com>
[ Daniel: major patch cleanup ]
Signed-off-by: Daniel Borkmann <daniel@iogearbox.net>
Reviewed-by: Toke Høiland-Jørgensen <toke@redhat.com>
Link: https://lore.kernel.org/bpf/20210512103451.989420-2-memxor@gmail.com
155 lines
3.8 KiB
C
155 lines
3.8 KiB
C
/* SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause) */
|
|
|
|
/*
|
|
* NETLINK Netlink attributes
|
|
*
|
|
* Copyright (c) 2003-2013 Thomas Graf <tgraf@suug.ch>
|
|
*/
|
|
|
|
#ifndef __LIBBPF_NLATTR_H
|
|
#define __LIBBPF_NLATTR_H
|
|
|
|
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <linux/netlink.h>
|
|
|
|
/* avoid multiple definition of netlink features */
|
|
#define __LINUX_NETLINK_H
|
|
|
|
/**
|
|
* Standard attribute types to specify validation policy
|
|
*/
|
|
enum {
|
|
LIBBPF_NLA_UNSPEC, /**< Unspecified type, binary data chunk */
|
|
LIBBPF_NLA_U8, /**< 8 bit integer */
|
|
LIBBPF_NLA_U16, /**< 16 bit integer */
|
|
LIBBPF_NLA_U32, /**< 32 bit integer */
|
|
LIBBPF_NLA_U64, /**< 64 bit integer */
|
|
LIBBPF_NLA_STRING, /**< NUL terminated character string */
|
|
LIBBPF_NLA_FLAG, /**< Flag */
|
|
LIBBPF_NLA_MSECS, /**< Micro seconds (64bit) */
|
|
LIBBPF_NLA_NESTED, /**< Nested attributes */
|
|
__LIBBPF_NLA_TYPE_MAX,
|
|
};
|
|
|
|
#define LIBBPF_NLA_TYPE_MAX (__LIBBPF_NLA_TYPE_MAX - 1)
|
|
|
|
/**
|
|
* @ingroup attr
|
|
* Attribute validation policy.
|
|
*
|
|
* See section @core_doc{core_attr_parse,Attribute Parsing} for more details.
|
|
*/
|
|
struct libbpf_nla_policy {
|
|
/** Type of attribute or LIBBPF_NLA_UNSPEC */
|
|
uint16_t type;
|
|
|
|
/** Minimal length of payload required */
|
|
uint16_t minlen;
|
|
|
|
/** Maximal length of payload allowed */
|
|
uint16_t maxlen;
|
|
};
|
|
|
|
/**
|
|
* @ingroup attr
|
|
* Iterate over a stream of attributes
|
|
* @arg pos loop counter, set to current attribute
|
|
* @arg head head of attribute stream
|
|
* @arg len length of attribute stream
|
|
* @arg rem initialized to len, holds bytes currently remaining in stream
|
|
*/
|
|
#define libbpf_nla_for_each_attr(pos, head, len, rem) \
|
|
for (pos = head, rem = len; \
|
|
nla_ok(pos, rem); \
|
|
pos = nla_next(pos, &(rem)))
|
|
|
|
/**
|
|
* libbpf_nla_data - head of payload
|
|
* @nla: netlink attribute
|
|
*/
|
|
static inline void *libbpf_nla_data(const struct nlattr *nla)
|
|
{
|
|
return (char *) nla + NLA_HDRLEN;
|
|
}
|
|
|
|
static inline uint8_t libbpf_nla_getattr_u8(const struct nlattr *nla)
|
|
{
|
|
return *(uint8_t *)libbpf_nla_data(nla);
|
|
}
|
|
|
|
static inline uint32_t libbpf_nla_getattr_u32(const struct nlattr *nla)
|
|
{
|
|
return *(uint32_t *)libbpf_nla_data(nla);
|
|
}
|
|
|
|
static inline const char *libbpf_nla_getattr_str(const struct nlattr *nla)
|
|
{
|
|
return (const char *)libbpf_nla_data(nla);
|
|
}
|
|
|
|
/**
|
|
* libbpf_nla_len - length of payload
|
|
* @nla: netlink attribute
|
|
*/
|
|
static inline int libbpf_nla_len(const struct nlattr *nla)
|
|
{
|
|
return nla->nla_len - NLA_HDRLEN;
|
|
}
|
|
|
|
int libbpf_nla_parse(struct nlattr *tb[], int maxtype, struct nlattr *head,
|
|
int len, struct libbpf_nla_policy *policy);
|
|
int libbpf_nla_parse_nested(struct nlattr *tb[], int maxtype,
|
|
struct nlattr *nla,
|
|
struct libbpf_nla_policy *policy);
|
|
|
|
int libbpf_nla_dump_errormsg(struct nlmsghdr *nlh);
|
|
|
|
static inline struct nlattr *nla_data(struct nlattr *nla)
|
|
{
|
|
return (struct nlattr *)((char *)nla + NLA_HDRLEN);
|
|
}
|
|
|
|
static inline struct nlattr *nh_tail(struct nlmsghdr *nh)
|
|
{
|
|
return (struct nlattr *)((char *)nh + NLMSG_ALIGN(nh->nlmsg_len));
|
|
}
|
|
|
|
static inline int nlattr_add(struct nlmsghdr *nh, size_t maxsz, int type,
|
|
const void *data, int len)
|
|
{
|
|
struct nlattr *nla;
|
|
|
|
if (NLMSG_ALIGN(nh->nlmsg_len) + NLA_ALIGN(NLA_HDRLEN + len) > maxsz)
|
|
return -EMSGSIZE;
|
|
if (!!data != !!len)
|
|
return -EINVAL;
|
|
|
|
nla = nh_tail(nh);
|
|
nla->nla_type = type;
|
|
nla->nla_len = NLA_HDRLEN + len;
|
|
if (data)
|
|
memcpy(nla_data(nla), data, len);
|
|
nh->nlmsg_len = NLMSG_ALIGN(nh->nlmsg_len) + NLA_ALIGN(nla->nla_len);
|
|
return 0;
|
|
}
|
|
|
|
static inline struct nlattr *nlattr_begin_nested(struct nlmsghdr *nh,
|
|
size_t maxsz, int type)
|
|
{
|
|
struct nlattr *tail;
|
|
|
|
tail = nh_tail(nh);
|
|
if (nlattr_add(nh, maxsz, type | NLA_F_NESTED, NULL, 0))
|
|
return NULL;
|
|
return tail;
|
|
}
|
|
|
|
static inline void nlattr_end_nested(struct nlmsghdr *nh, struct nlattr *tail)
|
|
{
|
|
tail->nla_len = (char *)nh_tail(nh) - (char *)tail;
|
|
}
|
|
|
|
#endif /* __LIBBPF_NLATTR_H */
|