// SPDX-License-Identifier: GPL-2.0-only
 /*
- * Copyright (c) 2010-2011, Code Aurora Forum. All rights reserved.
+ * Copyright (c) 2010-2011, 2020-2021, The Linux Foundation. All rights reserved.
  * Copyright (c) 2014, Sony Mobile Communications Inc.
  */
 
 #define PON_RT_STS                     0x10
 #define  PON_KPDPWR_N_SET              BIT(0)
 #define  PON_RESIN_N_SET               BIT(1)
+#define  PON_GEN3_RESIN_N_SET          BIT(6)
+#define  PON_GEN3_KPDPWR_N_SET         BIT(7)
 
 #define PON_PS_HOLD_RST_CTL            0x5a
 #define PON_PS_HOLD_RST_CTL2           0x5b
 #define  PON_DBC_DELAY_MASK            0x7
 
 struct pm8941_data {
-       unsigned int pull_up_bit;
-       unsigned int status_bit;
+       unsigned int    pull_up_bit;
+       unsigned int    status_bit;
+       bool            supports_ps_hold_poff_config;
+       bool            supports_debounce_config;
+       const char      *name;
+       const char      *phys;
 };
 
 struct pm8941_pwrkey {
 
        input_set_capability(pwrkey->input, EV_KEY, pwrkey->code);
 
-       pwrkey->input->name = "pm8941_pwrkey";
-       pwrkey->input->phys = "pm8941_pwrkey/input0";
-
-       req_delay = (req_delay << 6) / USEC_PER_SEC;
-       req_delay = ilog2(req_delay);
-
-       error = regmap_update_bits(pwrkey->regmap,
-                                  pwrkey->baseaddr + PON_DBC_CTL,
-                                  PON_DBC_DELAY_MASK,
-                                  req_delay);
-       if (error) {
-               dev_err(&pdev->dev, "failed to set debounce: %d\n", error);
-               return error;
+       pwrkey->input->name = pwrkey->data->name;
+       pwrkey->input->phys = pwrkey->data->phys;
+
+       if (pwrkey->data->supports_debounce_config) {
+               req_delay = (req_delay << 6) / USEC_PER_SEC;
+               req_delay = ilog2(req_delay);
+
+               error = regmap_update_bits(pwrkey->regmap,
+                                          pwrkey->baseaddr + PON_DBC_CTL,
+                                          PON_DBC_DELAY_MASK,
+                                          req_delay);
+               if (error) {
+                       dev_err(&pdev->dev, "failed to set debounce: %d\n",
+                               error);
+                       return error;
+               }
        }
 
-       error = regmap_update_bits(pwrkey->regmap,
-                                  pwrkey->baseaddr + PON_PULL_CTL,
-                                  pwrkey->data->pull_up_bit,
-                                  pull_up ? pwrkey->data->pull_up_bit : 0);
-       if (error) {
-               dev_err(&pdev->dev, "failed to set pull: %d\n", error);
-               return error;
+       if (pwrkey->data->pull_up_bit) {
+               error = regmap_update_bits(pwrkey->regmap,
+                                          pwrkey->baseaddr + PON_PULL_CTL,
+                                          pwrkey->data->pull_up_bit,
+                                          pull_up ? pwrkey->data->pull_up_bit :
+                                                    0);
+               if (error) {
+                       dev_err(&pdev->dev, "failed to set pull: %d\n", error);
+                       return error;
+               }
        }
 
        error = devm_request_threaded_irq(&pdev->dev, pwrkey->irq,
                                          NULL, pm8941_pwrkey_irq,
                                          IRQF_ONESHOT,
-                                         "pm8941_pwrkey", pwrkey);
+                                         pwrkey->data->name, pwrkey);
        if (error) {
                dev_err(&pdev->dev, "failed requesting IRQ: %d\n", error);
                return error;
                return error;
        }
 
-       pwrkey->reboot_notifier.notifier_call = pm8941_reboot_notify,
-       error = register_reboot_notifier(&pwrkey->reboot_notifier);
-       if (error) {
-               dev_err(&pdev->dev, "failed to register reboot notifier: %d\n",
-                       error);
-               return error;
+       if (pwrkey->data->supports_ps_hold_poff_config) {
+               pwrkey->reboot_notifier.notifier_call = pm8941_reboot_notify,
+               error = register_reboot_notifier(&pwrkey->reboot_notifier);
+               if (error) {
+                       dev_err(&pdev->dev, "failed to register reboot notifier: %d\n",
+                               error);
+                       return error;
+               }
        }
 
        platform_set_drvdata(pdev, pwrkey);
 {
        struct pm8941_pwrkey *pwrkey = platform_get_drvdata(pdev);
 
-       unregister_reboot_notifier(&pwrkey->reboot_notifier);
+       if (pwrkey->data->supports_ps_hold_poff_config)
+               unregister_reboot_notifier(&pwrkey->reboot_notifier);
 
        return 0;
 }
 static const struct pm8941_data pwrkey_data = {
        .pull_up_bit = PON_KPDPWR_PULL_UP,
        .status_bit = PON_KPDPWR_N_SET,
+       .name = "pm8941_pwrkey",
+       .phys = "pm8941_pwrkey/input0",
+       .supports_ps_hold_poff_config = true,
+       .supports_debounce_config = true,
 };
 
 static const struct pm8941_data resin_data = {
        .pull_up_bit = PON_RESIN_PULL_UP,
        .status_bit = PON_RESIN_N_SET,
+       .name = "pm8941_resin",
+       .phys = "pm8941_resin/input0",
+       .supports_ps_hold_poff_config = true,
+       .supports_debounce_config = true,
+};
+
+static const struct pm8941_data pon_gen3_pwrkey_data = {
+       .status_bit = PON_GEN3_KPDPWR_N_SET,
+       .name = "pmic_pwrkey",
+       .phys = "pmic_pwrkey/input0",
+       .supports_ps_hold_poff_config = false,
+       .supports_debounce_config = false,
+};
+
+static const struct pm8941_data pon_gen3_resin_data = {
+       .status_bit = PON_GEN3_RESIN_N_SET,
+       .name = "pmic_resin",
+       .phys = "pmic_resin/input0",
+       .supports_ps_hold_poff_config = false,
+       .supports_debounce_config = false,
 };
 
 static const struct of_device_id pm8941_pwr_key_id_table[] = {
        { .compatible = "qcom,pm8941-pwrkey", .data = &pwrkey_data },
        { .compatible = "qcom,pm8941-resin", .data = &resin_data },
+       { .compatible = "qcom,pmk8350-pwrkey", .data = &pon_gen3_pwrkey_data },
+       { .compatible = "qcom,pmk8350-resin", .data = &pon_gen3_resin_data },
        { }
 };
 MODULE_DEVICE_TABLE(of, pm8941_pwr_key_id_table);