drivers/pinctrl/pinctrl-at91.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * at91 pinctrl driver based on at91 pinmux core
   4  *
   5  * Copyright (C) 2011-2012 Jean-Christophe PLAGNIOL-VILLARD <plagnioj@jcrosoft.com>
   6  */
   7
   8 #include <linux/clk.h>
   9 #include <linux/err.h>
  10 #include <linux/init.h>
  11 #include <linux/of.h>
  12 #include <linux/of_device.h>
  13 #include <linux/of_address.h>
  14 #include <linux/of_irq.h>
  15 #include <linux/slab.h>
  16 #include <linux/interrupt.h>
  17 #include <linux/io.h>
  18 #include <linux/gpio/driver.h>
  19 #include <linux/pinctrl/machine.h>
  20 #include <linux/pinctrl/pinconf.h>
  21 #include <linux/pinctrl/pinctrl.h>
  22 #include <linux/pinctrl/pinmux.h>
  23 /* Since we request GPIOs from ourself */
  24 #include <linux/pinctrl/consumer.h>
  25
  26 #include "pinctrl-at91.h"
  27 #include "core.h"
  28
  29 #define MAX_GPIO_BANKS          5
  30 #define MAX_NB_GPIO_PER_BANK    32
  31
  32 struct at91_pinctrl_mux_ops;
  33
  34 struct at91_gpio_chip {
  35         struct gpio_chip        chip;
  36         struct pinctrl_gpio_range range;
  37         struct at91_gpio_chip   *next;          /* Bank sharing same clock */
  38         int                     pioc_hwirq;     /* PIO bank interrupt identifier on AIC */
  39         int                     pioc_virq;      /* PIO bank Linux virtual interrupt */
  40         int                     pioc_idx;       /* PIO bank index */
  41         void __iomem            *regbase;       /* PIO bank virtual address */
  42         struct clk              *clock;         /* associated clock */
  43         struct at91_pinctrl_mux_ops *ops;       /* ops */
  44 };
  45
  46 static struct at91_gpio_chip *gpio_chips[MAX_GPIO_BANKS];
  47
  48 static int gpio_banks;
  49
  50 #define PULL_UP         (1 << 0)
  51 #define MULTI_DRIVE     (1 << 1)
  52 #define DEGLITCH        (1 << 2)
  53 #define PULL_DOWN       (1 << 3)
  54 #define DIS_SCHMIT      (1 << 4)
  55 #define DRIVE_STRENGTH_SHIFT    5
  56 #define DRIVE_STRENGTH_MASK             0x3
  57 #define DRIVE_STRENGTH   (DRIVE_STRENGTH_MASK << DRIVE_STRENGTH_SHIFT)
  58 #define OUTPUT          (1 << 7)
  59 #define OUTPUT_VAL_SHIFT        8
  60 #define OUTPUT_VAL      (0x1 << OUTPUT_VAL_SHIFT)
  61 #define SLEWRATE_SHIFT  9
  62 #define SLEWRATE_MASK   0x1
  63 #define SLEWRATE        (SLEWRATE_MASK << SLEWRATE_SHIFT)
  64 #define DEBOUNCE        (1 << 16)
  65 #define DEBOUNCE_VAL_SHIFT      17
  66 #define DEBOUNCE_VAL    (0x3fff << DEBOUNCE_VAL_SHIFT)
  67
  68 /*
  69  * These defines will translated the dt binding settings to our internal
  70  * settings. They are not necessarily the same value as the register setting.
  71  * The actual drive strength current of low, medium and high must be looked up
  72  * from the corresponding device datasheet. This value is different for pins
  73  * that are even in the same banks. It is also dependent on VCC.
  74  * DRIVE_STRENGTH_DEFAULT is just a placeholder to avoid changing the drive
  75  * strength when there is no dt config for it.
  76  */
  77 enum drive_strength_bit {
  78         DRIVE_STRENGTH_BIT_DEF,
  79         DRIVE_STRENGTH_BIT_LOW,
  80         DRIVE_STRENGTH_BIT_MED,
  81         DRIVE_STRENGTH_BIT_HI,
  82 };
  83
  84 #define DRIVE_STRENGTH_BIT_MSK(name)    (DRIVE_STRENGTH_BIT_##name << \
  85                                          DRIVE_STRENGTH_SHIFT)
  86
  87 enum slewrate_bit {
  88         SLEWRATE_BIT_ENA,
  89         SLEWRATE_BIT_DIS,
  90 };
  91
  92 #define SLEWRATE_BIT_MSK(name)          (SLEWRATE_BIT_##name << SLEWRATE_SHIFT)
  93
  94 /**
  95  * struct at91_pmx_func - describes AT91 pinmux functions
  96  * @name: the name of this specific function
  97  * @groups: corresponding pin groups
  98  * @ngroups: the number of groups
  99  */
 100 struct at91_pmx_func {
 101         const char      *name;
 102         const char      **groups;
 103         unsigned        ngroups;
 104 };
 105
 106 enum at91_mux {
 107         AT91_MUX_GPIO = 0,
 108         AT91_MUX_PERIPH_A = 1,
 109         AT91_MUX_PERIPH_B = 2,
 110         AT91_MUX_PERIPH_C = 3,
 111         AT91_MUX_PERIPH_D = 4,
 112 };
 113
 114 /**
 115  * struct at91_pmx_pin - describes an At91 pin mux
 116  * @bank: the bank of the pin
 117  * @pin: the pin number in the @bank
 118  * @mux: the mux mode : gpio or periph_x of the pin i.e. alternate function.
 119  * @conf: the configuration of the pin: PULL_UP, MULTIDRIVE etc...
 120  */
 121 struct at91_pmx_pin {
 122         uint32_t        bank;
 123         uint32_t        pin;
 124         enum at91_mux   mux;
 125         unsigned long   conf;
 126 };
 127
 128 /**
 129  * struct at91_pin_group - describes an At91 pin group
 130  * @name: the name of this specific pin group
 131  * @pins_conf: the mux mode for each pin in this group. The size of this
 132  *      array is the same as pins.
 133  * @pins: an array of discrete physical pins used in this group, taken
 134  *      from the driver-local pin enumeration space
 135  * @npins: the number of pins in this group array, i.e. the number of
 136  *      elements in .pins so we can iterate over that array
 137  */
 138 struct at91_pin_group {
 139         const char              *name;
 140         struct at91_pmx_pin     *pins_conf;
 141         unsigned int            *pins;
 142         unsigned                npins;
 143 };
 144
 145 /**
 146  * struct at91_pinctrl_mux_ops - describes an AT91 mux ops group
 147  * on new IP with support for periph C and D the way to mux in
 148  * periph A and B has changed
 149  * So provide the right call back
 150  * if not present means the IP does not support it
 151  * @get_periph: return the periph mode configured
 152  * @mux_A_periph: mux as periph A
 153  * @mux_B_periph: mux as periph B
 154  * @mux_C_periph: mux as periph C
 155  * @mux_D_periph: mux as periph D
 156  * @get_deglitch: get deglitch status
 157  * @set_deglitch: enable/disable deglitch
 158  * @get_debounce: get debounce status
 159  * @set_debounce: enable/disable debounce
 160  * @get_pulldown: get pulldown status
 161  * @set_pulldown: enable/disable pulldown
 162  * @get_schmitt_trig: get schmitt trigger status
 163  * @disable_schmitt_trig: disable schmitt trigger
 164  * @get_drivestrength: get driver strength
 165  * @set_drivestrength: set driver strength
 166  * @get_slewrate: get slew rate
 167  * @set_slewrate: set slew rate
 168  * @irq_type: return irq type
 169  */
 170 struct at91_pinctrl_mux_ops {
 171         enum at91_mux (*get_periph)(void __iomem *pio, unsigned mask);
 172         void (*mux_A_periph)(void __iomem *pio, unsigned mask);
 173         void (*mux_B_periph)(void __iomem *pio, unsigned mask);
 174         void (*mux_C_periph)(void __iomem *pio, unsigned mask);
 175         void (*mux_D_periph)(void __iomem *pio, unsigned mask);
 176         bool (*get_deglitch)(void __iomem *pio, unsigned pin);
 177         void (*set_deglitch)(void __iomem *pio, unsigned mask, bool is_on);
 178         bool (*get_debounce)(void __iomem *pio, unsigned pin, u32 *div);
 179         void (*set_debounce)(void __iomem *pio, unsigned mask, bool is_on, u32 div);
 180         bool (*get_pulldown)(void __iomem *pio, unsigned pin);
 181         void (*set_pulldown)(void __iomem *pio, unsigned mask, bool is_on);
 182         bool (*get_schmitt_trig)(void __iomem *pio, unsigned pin);
 183         void (*disable_schmitt_trig)(void __iomem *pio, unsigned mask);
 184         unsigned (*get_drivestrength)(void __iomem *pio, unsigned pin);
 185         void (*set_drivestrength)(void __iomem *pio, unsigned pin,
 186                                         u32 strength);
 187         unsigned (*get_slewrate)(void __iomem *pio, unsigned pin);
 188         void (*set_slewrate)(void __iomem *pio, unsigned pin, u32 slewrate);
 189         /* irq */
 190         int (*irq_type)(struct irq_data *d, unsigned type);
 191 };
 192
 193 static int gpio_irq_type(struct irq_data *d, unsigned type);
 194 static int alt_gpio_irq_type(struct irq_data *d, unsigned type);
 195
 196 struct at91_pinctrl {
 197         struct device           *dev;
 198         struct pinctrl_dev      *pctl;
 199
 200         int                     nactive_banks;
 201
 202         uint32_t                *mux_mask;
 203         int                     nmux;
 204
 205         struct at91_pmx_func    *functions;
 206         int                     nfunctions;
 207
 208         struct at91_pin_group   *groups;
 209         int                     ngroups;
 210
 211         struct at91_pinctrl_mux_ops *ops;
 212 };
 213
 214 static inline const struct at91_pin_group *at91_pinctrl_find_group_by_name(
 215                                 const struct at91_pinctrl *info,
 216                                 const char *name)
 217 {
 218         const struct at91_pin_group *grp = NULL;
 219         int i;
 220
 221         for (i = 0; i < info->ngroups; i++) {
 222                 if (strcmp(info->groups[i].name, name))
 223                         continue;
 224
 225                 grp = &info->groups[i];
 226                 dev_dbg(info->dev, "%s: %d 0:%d\n", name, grp->npins, grp->pins[0]);
 227                 break;
 228         }
 229
 230         return grp;
 231 }
 232
 233 static int at91_get_groups_count(struct pinctrl_dev *pctldev)
 234 {
 235         struct at91_pinctrl *info = pinctrl_dev_get_drvdata(pctldev);
 236
 237         return info->ngroups;
 238 }
 239
 240 static const char *at91_get_group_name(struct pinctrl_dev *pctldev,
 241                                        unsigned selector)
 242 {
 243         struct at91_pinctrl *info = pinctrl_dev_get_drvdata(pctldev);
 244
 245         return info->groups[selector].name;
 246 }
 247
 248 static int at91_get_group_pins(struct pinctrl_dev *pctldev, unsigned selector,
 249                                const unsigned **pins,
 250                                unsigned *npins)
 251 {
 252         struct at91_pinctrl *info = pinctrl_dev_get_drvdata(pctldev);
 253
 254         if (selector >= info->ngroups)
 255                 return -EINVAL;
 256
 257         *pins = info->groups[selector].pins;
 258         *npins = info->groups[selector].npins;
 259
 260         return 0;
 261 }
 262
 263 static void at91_pin_dbg_show(struct pinctrl_dev *pctldev, struct seq_file *s,
 264                    unsigned offset)
 265 {
 266         seq_printf(s, "%s", dev_name(pctldev->dev));
 267 }
 268
 269 static int at91_dt_node_to_map(struct pinctrl_dev *pctldev,
 270                         struct device_node *np,
 271                         struct pinctrl_map **map, unsigned *num_maps)
 272 {
 273         struct at91_pinctrl *info = pinctrl_dev_get_drvdata(pctldev);
 274         const struct at91_pin_group *grp;
 275         struct pinctrl_map *new_map;
 276         struct device_node *parent;
 277         int map_num = 1;
 278         int i;
 279
 280         /*
 281          * first find the group of this node and check if we need to create
 282          * config maps for pins
 283          */
 284         grp = at91_pinctrl_find_group_by_name(info, np->name);
 285         if (!grp) {
 286                 dev_err(info->dev, "unable to find group for node %pOFn\n",
 287                         np);
 288                 return -EINVAL;
 289         }
 290
 291         map_num += grp->npins;
 292         new_map = devm_kcalloc(pctldev->dev, map_num, sizeof(*new_map),
 293                                GFP_KERNEL);
 294         if (!new_map)
 295                 return -ENOMEM;
 296
 297         *map = new_map;
 298         *num_maps = map_num;
 299
 300         /* create mux map */
 301         parent = of_get_parent(np);
 302         if (!parent) {
 303                 devm_kfree(pctldev->dev, new_map);
 304                 return -EINVAL;
 305         }
 306         new_map[0].type = PIN_MAP_TYPE_MUX_GROUP;
 307         new_map[0].data.mux.function = parent->name;
 308         new_map[0].data.mux.group = np->name;
 309         of_node_put(parent);
 310
 311         /* create config map */
 312         new_map++;
 313         for (i = 0; i < grp->npins; i++) {
 314                 new_map[i].type = PIN_MAP_TYPE_CONFIGS_PIN;
 315                 new_map[i].data.configs.group_or_pin =
 316                                 pin_get_name(pctldev, grp->pins[i]);
 317                 new_map[i].data.configs.configs = &grp->pins_conf[i].conf;
 318                 new_map[i].data.configs.num_configs = 1;
 319         }
 320
 321         dev_dbg(pctldev->dev, "maps: function %s group %s num %d\n",
 322                 (*map)->data.mux.function, (*map)->data.mux.group, map_num);
 323
 324         return 0;
 325 }
 326
 327 static void at91_dt_free_map(struct pinctrl_dev *pctldev,
 328                                 struct pinctrl_map *map, unsigned num_maps)
 329 {
 330 }
 331
 332 static const struct pinctrl_ops at91_pctrl_ops = {
 333         .get_groups_count       = at91_get_groups_count,
 334         .get_group_name         = at91_get_group_name,
 335         .get_group_pins         = at91_get_group_pins,
 336         .pin_dbg_show           = at91_pin_dbg_show,
 337         .dt_node_to_map         = at91_dt_node_to_map,
 338         .dt_free_map            = at91_dt_free_map,
 339 };
 340
 341 static void __iomem *pin_to_controller(struct at91_pinctrl *info,
 342                                  unsigned int bank)
 343 {
 344         if (!gpio_chips[bank])
 345                 return NULL;
 346
 347         return gpio_chips[bank]->regbase;
 348 }
 349
 350 static inline int pin_to_bank(unsigned pin)
 351 {
 352         return pin /= MAX_NB_GPIO_PER_BANK;
 353 }
 354
 355 static unsigned pin_to_mask(unsigned int pin)
 356 {
 357         return 1 << pin;
 358 }
 359
 360 static unsigned two_bit_pin_value_shift_amount(unsigned int pin)
 361 {
 362         /* return the shift value for a pin for "two bit" per pin registers,
 363          * i.e. drive strength */
 364         return 2*((pin >= MAX_NB_GPIO_PER_BANK/2)
 365                         ? pin - MAX_NB_GPIO_PER_BANK/2 : pin);
 366 }
 367
 368 static unsigned sama5d3_get_drive_register(unsigned int pin)
 369 {
 370         /* drive strength is split between two registers
 371          * with two bits per pin */
 372         return (pin >= MAX_NB_GPIO_PER_BANK/2)
 373                         ? SAMA5D3_PIO_DRIVER2 : SAMA5D3_PIO_DRIVER1;
 374 }
 375
 376 static unsigned at91sam9x5_get_drive_register(unsigned int pin)
 377 {
 378         /* drive strength is split between two registers
 379          * with two bits per pin */
 380         return (pin >= MAX_NB_GPIO_PER_BANK/2)
 381                         ? AT91SAM9X5_PIO_DRIVER2 : AT91SAM9X5_PIO_DRIVER1;
 382 }
 383
 384 static void at91_mux_disable_interrupt(void __iomem *pio, unsigned mask)
 385 {
 386         writel_relaxed(mask, pio + PIO_IDR);
 387 }
 388
 389 static unsigned at91_mux_get_pullup(void __iomem *pio, unsigned pin)
 390 {
 391         return !((readl_relaxed(pio + PIO_PUSR) >> pin) & 0x1);
 392 }
 393
 394 static void at91_mux_set_pullup(void __iomem *pio, unsigned mask, bool on)
 395 {
 396         if (on)
 397                 writel_relaxed(mask, pio + PIO_PPDDR);
 398
 399         writel_relaxed(mask, pio + (on ? PIO_PUER : PIO_PUDR));
 400 }
 401
 402 static bool at91_mux_get_output(void __iomem *pio, unsigned int pin, bool *val)
 403 {
 404         *val = (readl_relaxed(pio + PIO_ODSR) >> pin) & 0x1;
 405         return (readl_relaxed(pio + PIO_OSR) >> pin) & 0x1;
 406 }
 407
 408 static void at91_mux_set_output(void __iomem *pio, unsigned int mask,
 409                                 bool is_on, bool val)
 410 {
 411         writel_relaxed(mask, pio + (val ? PIO_SODR : PIO_CODR));
 412         writel_relaxed(mask, pio + (is_on ? PIO_OER : PIO_ODR));
 413 }
 414
 415 static unsigned at91_mux_get_multidrive(void __iomem *pio, unsigned pin)
 416 {
 417         return (readl_relaxed(pio + PIO_MDSR) >> pin) & 0x1;
 418 }
 419
 420 static void at91_mux_set_multidrive(void __iomem *pio, unsigned mask, bool on)
 421 {
 422         writel_relaxed(mask, pio + (on ? PIO_MDER : PIO_MDDR));
 423 }
 424
 425 static void at91_mux_set_A_periph(void __iomem *pio, unsigned mask)
 426 {
 427         writel_relaxed(mask, pio + PIO_ASR);
 428 }
 429
 430 static void at91_mux_set_B_periph(void __iomem *pio, unsigned mask)
 431 {
 432         writel_relaxed(mask, pio + PIO_BSR);
 433 }
 434
 435 static void at91_mux_pio3_set_A_periph(void __iomem *pio, unsigned mask)
 436 {
 437
 438         writel_relaxed(readl_relaxed(pio + PIO_ABCDSR1) & ~mask,
 439                                                 pio + PIO_ABCDSR1);
 440         writel_relaxed(readl_relaxed(pio + PIO_ABCDSR2) & ~mask,
 441                                                 pio + PIO_ABCDSR2);
 442 }
 443
 444 static void at91_mux_pio3_set_B_periph(void __iomem *pio, unsigned mask)
 445 {
 446         writel_relaxed(readl_relaxed(pio + PIO_ABCDSR1) | mask,
 447                                                 pio + PIO_ABCDSR1);
 448         writel_relaxed(readl_relaxed(pio + PIO_ABCDSR2) & ~mask,
 449                                                 pio + PIO_ABCDSR2);
 450 }
 451
 452 static void at91_mux_pio3_set_C_periph(void __iomem *pio, unsigned mask)
 453 {
 454         writel_relaxed(readl_relaxed(pio + PIO_ABCDSR1) & ~mask, pio + PIO_ABCDSR1);
 455         writel_relaxed(readl_relaxed(pio + PIO_ABCDSR2) | mask, pio + PIO_ABCDSR2);
 456 }
 457
 458 static void at91_mux_pio3_set_D_periph(void __iomem *pio, unsigned mask)
 459 {
 460         writel_relaxed(readl_relaxed(pio + PIO_ABCDSR1) | mask, pio + PIO_ABCDSR1);
 461         writel_relaxed(readl_relaxed(pio + PIO_ABCDSR2) | mask, pio + PIO_ABCDSR2);
 462 }
 463
 464 static enum at91_mux at91_mux_pio3_get_periph(void __iomem *pio, unsigned mask)
 465 {
 466         unsigned select;
 467
 468         if (readl_relaxed(pio + PIO_PSR) & mask)
 469                 return AT91_MUX_GPIO;
 470
 471         select = !!(readl_relaxed(pio + PIO_ABCDSR1) & mask);
 472         select |= (!!(readl_relaxed(pio + PIO_ABCDSR2) & mask) << 1);
 473
 474         return select + 1;
 475 }
 476
 477 static enum at91_mux at91_mux_get_periph(void __iomem *pio, unsigned mask)
 478 {
 479         unsigned select;
 480
 481         if (readl_relaxed(pio + PIO_PSR) & mask)
 482                 return AT91_MUX_GPIO;
 483
 484         select = readl_relaxed(pio + PIO_ABSR) & mask;
 485
 486         return select + 1;
 487 }
 488
 489 static bool at91_mux_get_deglitch(void __iomem *pio, unsigned pin)
 490 {
 491         return (readl_relaxed(pio + PIO_IFSR) >> pin) & 0x1;
 492 }
 493
 494 static void at91_mux_set_deglitch(void __iomem *pio, unsigned mask, bool is_on)
 495 {
 496         writel_relaxed(mask, pio + (is_on ? PIO_IFER : PIO_IFDR));
 497 }
 498
 499 static bool at91_mux_pio3_get_deglitch(void __iomem *pio, unsigned pin)
 500 {
 501         if ((readl_relaxed(pio + PIO_IFSR) >> pin) & 0x1)
 502                 return !((readl_relaxed(pio + PIO_IFSCSR) >> pin) & 0x1);
 503
 504         return false;
 505 }
 506
 507 static void at91_mux_pio3_set_deglitch(void __iomem *pio, unsigned mask, bool is_on)
 508 {
 509         if (is_on)
 510                 writel_relaxed(mask, pio + PIO_IFSCDR);
 511         at91_mux_set_deglitch(pio, mask, is_on);
 512 }
 513
 514 static bool at91_mux_pio3_get_debounce(void __iomem *pio, unsigned pin, u32 *div)
 515 {
 516         *div = readl_relaxed(pio + PIO_SCDR);
 517
 518         return ((readl_relaxed(pio + PIO_IFSR) >> pin) & 0x1) &&
 519                ((readl_relaxed(pio + PIO_IFSCSR) >> pin) & 0x1);
 520 }
 521
 522 static void at91_mux_pio3_set_debounce(void __iomem *pio, unsigned mask,
 523                                 bool is_on, u32 div)
 524 {
 525         if (is_on) {
 526                 writel_relaxed(mask, pio + PIO_IFSCER);
 527                 writel_relaxed(div & PIO_SCDR_DIV, pio + PIO_SCDR);
 528                 writel_relaxed(mask, pio + PIO_IFER);
 529         } else
 530                 writel_relaxed(mask, pio + PIO_IFSCDR);
 531 }
 532
 533 static bool at91_mux_pio3_get_pulldown(void __iomem *pio, unsigned pin)
 534 {
 535         return !((readl_relaxed(pio + PIO_PPDSR) >> pin) & 0x1);
 536 }
 537
 538 static void at91_mux_pio3_set_pulldown(void __iomem *pio, unsigned mask, bool is_on)
 539 {
 540         if (is_on)
 541                 writel_relaxed(mask, pio + PIO_PUDR);
 542
 543         writel_relaxed(mask, pio + (is_on ? PIO_PPDER : PIO_PPDDR));
 544 }
 545
 546 static void at91_mux_pio3_disable_schmitt_trig(void __iomem *pio, unsigned mask)
 547 {
 548         writel_relaxed(readl_relaxed(pio + PIO_SCHMITT) | mask, pio + PIO_SCHMITT);
 549 }
 550
 551 static bool at91_mux_pio3_get_schmitt_trig(void __iomem *pio, unsigned pin)
 552 {
 553         return (readl_relaxed(pio + PIO_SCHMITT) >> pin) & 0x1;
 554 }
 555
 556 static inline u32 read_drive_strength(void __iomem *reg, unsigned pin)
 557 {
 558         unsigned tmp = readl_relaxed(reg);
 559
 560         tmp = tmp >> two_bit_pin_value_shift_amount(pin);
 561
 562         return tmp & DRIVE_STRENGTH_MASK;
 563 }
 564
 565 static unsigned at91_mux_sama5d3_get_drivestrength(void __iomem *pio,
 566                                                         unsigned pin)
 567 {
 568         unsigned tmp = read_drive_strength(pio +
 569                                         sama5d3_get_drive_register(pin), pin);
 570
 571         /* SAMA5 strength is 1:1 with our defines,
 572          * except 0 is equivalent to low per datasheet */
 573         if (!tmp)
 574                 tmp = DRIVE_STRENGTH_BIT_MSK(LOW);
 575
 576         return tmp;
 577 }
 578
 579 static unsigned at91_mux_sam9x5_get_drivestrength(void __iomem *pio,
 580                                                         unsigned pin)
 581 {
 582         unsigned tmp = read_drive_strength(pio +
 583                                 at91sam9x5_get_drive_register(pin), pin);
 584
 585         /* strength is inverse in SAM9x5s hardware with the pinctrl defines
 586          * hardware: 0 = hi, 1 = med, 2 = low, 3 = rsvd */
 587         tmp = DRIVE_STRENGTH_BIT_MSK(HI) - tmp;
 588
 589         return tmp;
 590 }
 591
 592 static unsigned at91_mux_sam9x60_get_drivestrength(void __iomem *pio,
 593                                                    unsigned pin)
 594 {
 595         unsigned tmp = readl_relaxed(pio + SAM9X60_PIO_DRIVER1);
 596
 597         if (tmp & BIT(pin))
 598                 return DRIVE_STRENGTH_BIT_HI;
 599
 600         return DRIVE_STRENGTH_BIT_LOW;
 601 }
 602
 603 static unsigned at91_mux_sam9x60_get_slewrate(void __iomem *pio, unsigned pin)
 604 {
 605         unsigned tmp = readl_relaxed(pio + SAM9X60_PIO_SLEWR);
 606
 607         if ((tmp & BIT(pin)))
 608                 return SLEWRATE_BIT_ENA;
 609
 610         return SLEWRATE_BIT_DIS;
 611 }
 612
 613 static void set_drive_strength(void __iomem *reg, unsigned pin, u32 strength)
 614 {
 615         unsigned tmp = readl_relaxed(reg);
 616         unsigned shift = two_bit_pin_value_shift_amount(pin);
 617
 618         tmp &= ~(DRIVE_STRENGTH_MASK  <<  shift);
 619         tmp |= strength << shift;
 620
 621         writel_relaxed(tmp, reg);
 622 }
 623
 624 static void at91_mux_sama5d3_set_drivestrength(void __iomem *pio, unsigned pin,
 625                                                 u32 setting)
 626 {
 627         /* do nothing if setting is zero */
 628         if (!setting)
 629                 return;
 630
 631         /* strength is 1 to 1 with setting for SAMA5 */
 632         set_drive_strength(pio + sama5d3_get_drive_register(pin), pin, setting);
 633 }
 634
 635 static void at91_mux_sam9x5_set_drivestrength(void __iomem *pio, unsigned pin,
 636                                                 u32 setting)
 637 {
 638         /* do nothing if setting is zero */
 639         if (!setting)
 640                 return;
 641
 642         /* strength is inverse on SAM9x5s with our defines
 643          * 0 = hi, 1 = med, 2 = low, 3 = rsvd */
 644         setting = DRIVE_STRENGTH_BIT_MSK(HI) - setting;
 645
 646         set_drive_strength(pio + at91sam9x5_get_drive_register(pin), pin,
 647                                 setting);
 648 }
 649
 650 static void at91_mux_sam9x60_set_drivestrength(void __iomem *pio, unsigned pin,
 651                                                u32 setting)
 652 {
 653         unsigned int tmp;
 654
 655         if (setting <= DRIVE_STRENGTH_BIT_DEF ||
 656             setting == DRIVE_STRENGTH_BIT_MED ||
 657             setting > DRIVE_STRENGTH_BIT_HI)
 658                 return;
 659
 660         tmp = readl_relaxed(pio + SAM9X60_PIO_DRIVER1);
 661
 662         /* Strength is 0: low, 1: hi */
 663         if (setting == DRIVE_STRENGTH_BIT_LOW)
 664                 tmp &= ~BIT(pin);
 665         else
 666                 tmp |= BIT(pin);
 667
 668         writel_relaxed(tmp, pio + SAM9X60_PIO_DRIVER1);
 669 }
 670
 671 static void at91_mux_sam9x60_set_slewrate(void __iomem *pio, unsigned pin,
 672                                           u32 setting)
 673 {
 674         unsigned int tmp;
 675
 676         if (setting < SLEWRATE_BIT_ENA || setting > SLEWRATE_BIT_DIS)
 677                 return;
 678
 679         tmp = readl_relaxed(pio + SAM9X60_PIO_SLEWR);
 680
 681         if (setting == SLEWRATE_BIT_DIS)
 682                 tmp &= ~BIT(pin);
 683         else
 684                 tmp |= BIT(pin);
 685
 686         writel_relaxed(tmp, pio + SAM9X60_PIO_SLEWR);
 687 }
 688
 689 static struct at91_pinctrl_mux_ops at91rm9200_ops = {
 690         .get_periph     = at91_mux_get_periph,
 691         .mux_A_periph   = at91_mux_set_A_periph,
 692         .mux_B_periph   = at91_mux_set_B_periph,
 693         .get_deglitch   = at91_mux_get_deglitch,
 694         .set_deglitch   = at91_mux_set_deglitch,
 695         .irq_type       = gpio_irq_type,
 696 };
 697
 698 static struct at91_pinctrl_mux_ops at91sam9x5_ops = {
 699         .get_periph     = at91_mux_pio3_get_periph,
 700         .mux_A_periph   = at91_mux_pio3_set_A_periph,
 701         .mux_B_periph   = at91_mux_pio3_set_B_periph,
 702         .mux_C_periph   = at91_mux_pio3_set_C_periph,
 703         .mux_D_periph   = at91_mux_pio3_set_D_periph,
 704         .get_deglitch   = at91_mux_pio3_get_deglitch,
 705         .set_deglitch   = at91_mux_pio3_set_deglitch,
 706         .get_debounce   = at91_mux_pio3_get_debounce,
 707         .set_debounce   = at91_mux_pio3_set_debounce,
 708         .get_pulldown   = at91_mux_pio3_get_pulldown,
 709         .set_pulldown   = at91_mux_pio3_set_pulldown,
 710         .get_schmitt_trig = at91_mux_pio3_get_schmitt_trig,
 711         .disable_schmitt_trig = at91_mux_pio3_disable_schmitt_trig,
 712         .get_drivestrength = at91_mux_sam9x5_get_drivestrength,
 713         .set_drivestrength = at91_mux_sam9x5_set_drivestrength,
 714         .irq_type       = alt_gpio_irq_type,
 715 };
 716
 717 static const struct at91_pinctrl_mux_ops sam9x60_ops = {
 718         .get_periph     = at91_mux_pio3_get_periph,
 719         .mux_A_periph   = at91_mux_pio3_set_A_periph,
 720         .mux_B_periph   = at91_mux_pio3_set_B_periph,
 721         .mux_C_periph   = at91_mux_pio3_set_C_periph,
 722         .mux_D_periph   = at91_mux_pio3_set_D_periph,
 723         .get_deglitch   = at91_mux_pio3_get_deglitch,
 724         .set_deglitch   = at91_mux_pio3_set_deglitch,
 725         .get_debounce   = at91_mux_pio3_get_debounce,
 726         .set_debounce   = at91_mux_pio3_set_debounce,
 727         .get_pulldown   = at91_mux_pio3_get_pulldown,
 728         .set_pulldown   = at91_mux_pio3_set_pulldown,
 729         .get_schmitt_trig = at91_mux_pio3_get_schmitt_trig,
 730         .disable_schmitt_trig = at91_mux_pio3_disable_schmitt_trig,
 731         .get_drivestrength = at91_mux_sam9x60_get_drivestrength,
 732         .set_drivestrength = at91_mux_sam9x60_set_drivestrength,
 733         .get_slewrate   = at91_mux_sam9x60_get_slewrate,
 734         .set_slewrate   = at91_mux_sam9x60_set_slewrate,
 735         .irq_type       = alt_gpio_irq_type,
 736
 737 };
 738
 739 static struct at91_pinctrl_mux_ops sama5d3_ops = {
 740         .get_periph     = at91_mux_pio3_get_periph,
 741         .mux_A_periph   = at91_mux_pio3_set_A_periph,
 742         .mux_B_periph   = at91_mux_pio3_set_B_periph,
 743         .mux_C_periph   = at91_mux_pio3_set_C_periph,
 744         .mux_D_periph   = at91_mux_pio3_set_D_periph,
 745         .get_deglitch   = at91_mux_pio3_get_deglitch,
 746         .set_deglitch   = at91_mux_pio3_set_deglitch,
 747         .get_debounce   = at91_mux_pio3_get_debounce,
 748         .set_debounce   = at91_mux_pio3_set_debounce,
 749         .get_pulldown   = at91_mux_pio3_get_pulldown,
 750         .set_pulldown   = at91_mux_pio3_set_pulldown,
 751         .get_schmitt_trig = at91_mux_pio3_get_schmitt_trig,
 752         .disable_schmitt_trig = at91_mux_pio3_disable_schmitt_trig,
 753         .get_drivestrength = at91_mux_sama5d3_get_drivestrength,
 754         .set_drivestrength = at91_mux_sama5d3_set_drivestrength,
 755         .irq_type       = alt_gpio_irq_type,
 756 };
 757
 758 static void at91_pin_dbg(const struct device *dev, const struct at91_pmx_pin *pin)
 759 {
 760         if (pin->mux) {
 761                 dev_dbg(dev, "pio%c%d configured as periph%c with conf = 0x%lx\n",
 762                         pin->bank + 'A', pin->pin, pin->mux - 1 + 'A', pin->conf);
 763         } else {
 764                 dev_dbg(dev, "pio%c%d configured as gpio with conf = 0x%lx\n",
 765                         pin->bank + 'A', pin->pin, pin->conf);
 766         }
 767 }
 768
 769 static int pin_check_config(struct at91_pinctrl *info, const char *name,
 770                             int index, const struct at91_pmx_pin *pin)
 771 {
 772         int mux;
 773
 774         /* check if it's a valid config */
 775         if (pin->bank >= gpio_banks) {
 776                 dev_err(info->dev, "%s: pin conf %d bank_id %d >= nbanks %d\n",
 777                         name, index, pin->bank, gpio_banks);
 778                 return -EINVAL;
 779         }
 780
 781         if (!gpio_chips[pin->bank]) {
 782                 dev_err(info->dev, "%s: pin conf %d bank_id %d not enabled\n",
 783                         name, index, pin->bank);
 784                 return -ENXIO;
 785         }
 786
 787         if (pin->pin >= MAX_NB_GPIO_PER_BANK) {
 788                 dev_err(info->dev, "%s: pin conf %d pin_bank_id %d >= %d\n",
 789                         name, index, pin->pin, MAX_NB_GPIO_PER_BANK);
 790                 return -EINVAL;
 791         }
 792
 793         if (!pin->mux)
 794                 return 0;
 795
 796         mux = pin->mux - 1;
 797
 798         if (mux >= info->nmux) {
 799                 dev_err(info->dev, "%s: pin conf %d mux_id %d >= nmux %d\n",
 800                         name, index, mux, info->nmux);
 801                 return -EINVAL;
 802         }
 803
 804         if (!(info->mux_mask[pin->bank * info->nmux + mux] & 1 << pin->pin)) {
 805                 dev_err(info->dev, "%s: pin conf %d mux_id %d not supported for pio%c%d\n",
 806                         name, index, mux, pin->bank + 'A', pin->pin);
 807                 return -EINVAL;
 808         }
 809
 810         return 0;
 811 }
 812
 813 static void at91_mux_gpio_disable(void __iomem *pio, unsigned mask)
 814 {
 815         writel_relaxed(mask, pio + PIO_PDR);
 816 }
 817
 818 static void at91_mux_gpio_enable(void __iomem *pio, unsigned mask, bool input)
 819 {
 820         writel_relaxed(mask, pio + PIO_PER);
 821         writel_relaxed(mask, pio + (input ? PIO_ODR : PIO_OER));
 822 }
 823
 824 static int at91_pmx_set(struct pinctrl_dev *pctldev, unsigned selector,
 825                         unsigned group)
 826 {
 827         struct at91_pinctrl *info = pinctrl_dev_get_drvdata(pctldev);
 828         const struct at91_pmx_pin *pins_conf = info->groups[group].pins_conf;
 829         const struct at91_pmx_pin *pin;
 830         uint32_t npins = info->groups[group].npins;
 831         int i, ret;
 832         unsigned mask;
 833         void __iomem *pio;
 834
 835         dev_dbg(info->dev, "enable function %s group %s\n",
 836                 info->functions[selector].name, info->groups[group].name);
 837
 838         /* first check that all the pins of the group are valid with a valid
 839          * parameter */
 840         for (i = 0; i < npins; i++) {
 841                 pin = &pins_conf[i];
 842                 ret = pin_check_config(info, info->groups[group].name, i, pin);
 843                 if (ret)
 844                         return ret;
 845         }
 846
 847         for (i = 0; i < npins; i++) {
 848                 pin = &pins_conf[i];
 849                 at91_pin_dbg(info->dev, pin);
 850                 pio = pin_to_controller(info, pin->bank);
 851
 852                 if (!pio)
 853                         continue;
 854
 855                 mask = pin_to_mask(pin->pin);
 856                 at91_mux_disable_interrupt(pio, mask);
 857                 switch (pin->mux) {
 858                 case AT91_MUX_GPIO:
 859                         at91_mux_gpio_enable(pio, mask, 1);
 860                         break;
 861                 case AT91_MUX_PERIPH_A:
 862                         info->ops->mux_A_periph(pio, mask);
 863                         break;
 864                 case AT91_MUX_PERIPH_B:
 865                         info->ops->mux_B_periph(pio, mask);
 866                         break;
 867                 case AT91_MUX_PERIPH_C:
 868                         if (!info->ops->mux_C_periph)
 869                                 return -EINVAL;
 870                         info->ops->mux_C_periph(pio, mask);
 871                         break;
 872                 case AT91_MUX_PERIPH_D:
 873                         if (!info->ops->mux_D_periph)
 874                                 return -EINVAL;
 875                         info->ops->mux_D_periph(pio, mask);
 876                         break;
 877                 }
 878                 if (pin->mux)
 879                         at91_mux_gpio_disable(pio, mask);
 880         }
 881
 882         return 0;
 883 }
 884
 885 static int at91_pmx_get_funcs_count(struct pinctrl_dev *pctldev)
 886 {
 887         struct at91_pinctrl *info = pinctrl_dev_get_drvdata(pctldev);
 888
 889         return info->nfunctions;
 890 }
 891
 892 static const char *at91_pmx_get_func_name(struct pinctrl_dev *pctldev,
 893                                           unsigned selector)
 894 {
 895         struct at91_pinctrl *info = pinctrl_dev_get_drvdata(pctldev);
 896
 897         return info->functions[selector].name;
 898 }
 899
 900 static int at91_pmx_get_groups(struct pinctrl_dev *pctldev, unsigned selector,
 901                                const char * const **groups,
 902                                unsigned * const num_groups)
 903 {
 904         struct at91_pinctrl *info = pinctrl_dev_get_drvdata(pctldev);
 905
 906         *groups = info->functions[selector].groups;
 907         *num_groups = info->functions[selector].ngroups;
 908
 909         return 0;
 910 }
 911
 912 static int at91_gpio_request_enable(struct pinctrl_dev *pctldev,
 913                                     struct pinctrl_gpio_range *range,
 914                                     unsigned offset)
 915 {
 916         struct at91_pinctrl *npct = pinctrl_dev_get_drvdata(pctldev);
 917         struct at91_gpio_chip *at91_chip;
 918         struct gpio_chip *chip;
 919         unsigned mask;
 920
 921         if (!range) {
 922                 dev_err(npct->dev, "invalid range\n");
 923                 return -EINVAL;
 924         }
 925         if (!range->gc) {
 926                 dev_err(npct->dev, "missing GPIO chip in range\n");
 927                 return -EINVAL;
 928         }
 929         chip = range->gc;
 930         at91_chip = gpiochip_get_data(chip);
 931
 932         dev_dbg(npct->dev, "enable pin %u as GPIO\n", offset);
 933
 934         mask = 1 << (offset - chip->base);
 935
 936         dev_dbg(npct->dev, "enable pin %u as PIO%c%d 0x%x\n",
 937                 offset, 'A' + range->id, offset - chip->base, mask);
 938
 939         writel_relaxed(mask, at91_chip->regbase + PIO_PER);
 940
 941         return 0;
 942 }
 943
 944 static void at91_gpio_disable_free(struct pinctrl_dev *pctldev,
 945                                    struct pinctrl_gpio_range *range,
 946                                    unsigned offset)
 947 {
 948         struct at91_pinctrl *npct = pinctrl_dev_get_drvdata(pctldev);
 949
 950         dev_dbg(npct->dev, "disable pin %u as GPIO\n", offset);
 951         /* Set the pin to some default state, GPIO is usually default */
 952 }
 953
 954 static const struct pinmux_ops at91_pmx_ops = {
 955         .get_functions_count    = at91_pmx_get_funcs_count,
 956         .get_function_name      = at91_pmx_get_func_name,
 957         .get_function_groups    = at91_pmx_get_groups,
 958         .set_mux                = at91_pmx_set,
 959         .gpio_request_enable    = at91_gpio_request_enable,
 960         .gpio_disable_free      = at91_gpio_disable_free,
 961 };
 962
 963 static int at91_pinconf_get(struct pinctrl_dev *pctldev,
 964                              unsigned pin_id, unsigned long *config)
 965 {
 966         struct at91_pinctrl *info = pinctrl_dev_get_drvdata(pctldev);
 967         void __iomem *pio;
 968         unsigned pin;
 969         int div;
 970         bool out;
 971
 972         *config = 0;
 973         dev_dbg(info->dev, "%s:%d, pin_id=%d", __func__, __LINE__, pin_id);
 974         pio = pin_to_controller(info, pin_to_bank(pin_id));
 975
 976         if (!pio)
 977                 return -EINVAL;
 978
 979         pin = pin_id % MAX_NB_GPIO_PER_BANK;
 980
 981         if (at91_mux_get_multidrive(pio, pin))
 982                 *config |= MULTI_DRIVE;
 983
 984         if (at91_mux_get_pullup(pio, pin))
 985                 *config |= PULL_UP;
 986
 987         if (info->ops->get_deglitch && info->ops->get_deglitch(pio, pin))
 988                 *config |= DEGLITCH;
 989         if (info->ops->get_debounce && info->ops->get_debounce(pio, pin, &div))
 990                 *config |= DEBOUNCE | (div << DEBOUNCE_VAL_SHIFT);
 991         if (info->ops->get_pulldown && info->ops->get_pulldown(pio, pin))
 992                 *config |= PULL_DOWN;
 993         if (info->ops->get_schmitt_trig && info->ops->get_schmitt_trig(pio, pin))
 994                 *config |= DIS_SCHMIT;
 995         if (info->ops->get_drivestrength)
 996                 *config |= (info->ops->get_drivestrength(pio, pin)
 997                                 << DRIVE_STRENGTH_SHIFT);
 998         if (info->ops->get_slewrate)
 999                 *config |= (info->ops->get_slewrate(pio, pin) << SLEWRATE_SHIFT);
1000         if (at91_mux_get_output(pio, pin, &out))
1001                 *config |= OUTPUT | (out << OUTPUT_VAL_SHIFT);
1002
1003         return 0;
1004 }
1005
1006 static int at91_pinconf_set(struct pinctrl_dev *pctldev,
1007                              unsigned pin_id, unsigned long *configs,
1008                              unsigned num_configs)
1009 {
1010         struct at91_pinctrl *info = pinctrl_dev_get_drvdata(pctldev);
1011         unsigned mask;
1012         void __iomem *pio;
1013         int i;
1014         unsigned long config;
1015         unsigned pin;
1016
1017         for (i = 0; i < num_configs; i++) {
1018                 config = configs[i];
1019
1020                 dev_dbg(info->dev,
1021                         "%s:%d, pin_id=%d, config=0x%lx",
1022                         __func__, __LINE__, pin_id, config);
1023                 pio = pin_to_controller(info, pin_to_bank(pin_id));
1024
1025                 if (!pio)
1026                         return -EINVAL;
1027
1028                 pin = pin_id % MAX_NB_GPIO_PER_BANK;
1029                 mask = pin_to_mask(pin);
1030
1031                 if (config & PULL_UP && config & PULL_DOWN)
1032                         return -EINVAL;
1033
1034                 at91_mux_set_output(pio, mask, config & OUTPUT,
1035                                     (config & OUTPUT_VAL) >> OUTPUT_VAL_SHIFT);
1036                 at91_mux_set_pullup(pio, mask, config & PULL_UP);
1037                 at91_mux_set_multidrive(pio, mask, config & MULTI_DRIVE);
1038                 if (info->ops->set_deglitch)
1039                         info->ops->set_deglitch(pio, mask, config & DEGLITCH);
1040                 if (info->ops->set_debounce)
1041                         info->ops->set_debounce(pio, mask, config & DEBOUNCE,
1042                                 (config & DEBOUNCE_VAL) >> DEBOUNCE_VAL_SHIFT);
1043                 if (info->ops->set_pulldown)
1044                         info->ops->set_pulldown(pio, mask, config & PULL_DOWN);
1045                 if (info->ops->disable_schmitt_trig && config & DIS_SCHMIT)
1046                         info->ops->disable_schmitt_trig(pio, mask);
1047                 if (info->ops->set_drivestrength)
1048                         info->ops->set_drivestrength(pio, pin,
1049                                 (config & DRIVE_STRENGTH)
1050                                         >> DRIVE_STRENGTH_SHIFT);
1051                 if (info->ops->set_slewrate)
1052                         info->ops->set_slewrate(pio, pin,
1053                                 (config & SLEWRATE) >> SLEWRATE_SHIFT);
1054
1055         } /* for each config */
1056
1057         return 0;
1058 }
1059
1060 #define DBG_SHOW_FLAG(flag) do {                \
1061         if (config & flag) {                    \
1062                 if (num_conf)                   \
1063                         seq_puts(s, "|");       \
1064                 seq_puts(s, #flag);             \
1065                 num_conf++;                     \
1066         }                                       \
1067 } while (0)
1068
1069 #define DBG_SHOW_FLAG_MASKED(mask, flag, name) do { \
1070         if ((config & mask) == flag) {          \
1071                 if (num_conf)                   \
1072                         seq_puts(s, "|");       \
1073                 seq_puts(s, #name);             \
1074                 num_conf++;                     \
1075         }                                       \
1076 } while (0)
1077
1078 static void at91_pinconf_dbg_show(struct pinctrl_dev *pctldev,
1079                                    struct seq_file *s, unsigned pin_id)
1080 {
1081         unsigned long config;
1082         int val, num_conf = 0;
1083
1084         at91_pinconf_get(pctldev, pin_id, &config);
1085
1086         DBG_SHOW_FLAG(MULTI_DRIVE);
1087         DBG_SHOW_FLAG(PULL_UP);
1088         DBG_SHOW_FLAG(PULL_DOWN);
1089         DBG_SHOW_FLAG(DIS_SCHMIT);
1090         DBG_SHOW_FLAG(DEGLITCH);
1091         DBG_SHOW_FLAG_MASKED(DRIVE_STRENGTH, DRIVE_STRENGTH_BIT_MSK(LOW),
1092                              DRIVE_STRENGTH_LOW);
1093         DBG_SHOW_FLAG_MASKED(DRIVE_STRENGTH, DRIVE_STRENGTH_BIT_MSK(MED),
1094                              DRIVE_STRENGTH_MED);
1095         DBG_SHOW_FLAG_MASKED(DRIVE_STRENGTH, DRIVE_STRENGTH_BIT_MSK(HI),
1096                              DRIVE_STRENGTH_HI);
1097         DBG_SHOW_FLAG(SLEWRATE);
1098         DBG_SHOW_FLAG(DEBOUNCE);
1099         if (config & DEBOUNCE) {
1100                 val = config >> DEBOUNCE_VAL_SHIFT;
1101                 seq_printf(s, "(%d)", val);
1102         }
1103
1104         return;
1105 }
1106
1107 static void at91_pinconf_group_dbg_show(struct pinctrl_dev *pctldev,
1108                                          struct seq_file *s, unsigned group)
1109 {
1110 }
1111
1112 static const struct pinconf_ops at91_pinconf_ops = {
1113         .pin_config_get                 = at91_pinconf_get,
1114         .pin_config_set                 = at91_pinconf_set,
1115         .pin_config_dbg_show            = at91_pinconf_dbg_show,
1116         .pin_config_group_dbg_show      = at91_pinconf_group_dbg_show,
1117 };
1118
1119 static struct pinctrl_desc at91_pinctrl_desc = {
1120         .pctlops        = &at91_pctrl_ops,
1121         .pmxops         = &at91_pmx_ops,
1122         .confops        = &at91_pinconf_ops,
1123         .owner          = THIS_MODULE,
1124 };
1125
1126 static const char *gpio_compat = "atmel,at91rm9200-gpio";
1127
1128 static void at91_pinctrl_child_count(struct at91_pinctrl *info,
1129                                      struct device_node *np)
1130 {
1131         struct device_node *child;
1132
1133         for_each_child_of_node(np, child) {
1134                 if (of_device_is_compatible(child, gpio_compat)) {
1135                         if (of_device_is_available(child))
1136                                 info->nactive_banks++;
1137                 } else {
1138                         info->nfunctions++;
1139                         info->ngroups += of_get_child_count(child);
1140                 }
1141         }
1142 }
1143
1144 static int at91_pinctrl_mux_mask(struct at91_pinctrl *info,
1145                                  struct device_node *np)
1146 {
1147         int ret = 0;
1148         int size;
1149         const __be32 *list;
1150
1151         list = of_get_property(np, "atmel,mux-mask", &size);
1152         if (!list) {
1153                 dev_err(info->dev, "can not read the mux-mask of %d\n", size);
1154                 return -EINVAL;
1155         }
1156
1157         size /= sizeof(*list);
1158         if (!size || size % gpio_banks) {
1159                 dev_err(info->dev, "wrong mux mask array should be by %d\n", gpio_banks);
1160                 return -EINVAL;
1161         }
1162         info->nmux = size / gpio_banks;
1163
1164         info->mux_mask = devm_kcalloc(info->dev, size, sizeof(u32),
1165                                       GFP_KERNEL);
1166         if (!info->mux_mask)
1167                 return -ENOMEM;
1168
1169         ret = of_property_read_u32_array(np, "atmel,mux-mask",
1170                                           info->mux_mask, size);
1171         if (ret)
1172                 dev_err(info->dev, "can not read the mux-mask of %d\n", size);
1173         return ret;
1174 }
1175
1176 static int at91_pinctrl_parse_groups(struct device_node *np,
1177                                      struct at91_pin_group *grp,
1178                                      struct at91_pinctrl *info, u32 index)
1179 {
1180         struct at91_pmx_pin *pin;
1181         int size;
1182         const __be32 *list;
1183         int i, j;
1184
1185         dev_dbg(info->dev, "group(%d): %pOFn\n", index, np);
1186
1187         /* Initialise group */
1188         grp->name = np->name;
1189
1190         /*
1191          * the binding format is atmel,pins = <bank pin mux CONFIG ...>,
1192          * do sanity check and calculate pins number
1193          */
1194         list = of_get_property(np, "atmel,pins", &size);
1195         /* we do not check return since it's safe node passed down */
1196         size /= sizeof(*list);
1197         if (!size || size % 4) {
1198                 dev_err(info->dev, "wrong pins number or pins and configs should be by 4\n");
1199                 return -EINVAL;
1200         }
1201
1202         grp->npins = size / 4;
1203         pin = grp->pins_conf = devm_kcalloc(info->dev,
1204                                             grp->npins,
1205                                             sizeof(struct at91_pmx_pin),
1206                                             GFP_KERNEL);
1207         grp->pins = devm_kcalloc(info->dev, grp->npins, sizeof(unsigned int),
1208                                  GFP_KERNEL);
1209         if (!grp->pins_conf || !grp->pins)
1210                 return -ENOMEM;
1211
1212         for (i = 0, j = 0; i < size; i += 4, j++) {
1213                 pin->bank = be32_to_cpu(*list++);
1214                 pin->pin = be32_to_cpu(*list++);
1215                 grp->pins[j] = pin->bank * MAX_NB_GPIO_PER_BANK + pin->pin;
1216                 pin->mux = be32_to_cpu(*list++);
1217                 pin->conf = be32_to_cpu(*list++);
1218
1219                 at91_pin_dbg(info->dev, pin);
1220                 pin++;
1221         }
1222
1223         return 0;
1224 }
1225
1226 static int at91_pinctrl_parse_functions(struct device_node *np,
1227                                         struct at91_pinctrl *info, u32 index)
1228 {
1229         struct device_node *child;
1230         struct at91_pmx_func *func;
1231         struct at91_pin_group *grp;
1232         int ret;
1233         static u32 grp_index;
1234         u32 i = 0;
1235
1236         dev_dbg(info->dev, "parse function(%d): %pOFn\n", index, np);
1237
1238         func = &info->functions[index];
1239
1240         /* Initialise function */
1241         func->name = np->name;
1242         func->ngroups = of_get_child_count(np);
1243         if (func->ngroups == 0) {
1244                 dev_err(info->dev, "no groups defined\n");
1245                 return -EINVAL;
1246         }
1247         func->groups = devm_kcalloc(info->dev,
1248                         func->ngroups, sizeof(char *), GFP_KERNEL);
1249         if (!func->groups)
1250                 return -ENOMEM;
1251
1252         for_each_child_of_node(np, child) {
1253                 func->groups[i] = child->name;
1254                 grp = &info->groups[grp_index++];
1255                 ret = at91_pinctrl_parse_groups(child, grp, info, i++);
1256                 if (ret) {
1257                         of_node_put(child);
1258                         return ret;
1259                 }
1260         }
1261
1262         return 0;
1263 }
1264
1265 static const struct of_device_id at91_pinctrl_of_match[] = {
1266         { .compatible = "atmel,sama5d3-pinctrl", .data = &sama5d3_ops },
1267         { .compatible = "atmel,at91sam9x5-pinctrl", .data = &at91sam9x5_ops },
1268         { .compatible = "atmel,at91rm9200-pinctrl", .data = &at91rm9200_ops },
1269         { .compatible = "microchip,sam9x60-pinctrl", .data = &sam9x60_ops },
1270         { /* sentinel */ }
1271 };
1272
1273 static int at91_pinctrl_probe_dt(struct platform_device *pdev,
1274                                  struct at91_pinctrl *info)
1275 {
1276         int ret = 0;
1277         int i, j;
1278         uint32_t *tmp;
1279         struct device_node *np = pdev->dev.of_node;
1280         struct device_node *child;
1281
1282         if (!np)
1283                 return -ENODEV;
1284
1285         info->dev = &pdev->dev;
1286         info->ops = (struct at91_pinctrl_mux_ops *)
1287                 of_match_device(at91_pinctrl_of_match, &pdev->dev)->data;
1288         at91_pinctrl_child_count(info, np);
1289
1290         if (gpio_banks < 1) {
1291                 dev_err(&pdev->dev, "you need to specify at least one gpio-controller\n");
1292                 return -EINVAL;
1293         }
1294
1295         ret = at91_pinctrl_mux_mask(info, np);
1296         if (ret)
1297                 return ret;
1298
1299         dev_dbg(&pdev->dev, "nmux = %d\n", info->nmux);
1300
1301         dev_dbg(&pdev->dev, "mux-mask\n");
1302         tmp = info->mux_mask;
1303         for (i = 0; i < gpio_banks; i++) {
1304                 for (j = 0; j < info->nmux; j++, tmp++) {
1305                         dev_dbg(&pdev->dev, "%d:%d\t0x%x\n", i, j, tmp[0]);
1306                 }
1307         }
1308
1309         dev_dbg(&pdev->dev, "nfunctions = %d\n", info->nfunctions);
1310         dev_dbg(&pdev->dev, "ngroups = %d\n", info->ngroups);
1311         info->functions = devm_kcalloc(&pdev->dev,
1312                                         info->nfunctions,
1313                                         sizeof(struct at91_pmx_func),
1314                                         GFP_KERNEL);
1315         if (!info->functions)
1316                 return -ENOMEM;
1317
1318         info->groups = devm_kcalloc(&pdev->dev,
1319                                         info->ngroups,
1320                                         sizeof(struct at91_pin_group),
1321                                         GFP_KERNEL);
1322         if (!info->groups)
1323                 return -ENOMEM;
1324
1325         dev_dbg(&pdev->dev, "nbanks = %d\n", gpio_banks);
1326         dev_dbg(&pdev->dev, "nfunctions = %d\n", info->nfunctions);
1327         dev_dbg(&pdev->dev, "ngroups = %d\n", info->ngroups);
1328
1329         i = 0;
1330
1331         for_each_child_of_node(np, child) {
1332                 if (of_device_is_compatible(child, gpio_compat))
1333                         continue;
1334                 ret = at91_pinctrl_parse_functions(child, info, i++);
1335                 if (ret) {
1336                         dev_err(&pdev->dev, "failed to parse function\n");
1337                         of_node_put(child);
1338                         return ret;
1339                 }
1340         }
1341
1342         return 0;
1343 }
1344
1345 static int at91_pinctrl_probe(struct platform_device *pdev)
1346 {
1347         struct at91_pinctrl *info;
1348         struct pinctrl_pin_desc *pdesc;
1349         int ret, i, j, k, ngpio_chips_enabled = 0;
1350
1351         info = devm_kzalloc(&pdev->dev, sizeof(*info), GFP_KERNEL);
1352         if (!info)
1353                 return -ENOMEM;
1354
1355         ret = at91_pinctrl_probe_dt(pdev, info);
1356         if (ret)
1357                 return ret;
1358
1359         /*
1360          * We need all the GPIO drivers to probe FIRST, or we will not be able
1361          * to obtain references to the struct gpio_chip * for them, and we
1362          * need this to proceed.
1363          */
1364         for (i = 0; i < gpio_banks; i++)
1365                 if (gpio_chips[i])
1366                         ngpio_chips_enabled++;
1367
1368         if (ngpio_chips_enabled < info->nactive_banks) {
1369                 dev_warn(&pdev->dev,
1370                          "All GPIO chips are not registered yet (%d/%d)\n",
1371                          ngpio_chips_enabled, info->nactive_banks);
1372                 devm_kfree(&pdev->dev, info);
1373                 return -EPROBE_DEFER;
1374         }
1375
1376         at91_pinctrl_desc.name = dev_name(&pdev->dev);
1377         at91_pinctrl_desc.npins = gpio_banks * MAX_NB_GPIO_PER_BANK;
1378         at91_pinctrl_desc.pins = pdesc =
1379                 devm_kcalloc(&pdev->dev,
1380                              at91_pinctrl_desc.npins, sizeof(*pdesc),
1381                              GFP_KERNEL);
1382
1383         if (!at91_pinctrl_desc.pins)
1384                 return -ENOMEM;
1385
1386         for (i = 0, k = 0; i < gpio_banks; i++) {
1387                 for (j = 0; j < MAX_NB_GPIO_PER_BANK; j++, k++) {
1388                         pdesc->number = k;
1389                         pdesc->name = kasprintf(GFP_KERNEL, "pio%c%d", i + 'A', j);
1390                         pdesc++;
1391                 }
1392         }
1393
1394         platform_set_drvdata(pdev, info);
1395         info->pctl = devm_pinctrl_register(&pdev->dev, &at91_pinctrl_desc,
1396                                            info);
1397
1398         if (IS_ERR(info->pctl)) {
1399                 dev_err(&pdev->dev, "could not register AT91 pinctrl driver\n");
1400                 return PTR_ERR(info->pctl);
1401         }
1402
1403         /* We will handle a range of GPIO pins */
1404         for (i = 0; i < gpio_banks; i++)
1405                 if (gpio_chips[i])
1406                         pinctrl_add_gpio_range(info->pctl, &gpio_chips[i]->range);
1407
1408         dev_info(&pdev->dev, "initialized AT91 pinctrl driver\n");
1409
1410         return 0;
1411 }
1412
1413 static int at91_gpio_get_direction(struct gpio_chip *chip, unsigned offset)
1414 {
1415         struct at91_gpio_chip *at91_gpio = gpiochip_get_data(chip);
1416         void __iomem *pio = at91_gpio->regbase;
1417         unsigned mask = 1 << offset;
1418         u32 osr;
1419
1420         osr = readl_relaxed(pio + PIO_OSR);
1421         if (osr & mask)
1422                 return GPIO_LINE_DIRECTION_OUT;
1423
1424         return GPIO_LINE_DIRECTION_IN;
1425 }
1426
1427 static int at91_gpio_direction_input(struct gpio_chip *chip, unsigned offset)
1428 {
1429         struct at91_gpio_chip *at91_gpio = gpiochip_get_data(chip);
1430         void __iomem *pio = at91_gpio->regbase;
1431         unsigned mask = 1 << offset;
1432
1433         writel_relaxed(mask, pio + PIO_ODR);
1434         return 0;
1435 }
1436
1437 static int at91_gpio_get(struct gpio_chip *chip, unsigned offset)
1438 {
1439         struct at91_gpio_chip *at91_gpio = gpiochip_get_data(chip);
1440         void __iomem *pio = at91_gpio->regbase;
1441         unsigned mask = 1 << offset;
1442         u32 pdsr;
1443
1444         pdsr = readl_relaxed(pio + PIO_PDSR);
1445         return (pdsr & mask) != 0;
1446 }
1447
1448 static void at91_gpio_set(struct gpio_chip *chip, unsigned offset,
1449                                 int val)
1450 {
1451         struct at91_gpio_chip *at91_gpio = gpiochip_get_data(chip);
1452         void __iomem *pio = at91_gpio->regbase;
1453         unsigned mask = 1 << offset;
1454
1455         writel_relaxed(mask, pio + (val ? PIO_SODR : PIO_CODR));
1456 }
1457
1458 static void at91_gpio_set_multiple(struct gpio_chip *chip,
1459                                       unsigned long *mask, unsigned long *bits)
1460 {
1461         struct at91_gpio_chip *at91_gpio = gpiochip_get_data(chip);
1462         void __iomem *pio = at91_gpio->regbase;
1463
1464 #define BITS_MASK(bits) (((bits) == 32) ? ~0U : (BIT(bits) - 1))
1465         /* Mask additionally to ngpio as not all GPIO controllers have 32 pins */
1466         uint32_t set_mask = (*mask & *bits) & BITS_MASK(chip->ngpio);
1467         uint32_t clear_mask = (*mask & ~(*bits)) & BITS_MASK(chip->ngpio);
1468
1469         writel_relaxed(set_mask, pio + PIO_SODR);
1470         writel_relaxed(clear_mask, pio + PIO_CODR);
1471 }
1472
1473 static int at91_gpio_direction_output(struct gpio_chip *chip, unsigned offset,
1474                                 int val)
1475 {
1476         struct at91_gpio_chip *at91_gpio = gpiochip_get_data(chip);
1477         void __iomem *pio = at91_gpio->regbase;
1478         unsigned mask = 1 << offset;
1479
1480         writel_relaxed(mask, pio + (val ? PIO_SODR : PIO_CODR));
1481         writel_relaxed(mask, pio + PIO_OER);
1482
1483         return 0;
1484 }
1485
1486 #ifdef CONFIG_DEBUG_FS
1487 static void at91_gpio_dbg_show(struct seq_file *s, struct gpio_chip *chip)
1488 {
1489         enum at91_mux mode;
1490         int i;
1491         struct at91_gpio_chip *at91_gpio = gpiochip_get_data(chip);
1492         void __iomem *pio = at91_gpio->regbase;
1493         const char *gpio_label;
1494
1495         for_each_requested_gpio(chip, i, gpio_label) {
1496                 unsigned mask = pin_to_mask(i);
1497
1498                 mode = at91_gpio->ops->get_periph(pio, mask);
1499                 seq_printf(s, "[%s] GPIO%s%d: ",
1500                            gpio_label, chip->label, i);
1501                 if (mode == AT91_MUX_GPIO) {
1502                         seq_printf(s, "[gpio] ");
1503                         seq_printf(s, "%s ",
1504                                       readl_relaxed(pio + PIO_OSR) & mask ?
1505                                       "output" : "input");
1506                         seq_printf(s, "%s\n",
1507                                       readl_relaxed(pio + PIO_PDSR) & mask ?
1508                                       "set" : "clear");
1509                 } else {
1510                         seq_printf(s, "[periph %c]\n",
1511                                    mode + 'A' - 1);
1512                 }
1513         }
1514 }
1515 #else
1516 #define at91_gpio_dbg_show      NULL
1517 #endif
1518
1519 /* Several AIC controller irqs are dispatched through this GPIO handler.
1520  * To use any AT91_PIN_* as an externally triggered IRQ, first call
1521  * at91_set_gpio_input() then maybe enable its glitch filter.
1522  * Then just request_irq() with the pin ID; it works like any ARM IRQ
1523  * handler.
1524  * First implementation always triggers on rising and falling edges
1525  * whereas the newer PIO3 can be additionally configured to trigger on
1526  * level, edge with any polarity.
1527  *
1528  * Alternatively, certain pins may be used directly as IRQ0..IRQ6 after
1529  * configuring them with at91_set_a_periph() or at91_set_b_periph().
1530  * IRQ0..IRQ6 should be configurable, e.g. level vs edge triggering.
1531  */
1532
1533 static void gpio_irq_mask(struct irq_data *d)
1534 {
1535         struct at91_gpio_chip *at91_gpio = irq_data_get_irq_chip_data(d);
1536         void __iomem    *pio = at91_gpio->regbase;
1537         unsigned        mask = 1 << d->hwirq;
1538
1539         if (pio)
1540                 writel_relaxed(mask, pio + PIO_IDR);
1541 }
1542
1543 static void gpio_irq_unmask(struct irq_data *d)
1544 {
1545         struct at91_gpio_chip *at91_gpio = irq_data_get_irq_chip_data(d);
1546         void __iomem    *pio = at91_gpio->regbase;
1547         unsigned        mask = 1 << d->hwirq;
1548
1549         if (pio)
1550                 writel_relaxed(mask, pio + PIO_IER);
1551 }
1552
1553 static int gpio_irq_type(struct irq_data *d, unsigned type)
1554 {
1555         switch (type) {
1556         case IRQ_TYPE_NONE:
1557         case IRQ_TYPE_EDGE_BOTH:
1558                 return 0;
1559         default:
1560                 return -EINVAL;
1561         }
1562 }
1563
1564 /* Alternate irq type for PIO3 support */
1565 static int alt_gpio_irq_type(struct irq_data *d, unsigned type)
1566 {
1567         struct at91_gpio_chip *at91_gpio = irq_data_get_irq_chip_data(d);
1568         void __iomem    *pio = at91_gpio->regbase;
1569         unsigned        mask = 1 << d->hwirq;
1570
1571         switch (type) {
1572         case IRQ_TYPE_EDGE_RISING:
1573                 irq_set_handler_locked(d, handle_simple_irq);
1574                 writel_relaxed(mask, pio + PIO_ESR);
1575                 writel_relaxed(mask, pio + PIO_REHLSR);
1576                 break;
1577         case IRQ_TYPE_EDGE_FALLING:
1578                 irq_set_handler_locked(d, handle_simple_irq);
1579                 writel_relaxed(mask, pio + PIO_ESR);
1580                 writel_relaxed(mask, pio + PIO_FELLSR);
1581                 break;
1582         case IRQ_TYPE_LEVEL_LOW:
1583                 irq_set_handler_locked(d, handle_level_irq);
1584                 writel_relaxed(mask, pio + PIO_LSR);
1585                 writel_relaxed(mask, pio + PIO_FELLSR);
1586                 break;
1587         case IRQ_TYPE_LEVEL_HIGH:
1588                 irq_set_handler_locked(d, handle_level_irq);
1589                 writel_relaxed(mask, pio + PIO_LSR);
1590                 writel_relaxed(mask, pio + PIO_REHLSR);
1591                 break;
1592         case IRQ_TYPE_EDGE_BOTH:
1593                 /*
1594                  * disable additional interrupt modes:
1595                  * fall back to default behavior
1596                  */
1597                 irq_set_handler_locked(d, handle_simple_irq);
1598                 writel_relaxed(mask, pio + PIO_AIMDR);
1599                 return 0;
1600         case IRQ_TYPE_NONE:
1601         default:
1602                 pr_warn("AT91: No type for GPIO irq offset %d\n", d->irq);
1603                 return -EINVAL;
1604         }
1605
1606         /* enable additional interrupt modes */
1607         writel_relaxed(mask, pio + PIO_AIMER);
1608
1609         return 0;
1610 }
1611
1612 static void gpio_irq_ack(struct irq_data *d)
1613 {
1614         /* the interrupt is already cleared before by reading ISR */
1615 }
1616
1617 #ifdef CONFIG_PM
1618
1619 static u32 wakeups[MAX_GPIO_BANKS];
1620 static u32 backups[MAX_GPIO_BANKS];
1621
1622 static int gpio_irq_set_wake(struct irq_data *d, unsigned state)
1623 {
1624         struct at91_gpio_chip *at91_gpio = irq_data_get_irq_chip_data(d);
1625         unsigned        bank = at91_gpio->pioc_idx;
1626         unsigned mask = 1 << d->hwirq;
1627
1628         if (unlikely(bank >= MAX_GPIO_BANKS))
1629                 return -EINVAL;
1630
1631         if (state)
1632                 wakeups[bank] |= mask;
1633         else
1634                 wakeups[bank] &= ~mask;
1635
1636         irq_set_irq_wake(at91_gpio->pioc_virq, state);
1637
1638         return 0;
1639 }
1640
1641 void at91_pinctrl_gpio_suspend(void)
1642 {
1643         int i;
1644
1645         for (i = 0; i < gpio_banks; i++) {
1646                 void __iomem  *pio;
1647
1648                 if (!gpio_chips[i])
1649                         continue;
1650
1651                 pio = gpio_chips[i]->regbase;
1652
1653                 backups[i] = readl_relaxed(pio + PIO_IMR);
1654                 writel_relaxed(backups[i], pio + PIO_IDR);
1655                 writel_relaxed(wakeups[i], pio + PIO_IER);
1656
1657                 if (!wakeups[i])
1658                         clk_disable_unprepare(gpio_chips[i]->clock);
1659                 else
1660                         printk(KERN_DEBUG "GPIO-%c may wake for %08x\n",
1661                                'A'+i, wakeups[i]);
1662         }
1663 }
1664
1665 void at91_pinctrl_gpio_resume(void)
1666 {
1667         int i;
1668
1669         for (i = 0; i < gpio_banks; i++) {
1670                 void __iomem  *pio;
1671
1672                 if (!gpio_chips[i])
1673                         continue;
1674
1675                 pio = gpio_chips[i]->regbase;
1676
1677                 if (!wakeups[i])
1678                         clk_prepare_enable(gpio_chips[i]->clock);
1679
1680                 writel_relaxed(wakeups[i], pio + PIO_IDR);
1681                 writel_relaxed(backups[i], pio + PIO_IER);
1682         }
1683 }
1684
1685 #else
1686 #define gpio_irq_set_wake       NULL
1687 #endif /* CONFIG_PM */
1688
1689 static void gpio_irq_handler(struct irq_desc *desc)
1690 {
1691         struct irq_chip *chip = irq_desc_get_chip(desc);
1692         struct gpio_chip *gpio_chip = irq_desc_get_handler_data(desc);
1693         struct at91_gpio_chip *at91_gpio = gpiochip_get_data(gpio_chip);
1694         void __iomem    *pio = at91_gpio->regbase;
1695         unsigned long   isr;
1696         int             n;
1697
1698         chained_irq_enter(chip, desc);
1699         for (;;) {
1700                 /* Reading ISR acks pending (edge triggered) GPIO interrupts.
1701                  * When there are none pending, we're finished unless we need
1702                  * to process multiple banks (like ID_PIOCDE on sam9263).
1703                  */
1704                 isr = readl_relaxed(pio + PIO_ISR) & readl_relaxed(pio + PIO_IMR);
1705                 if (!isr) {
1706                         if (!at91_gpio->next)
1707                                 break;
1708                         at91_gpio = at91_gpio->next;
1709                         pio = at91_gpio->regbase;
1710                         gpio_chip = &at91_gpio->chip;
1711                         continue;
1712                 }
1713
1714                 for_each_set_bit(n, &isr, BITS_PER_LONG) {
1715                         generic_handle_irq(irq_find_mapping(
1716                                            gpio_chip->irq.domain, n));
1717                 }
1718         }
1719         chained_irq_exit(chip, desc);
1720         /* now it may re-trigger */
1721 }
1722
1723 static int at91_gpio_of_irq_setup(struct platform_device *pdev,
1724                                   struct at91_gpio_chip *at91_gpio)
1725 {
1726         struct gpio_chip        *gpiochip_prev = NULL;
1727         struct at91_gpio_chip   *prev = NULL;
1728         struct irq_data         *d = irq_get_irq_data(at91_gpio->pioc_virq);
1729         struct irq_chip         *gpio_irqchip;
1730         struct gpio_irq_chip    *girq;
1731         int i;
1732
1733         gpio_irqchip = devm_kzalloc(&pdev->dev, sizeof(*gpio_irqchip),
1734                                     GFP_KERNEL);
1735         if (!gpio_irqchip)
1736                 return -ENOMEM;
1737
1738         at91_gpio->pioc_hwirq = irqd_to_hwirq(d);
1739
1740         gpio_irqchip->name = "GPIO";
1741         gpio_irqchip->irq_ack = gpio_irq_ack;
1742         gpio_irqchip->irq_disable = gpio_irq_mask;
1743         gpio_irqchip->irq_mask = gpio_irq_mask;
1744         gpio_irqchip->irq_unmask = gpio_irq_unmask;
1745         gpio_irqchip->irq_set_wake = gpio_irq_set_wake,
1746         gpio_irqchip->irq_set_type = at91_gpio->ops->irq_type;
1747
1748         /* Disable irqs of this PIO controller */
1749         writel_relaxed(~0, at91_gpio->regbase + PIO_IDR);
1750
1751         /*
1752          * Let the generic code handle this edge IRQ, the the chained
1753          * handler will perform the actual work of handling the parent
1754          * interrupt.
1755          */
1756         girq = &at91_gpio->chip.irq;
1757         girq->chip = gpio_irqchip;
1758         girq->default_type = IRQ_TYPE_NONE;
1759         girq->handler = handle_edge_irq;
1760
1761         /*
1762          * The top level handler handles one bank of GPIOs, except
1763          * on some SoC it can handle up to three...
1764          * We only set up the handler for the first of the list.
1765          */
1766         gpiochip_prev = irq_get_handler_data(at91_gpio->pioc_virq);
1767         if (!gpiochip_prev) {
1768                 girq->parent_handler = gpio_irq_handler;
1769                 girq->num_parents = 1;
1770                 girq->parents = devm_kcalloc(&pdev->dev, 1,
1771                                              sizeof(*girq->parents),
1772                                              GFP_KERNEL);
1773                 if (!girq->parents)
1774                         return -ENOMEM;
1775                 girq->parents[0] = at91_gpio->pioc_virq;
1776                 return 0;
1777         }
1778
1779         prev = gpiochip_get_data(gpiochip_prev);
1780         /* we can only have 2 banks before */
1781         for (i = 0; i < 2; i++) {
1782                 if (prev->next) {
1783                         prev = prev->next;
1784                 } else {
1785                         prev->next = at91_gpio;
1786                         return 0;
1787                 }
1788         }
1789
1790         return -EINVAL;
1791 }
1792
1793 /* This structure is replicated for each GPIO block allocated at probe time */
1794 static const struct gpio_chip at91_gpio_template = {
1795         .request                = gpiochip_generic_request,
1796         .free                   = gpiochip_generic_free,
1797         .get_direction          = at91_gpio_get_direction,
1798         .direction_input        = at91_gpio_direction_input,
1799         .get                    = at91_gpio_get,
1800         .direction_output       = at91_gpio_direction_output,
1801         .set                    = at91_gpio_set,
1802         .set_multiple           = at91_gpio_set_multiple,
1803         .dbg_show               = at91_gpio_dbg_show,
1804         .can_sleep              = false,
1805         .ngpio                  = MAX_NB_GPIO_PER_BANK,
1806 };
1807
1808 static const struct of_device_id at91_gpio_of_match[] = {
1809         { .compatible = "atmel,at91sam9x5-gpio", .data = &at91sam9x5_ops, },
1810         { .compatible = "atmel,at91rm9200-gpio", .data = &at91rm9200_ops },
1811         { .compatible = "microchip,sam9x60-gpio", .data = &sam9x60_ops },
1812         { /* sentinel */ }
1813 };
1814
1815 static int at91_gpio_probe(struct platform_device *pdev)
1816 {
1817         struct device_node *np = pdev->dev.of_node;
1818         struct at91_gpio_chip *at91_chip = NULL;
1819         struct gpio_chip *chip;
1820         struct pinctrl_gpio_range *range;
1821         int ret = 0;
1822         int irq, i;
1823         int alias_idx = of_alias_get_id(np, "gpio");
1824         uint32_t ngpio;
1825         char **names;
1826
1827         BUG_ON(alias_idx >= ARRAY_SIZE(gpio_chips));
1828         if (gpio_chips[alias_idx]) {
1829                 ret = -EBUSY;
1830                 goto err;
1831         }
1832
1833         irq = platform_get_irq(pdev, 0);
1834         if (irq < 0) {
1835                 ret = irq;
1836                 goto err;
1837         }
1838
1839         at91_chip = devm_kzalloc(&pdev->dev, sizeof(*at91_chip), GFP_KERNEL);
1840         if (!at91_chip) {
1841                 ret = -ENOMEM;
1842                 goto err;
1843         }
1844
1845         at91_chip->regbase = devm_platform_ioremap_resource(pdev, 0);
1846         if (IS_ERR(at91_chip->regbase)) {
1847                 ret = PTR_ERR(at91_chip->regbase);
1848                 goto err;
1849         }
1850
1851         at91_chip->ops = (struct at91_pinctrl_mux_ops *)
1852                 of_match_device(at91_gpio_of_match, &pdev->dev)->data;
1853         at91_chip->pioc_virq = irq;
1854         at91_chip->pioc_idx = alias_idx;
1855
1856         at91_chip->clock = devm_clk_get(&pdev->dev, NULL);
1857         if (IS_ERR(at91_chip->clock)) {
1858                 dev_err(&pdev->dev, "failed to get clock, ignoring.\n");
1859                 ret = PTR_ERR(at91_chip->clock);
1860                 goto err;
1861         }
1862
1863         ret = clk_prepare_enable(at91_chip->clock);
1864         if (ret) {
1865                 dev_err(&pdev->dev, "failed to prepare and enable clock, ignoring.\n");
1866                 goto clk_enable_err;
1867         }
1868
1869         at91_chip->chip = at91_gpio_template;
1870
1871         chip = &at91_chip->chip;
1872         chip->of_node = np;
1873         chip->label = dev_name(&pdev->dev);
1874         chip->parent = &pdev->dev;
1875         chip->owner = THIS_MODULE;
1876         chip->base = alias_idx * MAX_NB_GPIO_PER_BANK;
1877
1878         if (!of_property_read_u32(np, "#gpio-lines", &ngpio)) {
1879                 if (ngpio >= MAX_NB_GPIO_PER_BANK)
1880                         pr_err("at91_gpio.%d, gpio-nb >= %d failback to %d\n",
1881                                alias_idx, MAX_NB_GPIO_PER_BANK, MAX_NB_GPIO_PER_BANK);
1882                 else
1883                         chip->ngpio = ngpio;
1884         }
1885
1886         names = devm_kcalloc(&pdev->dev, chip->ngpio, sizeof(char *),
1887                              GFP_KERNEL);
1888
1889         if (!names) {
1890                 ret = -ENOMEM;
1891                 goto clk_enable_err;
1892         }
1893
1894         for (i = 0; i < chip->ngpio; i++)
1895                 names[i] = kasprintf(GFP_KERNEL, "pio%c%d", alias_idx + 'A', i);
1896
1897         chip->names = (const char *const *)names;
1898
1899         range = &at91_chip->range;
1900         range->name = chip->label;
1901         range->id = alias_idx;
1902         range->pin_base = range->base = range->id * MAX_NB_GPIO_PER_BANK;
1903
1904         range->npins = chip->ngpio;
1905         range->gc = chip;
1906
1907         ret = at91_gpio_of_irq_setup(pdev, at91_chip);
1908         if (ret)
1909                 goto gpiochip_add_err;
1910
1911         ret = gpiochip_add_data(chip, at91_chip);
1912         if (ret)
1913                 goto gpiochip_add_err;
1914
1915         gpio_chips[alias_idx] = at91_chip;
1916         gpio_banks = max(gpio_banks, alias_idx + 1);
1917
1918         dev_info(&pdev->dev, "at address %p\n", at91_chip->regbase);
1919
1920         return 0;
1921
1922 gpiochip_add_err:
1923 clk_enable_err:
1924         clk_disable_unprepare(at91_chip->clock);
1925 err:
1926         dev_err(&pdev->dev, "Failure %i for GPIO %i\n", ret, alias_idx);
1927
1928         return ret;
1929 }
1930
1931 static struct platform_driver at91_gpio_driver = {
1932         .driver = {
1933                 .name = "gpio-at91",
1934                 .of_match_table = at91_gpio_of_match,
1935         },
1936         .probe = at91_gpio_probe,
1937 };
1938
1939 static struct platform_driver at91_pinctrl_driver = {
1940         .driver = {
1941                 .name = "pinctrl-at91",
1942                 .of_match_table = at91_pinctrl_of_match,
1943         },
1944         .probe = at91_pinctrl_probe,
1945 };
1946
1947 static struct platform_driver * const drivers[] = {
1948         &at91_gpio_driver,
1949         &at91_pinctrl_driver,
1950 };
1951
1952 static int __init at91_pinctrl_init(void)
1953 {
1954         return platform_register_drivers(drivers, ARRAY_SIZE(drivers));
1955 }
1956 arch_initcall(at91_pinctrl_init);