// SPDX-License-Identifier: GPL-2.0-only OR MIT
/* Copyright 2021 Alyssa Rosenzweig <alyssa@rosenzweig.io> */

#ifndef __APPLE_DCPEP_H__
#define __APPLE_DCPEP_H__

#include <linux/types.h>

#include "version_utils.h"

/* Fixed size of shared memory between DCP and AP */
#define DCP_SHMEM_SIZE 0x100000

/* DCP message contexts */
enum dcp_context_id {
	/* Callback */
	DCP_CONTEXT_CB = 0,

	/* Command */
	DCP_CONTEXT_CMD = 2,

	/* Asynchronous */
	DCP_CONTEXT_ASYNC = 3,

	/* Out-of-band callback */
	DCP_CONTEXT_OOBCB = 4,

	/* Out-of-band command */
	DCP_CONTEXT_OOBCMD = 6,

	/* Out-of-band Asynchronous */
	DCP_CONTEXT_OOBASYNC = 7,

	DCP_NUM_CONTEXTS
};

/* RTKit endpoint message types */
enum dcpep_type {
	/* Set shared memory */
	IOMFB_MESSAGE_TYPE_SET_SHMEM = 0,

	/* DCP is initialized */
	IOMFB_MESSAGE_TYPE_INITIALIZED = 1,

	/* Remote procedure call */
	IOMFB_MESSAGE_TYPE_MSG = 2,
};

#define IOMFB_MESSAGE_TYPE	GENMASK_ULL( 3,  0)

/* Message */
#define IOMFB_MSG_LENGTH	GENMASK_ULL(63, 32)
#define IOMFB_MSG_OFFSET	GENMASK_ULL(31, 16)
#define IOMFB_MSG_CONTEXT	GENMASK_ULL(11,  8)
#define IOMFB_MSG_ACK		BIT_ULL(6)

/* Set shmem */
#define IOMFB_SHMEM_DVA		GENMASK_ULL(63, 16)
#define IOMFB_SHMEM_FLAG	GENMASK_ULL( 7,  4)
#define IOMFB_SHMEM_FLAG_VALUE	4

struct dcp_packet_header {
	char tag[4];
	u32 in_len;
	u32 out_len;
} __packed;

#define DCP_IS_NULL(ptr) ((ptr) ? 1 : 0)
#define DCP_PACKET_ALIGNMENT (0x40)

enum iomfb_property_id {
    IOMFB_PROPERTY_NITS = 15, // divide by Brightness_Scale
};

#define IOMFB_BRIGHTNESS_MIN 0x10000000

/* Structures used in v12.0 firmware */

#define SWAP_SURFACES 4
#define MAX_PLANES 3

enum dcp_colorspace {
	DCP_COLORSPACE_BG_SRGB = 0,
	DCP_COLORSPACE_BG_BT2020 = 9,
	DCP_COLORSPACE_NATIVE = 12,
};

enum dcp_xfer_func {
	DCP_XFER_FUNC_SDR = 13,
	DCP_XFER_FUNC_HDR = 16,
};

struct dcp_iouserclient {
	/* Handle for the IOUserClient. macOS sets this to a kernel VA. */
	u64 handle;
	u32 unk;
	u8 flag1;
	u8 flag2;
	u8 padding[2];
} __packed;

struct dcp_rect {
	u32 x;
	u32 y;
	u32 w;
	u32 h;
} __packed;

/*
 * Update background color to struct dcp_swap.bg_color
 */
#define IOMFB_SET_BACKGROUND	BIT(31)

/* Information describing a plane of a planar compressed surface */
struct dcp_plane_info {
	u32 width;
	u32 height;
	u32 base;
	u32 offset;
	u32 stride;
	u32 size;
	u16 tile_size;
	u8 tile_w;
	u8 tile_h;
	u32 unk[13];
} __packed;

struct dcp_component_types {
	u8 count;
	u8 types[7];
} __packed;

struct dcp_allocate_bandwidth_req {
	u64 unk1;
	u64 unk2;
	u64 unk3;
	u8 unk1_null;
	u8 unk2_null;
	u8 padding[8];
} __packed;

struct dcp_allocate_bandwidth_resp {
	u64 unk1;
	u64 unk2;
	u32 ret;
} __packed;

struct dcp_rt_bandwidth {
	u64 unk1;
	u64 reg_scratch;
	u64 reg_doorbell;
	u32 unk2;
	u32 doorbell_bit;
	u32 padding[7];
} __packed;

struct frame_sync_props {
	u8 unk[28];
};

struct dcp_set_frame_sync_props_req {
	struct frame_sync_props props;
	u8 frame_sync_props_null;
	u8 padding[3];
} __packed;

struct dcp_set_frame_sync_props_resp {
	struct frame_sync_props props;
} __packed;

/* Method calls */

enum dcpep_method {
	dcpep_late_init_signal,
	dcpep_setup_video_limits,
	dcpep_set_create_dfb,
	dcpep_start_signal,
	dcpep_swap_start,
	dcpep_swap_submit,
	dcpep_set_display_device,
	dcpep_set_digital_out_mode,
	dcpep_create_default_fb,
	dcpep_set_display_refresh_properties,
	dcpep_flush_supports_power,
	dcpep_set_power_state,
	dcpep_first_client_open,
	dcpep_set_parameter_dcp,
	dcpep_enable_disable_video_power_savings,
	dcpep_is_main_display,
	iomfbep_a131_pmu_service_matched,
	iomfbep_a132_backlight_service_matched,
	iomfbep_a358_vi_set_temperature_hint,
	iomfbep_get_color_remap_mode,
	iomfbep_last_client_close,
	iomfbep_abort_swaps_dcp,
	iomfbep_set_matrix,
	dcpep_num_methods
};

#define IOMFB_METHOD(tag, name) [name] = { #name, tag }

struct dcp_method_entry {
	const char *name;
	char tag[4];
};

#define IOMFB_MAX_CB (1000)
struct apple_dcp;

typedef bool (*iomfb_cb_handler)(struct apple_dcp *, int, void *, void *);

/* Prototypes */

struct dcp_set_digital_out_mode_req {
	u32 color_mode_id;
	u32 timing_mode_id;
} __packed;

struct dcp_map_buf_req {
	u64 buffer;
	u8 unk;
	u8 buf_null;
	u8 vaddr_null;
	u8 dva_null;
} __packed;

struct dcp_map_buf_resp {
	u64 vaddr;
	u64 dva;
	u32 ret;
} __packed;

struct dcp_unmap_buf_resp {
	u64 buffer;
	u64 vaddr;
	u64 dva;
	u8 unk;
	u8 buf_null;
} __packed;

struct dcp_allocate_buffer_req {
	u32 unk0;
	u64 size;
	u32 unk2;
	u8 paddr_null;
	u8 dva_null;
	u8 dva_size_null;
	u8 padding;
} __packed;

struct dcp_allocate_buffer_resp {
	u64 paddr;
	u64 dva;
	u64 dva_size;
	u32 mem_desc_id;
} __packed;

struct dcp_map_physical_req {
	u64 paddr;
	u64 size;
	u32 flags;
	u8 dva_null;
	u8 dva_size_null;
	u8 padding[2];
} __packed;

struct dcp_map_physical_resp {
	u64 dva;
	u64 dva_size;
	u32 mem_desc_id;
} __packed;

struct dcp_swap_start_req {
	u32 swap_id;
	struct dcp_iouserclient client;
	u8 swap_id_null;
	u8 client_null;
	u8 padding[2];
} __packed;

struct dcp_swap_start_resp {
	u32 swap_id;
	struct dcp_iouserclient client;
	u32 ret;
} __packed;

struct dcp_get_uint_prop_req {
	char obj[4];
	char key[0x40];
	u64 value;
	u8 value_null;
	u8 padding[3];
} __packed;

struct dcp_get_uint_prop_resp {
	u64 value;
	u8 ret;
	u8 padding[3];
} __packed;

struct iomfb_sr_set_property_int_req {
	char obj[4];
	char key[0x40];
	u64 value;
	u8 value_null;
	u8 padding[3];
} __packed;

struct iomfb_set_fx_prop_req {
	char obj[4];
	char key[0x40];
	u32 value;
} __packed;

struct dcp_set_power_state_req {
	u64 unklong;
	u8 unkbool;
	u8 unkint_null;
	u8 padding[2];
} __packed;

struct dcp_set_power_state_resp {
	u32 unkint;
	u32 ret;
} __packed;

struct dcp_set_dcpav_prop_chunk_req {
	char data[0x1000];
	u32 offset;
	u32 length;
} __packed;

struct dcp_set_dcpav_prop_end_req {
	char key[0x40];
} __packed;

struct dcp_set_parameter_dcp {
	u32 param;
	u32 value[8];
	u32 count;
} __packed;

struct dcp_swap_complete_intent_gated {
	u32 swap_id;
	u8 unkBool;
	u32 unkInt;
	u32 width;
	u32 height;
} __packed;

struct dcp_read_edt_data_req {
	char key[0x40];
	u32 count;
	u32 value[8];
} __packed;

struct dcp_read_edt_data_resp {
	u32 value[8];
	u8 ret;
} __packed;

struct iomfb_property {
	u32 id;
	u32 value;
} __packed;

struct iomfb_get_color_remap_mode_req {
	u32 mode;
	u8 mode_null;
	u8 padding[3];
} __packed;

struct iomfb_get_color_remap_mode_resp {
	u32 mode;
	u32 ret;
} __packed;

struct iomfb_last_client_close_req {
	u8 unkint_null;
	u8 padding[3];
} __packed;

struct iomfb_last_client_close_resp {
	u32 unkint;
} __packed;

struct io_user_client {
	u64 addr;
	u32 unk;
	u8 flag1;
	u8 flag2;
	u8 pad[2];
} __packed;

struct iomfb_abort_swaps_dcp_req {
	struct io_user_client client;
	u8 client_null;
	u8 pad[3];
} __packed;

struct iomfb_abort_swaps_dcp_resp {
	struct io_user_client client;
	u32 ret;
} __packed;

struct iomfb_set_matrix_req {
	u32 unk_u32; // maybe length?
	u64 r[3];
	u64 g[3];
	u64 b[3];
	u8 matrix_null;
	u8 padding[3];
} __packed;

struct iomfb_set_matrix_resp {
	u32 ret;
} __packed;

struct dcpep_get_tiling_state_req {
	u32 event;
	u32 param;
	u32 value;
	u8 value_null;
	u8 padding[3];
} __packed;

struct dcpep_get_tiling_state_resp {
	u32 value;
	u32 ret;
} __packed;

#endif