/*      $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;
}