1 files changed, 136 insertions, 0 deletions
diff --git a/ring_tx.c b/ring_tx.c
new file mode 100644
index 0000000..dc777e5
--- /dev/null
+++ b/ring_tx.c
@@ -0,0 +1,136 @@
+/*
+ * netsniff-ng - the packet sniffing beast
+ * Copyright 2009, 2010 Daniel Borkmann.
+ * Copyright 2009, 2010 Emmanuel Roullit.
+ * Subject to the GPL, version 2.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/mman.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <arpa/inet.h>
+#include <linux/if_ether.h>
+
+#include "die.h"
+#include "xmalloc.h"
+#include "ring_tx.h"
+#include "built_in.h"
+
+void set_packet_loss_discard(int sock)
+{
+	int ret, discard = 1;
+	ret = setsockopt(sock, SOL_PACKET, PACKET_LOSS, (void *) &discard,
+			 sizeof(discard));
+	if (ret < 0)
+		panic("setsockopt: cannot set packet loss");
+}
+
+void destroy_tx_ring(int sock, struct ring *ring)
+{
+	fmemset(&ring->layout, 0, sizeof(ring->layout));
+	setsockopt(sock, SOL_PACKET, PACKET_TX_RING, &ring->layout,
+		   sizeof(ring->layout));
+
+	munmap(ring->mm_space, ring->mm_len);
+	ring->mm_len = 0;
+
+	xfree(ring->frames);
+}
+
+void setup_tx_ring_layout(int sock, struct ring *ring, unsigned int size,
+			  int jumbo_support)
+{
+	fmemset(&ring->layout, 0, sizeof(ring->layout));
+
+	ring->layout.tp_block_size = (jumbo_support ?
+				      getpagesize() << 4 :
+				      getpagesize() << 2);
+	ring->layout.tp_frame_size = (jumbo_support ?
+				      TPACKET_ALIGNMENT << 12 :
+				      TPACKET_ALIGNMENT << 7);
+	ring->layout.tp_block_nr = size / ring->layout.tp_block_size;
+	ring->layout.tp_frame_nr = ring->layout.tp_block_size /
+				   ring->layout.tp_frame_size *
+				   ring->layout.tp_block_nr;
+
+	bug_on(ring->layout.tp_block_size < ring->layout.tp_frame_size);
+	bug_on((ring->layout.tp_block_size % ring->layout.tp_frame_size) != 0);
+	bug_on((ring->layout.tp_block_size % getpagesize()) != 0);
+}
+
+void create_tx_ring(int sock, struct ring *ring, int verbose)
+{
+	int ret;
+
+	set_sockopt_tpacket(sock);
+retry:
+	ret = setsockopt(sock, SOL_PACKET, PACKET_TX_RING, &ring->layout,
+			 sizeof(ring->layout));
+	if (errno == ENOMEM && ring->layout.tp_block_nr > 1) {
+		ring->layout.tp_block_nr >>= 1;
+		ring->layout.tp_frame_nr = ring->layout.tp_block_size / 
+					   ring->layout.tp_frame_size * 
+					   ring->layout.tp_block_nr;
+		goto retry;
+	}
+
+	if (ret < 0)
+		panic("Cannot allocate TX_RING!\n");
+
+	ring->mm_len = ring->layout.tp_block_size * ring->layout.tp_block_nr;
+
+	if (verbose) {
+		printf("TX: %.2Lf MiB, %u Frames, each %u Byte allocated\n",
+		       (long double) ring->mm_len / (1 << 20),
+		       ring->layout.tp_frame_nr, ring->layout.tp_frame_size);
+	}
+}
+
+void mmap_tx_ring(int sock, struct ring *ring)
+{
+	ring->mm_space = mmap(0, ring->mm_len, PROT_READ | PROT_WRITE,
+			      MAP_SHARED | MAP_LOCKED | MAP_POPULATE, sock, 0);
+	if (ring->mm_space == MAP_FAILED) {
+		destroy_tx_ring(sock, ring);
+		panic("Cannot mmap TX_RING!\n");
+	}
+}
+
+void alloc_tx_ring_frames(struct ring *ring)
+{
+	int i;
+	size_t len = ring->layout.tp_frame_nr * sizeof(*ring->frames);
+
+	ring->frames = xmalloc_aligned(len, CO_CACHE_LINE_SIZE);
+	fmemset(ring->frames, 0, len);
+
+	for (i = 0; i < ring->layout.tp_frame_nr; ++i) {
+		ring->frames[i].iov_len = ring->layout.tp_frame_size;
+		ring->frames[i].iov_base = ring->mm_space +
+					   (i * ring->layout.tp_frame_size);
+	}
+}
+
+void bind_tx_ring(int sock, struct ring *ring, int ifindex)
+{
+	int ret;
+
+	fmemset(&ring->s_ll, 0, sizeof(ring->s_ll));
+
+	ring->s_ll.sll_family = AF_PACKET;
+	ring->s_ll.sll_protocol = htons(ETH_P_ALL);
+	ring->s_ll.sll_ifindex = ifindex;
+	ring->s_ll.sll_hatype = 0;
+	ring->s_ll.sll_halen = 0;
+	ring->s_ll.sll_pkttype = 0;
+
+	ret = bind(sock, (struct sockaddr *) &ring->s_ll, sizeof(ring->s_ll));
+	if (ret < 0) {
+		destroy_tx_ring(sock, ring);
+		panic("Cannot bind TX_RING!\n");
+	}
+}