usr.bin/tset/set.c

   1 /*-
   2  * Copyright (c) 1991, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by the University of
  16  *      California, Berkeley and its contributors.
  17  * 4. Neither the name of the University 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 REGENTS 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 REGENTS 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  * @(#)set.c    8.2 (Berkeley) 2/28/94
  34  * $FreeBSD: src/usr.bin/tset/set.c,v 1.7.2.1 2000/11/12 07:48:29 dg Exp $
  35  * $DragonFly: src/usr.bin/tset/set.c,v 1.2 2003/06/17 04:29:33 dillon Exp $
  36  */
  37
  38 #include <termios.h>
  39 #include <unistd.h>
  40 #include <stdio.h>
  41 #include "extern.h"
  42
  43 #define CHK(val, dft)   (val <= 0 ? dft : val)
  44
  45 int     set_tabs __P((void));
  46
  47 /*
  48  * Reset the terminal mode bits to a sensible state.  Very useful after
  49  * a child program dies in raw mode.
  50  */
  51 void
  52 reset_mode()
  53 {
  54         tcgetattr(STDERR_FILENO, &mode);
  55
  56 #if defined(VDISCARD) && defined(CDISCARD)
  57         mode.c_cc[VDISCARD] = CHK(mode.c_cc[VDISCARD], CDISCARD);
  58 #endif
  59         mode.c_cc[VEOF] = CHK(mode.c_cc[VEOF], CEOF);
  60         mode.c_cc[VERASE] = CHK(mode.c_cc[VERASE], CERASE);
  61 #if defined(VFLUSH) && defined(CFLUSH)
  62         mode.c_cc[VFLUSH] = CHK(mode.c_cc[VFLUSH], CFLUSH);
  63 #endif
  64         mode.c_cc[VINTR] = CHK(mode.c_cc[VINTR], CINTR);
  65         mode.c_cc[VKILL] = CHK(mode.c_cc[VKILL], CKILL);
  66 #if defined(VLNEXT) && defined(CLNEXT)
  67         mode.c_cc[VLNEXT] = CHK(mode.c_cc[VLNEXT], CLNEXT);
  68 #endif
  69         mode.c_cc[VQUIT] = CHK(mode.c_cc[VQUIT], CQUIT);
  70 #if defined(VREPRINT) && defined(CRPRNT)
  71         mode.c_cc[VREPRINT] = CHK(mode.c_cc[VREPRINT], CRPRNT);
  72 #endif
  73         mode.c_cc[VSTART] = CHK(mode.c_cc[VSTART], CSTART);
  74         mode.c_cc[VSTOP] = CHK(mode.c_cc[VSTOP], CSTOP);
  75         mode.c_cc[VSUSP] = CHK(mode.c_cc[VSUSP], CSUSP);
  76 #if defined(VWERASE) && defined(CWERASE)
  77         mode.c_cc[VWERASE] = CHK(mode.c_cc[VWERASE], CWERASE);
  78 #endif
  79
  80         mode.c_iflag &= ~(IGNBRK | PARMRK | INPCK | ISTRIP | INLCR | IGNCR
  81 #ifdef IUCLC
  82                           | IUCLC
  83 #endif
  84 #ifdef IXANY
  85                           | IXANY
  86 #endif
  87                           | IXOFF);
  88
  89         mode.c_iflag |= (BRKINT | IGNPAR | ICRNL | IXON
  90 #ifdef IMAXBEL
  91                          | IMAXBEL
  92 #endif
  93                          );
  94
  95         mode.c_oflag &= ~(0
  96 #ifdef OLCUC
  97                           | OLCUC
  98 #endif
  99 #ifdef OCRNL
 100                           | OCRNL
 101 #endif
 102 #ifdef ONOCR
 103                           | ONOCR
 104 #endif
 105 #ifdef ONLRET
 106                           | ONLRET
 107 #endif
 108 #ifdef OFILL
 109                           | OFILL
 110 #endif
 111 #ifdef OFDEL
 112                           | OFDEL
 113 #endif
 114 #ifdef NLDLY
 115                           | NLDLY | CRDLY | TABDLY | BSDLY | VTDLY | FFDLY
 116 #endif
 117                           );
 118
 119         mode.c_oflag |= (OPOST
 120 #ifdef ONLCR
 121                          | ONLCR
 122 #endif
 123                          );
 124
 125         mode.c_cflag &= ~(CSIZE | CSTOPB | PARENB | PARODD | CLOCAL);
 126         mode.c_cflag |= (CS8 | CREAD);
 127         mode.c_lflag &= ~(ECHONL | NOFLSH | TOSTOP
 128 #ifdef ECHOPTR
 129                           | ECHOPRT
 130 #endif
 131 #ifdef XCASE
 132                           | XCASE
 133 #endif
 134                           );
 135
 136         mode.c_lflag |= (ISIG | ICANON | ECHO | ECHOE | ECHOK
 137 #ifdef ECHOCTL
 138                          | ECHOCTL
 139 #endif
 140 #ifdef ECHOKE
 141                          | ECHOKE
 142 #endif
 143                          );
 144
 145         tcsetattr(STDERR_FILENO, TCSADRAIN, &mode);
 146 }
 147
 148 /*
 149  * Determine the erase, interrupt, and kill characters from the termcap
 150  * entry and command line and update their values in 'mode'.
 151  */
 152 void
 153 set_control_chars()
 154 {
 155         char *bp, *p, bs_char, buf[1024];
 156
 157         bp = buf;
 158         p = tgetstr("kb", &bp);
 159         if (p == NULL || p[1] != '\0')
 160                 p = tgetstr("bc", &bp);
 161         if (p != NULL && p[1] == '\0')
 162                 bs_char = p[0];
 163         else if (tgetflag("bs"))
 164                 bs_char = CTRL('h');
 165         else
 166                 bs_char = 0;
 167
 168         if (erasech == 0 && bs_char != 0 && !tgetflag("os"))
 169                 erasech = -1;
 170         if (erasech < 0)
 171                 erasech = (bs_char != 0) ? bs_char : CTRL('h');
 172
 173         if (mode.c_cc[VERASE] == 0 || erasech != 0)
 174                 mode.c_cc[VERASE] = erasech ? erasech : CERASE;
 175
 176         if (mode.c_cc[VINTR] == 0 || intrchar != 0)
 177                  mode.c_cc[VINTR] = intrchar ? intrchar : CINTR;
 178
 179         if (mode.c_cc[VKILL] == 0 || killch != 0)
 180                 mode.c_cc[VKILL] = killch ? killch : CKILL;
 181 }
 182
 183 /*
 184  * Set up various conversions in 'mode', including parity, tabs, returns,
 185  * echo, and case, according to the termcap entry.  If the program we're
 186  * running was named with a leading upper-case character, map external
 187  * uppercase to internal lowercase.
 188  */
 189 void
 190 set_conversions(usingupper)
 191         int usingupper;
 192 {
 193         if (tgetflag("UC") || usingupper) {
 194 #ifdef IUCLC
 195                 mode.c_iflag |= IUCLC;
 196                 mode.c_oflag |= OLCUC;
 197 #endif
 198         } else if (tgetflag("LC")) {
 199 #ifdef IUCLC
 200                 mode.c_iflag &= ~IUCLC;
 201                 mode.c_oflag &= ~OLCUC;
 202 #endif
 203         }
 204         mode.c_iflag &= ~(PARMRK | INPCK);
 205         mode.c_lflag |= ICANON;
 206         if (tgetflag("EP")) {
 207                 mode.c_cflag |= PARENB;
 208                 mode.c_cflag &= ~PARODD;
 209         }
 210         if (tgetflag("OP")) {
 211                 mode.c_cflag |= PARENB;
 212                 mode.c_cflag |= PARODD;
 213         }
 214
 215 #ifdef ONLCR
 216         mode.c_oflag |= ONLCR;
 217 #endif
 218         mode.c_iflag |= ICRNL;
 219         mode.c_lflag |= ECHO;
 220         mode.c_oflag |= OXTABS;
 221         if (tgetflag("NL")) {                   /* Newline, not linefeed. */
 222 #ifdef ONLCR
 223                 mode.c_oflag &= ~ONLCR;
 224 #endif
 225                 mode.c_iflag &= ~ICRNL;
 226         }
 227         if (tgetflag("HD"))                     /* Half duplex. */
 228                 mode.c_lflag &= ~ECHO;
 229         if (tgetflag("pt"))                     /* Print tabs. */
 230                 mode.c_oflag &= ~OXTABS;
 231         mode.c_lflag |= (ECHOE | ECHOK);
 232 }
 233
 234 /* Output startup string. */
 235 void
 236 set_init()
 237 {
 238         char *bp, buf[1024];
 239         int settle;
 240
 241         bp = buf;
 242         if (tgetstr("pc", &bp) != 0)            /* Get/set pad character. */
 243                 PC = buf[0];
 244
 245 #ifdef TAB3
 246         if (oldmode.c_oflag & (TAB3 | ONLCR | OCRNL | ONLRET)) {
 247                 oldmode.c_oflag &= (TAB3 | ONLCR | OCRNL | ONLRET);
 248                 tcsetattr(STDERR_FILENO, TCSADRAIN, &oldmode);
 249         }
 250 #endif
 251         settle = set_tabs();
 252
 253         if (isreset) {
 254                 bp = buf;
 255                 if (tgetstr("rs", &bp) != 0 || tgetstr("is", &bp) != 0) {
 256                         tputs(buf, 0, outc);
 257                         settle = 1;
 258                 }
 259                 bp = buf;
 260                 if (tgetstr("rf", &bp) != 0 || tgetstr("if", &bp) != 0) {
 261                         cat(buf);
 262                         settle = 1;
 263                 }
 264         }
 265
 266         if (settle) {
 267                 (void)putc('\r', stderr);
 268                 (void)fflush(stderr);
 269                 (void)sleep(1);                 /* Settle the terminal. */
 270         }
 271 }
 272
 273 /*
 274  * Set the hardware tabs on the terminal, using the ct (clear all tabs),
 275  * st (set one tab) and ch (horizontal cursor addressing) capabilities.
 276  * This is done before if and is, so they can patch in case we blow this.
 277  * Return nonzero if we set any tab stops, zero if not.
 278  */
 279 int
 280 set_tabs()
 281 {
 282         int c;
 283         char *capsp, *clear_tabs;
 284         char *set_column, *set_pos, *Set_tab, *tg_out;
 285         char caps[1024];
 286
 287         capsp = caps;
 288         Set_tab = tgetstr("st", &capsp);
 289
 290         if (Set_tab && (clear_tabs = tgetstr("ct", &capsp))) {
 291                 (void)putc('\r', stderr);       /* Force to left margin. */
 292                 tputs(clear_tabs, 0, outc);
 293         }
 294
 295         set_column = tgetstr("ch", &capsp);
 296         set_pos = set_column ? NULL : tgetstr("cm", &capsp);
 297
 298         if (Set_tab) {
 299                 for (c = 8; c < Columns; c += 8) {
 300                         /*
 301                          * Get to the right column.  "OOPS" is returned by
 302                          * tgoto() if it can't do the job.  (*snarl*)
 303                          */
 304                         tg_out = "OOPS";
 305                         if (set_column)
 306                                 tg_out = tgoto(set_column, 0, c);
 307                         if (*tg_out == 'O' && set_pos)
 308                                 tg_out = tgoto(set_pos, c, Lines - 1);
 309                         if (*tg_out != 'O')
 310                                 tputs(tg_out, 1, outc);
 311                         else
 312                                 (void)fprintf(stderr, "%s", "        ");
 313                         /* Set the tab. */
 314                         tputs(Set_tab, 0, outc);
 315                 }
 316                 putc('\r', stderr);
 317                 return (1);
 318         }
 319         return (0);
 320 }