/* SPDX-License-Identifier: GPL-2.0 */
#ifndef __LIBNETLINK_H__
#define __LIBNETLINK_H__ 1

#include <stdio.h>
#include <string.h>
#include <endian.h>
#include <asm/types.h>
#include <linux/netlink.h>
#include <linux/rtnetlink.h>
#include <linux/if_link.h>
#include <linux/if_addr.h>
#include <linux/neighbour.h>
#include <linux/netconf.h>
#include <arpa/inet.h>

struct rtnl_handle {
        int                     fd;
        struct sockaddr_nl      local;
        struct sockaddr_nl      peer;
        __u32                   seq;
        __u32                   dump;
        int                     proto;
        FILE                   *dump_fp;
#define RTNL_HANDLE_F_LISTEN_ALL_NSID           0x01
#define RTNL_HANDLE_F_SUPPRESS_NLERR            0x02
#define RTNL_HANDLE_F_STRICT_CHK                0x04
        int                     flags;
};

struct nlmsg_list {
        struct nlmsg_list *next;
        struct nlmsghdr   h;
};

struct nlmsg_chain {
        struct nlmsg_list *head;
        struct nlmsg_list *tail;
};

struct ipstats_req {
        struct nlmsghdr nlh;
        struct if_stats_msg ifsm;
        char buf[128];
};

extern int rcvbuf;

int rtnl_open(struct rtnl_handle *rth, unsigned int subscriptions)
        __attribute__((warn_unused_result));

int rtnl_open_byproto(struct rtnl_handle *rth, unsigned int subscriptions,
                             int protocol)
        __attribute__((warn_unused_result));
int rtnl_add_nl_group(struct rtnl_handle *rth, unsigned int group)
        __attribute__((warn_unused_result));
void rtnl_close(struct rtnl_handle *rth);
void rtnl_set_strict_dump(struct rtnl_handle *rth);

typedef int (*req_filter_fn_t)(struct nlmsghdr *nlh, int reqlen);

int rtnl_addrdump_req(struct rtnl_handle *rth, int family,
                      req_filter_fn_t filter_fn)
        __attribute__((warn_unused_result));
int rtnl_addrlbldump_req(struct rtnl_handle *rth, int family)
        __attribute__((warn_unused_result));
int rtnl_routedump_req(struct rtnl_handle *rth, int family,
                       req_filter_fn_t filter_fn)
        __attribute__((warn_unused_result));
int rtnl_ruledump_req(struct rtnl_handle *rth, int family)
        __attribute__((warn_unused_result));
int rtnl_neighdump_req(struct rtnl_handle *rth, int family,
                       req_filter_fn_t filter_fn)
        __attribute__((warn_unused_result));
int rtnl_neightbldump_req(struct rtnl_handle *rth, int family)
        __attribute__((warn_unused_result));
int rtnl_mdbdump_req(struct rtnl_handle *rth, int family)
        __attribute__((warn_unused_result));
int rtnl_brvlandump_req(struct rtnl_handle *rth, int family, __u32 dump_flags)
        __attribute__((warn_unused_result));
int rtnl_netconfdump_req(struct rtnl_handle *rth, int family)
        __attribute__((warn_unused_result));

int rtnl_linkdump_req(struct rtnl_handle *rth, int fam)
        __attribute__((warn_unused_result));
int rtnl_linkdump_req_filter(struct rtnl_handle *rth, int fam, __u32 filt_mask)
        __attribute__((warn_unused_result));

int rtnl_linkdump_req_filter_fn(struct rtnl_handle *rth, int fam,
                                req_filter_fn_t fn)
        __attribute__((warn_unused_result));
int rtnl_fdb_linkdump_req_filter_fn(struct rtnl_handle *rth,
                                    req_filter_fn_t filter_fn)
        __attribute__((warn_unused_result));
int rtnl_nsiddump_req_filter_fn(struct rtnl_handle *rth, int family,
                                req_filter_fn_t filter_fn)
        __attribute__((warn_unused_result));
int rtnl_statsdump_req_filter(struct rtnl_handle *rth, int fam, __u32 filt_mask,
                              int (*filter_fn)(struct ipstats_req *req,
                                               void *data),
                              void *filter_data)
        __attribute__((warn_unused_result));
int rtnl_dump_request(struct rtnl_handle *rth, int type, void *req,
                             int len)
        __attribute__((warn_unused_result));
int rtnl_dump_request_n(struct rtnl_handle *rth, struct nlmsghdr *n)
        __attribute__((warn_unused_result));

int rtnl_nexthopdump_req(struct rtnl_handle *rth, int family,
                         req_filter_fn_t filter_fn)
        __attribute__((warn_unused_result));
int rtnl_nexthop_bucket_dump_req(struct rtnl_handle *rth, int family,
                                 req_filter_fn_t filter_fn)
        __attribute__((warn_unused_result));

int rtnl_tunneldump_req(struct rtnl_handle *rth, int family, int ifindex,
                        __u8 flags)
        __attribute__((warn_unused_result));

struct rtnl_ctrl_data {
        int     nsid;
};

typedef int (*rtnl_filter_t)(struct nlmsghdr *n, void *);

/**
 * rtnl error handler called from
 *      rtnl_dump_done()
 *      rtnl_dump_error()
 *
 * Return value is a bitmask of the following values:
 * RTNL_LET_NLERR
 *      error handled as usual
 * RTNL_SUPPRESS_NLMSG_DONE_NLERR
 *      error in nlmsg_type == NLMSG_DONE will be suppressed
 * RTNL_SUPPRESS_NLMSG_ERROR_NLERR
 *      error in nlmsg_type == NLMSG_ERROR will be suppressed
 *      and nlmsg will be skipped
 * RTNL_SUPPRESS_NLERR - suppress error in both previous cases
 */
#define RTNL_LET_NLERR                          0x01
#define RTNL_SUPPRESS_NLMSG_DONE_NLERR          0x02
#define RTNL_SUPPRESS_NLMSG_ERROR_NLERR         0x04
#define RTNL_SUPPRESS_NLERR                     0x06
typedef int (*rtnl_err_hndlr_t)(struct nlmsghdr *n, void *);

typedef int (*rtnl_listen_filter_t)(struct rtnl_ctrl_data *,
                                    struct nlmsghdr *n, void *);

typedef int (*nl_ext_ack_fn_t)(const char *errmsg, uint32_t off,
                               const struct nlmsghdr *inner_nlh);

struct rtnl_dump_filter_arg {
        rtnl_filter_t filter;
        void *arg1;
        rtnl_err_hndlr_t errhndlr;
        void *arg2;
        __u16 nc_flags;
};

int rtnl_dump_filter_nc(struct rtnl_handle *rth,
                        rtnl_filter_t filter,
                        void *arg, __u16 nc_flags);
#define rtnl_dump_filter(rth, filter, arg) \
        rtnl_dump_filter_nc(rth, filter, arg, 0)
int rtnl_dump_filter_errhndlr_nc(struct rtnl_handle *rth,
                                 rtnl_filter_t filter,
                                 void *arg1,
                                 rtnl_err_hndlr_t errhndlr,
                                 void *arg2,
                                 __u16 nc_flags);
#define rtnl_dump_filter_errhndlr(rth, filter, farg, errhndlr, earg) \
        rtnl_dump_filter_errhndlr_nc(rth, filter, farg, errhndlr, earg, 0)

int rtnl_echo_talk(struct rtnl_handle *rtnl, struct nlmsghdr *n, int json,
                   int (*print_info)(struct nlmsghdr *n, void *arg))
        __attribute__((warn_unused_result));
int rtnl_talk(struct rtnl_handle *rtnl, struct nlmsghdr *n,
              struct nlmsghdr **answer)
        __attribute__((warn_unused_result));
int rtnl_talk_suppress_rtnl_errmsg(struct rtnl_handle *rtnl, struct nlmsghdr *n,
                                   struct nlmsghdr **answer)
        __attribute__((warn_unused_result));
int rtnl_send(struct rtnl_handle *rth, const void *buf, int)
        __attribute__((warn_unused_result));
int rtnl_send_check(struct rtnl_handle *rth, const void *buf, int)
        __attribute__((warn_unused_result));
int nl_dump_ext_ack(const struct nlmsghdr *nlh, nl_ext_ack_fn_t errfn);
int nl_dump_ext_ack_done(const struct nlmsghdr *nlh, unsigned int offset, int error);

int addattr(struct nlmsghdr *n, int maxlen, int type);
int addattr8(struct nlmsghdr *n, int maxlen, int type, __u8 data);
int addattr16(struct nlmsghdr *n, int maxlen, int type, __u16 data);
int addattr32(struct nlmsghdr *n, int maxlen, int type, __u32 data);
int addattr64(struct nlmsghdr *n, int maxlen, int type, __u64 data);
int addattrstrz(struct nlmsghdr *n, int maxlen, int type, const char *data);

int addattr_l(struct nlmsghdr *n, int maxlen, int type,
              const void *data, int alen);
int addraw_l(struct nlmsghdr *n, int maxlen, const void *data, int len);
struct rtattr *addattr_nest(struct nlmsghdr *n, int maxlen, int type);
int addattr_nest_end(struct nlmsghdr *n, struct rtattr *nest);
struct rtattr *addattr_nest_compat(struct nlmsghdr *n, int maxlen, int type,
                                   const void *data, int len);
int addattr_nest_compat_end(struct nlmsghdr *n, struct rtattr *nest);
int rta_addattr8(struct rtattr *rta, int maxlen, int type, __u8 data);
int rta_addattr16(struct rtattr *rta, int maxlen, int type, __u16 data);
int rta_addattr32(struct rtattr *rta, int maxlen, int type, __u32 data);
int rta_addattr64(struct rtattr *rta, int maxlen, int type, __u64 data);
int rta_addattr_l(struct rtattr *rta, int maxlen, int type,
                  const void *data, int alen);

int parse_rtattr(struct rtattr *tb[], int max, struct rtattr *rta, int len);
int parse_rtattr_flags(struct rtattr *tb[], int max, struct rtattr *rta,
                              int len, unsigned short flags);
struct rtattr *parse_rtattr_one(int type, struct rtattr *rta, int len);
int __parse_rtattr_nested_compat(struct rtattr *tb[], int max, struct rtattr *rta, int len);

struct rtattr *rta_nest(struct rtattr *rta, int maxlen, int type);
int rta_nest_end(struct rtattr *rta, struct rtattr *nest);

#define RTA_TAIL(rta) \
                ((struct rtattr *) (((void *) (rta)) + \
                                    RTA_ALIGN((rta)->rta_len)))

#define parse_rtattr_nested(tb, max, rta) \
        (parse_rtattr_flags((tb), (max), RTA_DATA(rta), RTA_PAYLOAD(rta), \
                            NLA_F_NESTED))

#define parse_rtattr_one_nested(type, rta) \
        (parse_rtattr_one(type, RTA_DATA(rta), RTA_PAYLOAD(rta)))

#define parse_rtattr_nested_compat(tb, max, rta, data, len) \
        ({ data = RTA_PAYLOAD(rta) >= len ? RTA_DATA(rta) : NULL;       \
                __parse_rtattr_nested_compat(tb, max, rta, len); })

static inline __u8 rta_getattr_u8(const struct rtattr *rta)
{
        return *(__u8 *)RTA_DATA(rta);
}
static inline __u16 rta_getattr_u16(const struct rtattr *rta)
{
        return *(__u16 *)RTA_DATA(rta);
}
static inline __be16 rta_getattr_be16(const struct rtattr *rta)
{
        return ntohs(rta_getattr_u16(rta));
}
static inline __u32 rta_getattr_u32(const struct rtattr *rta)
{
        return *(__u32 *)RTA_DATA(rta);
}
static inline __be32 rta_getattr_be32(const struct rtattr *rta)
{
        return ntohl(rta_getattr_u32(rta));
}
static inline __u64 rta_getattr_u64(const struct rtattr *rta)
{
        __u64 tmp;

        memcpy(&tmp, RTA_DATA(rta), sizeof(__u64));
        return tmp;
}
static inline __u64 rta_getattr_uint(const struct rtattr *rta)
{
        switch (RTA_PAYLOAD(rta)) {
        case sizeof(__u8):
                return rta_getattr_u8(rta);
        case sizeof(__u16):
                return rta_getattr_u16(rta);
        case sizeof(__u32):
                return rta_getattr_u32(rta);
        case sizeof(__u64):
                return rta_getattr_u64(rta);
        }
        return -1ULL;
}

static inline __be64 rta_getattr_be64(const struct rtattr *rta)
{
        return htobe64(rta_getattr_u64(rta));
}

static inline __s32 rta_getattr_s32(const struct rtattr *rta)
{
        return *(__s32 *)RTA_DATA(rta);
}
static inline __s64 rta_getattr_s64(const struct rtattr *rta)
{
        __s64 tmp;

        memcpy(&tmp, RTA_DATA(rta), sizeof(tmp));
        return tmp;
}
static inline const char *rta_getattr_str(const struct rtattr *rta)
{
        return (const char *)RTA_DATA(rta);
}
static inline int rta_type(const struct rtattr *rta)
{
        return rta->rta_type & NLA_TYPE_MASK;
}

int rtnl_listen_all_nsid(struct rtnl_handle *);
int rtnl_listen(struct rtnl_handle *, rtnl_listen_filter_t handler,
                void *jarg);
int rtnl_from_file(FILE *, rtnl_listen_filter_t handler,
                   void *jarg);

#define NLMSG_TAIL(nmsg) \
        ((struct rtattr *) (((void *) (nmsg)) + NLMSG_ALIGN((nmsg)->nlmsg_len)))

#ifndef IFA_RTA
#define IFA_RTA(r) \
        ((struct rtattr *)(((char *)(r)) + NLMSG_ALIGN(sizeof(struct ifaddrmsg))))
#endif
#ifndef IFA_PAYLOAD
#define IFA_PAYLOAD(n)  NLMSG_PAYLOAD(n, sizeof(struct ifaddrmsg))
#endif

#ifndef IFLA_RTA
#define IFLA_RTA(r) \
        ((struct rtattr *)(((char *)(r)) + NLMSG_ALIGN(sizeof(struct ifinfomsg))))
#endif
#ifndef IFLA_PAYLOAD
#define IFLA_PAYLOAD(n) NLMSG_PAYLOAD(n, sizeof(struct ifinfomsg))
#endif

#ifndef NDA_RTA
#define NDA_RTA(r) \
        ((struct rtattr *)(((char *)(r)) + NLMSG_ALIGN(sizeof(struct ndmsg))))
#endif
#ifndef NDA_PAYLOAD
#define NDA_PAYLOAD(n)  NLMSG_PAYLOAD(n, sizeof(struct ndmsg))
#endif

#ifndef NDTA_RTA
#define NDTA_RTA(r) \
        ((struct rtattr *)(((char *)(r)) + NLMSG_ALIGN(sizeof(struct ndtmsg))))
#endif
#ifndef NDTA_PAYLOAD
#define NDTA_PAYLOAD(n) NLMSG_PAYLOAD(n, sizeof(struct ndtmsg))
#endif

#ifndef NETNS_RTA
#define NETNS_RTA(r) \
        ((struct rtattr *)(((char *)(r)) + NLMSG_ALIGN(sizeof(struct rtgenmsg))))
#endif
#ifndef NETNS_PAYLOAD
#define NETNS_PAYLOAD(n)        NLMSG_PAYLOAD(n, sizeof(struct rtgenmsg))
#endif

#ifndef IFLA_STATS_RTA
#define IFLA_STATS_RTA(r) \
        ((struct rtattr *)(((char *)(r)) + NLMSG_ALIGN(sizeof(struct if_stats_msg))))
#endif

#ifndef BRVLAN_RTA
#define BRVLAN_RTA(r) \
        ((struct rtattr *)(((char *)(r)) + NLMSG_ALIGN(sizeof(struct br_vlan_msg))))
#endif

#ifndef TUNNEL_RTA
#define TUNNEL_RTA(r) \
        ((struct rtattr *)(((char *)(r)) + NLMSG_ALIGN(sizeof(struct tunnel_msg))))
#endif

/* User defined nlmsg_type which is used mostly for logging netlink
 * messages from dump file */
#define NLMSG_TSTAMP    15

#define rtattr_for_each_nested(attr, nest) \
        for ((attr) = (void *)RTA_DATA(nest); \
             RTA_OK(attr, RTA_PAYLOAD(nest) - ((char *)(attr) - (char *)RTA_DATA((nest)))); \
             (attr) = RTA_TAIL((attr)))

void nl_print_policy(const struct rtattr *attr, FILE *fp);

#endif /* __LIBNETLINK_H__ */