usr.bin/tput/tput.c

   1 /*-
   2  * Copyright (c) 1980, 1988, 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  * @(#) Copyright (c) 1980, 1988, 1993 The Regents of the University of California.  All rights reserved.
  34  * @(#)tput.c   8.2 (Berkeley) 3/19/94
  35  * $FreeBSD: src/usr.bin/tput/tput.c,v 1.7.6.3 2002/08/17 14:52:50 tjr Exp $
  36  * $DragonFly: src/usr.bin/tput/tput.c,v 1.5 2005/07/30 11:53:52 liamfoy Exp $
  37  */
  38
  39 #include <termios.h>
  40
  41 #include <err.h>
  42 #include <termcap.h>
  43 #include <stdio.h>
  44 #include <stdlib.h>
  45 #include <string.h>
  46 #include <unistd.h>
  47
  48 #undef putchar
  49 #define outc putchar
  50
  51 static void   prlongname(char *);
  52 static void   usage(void);
  53 static char **process(const char *, const char *, char **);
  54
  55 int
  56 main(int argc, char **argv)
  57 {
  58         int ch, exitval, n;
  59         char *cptr, *term, buf[1024], tbuf[1024];
  60         const char *p;
  61
  62         term = NULL;
  63         while ((ch = getopt(argc, argv, "T:")) != -1)
  64                 switch(ch) {
  65                 case 'T':
  66                         term = optarg;
  67                         break;
  68                 case '?':
  69                 default:
  70                         usage();
  71                 }
  72         argc -= optind;
  73         argv += optind;
  74
  75         if (!term && !(term = getenv("TERM")))
  76 errx(2, "no terminal type specified and no TERM environmental variable.");
  77         if (tgetent(tbuf, term) != 1)
  78                 err(3, "tgetent failure");
  79         for (exitval = 0; (p = *argv) != NULL; ++argv) {
  80                 switch (*p) {
  81                 case 'c':
  82                         if (!strcmp(p, "clear"))
  83                                 p = "cl";
  84                         break;
  85                 case 'i':
  86                         if (!strcmp(p, "init"))
  87                                 p = "is";
  88                         break;
  89                 case 'l':
  90                         if (!strcmp(p, "longname")) {
  91                                 prlongname(tbuf);
  92                                 continue;
  93                         }
  94                         break;
  95                 case 'r':
  96                         if (!strcmp(p, "reset"))
  97                                 p = "rs";
  98                         break;
  99                 }
 100                 cptr = buf;
 101                 if (tgetstr(p, &cptr))
 102                         argv = process(p, buf, argv);
 103                 else if ((n = tgetnum(p)) != -1)
 104                         (void)printf("%d\n", n);
 105                 else
 106                         exitval = !tgetflag(p);
 107         }
 108         exit(exitval);
 109 }
 110
 111 static void
 112 prlongname(char *buf)
 113 {
 114         int savech;
 115         char *p, *savep;
 116
 117         for (p = buf; *p && *p != ':'; ++p);
 118         savech = *(savep = p);
 119         for (*p = '\0'; p >= buf && *p != '|'; --p);
 120         (void)printf("%s\n", p + 1);
 121         *savep = savech;
 122 }
 123
 124 static char **
 125 process(const char *cap, const char *str, char **argv)
 126 {
 127         static char errfew[] =
 128             "not enough arguments (%d) for capability `%s'";
 129         static char errmany[] =
 130             "too many arguments (%d) for capability `%s'";
 131         static char erresc[] =
 132             "unknown %% escape `%c' for capability `%s'";
 133         const char *cp;
 134         int arg_need, arg_rows, arg_cols;
 135
 136         /* Count how many values we need for this capability. */
 137         for (cp = str, arg_need = 0; *cp != '\0'; cp++)
 138                 if (*cp == '%')
 139                             switch (*++cp) {
 140                             case 'd':
 141                             case '2':
 142                             case '3':
 143                             case '.':
 144                             case '+':
 145                                     arg_need++;
 146                                     break;
 147                             case '%':
 148                             case '>':
 149                             case 'i':
 150                             case 'r':
 151                             case 'n':
 152                             case 'B':
 153                             case 'D':
 154                                     break;
 155                             case 'p':
 156                                     if (cp[1]) {
 157                                         cp++;
 158                                         break;
 159                                     }
 160                             default:
 161                                 /*
 162                                  * hpux has lot's of them, but we complain
 163                                  */
 164                                  warnx(erresc, *cp, cap);
 165                             }
 166
 167         /* And print them. */
 168         switch (arg_need) {
 169         case 0:
 170                 (void)tputs(str, 1, outc);
 171                 break;
 172         case 1:
 173                 arg_cols = 0;
 174
 175                 if (*++argv == NULL || *argv[0] == '\0')
 176                         errx(2, errfew, 1, cap);
 177                 arg_rows = atoi(*argv);
 178
 179                 (void)tputs(tgoto(str, arg_cols, arg_rows), 1, outc);
 180                 break;
 181         case 2:
 182                 if (*++argv == NULL || *argv[0] == '\0')
 183                         errx(2, errfew, 2, cap);
 184                 arg_cols = atoi(*argv);
 185
 186                 if (*++argv == NULL || *argv[0] == '\0')
 187                         errx(2, errfew, 2, cap);
 188                 arg_rows = atoi(*argv);
 189
 190                 (void) tputs(tgoto(str, arg_cols, arg_rows), arg_rows, outc);
 191                 break;
 192
 193         default:
 194                 errx(2, errmany, arg_need, cap);
 195         }
 196         return (argv);
 197 }
 198
 199 static void
 200 usage(void)
 201 {
 202         (void)fprintf(stderr, "usage: tput [-T term] attribute ...\n");
 203         exit(2);
 204 }