NetBSD/usr.sbin/sti/sti.c

/*	$NetBSD: sti.c,v 1.8 2009/08/27 19:40:06 christos Exp $	*/

/*-
 * Copyright (c) 2005 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Christos Zoulas.
 *
 * 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. 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 FOUNDATION 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/cdefs.h>
#ifdef __RCSID
__RCSID("$NetBSD: sti.c,v 1.8 2009/08/27 19:40:06 christos Exp $");
#endif

#include <sys/param.h>
#include <sys/ioctl.h>
#include <ctype.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <err.h>
#include <errno.h>
#ifdef __RCSID
#include <vis.h>
#else
#define setprogname(a)
extern const char *__progname;
#define getprogname() __progname
#endif

static int
unescape(const char **pp, int *state)
{
	char ch, out;

#ifdef __RCSID
	while ((ch = *(*pp)++) != '\0') {
		switch(unvis(&out, ch, state, 0)) {
		case 0:
		case UNVIS_NOCHAR:
		        break;
		case UNVIS_VALID:
			return out;
		case UNVIS_VALIDPUSH:
		        (*pp)--;
			return out;
		case UNVIS_SYNBAD:
			errno = EILSEQ;
			return -1;
		}
	}
	if (unvis(&out, '\0', state, UNVIS_END) == UNVIS_VALID)
		return out;
#else
	switch ((ch = *(*pp)++)) {
	case '\0':
		goto out;
	case '^':
		ch = *(*pp)++;
		return CTRL(ch);
	case '\\':
		switch (ch = *(*pp)++) {
		case 'a': return '\a';
		case 'b': return '\b';
		case 'e': return '\e';
		case 'f': return '\f';
		case 't': return '\t';
		case 'n': return '\n';
		case 'r': return '\r';
		case 'v': return '\v';
		case '\\': return '\\';

		case '0': case '1': case '2': case '3':
		case '4': case '5': case '6': case '7':
			out = 0;
			if (ch >= '0' && ch < '8') {
				out = out * 8 + ch - '0';
				ch = *(*pp)++;
				if (ch >= '0' && ch < '8') {
					out = out * 8 + ch - '0';
					ch = *(*pp)++;
					if (ch >= '0' && ch < '8')
						out = out * 8 + ch - '0';
				}
			}
			return out;
		default:
		    break;
		}
		break;
	default:
		return ch;
	}
out:
#endif
	errno = ENODATA;
	return -1;
}

static void
sti(int fd, int c)
{
	char ch = c;

	if (ioctl(fd, TIOCSTI, &ch) == -1)
		err(1, "Cannot simulate terminal input");
}

static void
sendstr(int fd, const char *str)
{
	int c, state = 0;
	const char *ptr = str;

	while ((c = unescape(&ptr, &state)) != -1)
		sti(fd, c);

	if (c == -1 && errno != ENODATA)
		warn("Cannot decode `%s'", str);
}

int
main(int argc, char *argv[])
{
	const char *tty;
	char ttydev[MAXPATHLEN];
	int fd;

	setprogname(*argv);

	if (argc < 2) {
		(void)fprintf(stderr, "Usage: %s <tty> [arg ...]\n",
		    getprogname());
		return 1;
	}

	argc--;
	argv++;

	tty = *argv++;
	argc--;

	if (strncmp(tty, "/dev/", 5) == 0)
		(void)snprintf(ttydev, sizeof(ttydev), "%s", tty);
	else if (strncmp(tty, "tty", 3) == 0 || strncmp(tty, "pty", 3) == 0 ||
	    strncmp(tty, "pts/", 4) == 0)
		(void)snprintf(ttydev, sizeof(ttydev), "/dev/%s", tty);
	else if (isdigit((unsigned char)*tty))
		(void)snprintf(ttydev, sizeof(ttydev), "/dev/pts/%s", tty);
	else
		(void)snprintf(ttydev, sizeof(ttydev), "/dev/tty%s", tty);

	if ((fd = open(ttydev, O_RDWR)) == -1)
		err(1, "Cannot open `%s'", ttydev);

	if (argc == 0) {
		char *line;
#ifndef __RCSID
		line = malloc(10240);
		while (fgets(line, 10240, stdin) != NULL) {
			char *p;
			if ((p = strrchr(line, '\n')) != NULL)
				*p = '\0';
			sendstr(fd, line);
		}
		free(line);
#else
		while ((line = fparseln(stdin, NULL, NULL, NULL, 0)) != NULL) {
			sendstr(fd, line);
			free(line);
		}
#endif
	} else {
		for (; argc--; argv++) {
			sendstr(fd, *argv);
			if (argc != 0)
				sti(fd, ' ');
		}
	}

	(void)close(fd);
	return 0;
}