usr.bin/resizewin/resizewin.c

   1 /*
   2  * resizewin
   3  *
   4  * Query terminal for size and inform the kernel
   5  *
   6  * Copyright 2015 EMC / Isilon Storage Division
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  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  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  */
  29
  30 #include <sys/cdefs.h>
  31 __FBSDID("$FreeBSD$");
  32 #include <sys/ioctl.h>
  33 #include <sys/time.h>
  34 #include <err.h>
  35 #include <errno.h>
  36 #include <fcntl.h>
  37 #include <stdio.h>
  38 #include <stdlib.h>
  39 #include <strings.h>
  40 #include <termios.h>
  41 #include <unistd.h>
  42
  43 /* screen doesn't support ESC[18t (return terminal size) so do it the hard way */
  44 static const char query[] =
  45     "\0337"             /* Save cursor position */
  46     "\033[r"            /* Scroll whole screen */
  47     "\033[999;999H"     /* Move cursor */
  48     "\033[6n"           /* Get cursor position */
  49     "\0338";            /* Restore cursor position */
  50 int
  51 main(__unused int argc, __unused char **argv)
  52 {
  53         struct termios old, new;
  54         struct winsize w;
  55         int ret, fd, cnt, error;
  56         char data[20];
  57         struct timeval then, now;
  58
  59         error = 0;
  60
  61         if ((fd = open("/dev/tty", O_RDWR | O_NONBLOCK)) == -1)
  62                 exit(1);
  63
  64         /* Disable echo */
  65         if (tcgetattr(fd, &old) == -1)
  66                 exit(1);
  67
  68         new = old;
  69         new.c_cflag |= (CLOCAL | CREAD);
  70         new.c_lflag &= ~(ICANON | ECHO | ECHOE | ISIG);
  71         if (tcsetattr(fd, TCSANOW, &new) == -1)
  72                 exit(1);
  73
  74         /* Discard input received so far */
  75         error = tcflush(fd, TCIOFLUSH);
  76         if (error != 0)
  77                 warn("tcflush");
  78
  79         if (write(fd, query, sizeof(query)) != sizeof(query)) {
  80                 error = 1;
  81                 goto out;
  82         }
  83
  84         /* Read the response */
  85         bzero(data, sizeof(data));
  86         gettimeofday(&then, NULL);
  87         cnt = 0;
  88         while (1) {
  89                 ret = read(fd, data + cnt, 1);
  90
  91                 if (ret == -1) {
  92                         if (errno == EAGAIN) {
  93                                 gettimeofday(&now, NULL);
  94                                 timersub(&now, &then, &now);
  95                                 if (now.tv_sec >= 2) {
  96                                         warnx("timeout reading from terminal");
  97                                         error = 1;
  98                                         goto out;
  99                                 }
 100
 101                                 usleep(20000);
 102                                 continue;
 103                         }
 104                         error = 1;
 105                         goto out;
 106                 }
 107                 if (data[cnt] == 'R')
 108                         break;
 109
 110                 cnt++;
 111                 if (cnt == sizeof(data) - 2) {
 112                         warnx("response too long");
 113                         error = 1;
 114                         goto out;
 115                 }
 116         }
 117
 118         /* Parse */
 119         if (sscanf(data, "\033[%hu;%huR", &w.ws_row, &w.ws_col) != 2) {
 120                 error = 1;
 121                 warnx("unable to parse response");
 122                 goto out;
 123         }
 124
 125         /* Finally, what we want */
 126         if (ioctl(fd, TIOCSWINSZ, &w) == -1)
 127                 error = 1;
 128  out:
 129         /* Restore echo */
 130         tcsetattr(fd, TCSANOW, &old);
 131
 132         close(fd);
 133         exit(error);
 134 }