xutils.h

   1 /*
   2  * netsniff-ng - the packet sniffing beast
   3  * By Daniel Borkmann <daniel@netsniff-ng.org>
   4  * Copyright 2009, 2010 Daniel Borkmann.
   5  * Copyright 2009, 2010 Emmanuel Roullit.
   6  * Subject to the GPL, version 2.
   7  */
   8
   9 #ifndef XSYS_H
  10 #define XSYS_H
  11
  12 #define _GNU_SOURCE
  13 #include <errno.h>
  14 #include <stdint.h>
  15 #include <sys/socket.h>
  16 #include <unistd.h>
  17 #include <linux/ethtool.h>
  18 #include <linux/if.h>
  19 #include <linux/wireless.h>
  20 #include <poll.h>
  21 #include <sys/poll.h>
  22 #include <sched.h>
  23 #include <stdbool.h>
  24 #include <sys/resource.h>
  25 #include <sys/time.h>
  26
  27 #include "die.h"
  28 #include "built_in.h"
  29
  30 enum {
  31         sock_rmem_max = 0,
  32         sock_rmem_def,
  33         sock_wmem_max,
  34         sock_wmem_def,
  35 };
  36
  37 #define SMEM_SUG_MAX    104857600
  38 #define SMEM_SUG_DEF    4194304
  39
  40 extern int af_socket(int af);
  41 extern int pf_socket(void);
  42 extern int adjust_dbm_level(int in_dbm, int dbm_val);
  43 extern int wireless_sigqual(const char *ifname, struct iw_statistics *stats);
  44 extern int wireless_rangemax_sigqual(const char *ifname);
  45 extern u32 wireless_bitrate(const char *ifname);
  46 extern u32 ethtool_bitrate(const char *ifname);
  47 extern u32 device_bitrate(const char *ifname);
  48 extern int ethtool_drvinf(const char *ifname, struct ethtool_drvinfo *drvinf);
  49 extern int ethtool_link(const char *ifname);
  50 extern int device_mtu(const char *ifname);
  51 extern int device_address(const char *ifname, int af,
  52                           struct sockaddr_storage *ss);
  53 extern int device_irq_number(const char *ifname);
  54 extern int device_set_irq_affinity_list(int irq, unsigned long from,
  55                                         unsigned long to);
  56 extern int device_bind_irq_to_cpu(int irq, int cpu);
  57 extern void sock_print_net_stats(int sock, unsigned long skipped);
  58 extern int device_ifindex(const char *ifname);
  59 extern short device_get_flags(const char *ifname);
  60 extern void device_set_flags(const char *ifname, const short flags);
  61 extern void drop_privileges(bool enforce, uid_t uid, gid_t gid);
  62 extern void xlockme(void);
  63 extern void xunlockme(void);
  64 extern int set_nonblocking(int fd);
  65 extern int set_nonblocking_sloppy(int fd);
  66 extern int set_reuseaddr(int fd);
  67 extern void set_sock_prio(int fd, int prio);
  68 extern void set_tcp_cork(int fd);
  69 extern void set_tcp_uncork(int fd);
  70 extern void set_udp_cork(int fd);
  71 extern void set_udp_uncork(int fd);
  72 extern void set_sock_cork(int fd, int udp);
  73 extern void set_sock_uncork(int fd, int udp);
  74 extern void set_tcp_nodelay(int fd);
  75 extern void set_socket_keepalive(int fd);
  76 extern int set_ipv6_only(int fd);
  77 extern void set_mtu_disc_dont(int fd);
  78 extern int get_system_socket_mem(int which);
  79 extern void set_system_socket_mem(int which, int val);
  80 extern void register_signal(int signal, void (*handler)(int));
  81 extern void register_signal_f(int signal, void (*handler)(int), int flags);
  82 extern int get_tty_size(void);
  83 extern short enter_promiscuous_mode(char *ifname);
  84 extern void leave_promiscuous_mode(char *ifname, short oldflags);
  85 extern int device_up(char *ifname);
  86 extern int device_running(char *ifname);
  87 extern int device_up_and_running(char *ifname);
  88 extern int poll_error_maybe_die(int sock, struct pollfd *pfd);
  89 extern void set_epoll_descriptor(int fd_epoll, int action,
  90                                  int fd_toadd, int events);
  91 extern int set_epoll_descriptor2(int fd_epoll, int action,
  92                                  int fd_toadd, int events);
  93 extern void cpu_affinity(int cpu);
  94 extern int set_cpu_affinity(char *str, int inverted);
  95 extern int set_proc_prio(int prio);
  96 extern int set_sched_status(int policy, int priority);
  97 extern void ioprio_print(void);
  98 extern void set_ioprio_rt(void);
  99 extern void set_ioprio_be(void);
 100 extern void xusleep(const struct timespec *ts_delay);
 101 extern void xusleep2(long usecs);
 102 extern int xnanosleep(double seconds);
 103 extern int set_timeout(struct timeval *timeval, unsigned int msec);
 104 extern size_t strlcpy(char *dest, const char *src, size_t size);
 105 extern int slprintf(char *dst, size_t size, const char *fmt, ...)  __check_format_printf(3, 4);
 106 extern int slprintf_nocheck(char *dst, size_t size, const char *fmt, ...);
 107 extern noinline void *xmemset(void *s, int c, size_t n);
 108 extern char *getuint(char *in, uint32_t *out);
 109 extern char *strtrim_right(register char *p, register char c);
 110 extern char *strtrim_left(register char *p, register char c);
 111 extern void tputchar_safe(int c);
 112 extern void tputs_safe(const char *str, size_t len);
 113
 114 static inline int get_default_sched_policy(void)
 115 {
 116         return SCHED_FIFO;
 117 }
 118
 119 static inline int get_default_sched_prio(void)
 120 {
 121         return sched_get_priority_max(get_default_sched_policy());
 122 }
 123
 124 static inline int get_number_cpus(void)
 125 {
 126         return sysconf(_SC_NPROCESSORS_CONF);
 127 }
 128
 129 static inline int get_number_cpus_online(void)
 130 {
 131         return sysconf(_SC_NPROCESSORS_ONLN);
 132 }
 133
 134 static inline int get_default_proc_prio(void)
 135 {
 136         return -20;
 137 }
 138
 139 static inline struct timeval tv_subtract(struct timeval time1,
 140                                          struct timeval time2)
 141 {
 142         /* time1 - time2 */
 143         struct timeval result;
 144
 145         if ((time1.tv_sec < time2.tv_sec) || ((time1.tv_sec == time2.tv_sec) &&
 146             (time1.tv_usec <= time2.tv_usec))) {
 147                 result.tv_sec = result.tv_usec = 0;
 148         } else {
 149                 result.tv_sec = time1.tv_sec - time2.tv_sec;
 150                 if (time1.tv_usec < time2.tv_usec) {
 151                         result.tv_usec = time1.tv_usec + 1000000L -
 152                                          time2.tv_usec;
 153                         result.tv_sec--;
 154                 } else {
 155                         result.tv_usec = time1.tv_usec - time2.tv_usec;
 156                 }
 157         }
 158
 159         return result;
 160 }
 161
 162 static inline char *skips(char *p)
 163 {
 164         return strtrim_left(p, ' ');
 165 }
 166
 167 static inline char *skipchar(char *in, char c)
 168 {
 169         if (*in != c)
 170                 panic("Syntax error!\n");
 171         return ++in;
 172 }
 173
 174 #endif /* XSYS_H */