diff options
Diffstat (limited to 'examples/nfacct-add.c')
-rw-r--r-- | examples/nfacct-add.c | 22 |
1 files changed, 16 insertions, 6 deletions
diff --git a/examples/nfacct-add.c b/examples/nfacct-add.c index 64f94d1..40e0e31 100644 --- a/examples/nfacct-add.c +++ b/examples/nfacct-add.c @@ -10,18 +10,28 @@ int main(int argc, char *argv[]) char buf[MNL_SOCKET_BUFFER_SIZE]; struct nlmsghdr *nlh; uint32_t portid, seq; - struct nfacct nfacct = {}; + struct nfacct *nfacct; int ret; if (argc != 2) { fprintf(stderr, "Usage: %s [name]\n", argv[0]); exit(EXIT_FAILURE); } - strncpy(nfacct.name, argv[1], NFACCT_NAME_MAX); - nfacct.name[NFACCT_NAME_MAX-1] = '\0'; - nlh = nfacct_add(buf, &nfacct); - seq = nlh->nlmsg_seq = time(NULL); + nfacct = nfacct_alloc(); + if (nfacct == NULL) { + perror("OOM"); + exit(EXIT_FAILURE); + } + + nfacct_attr_set(nfacct, NFACCT_ATTR_NAME, argv[1]); + + seq = time(NULL); + nlh = nfacct_nlmsg_build_hdr(buf, NFNL_MSG_ACCT_NEW, + NLM_F_CREATE | NLM_F_ACK, seq); + nfacct_nlmsg_build_payload(nlh, nfacct); + + nfacct_free(nfacct); nl = mnl_socket_open(NETLINK_NETFILTER); if (nl == NULL) { @@ -42,7 +52,7 @@ int main(int argc, char *argv[]) ret = mnl_socket_recvfrom(nl, buf, sizeof(buf)); while (ret > 0) { - ret = mnl_cb_run(buf, ret, seq, portid, nfacct_list_cb, NULL); + ret = mnl_cb_run(buf, ret, seq, portid, NULL, NULL); if (ret <= 0) break; ret = mnl_socket_recvfrom(nl, buf, sizeof(buf)); |