xtables-addons/extensions/pknock/pknlusr.c

#include <sys/socket.h>
#include <unistd.h>
#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <linux/netlink.h>
#include <linux/connector.h>

#include "xt_pknock.h"

#define GROUP 1

int main(void)
{
	int status;
	int group = GROUP;
	struct sockaddr_nl local_addr = {.nl_family = AF_NETLINK};
	int sock_fd, buf_size;
	unsigned char *buf;

	sock_fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_CONNECTOR);

	if (sock_fd == -1) {
		perror("socket()");
		return 1;
	}

	local_addr.nl_groups = group;
	status = bind(sock_fd, (struct sockaddr *)&local_addr, sizeof(local_addr));
	if (status == -1) {
		close(sock_fd);
		perror("bind()");
		return 1;
	}

	buf_size = sizeof(struct xt_pknock_nl_msg) + sizeof(struct cn_msg) + sizeof(struct nlmsghdr);
	buf = malloc(buf_size);

	if (!buf) {
		perror("malloc()");
		return 1;
	}

	while(1) {
		struct xt_pknock_nl_msg *nlmsg;
		const char *ip;
		char ipbuf[48];

		memset(buf, 0, buf_size);
		status = recv(sock_fd, buf, buf_size, 0);
		if (status <= 0) {
			perror("recv()");
			return 1;
		}
		nlmsg = (struct xt_pknock_nl_msg *)(buf + sizeof(struct cn_msg) + sizeof(struct nlmsghdr));
		ip = inet_ntop(AF_INET, &nlmsg->peer_ip, ipbuf, sizeof(ipbuf));
		printf("rule_name: %s - ip %s\n", nlmsg->rule_name, ip);

	}

	close(sock_fd);

	free(buf);

	return 0;
}