// SPDX-License-Identifier: GPL-2.0-only
/*
 * Supports for the button array on SoC tablets originally running
 * Windows 8.
 *
 * (C) Copyright 2014 Intel Corporation
 */

#include <linux/module.h>
#include <linux/input.h>
#include <linux/init.h>
#include <linux/irq.h>
#include <linux/kernel.h>
#include <linux/acpi.h>
#include <linux/dmi.h>
#include <linux/gpio/consumer.h>
#include <linux/gpio_keys.h>
#include <linux/gpio.h>
#include <linux/platform_device.h>

static bool use_low_level_irq;
module_param(use_low_level_irq, bool, 0444);
MODULE_PARM_DESC(use_low_level_irq, "Use low-level triggered IRQ instead of edge triggered");

struct soc_button_info {
 const char *name;
 int acpi_index;
 unsigned int event_type;
 unsigned int event_code;
 bool autorepeat;
 bool wakeup;
 bool active_low;
};

struct soc_device_data {
 const struct soc_button_info *button_info;
 int (*check)(struct device *dev);
};

/*
 * Some of the buttons like volume up/down are auto repeat, while others
 * are not. To support both, we register two platform devices, and put
 * buttons into them based on whether the key should be auto repeat.
 */
#define BUTTON_TYPES 2

struct soc_button_data {
 struct platform_device *children[BUTTON_TYPES];
};

/*
 * Some 2-in-1s which use the soc_button_array driver have this ugly issue in
 * their DSDT where the _LID method modifies the irq-type settings of the GPIOs
 * used for the power and home buttons. The intend of this AML code is to
 * disable these buttons when the lid is closed.
 * The AML does this by directly poking the GPIO controllers registers. This is
 * problematic because when re-enabling the irq, which happens whenever _LID
 * gets called with the lid open (e.g. on boot and on resume), it sets the
 * irq-type to IRQ_TYPE_LEVEL_LOW. Where as the gpio-keys driver programs the
 * type to, and expects it to be, IRQ_TYPE_EDGE_BOTH.
 * To work around this we don't set gpio_keys_button.gpio on these 2-in-1s,
 * instead we get the irq for the GPIO ourselves, configure it as
 * IRQ_TYPE_LEVEL_LOW (to match how the _LID AML code configures it) and pass
 * the irq in gpio_keys_button.irq. Below is a list of affected devices.
 */
static const struct dmi_system_id dmi_use_low_level_irq[] = {
 {
  /*
   * Acer Switch 10 SW5-012. _LID method messes with home- and
   * power-button GPIO IRQ settings. When (re-)enabling the irq
   * it ors in its own flags without clearing the previous set
   * ones, leading to an irq-type of IRQ_TYPE_LEVEL_LOW |
   * IRQ_TYPE_LEVEL_HIGH causing a continuous interrupt storm.
 */
  .matches = {
   DMI_MATCH(DMI_SYS_VENDOR, "Acer"),
   DMI_MATCH(DMI_PRODUCT_NAME, "Aspire SW5-012"),
  },
 },
 {
  /* Acer Switch V 10 SW5-017, same issue as Acer Switch 10 SW5-012. */
  .matches = {
   DMI_MATCH(DMI_SYS_VENDOR, "Acer"),
   DMI_MATCH(DMI_PRODUCT_NAME, "SW5-017"),
  },
 },
 {
  /*
   * Acer One S1003. _LID method messes with power-button GPIO
   * IRQ settings, leading to a non working power-button.
 */
  .matches = {
   DMI_MATCH(DMI_SYS_VENDOR, "Acer"),
   DMI_MATCH(DMI_PRODUCT_NAME, "One S1003"),
  },
 },
 {
  /*
   * Lenovo Yoga Tab2 1051F/1051L, something messes with the home-button
   * IRQ settings, leading to a non working home-button.
 */
  .matches = {
   DMI_MATCH(DMI_SYS_VENDOR, "LENOVO"),
   DMI_MATCH(DMI_PRODUCT_NAME, "60073"),
   DMI_MATCH(DMI_PRODUCT_VERSION, "1051"),
  },
 },
 {} /* Terminating entry */
};

/*
 * Some devices have a wrong entry which points to a GPIO which is
 * required in another driver, so this driver must not claim it.
 */
static const struct dmi_system_id dmi_invalid_acpi_index[] = {
 {
  /*
   * Lenovo Yoga Book X90F / X90L, the PNP0C40 home button entry
   * points to a GPIO which is not a home button and which is
   * required by the lenovo-yogabook driver.
 */
  .matches = {
   DMI_EXACT_MATCH(DMI_SYS_VENDOR, "Intel Corporation"),
   DMI_EXACT_MATCH(DMI_PRODUCT_NAME, "CHERRYVIEW D1 PLATFORM"),
   DMI_EXACT_MATCH(DMI_PRODUCT_VERSION, "YETI-11"),
  },
  .driver_data = (void *)1l,
 },
 {} /* Terminating entry */
};

/*
 * Get the Nth GPIO number from the ACPI object.
 */
static int soc_button_lookup_gpio(struct device *dev, int acpi_index,
      int *gpio_ret, int *irq_ret)
{
 struct gpio_desc *desc;

 desc = gpiod_get_index(dev, NULL, acpi_index, GPIOD_ASIS);
 if (IS_ERR(desc))
  return PTR_ERR(desc);

 *gpio_ret = desc_to_gpio(desc);
 *irq_ret = gpiod_to_irq(desc);

 gpiod_put(desc);

 return 0;
}

static struct platform_device *
soc_button_device_create(struct platform_device *pdev,
    const struct soc_button_info *button_info,
    bool autorepeat)
{
 const struct soc_button_info *info;
 struct platform_device *pd;
 struct gpio_keys_button *gpio_keys;
 struct gpio_keys_platform_data *gpio_keys_pdata;
 const struct dmi_system_id *dmi_id;
 int invalid_acpi_index = -1;
 int error, gpio, irq;
 int n_buttons = 0;

 for (info = button_info; info->name; info++)
  if (info->autorepeat == autorepeat)
   n_buttons++;

 gpio_keys_pdata = devm_kzalloc(&pdev->dev,
           sizeof(*gpio_keys_pdata) +
     sizeof(*gpio_keys) * n_buttons,
           GFP_KERNEL);
 if (!gpio_keys_pdata)
  return ERR_PTR(-ENOMEM);

 gpio_keys = (void *)(gpio_keys_pdata + 1);
 n_buttons = 0;

 dmi_id = dmi_first_match(dmi_invalid_acpi_index);
 if (dmi_id)
  invalid_acpi_index = (long)dmi_id->driver_data;

 for (info = button_info; info->name; info++) {
  if (info->autorepeat != autorepeat)
   continue;

  if (info->acpi_index == invalid_acpi_index)
   continue;

  error = soc_button_lookup_gpio(&pdev->dev, info->acpi_index, &gpio, &irq);
  if (error || irq < 0) {
   /*
    * Skip GPIO if not present. Note we deliberately
    * ignore -EPROBE_DEFER errors here. On some devices
    * Intel is using so called virtual GPIOs which are not
    * GPIOs at all but some way for AML code to check some
    * random status bits without need a custom opregion.
    * In some cases the resources table we parse points to
    * such a virtual GPIO, since these are not real GPIOs
    * we do not have a driver for these so they will never
    * show up, therefore we ignore -EPROBE_DEFER.
 */
   continue;
  }

  /* See dmi_use_low_level_irq[] comment */
  if (!autorepeat && (use_low_level_irq ||
        dmi_check_system(dmi_use_low_level_irq))) {
   irq_set_irq_type(irq, IRQ_TYPE_LEVEL_LOW);
   gpio_keys[n_buttons].irq = irq;
   gpio_keys[n_buttons].gpio = -ENOENT;
  } else {
   gpio_keys[n_buttons].gpio = gpio;
  }

  gpio_keys[n_buttons].type = info->event_type;
  gpio_keys[n_buttons].code = info->event_code;
  gpio_keys[n_buttons].active_low = info->active_low;
  gpio_keys[n_buttons].desc = info->name;
  gpio_keys[n_buttons].wakeup = info->wakeup;
  /* These devices often use cheap buttons, use 50 ms debounce */
  gpio_keys[n_buttons].debounce_interval = 50;
  n_buttons++;
 }

 if (n_buttons == 0) {
  error = -ENODEV;
  goto err_free_mem;
 }

 gpio_keys_pdata->buttons = gpio_keys;
 gpio_keys_pdata->nbuttons = n_buttons;
 gpio_keys_pdata->rep = autorepeat;

 pd = platform_device_register_resndata(&pdev->dev, "gpio-keys",
            PLATFORM_DEVID_AUTO, NULL, 0,
            gpio_keys_pdata,
            sizeof(*gpio_keys_pdata));
 error = PTR_ERR_OR_ZERO(pd);
 if (error) {
  dev_err(&pdev->dev,
   "failed registering gpio-keys: %d\n", error);
  goto err_free_mem;
 }

 return pd;

err_free_mem:
 devm_kfree(&pdev->dev, gpio_keys_pdata);
 return ERR_PTR(error);
}

static int soc_button_get_acpi_object_int(const union acpi_object *obj)
{
 if (obj->type != ACPI_TYPE_INTEGER)
  return -1;

 return obj->integer.value;
}

/* Parse a single ACPI0011 _DSD button descriptor */
static int soc_button_parse_btn_desc(struct device *dev,
         const union acpi_object *desc,
         int collection_uid,
         struct soc_button_info *info)
{
 int upage, usage;

 if (desc->type != ACPI_TYPE_PACKAGE ||
     desc->package.count != 5 ||
     /* First byte should be 1 (control) */
     soc_button_get_acpi_object_int(&desc->package.elements[0]) != 1 ||
     /* Third byte should be collection uid */
     soc_button_get_acpi_object_int(&desc->package.elements[2]) !=
           collection_uid) {
  dev_err(dev, "Invalid ACPI Button Descriptor\n");
  return -ENODEV;
 }

 info->event_type = EV_KEY;
 info->active_low = true;
 info->acpi_index =
  soc_button_get_acpi_object_int(&desc->package.elements[1]);
 upage = soc_button_get_acpi_object_int(&desc->package.elements[3]);
 usage = soc_button_get_acpi_object_int(&desc->package.elements[4]);

 /*
  * The UUID: fa6bd625-9ce8-470d-a2c7-b3ca36c4282e descriptors use HID
  * usage page and usage codes, but otherwise the device is not HID
  * compliant: it uses one irq per button instead of generating HID
  * input reports and some buttons should generate wakeups where as
  * others should not, so we cannot use the HID subsystem.
  *
  * Luckily all devices only use a few usage page + usage combinations,
  * so we can simply check for the known combinations here.
 */
 if (upage == 0x01 && usage == 0x81) {
  info->name = "power";
  info->event_code = KEY_POWER;
  info->wakeup = true;
 } else if (upage == 0x01 && usage == 0xc6) {
  info->name = "airplane mode switch";
  info->event_type = EV_SW;
  info->event_code = SW_RFKILL_ALL;
  info->active_low = false;
 } else if (upage == 0x01 && usage == 0xca) {
  info->name = "rotation lock switch";
  info->event_type = EV_SW;
  info->event_code = SW_ROTATE_LOCK;
 } else if (upage == 0x07 && usage == 0xe3) {
  info->name = "home";
  info->event_code = KEY_LEFTMETA;
  info->wakeup = true;
 } else if (upage == 0x0c && usage == 0xe9) {
  info->name = "volume_up";
  info->event_code = KEY_VOLUMEUP;
  info->autorepeat = true;
 } else if (upage == 0x0c && usage == 0xea) {
  info->name = "volume_down";
  info->event_code = KEY_VOLUMEDOWN;
  info->autorepeat = true;
 } else {
  dev_warn(dev, "Unknown button index %d upage %02x usage %02x, ignoring\n",
    info->acpi_index, upage, usage);
  info->name = "unknown";
  info->event_code = KEY_RESERVED;
 }

 return 0;
}

/* ACPI0011 _DSD btns descriptors UUID: fa6bd625-9ce8-470d-a2c7-b3ca36c4282e */
static const u8 btns_desc_uuid[16] = {
 0x25, 0xd6, 0x6b, 0xfa, 0xe8, 0x9c, 0x0d, 0x47,
 0xa2, 0xc7, 0xb3, 0xca, 0x36, 0xc4, 0x28, 0x2e
};

/* Parse ACPI0011 _DSD button descriptors */
static struct soc_button_info *soc_button_get_button_info(struct device *dev)
{
 struct acpi_buffer buf = { ACPI_ALLOCATE_BUFFER };
 const union acpi_object *desc, *el0, *uuid, *btns_desc = NULL;
 struct soc_button_info *button_info;
 acpi_status status;
 int i, btn, collection_uid = -1;

 status = acpi_evaluate_object_typed(ACPI_HANDLE(dev), "_DSD", NULL,
         &buf, ACPI_TYPE_PACKAGE);
 if (ACPI_FAILURE(status)) {
  dev_err(dev, "ACPI _DSD object not found\n");
  return ERR_PTR(-ENODEV);
 }

 /* Look for the Button Descriptors UUID */
 desc = buf.pointer;
 for (i = 0; (i + 1) < desc->package.count; i += 2) {
  uuid = &desc->package.elements[i];

  if (uuid->type != ACPI_TYPE_BUFFER ||
      uuid->buffer.length != 16 ||
      desc->package.elements[i + 1].type != ACPI_TYPE_PACKAGE) {
   break;
  }

  if (memcmp(uuid->buffer.pointer, btns_desc_uuid, 16) == 0) {
   btns_desc = &desc->package.elements[i + 1];
   break;
  }
 }

 if (!btns_desc) {
  dev_err(dev, "ACPI Button Descriptors not found\n");
  button_info = ERR_PTR(-ENODEV);
  goto out;
 }

 /* The first package describes the collection */
 el0 = &btns_desc->package.elements[0];
 if (el0->type == ACPI_TYPE_PACKAGE &&
     el0->package.count == 5 &&
     /* First byte should be 0 (collection) */
     soc_button_get_acpi_object_int(&el0->package.elements[0]) == 0 &&
     /* Third byte should be 0 (top level collection) */
     soc_button_get_acpi_object_int(&el0->package.elements[2]) == 0) {
  collection_uid = soc_button_get_acpi_object_int(
      &el0->package.elements[1]);
 }
 if (collection_uid == -1) {
  dev_err(dev, "Invalid Button Collection Descriptor\n");
  button_info = ERR_PTR(-ENODEV);
  goto out;
 }

 /* There are package.count - 1 buttons + 1 terminating empty entry */
 button_info = devm_kcalloc(dev, btns_desc->package.count,
       sizeof(*button_info), GFP_KERNEL);
 if (!button_info) {
  button_info = ERR_PTR(-ENOMEM);
  goto out;
 }

 /* Parse the button descriptors */
 for (i = 1, btn = 0; i < btns_desc->package.count; i++, btn++) {
  if (soc_button_parse_btn_desc(dev,
           &btns_desc->package.elements[i],
           collection_uid,
           &button_info[btn])) {
   button_info = ERR_PTR(-ENODEV);
   goto out;
  }
 }

out:
 kfree(buf.pointer);
 return button_info;
}

static void soc_button_remove(struct platform_device *pdev)
{
 struct soc_button_data *priv = platform_get_drvdata(pdev);

 int i;

 for (i = 0; i < BUTTON_TYPES; i++)
  if (priv->children[i])
   platform_device_unregister(priv->children[i]);
}

static int soc_button_probe(struct platform_device *pdev)
{
 struct device *dev = &pdev->dev;
 const struct soc_device_data *device_data;
 const struct soc_button_info *button_info;
 struct soc_button_data *priv;
 struct platform_device *pd;
 int i;
 int error;

 device_data = acpi_device_get_match_data(dev);
 if (device_data && device_data->check) {
  error = device_data->check(dev);
  if (error)
   return error;
 }

 if (device_data && device_data->button_info) {
  button_info = device_data->button_info;
 } else {
  button_info = soc_button_get_button_info(dev);
  if (IS_ERR(button_info))
   return PTR_ERR(button_info);
 }

 error = gpiod_count(dev, NULL);
 if (error < 0) {
  dev_dbg(dev, "no GPIO attached, ignoring...\n");
  return -ENODEV;
 }

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

 platform_set_drvdata(pdev, priv);

 for (i = 0; i < BUTTON_TYPES; i++) {
  pd = soc_button_device_create(pdev, button_info, i == 0);
  if (IS_ERR(pd)) {
   error = PTR_ERR(pd);
   if (error != -ENODEV) {
    soc_button_remove(pdev);
    return error;
   }
   continue;
  }

  priv->children[i] = pd;
 }

 if (!priv->children[0] && !priv->children[1])
  return -ENODEV;

 if (!device_data || !device_data->button_info)
  devm_kfree(dev, button_info);

 return 0;
}

/*
 * Definition of buttons on the tablet. The ACPI index of each button
 * is defined in section 2.8.7.2 of "Windows ACPI Design Guide for SoC
 * Platforms"
 */
static const struct soc_button_info soc_button_PNP0C40[] = {
 { "power", 0, EV_KEY, KEY_POWER, false, true, true },
 { "home", 1, EV_KEY, KEY_LEFTMETA, false, true, true },
 { "volume_up", 2, EV_KEY, KEY_VOLUMEUP, true, false, true },
 { "volume_down", 3, EV_KEY, KEY_VOLUMEDOWN, true, false, true },
 { "rotation_lock", 4, EV_KEY, KEY_ROTATE_LOCK_TOGGLE, false, false, true },
 { }
};

static const struct soc_device_data soc_device_PNP0C40 = {
 .button_info = soc_button_PNP0C40,
};

static const struct soc_button_info soc_button_INT33D3[] = {
 { "tablet_mode", 0, EV_SW, SW_TABLET_MODE, false, false, false },
 { }
};

static const struct soc_device_data soc_device_INT33D3 = {
 .button_info = soc_button_INT33D3,
};

/*
 * Button info for Microsoft Surface 3 (non pro), this is identical to
 * the PNP0C40 info except that the home button is active-high.
 *
 * The Surface 3 Pro also has a MSHW0028 ACPI device, but that uses a custom
 * version of the drivers/platform/x86/intel/hid.c 5 button array ACPI API
 * instead. A check() callback is not necessary though as the Surface 3 Pro
 * MSHW0028 ACPI device's resource table does not contain any GPIOs.
 */
static const struct soc_button_info soc_button_MSHW0028[] = {
 { "power", 0, EV_KEY, KEY_POWER, false, true, true },
 { "home", 1, EV_KEY, KEY_LEFTMETA, false, true, false },
 { "volume_up", 2, EV_KEY, KEY_VOLUMEUP, true, false, true },
 { "volume_down", 3, EV_KEY, KEY_VOLUMEDOWN, true, false, true },
 { }
};

static const struct soc_device_data soc_device_MSHW0028 = {
 .button_info = soc_button_MSHW0028,
};

/*
 * Special device check for Surface Book 2 and Surface Pro (2017).
 * Both, the Surface Pro 4 (surfacepro3_button.c) and the above mentioned
 * devices use MSHW0040 for power and volume buttons, however the way they
 * have to be addressed differs. Make sure that we only load this drivers
 * for the correct devices by checking the OEM Platform Revision provided by
 * the _DSM method.
 */
#define MSHW0040_DSM_REVISION  0x01
#define MSHW0040_DSM_GET_OMPR  0x02 // get OEM Platform Revision
static const guid_t MSHW0040_DSM_UUID =
 GUID_INIT(0x6fd05c69, 0xcde3, 0x49f4, 0x95, 0xed, 0xab, 0x16, 0x65,
    0x49, 0x80, 0x35);

static int soc_device_check_MSHW0040(struct device *dev)
{
 acpi_handle handle = ACPI_HANDLE(dev);
 union acpi_object *result;
 u64 oem_platform_rev = 0; // valid revisions are nonzero

 // get OEM platform revision
 result = acpi_evaluate_dsm_typed(handle, &MSHW0040_DSM_UUID,
      MSHW0040_DSM_REVISION,
      MSHW0040_DSM_GET_OMPR, NULL,
      ACPI_TYPE_INTEGER);

 if (result) {
  oem_platform_rev = result->integer.value;
  ACPI_FREE(result);
 }

 /*
  * If the revision is zero here, the _DSM evaluation has failed. This
  * indicates that we have a Pro 4 or Book 1 and this driver should not
  * be used.
 */
 if (oem_platform_rev == 0)
  return -ENODEV;

 dev_dbg(dev, "OEM Platform Revision %llu\n", oem_platform_rev);

 return 0;
}

/*
 * Button infos for Microsoft Surface Book 2 and Surface Pro (2017).
 * Obtained from DSDT/testing.
 */
static const struct soc_button_info soc_button_MSHW0040[] = {
 { "power", 0, EV_KEY, KEY_POWER, false, true, true },
 { "volume_up", 2, EV_KEY, KEY_VOLUMEUP, true, false, true },
 { "volume_down", 4, EV_KEY, KEY_VOLUMEDOWN, true, false, true },
 { }
};

static const struct soc_device_data soc_device_MSHW0040 = {
 .button_info = soc_button_MSHW0040,
 .check = soc_device_check_MSHW0040,
};

static const struct acpi_device_id soc_button_acpi_match[] = {
 { "PNP0C40", (unsigned long)&soc_device_PNP0C40 },
 { "INT33D3", (unsigned long)&soc_device_INT33D3 },
 { "ID9001", (unsigned long)&soc_device_INT33D3 },
 { "ACPI0011", 0 },

 /* Microsoft Surface Devices (3th, 5th and 6th generation) */
 { "MSHW0028", (unsigned long)&soc_device_MSHW0028 },
 { "MSHW0040", (unsigned long)&soc_device_MSHW0040 },

 { }
};

MODULE_DEVICE_TABLE(acpi, soc_button_acpi_match);

static struct platform_driver soc_button_driver = {
 .probe          = soc_button_probe,
 .remove  = soc_button_remove,
 .driver  = {
  .name = KBUILD_MODNAME,
  .acpi_match_table = ACPI_PTR(soc_button_acpi_match),
 },
};
module_platform_driver(soc_button_driver);

MODULE_DESCRIPTION("Windows-compatible SoC Button Array driver");
MODULE_LICENSE("GPL");