// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (C) 2023 Linaro Ltd.
 *
 * Author: Dmitry Baryshkov <dmitry.baryshkov@linaro.org>
 */
#include <linux/auxiliary_bus.h>
#include <linux/export.h>
#include <linux/module.h>
#include <linux/of.h>

#include <drm/drm_bridge.h>
#include <drm/bridge/aux-bridge.h>

static DEFINE_IDA(drm_aux_hpd_bridge_ida);

struct drm_aux_hpd_bridge_data {
 struct drm_bridge bridge;
 struct device *dev;
};

static void drm_aux_hpd_bridge_release(struct device *dev)
{
 struct auxiliary_device *adev = to_auxiliary_dev(dev);

 ida_free(&drm_aux_hpd_bridge_ida, adev->id);

 of_node_put(adev->dev.platform_data);
 of_node_put(adev->dev.of_node);

 kfree(adev);
}

static void drm_aux_hpd_bridge_free_adev(void *_adev)
{
 auxiliary_device_uninit(_adev);
}

/**
 * devm_drm_dp_hpd_bridge_alloc - allocate a HPD DisplayPort bridge
 * @parent: device instance providing this bridge
 * @np: device node pointer corresponding to this bridge instance
 *
 * Creates a simple DRM bridge with the type set to
 * DRM_MODE_CONNECTOR_DisplayPort, which terminates the bridge chain and is
 * able to send the HPD events.
 *
 * Return: bridge auxiliary device pointer or an error pointer
 */
struct auxiliary_device *devm_drm_dp_hpd_bridge_alloc(struct device *parent, struct device_node *np)
{
 struct auxiliary_device *adev;
 int ret;

 adev = kzalloc(sizeof(*adev), GFP_KERNEL);
 if (!adev)
  return ERR_PTR(-ENOMEM);

 ret = ida_alloc(&drm_aux_hpd_bridge_ida, GFP_KERNEL);
 if (ret < 0) {
  kfree(adev);
  return ERR_PTR(ret);
 }

 adev->id = ret;
 adev->name = "dp_hpd_bridge";
 adev->dev.parent = parent;
 adev->dev.release = drm_aux_hpd_bridge_release;
 adev->dev.platform_data = of_node_get(np);

 device_set_of_node_from_dev(&adev->dev, parent);

 ret = auxiliary_device_init(adev);
 if (ret) {
  of_node_put(adev->dev.platform_data);
  of_node_put(adev->dev.of_node);
  ida_free(&drm_aux_hpd_bridge_ida, adev->id);
  kfree(adev);
  return ERR_PTR(ret);
 }

 ret = devm_add_action_or_reset(parent, drm_aux_hpd_bridge_free_adev, adev);
 if (ret)
  return ERR_PTR(ret);

 return adev;
}
EXPORT_SYMBOL_GPL(devm_drm_dp_hpd_bridge_alloc);

static void drm_aux_hpd_bridge_del_adev(void *_adev)
{
 auxiliary_device_delete(_adev);
}

/**
 * devm_drm_dp_hpd_bridge_add - register a HDP DisplayPort bridge
 * @dev: struct device to tie registration lifetime to
 * @adev: bridge auxiliary device to be registered
 *
 * Returns: zero on success or a negative errno
 */
int devm_drm_dp_hpd_bridge_add(struct device *dev, struct auxiliary_device *adev)
{
 int ret;

 ret = auxiliary_device_add(adev);
 if (ret)
  return ret;

 return devm_add_action_or_reset(dev, drm_aux_hpd_bridge_del_adev, adev);
}
EXPORT_SYMBOL_GPL(devm_drm_dp_hpd_bridge_add);

/**
 * drm_dp_hpd_bridge_register - allocate and register a HDP DisplayPort bridge
 * @parent: device instance providing this bridge
 * @np: device node pointer corresponding to this bridge instance
 *
 * Return: device instance that will handle created bridge or an error pointer
 */
struct device *drm_dp_hpd_bridge_register(struct device *parent, struct device_node *np)
{
 struct auxiliary_device *adev;
 int ret;

 adev = devm_drm_dp_hpd_bridge_alloc(parent, np);
 if (IS_ERR(adev))
  return ERR_CAST(adev);

 ret = devm_drm_dp_hpd_bridge_add(parent, adev);
 if (ret)
  return ERR_PTR(ret);

 return &adev->dev;
}
EXPORT_SYMBOL_GPL(drm_dp_hpd_bridge_register);

/**
 * drm_aux_hpd_bridge_notify - notify hot plug detection events
 * @dev: device created for the HPD bridge
 * @status: output connection status
 *
 * A wrapper around drm_bridge_hpd_notify() that is used to report hot plug
 * detection events for bridges created via drm_dp_hpd_bridge_register().
 *
 * This function shall be called in a context that can sleep.
 */
void drm_aux_hpd_bridge_notify(struct device *dev, enum drm_connector_status status)
{
 struct auxiliary_device *adev = to_auxiliary_dev(dev);
 struct drm_aux_hpd_bridge_data *data = auxiliary_get_drvdata(adev);

 if (!data)
  return;

 drm_bridge_hpd_notify(&data->bridge, status);
}
EXPORT_SYMBOL_GPL(drm_aux_hpd_bridge_notify);

static int drm_aux_hpd_bridge_attach(struct drm_bridge *bridge,
         struct drm_encoder *encoder,
         enum drm_bridge_attach_flags flags)
{
 return flags & DRM_BRIDGE_ATTACH_NO_CONNECTOR ? 0 : -EINVAL;
}

static const struct drm_bridge_funcs drm_aux_hpd_bridge_funcs = {
 .attach = drm_aux_hpd_bridge_attach,
};

static int drm_aux_hpd_bridge_probe(struct auxiliary_device *auxdev,
        const struct auxiliary_device_id *id)
{
 struct drm_aux_hpd_bridge_data *data;

 data = devm_drm_bridge_alloc(&auxdev->dev,
         struct drm_aux_hpd_bridge_data, bridge,
         &drm_aux_hpd_bridge_funcs);
 if (IS_ERR(data))
  return PTR_ERR(data);

 data->dev = &auxdev->dev;
 data->bridge.of_node = dev_get_platdata(data->dev);
 data->bridge.ops = DRM_BRIDGE_OP_HPD;
 data->bridge.type = id->driver_data;

 /* passthrough data, allow everything */
 data->bridge.interlace_allowed = true;
 data->bridge.ycbcr_420_allowed = true;

 auxiliary_set_drvdata(auxdev, data);

 return devm_drm_bridge_add(data->dev, &data->bridge);
}

static const struct auxiliary_device_id drm_aux_hpd_bridge_table[] = {
 { .name = KBUILD_MODNAME ".dp_hpd_bridge", .driver_data = DRM_MODE_CONNECTOR_DisplayPort, },
 {},
};
MODULE_DEVICE_TABLE(auxiliary, drm_aux_hpd_bridge_table);

static struct auxiliary_driver drm_aux_hpd_bridge_drv = {
 .name = "aux_hpd_bridge",
 .id_table = drm_aux_hpd_bridge_table,
 .probe = drm_aux_hpd_bridge_probe,
};
module_auxiliary_driver(drm_aux_hpd_bridge_drv);

MODULE_AUTHOR("Dmitry Baryshkov <dmitry.baryshkov@linaro.org>");
MODULE_DESCRIPTION("DRM HPD bridge");
MODULE_LICENSE("GPL");