usr.bin/tput/tput.c

   1 /*      $NetBSD: tput.c,v 1.18 2007/12/15 19:44:53 perry Exp $  */
   2
   3 /*-
   4  * Copyright (c) 1980, 1988, 1993
   5  *      The Regents of the University of California.  All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  * 3. Neither the name of the University nor the names of its contributors
  16  *    may be used to endorse or promote products derived from this software
  17  *    without specific prior written permission.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  29  * SUCH DAMAGE.
  30  */
  31
  32 #include <sys/cdefs.h>
  33 #ifndef lint
  34 __COPYRIGHT("@(#) Copyright (c) 1980, 1988, 1993\
  35  The Regents of the University of California.  All rights reserved.");
  36 #endif /* not lint */
  37
  38 #ifndef lint
  39 #if 0
  40 static char sccsid[] = "@(#)tput.c      8.3 (Berkeley) 4/28/95";
  41 #endif
  42 __RCSID("$NetBSD: tput.c,v 1.18 2007/12/15 19:44:53 perry Exp $");
  43 #endif /* not lint */
  44
  45 #include <termios.h>
  46
  47 #include <err.h>
  48 #include <stdio.h>
  49 #include <stdlib.h>
  50 #include <string.h>
  51 #include <termcap.h>
  52 #include <unistd.h>
  53
  54 static int    outc(int);
  55 static void   prlongname(char *);
  56 static void   setospeed(void);
  57 static void   usage(void) __dead;
  58 static char **process(const char *, char *, char **);
  59
  60 int
  61 main(int argc, char **argv)
  62 {
  63         int ch, exitval, n;
  64         char *cptr, *term, buf[1024], tbuf[1024];
  65         const char *p;
  66
  67         term = NULL;
  68         while ((ch = getopt(argc, argv, "T:")) != -1)
  69                 switch(ch) {
  70                 case 'T':
  71                         term = optarg;
  72                         break;
  73                 case '?':
  74                 default:
  75                         usage();
  76                 }
  77         argc -= optind;
  78         argv += optind;
  79
  80         if (!term && !(term = getenv("TERM")))
  81                 errx(2, "No terminal type specified and no TERM "
  82                     "variable set in the environment.");
  83         if (tgetent(tbuf, term) != 1)
  84                 err(2, "tgetent failure");
  85         setospeed();
  86         for (exitval = 0; (p = *argv) != NULL; ++argv) {
  87                 switch (*p) {
  88                 case 'c':
  89                         if (!strcmp(p, "clear"))
  90                                 p = "cl";
  91                         break;
  92                 case 'i':
  93                         if (!strcmp(p, "init"))
  94                                 p = "is";
  95                         break;
  96                 case 'l':
  97                         if (!strcmp(p, "longname")) {
  98                                 prlongname(tbuf);
  99                                 continue;
 100                         }
 101                         break;
 102                 case 'r':
 103                         if (!strcmp(p, "reset"))
 104                                 p = "rs";
 105                         break;
 106                 }
 107                 cptr = buf;
 108                 if (tgetstr(p, &cptr))
 109                         argv = process(p, buf, argv);
 110                 else if ((n = tgetnum(p)) != -1)
 111                         (void)printf("%d\n", n);
 112                 else
 113                         exitval = !tgetflag(p);
 114
 115                 if (argv == NULL)
 116                         break;
 117         }
 118         return argv ? exitval : 2;
 119 }
 120
 121 static void
 122 prlongname(char *buf)
 123 {
 124         int savech;
 125         char *p, *savep;
 126
 127         for (p = buf; *p && *p != ':'; ++p)
 128                 continue;
 129         savech = *(savep = p);
 130         for (*p = '\0'; p >= buf && *p != '|'; --p)
 131                 continue;
 132         (void)printf("%s\n", p + 1);
 133         *savep = savech;
 134 }
 135
 136 static char **
 137 process(const char *cap, char *str, char **argv)
 138 {
 139         static const char errfew[] =
 140             "Not enough arguments (%d) for capability `%s'";
 141         static const char errmany[] =
 142             "Too many arguments (%d) for capability `%s'";
 143         static const char erresc[] =
 144             "Unknown %% escape `%c' for capability `%s'";
 145         char *cp;
 146         int arg_need, arg_rows, arg_cols;
 147
 148         /* Count how many values we need for this capability. */
 149         for (cp = str, arg_need = 0; *cp != '\0'; cp++)
 150                 if (*cp == '%')
 151                             switch (*++cp) {
 152                             case 'd':
 153                             case '2':
 154                             case '3':
 155                             case '.':
 156                             case '+':
 157                                     arg_need++;
 158                                     break;
 159                             case '%':
 160                             case '>':
 161                             case 'i':
 162                             case 'r':
 163                             case 'n':
 164                             case 'B':
 165                             case 'D':
 166                                     break;
 167                             default:
 168                                 /*
 169                                  * hpux has lot's of them, but we complain
 170                                  */
 171                                  errx(2, erresc, *cp, cap);
 172                             }
 173
 174         /* And print them. */
 175         switch (arg_need) {
 176         case 0:
 177                 (void)tputs(str, 1, outc);
 178                 break;
 179         case 1:
 180                 arg_cols = 0;
 181
 182                 if (*++argv == NULL || *argv[0] == '\0')
 183                         errx(2, errfew, 1, cap);
 184                 arg_rows = atoi(*argv);
 185
 186                 (void)tputs(tgoto(str, arg_cols, arg_rows), 1, outc);
 187                 break;
 188         case 2:
 189                 if (*++argv == NULL || *argv[0] == '\0')
 190                         errx(2, errfew, 2, cap);
 191                 arg_rows = atoi(*argv);
 192
 193                 if (*++argv == NULL || *argv[0] == '\0')
 194                         errx(2, errfew, 2, cap);
 195                 arg_cols = atoi(*argv);
 196
 197                 (void)tputs(tgoto(str, arg_cols, arg_rows), arg_rows, outc);
 198                 break;
 199
 200         default:
 201                 errx(2, errmany, arg_need, cap);
 202         }
 203         return argv;
 204 }
 205
 206 static void
 207 setospeed(void)
 208 {
 209 #undef ospeed
 210         extern short ospeed;
 211         struct termios t;
 212
 213         if (tcgetattr(STDOUT_FILENO, &t) != -1)
 214                 ospeed = 0;
 215         else
 216                 ospeed = cfgetospeed(&t);
 217 }
 218
 219 static int
 220 outc(c)
 221         int c;
 222 {
 223         return putchar(c);
 224 }
 225
 226 static void
 227 usage(void)
 228 {
 229         (void)fprintf(stderr,
 230             "Usage: %s [-T term] attribute [attribute-args] ...\n",
 231             getprogname());
 232         exit(2);
 233 }