/* SPDX-License-Identifier: GPL-2.0 */

/***************************************************************************
 *   copyright            : (C) 2004 by Frank Mori Hess
 ***************************************************************************/

#ifndef _NI_USB_GPIB_H
#define _NI_USB_GPIB_H

#include <linux/mutex.h>
#include <linux/semaphore.h>
#include <linux/usb.h>
#include <linux/timer.h>
#include "gpibP.h"

enum {
        USB_VENDOR_ID_NI = 0x3923
};

enum {
        USB_DEVICE_ID_NI_USB_B = 0x702a,
        USB_DEVICE_ID_NI_USB_B_PREINIT = 0x702b,        // device id before firmware is loaded
        USB_DEVICE_ID_NI_USB_HS = 0x709b,
        USB_DEVICE_ID_NI_USB_HS_PLUS = 0x7618,
        USB_DEVICE_ID_KUSB_488A = 0x725c,
        USB_DEVICE_ID_MC_USB_488 = 0x725d
};

enum ni_usb_device {
        NIUSB_SUBDEV_TNT4882 = 1,
        NIUSB_SUBDEV_UNKNOWN2 = 2,
        NIUSB_SUBDEV_UNKNOWN3 = 3,
};

enum endpoint_addresses {
        NIUSB_B_BULK_OUT_ENDPOINT = 0x2,
        NIUSB_B_BULK_IN_ENDPOINT = 0x2,
        NIUSB_B_BULK_IN_ALT_ENDPOINT = 0x6,
        NIUSB_B_INTERRUPT_IN_ENDPOINT = 0x4,
};

enum hs_enpoint_addresses {
        NIUSB_HS_BULK_OUT_ENDPOINT = 0x2,
        NIUSB_HS_BULK_OUT_ALT_ENDPOINT = 0x6,
        NIUSB_HS_BULK_IN_ENDPOINT = 0x4,
        NIUSB_HS_BULK_IN_ALT_ENDPOINT = 0x8,
        NIUSB_HS_INTERRUPT_IN_ENDPOINT = 0x1,
};

enum hs_plus_endpoint_addresses {
        NIUSB_HS_PLUS_BULK_OUT_ENDPOINT = 0x1,
        NIUSB_HS_PLUS_BULK_OUT_ALT_ENDPOINT = 0x4,
        NIUSB_HS_PLUS_BULK_IN_ENDPOINT = 0x2,
        NIUSB_HS_PLUS_BULK_IN_ALT_ENDPOINT = 0x5,
        NIUSB_HS_PLUS_INTERRUPT_IN_ENDPOINT = 0x3,
};

struct ni_usb_urb_ctx {
        struct completion complete;
        unsigned timed_out : 1;
};

// struct which defines private_data for ni_usb devices
struct ni_usb_priv {
        struct usb_interface *bus_interface;
        int bulk_out_endpoint;
        int bulk_in_endpoint;
        int interrupt_in_endpoint;
        u8 eos_char;
        unsigned short eos_mode;
        unsigned int monitored_ibsta_bits;
        struct urb *bulk_urb;
        struct urb *interrupt_urb;
        u8 interrupt_buffer[0x11];
        struct mutex addressed_transfer_lock;   // protect transfer lock
        struct mutex bulk_transfer_lock;        // protect bulk message sends
        struct mutex control_transfer_lock;     // protect control messages
        struct mutex interrupt_transfer_lock;   //  protect interrupt messages
        struct timer_list bulk_timer;
        struct ni_usb_urb_ctx context;
        int product_id;
        unsigned short ren_state;
};

struct ni_usb_status_block {
        short id;
        unsigned short ibsta;
        short error_code;
        unsigned short count;
};

struct ni_usb_register {
        enum ni_usb_device device;
        short address;
        unsigned short value;
};

enum ni_usb_bulk_ids {
        NIUSB_IBCAC_ID = 0x1,
        NIUSB_UNKNOWN3_ID = 0x3, // device level function id?
        NIUSB_TERM_ID = 0x4,
        NIUSB_IBGTS_ID = 0x6,
        NIUSB_IBRPP_ID = 0x7,
        NIUSB_REG_READ_ID = 0x8,
        NIUSB_REG_WRITE_ID = 0x9,
        NIUSB_IBSIC_ID = 0xf,
        NIUSB_REGISTER_READ_DATA_START_ID = 0x34,
        NIUSB_REGISTER_READ_DATA_END_ID = 0x35,
        NIUSB_IBRD_DATA_ID = 0x36,
        NIUSB_IBRD_EXTENDED_DATA_ID = 0x37,
        NIUSB_IBRD_STATUS_ID = 0x38
};

enum ni_usb_error_codes {
        NIUSB_NO_ERROR = 0,
        /*
         * NIUSB_ABORTED_ERROR occurs when I/O is interrupted early by
         * doing a NI_USB_STOP_REQUEST on the control endpoint.
         */
        NIUSB_ABORTED_ERROR = 1,
        /*
         * NIUSB_READ_ATN_ERROR occurs when you do a board read while
         * ATN is set
         */
        NIUSB_ATN_STATE_ERROR = 2,
        /*
         * NIUSB_ADDRESSING_ERROR occurs when you do a board
         * read/write as CIC but are not in LACS/TACS
         */
        NIUSB_ADDRESSING_ERROR = 3,
        /*
         * NIUSB_EOSMODE_ERROR occurs on reads if any eos mode or char
         * bits are set when REOS is not set.
         * Have also seen error 4 if you try to send more than 16
         * command bytes at once on a usb-b.
         */
        NIUSB_EOSMODE_ERROR = 4,
        /*
         * NIUSB_NO_BUS_ERROR occurs when you try to write a command
         * byte but there are no devices connected to the gpib bus
         */
        NIUSB_NO_BUS_ERROR = 5,
        /*
         * NIUSB_NO_LISTENER_ERROR occurs when you do a board write as
         * CIC with no listener
         */
        NIUSB_NO_LISTENER_ERROR = 8,
        /* get NIUSB_TIMEOUT_ERROR on board read/write timeout */
        NIUSB_TIMEOUT_ERROR = 10,
};

enum ni_usb_control_requests {
        NI_USB_STOP_REQUEST = 0x20,
        NI_USB_WAIT_REQUEST = 0x21,
        NI_USB_POLL_READY_REQUEST = 0x40,
        NI_USB_SERIAL_NUMBER_REQUEST = 0x41,
        NI_USB_HS_PLUS_0x48_REQUEST = 0x48,
        NI_USB_HS_PLUS_LED_REQUEST = 0x4b,
        NI_USB_HS_PLUS_0xf8_REQUEST = 0xf8
};

static const unsigned int ni_usb_ibsta_monitor_mask =
        SRQI | LOK | REM | CIC | ATN | TACS | LACS | DTAS | DCAS;

static inline int nec7210_to_tnt4882_offset(int offset)
{
        return 2 * offset;
};

static inline int ni_usb_bulk_termination(u8 *buffer)
{
        int i = 0;

        buffer[i++] = NIUSB_TERM_ID;
        buffer[i++] = 0x0;
        buffer[i++] = 0x0;
        buffer[i++] = 0x0;
        return i;
}

enum ni_usb_unknown3_register {
        SERIAL_NUMBER_4_REG = 0x8,
        SERIAL_NUMBER_3_REG = 0x9,
        SERIAL_NUMBER_2_REG = 0xa,
        SERIAL_NUMBER_1_REG = 0xb,
};

static inline int ni_usb_bulk_register_write_header(u8 *buffer, int num_writes)
{
        int i = 0;

        buffer[i++] = NIUSB_REG_WRITE_ID;
        buffer[i++] = num_writes;
        buffer[i++] = 0x0;
        return i;
}

static inline int ni_usb_bulk_register_write(u8 *buffer, struct ni_usb_register reg)
{
        int i = 0;

        buffer[i++] = reg.device;
        buffer[i++] = reg.address;
        buffer[i++] = reg.value;
        return i;
}

static inline int ni_usb_bulk_register_read_header(u8 *buffer, int num_reads)
{
        int i = 0;

        buffer[i++] = NIUSB_REG_READ_ID;
        buffer[i++] = num_reads;
        return i;
}

static inline int ni_usb_bulk_register_read(u8 *buffer, int device, int address)
{
        int i = 0;

        buffer[i++] = device;
        buffer[i++] = address;
        return i;
}

#endif  // _NI_USB_GPIB_H