1 // SPDX-License-Identifier: GPL-2.0
3 * MMIO register bitfield-controlled multiplexer driver
4 * Based on the linux mmio multiplexer driver
10 #include <mux-internal.h>
13 #include <dm/device.h>
14 #include <dm/device-internal.h>
15 #include <dm/device_compat.h>
17 #include <dm/devres.h>
18 #include <dt-bindings/mux/mux.h>
19 #include <linux/bitops.h>
21 static int mux_mmio_set(struct mux_control *mux, int state)
23 struct regmap_field **fields = dev_get_priv(mux->dev);
25 return regmap_field_write(fields[mux_control_get_index(mux)], state);
28 static const struct mux_control_ops mux_mmio_ops = {
32 static const struct udevice_id mmio_mux_of_match[] = {
33 { .compatible = "mmio-mux" },
34 { .compatible = "reg-mux" },
38 static int mmio_mux_probe(struct udevice *dev)
40 struct regmap_field **fields;
41 struct mux_chip *mux_chip = dev_get_uclass_priv(dev);
42 struct regmap *regmap;
49 if (ofnode_device_is_compatible(dev_ofnode(dev), "mmio-mux"))
50 regmap = syscon_node_to_regmap(dev_ofnode(dev->parent));
52 regmap_init_mem(dev_ofnode(dev), ®map);
55 ret = PTR_ERR(regmap);
56 dev_err(dev, "failed to get regmap: %d\n", ret);
60 num_fields = dev_read_size(dev, "mux-reg-masks");
62 return log_msg_ret("mux-reg-masks missing", -EINVAL);
64 num_fields /= sizeof(u32);
65 if (num_fields == 0 || num_fields % 2)
67 num_fields = num_fields / 2;
69 ret = mux_alloc_controllers(dev, num_fields);
71 return log_msg_ret("mux_alloc_controllers", ret);
73 fields = devm_kmalloc(dev, num_fields * sizeof(*fields), __GFP_ZERO);
76 dev_set_priv(dev, fields);
78 mux_reg_masks = devm_kmalloc(dev, num_fields * 2 * sizeof(u32),
83 ret = dev_read_u32_array(dev, "mux-reg-masks", mux_reg_masks,
86 return log_msg_ret("mux-reg-masks read", ret);
88 idle_states = devm_kmalloc(dev, num_fields * sizeof(u32), __GFP_ZERO);
92 ret = dev_read_u32_array(dev, "idle-states", idle_states, num_fields);
94 log_err("idle-states");
95 devm_kfree(dev, idle_states);
99 for (i = 0; i < num_fields; i++) {
100 struct mux_control *mux = &mux_chip->mux[i];
101 struct reg_field field;
105 reg = mux_reg_masks[2 * i];
106 mask = mux_reg_masks[2 * i + 1];
109 field.msb = fls(mask) - 1;
110 field.lsb = ffs(mask) - 1;
112 if (mask != GENMASK(field.msb, field.lsb))
113 return log_msg_ret("invalid mask", -EINVAL);
115 fields[i] = devm_regmap_field_alloc(dev, regmap, field);
116 if (IS_ERR(fields[i])) {
117 ret = PTR_ERR(fields[i]);
118 return log_msg_ret("regmap_field_alloc", ret);
121 bits = 1 + field.msb - field.lsb;
122 mux->states = 1 << bits;
127 if (idle_states[i] != MUX_IDLE_AS_IS &&
128 idle_states[i] >= mux->states)
129 return log_msg_ret("idle-states range", -EINVAL);
131 mux->idle_state = idle_states[i];
134 devm_kfree(dev, mux_reg_masks);
136 devm_kfree(dev, idle_states);
141 U_BOOT_DRIVER(mmio_mux) = {
144 .of_match = mmio_mux_of_match,
145 .probe = mmio_mux_probe,
146 .ops = &mux_mmio_ops,