Initial revision

nsctl.c

@@ -0,0 +1,138 @@
+#include <stdio.h>
+#include <arpa/inet.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <malloc.h>
+#include <netdb.h>
+#include <string.h>
+#include <sys/time.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <signal.h>
+#include <stdarg.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <time.h>
+#include "control.h"
+
+struct { char *command; int pkt_type; int params; } commands[] = {
+	{ "load_plugin", PKT_LOAD_PLUGIN, 1 },
+	{ "unload_plugin", PKT_UNLOAD_PLUGIN, 1 },
+	{ "garden", PKT_GARDEN, 1 },
+	{ "ungarden", PKT_UNGARDEN, 1 },
+};
+
+char *dest_host = NULL;
+unsigned int dest_port = 1702;
+int udpfd;
+
+int main(int argc, char *argv[])
+{
+	int len = 0;
+	int dest_ip = 0;
+	int pkt_type = 0;
+	char *packet = NULL;
+	int i;
+
+	setbuf(stdout, NULL);
+
+	if (argc < 3)
+	{
+		printf("Usage: %s <host> <command> [args...]\n", argv[0]);
+		return 1;
+	}
+
+	dest_host = strdup(argv[1]);
+
+	{
+		// Init socket
+		int on = 1;
+		struct sockaddr_in addr;
+
+		memset(&addr, 0, sizeof(addr));
+		addr.sin_family = AF_INET;
+		addr.sin_port = htons(1703);
+		udpfd = socket(AF_INET, SOCK_DGRAM, 17);
+		setsockopt(udpfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
+		if (bind(udpfd, (void *) &addr, sizeof(addr)) < 0)
+		{
+			perror("bind");
+			return(1);
+		}
+	}
+
+	{
+		struct hostent *h = gethostbyname(dest_host);
+		if (h) dest_ip = ntohl(*(unsigned int *)h->h_addr);
+		if (!dest_ip) dest_ip = ntohl(inet_addr(dest_host));
+		if (!dest_ip)
+		{
+			printf("Can't resolve \"%s\"\n", dest_host);
+			return 0;
+		}
+	}
+
+	if (!(packet = calloc(1400, 1)))
+	{
+		perror("calloc");
+		return(1);
+	}
+
+	srand(time(NULL));
+
+	// Deal with command & params
+	for (i = 0; i < (sizeof(commands) / sizeof(commands[0])); i++)
+	{
+		if (strcasecmp(commands[i].command, argv[2]) == 0)
+		{
+			int p;
+			pkt_type = commands[i].pkt_type;
+			len = new_packet(pkt_type, packet);
+			if (argc < (commands[i].params + 3))
+			{
+				printf("Not enough parameters for %s\n", argv[2]);
+				return 1;
+			}
+			for (p = 0; p < commands[i].params; p++)
+			{
+				strncpy((packet + len), argv[p + 3], 1400 - len);
+				len += strlen(argv[p + 3]) + 1;
+			}
+			break;
+		}
+	}
+	if (!pkt_type)
+	{
+		printf("Unknown command\n");
+		return 1;
+	}
+
+	send_packet(udpfd, dest_ip, dest_port, packet, len);
+
+	{
+		int n;
+		fd_set r;
+		struct timeval timeout;
+
+		FD_ZERO(&r);
+		FD_SET(udpfd, &r);
+		timeout.tv_sec = 1;
+		timeout.tv_usec = 0;
+
+		n = select(udpfd + 1, &r, 0, 0, &timeout);
+		if (n <= 0)
+		{
+			printf("Timeout waiting for packet\n");
+			return 0;
+		}
+	}
+	if ((len = read_packet(udpfd, packet)))
+	{
+		printf("Received ");
+		dump_packet(packet, stdout);
+	}
+
+	return 0;
+}
+