File: [local] / src / usr.bin / tput / tput.c (download)
Revision 1.28, Wed Feb 8 15:56:32 2023 UTC (15 months, 3 weeks ago) by millert
Branch: MAIN
CVS Tags: OPENBSD_7_3_BASE, OPENBSD_7_3 Changes since 1.27: +20 -50 lines
Use _nc_tparm_analyze() instead of trying to do it ourselves in process().
This fixes tput when used with a capability that uses push/pop.
OK gkoehler@
|
/* $OpenBSD: tput.c,v 1.28 2023/02/08 15:56:32 millert Exp $ */
/*
* Copyright (c) 1999 Todd C. Miller <millert@openbsd.org>
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
/*-
* Copyright (c) 1980, 1988, 1993
* The Regents of the University of California. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include <sys/wait.h>
#include <ctype.h>
#include <err.h>
#include <curses.h>
#include <term.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <limits.h>
#include <string.h>
#define MAXIMUM(a, b) (((a) > (b)) ? (a) : (b))
#define NUM_PARM 9 /* must match tic.h */
static void init(void);
static char **process(char *, char *, char **);
static void reset(void);
static void set_margins(void);
static void usage(void);
extern char *__progname;
extern int _nc_tparm_analyze(const char *string, char *p_is_s[NUM_PARM], int *popcount);
int
main(int argc, char *argv[])
{
int ch, exitval, n, Sflag;
size_t len;
char *p, *term, *str;
char **oargv;
if (pledge("stdio rpath wpath tty", NULL) == -1)
err(1, "pledge");
oargv = argv;
term = NULL;
Sflag = exitval = 0;
while ((ch = getopt(argc, argv, "ST:")) != -1)
switch(ch) {
case 'T':
term = optarg;
break;
case 'S':
Sflag = 1;
break;
default:
usage();
}
argc -= optind;
argv += optind;
if (Sflag && argc > 0)
usage();
if (!term && !(term = getenv("TERM")))
errx(2, "No value for $TERM and no -T specified");
/*
* NOTE: tgetent() will call setupterm() and set ospeed for us
* (this is ncurses-specific behavior)
*/
if (tgetent(NULL, term) != 1)
errx(3, "Unknown terminal type `%s'", term);
if (strcmp(__progname, "clear") == 0) {
if (Sflag)
usage();
argv = oargv;
*argv = __progname;
*(argv+1) = NULL;
}
if (Sflag) {
char **av;
/* Build new argv based on stdin */
argc = n = 0;
av = NULL;
while ((str = fgetln(stdin, &len)) != NULL) {
if (str[len-1] != '\n')
errx(1, "premature EOF");
str[len-1] = '\0';
while ((p = strsep(&str, " \t")) != NULL) {
/* grow av as needed */
if (argc + 1 >= n) {
n += 64;
av = reallocarray(av, n,
sizeof(char *));
if (av == NULL)
errx(1, "out of memory");
}
if (*p != '\0' &&
(av[argc++] = strdup(p)) == NULL)
errx(1, "out of memory");
}
}
if (argc > 0) {
av[argc] = NULL;
argv = av;
}
}
while ((p = *argv++)) {
switch (*p) {
case 'i':
if (!strcmp(p, "init")) {
init();
continue;
}
break;
case 'l':
if (!strcmp(p, "longname")) {
puts(longname());
continue;
}
break;
case 'r':
if (!strcmp(p, "reset")) {
reset();
continue;
}
break;
}
/* First try as terminfo */
if ((str = tigetstr(p)) && str != (char *)-1)
argv = process(p, str, argv);
else if ((n = tigetnum(p)) != -2)
(void)printf("%d\n", n);
else if ((n = tigetflag(p)) != -1)
exitval = !n;
/* Then fall back on termcap */
else if ((str = tgetstr(p, NULL)))
argv = process(p, str, argv);
else if ((n = tgetnum(p)) != -1)
(void)printf("%d\n", n);
else if ((exitval = tgetflag(p)) != 0)
exitval = !exitval;
else {
warnx("Unknown terminfo capability `%s'", p);
exitval = 4;
}
}
exit(exitval);
}
static char **
process(char *cap, char *str, char **argv)
{
char *cp, *s, *nargv[NUM_PARM] = {0};
char *p_is_s[NUM_PARM];
int arg_need, i;
/* Count how many values we need for this capability. */
i = _nc_tparm_analyze(str, p_is_s, &arg_need);
if (arg_need == 0)
arg_need = i;
if (arg_need > NUM_PARM)
errx(2, "too many arguments (%d) for capability `%s'",
arg_need, cap);
for (i = 0; i < arg_need; i++) {
const char *errstr;
long l;
if (argv[i] == NULL)
errx(2, "not enough arguments (%d) for capability `%s'",
arg_need, cap);
if (p_is_s[i] != 0) {
nargv[i] = argv[i];
} else {
/* convert ascii representation of numbers to longs */
l = strtonum(argv[i], LONG_MIN, LONG_MAX, &errstr);
if (errstr != NULL)
errx(2, "capability `%s' is %s", cap, errstr);
nargv[i] = (char *)l;
}
}
s = tparm(str, nargv[0], nargv[1], nargv[2], nargv[3],
nargv[4], nargv[5], nargv[6], nargv[7], nargv[8]);
putp(s);
fflush(stdout);
return (argv + arg_need);
}
static void
init(void)
{
FILE *ifile;
size_t len;
char *buf;
int wstatus;
pid_t pid;
if (init_prog && !issetugid()) {
switch (pid = vfork()) {
case -1:
err(4, "vfork");
break;
case 0:
/* child */
execl(init_prog, init_prog, (char *)NULL);
_exit(127);
break;
default:
while (waitpid(pid, &wstatus, 0) == -1) {
if (errno != EINTR)
break;
}
/* parent */
break;
}
}
if (init_1string)
putp(init_1string);
if (init_2string)
putp(init_2string);
set_margins();
/* always use 8 space tabs */
if (init_tabs != 8 && clear_all_tabs && set_tab) {
int i;
putp(clear_all_tabs);
for (i = 0; i < (columns - 1) / 8; i++) {
if (parm_right_cursor)
putp(tparm(parm_right_cursor, 8));
else
fputs(" ", stdout);
putp(set_tab);
}
}
if (init_file && !issetugid() && (ifile = fopen(init_file, "r"))) {
while ((buf = fgetln(ifile, &len)) != NULL) {
if (buf[len-1] != '\n')
errx(1, "premature EOF reading %s", init_file);
buf[len-1] = '\0';
putp(buf);
}
fclose(ifile);
}
if (init_3string)
putp(init_3string);
fflush(stdout);
}
static void
reset(void)
{
FILE *rfile;
size_t len;
char *buf;
if (reset_1string)
putp(reset_1string);
if (reset_2string)
putp(reset_2string);
set_margins();
if (reset_file && !issetugid() && (rfile = fopen(reset_file, "r"))) {
while ((buf = fgetln(rfile, &len)) != NULL) {
if (buf[len-1] != '\n')
errx(1, "premature EOF reading %s", reset_file);
buf[len-1] = '\0';
putp(buf);
}
fclose(rfile);
}
if (reset_3string)
putp(reset_3string);
fflush(stdout);
}
static void
set_margins(void)
{
/*
* Four possibilities:
* 1) we have set_lr_margin and can set things with one call
* 2) we have set_{left,right}_margin_parm, use two calls
* 3) we have set_{left,right}_margin, set based on position
* 4) none of the above, leave things the way they are
*/
if (set_lr_margin) {
putp(tparm(set_lr_margin, 0, columns - 1));
} else if (set_left_margin_parm && set_right_margin_parm) {
putp(tparm(set_left_margin_parm, 0));
putp(tparm(set_right_margin_parm, columns - 1));
} else if (set_left_margin && set_right_margin && clear_margins) {
putp(clear_margins);
/* go to column 0 and set the left margin */
putp(carriage_return ? carriage_return : "\r");
putp(set_left_margin);
/* go to last column and set the right margin */
if (parm_right_cursor)
putp(tparm(parm_right_cursor, columns - 1));
else
printf("%*s", columns - 1, " ");
putp(set_right_margin);
putp(carriage_return ? carriage_return : "\r");
}
fflush(stdout);
}
static void
usage(void)
{
if (strcmp(__progname, "clear") == 0)
(void)fprintf(stderr, "usage: %s [-T term]\n", __progname);
else
(void)fprintf(stderr,
"usage: %s [-T term] attribute [attribute-arg ...] ...\n"
" %s [-T term] -S\n", __progname, __progname);
exit(1);
}