Blame SOURCES/arpwatch-2.1a15-lookupiselect.patch

1c6a1b
Note by jsynacek:
1c6a1b
This patch should be rewritten. There's no reason to be using a static variable
1c6a1b
and returning its content from a function (in iterate_dev()). Also, some things
1c6a1b
should be simplified (like iterate_dev()).
1c6a1b
1c6a1b
diff -up ./arpwatch.c.iselect ./arpwatch.c
1c6a1b
--- ./arpwatch.c.iselect	2012-10-15 16:01:24.701335291 +0200
1c6a1b
+++ ./arpwatch.c	2012-10-15 16:07:18.626322639 +0200
1c6a1b
@@ -162,50 +162,52 @@ void dropprivileges(const char* user)
1c6a1b
 }
1c6a1b
 
1c6a1b
 char *
1c6a1b
-get_first_dev(pcap_t **pd, int *linktype, char *errbuf)
1c6a1b
+try_dev(char *interface, pcap_t **pd, int *linktype, char *errbuf)
1c6a1b
 {
1c6a1b
-	static char interface[IF_NAMESIZE + 1];
1c6a1b
 	register int snaplen, timeout;
1c6a1b
-	pcap_if_t *alldevs;
1c6a1b
-	pcap_if_t *dev;
1c6a1b
-	char *ret = NULL;
1c6a1b
 
1c6a1b
 	snaplen = max(sizeof(struct ether_header),
1c6a1b
 				  sizeof(struct fddi_header)) + sizeof(struct ether_arp);
1c6a1b
 	timeout = 1000;
1c6a1b
 
1c6a1b
-	if (pcap_findalldevs(&alldevs, errbuf) == -1) {
1c6a1b
-		(void)fprintf(stderr, "%s: lookup_device: %s\n",
1c6a1b
-					  prog, errbuf);
1c6a1b
-		exit(1);
1c6a1b
+	*pd = pcap_open_live(interface, snaplen, 1, timeout, errbuf);
1c6a1b
+	if (NULL == *pd) {
1c6a1b
+		syslog(LOG_ERR, "pcap open %s: %s", interface,  errbuf);
1c6a1b
+		return NULL;
1c6a1b
 	}
1c6a1b
+	*linktype = pcap_datalink(*pd);
1c6a1b
+	/* Must be ethernet or fddi */
1c6a1b
+	if (*linktype != DLT_EN10MB && *linktype != DLT_FDDI) {
1c6a1b
+		syslog(LOG_ERR, "(%s) Link layer type %d not ethernet or fddi",
1c6a1b
+			   interface, *linktype);
1c6a1b
+		pcap_close(*pd);
1c6a1b
+		return NULL;
1c6a1b
+	}
1c6a1b
+	return interface;
1c6a1b
+}
1c6a1b
 
1c6a1b
-	for (dev = alldevs; dev; dev = dev->next) {
1c6a1b
-		strncpy(interface, dev->name, strlen(dev->name)+1);
1c6a1b
-
1c6a1b
-		*pd = pcap_open_live(interface, snaplen, 1, timeout, errbuf);
1c6a1b
-		if (*pd == NULL) {
1c6a1b
-			syslog(LOG_ERR, "pcap open %s: %s, trying next...", interface, errbuf);
1c6a1b
-			continue;
1c6a1b
-			/* exit(1); */
1c6a1b
-		}
1c6a1b
+char *
1c6a1b
+iterate_dev(char *arginterface, pcap_t **pd, int *linktype, char *errbuf)
1c6a1b
+{
1c6a1b
+	static char interface[64 + 1];
1c6a1b
+	pcap_if_t *alldevs;
1c6a1b
+	pcap_if_t *dev;
1c6a1b
 
1c6a1b
-		*linktype = pcap_datalink(*pd);
1c6a1b
-		/* Must be ethernet or fddi */
1c6a1b
-		if (*linktype != DLT_EN10MB && *linktype != DLT_FDDI) {
1c6a1b
-			syslog(LOG_ERR, "(%s) Link layer type %d not ethernet or fddi, trying next...",
1c6a1b
-				   interface, *linktype);
1c6a1b
-			pcap_close(*pd);
1c6a1b
+	if (NULL != arginterface) {
1c6a1b
+		return try_dev(arginterface, pd, linktype, errbuf);
1c6a1b
+	} else {
1c6a1b
+		if (pcap_findalldevs(&alldevs, errbuf) == -1) {
1c6a1b
+			(void)fprintf(stderr, "%s: lookup_device: %s\n",
1c6a1b
+						  prog, errbuf);
1c6a1b
+			exit(1);
1c6a1b
 		}
1c6a1b
-		else {
1c6a1b
-			/* First match, use it */
1c6a1b
-			ret = interface;
1c6a1b
-			break;
1c6a1b
+		for (dev = alldevs; dev && (arginterface == NULL); dev = dev->next) {
1c6a1b
+			strncpy(interface, dev->name, strlen(dev->name)+1);
1c6a1b
+			arginterface = try_dev(interface, pd, linktype, errbuf);
1c6a1b
 		}
1c6a1b
-
1c6a1b
+		pcap_freealldevs(alldevs);
1c6a1b
+		return arginterface;
1c6a1b
 	}
1c6a1b
-	pcap_freealldevs(alldevs);
1c6a1b
-	return (ret);
1c6a1b
 }
1c6a1b
 
1c6a1b
 int
1c6a1b
@@ -315,8 +317,8 @@ main(int argc, char **argv)
1c6a1b
 	} else {
1c6a1b
 
1c6a1b
 		/* Determine interface if not specified */
1c6a1b
-		if (interface == NULL &&
1c6a1b
-			(interface = get_first_dev(&pd, &linktype, errbuf)) == NULL) {
1c6a1b
+		interface = iterate_dev(interface, &pd, &linktype, errbuf);
1c6a1b
+		if (interface == NULL) {
1c6a1b
 			(void)fprintf(stderr, "%s: lookup_device: no suitable interface found\n",
1c6a1b
 						  prog);
1c6a1b
 			exit(1);