2 files changed, 23 insertions, 0 deletions

diff --git a/drivers/power/charger-manager.c b/drivers/power/charger-manager.c
index d1e99e7957d2..526e5c931294 100644
--- a/drivers/power/charger-manager.c
+++ b/drivers/power/charger-manager.c
@@ -1001,6 +1001,21 @@ static void charger_extcon_work(struct work_struct *work)
 {
         struct charger_cable *cable =
                         container_of(work, struct charger_cable, wq);
+        int ret;
+
+        if (cable->attached && cable->min_uA != 0 && cable->max_uA != 0) {
+                ret = regulator_set_current_limit(cable->charger->consumer,
+                                        cable->min_uA, cable->max_uA);
+                if (ret < 0) {
+                        pr_err("Cannot set current limit of %s (%s)\n",
+                                cable->charger->regulator_name, cable->name);
+                        return;
+                }
+
+                pr_info("Set current limit of %s : %duA ~ %duA\n",
+                                        cable->charger->regulator_name,
+                                        cable->min_uA, cable->max_uA);
+        }
 
         try_charger_enable(cable->cm, cable->attached);
 }
diff --git a/include/linux/power/charger-manager.h b/include/linux/power/charger-manager.h
index 6cb9fbc9a153..cd22029e32aa 100644
--- a/include/linux/power/charger-manager.h
+++ b/include/linux/power/charger-manager.h
@@ -94,6 +94,14 @@ struct charger_cable {
         bool attached;
 
         struct charger_regulator *charger;
+
+        /*
+         * Set min/max current of regulator to protect over-current issue
+         * according to a kind of charger cable when cable is attached.
+         */
+        int min_uA;
+        int max_uA;
+
         struct charger_manager *cm;
 };