xref: /netbsd-src/usr.bin/tput/tput.c (revision 404fbe5fb94ca1e054339640cabb2801ce52dd30)
1 /*	$NetBSD: tput.c,v 1.19 2008/07/21 14:19:27 lukem 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.19 2008/07/21 14:19:27 lukem 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 }
234