corking.c

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