1b6bc8
--- arpwatch-2.1a10/arpwatch.c	Sat Oct 14 05:07:35 2000
1b6bc8
+++ arpwatch-2.1a10/arpwatch.c	Sun Jun 10 16:22:57 2001
1b6bc8
@@ -62,7 +62,7 @@
1b6bc8
 #include <string.h>
1b6bc8
 #include <syslog.h>
1b6bc8
 #include <unistd.h>
1b6bc8
-
1b6bc8
+#include <pwd.h>
1b6bc8
 #include <pcap.h>
1b6bc8
 
1b6bc8
 #include "gnuc.h"
1b6bc8
@@ -141,6 +141,25 @@
1b6bc8
 int	sanity_fddi(struct fddi_header *, struct ether_arp *, int);
1b6bc8
 __dead	void usage(void) __attribute__((volatile));
1b6bc8
 
1b6bc8
+void dropprivileges(const char* user)
1b6bc8
+{
1b6bc8
+	struct passwd* pw;
1b6bc8
+	pw = getpwnam( user );
1b6bc8
+	if ( pw ) {
1b6bc8
+		if ( initgroups(pw->pw_name, NULL) != 0 || setgid(pw->pw_gid) != 0 ||
1b6bc8
+				 setuid(pw->pw_uid) != 0 ) {
1b6bc8
+			syslog(LOG_ERR, "Couldn't change to '%.32s' uid=%d gid=%d", user,
1b6bc8
+						 pw->pw_uid, pw->pw_gid);
1b6bc8
+			exit(1);
1b6bc8
+		}
1b6bc8
+	}
1b6bc8
+	else {
1b6bc8
+		syslog(LOG_ERR, "Couldn't find user '%.32s' in /etc/passwd", user);
1b6bc8
+		exit(1);
1b6bc8
+	}
1b6bc8
+	syslog(LOG_DEBUG, "Running as uid=%d gid=%d", getuid(), getgid());
1b6bc8
+}
1b6bc8
+
1b6bc8
 int
1b6bc8
 main(int argc, char **argv)
1b6bc8
 {
1b6bc8
@@ -153,6 +172,7 @@
1b6bc8
 	register char *interface, *rfilename;
1b6bc8
 	struct bpf_program code;
1b6bc8
 	char errbuf[PCAP_ERRBUF_SIZE];
1b6bc8
+	char* serveruser = NULL;
1b6bc8
 
1b6bc8
 	if (argv[0] == NULL)
1b6bc8
 		prog = "arpwatch";
1b6bc8
@@ -170,7 +190,7 @@
1b6bc8
 	interface = NULL;
1b6bc8
 	rfilename = NULL;
1b6bc8
 	pd = NULL;
1b6bc8
-	while ((op = getopt(argc, argv, "df:i:n:Nr:")) != EOF)
1b6bc8
+	while ((op = getopt(argc, argv, "df:i:n:Nr:u:")) != EOF)
1b6bc8
 		switch (op) {
1b6bc8
 
1b6bc8
 		case 'd':
1b6bc8
@@ -202,6 +222,16 @@
1b6bc8
 			rfilename = optarg;
1b6bc8
 			break;
1b6bc8
 
1b6bc8
+		case 'u':
1b6bc8
+			if ( optarg ) {
1b6bc8
+				serveruser = strdup(optarg);
1b6bc8
+			}
1b6bc8
+			else {
1b6bc8
+				fprintf(stderr, "%s: Need username after -u\n", prog);
1b6bc8
+				usage();
1b6bc8
+			}
1b6bc8
+			break;
1b6bc8
+
1b6bc8
 		default:
1b6bc8
 			usage();
1b6bc8
 		}
1b6bc8
@@ -283,8 +313,11 @@
1b6bc8
 	 * Revert to non-privileged user after opening sockets
1b6bc8
 	 * (not needed on most systems).
1b6bc8
 	 */
1b6bc8
-	setgid(getgid());
1b6bc8
-	setuid(getuid());
1b6bc8
+	/*setgid(getgid());*/
1b6bc8
+	/*setuid(getuid());*/
1b6bc8
+	if ( serveruser ) {
1b6bc8
+		dropprivileges( serveruser );
1b6bc8
+	}
1b6bc8
 
1b6bc8
 	/* Must be ethernet or fddi */
1b6bc8
 	linktype = pcap_datalink(pd);
1b6bc8
@@ -751,6 +784,6 @@
1b6bc8
 
1b6bc8
 	(void)fprintf(stderr, "Version %s\n", version);
1b6bc8
 	(void)fprintf(stderr, "usage: %s [-dN] [-f datafile] [-i interface]"
1b6bc8
-	    " [-n net[/width]] [-r file]\n", prog);
1b6bc8
+	    " [-n net[/width]] [-r file] [-u username]\n", prog);
1b6bc8
 	exit(1);
1b6bc8
 }