// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * AXP20x PMIC USB power supply status driver
 *
 * Copyright (C) 2015 Hans de Goede <hdegoede@redhat.com>
 * Copyright (C) 2014 Bruno Prémont <bonbons@linux-vserver.org>
 */

#include <linux/bitops.h>
#include <linux/device.h>
#include <linux/devm-helpers.h>
#include <linux/init.h>
#include <linux/interrupt.h>
#include <linux/kernel.h>
#include <linux/mfd/axp20x.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/platform_device.h>
#include <linux/pm.h>
#include <linux/power_supply.h>
#include <linux/regmap.h>
#include <linux/slab.h>
#include <linux/iio/consumer.h>
#include <linux/workqueue.h>

#define DRVNAME "axp20x-usb-power-supply"

#define AXP192_USB_OTG_STATUS           0x04

#define AXP20X_PWR_STATUS_VBUS_PRESENT  BIT(5)
#define AXP20X_PWR_STATUS_VBUS_USED     BIT(4)

#define AXP717_PWR_STATUS_VBUS_GOOD     BIT(5)

#define AXP20X_USB_STATUS_VBUS_VALID    BIT(2)

#define AXP717_PMU_FAULT_VBUS           BIT(5)
#define AXP717_PMU_FAULT_VSYS           BIT(3)

#define AXP20X_VBUS_VHOLD_uV(b)         (4000000 + (((b) >> 3) & 7) * 100000)
#define AXP20X_VBUS_VHOLD_MASK          GENMASK(5, 3)
#define AXP20X_VBUS_VHOLD_OFFSET        3

#define AXP20X_ADC_EN1_VBUS_CURR        BIT(2)
#define AXP20X_ADC_EN1_VBUS_VOLT        BIT(3)

#define AXP717_INPUT_VOL_LIMIT_MASK     GENMASK(3, 0)
#define AXP717_INPUT_CUR_LIMIT_MASK     GENMASK(5, 0)
#define AXP717_ADC_DATA_MASK            GENMASK(14, 0)

#define AXP717_ADC_EN_VBUS_VOLT         BIT(2)

/*
 * Note do not raise the debounce time, we must report Vusb high within
 * 100ms otherwise we get Vbus errors in musb.
 */
#define DEBOUNCE_TIME                   msecs_to_jiffies(50)

struct axp20x_usb_power;

struct axp_data {
        const struct power_supply_desc  *power_desc;
        const char * const              *irq_names;
        unsigned int                    num_irq_names;
        const int                       *curr_lim_table;
        int                             curr_lim_table_size;
        struct reg_field                curr_lim_fld;
        struct reg_field                vbus_valid_bit;
        struct reg_field                vbus_mon_bit;
        struct reg_field                usb_bc_en_bit;
        struct reg_field                usb_bc_det_fld;
        struct reg_field                vbus_disable_bit;
        bool                            vbus_needs_polling: 1;
        void (*axp20x_read_vbus)(struct work_struct *work);
        int (*axp20x_cfg_iio_chan)(struct platform_device *pdev,
                                   struct axp20x_usb_power *power);
        int (*axp20x_cfg_adc_reg)(struct axp20x_usb_power *power);
};

struct axp20x_usb_power {
        struct device *dev;
        struct regmap *regmap;
        struct regmap_field *curr_lim_fld;
        struct regmap_field *vbus_valid_bit;
        struct regmap_field *vbus_mon_bit;
        struct regmap_field *usb_bc_en_bit;
        struct regmap_field *usb_bc_det_fld;
        struct regmap_field *vbus_disable_bit;
        struct power_supply *supply;
        const struct axp_data *axp_data;
        struct iio_channel *vbus_v;
        struct iio_channel *vbus_i;
        struct delayed_work vbus_detect;
        int max_input_cur;
        unsigned int old_status;
        unsigned int online;
        unsigned int num_irqs;
        unsigned int irqs[] __counted_by(num_irqs);
};

static bool axp20x_usb_vbus_needs_polling(struct axp20x_usb_power *power)
{
        /*
         * Polling is only necessary while VBUS is offline. While online, a
         * present->absent transition implies an online->offline transition
         * and will trigger the VBUS_REMOVAL IRQ.
         */
        if (power->axp_data->vbus_needs_polling && !power->online)
                return true;

        return false;
}

static irqreturn_t axp20x_usb_power_irq(int irq, void *devid)
{
        struct axp20x_usb_power *power = devid;

        power_supply_changed(power->supply);

        mod_delayed_work(system_power_efficient_wq, &power->vbus_detect, DEBOUNCE_TIME);

        return IRQ_HANDLED;
}

static void axp20x_usb_power_poll_vbus(struct work_struct *work)
{
        struct axp20x_usb_power *power =
                container_of(work, struct axp20x_usb_power, vbus_detect.work);
        unsigned int val;
        int ret;

        ret = regmap_read(power->regmap, AXP20X_PWR_INPUT_STATUS, &val);
        if (ret)
                goto out;

        val &= (AXP20X_PWR_STATUS_VBUS_PRESENT | AXP20X_PWR_STATUS_VBUS_USED);
        if (val != power->old_status)
                power_supply_changed(power->supply);

        if (power->usb_bc_en_bit && (val & AXP20X_PWR_STATUS_VBUS_PRESENT) !=
                (power->old_status & AXP20X_PWR_STATUS_VBUS_PRESENT)) {
                dev_dbg(power->dev, "Cable status changed, re-enabling USB BC");
                ret = regmap_field_write(power->usb_bc_en_bit, 1);
                if (ret)
                        dev_err(power->dev, "failed to enable USB BC: errno %d",
                                ret);
        }

        power->old_status = val;
        power->online = val & AXP20X_PWR_STATUS_VBUS_USED;

out:
        if (axp20x_usb_vbus_needs_polling(power))
                mod_delayed_work(system_power_efficient_wq, &power->vbus_detect, DEBOUNCE_TIME);
}

static void axp717_usb_power_poll_vbus(struct work_struct *work)
{
        struct axp20x_usb_power *power =
                container_of(work, struct axp20x_usb_power, vbus_detect.work);
        unsigned int val;
        int ret;

        ret = regmap_read(power->regmap, AXP717_ON_INDICATE, &val);
        if (ret)
                return;

        val &= AXP717_PWR_STATUS_VBUS_GOOD;
        if (val != power->old_status)
                power_supply_changed(power->supply);

        power->old_status = val;
}

static int axp20x_get_usb_type(struct axp20x_usb_power *power,
                               union power_supply_propval *val)
{
        unsigned int reg;
        int ret;

        if (!power->usb_bc_det_fld)
                return -EINVAL;

        ret = regmap_field_read(power->usb_bc_det_fld, &reg);
        if (ret)
                return ret;

        switch (reg) {
        case 1:
                val->intval = POWER_SUPPLY_USB_TYPE_SDP;
                break;
        case 2:
                val->intval = POWER_SUPPLY_USB_TYPE_CDP;
                break;
        case 3:
                val->intval = POWER_SUPPLY_USB_TYPE_DCP;
                break;
        default:
                val->intval = POWER_SUPPLY_USB_TYPE_UNKNOWN;
                break;
        }

        return 0;
}

static int axp20x_usb_power_get_property(struct power_supply *psy,
        enum power_supply_property psp, union power_supply_propval *val)
{
        struct axp20x_usb_power *power = power_supply_get_drvdata(psy);
        unsigned int input, v;
        int ret;

        switch (psp) {
        case POWER_SUPPLY_PROP_VOLTAGE_MIN:
                ret = regmap_read(power->regmap, AXP20X_VBUS_IPSOUT_MGMT, &v);
                if (ret)
                        return ret;

                val->intval = AXP20X_VBUS_VHOLD_uV(v);
                return 0;
        case POWER_SUPPLY_PROP_VOLTAGE_NOW:
                if (IS_ENABLED(CONFIG_AXP20X_ADC)) {
                        /*
                         * IIO framework gives mV but Power Supply framework
                         * gives uV.
                         */
                        ret = iio_read_channel_processed_scale(power->vbus_v,
                                                               &val->intval, 1000);
                        if (ret)
                                return ret;

                        return 0;
                }

                ret = axp20x_read_variable_width(power->regmap,
                                                 AXP20X_VBUS_V_ADC_H, 12);
                if (ret < 0)
                        return ret;

                val->intval = ret * 1700; /* 1 step = 1.7 mV */
                return 0;
        case POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT:
                ret = regmap_field_read(power->curr_lim_fld, &v);
                if (ret)
                        return ret;

                if (v < power->axp_data->curr_lim_table_size)
                        val->intval = power->axp_data->curr_lim_table[v];
                else
                        val->intval = power->axp_data->curr_lim_table[
                                power->axp_data->curr_lim_table_size - 1];
                return 0;
        case POWER_SUPPLY_PROP_CURRENT_NOW:
                if (IS_ENABLED(CONFIG_AXP20X_ADC)) {
                        /*
                         * IIO framework gives mA but Power Supply framework
                         * gives uA.
                         */
                        ret = iio_read_channel_processed_scale(power->vbus_i,
                                                               &val->intval, 1000);
                        if (ret)
                                return ret;

                        return 0;
                }

                ret = axp20x_read_variable_width(power->regmap,
                                                 AXP20X_VBUS_I_ADC_H, 12);
                if (ret < 0)
                        return ret;

                val->intval = ret * 375; /* 1 step = 0.375 mA */
                return 0;

        case POWER_SUPPLY_PROP_USB_TYPE:
                return axp20x_get_usb_type(power, val);
        default:
                break;
        }

        /* All the properties below need the input-status reg value */
        ret = regmap_read(power->regmap, AXP20X_PWR_INPUT_STATUS, &input);
        if (ret)
                return ret;

        switch (psp) {
        case POWER_SUPPLY_PROP_HEALTH:
                if (!(input & AXP20X_PWR_STATUS_VBUS_PRESENT)) {
                        val->intval = POWER_SUPPLY_HEALTH_UNKNOWN;
                        break;
                }

                val->intval = POWER_SUPPLY_HEALTH_GOOD;

                if (power->vbus_valid_bit) {
                        ret = regmap_field_read(power->vbus_valid_bit, &v);
                        if (ret)
                                return ret;

                        if (v == 0)
                                val->intval = POWER_SUPPLY_HEALTH_UNSPEC_FAILURE;
                }

                break;
        case POWER_SUPPLY_PROP_PRESENT:
                val->intval = !!(input & AXP20X_PWR_STATUS_VBUS_PRESENT);
                break;
        case POWER_SUPPLY_PROP_ONLINE:
                val->intval = !!(input & AXP20X_PWR_STATUS_VBUS_USED);
                break;
        default:
                return -EINVAL;
        }

        return 0;
}

static int axp717_usb_power_get_property(struct power_supply *psy,
        enum power_supply_property psp, union power_supply_propval *val)
{
        struct axp20x_usb_power *power = power_supply_get_drvdata(psy);
        unsigned int v;
        int ret;

        switch (psp) {
        case POWER_SUPPLY_PROP_HEALTH:
                val->intval = POWER_SUPPLY_HEALTH_GOOD;
                ret = regmap_read(power->regmap, AXP717_ON_INDICATE, &v);
                if (ret)
                        return ret;

                if (!(v & AXP717_PWR_STATUS_VBUS_GOOD))
                        val->intval = POWER_SUPPLY_HEALTH_UNKNOWN;

                ret = regmap_read(power->regmap, AXP717_PMU_FAULT_VBUS, &v);
                if (ret)
                        return ret;

                v &= (AXP717_PMU_FAULT_VBUS | AXP717_PMU_FAULT_VSYS);
                if (v) {
                        val->intval = POWER_SUPPLY_HEALTH_OVERVOLTAGE;
                        regmap_write(power->regmap, AXP717_PMU_FAULT_VBUS, v);
                }

                break;
        case POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT:
                ret = regmap_read(power->regmap, AXP717_INPUT_CUR_LIMIT_CTRL, &v);
                if (ret)
                        return ret;

                /* 50ma step size with 100ma offset. */
                v &= AXP717_INPUT_CUR_LIMIT_MASK;
                val->intval = (v * 50000) + 100000;
                break;
        case POWER_SUPPLY_PROP_ONLINE:
        case POWER_SUPPLY_PROP_PRESENT:
                ret = regmap_read(power->regmap, AXP717_ON_INDICATE, &v);
                if (ret)
                        return ret;
                val->intval = !!(v & AXP717_PWR_STATUS_VBUS_GOOD);
                break;
        case POWER_SUPPLY_PROP_USB_TYPE:
                return axp20x_get_usb_type(power, val);
        case POWER_SUPPLY_PROP_VOLTAGE_MIN:
                ret = regmap_read(power->regmap, AXP717_INPUT_VOL_LIMIT_CTRL, &v);
                if (ret)
                        return ret;

                /* 80mv step size with 3.88v offset. */
                v &= AXP717_INPUT_VOL_LIMIT_MASK;
                val->intval = (v * 80000) + 3880000;
                break;
        case POWER_SUPPLY_PROP_VOLTAGE_NOW:
                if (IS_ENABLED(CONFIG_AXP20X_ADC)) {
                        /*
                         * IIO framework gives mV but Power Supply framework
                         * gives uV.
                         */
                        ret = iio_read_channel_processed_scale(power->vbus_v,
                                                               &val->intval, 1000);
                        if (ret)
                                return ret;

                        return 0;
                }

                ret = axp20x_read_variable_width(power->regmap,
                                                 AXP717_VBUS_V_H, 16);
                if (ret < 0)
                        return ret;

                val->intval = (ret % AXP717_ADC_DATA_MASK) * 1000;
                break;
        default:
                return -EINVAL;
        }

        return 0;

}

static int axp20x_usb_power_set_voltage_min(struct axp20x_usb_power *power,
                                            int intval)
{
        int val;

        switch (intval) {
        case 4000000:
        case 4100000:
        case 4200000:
        case 4300000:
        case 4400000:
        case 4500000:
        case 4600000:
        case 4700000:
                val = (intval - 4000000) / 100000;
                return regmap_update_bits(power->regmap,
                                          AXP20X_VBUS_IPSOUT_MGMT,
                                          AXP20X_VBUS_VHOLD_MASK,
                                          val << AXP20X_VBUS_VHOLD_OFFSET);
        default:
                return -EINVAL;
        }

        return -EINVAL;
}

static int axp717_usb_power_set_voltage_min(struct axp20x_usb_power *power,
                                            int intval)
{
        int val;

        /* Minimum value of 3.88v and maximum of 5.08v. */
        if (intval < 3880000 || intval > 5080000)
                return -EINVAL;

        /* step size of 80ma with 3.88v offset. */
        val = (intval - 3880000) / 80000;
        return regmap_update_bits(power->regmap,
                                  AXP717_INPUT_VOL_LIMIT_CTRL,
                                  AXP717_INPUT_VOL_LIMIT_MASK, val);
}

static int axp20x_usb_power_set_input_current_limit(struct axp20x_usb_power *power,
                                                    int intval)
{
        int ret;
        unsigned int reg;
        const unsigned int max = power->axp_data->curr_lim_table_size;

        if (intval == -1)
                return -EINVAL;

        if (power->max_input_cur && (intval > power->max_input_cur)) {
                dev_warn(power->dev,
                         "requested current %d clamped to max current %d\n",
                         intval, power->max_input_cur);
                intval = power->max_input_cur;
        }

        /*
         * BC1.2 detection can cause a race condition if we try to set a current
         * limit while it's in progress. When it finishes it will overwrite the
         * current limit we just set.
         */
        if (power->usb_bc_en_bit) {
                dev_dbg(power->dev,
                        "disabling BC1.2 detection because current limit was set");
                ret = regmap_field_write(power->usb_bc_en_bit, 0);
                if (ret)
                        return ret;
        }

        for (reg = max - 1; reg > 0; reg--)
                if (power->axp_data->curr_lim_table[reg] <= intval)
                        break;

        dev_dbg(power->dev, "setting input current limit reg to %d (%d uA), requested %d uA",
                reg, power->axp_data->curr_lim_table[reg], intval);

        return regmap_field_write(power->curr_lim_fld, reg);
}

static int axp717_usb_power_set_input_current_limit(struct axp20x_usb_power *power,
                                                    int intval)
{
        int tmp;

        /* Minimum value of 100mA and maximum value of 3.25A*/
        if (intval < 100000 || intval > 3250000)
                return -EINVAL;

        if (power->max_input_cur && (intval > power->max_input_cur)) {
                dev_warn(power->dev,
                         "requested current %d clamped to max current %d\n",
                         intval, power->max_input_cur);
                intval = power->max_input_cur;
        }

        /* Minimum value of 100mA with step size of 50mA. */
        tmp = (intval - 100000) / 50000;
        return regmap_update_bits(power->regmap,
                                  AXP717_INPUT_CUR_LIMIT_CTRL,
                                  AXP717_INPUT_CUR_LIMIT_MASK, tmp);
}

static int axp20x_usb_power_set_property(struct power_supply *psy,
                                         enum power_supply_property psp,
                                         const union power_supply_propval *val)
{
        struct axp20x_usb_power *power = power_supply_get_drvdata(psy);

        switch (psp) {
        case POWER_SUPPLY_PROP_ONLINE:
                if (!power->vbus_disable_bit)
                        return -EINVAL;

                return regmap_field_write(power->vbus_disable_bit, !val->intval);

        case POWER_SUPPLY_PROP_VOLTAGE_MIN:
                return axp20x_usb_power_set_voltage_min(power, val->intval);

        case POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT:
                return axp20x_usb_power_set_input_current_limit(power, val->intval);

        default:
                return -EINVAL;
        }
}

static int axp717_usb_power_set_property(struct power_supply *psy,
                                         enum power_supply_property psp,
                                         const union power_supply_propval *val)
{
        struct axp20x_usb_power *power = power_supply_get_drvdata(psy);

        switch (psp) {
        case POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT:
                return axp717_usb_power_set_input_current_limit(power, val->intval);

        case POWER_SUPPLY_PROP_VOLTAGE_MIN:
                return axp717_usb_power_set_voltage_min(power, val->intval);

        default:
                return -EINVAL;
        }

        return -EINVAL;
}

static int axp20x_usb_power_prop_writeable(struct power_supply *psy,
                                           enum power_supply_property psp)
{
        struct axp20x_usb_power *power = power_supply_get_drvdata(psy);

        /*
         * The VBUS path select flag works differently on AXP288 and newer:
         *  - On AXP20x and AXP22x, the flag enables VBUS (ignoring N_VBUSEN).
         *  - On AXP288 and AXP8xx, the flag disables VBUS (ignoring N_VBUSEN).
         * We only expose the control on variants where it can be used to force
         * the VBUS input offline.
         */
        if (psp == POWER_SUPPLY_PROP_ONLINE)
                return power->vbus_disable_bit != NULL;

        return psp == POWER_SUPPLY_PROP_VOLTAGE_MIN ||
               psp == POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT;
}

static int axp717_usb_power_prop_writeable(struct power_supply *psy,
                                           enum power_supply_property psp)
{
        return psp == POWER_SUPPLY_PROP_VOLTAGE_MIN ||
               psp == POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT;
}

static int axp20x_configure_iio_channels(struct platform_device *pdev,
                                         struct axp20x_usb_power *power)
{
        power->vbus_v = devm_iio_channel_get(&pdev->dev, "vbus_v");
        if (IS_ERR(power->vbus_v)) {
                if (PTR_ERR(power->vbus_v) == -ENODEV)
                        return -EPROBE_DEFER;
                return PTR_ERR(power->vbus_v);
        }

        power->vbus_i = devm_iio_channel_get(&pdev->dev, "vbus_i");
        if (IS_ERR(power->vbus_i)) {
                if (PTR_ERR(power->vbus_i) == -ENODEV)
                        return -EPROBE_DEFER;
                return PTR_ERR(power->vbus_i);
        }

        return 0;
}

static int axp717_configure_iio_channels(struct platform_device *pdev,
                                         struct axp20x_usb_power *power)
{
        power->vbus_v = devm_iio_channel_get(&pdev->dev, "vbus_v");
        if (IS_ERR(power->vbus_v)) {
                if (PTR_ERR(power->vbus_v) == -ENODEV)
                        return -EPROBE_DEFER;
                return PTR_ERR(power->vbus_v);
        }

        return 0;
}

static int axp20x_configure_adc_registers(struct axp20x_usb_power *power)
{
        /* Enable vbus voltage and current measurement */
        return regmap_update_bits(power->regmap, AXP20X_ADC_EN1,
                                  AXP20X_ADC_EN1_VBUS_CURR |
                                  AXP20X_ADC_EN1_VBUS_VOLT,
                                  AXP20X_ADC_EN1_VBUS_CURR |
                                  AXP20X_ADC_EN1_VBUS_VOLT);
}

static int axp717_configure_adc_registers(struct axp20x_usb_power *power)
{
        /* Enable vbus voltage measurement  */
        return regmap_update_bits(power->regmap, AXP717_ADC_CH_EN_CONTROL,
                                  AXP717_ADC_EN_VBUS_VOLT,
                                  AXP717_ADC_EN_VBUS_VOLT);
}

static enum power_supply_property axp20x_usb_power_properties[] = {
        POWER_SUPPLY_PROP_HEALTH,
        POWER_SUPPLY_PROP_PRESENT,
        POWER_SUPPLY_PROP_ONLINE,
        POWER_SUPPLY_PROP_VOLTAGE_MIN,
        POWER_SUPPLY_PROP_VOLTAGE_NOW,
        POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT,
        POWER_SUPPLY_PROP_CURRENT_NOW,
};

static enum power_supply_property axp22x_usb_power_properties[] = {
        POWER_SUPPLY_PROP_HEALTH,
        POWER_SUPPLY_PROP_PRESENT,
        POWER_SUPPLY_PROP_ONLINE,
        POWER_SUPPLY_PROP_VOLTAGE_MIN,
        POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT,
};

static enum power_supply_property axp717_usb_power_properties[] = {
        POWER_SUPPLY_PROP_HEALTH,
        POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT,
        POWER_SUPPLY_PROP_ONLINE,
        POWER_SUPPLY_PROP_PRESENT,
        POWER_SUPPLY_PROP_USB_TYPE,
        POWER_SUPPLY_PROP_VOLTAGE_MIN,
        POWER_SUPPLY_PROP_VOLTAGE_NOW,
};

static enum power_supply_property axp813_usb_power_properties[] = {
        POWER_SUPPLY_PROP_HEALTH,
        POWER_SUPPLY_PROP_PRESENT,
        POWER_SUPPLY_PROP_ONLINE,
        POWER_SUPPLY_PROP_VOLTAGE_MIN,
        POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT,
        POWER_SUPPLY_PROP_USB_TYPE,
};

static const struct power_supply_desc axp20x_usb_power_desc = {
        .name = "axp20x-usb",
        .type = POWER_SUPPLY_TYPE_USB,
        .properties = axp20x_usb_power_properties,
        .num_properties = ARRAY_SIZE(axp20x_usb_power_properties),
        .property_is_writeable = axp20x_usb_power_prop_writeable,
        .get_property = axp20x_usb_power_get_property,
        .set_property = axp20x_usb_power_set_property,
};

static const struct power_supply_desc axp22x_usb_power_desc = {
        .name = "axp20x-usb",
        .type = POWER_SUPPLY_TYPE_USB,
        .properties = axp22x_usb_power_properties,
        .num_properties = ARRAY_SIZE(axp22x_usb_power_properties),
        .property_is_writeable = axp20x_usb_power_prop_writeable,
        .get_property = axp20x_usb_power_get_property,
        .set_property = axp20x_usb_power_set_property,
};

static const struct power_supply_desc axp717_usb_power_desc = {
        .name = "axp20x-usb",
        .type = POWER_SUPPLY_TYPE_USB,
        .properties = axp717_usb_power_properties,
        .num_properties = ARRAY_SIZE(axp717_usb_power_properties),
        .property_is_writeable = axp717_usb_power_prop_writeable,
        .get_property = axp717_usb_power_get_property,
        .set_property = axp717_usb_power_set_property,
        .usb_types = BIT(POWER_SUPPLY_USB_TYPE_SDP) |
                     BIT(POWER_SUPPLY_USB_TYPE_CDP) |
                     BIT(POWER_SUPPLY_USB_TYPE_DCP) |
                     BIT(POWER_SUPPLY_USB_TYPE_UNKNOWN),
};

static const struct power_supply_desc axp813_usb_power_desc = {
        .name = "axp20x-usb",
        .type = POWER_SUPPLY_TYPE_USB,
        .properties = axp813_usb_power_properties,
        .num_properties = ARRAY_SIZE(axp813_usb_power_properties),
        .property_is_writeable = axp20x_usb_power_prop_writeable,
        .get_property = axp20x_usb_power_get_property,
        .set_property = axp20x_usb_power_set_property,
        .usb_types = BIT(POWER_SUPPLY_USB_TYPE_SDP) |
                     BIT(POWER_SUPPLY_USB_TYPE_CDP) |
                     BIT(POWER_SUPPLY_USB_TYPE_DCP) |
                     BIT(POWER_SUPPLY_USB_TYPE_UNKNOWN),
};

static const char * const axp20x_irq_names[] = {
        "VBUS_PLUGIN",
        "VBUS_REMOVAL",
        "VBUS_VALID",
        "VBUS_NOT_VALID",
};

static const char * const axp22x_irq_names[] = {
        "VBUS_PLUGIN",
        "VBUS_REMOVAL",
};

static const char * const axp717_irq_names[] = {
        "VBUS_PLUGIN",
        "VBUS_REMOVAL",
        "VBUS_OVER_V",
};

static int axp192_usb_curr_lim_table[] = {
        -1,
        -1,
        500000,
        100000,
};

static int axp20x_usb_curr_lim_table[] = {
        900000,
        500000,
        100000,
        -1,
};

static int axp221_usb_curr_lim_table[] = {
        900000,
        500000,
        -1,
        -1,
};

static int axp813_usb_curr_lim_table[] = {
        100000,
        500000,
        900000,
        1500000,
        2000000,
        2500000,
        3000000,
        3500000,
        4000000,
};

static const struct axp_data axp192_data = {
        .power_desc     = &axp20x_usb_power_desc,
        .irq_names      = axp20x_irq_names,
        .num_irq_names  = ARRAY_SIZE(axp20x_irq_names),
        .curr_lim_table = axp192_usb_curr_lim_table,
        .curr_lim_table_size = ARRAY_SIZE(axp192_usb_curr_lim_table),
        .curr_lim_fld   = REG_FIELD(AXP20X_VBUS_IPSOUT_MGMT, 0, 1),
        .vbus_valid_bit = REG_FIELD(AXP192_USB_OTG_STATUS, 2, 2),
        .vbus_mon_bit   = REG_FIELD(AXP20X_VBUS_MON, 3, 3),
        .axp20x_read_vbus = &axp20x_usb_power_poll_vbus,
        .axp20x_cfg_iio_chan = axp20x_configure_iio_channels,
        .axp20x_cfg_adc_reg = axp20x_configure_adc_registers,
};

static const struct axp_data axp202_data = {
        .power_desc     = &axp20x_usb_power_desc,
        .irq_names      = axp20x_irq_names,
        .num_irq_names  = ARRAY_SIZE(axp20x_irq_names),
        .curr_lim_table = axp20x_usb_curr_lim_table,
        .curr_lim_table_size = ARRAY_SIZE(axp20x_usb_curr_lim_table),
        .curr_lim_fld   = REG_FIELD(AXP20X_VBUS_IPSOUT_MGMT, 0, 1),
        .vbus_valid_bit = REG_FIELD(AXP20X_USB_OTG_STATUS, 2, 2),
        .vbus_mon_bit   = REG_FIELD(AXP20X_VBUS_MON, 3, 3),
        .axp20x_read_vbus = &axp20x_usb_power_poll_vbus,
        .axp20x_cfg_iio_chan = axp20x_configure_iio_channels,
        .axp20x_cfg_adc_reg = axp20x_configure_adc_registers,
};

static const struct axp_data axp221_data = {
        .power_desc     = &axp22x_usb_power_desc,
        .irq_names      = axp22x_irq_names,
        .num_irq_names  = ARRAY_SIZE(axp22x_irq_names),
        .curr_lim_table = axp221_usb_curr_lim_table,
        .curr_lim_table_size = ARRAY_SIZE(axp221_usb_curr_lim_table),
        .curr_lim_fld   = REG_FIELD(AXP20X_VBUS_IPSOUT_MGMT, 0, 1),
        .vbus_needs_polling = true,
        .axp20x_read_vbus = &axp20x_usb_power_poll_vbus,
        .axp20x_cfg_iio_chan = axp20x_configure_iio_channels,
        .axp20x_cfg_adc_reg = axp20x_configure_adc_registers,
};

static const struct axp_data axp223_data = {
        .power_desc     = &axp22x_usb_power_desc,
        .irq_names      = axp22x_irq_names,
        .num_irq_names  = ARRAY_SIZE(axp22x_irq_names),
        .curr_lim_table = axp20x_usb_curr_lim_table,
        .curr_lim_table_size = ARRAY_SIZE(axp20x_usb_curr_lim_table),
        .curr_lim_fld   = REG_FIELD(AXP20X_VBUS_IPSOUT_MGMT, 0, 1),
        .vbus_needs_polling = true,
        .axp20x_read_vbus = &axp20x_usb_power_poll_vbus,
        .axp20x_cfg_iio_chan = axp20x_configure_iio_channels,
        .axp20x_cfg_adc_reg = axp20x_configure_adc_registers,
};

static const struct axp_data axp717_data = {
        .power_desc     = &axp717_usb_power_desc,
        .irq_names      = axp717_irq_names,
        .num_irq_names  = ARRAY_SIZE(axp717_irq_names),
        .curr_lim_fld   = REG_FIELD(AXP717_INPUT_CUR_LIMIT_CTRL, 0, 5),
        .usb_bc_en_bit  = REG_FIELD(AXP717_MODULE_EN_CONTROL_1, 4, 4),
        .usb_bc_det_fld = REG_FIELD(AXP717_BC_DETECT, 5, 7),
        .vbus_mon_bit   = REG_FIELD(AXP717_ADC_CH_EN_CONTROL, 2, 2),
        .vbus_needs_polling = false,
        .axp20x_read_vbus = &axp717_usb_power_poll_vbus,
        .axp20x_cfg_iio_chan = axp717_configure_iio_channels,
        .axp20x_cfg_adc_reg = axp717_configure_adc_registers,
};

static const struct axp_data axp813_data = {
        .power_desc     = &axp813_usb_power_desc,
        .irq_names      = axp22x_irq_names,
        .num_irq_names  = ARRAY_SIZE(axp22x_irq_names),
        .curr_lim_table = axp813_usb_curr_lim_table,
        .curr_lim_table_size = ARRAY_SIZE(axp813_usb_curr_lim_table),
        .curr_lim_fld   = REG_FIELD(AXP22X_CHRG_CTRL3, 4, 7),
        .usb_bc_en_bit  = REG_FIELD(AXP288_BC_GLOBAL, 0, 0),
        .usb_bc_det_fld = REG_FIELD(AXP288_BC_DET_STAT, 5, 7),
        .vbus_disable_bit = REG_FIELD(AXP20X_VBUS_IPSOUT_MGMT, 7, 7),
        .vbus_needs_polling = true,
        .axp20x_read_vbus = &axp20x_usb_power_poll_vbus,
        .axp20x_cfg_iio_chan = axp20x_configure_iio_channels,
        .axp20x_cfg_adc_reg = axp20x_configure_adc_registers,
};

#ifdef CONFIG_PM_SLEEP
static int axp20x_usb_power_suspend(struct device *dev)
{
        struct axp20x_usb_power *power = dev_get_drvdata(dev);
        int i = 0;

        /*
         * Allow wake via VBUS_PLUGIN only.
         *
         * As nested threaded IRQs are not automatically disabled during
         * suspend, we must explicitly disable the remainder of the IRQs.
         */
        if (device_may_wakeup(&power->supply->dev))
                enable_irq_wake(power->irqs[i++]);
        while (i < power->num_irqs)
                disable_irq(power->irqs[i++]);

        return 0;
}

static int axp20x_usb_power_resume(struct device *dev)
{
        struct axp20x_usb_power *power = dev_get_drvdata(dev);
        int i = 0;

        if (device_may_wakeup(&power->supply->dev))
                disable_irq_wake(power->irqs[i++]);
        while (i < power->num_irqs)
                enable_irq(power->irqs[i++]);

        mod_delayed_work(system_power_efficient_wq, &power->vbus_detect, DEBOUNCE_TIME);

        return 0;
}
#endif

static SIMPLE_DEV_PM_OPS(axp20x_usb_power_pm_ops, axp20x_usb_power_suspend,
                                                  axp20x_usb_power_resume);

static int axp20x_regmap_field_alloc_optional(struct device *dev,
                                              struct regmap *regmap,
                                              struct reg_field fdesc,
                                              struct regmap_field **fieldp)
{
        struct regmap_field *field;

        if (fdesc.reg == 0) {
                *fieldp = NULL;
                return 0;
        }

        field = devm_regmap_field_alloc(dev, regmap, fdesc);
        if (IS_ERR(field))
                return PTR_ERR(field);

        *fieldp = field;
        return 0;
}

/* Optionally allow users to specify a maximum charging current. */
static void axp20x_usb_power_parse_dt(struct device *dev,
                                      struct axp20x_usb_power *power)
{
        int ret;

        ret = device_property_read_u32(dev, "input-current-limit-microamp",
                                       &power->max_input_cur);
        if (ret)
                dev_dbg(dev, "%s() no input-current-limit specified\n", __func__);
}

static int axp20x_usb_power_probe(struct platform_device *pdev)
{
        struct axp20x_dev *axp20x = dev_get_drvdata(pdev->dev.parent);
        struct power_supply_config psy_cfg = {};
        struct axp20x_usb_power *power;
        const struct axp_data *axp_data;
        int i, irq, ret;

        if (!of_device_is_available(pdev->dev.of_node))
                return -ENODEV;

        if (!axp20x) {
                dev_err(&pdev->dev, "Parent drvdata not set\n");
                return -EINVAL;
        }

        axp_data = of_device_get_match_data(&pdev->dev);

        power = devm_kzalloc(&pdev->dev,
                             struct_size(power, irqs, axp_data->num_irq_names),
                             GFP_KERNEL);
        if (!power)
                return -ENOMEM;

        platform_set_drvdata(pdev, power);

        power->dev = &pdev->dev;
        power->axp_data = axp_data;
        power->regmap = axp20x->regmap;
        power->num_irqs = axp_data->num_irq_names;

        power->curr_lim_fld = devm_regmap_field_alloc(&pdev->dev, power->regmap,
                                                      axp_data->curr_lim_fld);
        if (IS_ERR(power->curr_lim_fld))
                return PTR_ERR(power->curr_lim_fld);

        axp20x_usb_power_parse_dt(&pdev->dev, power);

        ret = axp20x_regmap_field_alloc_optional(&pdev->dev, power->regmap,
                                                 axp_data->vbus_valid_bit,
                                                 &power->vbus_valid_bit);
        if (ret)
                return ret;

        ret = axp20x_regmap_field_alloc_optional(&pdev->dev, power->regmap,
                                                 axp_data->vbus_mon_bit,
                                                 &power->vbus_mon_bit);
        if (ret)
                return ret;

        ret = axp20x_regmap_field_alloc_optional(&pdev->dev, power->regmap,
                                                 axp_data->usb_bc_en_bit,
                                                 &power->usb_bc_en_bit);
        if (ret)
                return ret;

        ret = axp20x_regmap_field_alloc_optional(&pdev->dev, power->regmap,
                                                 axp_data->usb_bc_det_fld,
                                                 &power->usb_bc_det_fld);
        if (ret)
                return ret;

        ret = axp20x_regmap_field_alloc_optional(&pdev->dev, power->regmap,
                                                 axp_data->vbus_disable_bit,
                                                 &power->vbus_disable_bit);
        if (ret)
                return ret;

        ret = devm_delayed_work_autocancel(&pdev->dev, &power->vbus_detect,
                                           axp_data->axp20x_read_vbus);
        if (ret)
                return ret;

        if (power->vbus_mon_bit) {
                /* Enable vbus valid checking */
                ret = regmap_field_write(power->vbus_mon_bit, 1);
                if (ret)
                        return ret;

                if (IS_ENABLED(CONFIG_AXP20X_ADC))
                        ret = axp_data->axp20x_cfg_iio_chan(pdev, power);
                else
                        ret = axp_data->axp20x_cfg_adc_reg(power);

                if (ret)
                        return ret;
        }

        if (power->usb_bc_en_bit) {
                /* Enable USB Battery Charging specification detection */
                ret = regmap_field_write(power->usb_bc_en_bit, 1);
                if (ret)
                        return ret;
        }

        psy_cfg.fwnode = dev_fwnode(&pdev->dev);
        psy_cfg.drv_data = power;

        power->supply = devm_power_supply_register(&pdev->dev,
                                                   axp_data->power_desc,
                                                   &psy_cfg);
        if (IS_ERR(power->supply))
                return PTR_ERR(power->supply);

        /* Request irqs after registering, as irqs may trigger immediately */
        for (i = 0; i < axp_data->num_irq_names; i++) {
                irq = platform_get_irq_byname(pdev, axp_data->irq_names[i]);
                if (irq < 0)
                        return irq;

                power->irqs[i] = regmap_irq_get_virq(axp20x->regmap_irqc, irq);
                ret = devm_request_any_context_irq(&pdev->dev, power->irqs[i],
                                                   axp20x_usb_power_irq, 0,
                                                   DRVNAME, power);
                if (ret < 0) {
                        dev_err(&pdev->dev, "Error requesting %s IRQ: %d\n",
                                axp_data->irq_names[i], ret);
                        return ret;
                }
        }

        if (axp20x_usb_vbus_needs_polling(power))
                queue_delayed_work(system_power_efficient_wq, &power->vbus_detect, 0);

        return 0;
}

static const struct of_device_id axp20x_usb_power_match[] = {
        {
                .compatible = "x-powers,axp192-usb-power-supply",
                .data = &axp192_data,
        }, {
                .compatible = "x-powers,axp202-usb-power-supply",
                .data = &axp202_data,
        }, {
                .compatible = "x-powers,axp221-usb-power-supply",
                .data = &axp221_data,
        }, {
                .compatible = "x-powers,axp223-usb-power-supply",
                .data = &axp223_data,
        }, {
                .compatible = "x-powers,axp717-usb-power-supply",
                .data = &axp717_data,
        }, {
                .compatible = "x-powers,axp813-usb-power-supply",
                .data = &axp813_data,
        }, { /* sentinel */ }
};
MODULE_DEVICE_TABLE(of, axp20x_usb_power_match);

static struct platform_driver axp20x_usb_power_driver = {
        .probe = axp20x_usb_power_probe,
        .driver = {
                .name           = DRVNAME,
                .of_match_table = axp20x_usb_power_match,
                .pm             = &axp20x_usb_power_pm_ops,
        },
};

module_platform_driver(axp20x_usb_power_driver);

MODULE_AUTHOR("Hans de Goede <hdegoede@redhat.com>");
MODULE_DESCRIPTION("AXP20x PMIC USB power supply status driver");
MODULE_LICENSE("GPL");