// SPDX-License-Identifier: GPL-2.0-only
/*
 * UFS Host driver for Synopsys Designware Core
 *
 * Copyright (C) 2015-2016 Synopsys, Inc. (www.synopsys.com)
 *
 * Authors: Joao Pinto <jpinto@synopsys.com>
 */

#include <linux/module.h>

#include <ufs/ufshcd.h>
#include <ufs/unipro.h>

#include "ufshcd-dwc.h"
#include "ufshci-dwc.h"

int ufshcd_dwc_dme_set_attrs(struct ufs_hba *hba,
    const struct ufshcd_dme_attr_val *v, int n)
{
 int ret = 0;
 int attr_node = 0;

 for (attr_node = 0; attr_node < n; attr_node++) {
  ret = ufshcd_dme_set_attr(hba, v[attr_node].attr_sel,
   ATTR_SET_NOR, v[attr_node].mib_val, v[attr_node].peer);

  if (ret)
   return ret;
 }

 return 0;
}
EXPORT_SYMBOL(ufshcd_dwc_dme_set_attrs);

/**
 * ufshcd_dwc_program_clk_div() - program clock divider.
 * @hba: Private Structure pointer
 * @divider_val: clock divider value to be programmed
 *
 */
static void ufshcd_dwc_program_clk_div(struct ufs_hba *hba, u32 divider_val)
{
 ufshcd_writel(hba, divider_val, DWC_UFS_REG_HCLKDIV);
}

/**
 * ufshcd_dwc_link_is_up() - check if link is up.
 * @hba: private structure pointer
 *
 * Return: 0 on success, non-zero value on failure.
 */
static int ufshcd_dwc_link_is_up(struct ufs_hba *hba)
{
 int dme_result = 0;

 ufshcd_dme_get(hba, UIC_ARG_MIB(VS_POWERSTATE), &dme_result);

 if (dme_result == UFSHCD_LINK_IS_UP) {
  ufshcd_set_link_active(hba);
  return 0;
 }

 return 1;
}

/**
 * ufshcd_dwc_connection_setup() - configure unipro attributes.
 * @hba: pointer to drivers private data
 *
 * This function configures both the local side (host) and the peer side
 * (device) unipro attributes to establish the connection to application/
 * cport.
 * This function is not required if the hardware is properly configured to
 * have this connection setup on reset. But invoking this function does no
 * harm and should be fine even working with any ufs device.
 *
 * Return: 0 on success non-zero value on failure.
 */
static int ufshcd_dwc_connection_setup(struct ufs_hba *hba)
{
 static const struct ufshcd_dme_attr_val setup_attrs[] = {
  { UIC_ARG_MIB(T_CONNECTIONSTATE), 0, DME_LOCAL },
  { UIC_ARG_MIB(N_DEVICEID), 0, DME_LOCAL },
  { UIC_ARG_MIB(N_DEVICEID_VALID), 0, DME_LOCAL },
  { UIC_ARG_MIB(T_PEERDEVICEID), 1, DME_LOCAL },
  { UIC_ARG_MIB(T_PEERCPORTID), 0, DME_LOCAL },
  { UIC_ARG_MIB(T_TRAFFICCLASS), 0, DME_LOCAL },
  { UIC_ARG_MIB(T_CPORTFLAGS), 0x6, DME_LOCAL },
  { UIC_ARG_MIB(T_CPORTMODE), 1, DME_LOCAL },
  { UIC_ARG_MIB(T_CONNECTIONSTATE), 1, DME_LOCAL },
  { UIC_ARG_MIB(T_CONNECTIONSTATE), 0, DME_PEER },
  { UIC_ARG_MIB(N_DEVICEID), 1, DME_PEER },
  { UIC_ARG_MIB(N_DEVICEID_VALID), 1, DME_PEER },
  { UIC_ARG_MIB(T_PEERDEVICEID), 1, DME_PEER },
  { UIC_ARG_MIB(T_PEERCPORTID), 0, DME_PEER },
  { UIC_ARG_MIB(T_TRAFFICCLASS), 0, DME_PEER },
  { UIC_ARG_MIB(T_CPORTFLAGS), 0x6, DME_PEER },
  { UIC_ARG_MIB(T_CPORTMODE), 1, DME_PEER },
  { UIC_ARG_MIB(T_CONNECTIONSTATE), 1, DME_PEER }
 };

 return ufshcd_dwc_dme_set_attrs(hba, setup_attrs, ARRAY_SIZE(setup_attrs));
}

/**
 * ufshcd_dwc_link_startup_notify() - program clock divider.
 * @hba: private structure pointer
 * @status: Callback notify status
 *
 * Return: 0 on success, non-zero value on failure.
 */
int ufshcd_dwc_link_startup_notify(struct ufs_hba *hba,
     enum ufs_notify_change_status status)
{
 int err = 0;

 if (status == PRE_CHANGE) {
  ufshcd_dwc_program_clk_div(hba, DWC_UFS_REG_HCLKDIV_DIV_125);

  err = ufshcd_vops_phy_initialization(hba);
  if (err) {
   dev_err(hba->dev, "Phy setup failed (%d)\n", err);
   goto out;
  }
 } else { /* POST_CHANGE */
  err = ufshcd_dwc_link_is_up(hba);
  if (err) {
   dev_err(hba->dev, "Link is not up\n");
   goto out;
  }

  err = ufshcd_dwc_connection_setup(hba);
  if (err)
   dev_err(hba->dev, "Connection setup failed (%d)\n",
         err);
 }

out:
 return err;
}
EXPORT_SYMBOL(ufshcd_dwc_link_startup_notify);

MODULE_AUTHOR("Joao Pinto <Joao.Pinto@synopsys.com>");
MODULE_DESCRIPTION("UFS Host driver for Synopsys Designware Core");
MODULE_LICENSE("Dual BSD/GPL");