/* SPDX-License-Identifier: GPL-2.0 */
#undef TRACE_SYSTEM
#define TRACE_SYSTEM spi-mem

#undef TRACE_SYSTEM_VAR
#define TRACE_SYSTEM_VAR spi_mem

#if !defined(_TRACE_SPI_MEM_H) || defined(TRACE_HEADER_MULTI_READ)
#define _TRACE_SPI_MEM_H

#include <linux/tracepoint.h>
#include <linux/spi/spi-mem.h>

#define decode_dtr(dtr) \
        __print_symbolic(dtr, \
                { 0, "S" }, \
                { 1, "D" })

TRACE_EVENT(spi_mem_start_op,
        TP_PROTO(struct spi_mem *mem, const struct spi_mem_op *op),
        TP_ARGS(mem, op),

        TP_STRUCT__entry(
                __string(name, mem->name)
                __dynamic_array(u8, op, 1 + op->addr.nbytes + op->dummy.nbytes)
                __dynamic_array(u8, data, op->data.dir == SPI_MEM_DATA_OUT ?
                                          min(op->data.nbytes, 64) : 0)
                __field(u32, data_len)
                __field(u32, max_freq)
                __field(u8, cmd_buswidth)
                __field(bool, cmd_dtr)
                __field(u8, addr_buswidth)
                __field(bool, addr_dtr)
                __field(u8, dummy_nbytes)
                __field(u8, data_buswidth)
                __field(bool, data_dtr)
        ),

        TP_fast_assign(
                int i;

                __assign_str(name);
                __entry->max_freq = op->max_freq ?: mem->spi->max_speed_hz;

                __entry->cmd_buswidth = op->cmd.buswidth;
                __entry->cmd_dtr = op->cmd.dtr;
                *((u8 *)__get_dynamic_array(op)) = op->cmd.opcode;

                __entry->addr_buswidth = op->addr.buswidth;
                __entry->addr_dtr = op->addr.dtr;
                for (i = 0; i < op->addr.nbytes; i++)
                        ((u8 *)__get_dynamic_array(op))[i + 1] =
                                op->addr.val >> (8 * (op->addr.nbytes - i - 1));

                memset(((u8 *)__get_dynamic_array(op)) + op->addr.nbytes + 1,
                       0xff, op->dummy.nbytes);

                __entry->data_len = op->data.nbytes;
                __entry->data_buswidth = op->data.buswidth;
                __entry->data_dtr = op->data.dtr;
                if (op->data.dir == SPI_MEM_DATA_OUT)
                        memcpy(__get_dynamic_array(data), op->data.buf.out,
                               __get_dynamic_array_len(data));
        ),

        TP_printk("%s %u%s-%u%s-%u%s @%u Hz op=[%*phD] len=%u tx=[%*phD]",
                __get_str(name),
                __entry->cmd_buswidth, decode_dtr(__entry->cmd_dtr),
                __entry->addr_buswidth, decode_dtr(__entry->addr_dtr),
                __entry->data_buswidth, decode_dtr(__entry->data_dtr),
                __entry->max_freq,
                __get_dynamic_array_len(op), __get_dynamic_array(op),
                __entry->data_len,
                __get_dynamic_array_len(data), __get_dynamic_array(data))
);

TRACE_EVENT(spi_mem_stop_op,
        TP_PROTO(struct spi_mem *mem, const struct spi_mem_op *op),
        TP_ARGS(mem, op),

        TP_STRUCT__entry(
                __string(name, mem->name)
                __dynamic_array(u8, data, op->data.dir == SPI_MEM_DATA_IN ?
                                          min(op->data.nbytes, 64) : 0)
                __field(u32, data_len)
        ),

        TP_fast_assign(
                __assign_str(name);
                __entry->data_len = op->data.nbytes;
                if (op->data.dir == SPI_MEM_DATA_IN)
                        memcpy(__get_dynamic_array(data), op->data.buf.in,
                               __get_dynamic_array_len(data));
        ),

        TP_printk("%s len=%u rx=[%*phD]",
                __get_str(name),
                __entry->data_len,
                __get_dynamic_array_len(data), __get_dynamic_array(data))
);


#endif /* _TRACE_SPI_MEM_H */

/* This part must be outside protection */
#include <trace/define_trace.h>