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
  40 #define colorize_str(fore, text)                                     \
  41                 colorize_start(fore) text colorize_end()
  42 #define colorize_full_str(fore, back, text)                          \
  43                 colorize_start_full(fore, back) text colorize_end()
  44
  45 static inline int get_tty_size(void)
  46 {
  47 #ifdef TIOCGSIZE
  48         struct ttysize ts = {0};
  49         int ret = ioctl(0, TIOCGSIZE, &ts);
  50         return (ret == 0 ? ts.ts_cols : DEFAULT_TTY_SIZE);
  51 #elif defined(TIOCGWINSZ)
  52         struct winsize ts = {0};
  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 static inline void set_tty_invisible(struct termios *orig)
  61 {
  62         struct termios now;
  63
  64         setvbuf(stdout, NULL, _IONBF ,0);
  65
  66         tcgetattr(0, orig);
  67         memcpy(&now, orig, sizeof(*orig));
  68         now.c_lflag &= ~(ISIG | ICANON | ECHO);
  69         now.c_cc[VMIN] = 1;
  70         now.c_cc[VTIME] = 2;
  71         tcsetattr(0, TCSANOW, &now);
  72 }
  73
  74 static inline void set_tty_visible(struct termios *orig)
  75 {
  76         tcsetattr(0, TCSANOW, orig);
  77 }
  78
  79 #endif /* TTY_H */