root/drivers/gpu/drm/rockchip/cdn-dp-core.c
// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) Rockchip Electronics Co., Ltd.
 * Author: Chris Zhong <zyw@rock-chips.com>
 */

#include <linux/clk.h>
#include <linux/component.h>
#include <linux/extcon.h>
#include <linux/firmware.h>
#include <linux/mfd/syscon.h>
#include <linux/phy/phy.h>
#include <linux/regmap.h>
#include <linux/reset.h>

#include <sound/hdmi-codec.h>

#include <drm/display/drm_dp_helper.h>
#include <drm/display/drm_hdmi_audio_helper.h>
#include <drm/drm_atomic_helper.h>
#include <drm/drm_bridge_connector.h>
#include <drm/drm_edid.h>
#include <drm/drm_of.h>
#include <drm/drm_print.h>
#include <drm/drm_probe_helper.h>
#include <drm/drm_simple_kms_helper.h>

#include "cdn-dp-core.h"
#include "cdn-dp-reg.h"

static inline struct cdn_dp_device *bridge_to_dp(struct drm_bridge *bridge)
{
        return container_of(bridge, struct cdn_dp_device, bridge);
}

static inline struct cdn_dp_device *encoder_to_dp(struct drm_encoder *encoder)
{
        struct rockchip_encoder *rkencoder = to_rockchip_encoder(encoder);

        return container_of(rkencoder, struct cdn_dp_device, encoder);
}

#define GRF_SOC_CON9            0x6224
#define DP_SEL_VOP_LIT          BIT(12)
#define GRF_SOC_CON26           0x6268
#define DPTX_HPD_SEL            (3 << 12)
#define DPTX_HPD_DEL            (2 << 12)
#define DPTX_HPD_SEL_MASK       (3 << 28)

#define CDN_FW_TIMEOUT_MS       (64 * 1000)
#define CDN_DPCD_TIMEOUT_MS     5000
#define CDN_DP_FIRMWARE         "rockchip/dptx.bin"
MODULE_FIRMWARE(CDN_DP_FIRMWARE);

struct cdn_dp_data {
        u8 max_phy;
};

static struct cdn_dp_data rk3399_cdn_dp = {
        .max_phy = 2,
};

static const struct of_device_id cdn_dp_dt_ids[] = {
        { .compatible = "rockchip,rk3399-cdn-dp",
                .data = (void *)&rk3399_cdn_dp },
        {}
};

MODULE_DEVICE_TABLE(of, cdn_dp_dt_ids);

static int cdn_dp_grf_write(struct cdn_dp_device *dp,
                            unsigned int reg, unsigned int val)
{
        int ret;

        ret = clk_prepare_enable(dp->grf_clk);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "Failed to prepare_enable grf clock\n");
                return ret;
        }

        ret = regmap_write(dp->grf, reg, val);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "Could not write to GRF: %d\n", ret);
                clk_disable_unprepare(dp->grf_clk);
                return ret;
        }

        clk_disable_unprepare(dp->grf_clk);

        return 0;
}

static int cdn_dp_clk_enable(struct cdn_dp_device *dp)
{
        int ret;
        unsigned long rate;

        ret = clk_prepare_enable(dp->pclk);
        if (ret < 0) {
                DRM_DEV_ERROR(dp->dev, "cannot enable dp pclk %d\n", ret);
                goto err_pclk;
        }

        ret = clk_prepare_enable(dp->core_clk);
        if (ret < 0) {
                DRM_DEV_ERROR(dp->dev, "cannot enable core_clk %d\n", ret);
                goto err_core_clk;
        }

        ret = pm_runtime_get_sync(dp->dev);
        if (ret < 0) {
                DRM_DEV_ERROR(dp->dev, "cannot get pm runtime %d\n", ret);
                goto err_pm_runtime_get;
        }

        reset_control_assert(dp->core_rst);
        reset_control_assert(dp->dptx_rst);
        reset_control_assert(dp->apb_rst);
        reset_control_deassert(dp->core_rst);
        reset_control_deassert(dp->dptx_rst);
        reset_control_deassert(dp->apb_rst);

        rate = clk_get_rate(dp->core_clk);
        if (!rate) {
                DRM_DEV_ERROR(dp->dev, "get clk rate failed\n");
                ret = -EINVAL;
                goto err_set_rate;
        }

        cdn_dp_set_fw_clk(dp, rate);
        cdn_dp_clock_reset(dp);

        return 0;

err_set_rate:
        pm_runtime_put(dp->dev);
err_pm_runtime_get:
        clk_disable_unprepare(dp->core_clk);
err_core_clk:
        clk_disable_unprepare(dp->pclk);
err_pclk:
        return ret;
}

static void cdn_dp_clk_disable(struct cdn_dp_device *dp)
{
        pm_runtime_put_sync(dp->dev);
        clk_disable_unprepare(dp->pclk);
        clk_disable_unprepare(dp->core_clk);
}

static int cdn_dp_get_port_lanes(struct cdn_dp_port *port)
{
        struct extcon_dev *edev = port->extcon;
        union extcon_property_value property;
        int dptx;
        u8 lanes;

        dptx = extcon_get_state(edev, EXTCON_DISP_DP);
        if (dptx > 0) {
                extcon_get_property(edev, EXTCON_DISP_DP,
                                    EXTCON_PROP_USB_SS, &property);
                if (property.intval)
                        lanes = 2;
                else
                        lanes = 4;
        } else {
                lanes = 0;
        }

        return lanes;
}

static int cdn_dp_get_sink_count(struct cdn_dp_device *dp, u8 *sink_count)
{
        int ret;
        u8 value;

        *sink_count = 0;
        ret = cdn_dp_dpcd_read(dp, DP_SINK_COUNT, &value, 1);
        if (ret)
                return ret;

        *sink_count = DP_GET_SINK_COUNT(value);
        return 0;
}

static struct cdn_dp_port *cdn_dp_connected_port(struct cdn_dp_device *dp)
{
        struct cdn_dp_port *port;
        int i, lanes;

        for (i = 0; i < dp->ports; i++) {
                port = dp->port[i];
                lanes = cdn_dp_get_port_lanes(port);
                if (lanes)
                        return port;
        }
        return NULL;
}

static bool cdn_dp_check_sink_connection(struct cdn_dp_device *dp)
{
        unsigned long timeout = jiffies + msecs_to_jiffies(CDN_DPCD_TIMEOUT_MS);
        struct cdn_dp_port *port;
        u8 sink_count = 0;

        if (dp->active_port < 0 || dp->active_port >= dp->ports) {
                DRM_DEV_ERROR(dp->dev, "active_port is wrong!\n");
                return false;
        }

        port = dp->port[dp->active_port];

        /*
         * Attempt to read sink count, retry in case the sink may not be ready.
         *
         * Sinks are *supposed* to come up within 1ms from an off state, but
         * some docks need more time to power up.
         */
        while (time_before(jiffies, timeout)) {
                if (!extcon_get_state(port->extcon, EXTCON_DISP_DP))
                        return false;

                if (!cdn_dp_get_sink_count(dp, &sink_count))
                        return sink_count ? true : false;

                usleep_range(5000, 10000);
        }

        DRM_DEV_ERROR(dp->dev, "Get sink capability timed out\n");
        return false;
}

static enum drm_connector_status
cdn_dp_bridge_detect(struct drm_bridge *bridge, struct drm_connector *connector)
{
        struct cdn_dp_device *dp = bridge_to_dp(bridge);
        enum drm_connector_status status = connector_status_disconnected;

        mutex_lock(&dp->lock);
        if (dp->connected)
                status = connector_status_connected;
        mutex_unlock(&dp->lock);

        return status;
}

static const struct drm_edid *
cdn_dp_bridge_edid_read(struct drm_bridge *bridge, struct drm_connector *connector)
{
        struct cdn_dp_device *dp = bridge_to_dp(bridge);
        const struct drm_edid *drm_edid;

        mutex_lock(&dp->lock);
        drm_edid = drm_edid_read_custom(connector, cdn_dp_get_edid_block, dp);
        mutex_unlock(&dp->lock);

        return drm_edid;
}

static enum drm_mode_status
cdn_dp_bridge_mode_valid(struct drm_bridge *bridge,
                         const struct drm_display_info *display_info,
                         const struct drm_display_mode *mode)
{
        struct cdn_dp_device *dp = bridge_to_dp(bridge);
        u32 requested, actual, rate, sink_max, source_max = 0;
        u8 lanes, bpc;

        /* If DP is disconnected, every mode is invalid */
        if (!dp->connected)
                return MODE_BAD;

        switch (display_info->bpc) {
        case 10:
                bpc = 10;
                break;
        case 6:
                bpc = 6;
                break;
        default:
                bpc = 8;
                break;
        }

        requested = mode->clock * bpc * 3 / 1000;

        source_max = dp->lanes;
        sink_max = drm_dp_max_lane_count(dp->dpcd);
        lanes = min(source_max, sink_max);

        source_max = drm_dp_bw_code_to_link_rate(CDN_DP_MAX_LINK_RATE);
        sink_max = drm_dp_max_link_rate(dp->dpcd);
        rate = min(source_max, sink_max);

        actual = rate * lanes / 100;

        /* efficiency is about 0.8 */
        actual = actual * 8 / 10;

        if (requested > actual) {
                DRM_DEV_DEBUG_KMS(dp->dev,
                                  "requested=%d, actual=%d, clock=%d\n",
                                  requested, actual, mode->clock);
                return MODE_CLOCK_HIGH;
        }

        return MODE_OK;
}

static int cdn_dp_firmware_init(struct cdn_dp_device *dp)
{
        int ret;
        const u32 *iram_data, *dram_data;
        const struct firmware *fw = dp->fw;
        const struct cdn_firmware_header *hdr;

        hdr = (struct cdn_firmware_header *)fw->data;
        if (fw->size != le32_to_cpu(hdr->size_bytes)) {
                DRM_DEV_ERROR(dp->dev, "firmware is invalid\n");
                return -EINVAL;
        }

        iram_data = (const u32 *)(fw->data + hdr->header_size);
        dram_data = (const u32 *)(fw->data + hdr->header_size + hdr->iram_size);

        ret = cdn_dp_load_firmware(dp, iram_data, hdr->iram_size,
                                   dram_data, hdr->dram_size);
        if (ret)
                return ret;

        ret = cdn_dp_set_firmware_active(dp, true);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "active ucpu failed: %d\n", ret);
                return ret;
        }

        return cdn_dp_event_config(dp);
}

static int cdn_dp_get_sink_capability(struct cdn_dp_device *dp)
{
        int ret;

        if (!cdn_dp_check_sink_connection(dp))
                return -ENODEV;

        ret = cdn_dp_dpcd_read(dp, DP_DPCD_REV, dp->dpcd,
                               DP_RECEIVER_CAP_SIZE);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "Failed to get caps %d\n", ret);
                return ret;
        }

        return 0;
}

static int cdn_dp_enable_phy(struct cdn_dp_device *dp, struct cdn_dp_port *port)
{
        union extcon_property_value property;
        int ret;

        if (!port->phy_enabled) {
                ret = phy_power_on(port->phy);
                if (ret) {
                        DRM_DEV_ERROR(dp->dev, "phy power on failed: %d\n",
                                      ret);
                        goto err_phy;
                }
                port->phy_enabled = true;
        }

        ret = cdn_dp_grf_write(dp, GRF_SOC_CON26,
                               DPTX_HPD_SEL_MASK | DPTX_HPD_SEL);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "Failed to write HPD_SEL %d\n", ret);
                goto err_power_on;
        }

        ret = cdn_dp_get_hpd_status(dp);
        if (ret <= 0) {
                if (!ret)
                        DRM_DEV_ERROR(dp->dev, "hpd does not exist\n");
                goto err_power_on;
        }

        ret = extcon_get_property(port->extcon, EXTCON_DISP_DP,
                                  EXTCON_PROP_USB_TYPEC_POLARITY, &property);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "get property failed\n");
                goto err_power_on;
        }

        port->lanes = cdn_dp_get_port_lanes(port);
        ret = cdn_dp_set_host_cap(dp, port->lanes, property.intval);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "set host capabilities failed: %d\n",
                              ret);
                goto err_power_on;
        }

        dp->active_port = port->id;
        return 0;

err_power_on:
        if (phy_power_off(port->phy))
                DRM_DEV_ERROR(dp->dev, "phy power off failed: %d", ret);
        else
                port->phy_enabled = false;

err_phy:
        cdn_dp_grf_write(dp, GRF_SOC_CON26,
                         DPTX_HPD_SEL_MASK | DPTX_HPD_DEL);
        return ret;
}

static int cdn_dp_disable_phy(struct cdn_dp_device *dp,
                              struct cdn_dp_port *port)
{
        int ret;

        if (port->phy_enabled) {
                ret = phy_power_off(port->phy);
                if (ret) {
                        DRM_DEV_ERROR(dp->dev, "phy power off failed: %d", ret);
                        return ret;
                }
        }

        port->phy_enabled = false;
        port->lanes = 0;
        dp->active_port = -1;
        return 0;
}

static int cdn_dp_disable(struct cdn_dp_device *dp)
{
        int ret, i;

        if (!dp->active)
                return 0;

        for (i = 0; i < dp->ports; i++)
                cdn_dp_disable_phy(dp, dp->port[i]);

        ret = cdn_dp_grf_write(dp, GRF_SOC_CON26,
                               DPTX_HPD_SEL_MASK | DPTX_HPD_DEL);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "Failed to clear hpd sel %d\n",
                              ret);
                return ret;
        }

        cdn_dp_set_firmware_active(dp, false);
        cdn_dp_clk_disable(dp);
        dp->active = false;
        dp->max_lanes = 0;
        dp->max_rate = 0;

        return 0;
}

static int cdn_dp_enable(struct cdn_dp_device *dp)
{
        int ret, i, lanes;
        struct cdn_dp_port *port;

        port = cdn_dp_connected_port(dp);
        if (!port) {
                DRM_DEV_ERROR(dp->dev,
                              "Can't enable without connection\n");
                return -ENODEV;
        }

        if (dp->active)
                return 0;

        ret = cdn_dp_clk_enable(dp);
        if (ret)
                return ret;

        ret = cdn_dp_firmware_init(dp);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "firmware init failed: %d", ret);
                goto err_clk_disable;
        }

        /* only enable the port that connected with downstream device */
        for (i = port->id; i < dp->ports; i++) {
                port = dp->port[i];
                lanes = cdn_dp_get_port_lanes(port);
                if (lanes) {
                        ret = cdn_dp_enable_phy(dp, port);
                        if (ret)
                                continue;

                        ret = cdn_dp_get_sink_capability(dp);
                        if (ret) {
                                cdn_dp_disable_phy(dp, port);
                        } else {
                                dp->active = true;
                                dp->lanes = port->lanes;
                                return 0;
                        }
                }
        }

err_clk_disable:
        cdn_dp_clk_disable(dp);
        return ret;
}

static void cdn_dp_bridge_mode_set(struct drm_bridge *bridge,
                                   const struct drm_display_mode *mode,
                                   const struct drm_display_mode *adjusted)
{
        struct cdn_dp_device *dp = bridge_to_dp(bridge);
        struct video_info *video = &dp->video_info;

        video->color_fmt = PXL_RGB;
        video->v_sync_polarity = !!(mode->flags & DRM_MODE_FLAG_NVSYNC);
        video->h_sync_polarity = !!(mode->flags & DRM_MODE_FLAG_NHSYNC);

        drm_mode_copy(&dp->mode, adjusted);
}

static bool cdn_dp_check_link_status(struct cdn_dp_device *dp)
{
        u8 link_status[DP_LINK_STATUS_SIZE];
        struct cdn_dp_port *port = cdn_dp_connected_port(dp);
        u8 sink_lanes = drm_dp_max_lane_count(dp->dpcd);

        if (!port || !dp->max_rate || !dp->max_lanes)
                return false;

        if (cdn_dp_dpcd_read(dp, DP_LANE0_1_STATUS, link_status,
                             DP_LINK_STATUS_SIZE)) {
                DRM_ERROR("Failed to get link status\n");
                return false;
        }

        /* if link training is requested we should perform it always */
        return drm_dp_channel_eq_ok(link_status, min(port->lanes, sink_lanes));
}

static void cdn_dp_display_info_update(struct cdn_dp_device *dp,
                                       struct drm_display_info *display_info)
{
        struct video_info *video = &dp->video_info;

        switch (display_info->bpc) {
        case 10:
                video->color_depth = 10;
                break;
        case 6:
                video->color_depth = 6;
                break;
        default:
                video->color_depth = 8;
                break;
        }
}

static void cdn_dp_bridge_atomic_enable(struct drm_bridge *bridge, struct drm_atomic_commit *state)
{
        struct cdn_dp_device *dp = bridge_to_dp(bridge);
        struct drm_connector *connector;
        int ret, val;

        connector = drm_atomic_get_new_connector_for_encoder(state, bridge->encoder);
        if (!connector)
                return;

        cdn_dp_display_info_update(dp, &connector->display_info);

        ret = drm_of_encoder_active_endpoint_id(dp->dev->of_node, &dp->encoder.encoder);
        if (ret < 0) {
                DRM_DEV_ERROR(dp->dev, "Could not get vop id, %d", ret);
                return;
        }

        DRM_DEV_DEBUG_KMS(dp->dev, "vop %s output to cdn-dp\n",
                          (ret) ? "LIT" : "BIG");
        if (ret)
                val = DP_SEL_VOP_LIT | (DP_SEL_VOP_LIT << 16);
        else
                val = DP_SEL_VOP_LIT << 16;

        ret = cdn_dp_grf_write(dp, GRF_SOC_CON9, val);
        if (ret)
                return;

        mutex_lock(&dp->lock);

        ret = cdn_dp_enable(dp);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "Failed to enable bridge %d\n",
                              ret);
                goto out;
        }
        if (!cdn_dp_check_link_status(dp)) {
                ret = cdn_dp_train_link(dp);
                if (ret) {
                        DRM_DEV_ERROR(dp->dev, "Failed link train %d\n", ret);
                        goto out;
                }
        }

        ret = cdn_dp_set_video_status(dp, CONTROL_VIDEO_IDLE);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "Failed to idle video %d\n", ret);
                goto out;
        }

        ret = cdn_dp_config_video(dp);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "Failed to config video %d\n", ret);
                goto out;
        }

        ret = cdn_dp_set_video_status(dp, CONTROL_VIDEO_VALID);
        if (ret) {
                DRM_DEV_ERROR(dp->dev, "Failed to valid video %d\n", ret);
                goto out;
        }

out:
        mutex_unlock(&dp->lock);
}

static void cdn_dp_bridge_atomic_disable(struct drm_bridge *bridge, struct drm_atomic_commit *state)
{
        struct cdn_dp_device *dp = bridge_to_dp(bridge);
        int ret;

        mutex_lock(&dp->lock);

        if (dp->active) {
                ret = cdn_dp_disable(dp);
                if (ret) {
                        DRM_DEV_ERROR(dp->dev, "Failed to disable bridge %d\n",
                                      ret);
                }
        }
        mutex_unlock(&dp->lock);

        /*
         * In the following 2 cases, we need to run the event_work to re-enable
         * the DP:
         * 1. If there is not just one port device is connected, and remove one
         *    device from a port, the DP will be disabled here, at this case,
         *    run the event_work to re-open DP for the other port.
         * 2. If re-training or re-config failed, the DP will be disabled here.
         *    run the event_work to re-connect it.
         */
        if (!dp->connected && cdn_dp_connected_port(dp))
                schedule_work(&dp->event_work);
}

static int cdn_dp_encoder_atomic_check(struct drm_encoder *encoder,
                                       struct drm_crtc_state *crtc_state,
                                       struct drm_connector_state *conn_state)
{
        struct rockchip_crtc_state *s = to_rockchip_crtc_state(crtc_state);

        s->output_mode = ROCKCHIP_OUT_MODE_AAAA;
        s->output_type = DRM_MODE_CONNECTOR_DisplayPort;

        return 0;
}

static const struct drm_encoder_helper_funcs cdn_dp_encoder_helper_funcs = {
        .atomic_check = cdn_dp_encoder_atomic_check,
};

static int cdn_dp_parse_dt(struct cdn_dp_device *dp)
{
        struct device *dev = dp->dev;
        struct device_node *np = dev->of_node;
        struct platform_device *pdev = to_platform_device(dev);

        dp->grf = syscon_regmap_lookup_by_phandle(np, "rockchip,grf");
        if (IS_ERR(dp->grf)) {
                DRM_DEV_ERROR(dev, "cdn-dp needs rockchip,grf property\n");
                return PTR_ERR(dp->grf);
        }

        dp->regs = devm_platform_ioremap_resource(pdev, 0);
        if (IS_ERR(dp->regs)) {
                DRM_DEV_ERROR(dev, "ioremap reg failed\n");
                return PTR_ERR(dp->regs);
        }

        dp->core_clk = devm_clk_get(dev, "core-clk");
        if (IS_ERR(dp->core_clk)) {
                DRM_DEV_ERROR(dev, "cannot get core_clk_dp\n");
                return PTR_ERR(dp->core_clk);
        }

        dp->pclk = devm_clk_get(dev, "pclk");
        if (IS_ERR(dp->pclk)) {
                DRM_DEV_ERROR(dev, "cannot get pclk\n");
                return PTR_ERR(dp->pclk);
        }

        dp->spdif_clk = devm_clk_get(dev, "spdif");
        if (IS_ERR(dp->spdif_clk)) {
                DRM_DEV_ERROR(dev, "cannot get spdif_clk\n");
                return PTR_ERR(dp->spdif_clk);
        }

        dp->grf_clk = devm_clk_get(dev, "grf");
        if (IS_ERR(dp->grf_clk)) {
                DRM_DEV_ERROR(dev, "cannot get grf clk\n");
                return PTR_ERR(dp->grf_clk);
        }

        dp->spdif_rst = devm_reset_control_get(dev, "spdif");
        if (IS_ERR(dp->spdif_rst)) {
                DRM_DEV_ERROR(dev, "no spdif reset control found\n");
                return PTR_ERR(dp->spdif_rst);
        }

        dp->dptx_rst = devm_reset_control_get(dev, "dptx");
        if (IS_ERR(dp->dptx_rst)) {
                DRM_DEV_ERROR(dev, "no uphy reset control found\n");
                return PTR_ERR(dp->dptx_rst);
        }

        dp->core_rst = devm_reset_control_get(dev, "core");
        if (IS_ERR(dp->core_rst)) {
                DRM_DEV_ERROR(dev, "no core reset control found\n");
                return PTR_ERR(dp->core_rst);
        }

        dp->apb_rst = devm_reset_control_get(dev, "apb");
        if (IS_ERR(dp->apb_rst)) {
                DRM_DEV_ERROR(dev, "no apb reset control found\n");
                return PTR_ERR(dp->apb_rst);
        }

        return 0;
}

static int cdn_dp_audio_prepare(struct drm_bridge *bridge,
                                struct drm_connector *connector,
                                struct hdmi_codec_daifmt *daifmt,
                                struct hdmi_codec_params *params)
{
        struct cdn_dp_device *dp = bridge_to_dp(bridge);
        struct audio_info audio = {
                .sample_width = params->sample_width,
                .sample_rate = params->sample_rate,
                .channels = params->channels,
        };
        int ret;

        mutex_lock(&dp->lock);
        if (!dp->active) {
                ret = -ENODEV;
                goto out;
        }

        switch (daifmt->fmt) {
        case HDMI_I2S:
                audio.format = AFMT_I2S;
                break;
        case HDMI_SPDIF:
                audio.format = AFMT_SPDIF;
                break;
        default:
                drm_err(bridge->dev, "Invalid format %d\n", daifmt->fmt);
                ret = -EINVAL;
                goto out;
        }

        ret = cdn_dp_audio_config(dp, &audio);
        if (!ret)
                dp->audio_info = audio;

out:
        mutex_unlock(&dp->lock);
        return ret;
}

static void cdn_dp_audio_shutdown(struct drm_bridge *bridge,
                                  struct drm_connector *connector)
{
        struct cdn_dp_device *dp = bridge_to_dp(bridge);
        int ret;

        mutex_lock(&dp->lock);
        if (!dp->active)
                goto out;

        ret = cdn_dp_audio_stop(dp, &dp->audio_info);
        if (!ret)
                dp->audio_info.format = AFMT_UNUSED;
out:
        mutex_unlock(&dp->lock);
}

static int cdn_dp_audio_mute_stream(struct drm_bridge *bridge,
                                    struct drm_connector *connector,
                                    bool enable, int direction)
{
        struct cdn_dp_device *dp = bridge_to_dp(bridge);
        int ret;

        mutex_lock(&dp->lock);
        if (!dp->active) {
                ret = -ENODEV;
                goto out;
        }

        ret = cdn_dp_audio_mute(dp, enable);

out:
        mutex_unlock(&dp->lock);
        return ret;
}

static const struct drm_bridge_funcs cdn_dp_bridge_funcs = {
        .atomic_duplicate_state = drm_atomic_helper_bridge_duplicate_state,
        .atomic_destroy_state = drm_atomic_helper_bridge_destroy_state,
        .atomic_reset = drm_atomic_helper_bridge_reset,
        .detect = cdn_dp_bridge_detect,
        .edid_read = cdn_dp_bridge_edid_read,
        .atomic_enable = cdn_dp_bridge_atomic_enable,
        .atomic_disable = cdn_dp_bridge_atomic_disable,
        .mode_valid = cdn_dp_bridge_mode_valid,
        .mode_set = cdn_dp_bridge_mode_set,

        .dp_audio_prepare = cdn_dp_audio_prepare,
        .dp_audio_mute_stream = cdn_dp_audio_mute_stream,
        .dp_audio_shutdown = cdn_dp_audio_shutdown,
};

static int cdn_dp_request_firmware(struct cdn_dp_device *dp)
{
        int ret;
        unsigned long timeout = jiffies + msecs_to_jiffies(CDN_FW_TIMEOUT_MS);
        unsigned long sleep = 1000;

        WARN_ON(!mutex_is_locked(&dp->lock));

        if (dp->fw_loaded)
                return 0;

        /* Drop the lock before getting the firmware to avoid blocking boot */
        mutex_unlock(&dp->lock);

        while (time_before(jiffies, timeout)) {
                ret = request_firmware(&dp->fw, CDN_DP_FIRMWARE, dp->dev);
                if (ret == -ENOENT) {
                        msleep(sleep);
                        sleep *= 2;
                        continue;
                } else if (ret) {
                        DRM_DEV_ERROR(dp->dev,
                                      "failed to request firmware: %d\n", ret);
                        goto out;
                }

                dp->fw_loaded = true;
                ret = 0;
                goto out;
        }

        DRM_DEV_ERROR(dp->dev, "Timed out trying to load firmware\n");
        ret = -ETIMEDOUT;
out:
        mutex_lock(&dp->lock);
        return ret;
}

static void cdn_dp_pd_event_work(struct work_struct *work)
{
        struct cdn_dp_device *dp = container_of(work, struct cdn_dp_device,
                                                event_work);
        int ret;

        mutex_lock(&dp->lock);

        if (dp->suspended)
                goto out;

        ret = cdn_dp_request_firmware(dp);
        if (ret)
                goto out;

        dp->connected = true;

        /* Not connected, notify userspace to disable the block */
        if (!cdn_dp_connected_port(dp)) {
                DRM_DEV_INFO(dp->dev, "Not connected; disabling cdn\n");
                dp->connected = false;

        /* Connected but not enabled, enable the block */
        } else if (!dp->active) {
                DRM_DEV_INFO(dp->dev, "Connected, not enabled; enabling cdn\n");
                ret = cdn_dp_enable(dp);
                if (ret) {
                        DRM_DEV_ERROR(dp->dev, "Enabling dp failed: %d\n", ret);
                        dp->connected = false;
                }

        /* Enabled and connected to a dongle without a sink, notify userspace */
        } else if (!cdn_dp_check_sink_connection(dp)) {
                DRM_DEV_INFO(dp->dev, "Connected without sink; assert hpd\n");
                dp->connected = false;

        /* Enabled and connected with a sink, re-train if requested */
        } else if (!cdn_dp_check_link_status(dp)) {
                unsigned int rate = dp->max_rate;
                unsigned int lanes = dp->max_lanes;
                struct drm_display_mode *mode = &dp->mode;

                DRM_DEV_INFO(dp->dev, "Connected with sink; re-train link\n");
                ret = cdn_dp_train_link(dp);
                if (ret) {
                        dp->connected = false;
                        DRM_DEV_ERROR(dp->dev, "Training link failed: %d\n", ret);
                        goto out;
                }

                /* If training result is changed, update the video config */
                if (mode->clock &&
                    (rate != dp->max_rate || lanes != dp->max_lanes)) {
                        ret = cdn_dp_config_video(dp);
                        if (ret) {
                                dp->connected = false;
                                DRM_DEV_ERROR(dp->dev, "Failed to configure video: %d\n", ret);
                        }
                }
        }

out:
        mutex_unlock(&dp->lock);
        drm_bridge_hpd_notify(&dp->bridge,
                              dp->connected ? connector_status_connected
                                            : connector_status_disconnected);
}

static int cdn_dp_pd_event(struct notifier_block *nb,
                           unsigned long event, void *priv)
{
        struct cdn_dp_port *port = container_of(nb, struct cdn_dp_port,
                                                event_nb);
        struct cdn_dp_device *dp = port->dp;

        /*
         * It would be nice to be able to just do the work inline right here.
         * However, we need to make a bunch of calls that might sleep in order
         * to turn on the block/phy, so use a worker instead.
         */
        schedule_work(&dp->event_work);

        return NOTIFY_DONE;
}

static int cdn_dp_bind(struct device *dev, struct device *master, void *data)
{
        struct cdn_dp_device *dp = dev_get_drvdata(dev);
        struct drm_encoder *encoder;
        struct drm_connector *connector;
        struct cdn_dp_port *port;
        struct drm_device *drm_dev = data;
        int ret, i;

        ret = cdn_dp_parse_dt(dp);
        if (ret < 0)
                return ret;

        dp->drm_dev = drm_dev;
        dp->connected = false;
        dp->active = false;
        dp->active_port = -1;
        dp->fw_loaded = false;

        INIT_WORK(&dp->event_work, cdn_dp_pd_event_work);

        encoder = &dp->encoder.encoder;

        encoder->possible_crtcs = drm_of_find_possible_crtcs(drm_dev,
                                                             dev->of_node);
        DRM_DEBUG_KMS("possible_crtcs = 0x%x\n", encoder->possible_crtcs);

        ret = drm_simple_encoder_init(drm_dev, encoder,
                                      DRM_MODE_ENCODER_TMDS);
        if (ret) {
                DRM_ERROR("failed to initialize encoder with drm\n");
                return ret;
        }

        drm_encoder_helper_add(encoder, &cdn_dp_encoder_helper_funcs);

        dp->bridge.ops =
                        DRM_BRIDGE_OP_DETECT |
                        DRM_BRIDGE_OP_EDID |
                        DRM_BRIDGE_OP_HPD |
                        DRM_BRIDGE_OP_DP_AUDIO;
        dp->bridge.of_node = dp->dev->of_node;
        dp->bridge.type = DRM_MODE_CONNECTOR_DisplayPort;
        dp->bridge.hdmi_audio_dev = dp->dev;
        dp->bridge.hdmi_audio_max_i2s_playback_channels = 8;
        dp->bridge.hdmi_audio_spdif_playback = 1;
        dp->bridge.hdmi_audio_dai_port = -1;

        ret = devm_drm_bridge_add(dev, &dp->bridge);
        if (ret)
                return ret;

        ret = drm_bridge_attach(encoder, &dp->bridge, NULL, DRM_BRIDGE_ATTACH_NO_CONNECTOR);
        if (ret)
                return ret;

        connector = drm_bridge_connector_init(drm_dev, encoder);
        if (IS_ERR(connector)) {
                ret = PTR_ERR(connector);
                dev_err(dp->dev, "failed to init bridge connector: %d\n", ret);
                return ret;
        }

        for (i = 0; i < dp->ports; i++) {
                port = dp->port[i];

                port->event_nb.notifier_call = cdn_dp_pd_event;
                ret = devm_extcon_register_notifier(dp->dev, port->extcon,
                                                    EXTCON_DISP_DP,
                                                    &port->event_nb);
                if (ret) {
                        DRM_DEV_ERROR(dev,
                                      "register EXTCON_DISP_DP notifier err\n");
                        return ret;
                }
        }

        pm_runtime_enable(dev);

        schedule_work(&dp->event_work);

        return 0;
}

static void cdn_dp_unbind(struct device *dev, struct device *master, void *data)
{
        struct cdn_dp_device *dp = dev_get_drvdata(dev);
        struct drm_encoder *encoder = &dp->encoder.encoder;

        cancel_work_sync(&dp->event_work);
        encoder->funcs->destroy(encoder);

        pm_runtime_disable(dev);
        if (dp->fw_loaded)
                release_firmware(dp->fw);
}

static const struct component_ops cdn_dp_component_ops = {
        .bind = cdn_dp_bind,
        .unbind = cdn_dp_unbind,
};

static int cdn_dp_suspend(struct device *dev)
{
        struct cdn_dp_device *dp = dev_get_drvdata(dev);
        int ret = 0;

        mutex_lock(&dp->lock);
        if (dp->active)
                ret = cdn_dp_disable(dp);
        dp->suspended = true;
        mutex_unlock(&dp->lock);

        return ret;
}

static __maybe_unused int cdn_dp_resume(struct device *dev)
{
        struct cdn_dp_device *dp = dev_get_drvdata(dev);

        mutex_lock(&dp->lock);
        dp->suspended = false;
        if (dp->fw_loaded)
                schedule_work(&dp->event_work);
        mutex_unlock(&dp->lock);

        return 0;
}

static int cdn_dp_probe(struct platform_device *pdev)
{
        struct device *dev = &pdev->dev;
        const struct of_device_id *match;
        struct cdn_dp_data *dp_data;
        struct cdn_dp_port *port;
        struct cdn_dp_device *dp;
        struct extcon_dev *extcon;
        struct phy *phy;
        int ret;
        int i;

        dp = devm_drm_bridge_alloc(dev, struct cdn_dp_device, bridge,
                                   &cdn_dp_bridge_funcs);
        if (IS_ERR(dp))
                return PTR_ERR(dp);
        dp->dev = dev;

        match = of_match_node(cdn_dp_dt_ids, pdev->dev.of_node);
        dp_data = (struct cdn_dp_data *)match->data;

        for (i = 0; i < dp_data->max_phy; i++) {
                extcon = extcon_get_edev_by_phandle(dev, i);
                phy = devm_of_phy_get_by_index(dev, dev->of_node, i);

                if (PTR_ERR(extcon) == -EPROBE_DEFER ||
                    PTR_ERR(phy) == -EPROBE_DEFER)
                        return -EPROBE_DEFER;

                if (IS_ERR(extcon) || IS_ERR(phy))
                        continue;

                port = devm_kzalloc(dev, sizeof(*port), GFP_KERNEL);
                if (!port)
                        return -ENOMEM;

                port->extcon = extcon;
                port->phy = phy;
                port->dp = dp;
                port->id = i;
                dp->port[dp->ports++] = port;
        }

        if (!dp->ports) {
                DRM_DEV_ERROR(dev, "missing extcon or phy\n");
                return -EINVAL;
        }

        mutex_init(&dp->lock);
        dev_set_drvdata(dev, dp);

        ret = component_add(dev, &cdn_dp_component_ops);
        if (ret)
                return ret;

        return 0;
}

static void cdn_dp_remove(struct platform_device *pdev)
{
        struct cdn_dp_device *dp = platform_get_drvdata(pdev);

        platform_device_unregister(dp->audio_pdev);
        cdn_dp_suspend(dp->dev);
        component_del(&pdev->dev, &cdn_dp_component_ops);
}

static void cdn_dp_shutdown(struct platform_device *pdev)
{
        struct cdn_dp_device *dp = platform_get_drvdata(pdev);

        cdn_dp_suspend(dp->dev);
}

static const struct dev_pm_ops cdn_dp_pm_ops = {
        SET_SYSTEM_SLEEP_PM_OPS(cdn_dp_suspend,
                                cdn_dp_resume)
};

struct platform_driver cdn_dp_driver = {
        .probe = cdn_dp_probe,
        .remove = cdn_dp_remove,
        .shutdown = cdn_dp_shutdown,
        .driver = {
                   .name = "cdn-dp",
                   .of_match_table = cdn_dp_dt_ids,
                   .pm = &cdn_dp_pm_ops,
        },
};