From d7c59006cbfac80c9fb32c2674ec352c7717ea56 Mon Sep 17 00:00:00 2001
From: Bjorn Andersson <bjorn.andersson@linaro.org>
Date: Wed, 1 Jun 2016 06:44:36 -0700
Subject: [PATCH] qrtr-cfg: Introduce node-id configuration tool

The final version of QRTR that was merged into v4.7 of the Linux kernel
requires a configuration step where we set the local node address.

Signed-off-by: Bjorn Andersson <bjorn.andersson@linaro.org>
---
 Makefile  |  5 ++++-
 src/cfg.c | 65 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 69 insertions(+), 1 deletion(-)
 create mode 100644 src/cfg.c

diff --git a/Makefile b/Makefile
index bac45d3..675195c 100644
--- a/Makefile
+++ b/Makefile
@@ -15,6 +15,9 @@ CC := $(CROSS_COMPILE)gcc
 endif
 SFLAGS := -I$(shell $(CC) -print-file-name=include) -Wno-non-pointer-null
 
+$(proj)-cfg-srcs := \
+	src/cfg.c
+
 $(proj)-ns-srcs := \
 	src/ns.c \
 	src/map.c \
@@ -31,7 +34,7 @@ lib$(proj).so-srcs := \
 
 lib$(proj).so-cflags := -fPIC -Isrc
 
-targets := $(proj)-ns $(proj)-lookup lib$(proj).so
+targets := $(proj)-ns $(proj)-cfg $(proj)-lookup lib$(proj).so
 
 out := out
 src_to_obj = $(patsubst %.c,$(out)/obj/%.o,$(1))
diff --git a/src/cfg.c b/src/cfg.c
new file mode 100644
index 0000000..79dec37
--- /dev/null
+++ b/src/cfg.c
@@ -0,0 +1,65 @@
+#include <err.h>
+#include <limits.h>
+#include <linux/netlink.h>
+#include <linux/rtnetlink.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/socket.h>
+#include <sys/types.h>
+
+#include "qrtr.h"
+
+static void usage(void)
+{
+	extern char *__progname;
+
+	fprintf(stderr, "%s <node-id>\n", __progname);
+	exit(1);
+}
+
+int main(int argc, char **argv)
+{
+	struct {
+		struct nlmsghdr nh;
+		struct ifaddrmsg ifa;
+		char attrbuf[32];
+	} req;
+	struct rtattr *rta;
+	unsigned long addrul;
+	uint32_t addr;
+	char *ep;
+	int sock;
+	int ret;
+
+	if (argc != 2)
+		usage();
+
+	addrul = strtoul(argv[1], &ep, 10);
+	if (argv[1][0] == '\0' || *ep != '\0' || addrul >= UINT_MAX)
+		usage();
+	addr = addrul;
+
+	sock = socket(AF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE);
+	if (sock < 0)
+		err(1, "failed to create netlink socket");
+
+	memset(&req, 0, sizeof(req));
+	req.nh.nlmsg_len = NLMSG_SPACE(sizeof(struct ifaddrmsg));
+	req.nh.nlmsg_flags = NLM_F_REQUEST;
+	req.nh.nlmsg_type = RTM_NEWADDR;
+	req.ifa.ifa_family = AF_QIPCRTR;
+
+	rta = (struct rtattr *)(((char *) &req) + req.nh.nlmsg_len);
+	rta->rta_type = IFA_LOCAL;
+	rta->rta_len = RTA_LENGTH(sizeof(addr));
+	memcpy(RTA_DATA(rta), &addr, sizeof(addr));
+
+	req.nh.nlmsg_len += rta->rta_len;
+
+	ret = send(sock, &req, req.nh.nlmsg_len, 0);
+	if (ret < 0)
+		err(1, "failed to send netlink request");
+
+	return 0;
+}