corking.c

   1 /*
   2  * netsniff-ng - the packet sniffing beast
   3  * Copyright 2009, 2010 Daniel Borkmann.
   4  * Subject to the GPL, version 2.
   5  */
   6
   7 #include <sys/types.h>
   8 #include <sys/socket.h>
   9 #include <linux/udp.h>
  10 #include <linux/tcp.h>
  11 #include <linux/in.h>
  12
  13 #include "corking.h"
  14 #include "die.h"
  15
  16 void set_udp_cork(int fd)
  17 {
  18         int ret, state = 1;
  19
  20         ret = setsockopt(fd, IPPROTO_UDP, UDP_CORK, &state, sizeof(state));
  21         if (unlikely(ret))
  22                 panic("Cannot cork UDP socket!\n");
  23 }
  24
  25 void set_udp_uncork(int fd)
  26 {
  27         int ret, state = 0;
  28
  29         ret = setsockopt(fd, IPPROTO_UDP, UDP_CORK, &state, sizeof(state));
  30         if (unlikely(ret))
  31                 panic("Cannot uncork UDP socket!\n");
  32 }
  33
  34 void set_tcp_cork(int fd)
  35 {
  36         int ret, state = 1;
  37
  38         ret = setsockopt(fd, IPPROTO_TCP, TCP_CORK, &state, sizeof(state));
  39         if (unlikely(ret))
  40                 panic("Cannot cork TCP socket!\n");
  41 }
  42
  43 void set_tcp_uncork(int fd)
  44 {
  45         int ret, state = 0;
  46
  47         ret = setsockopt(fd, IPPROTO_TCP, TCP_CORK, &state, sizeof(state));
  48         if (unlikely(ret))
  49                 panic("Cannot uncork TCP socket!\n");
  50 }
  51
  52 void set_sock_cork(int fd, bool is_udp)
  53 {
  54         if (is_udp)
  55                 set_udp_cork(fd);
  56         else
  57                 set_tcp_cork(fd);
  58 }
  59
  60 void set_sock_uncork(int fd, bool is_udp)
  61 {
  62         if (is_udp)
  63                 set_udp_uncork(fd);
  64         else
  65                 set_tcp_uncork(fd);
  66 }