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