get_window_size.c revision 178826 
1/*
2 * Copyright (c) 1995, 1996, 1997, 1998 Kungliga Tekniska H�gskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 *
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * 3. Neither the name of the Institute nor the names of its contributors
18 *    may be used to endorse or promote products derived from this software
19 *    without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34#ifdef HAVE_CONFIG_H
35#include <config.h>
36RCSID("$Id: get_window_size.c 21005 2007-06-08 01:54:35Z lha $");
37#endif
38
39#include <stdlib.h>
40#ifdef HAVE_UNISTD_H
41#include <unistd.h>
42#endif
43#ifdef HAVE_SYS_TYPES_H
44#include <sys/types.h>
45#endif
46
47#if 0 /* Where were those needed? /confused */
48#ifdef HAVE_SYS_PROC_H
49#include <sys/proc.h>
50#endif
51
52#ifdef HAVE_SYS_TTY_H
53#include <sys/tty.h>
54#endif
55#endif
56
57#ifdef HAVE_TERMIOS_H
58#include <termios.h>
59#endif
60
61#include "roken.h"
62
63int ROKEN_LIB_FUNCTION
64get_window_size(int fd, struct winsize *wp)
65{
66    int ret = -1;
67
68    memset(wp, 0, sizeof(*wp));
69
70#if defined(TIOCGWINSZ)
71    ret = ioctl(fd, TIOCGWINSZ, wp);
72#elif defined(TIOCGSIZE)
73    {
74	struct ttysize ts;
75
76	ret = ioctl(fd, TIOCGSIZE, &ts);
77	if(ret == 0) {
78	    wp->ws_row = ts.ts_lines;
79	    wp->ws_col = ts.ts_cols;
80	}
81    }
82#elif defined(HAVE__SCRSIZE)
83    {
84	int dst[2];
85
86	_scrsize(dst);
87	wp->ws_row = dst[1];
88	wp->ws_col = dst[0];
89	ret = 0;
90    }
91#endif
92    if (ret != 0) {
93        char *s;
94        if((s = getenv("COLUMNS")))
95	    wp->ws_col = atoi(s);
96	if((s = getenv("LINES")))
97	    wp->ws_row = atoi(s);
98	if(wp->ws_col > 0 && wp->ws_row > 0)
99	    ret = 0;
100    }
101    return ret;
102}
103