Imported Upstream version 1.8.6p8
[debian/sudo] / 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 }