diff --git a/lib/libpfctl/libpfctl.h b/lib/libpfctl/libpfctl.h --- a/lib/libpfctl/libpfctl.h +++ b/lib/libpfctl/libpfctl.h @@ -413,6 +413,8 @@ const char *anchor, const char *anchor_call, uint32_t ticket, uint32_t pool_ticket); int pfctl_set_keepcounters(int dev, bool keep); +typedef int (*pfctl_get_state_fn)(struct pfctl_state *, void *); +int pfctl_get_states_iter(pfctl_get_state_fn f, void *arg); int pfctl_get_states(int dev, struct pfctl_states *states); void pfctl_free_states(struct pfctl_states *states); int pfctl_clear_states(int dev, const struct pfctl_kill *kill, diff --git a/lib/libpfctl/libpfctl.c b/lib/libpfctl/libpfctl.c --- a/lib/libpfctl/libpfctl.c +++ b/lib/libpfctl/libpfctl.c @@ -1203,10 +1203,11 @@ }; static int -pfctl_get_states_nl(struct snl_state *ss, struct pfctl_states *states) +pfctl_get_states_nl(struct snl_state *ss, pfctl_get_state_fn f, void *arg) { SNL_VERIFY_PARSERS(all_parsers); int family_id = snl_get_genl_family(ss, PFNL_FAMILY_NAME); + int ret; struct nlmsghdr *hdr; struct snl_writer nw; @@ -1219,42 +1220,71 @@ snl_send_message(ss, hdr); - bzero(states, sizeof(*states)); - TAILQ_INIT(&states->states); - struct snl_errmsg_data e = {}; while ((hdr = snl_read_reply_multi(ss, seq_id, &e)) != NULL) { - struct pfctl_state *s = malloc(sizeof(*s)); - bzero(s, sizeof(*s)); - if (s == NULL) { - pfctl_free_states(states); - return (ENOMEM); - } - if (!snl_parse_nlmsg(ss, hdr, &state_parser, s)) + struct pfctl_state s; + bzero(&s, sizeof(s)); + if (!snl_parse_nlmsg(ss, hdr, &state_parser, &s)) continue; - s->key[1].af = s->key[0].af; - s->key[1].proto = s->key[0].proto; + s.key[1].af = s.key[0].af; + s.key[1].proto = s.key[0].proto; - TAILQ_INSERT_TAIL(&states->states, s, entry); + ret = f(&s, arg); + if (ret != 0) + return (ret); } return (0); } int -pfctl_get_states(int dev __unused, struct pfctl_states *states) +pfctl_get_states_iter(pfctl_get_state_fn f, void *arg) { struct snl_state ss = {}; int error; snl_init(&ss, NETLINK_GENERIC); - error = pfctl_get_states_nl(&ss, states); + error = pfctl_get_states_nl(&ss, f, arg); snl_free(&ss); return (error); } +static int +pfctl_append_states(struct pfctl_state *s, void *arg) +{ + struct pfctl_state *new; + struct pfctl_states *states = (struct pfctl_states *)arg; + + new = malloc(sizeof(*s)); + if (new == NULL) + return (ENOMEM); + + memcpy(new, s, sizeof(*s)); + + TAILQ_INSERT_TAIL(&states->states, s, entry); + + return (0); +} + +int +pfctl_get_states(int dev __unused, struct pfctl_states *states) +{ + int ret; + + bzero(states, sizeof(*states)); + TAILQ_INIT(&states->states); + + ret = pfctl_get_states_iter(pfctl_append_states, states); + if (ret != 0) { + pfctl_free_states(states); + return (ret); + } + + return (0); +} + void pfctl_free_states(struct pfctl_states *states) { diff --git a/sbin/pfctl/pfctl.c b/sbin/pfctl/pfctl.c --- a/sbin/pfctl/pfctl.c +++ b/sbin/pfctl/pfctl.c @@ -1514,30 +1514,41 @@ return (0); } +struct pfctl_show_state_arg { + int opts; + int dotitle; + const char *iface; +}; + +static int +pfctl_show_state(struct pfctl_state *s, void *arg) +{ + struct pfctl_show_state_arg *a = (struct pfctl_show_state_arg *)arg; + + if (a->iface != NULL && strcmp(s->ifname, a->iface)) + return (0); + + if (a->dotitle) { + pfctl_print_title("STATES:"); + a->dotitle = 0; + } + print_state(s, a->opts); + + return (0); +} + int pfctl_show_states(int dev, const char *iface, int opts) { - struct pfctl_states states; - struct pfctl_state *s; - int dotitle = (opts & PF_OPT_SHOWALL); + struct pfctl_show_state_arg arg; - memset(&states, 0, sizeof(states)); + arg.opts = opts; + arg.dotitle = opts & PF_OPT_SHOWALL; + arg.iface = iface; - if (pfctl_get_states(dev, &states)) + if (pfctl_get_states_iter(pfctl_show_state, &arg)) return (-1); - TAILQ_FOREACH(s, &states.states, entry) { - if (iface != NULL && strcmp(s->ifname, iface)) - continue; - if (dotitle) { - pfctl_print_title("STATES:"); - dotitle = 0; - } - print_state(s, opts); - } - - pfctl_free_states(&states); - return (0); }