1 files changed, 40 insertions, 0 deletions
diff --git a/rdns_scan/zmap4rdns/src/socket.h b/rdns_scan/zmap4rdns/src/socket.h
new file mode 100644
index 0000000..8b9fe84
--- /dev/null
+++ b/rdns_scan/zmap4rdns/src/socket.h
@@ -0,0 +1,40 @@
+/*
+ * 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
+ */
+
+#ifndef ZMAP_SOCKET_H
+#define ZMAP_SOCKET_H
+
+#include <stdint.h>
+
+#include "../lib/includes.h"
+
+#ifdef PFRING
+
+#include <pfring_zc.h>
+
+typedef union {
+	int sock;
+	struct {
+		pfring_zc_queue *queue;
+		pfring_zc_pkt_buff **buffers;
+		int idx;
+	} pf;
+} sock_t;
+
+#else
+
+typedef struct {
+	int sock;
+} sock_t;
+
+#endif /* PFRING */
+
+sock_t get_dryrun_socket(void);
+sock_t get_socket(uint32_t id);
+
+#endif /* ZMAP_SOCKET_H */