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  * 4. Neither the name of the University nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS 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 REGENTS 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 
32 __FBSDID("$FreeBSD: stable/9/usr.bin/tput/tput.c 216370 2010-12-11 08:32:16Z joel $");
33 
34 #ifndef lint
35 static const char copyright[] =
36 "@(#) Copyright (c) 1980, 1988, 1993\n\
37 	The Regents of the University of California.  All rights reserved.\n";
38 #endif
39 
40 #ifndef lint
41 static const char sccsid[] = "@(#)tput.c	8.2 (Berkeley) 3/19/94";
42 #endif
43 
44 #include <termios.h>
45 
46 #include <err.h>
47 #include <termcap.h>
48 #include <stdio.h>
49 #include <stdlib.h>
50 #include <string.h>
51 #include <unistd.h>
52 
53 #undef putchar
54 #define outc putchar
55 
56 static void   prlongname(char *);
57 static void   usage(void);
58 static char **process(const char *, char *, char **);
59 
60 int
main(int argc,char ** argv)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 (argc < 1)
81 		usage();
82 
83 	if (!term && !(term = getenv("TERM")))
84 errx(2, "no terminal type specified and no TERM environmental variable.");
85 	if (tgetent(tbuf, term) != 1)
86 		err(3, "tgetent failure");
87 	for (exitval = 0; (p = *argv) != NULL; ++argv) {
88 		switch (*p) {
89 		case 'c':
90 			if (!strcmp(p, "clear"))
91 				p = "cl";
92 			break;
93 		case 'i':
94 			if (!strcmp(p, "init"))
95 				p = "is";
96 			break;
97 		case 'l':
98 			if (!strcmp(p, "longname")) {
99 				prlongname(tbuf);
100 				continue;
101 			}
102 			break;
103 		case 'r':
104 			if (!strcmp(p, "reset"))
105 				p = "rs";
106 			break;
107 		}
108 		cptr = buf;
109 		if (tgetstr(p, &cptr))
110 			argv = process(p, buf, argv);
111 		else if ((n = tgetnum(p)) != -1)
112 			(void)printf("%d\n", n);
113 		else
114 			exitval = !tgetflag(p);
115 	}
116 	exit(exitval);
117 }
118 
119 static void
prlongname(char * buf)120 prlongname(char *buf)
121 {
122 	int savech;
123 	char *p, *savep;
124 
125 	for (p = buf; *p && *p != ':'; ++p);
126 	savech = *(savep = p);
127 	for (*p = '\0'; p >= buf && *p != '|'; --p);
128 	(void)printf("%s\n", p + 1);
129 	*savep = savech;
130 }
131 
132 static char **
process(const char * cap,char * str,char ** argv)133 process(const char *cap, char *str, char **argv)
134 {
135 	static const char errfew[] =
136 	    "not enough arguments (%d) for capability `%s'";
137 	static const char errmany[] =
138 	    "too many arguments (%d) for capability `%s'";
139 	static const char erresc[] =
140 	    "unknown %% escape `%c' for capability `%s'";
141 	char *cp;
142 	int arg_need, arg_rows, arg_cols;
143 
144 	/* Count how many values we need for this capability. */
145 	for (cp = str, arg_need = 0; *cp != '\0'; cp++)
146 		if (*cp == '%')
147 			    switch (*++cp) {
148 			    case 'd':
149 			    case '2':
150 			    case '3':
151 			    case '.':
152 			    case '+':
153 				    arg_need++;
154 				    break;
155 			    case '%':
156 			    case '>':
157 			    case 'i':
158 			    case 'r':
159 			    case 'n':
160 			    case 'B':
161 			    case 'D':
162 				    break;
163 			    case 'p':
164 				    if (cp[1]) {
165 					cp++;
166 					break;
167 				    }
168 			    default:
169 				/*
170 				 * hpux has lot's of them, but we complain
171 				 */
172 				 warnx(erresc, *cp, cap);
173 			    }
174 
175 	/* And print them. */
176 	switch (arg_need) {
177 	case 0:
178 		(void)tputs(str, 1, outc);
179 		break;
180 	case 1:
181 		arg_cols = 0;
182 
183 		if (*++argv == NULL || *argv[0] == '\0')
184 			errx(2, errfew, 1, cap);
185 		arg_rows = atoi(*argv);
186 
187 		(void)tputs(tgoto(str, arg_cols, arg_rows), 1, outc);
188 		break;
189 	case 2:
190 		if (*++argv == NULL || *argv[0] == '\0')
191 			errx(2, errfew, 2, cap);
192 		arg_cols = atoi(*argv);
193 
194 		if (*++argv == NULL || *argv[0] == '\0')
195 			errx(2, errfew, 2, cap);
196 		arg_rows = atoi(*argv);
197 
198 		(void) tputs(tgoto(str, arg_cols, arg_rows), arg_rows, outc);
199 		break;
200 
201 	default:
202 		errx(2, errmany, arg_need, cap);
203 	}
204 	return (argv);
205 }
206 
207 static void
usage(void)208 usage(void)
209 {
210 	(void)fprintf(stderr, "usage: tput [-T term] attribute ...\n");
211 	exit(2);
212 }
213