[PATCH] md: fix BUG when raid10 rebuilds without enough drives
[linux-2.6] / drivers / md / dm-stripe.c
1 /*
2  * Copyright (C) 2001-2003 Sistina Software (UK) Limited.
3  *
4  * This file is released under the GPL.
5  */
6
7 #include "dm.h"
8
9 #include <linux/module.h>
10 #include <linux/init.h>
11 #include <linux/blkdev.h>
12 #include <linux/bio.h>
13 #include <linux/slab.h>
14
15 struct stripe {
16         struct dm_dev *dev;
17         sector_t physical_start;
18 };
19
20 struct stripe_c {
21         uint32_t stripes;
22
23         /* The size of this target / num. stripes */
24         sector_t stripe_width;
25
26         /* stripe chunk size */
27         uint32_t chunk_shift;
28         sector_t chunk_mask;
29
30         struct stripe stripe[0];
31 };
32
33 static inline struct stripe_c *alloc_context(unsigned int stripes)
34 {
35         size_t len;
36
37         if (array_too_big(sizeof(struct stripe_c), sizeof(struct stripe),
38                           stripes))
39                 return NULL;
40
41         len = sizeof(struct stripe_c) + (sizeof(struct stripe) * stripes);
42
43         return kmalloc(len, GFP_KERNEL);
44 }
45
46 /*
47  * Parse a single <dev> <sector> pair
48  */
49 static int get_stripe(struct dm_target *ti, struct stripe_c *sc,
50                       unsigned int stripe, char **argv)
51 {
52         sector_t start;
53
54         if (sscanf(argv[1], SECTOR_FORMAT, &start) != 1)
55                 return -EINVAL;
56
57         if (dm_get_device(ti, argv[0], start, sc->stripe_width,
58                           dm_table_get_mode(ti->table),
59                           &sc->stripe[stripe].dev))
60                 return -ENXIO;
61
62         sc->stripe[stripe].physical_start = start;
63         return 0;
64 }
65
66 /*
67  * Construct a striped mapping.
68  * <number of stripes> <chunk size (2^^n)> [<dev_path> <offset>]+
69  */
70 static int stripe_ctr(struct dm_target *ti, unsigned int argc, char **argv)
71 {
72         struct stripe_c *sc;
73         sector_t width;
74         uint32_t stripes;
75         uint32_t chunk_size;
76         char *end;
77         int r;
78         unsigned int i;
79
80         if (argc < 2) {
81                 ti->error = "dm-stripe: Not enough arguments";
82                 return -EINVAL;
83         }
84
85         stripes = simple_strtoul(argv[0], &end, 10);
86         if (*end) {
87                 ti->error = "dm-stripe: Invalid stripe count";
88                 return -EINVAL;
89         }
90
91         chunk_size = simple_strtoul(argv[1], &end, 10);
92         if (*end) {
93                 ti->error = "dm-stripe: Invalid chunk_size";
94                 return -EINVAL;
95         }
96
97         /*
98          * chunk_size is a power of two
99          */
100         if (!chunk_size || (chunk_size & (chunk_size - 1)) ||
101             (chunk_size < (PAGE_SIZE >> SECTOR_SHIFT))) {
102                 ti->error = "dm-stripe: Invalid chunk size";
103                 return -EINVAL;
104         }
105
106         width = ti->len;
107         if (sector_div(width, stripes)) {
108                 ti->error = "dm-stripe: Target length not divisable by "
109                     "number of stripes";
110                 return -EINVAL;
111         }
112
113         /*
114          * Do we have enough arguments for that many stripes ?
115          */
116         if (argc != (2 + 2 * stripes)) {
117                 ti->error = "dm-stripe: Not enough destinations "
118                         "specified";
119                 return -EINVAL;
120         }
121
122         sc = alloc_context(stripes);
123         if (!sc) {
124                 ti->error = "dm-stripe: Memory allocation for striped context "
125                     "failed";
126                 return -ENOMEM;
127         }
128
129         sc->stripes = stripes;
130         sc->stripe_width = width;
131         ti->split_io = chunk_size;
132
133         sc->chunk_mask = ((sector_t) chunk_size) - 1;
134         for (sc->chunk_shift = 0; chunk_size; sc->chunk_shift++)
135                 chunk_size >>= 1;
136         sc->chunk_shift--;
137
138         /*
139          * Get the stripe destinations.
140          */
141         for (i = 0; i < stripes; i++) {
142                 argv += 2;
143
144                 r = get_stripe(ti, sc, i, argv);
145                 if (r < 0) {
146                         ti->error = "dm-stripe: Couldn't parse stripe "
147                                 "destination";
148                         while (i--)
149                                 dm_put_device(ti, sc->stripe[i].dev);
150                         kfree(sc);
151                         return r;
152                 }
153         }
154
155         ti->private = sc;
156         return 0;
157 }
158
159 static void stripe_dtr(struct dm_target *ti)
160 {
161         unsigned int i;
162         struct stripe_c *sc = (struct stripe_c *) ti->private;
163
164         for (i = 0; i < sc->stripes; i++)
165                 dm_put_device(ti, sc->stripe[i].dev);
166
167         kfree(sc);
168 }
169
170 static int stripe_map(struct dm_target *ti, struct bio *bio,
171                       union map_info *map_context)
172 {
173         struct stripe_c *sc = (struct stripe_c *) ti->private;
174
175         sector_t offset = bio->bi_sector - ti->begin;
176         sector_t chunk = offset >> sc->chunk_shift;
177         uint32_t stripe = sector_div(chunk, sc->stripes);
178
179         bio->bi_bdev = sc->stripe[stripe].dev->bdev;
180         bio->bi_sector = sc->stripe[stripe].physical_start +
181             (chunk << sc->chunk_shift) + (offset & sc->chunk_mask);
182         return 1;
183 }
184
185 static int stripe_status(struct dm_target *ti,
186                          status_type_t type, char *result, unsigned int maxlen)
187 {
188         struct stripe_c *sc = (struct stripe_c *) ti->private;
189         unsigned int sz = 0;
190         unsigned int i;
191
192         switch (type) {
193         case STATUSTYPE_INFO:
194                 result[0] = '\0';
195                 break;
196
197         case STATUSTYPE_TABLE:
198                 DMEMIT("%d " SECTOR_FORMAT, sc->stripes, sc->chunk_mask + 1);
199                 for (i = 0; i < sc->stripes; i++)
200                         DMEMIT(" %s " SECTOR_FORMAT, sc->stripe[i].dev->name,
201                                sc->stripe[i].physical_start);
202                 break;
203         }
204         return 0;
205 }
206
207 static struct target_type stripe_target = {
208         .name   = "striped",
209         .version= {1, 0, 2},
210         .module = THIS_MODULE,
211         .ctr    = stripe_ctr,
212         .dtr    = stripe_dtr,
213         .map    = stripe_map,
214         .status = stripe_status,
215 };
216
217 int __init dm_stripe_init(void)
218 {
219         int r;
220
221         r = dm_register_target(&stripe_target);
222         if (r < 0)
223                 DMWARN("striped target registration failed");
224
225         return r;
226 }
227
228 void dm_stripe_exit(void)
229 {
230         if (dm_unregister_target(&stripe_target))
231                 DMWARN("striped target unregistration failed");
232
233         return;
234 }