git.gag.com Git - debian/sudo/blob - common/ttysize.c

   1 /*
   2  * Copyright (c) 2010 Todd C. Miller <Todd.Miller@courtesan.com>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  16  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  17  */
  18
  19 #include <config.h>
  20
  21 #include <sys/types.h>
  22 #include <sys/ioctl.h>
  23 #include <stdio.h>
  24 #ifdef STDC_HEADERS
  25 # include <stdlib.h>
  26 # include <stddef.h>
  27 #else
  28 # ifdef HAVE_STDLIB_H
  29 #  include <stdlib.h>
  30 # endif
  31 #endif /* STDC_HEADERS */
  32 #ifdef HAVE_UNISTD_H
  33 # include <unistd.h>
  34 #endif /* HAVE_UNISTD_H */
  35 #include <termios.h>
  36
  37 #include "missing.h"
  38 #include "sudo_debug.h"
  39
  40 /* Compatibility with older tty systems. */
  41 #if !defined(TIOCGWINSZ) && defined(TIOCGSIZE)
  42 # define TIOCGWINSZ     TIOCGSIZE
  43 # define winsize        ttysize
  44 # define ws_col         ts_cols
  45 # define ws_row         ts_lines
  46 #endif
  47
  48 #ifdef TIOCGWINSZ
  49 static int
  50 get_ttysize_ioctl(int *rowp, int *colp)
  51 {
  52     struct winsize wsize;
  53     debug_decl(get_ttysize_ioctl, SUDO_DEBUG_EXEC)
  54
  55     if (ioctl(STDERR_FILENO, TIOCGWINSZ, &wsize) == 0 &&
  56         wsize.ws_row != 0 && wsize.ws_col  != 0) {
  57         *rowp = wsize.ws_row;
  58         *colp = wsize.ws_col;
  59         debug_return_int(0);
  60     }
  61     debug_return_int(-1);
  62 }
  63 #else
  64 static int
  65 get_ttysize_ioctl(int *rowp, int *colp)
  66 {
  67     return -1;
  68 }
  69 #endif /* TIOCGWINSZ */
  70
  71 void
  72 get_ttysize(int *rowp, int *colp)
  73 {
  74     debug_decl(fork_cmnd, SUDO_DEBUG_EXEC)
  75
  76     if (get_ttysize_ioctl(rowp, colp) == -1) {
  77         char *p;
  78
  79         /* Fall back on $LINES and $COLUMNS. */
  80         if ((p = getenv("LINES")) == NULL || (*rowp = atoi(p)) <= 0)
  81             *rowp = 24;
  82         if ((p = getenv("COLUMNS")) == NULL || (*colp = atoi(p)) <= 0)
  83             *colp = 80;
  84     }
  85
  86     debug_return;
  87 }