1 files changed, 81 insertions, 2 deletions

diff --git a/drivers/gpio/gpiolib.c b/drivers/gpio/gpiolib.c
index a971e3d043b..a0fa0b25d21 100644
--- a/drivers/gpio/gpiolib.c
+++ b/drivers/gpio/gpiolib.c
@@ -58,6 +58,8 @@ struct gpio_desc {
 #define FLAG_TRIG_FALL  5       /* trigger on falling edge */
 #define FLAG_TRIG_RISE  6       /* trigger on rising edge */
 #define FLAG_ACTIVE_LOW 7       /* sysfs value has active low */
+#define FLAG_OPEN_DRAIN 8       /* Gpio is open drain type */
+#define FLAG_OPEN_SOURCE 9      /* Gpio is open source type */
 
 #define ID_SHIFT        16      /* add new flags before this one */
 
@@ -1257,6 +1259,8 @@ void gpio_free(unsigned gpio)
                 module_put(desc->chip->owner);
                 clear_bit(FLAG_ACTIVE_LOW, &desc->flags);
                 clear_bit(FLAG_REQUESTED, &desc->flags);
+                clear_bit(FLAG_OPEN_DRAIN, &desc->flags);
+                clear_bit(FLAG_OPEN_SOURCE, &desc->flags);
         } else
                 WARN_ON(extra_checks);
 
@@ -1278,6 +1282,12 @@ int gpio_request_one(unsigned gpio, unsigned long flags, const char *label)
         if (err)
                 return err;
 
+        if (flags & GPIOF_OPEN_DRAIN)
+                set_bit(FLAG_OPEN_DRAIN, &gpio_desc[gpio].flags);
+
+        if (flags & GPIOF_OPEN_SOURCE)
+                set_bit(FLAG_OPEN_SOURCE, &gpio_desc[gpio].flags);
+
         if (flags & GPIOF_DIR_IN)
                 err = gpio_direction_input(gpio);
         else
@@ -1427,6 +1437,14 @@ int gpio_direction_output(unsigned gpio, int value)
         struct gpio_desc        *desc = &gpio_desc[gpio];
         int                     status = -EINVAL;
 
+        /* Open drain pin should not be driven to 1 */
+        if (value && test_bit(FLAG_OPEN_DRAIN,  &desc->flags))
+                return gpio_direction_input(gpio);
+
+        /* Open source pin should not be driven to 0 */
+        if (!value && test_bit(FLAG_OPEN_SOURCE,  &desc->flags))
+                return gpio_direction_input(gpio);
+
         spin_lock_irqsave(&gpio_lock, flags);
 
         if (!gpio_is_valid(gpio))
@@ -1563,6 +1581,57 @@ int __gpio_get_value(unsigned gpio)
 }
 EXPORT_SYMBOL_GPL(__gpio_get_value);
 
+/*
+ *  _gpio_set_open_drain_value() - Set the open drain gpio's value.
+ * @gpio: Gpio whose state need to be set.
+ * @chip: Gpio chip.
+ * @value: Non-zero for setting it HIGH otherise it will set to LOW.
+ */
+static void _gpio_set_open_drain_value(unsigned gpio,
+                        struct gpio_chip *chip, int value)
+{
+        int err = 0;
+        if (value) {
+                err = chip->direction_input(chip, gpio - chip->base);
+                if (!err)
+                        clear_bit(FLAG_IS_OUT, &gpio_desc[gpio].flags);
+        } else {
+                err = chip->direction_output(chip, gpio - chip->base, 0);
+                if (!err)
+                        set_bit(FLAG_IS_OUT, &gpio_desc[gpio].flags);
+        }
+        trace_gpio_direction(gpio, value, err);
+        if (err < 0)
+                pr_err("%s: Error in set_value for open drain gpio%d err %d\n",
+                                        __func__, gpio, err);
+}
+
+/*
+ *  _gpio_set_open_source() - Set the open source gpio's value.
+ * @gpio: Gpio whose state need to be set.
+ * @chip: Gpio chip.
+ * @value: Non-zero for setting it HIGH otherise it will set to LOW.
+ */
+static void _gpio_set_open_source_value(unsigned gpio,
+                        struct gpio_chip *chip, int value)
+{
+        int err = 0;
+        if (value) {
+                err = chip->direction_output(chip, gpio - chip->base, 1);
+                if (!err)
+                        set_bit(FLAG_IS_OUT, &gpio_desc[gpio].flags);
+        } else {
+                err = chip->direction_input(chip, gpio - chip->base);
+                if (!err)
+                        clear_bit(FLAG_IS_OUT, &gpio_desc[gpio].flags);
+        }
+        trace_gpio_direction(gpio, !value, err);
+        if (err < 0)
+                pr_err("%s: Error in set_value for open source gpio%d err %d\n",
+                                        __func__, gpio, err);
+}
+
+
 /**
  * __gpio_set_value() - assign a gpio's value
  * @gpio: gpio whose value will be assigned
@@ -1579,7 +1648,12 @@ void __gpio_set_value(unsigned gpio, int value)
         chip = gpio_to_chip(gpio);
         WARN_ON(chip->can_sleep);
         trace_gpio_value(gpio, 0, value);
-        chip->set(chip, gpio - chip->base, value);
+        if (test_bit(FLAG_OPEN_DRAIN,  &gpio_desc[gpio].flags))
+                _gpio_set_open_drain_value(gpio, chip, value);
+        else if (test_bit(FLAG_OPEN_SOURCE,  &gpio_desc[gpio].flags))
+                _gpio_set_open_source_value(gpio, chip, value);
+        else
+                chip->set(chip, gpio - chip->base, value);
 }
 EXPORT_SYMBOL_GPL(__gpio_set_value);
 
@@ -1646,7 +1720,12 @@ void gpio_set_value_cansleep(unsigned gpio, int value)
         might_sleep_if(extra_checks);
         chip = gpio_to_chip(gpio);
         trace_gpio_value(gpio, 0, value);
-        chip->set(chip, gpio - chip->base, value);
+        if (test_bit(FLAG_OPEN_DRAIN,  &gpio_desc[gpio].flags))
+                _gpio_set_open_drain_value(gpio, chip, value);
+        else if (test_bit(FLAG_OPEN_SOURCE,  &gpio_desc[gpio].flags))
+                _gpio_set_open_source_value(gpio, chip, value);
+        else
+                chip->set(chip, gpio - chip->base, value);
 }
 EXPORT_SYMBOL_GPL(gpio_set_value_cansleep);