1 files changed, 66 insertions, 0 deletions
diff --git a/rdns_scan/zmap4rdns/src/socket-bsd.c b/rdns_scan/zmap4rdns/src/socket-bsd.c
new file mode 100644
index 0000000..e55cd72
--- /dev/null
+++ b/rdns_scan/zmap4rdns/src/socket-bsd.c
@@ -0,0 +1,66 @@
+/*
+ * ZMap Copyright 2013 Regents of the University of Michigan
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not
+ * use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ */
+
+#include "socket.h"
+
+#include <errno.h>
+
+#include "../lib/includes.h"
+#include "../lib/logger.h"
+
+#include <sys/types.h>
+#include <sys/time.h>
+#include <sys/ioctl.h>
+#include <fcntl.h>
+#include <net/bpf.h>
+
+#include "state.h"
+
+sock_t get_socket(UNUSED uint32_t id)
+{
+	char file[32];
+	int bpf;
+	// Assume failure
+	sock_t ret;
+	ret.sock = -1;
+
+	if(zconf.send_ip_pkts && !zconf.dryrun) {
+		log_fatal("socket", "iplayer not supported on bsd");
+	}
+
+	// Try to find a valid bpf
+	for (int i = 0; i < 128; i++) {
+		snprintf(file, sizeof(file), "/dev/bpf%d", i);
+		bpf = open(file, O_WRONLY);
+		if (bpf != -1 || errno != EBUSY)
+			break;
+	}
+
+	// Make sure it worked
+	if (bpf < 0) {
+		return ret;
+	}
+
+	// Set up an ifreq to bind to
+	struct ifreq ifr;
+	memset(&ifr, 0, sizeof(ifr));
+	strlcpy(ifr.ifr_name, zconf.iface, sizeof(ifr.ifr_name));
+
+	// Bind the bpf to the interface
+	if (ioctl(bpf, BIOCSETIF, (char *)&ifr) < 0) {
+		return ret;
+	}
+
+	// Enable writing the address in
+	int write_addr_enable = 1;
+	if (ioctl(bpf, BIOCSHDRCMPLT, &write_addr_enable) < 0) {
+		return ret;
+	}
+	ret.sock = bpf;
+	return ret;
+}