summaryrefslogtreecommitdiff
path: root/ring_tx.c
diff options
context:
space:
mode:
authorDaniel Borkmann <dborkman@redhat.com>2013-03-15 10:41:48 +0100
committerDaniel Borkmann <dborkman@redhat.com>2013-03-15 10:41:48 +0100
commit1a9fbac03c684f29cff9ac44875bd9504a89f54e (patch)
tree1b2e40dbe5dc1899ef5b62c4325c9b94c9c450fc /ring_tx.c
all: import netsniff-ng 0.5.8-rc0 source
We decided to get rid of the old Git history and start a new one for several reasons: *) Allow / enforce only high-quality commits (which was not the case for many commits in the history), have a policy that is more close to the one from the Linux kernel. With high quality commits, we mean code that is logically split into commits and commit messages that are signed-off and have a proper subject and message body. We do not allow automatic Github merges anymore, since they are total bullshit. However, we will either cherry-pick your patches or pull them manually. *) The old archive was about ~27MB for no particular good reason. This basically derived from the bad decision that also some PDF files where stored there. From this moment onwards, no binary objects are allowed to be stored in this repository anymore. The old archive is not wiped away from the Internet. You will still be able to find it, e.g. on git.cryptoism.org etc. Signed-off-by: Daniel Borkmann <dborkman@redhat.com> Signed-off-by: Tobias Klauser <tklauser@distanz.ch>
Diffstat (limited to 'ring_tx.c')
-rw-r--r--ring_tx.c136
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");
+ }
+}