/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 1998, 2001 Nicolas Souchu
 * 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 AUTHOR 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 AUTHOR 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>
/*
 * Generic I2C bit-banging code
 *
 * Example:
 *
 *      iicbus
 *       /  \ 
 *    iicbb pcf
 *     |  \
 *   bti2c lpbb
 *
 * From Linux I2C generic interface
 * (c) 1998 Gerd Knorr <kraxel@cs.tu-berlin.de>
 *
 */

#include "opt_platform.h"

#include <sys/param.h>
#include <sys/kernel.h>
#include <sys/systm.h>
#include <sys/module.h>
#include <sys/bus.h>
#include <sys/sysctl.h>
#include <sys/uio.h>
#include <machine/cpu.h>

#ifdef FDT
#include <dev/ofw/ofw_bus.h>
#include <dev/ofw/ofw_bus_subr.h>
#include <dev/fdt/fdt_common.h>
#endif

#include <dev/iicbus/iiconf.h>
#include <dev/iicbus/iicbus.h>

#include <dev/smbus/smbconf.h>

#include "iicbus_if.h"
#include "iicbb_if.h"

/* Based on the SMBus specification. */
#define DEFAULT_SCL_LOW_TIMEOUT (25 * 1000)

struct iicbb_softc {
        device_t iicbus;
        u_int udelay;           /* signal toggle delay in usec */
        u_int io_latency;       /* approximate pin toggling latency */
        u_int scl_low_timeout;
};

static int iicbb_attach(device_t);
static void iicbb_child_detached(device_t, device_t);
static int iicbb_print_child(device_t, device_t);
static int iicbb_probe(device_t);

static int iicbb_callback(device_t, int, caddr_t);
static int iicbb_start(device_t, u_char, int);
static int iicbb_repstart(device_t, u_char, int);
static int iicbb_stop(device_t);
static int iicbb_write(device_t, const char *, int, int *, int);
static int iicbb_read(device_t, char *, int, int *, int, int);
static int iicbb_reset(device_t, u_char, u_char, u_char *);
static int iicbb_transfer(device_t dev, struct iic_msg *msgs, uint32_t nmsgs);
static void iicbb_set_speed(struct iicbb_softc *sc, u_char);
#ifdef FDT
static phandle_t iicbb_get_node(device_t, device_t);
#endif

static device_method_t iicbb_methods[] = {
        /* device interface */
        DEVMETHOD(device_probe,         iicbb_probe),
        DEVMETHOD(device_attach,        iicbb_attach),
        DEVMETHOD(device_detach,        bus_generic_detach),

        /* bus interface */
        DEVMETHOD(bus_child_detached,   iicbb_child_detached),
        DEVMETHOD(bus_print_child,      iicbb_print_child),

        /* iicbus interface */
        DEVMETHOD(iicbus_callback,      iicbb_callback),
        DEVMETHOD(iicbus_start,         iicbb_start),
        DEVMETHOD(iicbus_repeated_start, iicbb_repstart),
        DEVMETHOD(iicbus_stop,          iicbb_stop),
        DEVMETHOD(iicbus_write,         iicbb_write),
        DEVMETHOD(iicbus_read,          iicbb_read),
        DEVMETHOD(iicbus_reset,         iicbb_reset),
        DEVMETHOD(iicbus_transfer,      iicbb_transfer),

#ifdef FDT
        /* ofw_bus interface */
        DEVMETHOD(ofw_bus_get_node,     iicbb_get_node),
#endif

        DEVMETHOD_END
};

driver_t iicbb_driver = {
        "iicbb",
        iicbb_methods,
        sizeof(struct iicbb_softc),
};

static int
iicbb_probe(device_t dev)
{
        device_set_desc(dev, "I2C bit-banging driver");

        return (0);
}

static int
iicbb_attach(device_t dev)
{
        struct iicbb_softc *sc = (struct iicbb_softc *)device_get_softc(dev);

        sc->iicbus = device_add_child(dev, "iicbus", DEVICE_UNIT_ANY);
        if (!sc->iicbus)
                return (ENXIO);

        sc->scl_low_timeout = DEFAULT_SCL_LOW_TIMEOUT;

        SYSCTL_ADD_UINT(device_get_sysctl_ctx(dev),
            SYSCTL_CHILDREN(device_get_sysctl_tree(dev)), OID_AUTO,
            "delay", CTLFLAG_RD, &sc->udelay,
            0, "Signal change delay controlled by bus frequency, microseconds");

        SYSCTL_ADD_UINT(device_get_sysctl_ctx(dev),
            SYSCTL_CHILDREN(device_get_sysctl_tree(dev)), OID_AUTO,
            "scl_low_timeout", CTLFLAG_RWTUN, &sc->scl_low_timeout,
            0, "SCL low timeout, microseconds");
        SYSCTL_ADD_UINT(device_get_sysctl_ctx(dev),
            SYSCTL_CHILDREN(device_get_sysctl_tree(dev)), OID_AUTO,
            "io_latency", CTLFLAG_RWTUN, &sc->io_latency,
            0, "Estimate of pin toggling latency, microseconds");

        bus_attach_children(dev);
        return (0);
}

#ifdef FDT
static phandle_t
iicbb_get_node(device_t bus, device_t dev)
{

        /* We only have one child, the I2C bus, which needs our own node. */
        return (ofw_bus_get_node(bus));
}
#endif

static void
iicbb_child_detached( device_t dev, device_t child )
{
        struct iicbb_softc *sc = (struct iicbb_softc *)device_get_softc(dev);

        if (child == sc->iicbus)
                sc->iicbus = NULL;
}

static int
iicbb_print_child(device_t bus, device_t dev)
{
        int error;
        int retval = 0;
        u_char oldaddr;

        retval += bus_print_child_header(bus, dev);
        /* retrieve the interface I2C address */
        error = IICBB_RESET(device_get_parent(bus), IIC_FASTEST, 0, &oldaddr);
        if (error == IIC_ENOADDR) {
                retval += printf(" on %s master-only\n",
                                 device_get_nameunit(bus));
        } else {
                /* restore the address */
                IICBB_RESET(device_get_parent(bus), IIC_FASTEST, oldaddr, NULL);

                retval += printf(" on %s addr 0x%x\n",
                                 device_get_nameunit(bus), oldaddr & 0xff);
        }

        return (retval);
}

#define IICBB_DEBUG
#ifdef IICBB_DEBUG
static int i2c_debug = 0;

SYSCTL_DECL(_hw_i2c);
SYSCTL_INT(_hw_i2c, OID_AUTO, iicbb_debug, CTLFLAG_RWTUN,
    &i2c_debug, 0, "Enable i2c bit-banging driver debug");

#define I2C_DEBUG(x)    do {            \
                if (i2c_debug) (x);     \
        } while (0)
#else
#define I2C_DEBUG(x)
#endif

#define I2C_GETSDA(dev)         (IICBB_GETSDA(device_get_parent(dev)))
#define I2C_SETSDA(dev, x)      (IICBB_SETSDA(device_get_parent(dev), x))
#define I2C_GETSCL(dev)         (IICBB_GETSCL(device_get_parent(dev)))
#define I2C_SETSCL(dev, x)      (IICBB_SETSCL(device_get_parent(dev), x))

static int
iicbb_waitforscl(device_t dev)
{
        struct iicbb_softc *sc = device_get_softc(dev);
        sbintime_t fast_timeout;
        sbintime_t now, timeout;

        /* Spin for up to 1 ms, then switch to pause. */
        now = sbinuptime();
        fast_timeout = now + SBT_1MS;
        timeout = now + sc->scl_low_timeout * SBT_1US;
        do {
                if (I2C_GETSCL(dev))
                        return (0);
                cpu_spinwait();
                now = sbinuptime();
        } while (now < fast_timeout);
        do {
                I2C_DEBUG(printf("."));
                pause_sbt("iicbb-scl-low", SBT_1MS, 0, C_PREL(2));
                if (I2C_GETSCL(dev))
                        return (0);
                now = sbinuptime();
        } while (now < timeout);

        I2C_DEBUG(printf("*"));
        return (IIC_ETIMEOUT);
}

/* Start the high phase of the clock. */
static int
iicbb_clockin(device_t dev, int sda)
{

        /*
         * Precondition: SCL is low.
         * Action:
         * - set SDA to the value;
         * - release SCL and wait until it's high.
         * The caller is responsible for keeping SCL high for udelay.
         *
         * There should be a data set-up time, 250 ns minimum, between setting
         * SDA and raising SCL.  It's expected that the I/O access latency will
         * naturally provide that delay.
         */
        I2C_SETSDA(dev, sda);
        I2C_SETSCL(dev, 1);
        return (iicbb_waitforscl(dev));
}

/*
 * End the high phase of the clock and wait out the low phase
 * as nothing interesting happens during it anyway.
 */
static void
iicbb_clockout(device_t dev)
{
        struct iicbb_softc *sc = device_get_softc(dev);

        /*
         * Precondition: SCL is high.
         * Action:
         * - pull SCL low and hold for udelay.
         */
        I2C_SETSCL(dev, 0);
        DELAY(sc->udelay);
}

static int
iicbb_sendbit(device_t dev, int bit)
{
        struct iicbb_softc *sc = device_get_softc(dev);
        int err;

        err = iicbb_clockin(dev, bit);
        if (err != 0)
                return (err);
        DELAY(sc->udelay);
        iicbb_clockout(dev);
        return (0);
}

/*
 * Waiting for ACKNOWLEDGE.
 *
 * When a chip is being addressed or has received data it will issue an
 * ACKNOWLEDGE pulse. Therefore the MASTER must release the DATA line
 * (set it to high level) and then release the CLOCK line.
 * Now it must wait for the SLAVE to pull the DATA line low.
 * Actually on the bus this looks like a START condition so nothing happens
 * because of the fact that the IC's that have not been addressed are doing
 * nothing.
 *
 * When the SLAVE has pulled this line low the MASTER will take the CLOCK
 * line low and then the SLAVE will release the SDA (data) line.
 */
static int
iicbb_getack(device_t dev)
{
        struct iicbb_softc *sc = device_get_softc(dev);
        int noack, err;
        int t = 0;

        /* Release SDA so that the slave can drive it. */
        err = iicbb_clockin(dev, 1);
        if (err != 0) {
                I2C_DEBUG(printf("! "));
                return (err);
        }

        /* Sample SDA until ACK (low) or udelay runs out. */
        do {
                noack = I2C_GETSDA(dev);
                if (!noack)
                        break;
                DELAY(1);
                t++;
        } while(t < sc->udelay);

        DELAY(sc->udelay - t);
        iicbb_clockout(dev);

        I2C_DEBUG(printf("%c ", noack ? '-' : '+'));
        return (noack ? IIC_ENOACK : 0);
}

static int
iicbb_sendbyte(device_t dev, uint8_t data)
{
        int err, i;

        for (i = 7; i >= 0; i--) {
                err = iicbb_sendbit(dev, (data & (1 << i)) != 0);
                if (err != 0) {
                        I2C_DEBUG(printf("w!"));
                        return (err);
                }
        }
        I2C_DEBUG(printf("w%02x", data));
        return (0);
}

static int
iicbb_readbyte(device_t dev, bool last, uint8_t *data)
{
        struct iicbb_softc *sc = device_get_softc(dev);
        int i, err;

        /*
         * Release SDA so that the slave can drive it.
         * We do not use iicbb_clockin() here because we need to release SDA
         * only once and then we just pulse the SCL.
         */
        *data = 0;
        I2C_SETSDA(dev, 1);
        for (i = 7; i >= 0; i--) {
                I2C_SETSCL(dev, 1);
                err = iicbb_waitforscl(dev);
                if (err != 0) {
                        I2C_DEBUG(printf("r! "));
                        return (err);
                }
                DELAY((sc->udelay + 1) / 2);
                if (I2C_GETSDA(dev))
                        *data |= 1 << i;
                DELAY((sc->udelay + 1) / 2);
                iicbb_clockout(dev);
        }

        /*
         * Send master->slave ACK (low) for more data,
         * NoACK (high) otherwise.
         */
        iicbb_sendbit(dev, last);
        I2C_DEBUG(printf("r%02x%c ", *data, last ? '-' : '+'));
        return (0);
}

static int
iicbb_callback(device_t dev, int index, caddr_t data)
{
        return (IICBB_CALLBACK(device_get_parent(dev), index, data));
}

static int
iicbb_reset(device_t dev, u_char speed, u_char addr, u_char *oldaddr)
{
        iicbb_set_speed(device_get_softc(dev), speed);
        return (IICBB_RESET(device_get_parent(dev), speed, addr, oldaddr));
}

static int
iicbb_start_impl(device_t dev, u_char slave, bool repstart)
{
        struct iicbb_softc *sc = device_get_softc(dev);
        int error;

        if (!repstart) {
                I2C_DEBUG(printf("<<"));

                /* SCL must be high on the idle bus. */
                if (iicbb_waitforscl(dev) != 0) {
                        I2C_DEBUG(printf("C!\n"));
                        return (IIC_EBUSERR);
                }
        } else {
                I2C_DEBUG(printf("<"));
                error = iicbb_clockin(dev, 1);
                if (error != 0)
                        return (error);

                /* SDA will go low in the middle of the SCL high phase. */
                DELAY((sc->udelay + 1) / 2);
        }

        /*
         * SDA must be high after the earlier stop condition or the end
         * of Ack/NoAck pulse.
         */
        if (!I2C_GETSDA(dev)) {
                I2C_DEBUG(printf("D!\n"));
                return (IIC_EBUSERR);
        }

        /* Start: SDA high->low. */
        I2C_SETSDA(dev, 0);

        /* Wait the second half of the SCL high phase. */
        DELAY((sc->udelay + 1) / 2);

        /* Pull SCL low to keep the bus reserved. */
        iicbb_clockout(dev);

        /* send address */
        error = iicbb_sendbyte(dev, slave);

        /* check for ack */
        if (error == 0)
                error = iicbb_getack(dev);
        if (error != 0)
                (void)iicbb_stop(dev);
        return (error);
}

/* NB: the timeout is ignored. */
static int
iicbb_start(device_t dev, u_char slave, int timeout)
{
        return (iicbb_start_impl(dev, slave, false));
}

/* NB: the timeout is ignored. */
static int
iicbb_repstart(device_t dev, u_char slave, int timeout)
{
        return (iicbb_start_impl(dev, slave, true));
}

static int
iicbb_stop(device_t dev)
{
        struct iicbb_softc *sc = device_get_softc(dev);
        int err = 0;

        /*
         * Stop: SDA goes from low to high in the middle of the SCL high phase.
         */
        err = iicbb_clockin(dev, 0);
        if (err != 0)
                return (err);
        DELAY((sc->udelay + 1) / 2);
        I2C_SETSDA(dev, 1);
        DELAY((sc->udelay + 1) / 2);

        I2C_DEBUG(printf("%s>>", err != 0 ? "!" : ""));
        I2C_DEBUG(printf("\n"));
        return (err);
}

/* NB: the timeout is ignored. */
static int
iicbb_write(device_t dev, const char *buf, int len, int *sent, int timeout)
{
        int bytes, error = 0;

        bytes = 0;
        while (len > 0) {
                /* send byte */
                iicbb_sendbyte(dev, (uint8_t)*buf++);

                /* check for ack */
                error = iicbb_getack(dev);
                if (error != 0)
                        break;
                bytes++;
                len--;
        }

        *sent = bytes;
        return (error);
}

/* NB: whatever delay is, it's ignored. */
static int
iicbb_read(device_t dev, char *buf, int len, int *read, int last, int delay)
{
        int bytes = 0;
        int err = 0;

        while (len > 0) {
                err = iicbb_readbyte(dev, (len == 1) ? last : 0,
                    (uint8_t *)buf);
                if (err != 0)
                        break;
                buf++;
                bytes++;
                len--;
        }

        *read = bytes;
        return (err);
}

static int
iicbb_transfer(device_t dev, struct iic_msg *msgs, uint32_t nmsgs)
{
        int error;

        error = IICBB_PRE_XFER(device_get_parent(dev));
        if (error)
                return (error);

        error = iicbus_transfer_gen(dev, msgs, nmsgs);

        IICBB_POST_XFER(device_get_parent(dev));
        return (error);
}

static void
iicbb_set_speed(struct iicbb_softc *sc, u_char speed)
{
        u_int busfreq;
        int period;

        /*
         * udelay is half a period, the clock is held high or low for this long.
         */
        busfreq = IICBUS_GET_FREQUENCY(sc->iicbus, speed);
        period = 1000000 / 2 / busfreq; /* Hz -> uS */
        period -= sc->io_latency;
        sc->udelay = MAX(period, 1);
}

DRIVER_MODULE(iicbus, iicbb, iicbus_driver, 0, 0);

MODULE_DEPEND(iicbb, iicbus, IICBUS_MINVER, IICBUS_PREFVER, IICBUS_MAXVER);
MODULE_VERSION(iicbb, IICBB_MODVER);