Initial import.

nethogs.cpp

@@ -0,0 +1,175 @@
+/* nethogs.cpp
+ *
+ * Updates:
+ *
+ *     14/02/04 (Fabian) 
+ *               -Refresh delay
+ *               -Help
+ *               -Handling command-line options
+ *
+ *     06/04/04 (Fabian)
+ *		 -getLocal
+ *		 -I/O by balance	
+ *		 -forceExit
+ *     
+ *     10/05/04 (Arnout)
+ *               -cleanups
+ *               -splitting out incoming and outgoing traffic 
+ *                (based on 'I/O by balance' by Fabian)
+ */
+
+#include "nethogs.h"
+
+#include <iostream>
+#include <stdio.h>
+#include <stdlib.h>
+#include <assert.h>
+#include <unistd.h>
+#include <signal.h>
+#include <string>
+#include <string.h>
+
+#include <ncurses.h>
+
+extern "C" {
+	#include <pcap.h>
+}
+
+#include "packet.h"
+#include "connection.h"
+#include "process.h"
+#include "refresh.h"
+
+bool needrefresh = false;
+unsigned refreshdelay = 1;
+
+const char version[] = " version " VERSION "." SUBVERSION "." MINORVERSION;
+
+timeval curtime;
+std::string * caption;
+
+
+
+void process (u_char * args, const struct pcap_pkthdr * header, const u_char * m_packet)
+{
+	curtime = header->ts;
+
+	Packet * packet = getPacket (header, m_packet);
+	if (packet == NULL)
+		return;
+
+	Connection * connection = findConnection(packet);
+	if (connection != NULL)
+	{
+		connection->add(packet);
+		return;
+	}
+	connection = new Connection (packet);
+	Process * process = getProcess(connection);
+	//process->addConnection (connection);
+	
+	if (needrefresh)
+	{
+		do_refresh();
+		needrefresh = false;
+	}
+
+	return;
+}
+
+void quit_cb (int i)
+{
+	procclean();
+	clear();
+	endwin();
+	exit(0);
+}
+
+void forceExit(const char *msg)
+{
+        clear();
+        endwin();
+        std::cerr << msg << std::endl;
+        exit(0);
+}
+
+static void versiondisplay(void)
+{
+
+	std::cerr << version << "\n";
+}
+
+static void help(void)
+{
+	std::cerr << "usage: nethogs [-V] [-d] [device]\n";
+	std::cerr << "		-V : prints version.\n";
+	std::cerr << "		-d : delay for update refresh rate in seconds. default is 1.\n";
+	std::cerr << "		device : device to monitor. default is eth0\n";
+}
+
+int main (int argc, char** argv)
+{
+	char* dev = strdup("eth0");
+
+	for (argv++; *argv; argv++)
+	{
+		if (**argv=='-')
+		{
+			(*argv)++;
+			switch(**argv)
+			{
+				case 'V': versiondisplay();
+					  exit(0);
+				case 'h': help();
+					  exit(0);
+				case 'd': if (argv[1])
+					  {
+						argv++;
+						refreshdelay=atoi(*argv);
+					  }
+					  break;
+				default : help();
+					  exit(0);
+			}
+		}
+		else
+		{
+			dev = strdup(*argv);
+		}
+	}
+#if DEBUG
+#else
+	initscr();
+	raw();
+	noecho();
+	cbreak();
+#endif
+	getLocal(dev);
+
+	caption = new std::string ("NetHogs");
+	caption->append(version);
+	caption->append(", running at ");
+	caption->append(dev);
+
+	if (NEEDROOT && (getuid() != 0))
+		forceExit("You need to be root to run NetHogs !");
+
+	char errbuf[PCAP_ERRBUF_SIZE];
+
+	pcap_t * handle;
+	handle = pcap_open_live(dev, BUFSIZ, 0, 1000, errbuf);
+
+	signal (SIGALRM, &alarm_cb);
+	signal (SIGINT, &quit_cb);
+	alarm (refreshdelay);
+	while (1)
+	{
+		pcap_dispatch (handle, -1, process, NULL);
+		if (needrefresh)
+		{
+			do_refresh(); 
+			needrefresh = false;
+		}
+	}
+}
+