mtd: ofpart: grab device tree node directly from master device node
[cascardo/linux.git] / drivers / mtd / ofpart.c
1 /*
2  * Flash partitions described by the OF (or flattened) device tree
3  *
4  * Copyright © 2006 MontaVista Software Inc.
5  * Author: Vitaly Wool <vwool@ru.mvista.com>
6  *
7  * Revised to handle newer style flash binding by:
8  *   Copyright © 2007 David Gibson, IBM Corporation.
9  *
10  * This program is free software; you can redistribute  it and/or modify it
11  * under  the terms of  the GNU General  Public License as published by the
12  * Free Software Foundation;  either version 2 of the  License, or (at your
13  * option) any later version.
14  */
15
16 #include <linux/module.h>
17 #include <linux/init.h>
18 #include <linux/of.h>
19 #include <linux/mtd/mtd.h>
20 #include <linux/slab.h>
21 #include <linux/mtd/partitions.h>
22
23 static bool node_has_compatible(struct device_node *pp)
24 {
25         return of_get_property(pp, "compatible", NULL);
26 }
27
28 static int parse_ofpart_partitions(struct mtd_info *master,
29                                    struct mtd_partition **pparts,
30                                    struct mtd_part_parser_data *data)
31 {
32         struct device_node *mtd_node;
33         struct device_node *ofpart_node;
34         const char *partname;
35         struct device_node *pp;
36         int nr_parts, i, ret = 0;
37         bool dedicated = true;
38
39
40         /*
41          * of_node can be provided through auxiliary parser data or (preferred)
42          * by assigning the master device node
43          */
44         mtd_node = data && data->of_node ? data->of_node : mtd_get_of_node(master);
45         if (!mtd_node)
46                 return 0;
47
48         ofpart_node = of_get_child_by_name(mtd_node, "partitions");
49         if (!ofpart_node) {
50                 pr_warn("%s: 'partitions' subnode not found on %s. Trying to parse direct subnodes as partitions.\n",
51                         master->name, mtd_node->full_name);
52                 ofpart_node = mtd_node;
53                 dedicated = false;
54         }
55
56         /* First count the subnodes */
57         nr_parts = 0;
58         for_each_child_of_node(ofpart_node,  pp) {
59                 if (!dedicated && node_has_compatible(pp))
60                         continue;
61
62                 nr_parts++;
63         }
64
65         if (nr_parts == 0)
66                 return 0;
67
68         *pparts = kzalloc(nr_parts * sizeof(**pparts), GFP_KERNEL);
69         if (!*pparts)
70                 return -ENOMEM;
71
72         i = 0;
73         for_each_child_of_node(ofpart_node,  pp) {
74                 const __be32 *reg;
75                 int len;
76                 int a_cells, s_cells;
77
78                 if (!dedicated && node_has_compatible(pp))
79                         continue;
80
81                 reg = of_get_property(pp, "reg", &len);
82                 if (!reg) {
83                         if (dedicated) {
84                                 pr_debug("%s: ofpart partition %s (%s) missing reg property.\n",
85                                          master->name, pp->full_name,
86                                          mtd_node->full_name);
87                                 goto ofpart_fail;
88                         } else {
89                                 nr_parts--;
90                                 continue;
91                         }
92                 }
93
94                 a_cells = of_n_addr_cells(pp);
95                 s_cells = of_n_size_cells(pp);
96                 if (len / 4 != a_cells + s_cells) {
97                         pr_debug("%s: ofpart partition %s (%s) error parsing reg property.\n",
98                                  master->name, pp->full_name,
99                                  mtd_node->full_name);
100                         goto ofpart_fail;
101                 }
102
103                 (*pparts)[i].offset = of_read_number(reg, a_cells);
104                 (*pparts)[i].size = of_read_number(reg + a_cells, s_cells);
105
106                 partname = of_get_property(pp, "label", &len);
107                 if (!partname)
108                         partname = of_get_property(pp, "name", &len);
109                 (*pparts)[i].name = partname;
110
111                 if (of_get_property(pp, "read-only", &len))
112                         (*pparts)[i].mask_flags |= MTD_WRITEABLE;
113
114                 if (of_get_property(pp, "lock", &len))
115                         (*pparts)[i].mask_flags |= MTD_POWERUP_LOCK;
116
117                 i++;
118         }
119
120         if (!nr_parts)
121                 goto ofpart_none;
122
123         return nr_parts;
124
125 ofpart_fail:
126         pr_err("%s: error parsing ofpart partition %s (%s)\n",
127                master->name, pp->full_name, mtd_node->full_name);
128         ret = -EINVAL;
129 ofpart_none:
130         of_node_put(pp);
131         kfree(*pparts);
132         *pparts = NULL;
133         return ret;
134 }
135
136 static struct mtd_part_parser ofpart_parser = {
137         .owner = THIS_MODULE,
138         .parse_fn = parse_ofpart_partitions,
139         .name = "ofpart",
140 };
141
142 static int parse_ofoldpart_partitions(struct mtd_info *master,
143                                       struct mtd_partition **pparts,
144                                       struct mtd_part_parser_data *data)
145 {
146         struct device_node *dp;
147         int i, plen, nr_parts;
148         const struct {
149                 __be32 offset, len;
150         } *part;
151         const char *names;
152
153         /*
154          * of_node can be provided through auxiliary parser data or (preferred)
155          * by assigning the master device node
156          */
157         dp = data && data->of_node ? data->of_node : mtd_get_of_node(master);
158         if (!dp)
159                 return 0;
160
161         part = of_get_property(dp, "partitions", &plen);
162         if (!part)
163                 return 0; /* No partitions found */
164
165         pr_warning("Device tree uses obsolete partition map binding: %s\n",
166                         dp->full_name);
167
168         nr_parts = plen / sizeof(part[0]);
169
170         *pparts = kzalloc(nr_parts * sizeof(*(*pparts)), GFP_KERNEL);
171         if (!*pparts)
172                 return -ENOMEM;
173
174         names = of_get_property(dp, "partition-names", &plen);
175
176         for (i = 0; i < nr_parts; i++) {
177                 (*pparts)[i].offset = be32_to_cpu(part->offset);
178                 (*pparts)[i].size   = be32_to_cpu(part->len) & ~1;
179                 /* bit 0 set signifies read only partition */
180                 if (be32_to_cpu(part->len) & 1)
181                         (*pparts)[i].mask_flags = MTD_WRITEABLE;
182
183                 if (names && (plen > 0)) {
184                         int len = strlen(names) + 1;
185
186                         (*pparts)[i].name = names;
187                         plen -= len;
188                         names += len;
189                 } else {
190                         (*pparts)[i].name = "unnamed";
191                 }
192
193                 part++;
194         }
195
196         return nr_parts;
197 }
198
199 static struct mtd_part_parser ofoldpart_parser = {
200         .owner = THIS_MODULE,
201         .parse_fn = parse_ofoldpart_partitions,
202         .name = "ofoldpart",
203 };
204
205 static int __init ofpart_parser_init(void)
206 {
207         register_mtd_parser(&ofpart_parser);
208         register_mtd_parser(&ofoldpart_parser);
209         return 0;
210 }
211
212 static void __exit ofpart_parser_exit(void)
213 {
214         deregister_mtd_parser(&ofpart_parser);
215         deregister_mtd_parser(&ofoldpart_parser);
216 }
217
218 module_init(ofpart_parser_init);
219 module_exit(ofpart_parser_exit);
220
221 MODULE_LICENSE("GPL");
222 MODULE_DESCRIPTION("Parser for MTD partitioning information in device tree");
223 MODULE_AUTHOR("Vitaly Wool, David Gibson");
224 /*
225  * When MTD core cannot find the requested parser, it tries to load the module
226  * with the same name. Since we provide the ofoldpart parser, we should have
227  * the corresponding alias.
228  */
229 MODULE_ALIAS("ofoldpart");