// SPDX-License-Identifier: GPL-2.0+
/*
 * PCI Hot Plug Controller Driver for RPA-compliant PPC64 platform.
 * Copyright (C) 2003 Linda Xie <lxie@us.ibm.com>
 *
 * All rights reserved.
 *
 * Send feedback to <lxie@us.ibm.com>
 *
 */
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/of.h>
#include <linux/pci.h>
#include <linux/pci_hotplug.h>
#include <linux/smp.h>
#include <linux/init.h>
#include <linux/vmalloc.h>
#include <asm/firmware.h>
#include <asm/eeh.h>       /* for eeh_add_device() */
#include <asm/rtas.h>  /* rtas_call */
#include <asm/pci-bridge.h> /* for pci_controller */
#include <asm/prom.h>
#include "../pci.h"  /* for pci_add_new_bus */
    /* and pci_do_scan_bus */
#include "rpaphp.h"

bool rpaphp_debug;
LIST_HEAD(rpaphp_slot_head);
EXPORT_SYMBOL_GPL(rpaphp_slot_head);

#define DRIVER_VERSION "0.1"
#define DRIVER_AUTHOR "Linda Xie <lxie@us.ibm.com>"
#define DRIVER_DESC "RPA HOT Plug PCI Controller Driver"

#define MAX_LOC_CODE 128

MODULE_AUTHOR(DRIVER_AUTHOR);
MODULE_DESCRIPTION(DRIVER_DESC);
MODULE_LICENSE("GPL");

module_param_named(debug, rpaphp_debug, bool, 0644);

/**
 * set_attention_status - set attention LED
 * @hotplug_slot: target &hotplug_slot
 * @value: LED control value
 *
 * echo 0 > attention -- set LED OFF
 * echo 1 > attention -- set LED ON
 * echo 2 > attention -- set LED ID(identify, light is blinking)
 */
static int set_attention_status(struct hotplug_slot *hotplug_slot, u8 value)
{
 int rc;
 struct slot *slot = to_slot(hotplug_slot);

 switch (value) {
 case 0:
 case 1:
 case 2:
  break;
 default:
  value = 1;
  break;
 }

 rc = rtas_set_indicator(DR_INDICATOR, slot->index, value);
 if (!rc)
  slot->attention_status = value;

 return rc;
}

/**
 * get_power_status - get power status of a slot
 * @hotplug_slot: slot to get status
 * @value: pointer to store status
 */
static int get_power_status(struct hotplug_slot *hotplug_slot, u8 *value)
{
 int retval, level;
 struct slot *slot = to_slot(hotplug_slot);

 retval = rtas_get_power_level(slot->power_domain, &level);
 if (!retval)
  *value = level;
 return retval;
}

/**
 * get_attention_status - get attention LED status
 * @hotplug_slot: slot to get status
 * @value: pointer to store status
 */
static int get_attention_status(struct hotplug_slot *hotplug_slot, u8 *value)
{
 struct slot *slot = to_slot(hotplug_slot);
 *value = slot->attention_status;
 return 0;
}

static int get_adapter_status(struct hotplug_slot *hotplug_slot, u8 *value)
{
 struct slot *slot = to_slot(hotplug_slot);
 int rc, state;

 rc = rpaphp_get_sensor_state(slot, &state);

 *value = NOT_VALID;
 if (rc)
  return rc;

 if (state == EMPTY)
  *value = EMPTY;
 else if (state == PRESENT)
  *value = slot->state;

 return 0;
}

static enum pci_bus_speed get_max_bus_speed(struct slot *slot)
{
 enum pci_bus_speed speed;
 switch (slot->type) {
 case 1:
 case 2:
 case 3:
 case 4:
 case 5:
 case 6:
  speed = PCI_SPEED_33MHz; /* speed for case 1-6 */
  break;
 case 7:
 case 8:
  speed = PCI_SPEED_66MHz;
  break;
 case 11:
 case 14:
  speed = PCI_SPEED_66MHz_PCIX;
  break;
 case 12:
 case 15:
  speed = PCI_SPEED_100MHz_PCIX;
  break;
 case 13:
 case 16:
  speed = PCI_SPEED_133MHz_PCIX;
  break;
 default:
  speed = PCI_SPEED_UNKNOWN;
  break;
 }

 return speed;
}

static int get_children_props(struct device_node *dn, const __be32 **drc_indexes,
         const __be32 **drc_names, const __be32 **drc_types,
         const __be32 **drc_power_domains)
{
 const __be32 *indexes, *names, *types, *domains;

 indexes = of_get_property(dn, "ibm,drc-indexes", NULL);
 names = of_get_property(dn, "ibm,drc-names", NULL);
 types = of_get_property(dn, "ibm,drc-types", NULL);
 domains = of_get_property(dn, "ibm,drc-power-domains", NULL);

 if (!indexes || !names || !types || !domains) {
  /* Slot does not have dynamically-removable children */
  return -EINVAL;
 }
 if (drc_indexes)
  *drc_indexes = indexes;
 if (drc_names)
  /* &drc_names[1] contains NULL terminated slot names */
  *drc_names = names;
 if (drc_types)
  /* &drc_types[1] contains NULL terminated slot types */
  *drc_types = types;
 if (drc_power_domains)
  *drc_power_domains = domains;

 return 0;
}


/* Verify the existence of 'drc_name' and/or 'drc_type' within the
 * current node.  First obtain its my-drc-index property.  Next,
 * obtain the DRC info from its parent.  Use the my-drc-index for
 * correlation, and obtain/validate the requested properties.
 */

static int rpaphp_check_drc_props_v1(struct device_node *dn, char *drc_name,
    char *drc_type, unsigned int my_index)
{
 char *name_tmp, *type_tmp;
 const __be32 *indexes, *names;
 const __be32 *types, *domains;
 int i, rc;

 rc = get_children_props(dn->parent, &indexes, &names, &types, &domains);
 if (rc < 0) {
  return -EINVAL;
 }

 name_tmp = (char *) &names[1];
 type_tmp = (char *) &types[1];

 /* Iterate through parent properties, looking for my-drc-index */
 for (i = 0; i < be32_to_cpu(indexes[0]); i++) {
  if (be32_to_cpu(indexes[i + 1]) == my_index)
   break;

  name_tmp += (strlen(name_tmp) + 1);
  type_tmp += (strlen(type_tmp) + 1);
 }

 if (((drc_name == NULL) || (drc_name && !strcmp(drc_name, name_tmp))) &&
     ((drc_type == NULL) || (drc_type && !strcmp(drc_type, type_tmp))))
  return 0;

 return -EINVAL;
}

static int rpaphp_check_drc_props_v2(struct device_node *dn, char *drc_name,
    char *drc_type, unsigned int my_index)
{
 struct property *info;
 unsigned int entries;
 struct of_drc_info drc;
 const __be32 *value;
 char cell_drc_name[MAX_DRC_NAME_LEN];
 int j;

 info = of_find_property(dn->parent, "ibm,drc-info", NULL);
 if (info == NULL)
  return -EINVAL;

 value = of_prop_next_u32(info, NULL, &entries);
 if (!value)
  return -EINVAL;
 else
  value++;

 for (j = 0; j < entries; j++) {
  of_read_drc_info_cell(&info, &value, &drc);

  /* Should now know end of current entry */

  /* Found it */
  if (my_index >= drc.drc_index_start && my_index <= drc.last_drc_index) {
   int index = my_index - drc.drc_index_start;
   sprintf(cell_drc_name, "%s%d", drc.drc_name_prefix,
    drc.drc_name_suffix_start + index);
   break;
  }
 }

 if (((drc_name == NULL) ||
      (drc_name && !strcmp(drc_name, cell_drc_name))) &&
     ((drc_type == NULL) ||
      (drc_type && !strcmp(drc_type, drc.drc_type))))
  return 0;

 return -EINVAL;
}

int rpaphp_check_drc_props(struct device_node *dn, char *drc_name,
   char *drc_type)
{
 const __be32 *my_index;

 my_index = of_get_property(dn, "ibm,my-drc-index", NULL);
 if (!my_index) {
  /* Node isn't DLPAR/hotplug capable */
  return -EINVAL;
 }

 if (of_property_present(dn->parent, "ibm,drc-info"))
  return rpaphp_check_drc_props_v2(dn, drc_name, drc_type,
      be32_to_cpu(*my_index));
 else
  return rpaphp_check_drc_props_v1(dn, drc_name, drc_type,
      be32_to_cpu(*my_index));
}
EXPORT_SYMBOL_GPL(rpaphp_check_drc_props);


static int is_php_type(char *drc_type)
{
 char *endptr;

 /* PCI Hotplug nodes have an integer for drc_type */
 simple_strtoul(drc_type, &endptr, 10);
 if (endptr == drc_type)
  return 0;

 return 1;
}

/**
 * is_php_dn() - return 1 if this is a hotpluggable pci slot, else 0
 * @dn: target &device_node
 * @indexes: passed to get_children_props()
 * @names: passed to get_children_props()
 * @types: returned from get_children_props()
 * @power_domains:
 *
 * This routine will return true only if the device node is
 * a hotpluggable slot. This routine will return false
 * for built-in pci slots (even when the built-in slots are
 * dlparable.)
 */
static int is_php_dn(struct device_node *dn, const __be32 **indexes,
       const __be32 **names, const __be32 **types,
       const __be32 **power_domains)
{
 const __be32 *drc_types;
 int rc;

 rc = get_children_props(dn, indexes, names, &drc_types, power_domains);
 if (rc < 0)
  return 0;

 if (!is_php_type((char *) &drc_types[1]))
  return 0;

 *types = drc_types;
 return 1;
}

static int rpaphp_drc_info_add_slot(struct device_node *dn)
{
 struct slot *slot;
 struct property *info;
 struct of_drc_info drc;
 char drc_name[MAX_DRC_NAME_LEN];
 const __be32 *cur;
 u32 count;
 int retval = 0;

 info = of_find_property(dn, "ibm,drc-info", NULL);
 if (!info)
  return 0;

 cur = of_prop_next_u32(info, NULL, &count);
 if (cur)
  cur++;
 else
  return 0;

 of_read_drc_info_cell(&info, &cur, &drc);
 if (!is_php_type(drc.drc_type))
  return 0;

 sprintf(drc_name, "%s%d", drc.drc_name_prefix, drc.drc_name_suffix_start);

 slot = alloc_slot_struct(dn, drc.drc_index_start, drc_name, drc.drc_power_domain);
 if (!slot)
  return -ENOMEM;

 slot->type = simple_strtoul(drc.drc_type, NULL, 10);
 retval = rpaphp_enable_slot(slot);
 if (!retval)
  retval = rpaphp_register_slot(slot);

 if (retval)
  dealloc_slot_struct(slot);

 return retval;
}

static int rpaphp_drc_add_slot(struct device_node *dn)
{
 struct slot *slot;
 int retval = 0;
 int i;
 const __be32 *indexes, *names, *types, *power_domains;
 char *name, *type;

 /* If this is not a hotplug slot, return without doing anything. */
 if (!is_php_dn(dn, &indexes, &names, &types, &power_domains))
  return 0;

 dbg("Entry %s: dn=%pOF\n", __func__, dn);

 /* register PCI devices */
 name = (char *) &names[1];
 type = (char *) &types[1];
 for (i = 0; i < be32_to_cpu(indexes[0]); i++) {
  int index;

  index = be32_to_cpu(indexes[i + 1]);
  slot = alloc_slot_struct(dn, index, name,
      be32_to_cpu(power_domains[i + 1]));
  if (!slot)
   return -ENOMEM;

  slot->type = simple_strtoul(type, NULL, 10);

  dbg("Found drc-index:0x%x drc-name:%s drc-type:%s\n",
    index, name, type);

  retval = rpaphp_enable_slot(slot);
  if (!retval)
   retval = rpaphp_register_slot(slot);

  if (retval)
   dealloc_slot_struct(slot);

  name += strlen(name) + 1;
  type += strlen(type) + 1;
 }
 dbg("%s - Exit: rc[%d]\n", __func__, retval);

 /* XXX FIXME: reports a failure only if last entry in loop failed */
 return retval;
}

/**
 * rpaphp_add_slot -- declare a hotplug slot to the hotplug subsystem.
 * @dn: device node of slot
 *
 * This subroutine will register a hotpluggable slot with the
 * PCI hotplug infrastructure. This routine is typically called
 * during boot time, if the hotplug slots are present at boot time,
 * or is called later, by the dlpar add code, if the slot is
 * being dynamically added during runtime.
 *
 * If the device node points at an embedded (built-in) slot, this
 * routine will just return without doing anything, since embedded
 * slots cannot be hotplugged.
 *
 * To remove a slot, it suffices to call rpaphp_deregister_slot().
 */
int rpaphp_add_slot(struct device_node *dn)
{
 if (!of_node_name_eq(dn, "pci"))
  return 0;

 if (of_property_present(dn, "ibm,drc-info"))
  return rpaphp_drc_info_add_slot(dn);
 else
  return rpaphp_drc_add_slot(dn);
}
EXPORT_SYMBOL_GPL(rpaphp_add_slot);

static void __exit cleanup_slots(void)
{
 struct slot *slot, *next;

 /*
  * Unregister all of our slots with the pci_hotplug subsystem,
  * and free up all memory that we had allocated.
 */

 list_for_each_entry_safe(slot, next, &rpaphp_slot_head,
     rpaphp_slot_list) {
  list_del(&slot->rpaphp_slot_list);
  pci_hp_deregister(&slot->hotplug_slot);
  dealloc_slot_struct(slot);
 }
}

static int __init rpaphp_init(void)
{
 struct device_node *dn;

 info(DRIVER_DESC " version: " DRIVER_VERSION "\n");

 for_each_node_by_name(dn, "pci")
  rpaphp_add_slot(dn);

 return 0;
}

static void __exit rpaphp_exit(void)
{
 cleanup_slots();
}

static int enable_slot(struct hotplug_slot *hotplug_slot)
{
 struct slot *slot = to_slot(hotplug_slot);
 int state;
 int retval;

 if (slot->state == CONFIGURED)
  return 0;

 retval = rpaphp_get_sensor_state(slot, &state);
 if (retval)
  return retval;

 if (state == PRESENT) {
  pseries_eeh_init_edev_recursive(PCI_DN(slot->dn));

  pci_lock_rescan_remove();
  pci_hp_add_devices(slot->bus);
  pci_unlock_rescan_remove();
  slot->state = CONFIGURED;
 } else if (state == EMPTY) {
  slot->state = EMPTY;
 } else {
  err("%s: slot[%s] is in invalid state\n", __func__, slot->name);
  slot->state = NOT_VALID;
  return -EINVAL;
 }

 slot->bus->max_bus_speed = get_max_bus_speed(slot);
 return 0;
}

static int disable_slot(struct hotplug_slot *hotplug_slot)
{
 struct slot *slot = to_slot(hotplug_slot);
 if (slot->state == NOT_CONFIGURED)
  return -EINVAL;

 pci_lock_rescan_remove();
 pci_hp_remove_devices(slot->bus);
 pci_unlock_rescan_remove();
 vm_unmap_aliases();

 slot->state = NOT_CONFIGURED;
 return 0;
}

const struct hotplug_slot_ops rpaphp_hotplug_slot_ops = {
 .enable_slot = enable_slot,
 .disable_slot = disable_slot,
 .set_attention_status = set_attention_status,
 .get_power_status = get_power_status,
 .get_attention_status = get_attention_status,
 .get_adapter_status = get_adapter_status,
};

module_init(rpaphp_init);
module_exit(rpaphp_exit);