xref: /linux/drivers/pinctrl/sunxi/pinctrl-sunxi.c (revision 3aed61d1eb06b8b19b7bb09d49b222ebc3f83347)
1 /*
2  * Allwinner A1X SoCs pinctrl driver.
3  *
4  * Copyright (C) 2012 Maxime Ripard
5  *
6  * Maxime Ripard <maxime.ripard@free-electrons.com>
7  *
8  * This file is licensed under the terms of the GNU General Public
9  * License version 2.  This program is licensed "as is" without any
10  * warranty of any kind, whether express or implied.
11  */
12 
13 #include <linux/io.h>
14 #include <linux/clk.h>
15 #include <linux/gpio.h>
16 #include <linux/irqdomain.h>
17 #include <linux/irqchip/chained_irq.h>
18 #include <linux/module.h>
19 #include <linux/of.h>
20 #include <linux/of_address.h>
21 #include <linux/of_device.h>
22 #include <linux/of_irq.h>
23 #include <linux/pinctrl/consumer.h>
24 #include <linux/pinctrl/machine.h>
25 #include <linux/pinctrl/pinctrl.h>
26 #include <linux/pinctrl/pinconf-generic.h>
27 #include <linux/pinctrl/pinmux.h>
28 #include <linux/platform_device.h>
29 #include <linux/slab.h>
30 
31 #include "../core.h"
32 #include "../../gpio/gpiolib.h"
33 #include "pinctrl-sunxi.h"
34 
35 static struct irq_chip sunxi_pinctrl_edge_irq_chip;
36 static struct irq_chip sunxi_pinctrl_level_irq_chip;
37 
38 static struct sunxi_pinctrl_group *
39 sunxi_pinctrl_find_group_by_name(struct sunxi_pinctrl *pctl, const char *group)
40 {
41 	int i;
42 
43 	for (i = 0; i < pctl->ngroups; i++) {
44 		struct sunxi_pinctrl_group *grp = pctl->groups + i;
45 
46 		if (!strcmp(grp->name, group))
47 			return grp;
48 	}
49 
50 	return NULL;
51 }
52 
53 static struct sunxi_pinctrl_function *
54 sunxi_pinctrl_find_function_by_name(struct sunxi_pinctrl *pctl,
55 				    const char *name)
56 {
57 	struct sunxi_pinctrl_function *func = pctl->functions;
58 	int i;
59 
60 	for (i = 0; i < pctl->nfunctions; i++) {
61 		if (!func[i].name)
62 			break;
63 
64 		if (!strcmp(func[i].name, name))
65 			return func + i;
66 	}
67 
68 	return NULL;
69 }
70 
71 static struct sunxi_desc_function *
72 sunxi_pinctrl_desc_find_function_by_name(struct sunxi_pinctrl *pctl,
73 					 const char *pin_name,
74 					 const char *func_name)
75 {
76 	int i;
77 
78 	for (i = 0; i < pctl->desc->npins; i++) {
79 		const struct sunxi_desc_pin *pin = pctl->desc->pins + i;
80 
81 		if (!strcmp(pin->pin.name, pin_name)) {
82 			struct sunxi_desc_function *func = pin->functions;
83 
84 			while (func->name) {
85 				if (!strcmp(func->name, func_name))
86 					return func;
87 
88 				func++;
89 			}
90 		}
91 	}
92 
93 	return NULL;
94 }
95 
96 static struct sunxi_desc_function *
97 sunxi_pinctrl_desc_find_function_by_pin(struct sunxi_pinctrl *pctl,
98 					const u16 pin_num,
99 					const char *func_name)
100 {
101 	int i;
102 
103 	for (i = 0; i < pctl->desc->npins; i++) {
104 		const struct sunxi_desc_pin *pin = pctl->desc->pins + i;
105 
106 		if (pin->pin.number == pin_num) {
107 			struct sunxi_desc_function *func = pin->functions;
108 
109 			while (func->name) {
110 				if (!strcmp(func->name, func_name))
111 					return func;
112 
113 				func++;
114 			}
115 		}
116 	}
117 
118 	return NULL;
119 }
120 
121 static int sunxi_pctrl_get_groups_count(struct pinctrl_dev *pctldev)
122 {
123 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
124 
125 	return pctl->ngroups;
126 }
127 
128 static const char *sunxi_pctrl_get_group_name(struct pinctrl_dev *pctldev,
129 					      unsigned group)
130 {
131 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
132 
133 	return pctl->groups[group].name;
134 }
135 
136 static int sunxi_pctrl_get_group_pins(struct pinctrl_dev *pctldev,
137 				      unsigned group,
138 				      const unsigned **pins,
139 				      unsigned *num_pins)
140 {
141 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
142 
143 	*pins = (unsigned *)&pctl->groups[group].pin;
144 	*num_pins = 1;
145 
146 	return 0;
147 }
148 
149 static int sunxi_pctrl_dt_node_to_map(struct pinctrl_dev *pctldev,
150 				      struct device_node *node,
151 				      struct pinctrl_map **map,
152 				      unsigned *num_maps)
153 {
154 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
155 	unsigned long *pinconfig;
156 	struct property *prop;
157 	const char *function;
158 	const char *group;
159 	int ret, nmaps, i = 0;
160 	u32 val;
161 
162 	*map = NULL;
163 	*num_maps = 0;
164 
165 	ret = of_property_read_string(node, "allwinner,function", &function);
166 	if (ret) {
167 		dev_err(pctl->dev,
168 			"missing allwinner,function property in node %s\n",
169 			node->name);
170 		return -EINVAL;
171 	}
172 
173 	nmaps = of_property_count_strings(node, "allwinner,pins") * 2;
174 	if (nmaps < 0) {
175 		dev_err(pctl->dev,
176 			"missing allwinner,pins property in node %s\n",
177 			node->name);
178 		return -EINVAL;
179 	}
180 
181 	*map = kmalloc(nmaps * sizeof(struct pinctrl_map), GFP_KERNEL);
182 	if (!*map)
183 		return -ENOMEM;
184 
185 	of_property_for_each_string(node, "allwinner,pins", prop, group) {
186 		struct sunxi_pinctrl_group *grp =
187 			sunxi_pinctrl_find_group_by_name(pctl, group);
188 		int j = 0, configlen = 0;
189 
190 		if (!grp) {
191 			dev_err(pctl->dev, "unknown pin %s", group);
192 			continue;
193 		}
194 
195 		if (!sunxi_pinctrl_desc_find_function_by_name(pctl,
196 							      grp->name,
197 							      function)) {
198 			dev_err(pctl->dev, "unsupported function %s on pin %s",
199 				function, group);
200 			continue;
201 		}
202 
203 		(*map)[i].type = PIN_MAP_TYPE_MUX_GROUP;
204 		(*map)[i].data.mux.group = group;
205 		(*map)[i].data.mux.function = function;
206 
207 		i++;
208 
209 		(*map)[i].type = PIN_MAP_TYPE_CONFIGS_GROUP;
210 		(*map)[i].data.configs.group_or_pin = group;
211 
212 		if (of_find_property(node, "allwinner,drive", NULL))
213 			configlen++;
214 		if (of_find_property(node, "allwinner,pull", NULL))
215 			configlen++;
216 
217 		pinconfig = kzalloc(configlen * sizeof(*pinconfig), GFP_KERNEL);
218 		if (!pinconfig) {
219 			kfree(*map);
220 			return -ENOMEM;
221 		}
222 
223 		if (!of_property_read_u32(node, "allwinner,drive", &val)) {
224 			u16 strength = (val + 1) * 10;
225 			pinconfig[j++] =
226 				pinconf_to_config_packed(PIN_CONFIG_DRIVE_STRENGTH,
227 							 strength);
228 		}
229 
230 		if (!of_property_read_u32(node, "allwinner,pull", &val)) {
231 			enum pin_config_param pull = PIN_CONFIG_END;
232 			if (val == 1)
233 				pull = PIN_CONFIG_BIAS_PULL_UP;
234 			else if (val == 2)
235 				pull = PIN_CONFIG_BIAS_PULL_DOWN;
236 			pinconfig[j++] = pinconf_to_config_packed(pull, 0);
237 		}
238 
239 		(*map)[i].data.configs.configs = pinconfig;
240 		(*map)[i].data.configs.num_configs = configlen;
241 
242 		i++;
243 	}
244 
245 	*num_maps = nmaps;
246 
247 	return 0;
248 }
249 
250 static void sunxi_pctrl_dt_free_map(struct pinctrl_dev *pctldev,
251 				    struct pinctrl_map *map,
252 				    unsigned num_maps)
253 {
254 	int i;
255 
256 	for (i = 0; i < num_maps; i++) {
257 		if (map[i].type == PIN_MAP_TYPE_CONFIGS_GROUP)
258 			kfree(map[i].data.configs.configs);
259 	}
260 
261 	kfree(map);
262 }
263 
264 static const struct pinctrl_ops sunxi_pctrl_ops = {
265 	.dt_node_to_map		= sunxi_pctrl_dt_node_to_map,
266 	.dt_free_map		= sunxi_pctrl_dt_free_map,
267 	.get_groups_count	= sunxi_pctrl_get_groups_count,
268 	.get_group_name		= sunxi_pctrl_get_group_name,
269 	.get_group_pins		= sunxi_pctrl_get_group_pins,
270 };
271 
272 static int sunxi_pconf_group_get(struct pinctrl_dev *pctldev,
273 				 unsigned group,
274 				 unsigned long *config)
275 {
276 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
277 
278 	*config = pctl->groups[group].config;
279 
280 	return 0;
281 }
282 
283 static int sunxi_pconf_group_set(struct pinctrl_dev *pctldev,
284 				 unsigned group,
285 				 unsigned long *configs,
286 				 unsigned num_configs)
287 {
288 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
289 	struct sunxi_pinctrl_group *g = &pctl->groups[group];
290 	unsigned long flags;
291 	unsigned pin = g->pin - pctl->desc->pin_base;
292 	u32 val, mask;
293 	u16 strength;
294 	u8 dlevel;
295 	int i;
296 
297 	spin_lock_irqsave(&pctl->lock, flags);
298 
299 	for (i = 0; i < num_configs; i++) {
300 		switch (pinconf_to_config_param(configs[i])) {
301 		case PIN_CONFIG_DRIVE_STRENGTH:
302 			strength = pinconf_to_config_argument(configs[i]);
303 			if (strength > 40) {
304 				spin_unlock_irqrestore(&pctl->lock, flags);
305 				return -EINVAL;
306 			}
307 			/*
308 			 * We convert from mA to what the register expects:
309 			 *   0: 10mA
310 			 *   1: 20mA
311 			 *   2: 30mA
312 			 *   3: 40mA
313 			 */
314 			dlevel = strength / 10 - 1;
315 			val = readl(pctl->membase + sunxi_dlevel_reg(pin));
316 			mask = DLEVEL_PINS_MASK << sunxi_dlevel_offset(pin);
317 			writel((val & ~mask)
318 				| dlevel << sunxi_dlevel_offset(pin),
319 				pctl->membase + sunxi_dlevel_reg(pin));
320 			break;
321 		case PIN_CONFIG_BIAS_PULL_UP:
322 			val = readl(pctl->membase + sunxi_pull_reg(pin));
323 			mask = PULL_PINS_MASK << sunxi_pull_offset(pin);
324 			writel((val & ~mask) | 1 << sunxi_pull_offset(pin),
325 				pctl->membase + sunxi_pull_reg(pin));
326 			break;
327 		case PIN_CONFIG_BIAS_PULL_DOWN:
328 			val = readl(pctl->membase + sunxi_pull_reg(pin));
329 			mask = PULL_PINS_MASK << sunxi_pull_offset(pin);
330 			writel((val & ~mask) | 2 << sunxi_pull_offset(pin),
331 				pctl->membase + sunxi_pull_reg(pin));
332 			break;
333 		default:
334 			break;
335 		}
336 		/* cache the config value */
337 		g->config = configs[i];
338 	} /* for each config */
339 
340 	spin_unlock_irqrestore(&pctl->lock, flags);
341 
342 	return 0;
343 }
344 
345 static const struct pinconf_ops sunxi_pconf_ops = {
346 	.pin_config_group_get	= sunxi_pconf_group_get,
347 	.pin_config_group_set	= sunxi_pconf_group_set,
348 };
349 
350 static int sunxi_pmx_get_funcs_cnt(struct pinctrl_dev *pctldev)
351 {
352 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
353 
354 	return pctl->nfunctions;
355 }
356 
357 static const char *sunxi_pmx_get_func_name(struct pinctrl_dev *pctldev,
358 					   unsigned function)
359 {
360 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
361 
362 	return pctl->functions[function].name;
363 }
364 
365 static int sunxi_pmx_get_func_groups(struct pinctrl_dev *pctldev,
366 				     unsigned function,
367 				     const char * const **groups,
368 				     unsigned * const num_groups)
369 {
370 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
371 
372 	*groups = pctl->functions[function].groups;
373 	*num_groups = pctl->functions[function].ngroups;
374 
375 	return 0;
376 }
377 
378 static void sunxi_pmx_set(struct pinctrl_dev *pctldev,
379 				 unsigned pin,
380 				 u8 config)
381 {
382 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
383 	unsigned long flags;
384 	u32 val, mask;
385 
386 	spin_lock_irqsave(&pctl->lock, flags);
387 
388 	pin -= pctl->desc->pin_base;
389 	val = readl(pctl->membase + sunxi_mux_reg(pin));
390 	mask = MUX_PINS_MASK << sunxi_mux_offset(pin);
391 	writel((val & ~mask) | config << sunxi_mux_offset(pin),
392 		pctl->membase + sunxi_mux_reg(pin));
393 
394 	spin_unlock_irqrestore(&pctl->lock, flags);
395 }
396 
397 static int sunxi_pmx_set_mux(struct pinctrl_dev *pctldev,
398 			     unsigned function,
399 			     unsigned group)
400 {
401 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
402 	struct sunxi_pinctrl_group *g = pctl->groups + group;
403 	struct sunxi_pinctrl_function *func = pctl->functions + function;
404 	struct sunxi_desc_function *desc =
405 		sunxi_pinctrl_desc_find_function_by_name(pctl,
406 							 g->name,
407 							 func->name);
408 
409 	if (!desc)
410 		return -EINVAL;
411 
412 	sunxi_pmx_set(pctldev, g->pin, desc->muxval);
413 
414 	return 0;
415 }
416 
417 static int
418 sunxi_pmx_gpio_set_direction(struct pinctrl_dev *pctldev,
419 			struct pinctrl_gpio_range *range,
420 			unsigned offset,
421 			bool input)
422 {
423 	struct sunxi_pinctrl *pctl = pinctrl_dev_get_drvdata(pctldev);
424 	struct sunxi_desc_function *desc;
425 	const char *func;
426 
427 	if (input)
428 		func = "gpio_in";
429 	else
430 		func = "gpio_out";
431 
432 	desc = sunxi_pinctrl_desc_find_function_by_pin(pctl, offset, func);
433 	if (!desc)
434 		return -EINVAL;
435 
436 	sunxi_pmx_set(pctldev, offset, desc->muxval);
437 
438 	return 0;
439 }
440 
441 static const struct pinmux_ops sunxi_pmx_ops = {
442 	.get_functions_count	= sunxi_pmx_get_funcs_cnt,
443 	.get_function_name	= sunxi_pmx_get_func_name,
444 	.get_function_groups	= sunxi_pmx_get_func_groups,
445 	.set_mux		= sunxi_pmx_set_mux,
446 	.gpio_set_direction	= sunxi_pmx_gpio_set_direction,
447 };
448 
449 static int sunxi_pinctrl_gpio_request(struct gpio_chip *chip, unsigned offset)
450 {
451 	return pinctrl_request_gpio(chip->base + offset);
452 }
453 
454 static void sunxi_pinctrl_gpio_free(struct gpio_chip *chip, unsigned offset)
455 {
456 	pinctrl_free_gpio(chip->base + offset);
457 }
458 
459 static int sunxi_pinctrl_gpio_direction_input(struct gpio_chip *chip,
460 					unsigned offset)
461 {
462 	return pinctrl_gpio_direction_input(chip->base + offset);
463 }
464 
465 static int sunxi_pinctrl_gpio_get(struct gpio_chip *chip, unsigned offset)
466 {
467 	struct sunxi_pinctrl *pctl = dev_get_drvdata(chip->dev);
468 	u32 reg = sunxi_data_reg(offset);
469 	u8 index = sunxi_data_offset(offset);
470 	u32 set_mux = pctl->desc->irq_read_needs_mux &&
471 			test_bit(FLAG_USED_AS_IRQ, &chip->desc[offset].flags);
472 	u32 val;
473 
474 	if (set_mux)
475 		sunxi_pmx_set(pctl->pctl_dev, offset, SUN4I_FUNC_INPUT);
476 
477 	val = (readl(pctl->membase + reg) >> index) & DATA_PINS_MASK;
478 
479 	if (set_mux)
480 		sunxi_pmx_set(pctl->pctl_dev, offset, SUN4I_FUNC_IRQ);
481 
482 	return val;
483 }
484 
485 static void sunxi_pinctrl_gpio_set(struct gpio_chip *chip,
486 				unsigned offset, int value)
487 {
488 	struct sunxi_pinctrl *pctl = dev_get_drvdata(chip->dev);
489 	u32 reg = sunxi_data_reg(offset);
490 	u8 index = sunxi_data_offset(offset);
491 	unsigned long flags;
492 	u32 regval;
493 
494 	spin_lock_irqsave(&pctl->lock, flags);
495 
496 	regval = readl(pctl->membase + reg);
497 
498 	if (value)
499 		regval |= BIT(index);
500 	else
501 		regval &= ~(BIT(index));
502 
503 	writel(regval, pctl->membase + reg);
504 
505 	spin_unlock_irqrestore(&pctl->lock, flags);
506 }
507 
508 static int sunxi_pinctrl_gpio_direction_output(struct gpio_chip *chip,
509 					unsigned offset, int value)
510 {
511 	sunxi_pinctrl_gpio_set(chip, offset, value);
512 	return pinctrl_gpio_direction_output(chip->base + offset);
513 }
514 
515 static int sunxi_pinctrl_gpio_of_xlate(struct gpio_chip *gc,
516 				const struct of_phandle_args *gpiospec,
517 				u32 *flags)
518 {
519 	int pin, base;
520 
521 	base = PINS_PER_BANK * gpiospec->args[0];
522 	pin = base + gpiospec->args[1];
523 
524 	if (pin > gc->ngpio)
525 		return -EINVAL;
526 
527 	if (flags)
528 		*flags = gpiospec->args[2];
529 
530 	return pin;
531 }
532 
533 static int sunxi_pinctrl_gpio_to_irq(struct gpio_chip *chip, unsigned offset)
534 {
535 	struct sunxi_pinctrl *pctl = dev_get_drvdata(chip->dev);
536 	struct sunxi_desc_function *desc;
537 	unsigned pinnum = pctl->desc->pin_base + offset;
538 	unsigned irqnum;
539 
540 	if (offset >= chip->ngpio)
541 		return -ENXIO;
542 
543 	desc = sunxi_pinctrl_desc_find_function_by_pin(pctl, pinnum, "irq");
544 	if (!desc)
545 		return -EINVAL;
546 
547 	irqnum = desc->irqbank * IRQ_PER_BANK + desc->irqnum;
548 
549 	dev_dbg(chip->dev, "%s: request IRQ for GPIO %d, return %d\n",
550 		chip->label, offset + chip->base, irqnum);
551 
552 	return irq_find_mapping(pctl->domain, irqnum);
553 }
554 
555 static int sunxi_pinctrl_irq_request_resources(struct irq_data *d)
556 {
557 	struct sunxi_pinctrl *pctl = irq_data_get_irq_chip_data(d);
558 	struct sunxi_desc_function *func;
559 	int ret;
560 
561 	func = sunxi_pinctrl_desc_find_function_by_pin(pctl,
562 					pctl->irq_array[d->hwirq], "irq");
563 	if (!func)
564 		return -EINVAL;
565 
566 	ret = gpiochip_lock_as_irq(pctl->chip,
567 			pctl->irq_array[d->hwirq] - pctl->desc->pin_base);
568 	if (ret) {
569 		dev_err(pctl->dev, "unable to lock HW IRQ %lu for IRQ\n",
570 			irqd_to_hwirq(d));
571 		return ret;
572 	}
573 
574 	/* Change muxing to INT mode */
575 	sunxi_pmx_set(pctl->pctl_dev, pctl->irq_array[d->hwirq], func->muxval);
576 
577 	return 0;
578 }
579 
580 static void sunxi_pinctrl_irq_release_resources(struct irq_data *d)
581 {
582 	struct sunxi_pinctrl *pctl = irq_data_get_irq_chip_data(d);
583 
584 	gpiochip_unlock_as_irq(pctl->chip,
585 			      pctl->irq_array[d->hwirq] - pctl->desc->pin_base);
586 }
587 
588 static int sunxi_pinctrl_irq_set_type(struct irq_data *d, unsigned int type)
589 {
590 	struct sunxi_pinctrl *pctl = irq_data_get_irq_chip_data(d);
591 	u32 reg = sunxi_irq_cfg_reg(d->hwirq);
592 	u8 index = sunxi_irq_cfg_offset(d->hwirq);
593 	unsigned long flags;
594 	u32 regval;
595 	u8 mode;
596 
597 	switch (type) {
598 	case IRQ_TYPE_EDGE_RISING:
599 		mode = IRQ_EDGE_RISING;
600 		break;
601 	case IRQ_TYPE_EDGE_FALLING:
602 		mode = IRQ_EDGE_FALLING;
603 		break;
604 	case IRQ_TYPE_EDGE_BOTH:
605 		mode = IRQ_EDGE_BOTH;
606 		break;
607 	case IRQ_TYPE_LEVEL_HIGH:
608 		mode = IRQ_LEVEL_HIGH;
609 		break;
610 	case IRQ_TYPE_LEVEL_LOW:
611 		mode = IRQ_LEVEL_LOW;
612 		break;
613 	default:
614 		return -EINVAL;
615 	}
616 
617 	spin_lock_irqsave(&pctl->lock, flags);
618 
619 	if (type & IRQ_TYPE_LEVEL_MASK)
620 		__irq_set_chip_handler_name_locked(d->irq,
621 						   &sunxi_pinctrl_level_irq_chip,
622 						   handle_fasteoi_irq, NULL);
623 	else
624 		__irq_set_chip_handler_name_locked(d->irq,
625 						   &sunxi_pinctrl_edge_irq_chip,
626 						   handle_edge_irq, NULL);
627 
628 	regval = readl(pctl->membase + reg);
629 	regval &= ~(IRQ_CFG_IRQ_MASK << index);
630 	writel(regval | (mode << index), pctl->membase + reg);
631 
632 	spin_unlock_irqrestore(&pctl->lock, flags);
633 
634 	return 0;
635 }
636 
637 static void sunxi_pinctrl_irq_ack(struct irq_data *d)
638 {
639 	struct sunxi_pinctrl *pctl = irq_data_get_irq_chip_data(d);
640 	u32 status_reg = sunxi_irq_status_reg(d->hwirq);
641 	u8 status_idx = sunxi_irq_status_offset(d->hwirq);
642 
643 	/* Clear the IRQ */
644 	writel(1 << status_idx, pctl->membase + status_reg);
645 }
646 
647 static void sunxi_pinctrl_irq_mask(struct irq_data *d)
648 {
649 	struct sunxi_pinctrl *pctl = irq_data_get_irq_chip_data(d);
650 	u32 reg = sunxi_irq_ctrl_reg(d->hwirq);
651 	u8 idx = sunxi_irq_ctrl_offset(d->hwirq);
652 	unsigned long flags;
653 	u32 val;
654 
655 	spin_lock_irqsave(&pctl->lock, flags);
656 
657 	/* Mask the IRQ */
658 	val = readl(pctl->membase + reg);
659 	writel(val & ~(1 << idx), pctl->membase + reg);
660 
661 	spin_unlock_irqrestore(&pctl->lock, flags);
662 }
663 
664 static void sunxi_pinctrl_irq_unmask(struct irq_data *d)
665 {
666 	struct sunxi_pinctrl *pctl = irq_data_get_irq_chip_data(d);
667 	u32 reg = sunxi_irq_ctrl_reg(d->hwirq);
668 	u8 idx = sunxi_irq_ctrl_offset(d->hwirq);
669 	unsigned long flags;
670 	u32 val;
671 
672 	spin_lock_irqsave(&pctl->lock, flags);
673 
674 	/* Unmask the IRQ */
675 	val = readl(pctl->membase + reg);
676 	writel(val | (1 << idx), pctl->membase + reg);
677 
678 	spin_unlock_irqrestore(&pctl->lock, flags);
679 }
680 
681 static void sunxi_pinctrl_irq_ack_unmask(struct irq_data *d)
682 {
683 	sunxi_pinctrl_irq_ack(d);
684 	sunxi_pinctrl_irq_unmask(d);
685 }
686 
687 static struct irq_chip sunxi_pinctrl_edge_irq_chip = {
688 	.name		= "sunxi_pio_edge",
689 	.irq_ack	= sunxi_pinctrl_irq_ack,
690 	.irq_mask	= sunxi_pinctrl_irq_mask,
691 	.irq_unmask	= sunxi_pinctrl_irq_unmask,
692 	.irq_request_resources = sunxi_pinctrl_irq_request_resources,
693 	.irq_release_resources = sunxi_pinctrl_irq_release_resources,
694 	.irq_set_type	= sunxi_pinctrl_irq_set_type,
695 	.flags		= IRQCHIP_SKIP_SET_WAKE,
696 };
697 
698 static struct irq_chip sunxi_pinctrl_level_irq_chip = {
699 	.name		= "sunxi_pio_level",
700 	.irq_eoi	= sunxi_pinctrl_irq_ack,
701 	.irq_mask	= sunxi_pinctrl_irq_mask,
702 	.irq_unmask	= sunxi_pinctrl_irq_unmask,
703 	/* Define irq_enable / disable to avoid spurious irqs for drivers
704 	 * using these to suppress irqs while they clear the irq source */
705 	.irq_enable	= sunxi_pinctrl_irq_ack_unmask,
706 	.irq_disable	= sunxi_pinctrl_irq_mask,
707 	.irq_request_resources = sunxi_pinctrl_irq_request_resources,
708 	.irq_release_resources = sunxi_pinctrl_irq_release_resources,
709 	.irq_set_type	= sunxi_pinctrl_irq_set_type,
710 	.flags		= IRQCHIP_SKIP_SET_WAKE | IRQCHIP_EOI_THREADED |
711 			  IRQCHIP_EOI_IF_HANDLED,
712 };
713 
714 static int sunxi_pinctrl_irq_of_xlate(struct irq_domain *d,
715 				      struct device_node *node,
716 				      const u32 *intspec,
717 				      unsigned int intsize,
718 				      unsigned long *out_hwirq,
719 				      unsigned int *out_type)
720 {
721 	struct sunxi_desc_function *desc;
722 	int pin, base;
723 
724 	if (intsize < 3)
725 		return -EINVAL;
726 
727 	base = PINS_PER_BANK * intspec[0];
728 	pin = base + intspec[1];
729 
730 	desc = sunxi_pinctrl_desc_find_function_by_pin(d->host_data,
731 						       pin, "irq");
732 	if (!desc)
733 		return -EINVAL;
734 
735 	*out_hwirq = desc->irqbank * PINS_PER_BANK + desc->irqnum;
736 	*out_type = intspec[2];
737 
738 	return 0;
739 }
740 
741 static struct irq_domain_ops sunxi_pinctrl_irq_domain_ops = {
742 	.xlate		= sunxi_pinctrl_irq_of_xlate,
743 };
744 
745 static void sunxi_pinctrl_irq_handler(unsigned __irq, struct irq_desc *desc)
746 {
747 	unsigned int irq = irq_desc_get_irq(desc);
748 	struct irq_chip *chip = irq_desc_get_chip(desc);
749 	struct sunxi_pinctrl *pctl = irq_desc_get_handler_data(desc);
750 	unsigned long bank, reg, val;
751 
752 	for (bank = 0; bank < pctl->desc->irq_banks; bank++)
753 		if (irq == pctl->irq[bank])
754 			break;
755 
756 	if (bank == pctl->desc->irq_banks)
757 		return;
758 
759 	reg = sunxi_irq_status_reg_from_bank(bank);
760 	val = readl(pctl->membase + reg);
761 
762 	if (val) {
763 		int irqoffset;
764 
765 		chained_irq_enter(chip, desc);
766 		for_each_set_bit(irqoffset, &val, IRQ_PER_BANK) {
767 			int pin_irq = irq_find_mapping(pctl->domain,
768 						       bank * IRQ_PER_BANK + irqoffset);
769 			generic_handle_irq(pin_irq);
770 		}
771 		chained_irq_exit(chip, desc);
772 	}
773 }
774 
775 static int sunxi_pinctrl_add_function(struct sunxi_pinctrl *pctl,
776 					const char *name)
777 {
778 	struct sunxi_pinctrl_function *func = pctl->functions;
779 
780 	while (func->name) {
781 		/* function already there */
782 		if (strcmp(func->name, name) == 0) {
783 			func->ngroups++;
784 			return -EEXIST;
785 		}
786 		func++;
787 	}
788 
789 	func->name = name;
790 	func->ngroups = 1;
791 
792 	pctl->nfunctions++;
793 
794 	return 0;
795 }
796 
797 static int sunxi_pinctrl_build_state(struct platform_device *pdev)
798 {
799 	struct sunxi_pinctrl *pctl = platform_get_drvdata(pdev);
800 	int i;
801 
802 	pctl->ngroups = pctl->desc->npins;
803 
804 	/* Allocate groups */
805 	pctl->groups = devm_kzalloc(&pdev->dev,
806 				    pctl->ngroups * sizeof(*pctl->groups),
807 				    GFP_KERNEL);
808 	if (!pctl->groups)
809 		return -ENOMEM;
810 
811 	for (i = 0; i < pctl->desc->npins; i++) {
812 		const struct sunxi_desc_pin *pin = pctl->desc->pins + i;
813 		struct sunxi_pinctrl_group *group = pctl->groups + i;
814 
815 		group->name = pin->pin.name;
816 		group->pin = pin->pin.number;
817 	}
818 
819 	/*
820 	 * We suppose that we won't have any more functions than pins,
821 	 * we'll reallocate that later anyway
822 	 */
823 	pctl->functions = devm_kzalloc(&pdev->dev,
824 				pctl->desc->npins * sizeof(*pctl->functions),
825 				GFP_KERNEL);
826 	if (!pctl->functions)
827 		return -ENOMEM;
828 
829 	/* Count functions and their associated groups */
830 	for (i = 0; i < pctl->desc->npins; i++) {
831 		const struct sunxi_desc_pin *pin = pctl->desc->pins + i;
832 		struct sunxi_desc_function *func = pin->functions;
833 
834 		while (func->name) {
835 			/* Create interrupt mapping while we're at it */
836 			if (!strcmp(func->name, "irq")) {
837 				int irqnum = func->irqnum + func->irqbank * IRQ_PER_BANK;
838 				pctl->irq_array[irqnum] = pin->pin.number;
839 			}
840 
841 			sunxi_pinctrl_add_function(pctl, func->name);
842 			func++;
843 		}
844 	}
845 
846 	pctl->functions = krealloc(pctl->functions,
847 				pctl->nfunctions * sizeof(*pctl->functions),
848 				GFP_KERNEL);
849 
850 	for (i = 0; i < pctl->desc->npins; i++) {
851 		const struct sunxi_desc_pin *pin = pctl->desc->pins + i;
852 		struct sunxi_desc_function *func = pin->functions;
853 
854 		while (func->name) {
855 			struct sunxi_pinctrl_function *func_item;
856 			const char **func_grp;
857 
858 			func_item = sunxi_pinctrl_find_function_by_name(pctl,
859 									func->name);
860 			if (!func_item)
861 				return -EINVAL;
862 
863 			if (!func_item->groups) {
864 				func_item->groups =
865 					devm_kzalloc(&pdev->dev,
866 						     func_item->ngroups * sizeof(*func_item->groups),
867 						     GFP_KERNEL);
868 				if (!func_item->groups)
869 					return -ENOMEM;
870 			}
871 
872 			func_grp = func_item->groups;
873 			while (*func_grp)
874 				func_grp++;
875 
876 			*func_grp = pin->pin.name;
877 			func++;
878 		}
879 	}
880 
881 	return 0;
882 }
883 
884 int sunxi_pinctrl_init(struct platform_device *pdev,
885 		       const struct sunxi_pinctrl_desc *desc)
886 {
887 	struct device_node *node = pdev->dev.of_node;
888 	struct pinctrl_desc *pctrl_desc;
889 	struct pinctrl_pin_desc *pins;
890 	struct sunxi_pinctrl *pctl;
891 	struct resource *res;
892 	int i, ret, last_pin;
893 	struct clk *clk;
894 
895 	pctl = devm_kzalloc(&pdev->dev, sizeof(*pctl), GFP_KERNEL);
896 	if (!pctl)
897 		return -ENOMEM;
898 	platform_set_drvdata(pdev, pctl);
899 
900 	spin_lock_init(&pctl->lock);
901 
902 	res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
903 	pctl->membase = devm_ioremap_resource(&pdev->dev, res);
904 	if (IS_ERR(pctl->membase))
905 		return PTR_ERR(pctl->membase);
906 
907 	pctl->dev = &pdev->dev;
908 	pctl->desc = desc;
909 
910 	pctl->irq_array = devm_kcalloc(&pdev->dev,
911 				       IRQ_PER_BANK * pctl->desc->irq_banks,
912 				       sizeof(*pctl->irq_array),
913 				       GFP_KERNEL);
914 	if (!pctl->irq_array)
915 		return -ENOMEM;
916 
917 	ret = sunxi_pinctrl_build_state(pdev);
918 	if (ret) {
919 		dev_err(&pdev->dev, "dt probe failed: %d\n", ret);
920 		return ret;
921 	}
922 
923 	pins = devm_kzalloc(&pdev->dev,
924 			    pctl->desc->npins * sizeof(*pins),
925 			    GFP_KERNEL);
926 	if (!pins)
927 		return -ENOMEM;
928 
929 	for (i = 0; i < pctl->desc->npins; i++)
930 		pins[i] = pctl->desc->pins[i].pin;
931 
932 	pctrl_desc = devm_kzalloc(&pdev->dev,
933 				  sizeof(*pctrl_desc),
934 				  GFP_KERNEL);
935 	if (!pctrl_desc)
936 		return -ENOMEM;
937 
938 	pctrl_desc->name = dev_name(&pdev->dev);
939 	pctrl_desc->owner = THIS_MODULE;
940 	pctrl_desc->pins = pins;
941 	pctrl_desc->npins = pctl->desc->npins;
942 	pctrl_desc->confops = &sunxi_pconf_ops;
943 	pctrl_desc->pctlops = &sunxi_pctrl_ops;
944 	pctrl_desc->pmxops =  &sunxi_pmx_ops;
945 
946 	pctl->pctl_dev = pinctrl_register(pctrl_desc,
947 					  &pdev->dev, pctl);
948 	if (IS_ERR(pctl->pctl_dev)) {
949 		dev_err(&pdev->dev, "couldn't register pinctrl driver\n");
950 		return PTR_ERR(pctl->pctl_dev);
951 	}
952 
953 	pctl->chip = devm_kzalloc(&pdev->dev, sizeof(*pctl->chip), GFP_KERNEL);
954 	if (!pctl->chip) {
955 		ret = -ENOMEM;
956 		goto pinctrl_error;
957 	}
958 
959 	last_pin = pctl->desc->pins[pctl->desc->npins - 1].pin.number;
960 	pctl->chip->owner = THIS_MODULE;
961 	pctl->chip->request = sunxi_pinctrl_gpio_request,
962 	pctl->chip->free = sunxi_pinctrl_gpio_free,
963 	pctl->chip->direction_input = sunxi_pinctrl_gpio_direction_input,
964 	pctl->chip->direction_output = sunxi_pinctrl_gpio_direction_output,
965 	pctl->chip->get = sunxi_pinctrl_gpio_get,
966 	pctl->chip->set = sunxi_pinctrl_gpio_set,
967 	pctl->chip->of_xlate = sunxi_pinctrl_gpio_of_xlate,
968 	pctl->chip->to_irq = sunxi_pinctrl_gpio_to_irq,
969 	pctl->chip->of_gpio_n_cells = 3,
970 	pctl->chip->can_sleep = false,
971 	pctl->chip->ngpio = round_up(last_pin, PINS_PER_BANK) -
972 			    pctl->desc->pin_base;
973 	pctl->chip->label = dev_name(&pdev->dev);
974 	pctl->chip->dev = &pdev->dev;
975 	pctl->chip->base = pctl->desc->pin_base;
976 
977 	ret = gpiochip_add(pctl->chip);
978 	if (ret)
979 		goto pinctrl_error;
980 
981 	for (i = 0; i < pctl->desc->npins; i++) {
982 		const struct sunxi_desc_pin *pin = pctl->desc->pins + i;
983 
984 		ret = gpiochip_add_pin_range(pctl->chip, dev_name(&pdev->dev),
985 					     pin->pin.number - pctl->desc->pin_base,
986 					     pin->pin.number, 1);
987 		if (ret)
988 			goto gpiochip_error;
989 	}
990 
991 	clk = devm_clk_get(&pdev->dev, NULL);
992 	if (IS_ERR(clk)) {
993 		ret = PTR_ERR(clk);
994 		goto gpiochip_error;
995 	}
996 
997 	ret = clk_prepare_enable(clk);
998 	if (ret)
999 		goto gpiochip_error;
1000 
1001 	pctl->irq = devm_kcalloc(&pdev->dev,
1002 				 pctl->desc->irq_banks,
1003 				 sizeof(*pctl->irq),
1004 				 GFP_KERNEL);
1005 	if (!pctl->irq) {
1006 		ret = -ENOMEM;
1007 		goto clk_error;
1008 	}
1009 
1010 	for (i = 0; i < pctl->desc->irq_banks; i++) {
1011 		pctl->irq[i] = platform_get_irq(pdev, i);
1012 		if (pctl->irq[i] < 0) {
1013 			ret = pctl->irq[i];
1014 			goto clk_error;
1015 		}
1016 	}
1017 
1018 	pctl->domain = irq_domain_add_linear(node,
1019 					     pctl->desc->irq_banks * IRQ_PER_BANK,
1020 					     &sunxi_pinctrl_irq_domain_ops,
1021 					     pctl);
1022 	if (!pctl->domain) {
1023 		dev_err(&pdev->dev, "Couldn't register IRQ domain\n");
1024 		ret = -ENOMEM;
1025 		goto clk_error;
1026 	}
1027 
1028 	for (i = 0; i < (pctl->desc->irq_banks * IRQ_PER_BANK); i++) {
1029 		int irqno = irq_create_mapping(pctl->domain, i);
1030 
1031 		irq_set_chip_and_handler(irqno, &sunxi_pinctrl_edge_irq_chip,
1032 					 handle_edge_irq);
1033 		irq_set_chip_data(irqno, pctl);
1034 	};
1035 
1036 	for (i = 0; i < pctl->desc->irq_banks; i++) {
1037 		/* Mask and clear all IRQs before registering a handler */
1038 		writel(0, pctl->membase + sunxi_irq_ctrl_reg_from_bank(i));
1039 		writel(0xffffffff,
1040 			pctl->membase + sunxi_irq_status_reg_from_bank(i));
1041 
1042 		irq_set_chained_handler_and_data(pctl->irq[i],
1043 						 sunxi_pinctrl_irq_handler,
1044 						 pctl);
1045 	}
1046 
1047 	dev_info(&pdev->dev, "initialized sunXi PIO driver\n");
1048 
1049 	return 0;
1050 
1051 clk_error:
1052 	clk_disable_unprepare(clk);
1053 gpiochip_error:
1054 	gpiochip_remove(pctl->chip);
1055 pinctrl_error:
1056 	pinctrl_unregister(pctl->pctl_dev);
1057 	return ret;
1058 }
1059