/* SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause */
/* Copyright (c) 2010-2012 Broadcom. All rights reserved. */

#ifndef VCHIQ_CORE_H
#define VCHIQ_CORE_H

#include <linux/mutex.h>
#include <linux/completion.h>
#include <linux/dma-mapping.h>
#include <linux/dev_printk.h>
#include <linux/kthread.h>
#include <linux/kref.h>
#include <linux/rcupdate.h>
#include <linux/seq_file.h>
#include <linux/spinlock_types.h>
#include <linux/wait.h>

#include "vchiq.h"
#include "vchiq_cfg.h"

/* Do this so that we can test-build the code on non-rpi systems */
#if IS_ENABLED(CONFIG_RASPBERRYPI_FIRMWARE)

#else

#ifndef dsb
#define dsb(a)
#endif

#endif  /* IS_ENABLED(CONFIG_RASPBERRYPI_FIRMWARE) */

#define VCHIQ_SERVICE_HANDLE_INVALID 0

#define VCHIQ_SLOT_SIZE     4096
#define VCHIQ_MAX_MSG_SIZE  (VCHIQ_SLOT_SIZE - sizeof(struct vchiq_header))

#define VCHIQ_SLOT_MASK        (VCHIQ_SLOT_SIZE - 1)
#define VCHIQ_SLOT_QUEUE_MASK  (VCHIQ_MAX_SLOTS_PER_SIDE - 1)
#define VCHIQ_SLOT_ZERO_SLOTS  DIV_ROUND_UP(sizeof(struct vchiq_slot_zero), \
                                            VCHIQ_SLOT_SIZE)

#define BITSET_SIZE(b)        ((b + 31) >> 5)
#define BITSET_WORD(b)        (b >> 5)
#define BITSET_BIT(b)         (1 << (b & 31))
#define BITSET_IS_SET(bs, b)  (bs[BITSET_WORD(b)] & BITSET_BIT(b))
#define BITSET_SET(bs, b)     (bs[BITSET_WORD(b)] |= BITSET_BIT(b))

enum {
        DEBUG_ENTRIES,
#if VCHIQ_ENABLE_DEBUG
        DEBUG_SLOT_HANDLER_COUNT,
        DEBUG_SLOT_HANDLER_LINE,
        DEBUG_PARSE_LINE,
        DEBUG_PARSE_HEADER,
        DEBUG_PARSE_MSGID,
        DEBUG_AWAIT_COMPLETION_LINE,
        DEBUG_DEQUEUE_MESSAGE_LINE,
        DEBUG_SERVICE_CALLBACK_LINE,
        DEBUG_MSG_QUEUE_FULL_COUNT,
        DEBUG_COMPLETION_QUEUE_FULL_COUNT,
#endif
        DEBUG_MAX
};

#if VCHIQ_ENABLE_DEBUG

#define DEBUG_INITIALISE(local) int *debug_ptr = (local)->debug
#define DEBUG_TRACE(d) \
        do { debug_ptr[DEBUG_ ## d] = __LINE__; dsb(sy); } while (0)
#define DEBUG_VALUE(d, v) \
        do { debug_ptr[DEBUG_ ## d] = (v); dsb(sy); } while (0)
#define DEBUG_COUNT(d) \
        do { debug_ptr[DEBUG_ ## d]++; dsb(sy); } while (0)

#else /* VCHIQ_ENABLE_DEBUG */

#define DEBUG_INITIALISE(local)
#define DEBUG_TRACE(d)
#define DEBUG_VALUE(d, v)
#define DEBUG_COUNT(d)

#endif /* VCHIQ_ENABLE_DEBUG */

enum vchiq_connstate {
        VCHIQ_CONNSTATE_DISCONNECTED,
        VCHIQ_CONNSTATE_CONNECTING,
        VCHIQ_CONNSTATE_CONNECTED,
        VCHIQ_CONNSTATE_PAUSING,
        VCHIQ_CONNSTATE_PAUSE_SENT,
        VCHIQ_CONNSTATE_PAUSED,
        VCHIQ_CONNSTATE_RESUMING,
        VCHIQ_CONNSTATE_PAUSE_TIMEOUT,
        VCHIQ_CONNSTATE_RESUME_TIMEOUT
};

enum {
        VCHIQ_SRVSTATE_FREE,
        VCHIQ_SRVSTATE_HIDDEN,
        VCHIQ_SRVSTATE_LISTENING,
        VCHIQ_SRVSTATE_OPENING,
        VCHIQ_SRVSTATE_OPEN,
        VCHIQ_SRVSTATE_OPENSYNC,
        VCHIQ_SRVSTATE_CLOSESENT,
        VCHIQ_SRVSTATE_CLOSERECVD,
        VCHIQ_SRVSTATE_CLOSEWAIT,
        VCHIQ_SRVSTATE_CLOSED
};

enum vchiq_bulk_dir {
        VCHIQ_BULK_TRANSMIT,
        VCHIQ_BULK_RECEIVE
};

struct vchiq_bulk {
        short mode;
        short dir;
        void *cb_data;
        void __user *cb_userdata;
        struct bulk_waiter *waiter;
        dma_addr_t dma_addr;
        int size;
        void *remote_data;
        int remote_size;
        int actual;
        void *offset;
        void __user *uoffset;
};

struct vchiq_bulk_queue {
        int local_insert;  /* Where to insert the next local bulk */
        int remote_insert; /* Where to insert the next remote bulk (master) */
        int process;       /* Bulk to transfer next */
        int remote_notify; /* Bulk to notify the remote client of next (mstr) */
        int remove;        /* Bulk to notify the local client of, and remove, next */
        struct vchiq_bulk bulks[VCHIQ_NUM_SERVICE_BULKS];
};

/*
 * Remote events provide a way of presenting several virtual doorbells to a
 * peer (ARM host to VPU) using only one physical doorbell. They can be thought
 * of as a way for the peer to signal a semaphore, in this case implemented as
 * a workqueue.
 *
 * Remote events remain signalled until acknowledged by the receiver, and they
 * are non-counting. They are designed in such a way as to minimise the number
 * of interrupts and avoid unnecessary waiting.
 *
 * A remote_event is as small data structures that live in shared memory. It
 * comprises two booleans - armed and fired:
 *
 * The sender sets fired when they signal the receiver.
 * If fired is set, the receiver has been signalled and need not wait.
 * The receiver sets the armed field before they begin to wait.
 * If armed is set, the receiver is waiting and wishes to be woken by interrupt.
 */
struct remote_event {
        int armed;
        int fired;
        u32 __unused;
};

struct opaque_platform_state;

struct vchiq_slot {
        char data[VCHIQ_SLOT_SIZE];
};

struct vchiq_slot_info {
        /* Use two counters rather than one to avoid the need for a mutex. */
        short use_count;
        short release_count;
};

/*
 * VCHIQ is a reliable connection-oriented datagram protocol.
 *
 * A VCHIQ service is equivalent to a TCP connection, except:
 * + FOURCCs are used for the rendezvous, and port numbers are assigned at the
 *   time the connection is established.
 * + There is less of a distinction between server and client sockets, the only
 *   difference being which end makes the first move.
 * + For a multi-client server, the server creates new "listening" services as
 *   the existing one becomes connected - there is no need to specify the
 *   maximum number of clients up front.
 * + Data transfer is reliable but packetized (messages have defined ends).
 * + Messages can be either short (capable of fitting in a slot) and in-band,
 *   or copied between external buffers (bulk transfers).
 */
struct vchiq_service {
        struct vchiq_service_base base;
        unsigned int handle;
        struct kref ref_count;
        struct rcu_head rcu;
        int srvstate;
        void (*userdata_term)(void *userdata);
        unsigned int localport;
        unsigned int remoteport;
        int public_fourcc;
        int client_id;
        char auto_close;
        char sync;
        char closing;
        char trace;
        atomic_t poll_flags;
        short version;
        short version_min;
        short peer_version;

        struct vchiq_state *state;
        struct vchiq_instance *instance;

        int service_use_count;

        struct vchiq_bulk_queue bulk_tx;
        struct vchiq_bulk_queue bulk_rx;

        struct completion remove_event;
        struct completion bulk_remove_event;
        struct mutex bulk_mutex;

        struct service_stats_struct {
                int quota_stalls;
                int slot_stalls;
                int bulk_stalls;
                int error_count;
                int ctrl_tx_count;
                int ctrl_rx_count;
                int bulk_tx_count;
                int bulk_rx_count;
                int bulk_aborted_count;
                u64 ctrl_tx_bytes;
                u64 ctrl_rx_bytes;
                u64 bulk_tx_bytes;
                u64 bulk_rx_bytes;
        } stats;

        int msg_queue_read;
        int msg_queue_write;
        struct completion msg_queue_pop;
        struct completion msg_queue_push;
        struct vchiq_header *msg_queue[VCHIQ_MAX_SLOTS];
};

/*
 * The quota information is outside struct vchiq_service so that it can
 * be statically allocated, since for accounting reasons a service's slot
 * usage is carried over between users of the same port number.
 */
struct vchiq_service_quota {
        unsigned short slot_quota;
        unsigned short slot_use_count;
        unsigned short message_quota;
        unsigned short message_use_count;
        struct completion quota_event;
        int previous_tx_index;
};

struct vchiq_shared_state {
        /* A non-zero value here indicates that the content is valid. */
        int initialised;

        /* The first and last (inclusive) slots allocated to the owner. */
        int slot_first;
        int slot_last;

        /* The slot allocated to synchronous messages from the owner. */
        int slot_sync;

        /*
         * Signalling this event indicates that owner's slot handler thread
         * should run.
         */
        struct remote_event trigger;

        /*
         * Indicates the byte position within the stream where the next message
         * will be written. The least significant bits are an index into the
         * slot. The next bits are the index of the slot in slot_queue.
         */
        int tx_pos;

        /* This event should be signalled when a slot is recycled. */
        struct remote_event recycle;

        /* The slot_queue index where the next recycled slot will be written. */
        int slot_queue_recycle;

        /* This event should be signalled when a synchronous message is sent. */
        struct remote_event sync_trigger;

        /*
         * This event should be signalled when a synchronous message has been
         * released.
         */
        struct remote_event sync_release;

        /* A circular buffer of slot indexes. */
        int slot_queue[VCHIQ_MAX_SLOTS_PER_SIDE];

        /* Debugging state */
        int debug[DEBUG_MAX];
};

/*
 * vchiq_slot_zero describes the memory shared between the ARM host and the
 * VideoCore VPU. The "master" and "slave" states are owned by the respective
 * sides but visible to the other; the slots are shared, and the remaining
 * fields are read-only.
 *
 * In the configuration used by this implementation, the memory is allocated
 * by the host, the VPU is the master (the side which controls the DMA for bulk
 * transfers), and the host is the slave.
 *
 * The ownership of slots changes with use:
 * + When empty they are owned by the sender.
 * + When partially filled they are shared with the receiver.
 * + When completely full they are owned by the receiver.
 * + When the receiver has finished processing the contents, they are recycled
 *   back to the sender.
 */
struct vchiq_slot_zero {
        int magic;
        short version;
        short version_min;
        int slot_zero_size;
        int slot_size;
        int max_slots;
        int max_slots_per_side;
        int platform_data[2];
        struct vchiq_shared_state master;
        struct vchiq_shared_state slave;
        struct vchiq_slot_info slots[VCHIQ_MAX_SLOTS];
};

/*
 * This is the private runtime state used by each side. The same structure was
 * originally used by both sides, but implementations have since diverged.
 */
struct vchiq_state {
        struct device *dev;
        int id;
        int initialised;
        enum vchiq_connstate conn_state;
        short version_common;

        struct vchiq_shared_state *local;
        struct vchiq_shared_state *remote;
        struct vchiq_slot *slot_data;

        unsigned short default_slot_quota;
        unsigned short default_message_quota;

        /* Event indicating connect message received */
        struct completion connect;

        /* Mutex protecting services */
        struct mutex mutex;
        struct vchiq_instance **instance;

        /* Processes all incoming messages which aren't synchronous */
        struct task_struct *slot_handler_thread;

        /*
         * Slots which have been fully processed and released by the (peer)
         * receiver are added to the receiver queue, which is asynchronously
         * processed by the recycle thread.
         */
        struct task_struct *recycle_thread;

        /*
         * Processes incoming synchronous messages
         *
         * The synchronous message channel is shared between all synchronous
         * services, and provides a way for urgent messages to bypass
         * potentially long queues of asynchronous messages in the normal slots.
         *
         * There can be only one outstanding synchronous message in
         * each direction, and as a precious shared resource synchronous
         * services should be used sparingly.
         */
        struct task_struct *sync_thread;

        /* Local implementation of the trigger remote event */
        wait_queue_head_t trigger_event;

        /* Local implementation of the recycle remote event */
        wait_queue_head_t recycle_event;

        /* Local implementation of the sync trigger remote event */
        wait_queue_head_t sync_trigger_event;

        /* Local implementation of the sync release remote event */
        wait_queue_head_t sync_release_event;

        char *tx_data;
        char *rx_data;
        struct vchiq_slot_info *rx_info;

        struct mutex slot_mutex;

        struct mutex recycle_mutex;

        struct mutex sync_mutex;

        spinlock_t msg_queue_spinlock;

        spinlock_t bulk_waiter_spinlock;

        spinlock_t quota_spinlock;

        /*
         * Indicates the byte position within the stream from where the next
         * message will be read. The least significant bits are an index into
         * the slot.The next bits are the index of the slot in
         * remote->slot_queue.
         */
        int rx_pos;

        /*
         * A cached copy of local->tx_pos. Only write to local->tx_pos, and read
         * from remote->tx_pos.
         */
        int local_tx_pos;

        /* The slot_queue index of the slot to become available next. */
        int slot_queue_available;

        /* A flag to indicate if any poll has been requested */
        int poll_needed;

        /* Ths index of the previous slot used for data messages. */
        int previous_data_index;

        /* The number of slots occupied by data messages. */
        unsigned short data_use_count;

        /* The maximum number of slots to be occupied by data messages. */
        unsigned short data_quota;

        /* An array of bit sets indicating which services must be polled. */
        atomic_t poll_services[BITSET_SIZE(VCHIQ_MAX_SERVICES)];

        /* The number of the first unused service */
        int unused_service;

        /* Signalled when a free slot becomes available. */
        struct completion slot_available_event;

        /* Signalled when a free data slot becomes available. */
        struct completion data_quota_event;

        struct state_stats_struct {
                int slot_stalls;
                int data_stalls;
                int ctrl_tx_count;
                int ctrl_rx_count;
                int error_count;
        } stats;

        struct vchiq_service __rcu *services[VCHIQ_MAX_SERVICES];
        struct vchiq_service_quota service_quotas[VCHIQ_MAX_SERVICES];
        struct vchiq_slot_info slot_info[VCHIQ_MAX_SLOTS];

        struct opaque_platform_state *platform_state;
};

struct pagelist {
        u32 length;
        u16 type;
        u16 offset;
        u32 addrs[1];   /* N.B. 12 LSBs hold the number
                         * of following pages at consecutive
                         * addresses.
                         */
};

struct vchiq_pagelist_info {
        struct pagelist *pagelist;
        size_t pagelist_buffer_size;
        dma_addr_t dma_addr;
        enum dma_data_direction dma_dir;
        unsigned int num_pages;
        unsigned int pages_need_release;
        struct page **pages;
        struct scatterlist *scatterlist;
        unsigned int scatterlist_mapped;
};

static inline bool vchiq_remote_initialised(const struct vchiq_state *state)
{
        return state->remote && state->remote->initialised;
}

struct bulk_waiter {
        struct vchiq_bulk *bulk;
        struct completion event;
        int actual;
};

struct vchiq_config {
        unsigned int max_msg_size;
        unsigned int bulk_threshold;    /* The message size above which it
                                         * is better to use a bulk transfer
                                         * (<= max_msg_size)
                                         */
        unsigned int max_outstanding_bulks;
        unsigned int max_services;
        short version;      /* The version of VCHIQ */
        short version_min;  /* The minimum compatible version of VCHIQ */
};

extern spinlock_t bulk_waiter_spinlock;

extern const char *
get_conn_state_name(enum vchiq_connstate conn_state);

extern struct vchiq_slot_zero *
vchiq_init_slots(struct device *dev, void *mem_base, int mem_size);

extern int
vchiq_init_state(struct vchiq_state *state, struct vchiq_slot_zero *slot_zero, struct device *dev);

extern int
vchiq_connect_internal(struct vchiq_state *state, struct vchiq_instance *instance);

struct vchiq_service *
vchiq_add_service_internal(struct vchiq_state *state,
                           const struct vchiq_service_params_kernel *params,
                           int srvstate, struct vchiq_instance *instance,
                           void (*userdata_term)(void *userdata));

extern int
vchiq_open_service_internal(struct vchiq_service *service, int client_id);

extern int
vchiq_close_service_internal(struct vchiq_service *service, int close_recvd);

extern void
vchiq_terminate_service_internal(struct vchiq_service *service);

extern void
vchiq_free_service_internal(struct vchiq_service *service);

extern void
vchiq_shutdown_internal(struct vchiq_state *state, struct vchiq_instance *instance);

extern void
remote_event_pollall(struct vchiq_state *state);

extern int
vchiq_bulk_xfer_waiting(struct vchiq_instance *instance, unsigned int handle,
                        struct bulk_waiter *userdata);

extern int
vchiq_bulk_xfer_blocking(struct vchiq_instance *instance, unsigned int handle,
                         struct vchiq_bulk *bulk);

extern int
vchiq_bulk_xfer_callback(struct vchiq_instance *instance, unsigned int handle,
                         struct vchiq_bulk *bulk);

extern void
vchiq_dump_state(struct seq_file *f, struct vchiq_state *state);

extern void
request_poll(struct vchiq_state *state, struct vchiq_service *service,
             int poll_type);

struct vchiq_service *handle_to_service(struct vchiq_instance *instance, unsigned int handle);

extern struct vchiq_service *
find_service_by_handle(struct vchiq_instance *instance, unsigned int handle);

extern struct vchiq_service *
find_service_by_port(struct vchiq_state *state, unsigned int localport);

extern struct vchiq_service *
find_service_for_instance(struct vchiq_instance *instance, unsigned int handle);

extern struct vchiq_service *
find_closed_service_for_instance(struct vchiq_instance *instance, unsigned int handle);

extern struct vchiq_service *
__next_service_by_instance(struct vchiq_state *state,
                           struct vchiq_instance *instance,
                           int *pidx);

extern struct vchiq_service *
next_service_by_instance(struct vchiq_state *state,
                         struct vchiq_instance *instance,
                         int *pidx);

extern void
vchiq_service_get(struct vchiq_service *service);

extern void
vchiq_service_put(struct vchiq_service *service);

extern int
vchiq_queue_message(struct vchiq_instance *instance, unsigned int handle,
                    ssize_t (*copy_callback)(void *context, void *dest,
                                             size_t offset, size_t maxsize),
                    void *context,
                    size_t size);

void vchiq_dump_platform_state(struct seq_file *f);

void vchiq_dump_platform_instances(struct vchiq_state *state, struct seq_file *f);

void vchiq_dump_platform_service_state(struct seq_file *f, struct vchiq_service *service);

int vchiq_use_service_internal(struct vchiq_service *service);

int vchiq_release_service_internal(struct vchiq_service *service);

void vchiq_on_remote_use(struct vchiq_state *state);

void vchiq_on_remote_release(struct vchiq_state *state);

int vchiq_platform_init_state(struct vchiq_state *state);

int vchiq_check_service(struct vchiq_service *service);

int vchiq_send_remote_use(struct vchiq_state *state);

int vchiq_send_remote_use_active(struct vchiq_state *state);

void vchiq_platform_conn_state_changed(struct vchiq_state *state,
                                       enum vchiq_connstate oldstate,
                                  enum vchiq_connstate newstate);

void vchiq_set_conn_state(struct vchiq_state *state, enum vchiq_connstate newstate);

void vchiq_log_dump_mem(struct device *dev, const char *label, u32 addr,
                        const void *void_mem, size_t num_bytes);

int vchiq_remove_service(struct vchiq_instance *instance, unsigned int service);

int vchiq_get_client_id(struct vchiq_instance *instance, unsigned int service);

void vchiq_get_config(struct vchiq_config *config);

int vchiq_set_service_option(struct vchiq_instance *instance, unsigned int service,
                             enum vchiq_service_option option, int value);

#endif