pf: convert DIOCRCLRADDRS to netlink

Sponsored by:	Rubicon Communications, LLC ("Netgate")
This commit is contained in:
Kristof Provost
2025-05-06 17:15:53 +02:00
parent 681a04dbda
commit b543f426c2
5 changed files with 90 additions and 14 deletions
+42
View File
@@ -3324,3 +3324,45 @@ pfctl_clear_tstats(struct pfctl_handle *h, const struct pfr_table *filter,
return (e.error); return (e.error);
} }
int
pfctl_clear_addrs(struct pfctl_handle *h, const struct pfr_table *filter,
int *ndel, int flags)
{
struct snl_writer nw;
struct snl_errmsg_data e = {};
struct nlmsghdr *hdr;
uint64_t del;
uint32_t seq_id;
int family_id;
family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
if (family_id == 0)
return (ENOTSUP);
snl_init_writer(&h->ss, &nw);
hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_CLR_ADDRS);
snl_add_msg_attr_string(&nw, PF_T_ANCHOR, filter->pfrt_anchor);
snl_add_msg_attr_string(&nw, PF_T_NAME, filter->pfrt_name);
snl_add_msg_attr_u32(&nw, PF_T_TABLE_FLAGS, filter->pfrt_flags);
snl_add_msg_attr_u32(&nw, PF_T_FLAGS, flags);
if ((hdr = snl_finalize_msg(&nw)) == NULL)
return (ENXIO);
seq_id = hdr->nlmsg_seq;
if (!snl_send_message(&h->ss, hdr))
return (ENXIO);
while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
if (!snl_parse_nlmsg(&h->ss, hdr, &tstats_clr_parser, &del))
continue;
if (ndel)
*ndel = (uint32_t)del;
}
return (e.error);
}
+2
View File
@@ -556,5 +556,7 @@ int pfctl_get_tstats(struct pfctl_handle *h, const struct pfr_table *filter,
pfctl_get_tstats_fn fn, void *arg); pfctl_get_tstats_fn fn, void *arg);
int pfctl_clear_tstats(struct pfctl_handle *h, const struct pfr_table *filter, int pfctl_clear_tstats(struct pfctl_handle *h, const struct pfr_table *filter,
int *nzero, int flags); int *nzero, int flags);
int pfctl_clear_addrs(struct pfctl_handle *h, const struct pfr_table *filter,
int *ndel, int flags);
#endif #endif
+1 -14
View File
@@ -113,20 +113,7 @@ pfr_get_tables(struct pfr_table *filter, struct pfr_table *tbl, int *size,
int int
pfr_clr_addrs(struct pfr_table *tbl, int *ndel, int flags) pfr_clr_addrs(struct pfr_table *tbl, int *ndel, int flags)
{ {
struct pfioc_table io; return (pfctl_clear_addrs(pfh, tbl, ndel, flags));
if (tbl == NULL) {
errno = EINVAL;
return (-1);
}
bzero(&io, sizeof io);
io.pfrio_flags = flags;
io.pfrio_table = *tbl;
if (ioctl(dev, DIOCRCLRADDRS, &io))
return (-1);
if (ndel != NULL)
*ndel = io.pfrio_ndel;
return (0);
} }
int int
+44
View File
@@ -2073,6 +2073,43 @@ pf_handle_clear_tstats(struct nlmsghdr *hdr, struct nl_pstate *npt)
return (error); return (error);
} }
static int
pf_handle_clear_addrs(struct nlmsghdr *hdr, struct nl_pstate *npt)
{
struct pfioc_table attrs = { 0 };
struct nl_writer *nw = npt->nw;
struct genlmsghdr *ghdr_new;
int error;
int ndel;
error = nl_parse_nlmsg(hdr, &table_parser, npt, &attrs);
if (error != 0)
return (error);
PF_RULES_WLOCK();
error = pfr_clr_addrs(&attrs.pfrio_table, &ndel,
attrs.pfrio_flags | PFR_FLAG_USERIOCTL);
PF_RULES_WUNLOCK();
if (error)
return (error);
if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr)))
return (ENOMEM);
ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr);
ghdr_new->cmd = PFNL_CMD_CLR_ADDRS;
ghdr_new->version = 0;
ghdr_new->reserved = 0;
nlattr_add_u64(nw, PF_T_NBR_DELETED, ndel);
if (!nlmsg_end(nw))
return (ENOMEM);
return (error);
}
static const struct nlhdr_parser *all_parsers[] = { static const struct nlhdr_parser *all_parsers[] = {
&state_parser, &state_parser,
&addrule_parser, &addrule_parser,
@@ -2302,6 +2339,13 @@ static const struct genl_cmd pf_cmds[] = {
.cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL,
.cmd_priv = PRIV_NETINET_PF, .cmd_priv = PRIV_NETINET_PF,
}, },
{
.cmd_num = PFNL_CMD_CLR_ADDRS,
.cmd_name = "CRL_ADDRS",
.cmd_cb = pf_handle_clear_addrs,
.cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL,
.cmd_priv = PRIV_NETINET_PF,
},
}; };
void void
+1
View File
@@ -66,6 +66,7 @@ enum {
PFNL_CMD_DEL_TABLE = 28, PFNL_CMD_DEL_TABLE = 28,
PFNL_CMD_GET_TSTATS = 29, PFNL_CMD_GET_TSTATS = 29,
PFNL_CMD_CLR_TSTATS = 30, PFNL_CMD_CLR_TSTATS = 30,
PFNL_CMD_CLR_ADDRS = 31,
__PFNL_CMD_MAX, __PFNL_CMD_MAX,
}; };
#define PFNL_CMD_MAX (__PFNL_CMD_MAX -1) #define PFNL_CMD_MAX (__PFNL_CMD_MAX -1)