kill hotplug2 before running init
[openwrt.git] / target / linux / generic-2.6 / patches-2.6.25 / 065-rootfs_split.patch
1 Index: linux-2.6.23-rc6/drivers/mtd/Kconfig
2 ===================================================================
3 --- linux-2.6.23-rc6.orig/drivers/mtd/Kconfig   2007-09-21 16:23:54.000000000 +0800
4 +++ linux-2.6.23-rc6/drivers/mtd/Kconfig        2007-09-21 16:23:59.000000000 +0800
5 @@ -47,6 +47,16 @@
6           devices. Partitioning on NFTL 'devices' is a different - that's the
7           'normal' form of partitioning used on a block device.
8  
9 +config MTD_ROOTFS_ROOT_DEV
10 +       bool "Automatically set 'rootfs' partition to be root filesystem"
11 +       depends on MTD_PARTITIONS
12 +       default y
13 +
14 +config MTD_ROOTFS_SPLIT
15 +       bool "Automatically split 'rootfs' partition for squashfs"
16 +       depends on MTD_PARTITIONS
17 +       default y
18 +
19  config MTD_REDBOOT_PARTS
20         tristate "RedBoot partition table parsing"
21         depends on MTD_PARTITIONS
22 Index: linux-2.6.23-rc6/drivers/mtd/mtdpart.c
23 ===================================================================
24 --- linux-2.6.23-rc6.orig/drivers/mtd/mtdpart.c 2007-09-21 16:23:54.000000000 +0800
25 +++ linux-2.6.23-rc6/drivers/mtd/mtdpart.c      2007-09-21 16:23:59.000000000 +0800
26 @@ -20,6 +20,8 @@
27  #include <linux/mtd/mtd.h>
28  #include <linux/mtd/partitions.h>
29  #include <linux/mtd/compatmac.h>
30 +#include <linux/squashfs_fs.h>
31 +#include <linux/root_dev.h>
32  
33  /* Our partition linked list */
34  static LIST_HEAD(mtd_partitions);
35 @@ -39,7 +41,7 @@
36   * the pointer to that structure with this macro.
37   */
38  #define PART(x)  ((struct mtd_part *)(x))
39 -
40 +#define IS_PART(mtd) (mtd->read == part_read)
41  
42  /*
43   * MTD methods which simply translate the effective address and pass through
44 @@ -322,6 +324,316 @@
45         return 0;
46  }
47  
48 +static u_int32_t cur_offset = 0;
49 +static int add_one_partition(struct mtd_info *master, const struct mtd_partition *part,
50 +               int i, struct mtd_part **slp)
51 +{
52 +       struct mtd_part *slave;
53 +
54 +       /* allocate the partition structure */
55 +       slave = kzalloc (sizeof(*slave), GFP_KERNEL);
56 +       if (!slave) {
57 +               printk ("memory allocation error while creating partitions for \"%s\"\n",
58 +                       master->name);
59 +               del_mtd_partitions(master);
60 +               return -ENOMEM;
61 +       }
62 +       list_add(&slave->list, &mtd_partitions);
63 +
64 +       /* set up the MTD object for this partition */
65 +       slave->mtd.type = master->type;
66 +       slave->mtd.flags = master->flags & ~part->mask_flags;
67 +       slave->mtd.size = part->size;
68 +       slave->mtd.writesize = master->writesize;
69 +       slave->mtd.oobsize = master->oobsize;
70 +       slave->mtd.oobavail = master->oobavail;
71 +       slave->mtd.subpage_sft = master->subpage_sft;
72 +
73 +       slave->mtd.name = part->name;
74 +       slave->mtd.owner = master->owner;
75 +
76 +       slave->mtd.read = part_read;
77 +       slave->mtd.write = part_write;
78 +
79 +       if (master->panic_write)
80 +               slave->mtd.panic_write = part_panic_write;
81 +
82 +       slave->mtd.refresh_device = part->refresh_partition;
83 +
84 +       if(master->point && master->unpoint){
85 +               slave->mtd.point = part_point;
86 +               slave->mtd.unpoint = part_unpoint;
87 +       }
88 +
89 +       if (master->read_oob)
90 +               slave->mtd.read_oob = part_read_oob;
91 +       if (master->write_oob)
92 +               slave->mtd.write_oob = part_write_oob;
93 +       if(master->read_user_prot_reg)
94 +               slave->mtd.read_user_prot_reg = part_read_user_prot_reg;
95 +       if(master->read_fact_prot_reg)
96 +               slave->mtd.read_fact_prot_reg = part_read_fact_prot_reg;
97 +       if(master->write_user_prot_reg)
98 +               slave->mtd.write_user_prot_reg = part_write_user_prot_reg;
99 +       if(master->lock_user_prot_reg)
100 +               slave->mtd.lock_user_prot_reg = part_lock_user_prot_reg;
101 +       if(master->get_user_prot_info)
102 +               slave->mtd.get_user_prot_info = part_get_user_prot_info;
103 +       if(master->get_fact_prot_info)
104 +               slave->mtd.get_fact_prot_info = part_get_fact_prot_info;
105 +       if (master->sync)
106 +               slave->mtd.sync = part_sync;
107 +       if (!i && master->suspend && master->resume) {
108 +                       slave->mtd.suspend = part_suspend;
109 +                       slave->mtd.resume = part_resume;
110 +       }
111 +       if (master->writev)
112 +               slave->mtd.writev = part_writev;
113 +       if (master->lock)
114 +               slave->mtd.lock = part_lock;
115 +       if (master->unlock)
116 +               slave->mtd.unlock = part_unlock;
117 +       if (master->block_isbad)
118 +               slave->mtd.block_isbad = part_block_isbad;
119 +       if (master->block_markbad)
120 +               slave->mtd.block_markbad = part_block_markbad;
121 +       slave->mtd.erase = part_erase;
122 +       slave->master = master;
123 +       slave->offset = part->offset;
124 +       slave->index = i;
125 +
126 +       if (slave->offset == MTDPART_OFS_APPEND)
127 +               slave->offset = cur_offset;
128 +       if (slave->offset == MTDPART_OFS_NXTBLK) {
129 +               slave->offset = cur_offset;
130 +               if ((cur_offset % master->erasesize) != 0) {
131 +                       /* Round up to next erasesize */
132 +                       slave->offset = ((cur_offset / master->erasesize) + 1) * master->erasesize;
133 +                       printk(KERN_NOTICE "Moving partition %d: "
134 +                              "0x%08x -> 0x%08x\n", i,
135 +                              cur_offset, slave->offset);
136 +               }
137 +       }
138 +       if (slave->mtd.size == MTDPART_SIZ_FULL)
139 +               slave->mtd.size = master->size - slave->offset;
140 +       cur_offset = slave->offset + slave->mtd.size;
141 +
142 +       printk (KERN_NOTICE "0x%08x-0x%08x : \"%s\"\n", slave->offset,
143 +               slave->offset + slave->mtd.size, slave->mtd.name);
144 +
145 +       /* let's do some sanity checks */
146 +       if (slave->offset >= master->size) {
147 +                       /* let's register it anyway to preserve ordering */
148 +               slave->offset = 0;
149 +               slave->mtd.size = 0;
150 +               printk ("mtd: partition \"%s\" is out of reach -- disabled\n",
151 +                       part->name);
152 +       }
153 +       if (slave->offset + slave->mtd.size > master->size) {
154 +               slave->mtd.size = master->size - slave->offset;
155 +               printk ("mtd: partition \"%s\" extends beyond the end of device \"%s\" -- size truncated to %#x\n",
156 +                       part->name, master->name, slave->mtd.size);
157 +       }
158 +       if (master->numeraseregions>1) {
159 +               /* Deal with variable erase size stuff */
160 +               int i;
161 +               struct mtd_erase_region_info *regions = master->eraseregions;
162 +
163 +               /* Find the first erase regions which is part of this partition. */
164 +               for (i=0; i < master->numeraseregions && slave->offset >= regions[i].offset; i++)
165 +                       ;
166 +
167 +               for (i--; i < master->numeraseregions && slave->offset + slave->mtd.size > regions[i].offset; i++) {
168 +                       if (slave->mtd.erasesize < regions[i].erasesize) {
169 +                               slave->mtd.erasesize = regions[i].erasesize;
170 +                       }
171 +               }
172 +       } else {
173 +               /* Single erase size */
174 +               slave->mtd.erasesize = master->erasesize;
175 +       }
176 +
177 +       if ((slave->mtd.flags & MTD_WRITEABLE) &&
178 +           (slave->offset % slave->mtd.erasesize)) {
179 +               /* Doesn't start on a boundary of major erase size */
180 +               /* FIXME: Let it be writable if it is on a boundary of _minor_ erase size though */
181 +               slave->mtd.flags &= ~MTD_WRITEABLE;
182 +               printk ("mtd: partition \"%s\" doesn't start on an erase block boundary -- force read-only\n",
183 +                       part->name);
184 +       }
185 +       if ((slave->mtd.flags & MTD_WRITEABLE) &&
186 +           (slave->mtd.size % slave->mtd.erasesize)) {
187 +               slave->mtd.flags &= ~MTD_WRITEABLE;
188 +               printk ("mtd: partition \"%s\" doesn't end on an erase block -- force read-only\n",
189 +                       part->name);
190 +       }
191 +
192 +       slave->mtd.ecclayout = master->ecclayout;
193 +       if (master->block_isbad) {
194 +               uint32_t offs = 0;
195 +
196 +               while(offs < slave->mtd.size) {
197 +                       if (master->block_isbad(master,
198 +                                               offs + slave->offset))
199 +                               slave->mtd.ecc_stats.badblocks++;
200 +                       offs += slave->mtd.erasesize;
201 +               }
202 +       }
203 +
204 +       if(part->mtdp)
205 +       {       /* store the object pointer (caller may or may not register it */
206 +               *part->mtdp = &slave->mtd;
207 +               slave->registered = 0;
208 +       }
209 +       else
210 +       {
211 +               /* register our partition */
212 +               add_mtd_device(&slave->mtd);
213 +               slave->registered = 1;
214 +       }
215 +
216 +       if (slp)
217 +               *slp = slave;
218 +
219 +       return 0;
220 +}
221 +
222 +#ifdef CONFIG_MTD_ROOTFS_SPLIT
223 +#define ROOTFS_SPLIT_NAME "rootfs_data"
224 +#define ROOTFS_REMOVED_NAME "<removed>"
225 +static int split_squashfs(struct mtd_info *master, int offset, int *split_offset)
226 +{
227 +       char buf[512];
228 +       struct squashfs_super_block *sb = (struct squashfs_super_block *) buf;
229 +       int len, ret;
230 +
231 +       ret = master->read(master, offset, sizeof(*sb), &len, buf);
232 +       if (ret || (len != sizeof(*sb))) {
233 +               printk(KERN_ALERT "split_squashfs: error occured while reading "
234 +                       "from \"%s\"\n", master->name);
235 +               return -EINVAL;
236 +       }
237 +
238 +       if (*((u32 *) buf) != SQUASHFS_MAGIC) {
239 +               printk(KERN_ALERT "split_squashfs: no squashfs found in \"%s\"\n",
240 +                       master->name);
241 +               *split_offset = 0;
242 +               return 0;
243 +       }
244 +
245 +       if (sb->bytes_used <= 0) {
246 +               printk(KERN_ALERT "split_squashfs: squashfs is empty in \"%s\"\n",
247 +                       master->name);
248 +               *split_offset = 0;
249 +               return 0;
250 +       }
251 +
252 +       len = (u32) sb->bytes_used;
253 +       len += (offset & 0x000fffff);
254 +       len +=  (master->erasesize - 1);
255 +       len &= ~(master->erasesize - 1);
256 +       len -= (offset & 0x000fffff);
257 +       *split_offset = offset + len;
258 +
259 +       return 0;
260 +}
261 +
262 +static int split_rootfs_data(struct mtd_info *master, struct mtd_info *rpart, struct mtd_partition *part,
263 +               int index)
264 +{
265 +       struct mtd_partition *dpart;
266 +       struct mtd_part *slave = NULL;
267 +       int split_offset = 0;
268 +       int ret;
269 +
270 +       ret = split_squashfs(master, part->offset, &split_offset);
271 +       if (ret)
272 +               return ret;
273 +
274 +       if (split_offset <= 0)
275 +               return 0;
276 +
277 +       dpart = kmalloc(sizeof(*part)+sizeof(ROOTFS_SPLIT_NAME)+1, GFP_KERNEL);
278 +       if (dpart == NULL) {
279 +               printk(KERN_INFO "split_squashfs: no memory for partition \"%s\"\n",
280 +                       ROOTFS_SPLIT_NAME);
281 +               return -ENOMEM;
282 +       }
283 +
284 +       memcpy(dpart, part, sizeof(*part));
285 +       dpart->name = (unsigned char *)&dpart[1];
286 +       strcpy(dpart->name, ROOTFS_SPLIT_NAME);
287 +
288 +       dpart->size -= split_offset - dpart->offset;
289 +       dpart->offset = split_offset;
290 +
291 +       if (dpart == NULL)
292 +               return 1;
293 +
294 +       printk(KERN_INFO "mtd: partition \"%s\" created automatically, ofs=%X, len=%X \n",
295 +               ROOTFS_SPLIT_NAME, dpart->offset, dpart->size);
296 +
297 +       ret = add_one_partition(master, dpart, index, &slave);
298 +       if (ret)
299 +               kfree(dpart);
300 +       else if (slave)
301 +               rpart->split = &slave->mtd;
302 +
303 +       return ret;
304 +}
305 +
306 +static int refresh_rootfs_split(struct mtd_info *mtd)
307 +{
308 +       struct mtd_partition tpart;
309 +       struct mtd_part *part;
310 +       int index = 0;
311 +       int offset, size;
312 +       int ret;
313 +
314 +       part = PART(mtd);
315 +
316 +       /* check for the new squashfs offset first */
317 +       ret = split_squashfs(part->master, part->offset, &offset);
318 +       if (ret)
319 +               return ret;
320 +
321 +       if ((offset > 0) && !mtd->split) {
322 +               printk(KERN_INFO "%s: creating new split partition for \"%s\"\n", __func__, mtd->name);
323 +               /* if we don't have a rootfs split partition, create a new one */
324 +               tpart.name = mtd->name;
325 +               tpart.size = mtd->size;
326 +               tpart.offset = part->offset;
327 +
328 +               /* find the index of the last partition */
329 +               if (!list_empty(&mtd_partitions))
330 +                       index = list_first_entry(&mtd_partitions, struct mtd_part, list)->index + 1;
331 +
332 +               return split_rootfs_data(part->master, &part->mtd, &tpart, index);
333 +       } else if ((offset > 0) && mtd->split) {
334 +               /* update the offsets of the existing partition */
335 +               size = mtd->size + part->offset - offset;
336 +
337 +               part = PART(mtd->split);
338 +               part->offset = offset;
339 +               part->mtd.size = size;
340 +               printk(KERN_INFO "%s: %s partition \"" ROOTFS_SPLIT_NAME "\", offset: 0x%06x (0x%06x)\n",
341 +                       __func__, (!strcmp(part->mtd.name, ROOTFS_SPLIT_NAME) ? "updating" : "creating"),
342 +                       part->offset, part->mtd.size);
343 +               strcpy(part->mtd.name, ROOTFS_SPLIT_NAME);
344 +       } else if ((offset <= 0) && mtd->split) {
345 +               printk(KERN_INFO "%s: removing partition \"%s\"\n", __func__, mtd->split->name);
346 +
347 +               /* mark existing partition as removed */
348 +               part = PART(mtd->split);
349 +               strcpy(part->mtd.name, ROOTFS_REMOVED_NAME);
350 +               part->offset = 0;
351 +               part->mtd.size = 0;
352 +       }
353 +
354 +       return 0;
355 +}
356 +#endif /* CONFIG_MTD_ROOTFS_SPLIT */
357 +
358  /*
359   * This function, given a master MTD object and a partition table, creates
360   * and registers slave MTD objects which are bound to the master according to
361 @@ -334,171 +646,31 @@
362                        int nbparts)
363  {
364         struct mtd_part *slave;
365 -       u_int32_t cur_offset = 0;
366 -       int i;
367 +       struct mtd_partition *part;
368 +       int i, j, ret = 0;
369  
370         printk (KERN_NOTICE "Creating %d MTD partitions on \"%s\":\n", nbparts, master->name);
371  
372 -       for (i = 0; i < nbparts; i++) {
373 -
374 -               /* allocate the partition structure */
375 -               slave = kzalloc (sizeof(*slave), GFP_KERNEL);
376 -               if (!slave) {
377 -                       printk ("memory allocation error while creating partitions for \"%s\"\n",
378 -                               master->name);
379 -                       del_mtd_partitions(master);
380 -                       return -ENOMEM;
381 -               }
382 -               list_add(&slave->list, &mtd_partitions);
383 -
384 -               /* set up the MTD object for this partition */
385 -               slave->mtd.type = master->type;
386 -               slave->mtd.flags = master->flags & ~parts[i].mask_flags;
387 -               slave->mtd.size = parts[i].size;
388 -               slave->mtd.writesize = master->writesize;
389 -               slave->mtd.oobsize = master->oobsize;
390 -               slave->mtd.oobavail = master->oobavail;
391 -               slave->mtd.subpage_sft = master->subpage_sft;
392 -
393 -               slave->mtd.name = parts[i].name;
394 -               slave->mtd.owner = master->owner;
395 -
396 -               slave->mtd.read = part_read;
397 -               slave->mtd.write = part_write;
398 -
399 -               if (master->panic_write)
400 -                       slave->mtd.panic_write = part_panic_write;
401 -
402 -               if(master->point && master->unpoint){
403 -                       slave->mtd.point = part_point;
404 -                       slave->mtd.unpoint = part_unpoint;
405 -               }
406 -
407 -               if (master->read_oob)
408 -                       slave->mtd.read_oob = part_read_oob;
409 -               if (master->write_oob)
410 -                       slave->mtd.write_oob = part_write_oob;
411 -               if(master->read_user_prot_reg)
412 -                       slave->mtd.read_user_prot_reg = part_read_user_prot_reg;
413 -               if(master->read_fact_prot_reg)
414 -                       slave->mtd.read_fact_prot_reg = part_read_fact_prot_reg;
415 -               if(master->write_user_prot_reg)
416 -                       slave->mtd.write_user_prot_reg = part_write_user_prot_reg;
417 -               if(master->lock_user_prot_reg)
418 -                       slave->mtd.lock_user_prot_reg = part_lock_user_prot_reg;
419 -               if(master->get_user_prot_info)
420 -                       slave->mtd.get_user_prot_info = part_get_user_prot_info;
421 -               if(master->get_fact_prot_info)
422 -                       slave->mtd.get_fact_prot_info = part_get_fact_prot_info;
423 -               if (master->sync)
424 -                       slave->mtd.sync = part_sync;
425 -               if (!i && master->suspend && master->resume) {
426 -                               slave->mtd.suspend = part_suspend;
427 -                               slave->mtd.resume = part_resume;
428 -               }
429 -               if (master->writev)
430 -                       slave->mtd.writev = part_writev;
431 -               if (master->lock)
432 -                       slave->mtd.lock = part_lock;
433 -               if (master->unlock)
434 -                       slave->mtd.unlock = part_unlock;
435 -               if (master->block_isbad)
436 -                       slave->mtd.block_isbad = part_block_isbad;
437 -               if (master->block_markbad)
438 -                       slave->mtd.block_markbad = part_block_markbad;
439 -               slave->mtd.erase = part_erase;
440 -               slave->master = master;
441 -               slave->offset = parts[i].offset;
442 -               slave->index = i;
443 -
444 -               if (slave->offset == MTDPART_OFS_APPEND)
445 -                       slave->offset = cur_offset;
446 -               if (slave->offset == MTDPART_OFS_NXTBLK) {
447 -                       slave->offset = cur_offset;
448 -                       if ((cur_offset % master->erasesize) != 0) {
449 -                               /* Round up to next erasesize */
450 -                               slave->offset = ((cur_offset / master->erasesize) + 1) * master->erasesize;
451 -                               printk(KERN_NOTICE "Moving partition %d: "
452 -                                      "0x%08x -> 0x%08x\n", i,
453 -                                      cur_offset, slave->offset);
454 -                       }
455 -               }
456 -               if (slave->mtd.size == MTDPART_SIZ_FULL)
457 -                       slave->mtd.size = master->size - slave->offset;
458 -               cur_offset = slave->offset + slave->mtd.size;
459 -
460 -               printk (KERN_NOTICE "0x%08x-0x%08x : \"%s\"\n", slave->offset,
461 -                       slave->offset + slave->mtd.size, slave->mtd.name);
462 -
463 -               /* let's do some sanity checks */
464 -               if (slave->offset >= master->size) {
465 -                               /* let's register it anyway to preserve ordering */
466 -                       slave->offset = 0;
467 -                       slave->mtd.size = 0;
468 -                       printk ("mtd: partition \"%s\" is out of reach -- disabled\n",
469 -                               parts[i].name);
470 -               }
471 -               if (slave->offset + slave->mtd.size > master->size) {
472 -                       slave->mtd.size = master->size - slave->offset;
473 -                       printk ("mtd: partition \"%s\" extends beyond the end of device \"%s\" -- size truncated to %#x\n",
474 -                               parts[i].name, master->name, slave->mtd.size);
475 -               }
476 -               if (master->numeraseregions>1) {
477 -                       /* Deal with variable erase size stuff */
478 -                       int i;
479 -                       struct mtd_erase_region_info *regions = master->eraseregions;
480 -
481 -                       /* Find the first erase regions which is part of this partition. */
482 -                       for (i=0; i < master->numeraseregions && slave->offset >= regions[i].offset; i++)
483 -                               ;
484 -
485 -                       for (i--; i < master->numeraseregions && slave->offset + slave->mtd.size > regions[i].offset; i++) {
486 -                               if (slave->mtd.erasesize < regions[i].erasesize) {
487 -                                       slave->mtd.erasesize = regions[i].erasesize;
488 -                               }
489 +       for (i = 0, j = 0; i < nbparts; i++) {
490 +               part = (struct mtd_partition *) &parts[i];
491 +               ret = add_one_partition(master, part, j, &slave);
492 +               if (ret)
493 +                       return ret;
494 +               j++;
495 +
496 +               if (strcmp(part->name, "rootfs") == 0 && slave->registered) {
497 +#ifdef CONFIG_MTD_ROOTFS_ROOT_DEV
498 +                       if (ROOT_DEV == 0) {
499 +                               printk(KERN_NOTICE "mtd: partition \"rootfs\" "
500 +                                       "set to be root filesystem\n");
501 +                               ROOT_DEV = MKDEV(MTD_BLOCK_MAJOR, slave->mtd.index);
502                         }
503 -               } else {
504 -                       /* Single erase size */
505 -                       slave->mtd.erasesize = master->erasesize;
506 -               }
507 -
508 -               if ((slave->mtd.flags & MTD_WRITEABLE) &&
509 -                   (slave->offset % slave->mtd.erasesize)) {
510 -                       /* Doesn't start on a boundary of major erase size */
511 -                       /* FIXME: Let it be writable if it is on a boundary of _minor_ erase size though */
512 -                       slave->mtd.flags &= ~MTD_WRITEABLE;
513 -                       printk ("mtd: partition \"%s\" doesn't start on an erase block boundary -- force read-only\n",
514 -                               parts[i].name);
515 -               }
516 -               if ((slave->mtd.flags & MTD_WRITEABLE) &&
517 -                   (slave->mtd.size % slave->mtd.erasesize)) {
518 -                       slave->mtd.flags &= ~MTD_WRITEABLE;
519 -                       printk ("mtd: partition \"%s\" doesn't end on an erase block -- force read-only\n",
520 -                               parts[i].name);
521 -               }
522 -
523 -               slave->mtd.ecclayout = master->ecclayout;
524 -               if (master->block_isbad) {
525 -                       uint32_t offs = 0;
526 -
527 -                       while(offs < slave->mtd.size) {
528 -                               if (master->block_isbad(master,
529 -                                                       offs + slave->offset))
530 -                                       slave->mtd.ecc_stats.badblocks++;
531 -                               offs += slave->mtd.erasesize;
532 -                       }
533 -               }
534 -
535 -               if(parts[i].mtdp)
536 -               {       /* store the object pointer (caller may or may not register it */
537 -                       *parts[i].mtdp = &slave->mtd;
538 -                       slave->registered = 0;
539 -               }
540 -               else
541 -               {
542 -                       /* register our partition */
543 -                       add_mtd_device(&slave->mtd);
544 -                       slave->registered = 1;
545 +#endif
546 +#ifdef CONFIG_MTD_ROOTFS_SPLIT
547 +                       ret = split_rootfs_data(master, &slave->mtd, part, j);
548 +                       if (ret == 0)
549 +                               j++;
550 +#endif
551                 }
552         }
553  
554 @@ -574,6 +746,32 @@
555         return ret;
556  }
557  
558 +int refresh_mtd_partitions(struct mtd_info *mtd)
559 +{
560 +       int ret = 0;
561 +
562 +       if (IS_PART(mtd)) {
563 +               struct mtd_part *part;
564 +               struct mtd_info *master;
565 +
566 +               part = PART(mtd);
567 +               master = part->master;
568 +               if (master->refresh_device)
569 +                       ret = master->refresh_device(master);
570 +       }
571 +
572 +       if (!ret && mtd->refresh_device)
573 +               ret = mtd->refresh_device(mtd);
574 +
575 +#ifdef CONFIG_MTD_ROOTFS_SPLIT
576 +       if (!ret && IS_PART(mtd) && !strcmp(mtd->name, "rootfs"))
577 +               refresh_rootfs_split(mtd);
578 +#endif
579 +
580 +       return 0;
581 +}
582 +
583  EXPORT_SYMBOL_GPL(parse_mtd_partitions);
584 +EXPORT_SYMBOL_GPL(refresh_mtd_partitions);
585  EXPORT_SYMBOL_GPL(register_mtd_parser);
586  EXPORT_SYMBOL_GPL(deregister_mtd_parser);
587 Index: linux-2.6.23-rc6/drivers/mtd/devices/block2mtd.c
588 ===================================================================
589 --- linux-2.6.23-rc6.orig/drivers/mtd/devices/block2mtd.c       2007-09-21 16:23:59.000000000 +0800
590 +++ linux-2.6.23-rc6/drivers/mtd/devices/block2mtd.c    2007-09-21 16:23:59.000000000 +0800
591 @@ -34,6 +34,8 @@
592         struct block_device *blkdev;
593         struct mtd_info mtd;
594         struct mutex write_mutex;
595 +       rwlock_t bdev_mutex;
596 +       char devname[0];
597  };
598  
599  
600 @@ -86,6 +88,12 @@
601         size_t len = instr->len;
602         int err;
603  
604 +       read_lock(&dev->bdev_mutex);
605 +       if (!dev->blkdev) {
606 +               err = -EINVAL;
607 +               goto done;
608 +       }
609 +
610         instr->state = MTD_ERASING;
611         mutex_lock(&dev->write_mutex);
612         err = _block2mtd_erase(dev, from, len);
613 @@ -98,6 +106,10 @@
614  
615         instr->state = MTD_ERASE_DONE;
616         mtd_erase_callback(instr);
617 +
618 +done:
619 +       read_unlock(&dev->bdev_mutex);
620 +
621         return err;
622  }
623  
624 @@ -109,10 +121,14 @@
625         struct page *page;
626         int index = from >> PAGE_SHIFT;
627         int offset = from & (PAGE_SIZE-1);
628 -       int cpylen;
629 +       int cpylen, err = 0;
630 +
631 +       read_lock(&dev->bdev_mutex);
632 +       if (!dev->blkdev || (from > mtd->size)) {
633 +               err = -EINVAL;
634 +               goto done;
635 +       }
636  
637 -       if (from > mtd->size)
638 -               return -EINVAL;
639         if (from + len > mtd->size)
640                 len = mtd->size - from;
641  
642 @@ -127,10 +143,14 @@
643                 len = len - cpylen;
644  
645                 page = page_read(dev->blkdev->bd_inode->i_mapping, index);
646 -               if (!page)
647 -                       return -ENOMEM;
648 -               if (IS_ERR(page))
649 -                       return PTR_ERR(page);
650 +               if (!page) {
651 +                       err = -ENOMEM;
652 +                       goto done;
653 +               }
654 +               if (IS_ERR(page)) {
655 +                       err = PTR_ERR(page);
656 +                       goto done;
657 +               }
658  
659                 memcpy(buf, page_address(page) + offset, cpylen);
660                 page_cache_release(page);
661 @@ -141,7 +161,10 @@
662                 offset = 0;
663                 index++;
664         }
665 -       return 0;
666 +
667 +done:
668 +       read_unlock(&dev->bdev_mutex);
669 +       return err;
670  }
671  
672  
673 @@ -193,12 +216,22 @@
674                 size_t *retlen, const u_char *buf)
675  {
676         struct block2mtd_dev *dev = mtd->priv;
677 -       int err;
678 +       int err = 0;
679 +
680 +       read_lock(&dev->bdev_mutex);
681 +       if (!dev->blkdev) {
682 +               err = -EINVAL;
683 +               goto done;
684 +       }
685  
686         if (!len)
687 -               return 0;
688 -       if (to >= mtd->size)
689 -               return -ENOSPC;
690 +               goto done;
691 +
692 +       if (to >= mtd->size) {
693 +               err = -ENOSPC;
694 +               goto done;
695 +       }
696 +
697         if (to + len > mtd->size)
698                 len = mtd->size - to;
699  
700 @@ -207,6 +240,9 @@
701         mutex_unlock(&dev->write_mutex);
702         if (err > 0)
703                 err = 0;
704 +
705 +done:
706 +       read_unlock(&dev->bdev_mutex);
707         return err;
708  }
709  
710 @@ -215,51 +251,29 @@
711  static void block2mtd_sync(struct mtd_info *mtd)
712  {
713         struct block2mtd_dev *dev = mtd->priv;
714 -       sync_blockdev(dev->blkdev);
715 -       return;
716 -}
717 -
718 -
719 -static void block2mtd_free_device(struct block2mtd_dev *dev)
720 -{
721 -       if (!dev)
722 -               return;
723 -
724 -       kfree(dev->mtd.name);
725  
726 -       if (dev->blkdev) {
727 -               invalidate_mapping_pages(dev->blkdev->bd_inode->i_mapping,
728 -                                       0, -1);
729 -               close_bdev_excl(dev->blkdev);
730 -       }
731 +       read_lock(&dev->bdev_mutex);
732 +       if (dev->blkdev)
733 +               sync_blockdev(dev->blkdev);
734 +       read_unlock(&dev->bdev_mutex);
735  
736 -       kfree(dev);
737 +       return;
738  }
739  
740  
741 -/* FIXME: ensure that mtd->size % erase_size == 0 */
742 -static struct block2mtd_dev *add_device(char *devname, int erase_size, char *mtdname)
743 +static int _open_bdev(struct block2mtd_dev *dev)
744  {
745         struct block_device *bdev;
746 -       struct block2mtd_dev *dev;
747 -       struct mtd_partition *part;
748 -
749 -       if (!devname)
750 -               return NULL;
751 -
752 -       dev = kzalloc(sizeof(struct block2mtd_dev), GFP_KERNEL);
753 -       if (!dev)
754 -               return NULL;
755  
756         /* Get a handle on the device */
757 -       bdev = open_bdev_excl(devname, O_RDWR, NULL);
758 +       bdev = open_bdev_excl(dev->devname, O_RDWR, NULL);
759  #ifndef MODULE
760         if (IS_ERR(bdev)) {
761  
762                 /* We might not have rootfs mounted at this point. Try
763                    to resolve the device name by other means. */
764  
765 -               dev_t devt = name_to_dev_t(devname);
766 +               dev_t devt = name_to_dev_t(dev->devname);
767                 if (devt) {
768                         bdev = open_by_devnum(devt, FMODE_WRITE | FMODE_READ);
769                 }
770 @@ -267,17 +281,96 @@
771  #endif
772  
773         if (IS_ERR(bdev)) {
774 -               ERROR("error: cannot open device %s", devname);
775 -               goto devinit_err;
776 +               ERROR("error: cannot open device %s", dev->devname);
777 +               return 1;
778         }
779         dev->blkdev = bdev;
780  
781         if (MAJOR(bdev->bd_dev) == MTD_BLOCK_MAJOR) {
782                 ERROR("attempting to use an MTD device as a block device");
783 -               goto devinit_err;
784 +               return 1;
785         }
786  
787 +       return 0;
788 +}
789 +
790 +static void _close_bdev(struct block2mtd_dev *dev)
791 +{
792 +       struct block_device *bdev;
793 +
794 +       if (!dev->blkdev)
795 +               return;
796 +
797 +       bdev = dev->blkdev;
798 +       invalidate_mapping_pages(dev->blkdev->bd_inode->i_mapping, 0, -1);
799 +       close_bdev_excl(dev->blkdev);
800 +       dev->blkdev = NULL;
801 +}
802 +
803 +static void block2mtd_free_device(struct block2mtd_dev *dev)
804 +{
805 +       if (!dev)
806 +               return;
807 +
808 +       kfree(dev->mtd.name);
809 +       _close_bdev(dev);
810 +       kfree(dev);
811 +}
812 +
813 +
814 +static int block2mtd_refresh(struct mtd_info *mtd)
815 +{
816 +       struct block2mtd_dev *dev = mtd->priv;
817 +       struct block_device *bdev;
818 +       dev_t devt;
819 +       int err = 0;
820 +
821 +       /* no other mtd function can run at this point */
822 +       write_lock(&dev->bdev_mutex);
823 +
824 +       /* get the device number for the whole disk */
825 +       devt = MKDEV(MAJOR(dev->blkdev->bd_dev), 0);
826 +
827 +       /* close the old block device */
828 +       _close_bdev(dev);
829 +
830 +       /* open the whole disk, issue a partition rescan, then */
831 +       bdev = open_by_devnum(devt, FMODE_WRITE | FMODE_READ);
832 +       if (!bdev || !bdev->bd_disk)
833 +               err = -EINVAL;
834 +       else {
835 +               err = rescan_partitions(bdev->bd_disk, bdev);
836 +       }
837 +       if (bdev)
838 +               close_bdev_excl(bdev);
839 +
840 +       /* try to open the partition block device again */
841 +       _open_bdev(dev);
842 +       write_unlock(&dev->bdev_mutex);
843 +
844 +       return err;
845 +}
846 +
847 +/* FIXME: ensure that mtd->size % erase_size == 0 */
848 +static struct block2mtd_dev *add_device(char *devname, int erase_size, char *mtdname)
849 +{
850 +       struct block2mtd_dev *dev;
851 +       struct mtd_partition *part;
852 +
853 +       if (!devname)
854 +               return NULL;
855 +
856 +       dev = kzalloc(sizeof(struct block2mtd_dev) + strlen(devname) + 1, GFP_KERNEL);
857 +       if (!dev)
858 +               return NULL;
859 +
860 +       strcpy(dev->devname, devname);
861 +
862 +       if (_open_bdev(dev))
863 +               goto devinit_err;
864 +
865         mutex_init(&dev->write_mutex);
866 +       rwlock_init(&dev->bdev_mutex);
867  
868         /* Setup the MTD structure */
869         /* make the name contain the block device in */
870 @@ -304,6 +397,7 @@
871         dev->mtd.read = block2mtd_read;
872         dev->mtd.priv = dev;
873         dev->mtd.owner = THIS_MODULE;
874 +       dev->mtd.refresh_device = block2mtd_refresh;
875         
876         part = kzalloc(sizeof(struct mtd_partition), GFP_KERNEL);
877         part->name = dev->mtd.name;
878 Index: linux-2.6.23-rc6/drivers/mtd/mtdchar.c
879 ===================================================================
880 --- linux-2.6.23-rc6.orig/drivers/mtd/mtdchar.c 2007-09-21 16:23:54.000000000 +0800
881 +++ linux-2.6.23-rc6/drivers/mtd/mtdchar.c      2007-09-21 16:23:59.000000000 +0800
882 @@ -17,6 +17,7 @@
883  
884  #include <linux/mtd/mtd.h>
885  #include <linux/mtd/compatmac.h>
886 +#include <linux/mtd/partitions.h>
887  
888  #include <asm/uaccess.h>
889  
890 @@ -753,6 +754,13 @@
891                 file->f_pos = 0;
892                 break;
893         }
894 +#ifdef CONFIG_MTD_PARTITIONS
895 +       case MTDREFRESH:
896 +       {
897 +               ret = refresh_mtd_partitions(mtd);
898 +               break;
899 +       }
900 +#endif
901  
902         default:
903                 ret = -ENOTTY;
904 Index: linux-2.6.23-rc6/include/linux/mtd/mtd.h
905 ===================================================================
906 --- linux-2.6.23-rc6.orig/include/linux/mtd/mtd.h       2007-09-21 16:23:54.000000000 +0800
907 +++ linux-2.6.23-rc6/include/linux/mtd/mtd.h    2007-09-21 16:23:59.000000000 +0800
908 @@ -98,6 +98,7 @@
909         uint8_t         *oobbuf;
910  };
911  
912 +struct mtd_info;
913  struct mtd_info {
914         u_char type;
915         u_int32_t flags;
916 @@ -195,6 +196,9 @@
917         struct module *owner;
918         int usecount;
919  
920 +       int (*refresh_device)(struct mtd_info *mtd);
921 +       struct mtd_info *split;
922 +
923         /* If the driver is something smart, like UBI, it may need to maintain
924          * its own reference counting. The below functions are only for driver.
925          * The driver may register its callbacks. These callbacks are not
926 Index: linux-2.6.23-rc6/include/linux/mtd/partitions.h
927 ===================================================================
928 --- linux-2.6.23-rc6.orig/include/linux/mtd/partitions.h        2007-09-21 16:23:54.000000000 +0800
929 +++ linux-2.6.23-rc6/include/linux/mtd/partitions.h     2007-09-21 16:23:59.000000000 +0800
930 @@ -36,6 +36,7 @@
931   * erasesize aligned (e.g. use MTDPART_OFS_NEXTBLK).
932   */
933  
934 +struct mtd_partition;
935  struct mtd_partition {
936         char *name;                     /* identifier string */
937         u_int32_t size;                 /* partition size */
938 @@ -43,6 +44,7 @@
939         u_int32_t mask_flags;           /* master MTD flags to mask out for this partition */
940         struct nand_ecclayout *ecclayout;       /* out of band layout for this partition (NAND only)*/
941         struct mtd_info **mtdp;         /* pointer to store the MTD object */
942 +       int (*refresh_partition)(struct mtd_info *);
943  };
944  
945  #define MTDPART_OFS_NXTBLK     (-2)
946 @@ -52,6 +54,7 @@
947  
948  int add_mtd_partitions(struct mtd_info *, const struct mtd_partition *, int);
949  int del_mtd_partitions(struct mtd_info *);
950 +int refresh_mtd_partitions(struct mtd_info *);
951  
952  /*
953   * Functions dealing with the various ways of partitioning the space
954 Index: linux-2.6.23-rc6/include/mtd/mtd-abi.h
955 ===================================================================
956 --- linux-2.6.23-rc6.orig/include/mtd/mtd-abi.h 2007-09-21 16:23:54.000000000 +0800
957 +++ linux-2.6.23-rc6/include/mtd/mtd-abi.h      2007-09-21 16:23:59.000000000 +0800
958 @@ -95,6 +95,7 @@
959  #define ECCGETLAYOUT           _IOR('M', 17, struct nand_ecclayout)
960  #define ECCGETSTATS            _IOR('M', 18, struct mtd_ecc_stats)
961  #define MTDFILEMODE            _IO('M', 19)
962 +#define MTDREFRESH             _IO('M', 23)
963  
964  /*
965   * Obsolete legacy interface. Keep it in order not to break userspace