/* $NetBSD: ttycons.c,v 1.20 2014/07/25 08:10:35 dholland Exp $ */

/*-
 * Copyright (c) 2007 Jared D. McNeill <jmcneill@invisible.ca>
 * 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.
 *
 * 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>
__KERNEL_RCSID(0, "$NetBSD: ttycons.c,v 1.20 2014/07/25 08:10:35 dholland Exp $");

#include <sys/param.h>
#include <sys/conf.h>
#include <sys/proc.h>
#include <sys/systm.h>
#include <sys/device.h>
#include <sys/kauth.h>
#include <sys/termios.h>
#include <sys/tty.h>

#include <dev/cons.h>

#include <machine/mainbus.h>
#include <machine/thunk.h>

static int      ttycons_match(device_t, cfdata_t, void *);
static void     ttycons_attach(device_t, device_t, void *);

void            ttycons_consinit(void);

struct ttycons_softc {
        device_t        sc_dev;
        struct tty      *sc_tty;
        void            *sc_rd_sih;
        void            *sc_ctrlc_sih;
        void            *sc_ctrlz_sih;
        u_char          sc_buf[1024];
};

dev_type_cngetc(ttycons_cngetc);
dev_type_cnputc(ttycons_cnputc);
dev_type_cnpollc(ttycons_cnpollc);

static struct cnm_state ttycons_cnm_state;
struct consdev ttycons_consdev = {
        .cn_getc = ttycons_cngetc,
        .cn_putc = ttycons_cnputc,
        .cn_pollc = ttycons_cnpollc,
        .cn_dev = NODEV,
        .cn_pri = CN_NORMAL,
};

CFATTACH_DECL_NEW(ttycons, sizeof(struct ttycons_softc),
    ttycons_match, ttycons_attach, NULL, NULL);

extern struct cfdriver ttycons_cd;

dev_type_open(ttycons_open);
dev_type_close(ttycons_close);
dev_type_read(ttycons_read);
dev_type_write(ttycons_write);
dev_type_ioctl(ttycons_ioctl);
dev_type_stop(ttycons_stop);
dev_type_tty(ttycons_tty);
dev_type_poll(ttycons_poll);

const struct cdevsw ttycons_cdevsw = {
        .d_open = ttycons_open,
        .d_close = ttycons_close,
        .d_read = ttycons_read,
        .d_write = ttycons_write,
        .d_ioctl = ttycons_ioctl,
        .d_stop = ttycons_stop,
        .d_tty = ttycons_tty,
        .d_poll = ttycons_poll,
        .d_kqfilter = ttykqfilter,
        .d_flag = D_TTY,
        .d_discard = nodiscard,
};

static void     ttycons_start(struct tty *);
static int      ttycons_param(struct tty *, struct termios *);

static int      ttycons_intr(void *);
static void     ttycons_softintr(void *);

static sigfunc_t  ttycons_ctrlc;
static void     ttycons_softctrlc(void *);
static sigfunc_t  ttycons_ctrlz;
static void     ttycons_softctrlz(void *);

static int
ttycons_match(device_t parent, cfdata_t match, void *opaque)
{
        struct thunkbus_attach_args *taa = opaque;

        if (taa->taa_type != THUNKBUS_TYPE_TTYCONS)
                return 0;

        return 1;
}

static void
ttycons_attach(device_t parent, device_t self, void *opaque)
{
        struct ttycons_softc *sc = device_private(self);
        int maj;

        aprint_naive("\n");
        aprint_normal(": console\n");

        sc->sc_dev = self;
        sc->sc_tty = tty_alloc();
        tty_attach(sc->sc_tty);
        sc->sc_tty->t_oproc = ttycons_start;
        sc->sc_tty->t_param = ttycons_param;
        sc->sc_tty->t_sc = sc;

        maj = cdevsw_lookup_major(&ttycons_cdevsw);
        cn_tab->cn_dev = makedev(maj, device_unit(self));
        sc->sc_tty->t_dev = cn_tab->cn_dev;

        sc->sc_rd_sih = softint_establish(SOFTINT_SERIAL,
            ttycons_softintr, sc);
        if (sc->sc_rd_sih == NULL)
                panic("couldn't establish ttycons intr handler\n");

        sc->sc_ctrlc_sih = softint_establish(SOFTINT_SERIAL,
            ttycons_softctrlc, sc);
        if (sc->sc_ctrlc_sih == NULL)
                panic("couldn't establish ttycons ctrlc handler\n");
        sc->sc_ctrlz_sih = softint_establish(SOFTINT_SERIAL,
            ttycons_softctrlz, sc);
        if (sc->sc_ctrlz_sih == NULL)
                panic("couldn't establish ttycons ctrlz handler\n");

        sigio_intr_establish(ttycons_intr, sc);
        signal_intr_establish(SIGINT,  ttycons_ctrlc);
        signal_intr_establish(SIGTSTP, ttycons_ctrlz);

        if (thunk_set_stdin_sigio(true) != 0)
                panic("couldn't enable stdin async mode");
}

void
ttycons_consinit(void)
{
        struct thunk_termios t;

        thunk_tcgetattr(0, &t);
        t.c_lflag &= ~(ECHO|ICANON);
        t.c_cc[VTIME] = 0;
        t.c_cc[VMIN] = 1;
        thunk_tcsetattr(0, TCSANOW, &t);

        cn_tab = &ttycons_consdev;
        cn_init_magic(&ttycons_cnm_state);
        cn_set_magic("\047\001");
}

int
ttycons_cngetc(dev_t dev)
{
        return thunk_getchar();
}

void
ttycons_cnputc(dev_t dev, int c)
{
        thunk_putchar(c);
}

void
ttycons_cnpollc(dev_t dev, int on)
{
}

int
ttycons_open(dev_t dev, int flag, int mode, lwp_t *l)
{
        struct ttycons_softc *sc;
        struct tty *t;

        sc = device_lookup_private(&ttycons_cd, minor(dev));
        if (sc == NULL)
                return ENXIO;
        t = sc->sc_tty;

        if (kauth_authorize_device_tty(l->l_cred, KAUTH_DEVICE_TTY_OPEN, t))
                return EBUSY;

        if ((t->t_state & TS_ISOPEN) == 0 && t->t_wopen == 0) {
                t->t_dev = dev;
                ttychars(t);
                t->t_iflag = TTYDEF_IFLAG;
                t->t_oflag = TTYDEF_OFLAG;
                t->t_cflag = TTYDEF_CFLAG;
                t->t_lflag = TTYDEF_LFLAG;
                t->t_ispeed = t->t_ospeed = TTYDEF_SPEED;
                ttycons_param(t, &t->t_termios);
                ttsetwater(t);
        }
        t->t_state |= TS_CARR_ON;

        return t->t_linesw->l_open(dev, t);
}

int
ttycons_close(dev_t dev, int flag, int mode, lwp_t *l)
{
        struct ttycons_softc *sc;
        struct tty *t;

        sc = device_lookup_private(&ttycons_cd, minor(dev));
        t = sc->sc_tty;

        if (t == NULL)
                return 0;

        t->t_linesw->l_close(t, flag);
        ttyclose(t);

        return 0;
}

int
ttycons_read(dev_t dev, struct uio *uio, int flag)
{
        struct ttycons_softc *sc;
        struct tty *t;

        sc = device_lookup_private(&ttycons_cd, minor(dev));
        t = sc->sc_tty;

        return t->t_linesw->l_read(t, uio, flag);
}

int
ttycons_write(dev_t dev, struct uio *uio, int flag)
{
        struct ttycons_softc *sc;
        struct tty *t;

        sc = device_lookup_private(&ttycons_cd, minor(dev));
        t = sc->sc_tty;

        return t->t_linesw->l_write(t, uio, flag);
}

int
ttycons_poll(dev_t dev, int events, lwp_t *l)
{
        struct ttycons_softc *sc;
        struct tty *t;

        sc = device_lookup_private(&ttycons_cd, minor(dev));
        t = sc->sc_tty;

        return t->t_linesw->l_poll(t, events, l);
}

struct tty *
ttycons_tty(dev_t dev)
{
        struct ttycons_softc *sc;

        sc = device_lookup_private(&ttycons_cd, minor(dev));

        return sc->sc_tty;
}

int
ttycons_ioctl(dev_t dev, u_long cmd, void *data, int flag, lwp_t *l)
{
        struct ttycons_softc *sc;
        struct tty *t;
        int error;

        sc = device_lookup_private(&ttycons_cd, minor(dev));
        t = sc->sc_tty;

        error = t->t_linesw->l_ioctl(t, cmd, data, flag, l);
        if (error != EPASSTHROUGH)
                return error;

        error = ttioctl(t, cmd, data, flag, l);
        if (error != EPASSTHROUGH)
                return error;

        return EPASSTHROUGH;
}

static void
ttycons_start(struct tty *t)
{
        struct ttycons_softc *sc = t->t_sc;
        u_char *p = sc->sc_buf;
        int s, len, brem;

        s = spltty();
        if (t->t_state & (TS_TIMEOUT|TS_BUSY|TS_TTSTOP)) {
                splx(s);
                return;
        }
        t->t_state |= TS_BUSY;
        splx(s);

        brem = q_to_b(&t->t_outq, sc->sc_buf, sizeof(sc->sc_buf));

        while (brem > 0) {
                len = thunk_write(1, p, brem);
                if (len > 0) {
                        p += len;
                        brem -= len;
                }
        }

        s = spltty();
        t->t_state &= ~TS_BUSY;
        if (ttypull(t)) {
                t->t_state |= TS_TIMEOUT;
                callout_schedule(&t->t_rstrt_ch, 1);
        }
        splx(s);
}

void
ttycons_stop(struct tty *t, int flag)
{
}

static int
ttycons_param(struct tty *t, struct termios *c)
{
        t->t_ispeed = c->c_ispeed;
        t->t_ospeed = c->c_ospeed;
        t->t_cflag = c->c_cflag;
        return 0;
}

static int
ttycons_intr(void *priv)
{
        struct ttycons_softc *sc = priv;

        softint_schedule(sc->sc_rd_sih);

        return 0;
}

static void
ttycons_softintr(void *priv)
{
        struct ttycons_softc *sc = priv;
        struct tty *t = sc->sc_tty;
        unsigned char ch;
        int c;

        while ((c = thunk_pollchar()) >= 0) {
                ch = (unsigned char)c;
                cn_check_magic(t->t_dev, ch, ttycons_cnm_state);
                t->t_linesw->l_rint(ch, t);
        }
}


/*
 * handle SIGINT signal from trap.c
 *
 * argument 'pc' and 'va' are not used.
 */
static void
ttycons_ctrlc(siginfo_t *info, vaddr_t from_userland, vaddr_t pc, vaddr_t va)
{
        struct ttycons_softc *sc;

        sc = device_lookup_private(&ttycons_cd, minor(cn_tab->cn_dev));
        if (sc)
                softint_schedule(sc->sc_ctrlc_sih);

}

static void
ttycons_softctrlc(void *priv)
{
        struct ttycons_softc *sc = priv;
        struct tty *t = sc->sc_tty;
        unsigned char ch = 3;   /* ETX */

        cn_check_magic(t->t_dev, ch, ttycons_cnm_state);
        t->t_linesw->l_rint(ch, t);
}

/*
 * handle SIGTSTP signal from trap.c
 *
 * argument 'pc' and 'va' are not used.
 */
static void
ttycons_ctrlz(siginfo_t *info, vaddr_t from_userland, vaddr_t pc, vaddr_t va)
{
        struct ttycons_softc *sc;

        sc = device_lookup_private(&ttycons_cd, minor(cn_tab->cn_dev));
        if (sc)
                softint_schedule(sc->sc_ctrlz_sih);
}

static void
ttycons_softctrlz(void *priv)
{
        struct ttycons_softc *sc = priv;
        struct tty *t = sc->sc_tty;
        unsigned char ch = 26;  /* SUB */

        cn_check_magic(t->t_dev, ch, ttycons_cnm_state);
        t->t_linesw->l_rint(ch, t);
}