/* $NetBSD: iomdkbc.c,v 1.7 2020/11/20 18:18:51 thorpej Exp $ */

/*-
 * Copyright (c) 2004 Ben Harris
 * 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. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 AUTHOR 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: iomdkbc.c,v 1.7 2020/11/20 18:18:51 thorpej Exp $");

#include <sys/param.h>
#include <sys/device.h>
#include <sys/kmem.h>
#include <sys/systm.h>

#include <dev/pckbport/pckbportvar.h>

#include <sys/bus.h>
#include <machine/intr.h>

#include <arch/arm/iomd/iomdreg.h>
#include <arch/arm/iomd/iomdvar.h>
#include <arch/arm/iomd/iomdkbcvar.h>

/* XXX belongs in iomdreg.h */
#define IOMDKBC_TXE     0x80
#define IOMDKBC_TXB     0x40
#define IOMDKBC_RXF     0x20
#define IOMDKBC_RXB     0x10
#define IOMDKBC_ENA     0x08
#define IOMDKBC_RXP     0x04
#define IOMDKBC_DATA    0x02
#define IOMDKBC_CLK     0x01

#define IOMDKBC_DR      0x00
#define IOMDKBC_CR      0x01

#define KBC_DEVCMD_ACK 0xfa
#define KBC_DEVCMD_RESEND 0xfe

struct iomdkbc_softc {
        device_t                sc_dev;
        struct iomdkbc_internal *sc_id;
};

struct iomdkbc_internal {
        struct iomdkbc_softc    *t_sc;
        struct pckbport_tag     *t_pt;

        int                     t_haveport[PCKBPORT_NSLOTS];
        bus_space_tag_t         t_iot;
        bus_space_handle_t      t_ioh[PCKBPORT_NSLOTS];

        void    *t_rxih[PCKBPORT_NSLOTS];
        int     t_rxirq[PCKBPORT_NSLOTS];
};

static int iomdkbc_match(device_t , cfdata_t , void *);
static void iomdkbc_attach(device_t , device_t , void *);

static int iomdkbc_xt_translation(void *, pckbport_slot_t, int);
static int iomdkbc_send_devcmd(void *, pckbport_slot_t, u_char);
static int iomdkbc_poll_data1(void *, pckbport_slot_t);
static void iomdkbc_slot_enable(void *, pckbport_slot_t, int);
static void iomdkbc_intr_establish(void *, pckbport_slot_t);
static void iomdkbc_set_poll(void *, pckbport_slot_t, int);

static int iomdkbc_intr(void *);

CFATTACH_DECL_NEW(iomdkbc, sizeof(struct iomdkbc_softc),
    iomdkbc_match, iomdkbc_attach, NULL, NULL);

static struct pckbport_accessops const iomdkbc_ops = {
        iomdkbc_xt_translation,
        iomdkbc_send_devcmd,
        iomdkbc_poll_data1,
        iomdkbc_slot_enable,
        iomdkbc_intr_establish,
        iomdkbc_set_poll
};

static struct iomdkbc_internal iomdkbc_cntag;

static int
iomdkbc_match(device_t parent, cfdata_t cf, void *aux)
{
        struct kbd_attach_args *ka = aux;
        struct opms_attach_args *pa = aux;

        if (strcmp(ka->ka_name, "kbd") == 0 ||
            strcmp(pa->pa_name, "opms") == 0)
                return 1;
        return 0;
}

static void
iomdkbc_attach(device_t parent, device_t self, void *aux)
{
        struct kbd_attach_args *ka = aux;
        struct opms_attach_args *pa = aux;
        struct iomdkbc_softc *sc = device_private(self);
        struct iomdkbc_internal *t;

        sc->sc_dev = self;
        aprint_normal("\n");

        t = NULL;
        if (strcmp(ka->ka_name, "kbd") == 0) {
                /* XXX not really right, but good enough. */
                if (iomdkbc_cntag.t_haveport[PCKBPORT_KBD_SLOT]) {
                        /* Have an iomdkbc as console.  Assume it's this one.*/
                        t = &iomdkbc_cntag;
                } else {
                        t = kmem_zalloc(sizeof(struct iomdkbc_internal),
                            KM_SLEEP);
                        t->t_haveport[PCKBPORT_KBD_SLOT] = 1;
                        t->t_iot = ka->ka_iot;
                        t->t_ioh[PCKBPORT_KBD_SLOT] = ka->ka_ioh;
                }
                t->t_rxih[PCKBPORT_KBD_SLOT] = intr_claim(ka->ka_rxirq,
                    IPL_TTY, device_xname(sc->sc_dev), iomdkbc_intr, t);
                t->t_rxirq[PCKBPORT_KBD_SLOT] = ka->ka_rxirq;
                disable_irq(t->t_rxirq[PCKBPORT_KBD_SLOT]);
                sc->sc_id = t;
                t->t_sc = sc;
                t->t_pt = pckbport_attach(t, &iomdkbc_ops);
                pckbport_attach_slot(sc->sc_dev, t->t_pt, PCKBPORT_KBD_SLOT);
        }

        if (strcmp(pa->pa_name, "opms") == 0) {
                if (t == NULL) {
                        t = kmem_zalloc(sizeof(struct iomdkbc_internal),
                            KM_SLEEP);
                }
                t->t_haveport[PCKBPORT_AUX_SLOT] = 1;
                t->t_iot = pa->pa_iot;
                t->t_ioh[PCKBPORT_AUX_SLOT] = pa->pa_ioh;
                t->t_rxih[PCKBPORT_AUX_SLOT] = intr_claim(pa->pa_irq,
                    IPL_TTY, device_xname(sc->sc_dev), iomdkbc_intr, t);
                t->t_rxirq[PCKBPORT_AUX_SLOT] = pa->pa_irq;
                disable_irq(t->t_rxirq[PCKBPORT_AUX_SLOT]);
                sc->sc_id = t;
                t->t_sc = sc;
                if (t->t_pt == NULL)
                        t->t_pt = pckbport_attach(t, &iomdkbc_ops);
                pckbport_attach_slot(sc->sc_dev, t->t_pt, PCKBPORT_AUX_SLOT);
        }
}

static int
iomdkbc_send_devcmd(void *cookie, pckbport_slot_t slot, u_char cmd)
{
        struct iomdkbc_internal *t = cookie;
        bus_space_tag_t iot = t->t_iot;
        bus_space_handle_t ioh = t->t_ioh[slot];
        int timeout;

        timeout = 10000;
        while ((bus_space_read_1(iot, ioh, IOMDKBC_CR) &
                   IOMDKBC_TXE) == 0) {
                DELAY(10);
                if (--timeout == 0) return 0;
        }

        bus_space_write_1(iot, ioh, IOMDKBC_DR, cmd);
        return 1;
}

static int
iomdkbc_poll_data1(void *cookie, pckbport_slot_t slot)
{
        struct iomdkbc_internal *t = cookie;
        bus_space_tag_t iot = t->t_iot;
        bus_space_handle_t ioh = t->t_ioh[slot];
        int timeout;

        timeout = 10000;
        while ((bus_space_read_1(iot, ioh, IOMDKBC_CR) &
                   IOMDKBC_RXF) == 0) {
                DELAY(10);
                if (--timeout == 0) return -1;
        }

        return bus_space_read_1(iot, ioh, IOMDKBC_DR);
}

/*
 * switch scancode translation on / off
 * return nonzero on success
 */
static int
iomdkbc_xt_translation(void *cookie, pckbport_slot_t slot, int on)
{

        if (on)
                return 0; /* Can't do XT translation */
        else
                return 1;
}

static void
iomdkbc_slot_enable(void *cookie, pckbport_slot_t slot, int on)
{
        struct iomdkbc_internal *t = cookie;
        bus_space_tag_t iot = t->t_iot;
        bus_space_handle_t ioh = t->t_ioh[slot];

        bus_space_write_1(iot, ioh, IOMDKBC_CR, on ? IOMDKBC_ENA : 0);
}


static void
iomdkbc_intr_establish(void *cookie, pckbport_slot_t slot)
{
        struct iomdkbc_internal *t = cookie;

        enable_irq(t->t_rxirq[slot]);
}

static void
iomdkbc_set_poll(void *cookie, pckbport_slot_t slot, int on)
{
        struct iomdkbc_internal *t = cookie;

        if (on)
                disable_irq(t->t_rxirq[slot]);
        else
                enable_irq(t->t_rxirq[slot]);
}

static int
iomdkbc_intr(void *self)
{
        struct iomdkbc_internal *t = self;
        bus_space_tag_t iot = t->t_iot;
        bus_space_handle_t ioh;
        unsigned i;
        int stat, ret;

        ret = 0;
        for (i = 0; i < PCKBPORT_NSLOTS; i++)
                if (t->t_haveport[i]) {
                        ioh = t->t_ioh[i];
                        stat = bus_space_read_1(iot, ioh, IOMDKBC_CR);
                        if ((stat & IOMDKBC_RXF) == 0)
                                continue;
                        pckbportintr(t->t_pt, i,
                            bus_space_read_1(iot, ioh, IOMDKBC_DR));
                        ret = 1;
                }

        return ret;
}

int
iomdkbc_cnattach(bus_space_tag_t iot, bus_addr_t addr, int slot)
{
        struct iomdkbc_internal *t = &iomdkbc_cntag;

        t->t_iot = iot;
        bus_space_map(iot, addr, 2, 0, &t->t_ioh[slot]);
        t->t_haveport[slot] = 1;
        iomdkbc_slot_enable(t, slot, 1);
        return pckbport_cnattach(t, &iomdkbc_ops, slot);
}