Page MenuHomeFreeBSD
Files F107139141

D42091.diff
No OneTemporary
Actions

D42091.diff
View Options

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);
}

File Metadata

Mime Type
text/plain
Expires
Sat, Jan 11, 5:41 PM (19 h, 37 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
15756230
Default Alt Text
D42091.diff (4 KB)

Event Timeline