X-Git-Url: https://tinc-vpn.org/git/browse?a=blobdiff_plain;f=src%2Ftincctl.c;h=000d383eb0dd4f863cad694d5b3f8ec43d3042c1;hb=453c44e7b27d4259461795ab4ec6ef264085dd28;hp=ede186ad4be94c1000c300cfee183ee4f4891b71;hpb=5d78e497f1c352c8d490eed1d44d128523a34572;p=tinc

diff --git a/src/tincctl.c b/src/tincctl.c
index ede186ad..000d383e 100644
--- a/src/tincctl.c
+++ b/src/tincctl.c
@@ -1,6 +1,6 @@
 /*
     tincctl.c -- Controlling a running tincd
-    Copyright (C) 2007-2009 Guus Sliepen <guus@tinc-vpn.org>
+    Copyright (C) 2007-2011 Guus Sliepen <guus@tinc-vpn.org>
 
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
@@ -26,6 +26,8 @@
 #include "control_common.h"
 #include "rsagen.h"
 #include "utils.h"
+#include "tincctl.h"
+#include "top.h"
 
 /* The name this program was run with. */
 char *program_name = NULL;
@@ -92,6 +94,11 @@ static void usage(bool status) {
 				"  debug N                    Set debug level\n"
 				"  retry                      Retry all outgoing connections\n"
 				"  reload                     Partial reload of configuration\n"
+				"  disconnect NODE            Close meta connection with NODE\n"
+#ifdef HAVE_CURSES
+				"  top                        Show real-time statistics\n"
+#endif
+				"  pcap                       Dump traffic in pcap format\n"
 				"\n");
 		printf("Report bugs to tinc@tinc-vpn.org.\n");
 	}
@@ -295,9 +302,10 @@ static void make_names(void) {
 	}
 }
 
-static bool recvline(int fd, char *line, size_t len) {
-	static char buffer[4096];
-	static size_t blen = 0;
+static char buffer[4096];
+static size_t blen = 0;
+
+bool recvline(int fd, char *line, size_t len) {
 	char *newline = NULL;
 
 	while(!(newline = memchr(buffer, '\n', blen))) {
@@ -322,7 +330,24 @@ static bool recvline(int fd, char *line, size_t len) {
 	return true;
 }
 
-static bool sendline(int fd, char *format, ...) {
+bool recvdata(int fd, char *data, size_t len) {
+	while(blen < len) {
+		int result = recv(fd, buffer + blen, sizeof buffer - blen, 0);
+		if(result == -1 && errno == EINTR)
+			continue;
+		else if(result <= 0)
+			return false;
+		blen += result;
+	}
+
+	memcpy(data, buffer, len);
+	memmove(buffer, buffer + len, blen - len);
+	blen -= len;
+
+	return true;
+}
+
+bool sendline(int fd, char *format, ...) {
 	static char buffer[4096];
 	char *p = buffer;
 	size_t blen = 0;
@@ -351,6 +376,57 @@ static bool sendline(int fd, char *format, ...) {
 	return true;	
 }
 
+void pcap(int fd, FILE *out) {
+	sendline(fd, "%d %d", CONTROL, REQ_PCAP);
+	char data[9018];
+
+	struct {
+		uint32_t magic;
+		uint16_t major;
+		uint16_t minor;
+		uint32_t tz_offset;
+		uint32_t tz_accuracy;
+		uint32_t snaplen;
+		uint32_t ll_type;
+	} header = {
+		0xa1b2c3d4,
+		2, 4,
+		0, 0,
+		sizeof data,
+		1,
+	};
+
+	struct {
+		uint32_t tv_sec;
+		uint32_t tv_usec;
+		uint32_t len;
+		uint32_t origlen;
+	} packet;
+
+	struct timeval tv;
+
+	fwrite(&header, sizeof header, 1, out);
+	fflush(out);
+
+	char line[32];
+	while(recvline(fd, line, sizeof line)) {
+		int code, req, len;
+		int n = sscanf(line, "%d %d %d", &code, &req, &len);
+		gettimeofday(&tv, NULL);
+		if(n != 3 || code != CONTROL || req != REQ_PCAP || len < 0 || len > sizeof data)
+			break;
+		if(!recvdata(fd, data, len))
+			break;
+		packet.tv_sec = tv.tv_sec;
+		packet.tv_usec = tv.tv_usec;
+		packet.len = len;
+		packet.origlen = len;
+		fwrite(&packet, sizeof packet, 1, out);
+		fwrite(data, len, 1, out);
+		fflush(out);
+	}
+}
+
 int main(int argc, char *argv[], char *envp[]) {
 	int fd;
 	int result;
@@ -516,7 +592,6 @@ int main(int argc, char *argv[], char *envp[]) {
 		}
 
 		bool do_graph = false;
-		int dumps = 1;
 
 		if(!strcasecmp(argv[optind+1], "nodes"))
 			sendline(fd, "%d %d", CONTROL, REQ_DUMP_NODES);
@@ -530,7 +605,6 @@ int main(int argc, char *argv[], char *envp[]) {
 			sendline(fd, "%d %d", CONTROL, REQ_DUMP_NODES);
 			sendline(fd, "%d %d", CONTROL, REQ_DUMP_EDGES);
 			do_graph = true;
-			dumps = 2;
 			printf("digraph {\n");
 		} else {
 			fprintf(stderr, "Unknown dump type '%s'.\n", argv[optind+1]);
@@ -595,6 +669,48 @@ int main(int argc, char *argv[], char *envp[]) {
 		return 0;
 	}
 
+	if(!strcasecmp(argv[optind], "connect")) {
+		if(argc != optind + 2) {
+			fprintf(stderr, "Invalid arguments.\n");
+			return 1;
+		}
+		char *name = argv[optind + 1];
+
+		sendline(fd, "%d %d %s", CONTROL, REQ_CONNECT, name);
+		if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_CONNECT || result) {
+			fprintf(stderr, "Could not connect to %s\n", name);
+			return 1;
+		}
+		return 0;
+	}
+
+	if(!strcasecmp(argv[optind], "disconnect")) {
+		if(argc != optind + 2) {
+			fprintf(stderr, "Invalid arguments.\n");
+			return 1;
+		}
+		char *name = argv[optind + 1];
+
+		sendline(fd, "%d %d %s", CONTROL, REQ_DISCONNECT, name);
+		if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_DISCONNECT || result) {
+			fprintf(stderr, "Could not disconnect %s\n", name);
+			return 1;
+		}
+		return 0;
+	}
+
+#ifdef HAVE_CURSES
+	if(!strcasecmp(argv[optind], "top")) {
+		top(fd);
+		return 0;
+	}
+#endif
+
+	if(!strcasecmp(argv[optind], "pcap")) {
+		pcap(fd, stdout);
+		return 0;
+	}
+
 	fprintf(stderr, "Unknown command `%s'.\n", argv[optind]);
 	usage(true);