src/tty.h

   1 /*
   2  * netsniff-ng - the packet sniffing beast
   3  * By Daniel Borkmann <daniel@netsniff-ng.org>
   4  * Copyright 2009, 2010 Daniel Borkmann.
   5  * Subject to the GPL, version 2.
   6  */
   7
   8 #ifndef TTY_H
   9 #define TTY_H
  10
  11 #include <termios.h>
  12 #include <string.h>
  13 #include <sys/ioctl.h>
  14
  15 #define DEFAULT_TTY_SIZE        80
  16
  17 #define __reset                 "0"
  18 #define __bold                  "1"
  19 #define __black                 "30"
  20 #define __red                   "31"
  21 #define __green                 "32"
  22 #define __yellow                "33"
  23 #define __blue                  "34"
  24 #define __magenta               "35"
  25 #define __cyan                  "36"
  26 #define __white                 "37"
  27 #define __on_black              "40"
  28 #define __on_red                "41"
  29 #define __on_green              "42"
  30 #define __on_yellow             "43"
  31 #define __on_blue               "44"
  32 #define __on_magenta            "45"
  33 #define __on_cyan               "46"
  34 #define __on_white              "47"
  35
  36 #define colorize_start(fore)            "\033[" __##fore "m"
  37 #define colorize_start_full(fore, back) "\033[" __##fore ";" __on_##back "m"
  38 #define colorize_end()                  "\033[" __reset "m"
  39 #define colorize_str(fore, text)                                     \
  40                 colorize_start(fore) text colorize_end()
  41 #define colorize_full_str(fore, back, text)                          \
  42                 colorize_start_full(fore, back) text colorize_end()
  43
  44 static inline int get_tty_size(void)
  45 {
  46 #ifdef TIOCGSIZE
  47         struct ttysize ts = {0};
  48         int ret = ioctl(0, TIOCGSIZE, &ts);
  49         return (ret == 0 ? ts.ts_cols : DEFAULT_TTY_SIZE);
  50 #elif defined(TIOCGWINSZ)
  51         struct winsize ts;
  52         memset(&ts, 0, sizeof(ts));
  53         int ret = ioctl(0, TIOCGWINSZ, &ts);
  54         return (ret == 0 ? ts.ws_col : DEFAULT_TTY_SIZE);
  55 #else
  56         return DEFAULT_TTY_SIZE;
  57 #endif /* TIOCGSIZE */
  58 }
  59
  60 #endif /* TTY_H */