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

/*-
* Copyright (c) 2007 Jared D. McNeill <[email protected]>
* 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);
}