xref: /linux/drivers/block/aoe/aoedev.c (revision e58e871becec2d3b04ed91c0c16fe8deac9c9dfa)
1 /* Copyright (c) 2013 Coraid, Inc.  See COPYING for GPL terms. */
2 /*
3  * aoedev.c
4  * AoE device utility functions; maintains device list.
5  */
6 
7 #include <linux/hdreg.h>
8 #include <linux/blkdev.h>
9 #include <linux/netdevice.h>
10 #include <linux/delay.h>
11 #include <linux/slab.h>
12 #include <linux/bitmap.h>
13 #include <linux/kdev_t.h>
14 #include <linux/moduleparam.h>
15 #include <linux/string.h>
16 #include "aoe.h"
17 
18 static void dummy_timer(ulong);
19 static void freetgt(struct aoedev *d, struct aoetgt *t);
20 static void skbpoolfree(struct aoedev *d);
21 
22 static int aoe_dyndevs = 1;
23 module_param(aoe_dyndevs, int, 0644);
24 MODULE_PARM_DESC(aoe_dyndevs, "Use dynamic minor numbers for devices.");
25 
26 static struct aoedev *devlist;
27 static DEFINE_SPINLOCK(devlist_lock);
28 
29 /* Because some systems will have one, many, or no
30  *   - partitions,
31  *   - slots per shelf,
32  *   - or shelves,
33  * we need some flexibility in the way the minor numbers
34  * are allocated.  So they are dynamic.
35  */
36 #define N_DEVS ((1U<<MINORBITS)/AOE_PARTITIONS)
37 
38 static DEFINE_SPINLOCK(used_minors_lock);
39 static DECLARE_BITMAP(used_minors, N_DEVS);
40 
41 static int
42 minor_get_dyn(ulong *sysminor)
43 {
44 	ulong flags;
45 	ulong n;
46 	int error = 0;
47 
48 	spin_lock_irqsave(&used_minors_lock, flags);
49 	n = find_first_zero_bit(used_minors, N_DEVS);
50 	if (n < N_DEVS)
51 		set_bit(n, used_minors);
52 	else
53 		error = -1;
54 	spin_unlock_irqrestore(&used_minors_lock, flags);
55 
56 	*sysminor = n * AOE_PARTITIONS;
57 	return error;
58 }
59 
60 static int
61 minor_get_static(ulong *sysminor, ulong aoemaj, int aoemin)
62 {
63 	ulong flags;
64 	ulong n;
65 	int error = 0;
66 	enum {
67 		/* for backwards compatibility when !aoe_dyndevs,
68 		 * a static number of supported slots per shelf */
69 		NPERSHELF = 16,
70 	};
71 
72 	if (aoemin >= NPERSHELF) {
73 		pr_err("aoe: %s %d slots per shelf\n",
74 			"static minor device numbers support only",
75 			NPERSHELF);
76 		error = -1;
77 		goto out;
78 	}
79 
80 	n = aoemaj * NPERSHELF + aoemin;
81 	if (n >= N_DEVS) {
82 		pr_err("aoe: %s with e%ld.%d\n",
83 			"cannot use static minor device numbers",
84 			aoemaj, aoemin);
85 		error = -1;
86 		goto out;
87 	}
88 
89 	spin_lock_irqsave(&used_minors_lock, flags);
90 	if (test_bit(n, used_minors)) {
91 		pr_err("aoe: %s %lu\n",
92 			"existing device already has static minor number",
93 			n);
94 		error = -1;
95 	} else
96 		set_bit(n, used_minors);
97 	spin_unlock_irqrestore(&used_minors_lock, flags);
98 	*sysminor = n * AOE_PARTITIONS;
99 out:
100 	return error;
101 }
102 
103 static int
104 minor_get(ulong *sysminor, ulong aoemaj, int aoemin)
105 {
106 	if (aoe_dyndevs)
107 		return minor_get_dyn(sysminor);
108 	else
109 		return minor_get_static(sysminor, aoemaj, aoemin);
110 }
111 
112 static void
113 minor_free(ulong minor)
114 {
115 	ulong flags;
116 
117 	minor /= AOE_PARTITIONS;
118 	BUG_ON(minor >= N_DEVS);
119 
120 	spin_lock_irqsave(&used_minors_lock, flags);
121 	BUG_ON(!test_bit(minor, used_minors));
122 	clear_bit(minor, used_minors);
123 	spin_unlock_irqrestore(&used_minors_lock, flags);
124 }
125 
126 /*
127  * Users who grab a pointer to the device with aoedev_by_aoeaddr
128  * automatically get a reference count and must be responsible
129  * for performing a aoedev_put.  With the addition of async
130  * kthread processing I'm no longer confident that we can
131  * guarantee consistency in the face of device flushes.
132  *
133  * For the time being, we only bother to add extra references for
134  * frames sitting on the iocq.  When the kthreads finish processing
135  * these frames, they will aoedev_put the device.
136  */
137 
138 void
139 aoedev_put(struct aoedev *d)
140 {
141 	ulong flags;
142 
143 	spin_lock_irqsave(&devlist_lock, flags);
144 	d->ref--;
145 	spin_unlock_irqrestore(&devlist_lock, flags);
146 }
147 
148 static void
149 dummy_timer(ulong vp)
150 {
151 	struct aoedev *d;
152 
153 	d = (struct aoedev *)vp;
154 	if (d->flags & DEVFL_TKILL)
155 		return;
156 	d->timer.expires = jiffies + HZ;
157 	add_timer(&d->timer);
158 }
159 
160 static void
161 aoe_failip(struct aoedev *d)
162 {
163 	struct request *rq;
164 	struct bio *bio;
165 	unsigned long n;
166 
167 	aoe_failbuf(d, d->ip.buf);
168 
169 	rq = d->ip.rq;
170 	if (rq == NULL)
171 		return;
172 	while ((bio = d->ip.nxbio)) {
173 		bio->bi_error = -EIO;
174 		d->ip.nxbio = bio->bi_next;
175 		n = (unsigned long) rq->special;
176 		rq->special = (void *) --n;
177 	}
178 	if ((unsigned long) rq->special == 0)
179 		aoe_end_request(d, rq, 0);
180 }
181 
182 static void
183 downdev_frame(struct list_head *pos)
184 {
185 	struct frame *f;
186 
187 	f = list_entry(pos, struct frame, head);
188 	list_del(pos);
189 	if (f->buf) {
190 		f->buf->nframesout--;
191 		aoe_failbuf(f->t->d, f->buf);
192 	}
193 	aoe_freetframe(f);
194 }
195 
196 void
197 aoedev_downdev(struct aoedev *d)
198 {
199 	struct aoetgt *t, **tt, **te;
200 	struct list_head *head, *pos, *nx;
201 	struct request *rq;
202 	int i;
203 
204 	d->flags &= ~DEVFL_UP;
205 
206 	/* clean out active and to-be-retransmitted buffers */
207 	for (i = 0; i < NFACTIVE; i++) {
208 		head = &d->factive[i];
209 		list_for_each_safe(pos, nx, head)
210 			downdev_frame(pos);
211 	}
212 	head = &d->rexmitq;
213 	list_for_each_safe(pos, nx, head)
214 		downdev_frame(pos);
215 
216 	/* reset window dressings */
217 	tt = d->targets;
218 	te = tt + d->ntargets;
219 	for (; tt < te && (t = *tt); tt++) {
220 		aoecmd_wreset(t);
221 		t->nout = 0;
222 	}
223 
224 	/* clean out the in-process request (if any) */
225 	aoe_failip(d);
226 
227 	/* fast fail all pending I/O */
228 	if (d->blkq) {
229 		while ((rq = blk_peek_request(d->blkq))) {
230 			blk_start_request(rq);
231 			aoe_end_request(d, rq, 1);
232 		}
233 	}
234 
235 	if (d->gd)
236 		set_capacity(d->gd, 0);
237 }
238 
239 /* return whether the user asked for this particular
240  * device to be flushed
241  */
242 static int
243 user_req(char *s, size_t slen, struct aoedev *d)
244 {
245 	const char *p;
246 	size_t lim;
247 
248 	if (!d->gd)
249 		return 0;
250 	p = kbasename(d->gd->disk_name);
251 	lim = sizeof(d->gd->disk_name);
252 	lim -= p - d->gd->disk_name;
253 	if (slen < lim)
254 		lim = slen;
255 
256 	return !strncmp(s, p, lim);
257 }
258 
259 static void
260 freedev(struct aoedev *d)
261 {
262 	struct aoetgt **t, **e;
263 	int freeing = 0;
264 	unsigned long flags;
265 
266 	spin_lock_irqsave(&d->lock, flags);
267 	if (d->flags & DEVFL_TKILL
268 	&& !(d->flags & DEVFL_FREEING)) {
269 		d->flags |= DEVFL_FREEING;
270 		freeing = 1;
271 	}
272 	spin_unlock_irqrestore(&d->lock, flags);
273 	if (!freeing)
274 		return;
275 
276 	del_timer_sync(&d->timer);
277 	if (d->gd) {
278 		aoedisk_rm_debugfs(d);
279 		aoedisk_rm_sysfs(d);
280 		del_gendisk(d->gd);
281 		put_disk(d->gd);
282 		blk_cleanup_queue(d->blkq);
283 	}
284 	t = d->targets;
285 	e = t + d->ntargets;
286 	for (; t < e && *t; t++)
287 		freetgt(d, *t);
288 	if (d->bufpool)
289 		mempool_destroy(d->bufpool);
290 	skbpoolfree(d);
291 	minor_free(d->sysminor);
292 
293 	spin_lock_irqsave(&d->lock, flags);
294 	d->flags |= DEVFL_FREED;
295 	spin_unlock_irqrestore(&d->lock, flags);
296 }
297 
298 enum flush_parms {
299 	NOT_EXITING = 0,
300 	EXITING = 1,
301 };
302 
303 static int
304 flush(const char __user *str, size_t cnt, int exiting)
305 {
306 	ulong flags;
307 	struct aoedev *d, **dd;
308 	char buf[16];
309 	int all = 0;
310 	int specified = 0;	/* flush a specific device */
311 	unsigned int skipflags;
312 
313 	skipflags = DEVFL_GDALLOC | DEVFL_NEWSIZE | DEVFL_TKILL;
314 
315 	if (!exiting && cnt >= 3) {
316 		if (cnt > sizeof buf)
317 			cnt = sizeof buf;
318 		if (copy_from_user(buf, str, cnt))
319 			return -EFAULT;
320 		all = !strncmp(buf, "all", 3);
321 		if (!all)
322 			specified = 1;
323 	}
324 
325 	flush_scheduled_work();
326 	/* pass one: without sleeping, do aoedev_downdev */
327 	spin_lock_irqsave(&devlist_lock, flags);
328 	for (d = devlist; d; d = d->next) {
329 		spin_lock(&d->lock);
330 		if (exiting) {
331 			/* unconditionally take each device down */
332 		} else if (specified) {
333 			if (!user_req(buf, cnt, d))
334 				goto cont;
335 		} else if ((!all && (d->flags & DEVFL_UP))
336 		|| d->flags & skipflags
337 		|| d->nopen
338 		|| d->ref)
339 			goto cont;
340 
341 		aoedev_downdev(d);
342 		d->flags |= DEVFL_TKILL;
343 cont:
344 		spin_unlock(&d->lock);
345 	}
346 	spin_unlock_irqrestore(&devlist_lock, flags);
347 
348 	/* pass two: call freedev, which might sleep,
349 	 * for aoedevs marked with DEVFL_TKILL
350 	 */
351 restart:
352 	spin_lock_irqsave(&devlist_lock, flags);
353 	for (d = devlist; d; d = d->next) {
354 		spin_lock(&d->lock);
355 		if (d->flags & DEVFL_TKILL
356 		&& !(d->flags & DEVFL_FREEING)) {
357 			spin_unlock(&d->lock);
358 			spin_unlock_irqrestore(&devlist_lock, flags);
359 			freedev(d);
360 			goto restart;
361 		}
362 		spin_unlock(&d->lock);
363 	}
364 
365 	/* pass three: remove aoedevs marked with DEVFL_FREED */
366 	for (dd = &devlist, d = *dd; d; d = *dd) {
367 		struct aoedev *doomed = NULL;
368 
369 		spin_lock(&d->lock);
370 		if (d->flags & DEVFL_FREED) {
371 			*dd = d->next;
372 			doomed = d;
373 		} else {
374 			dd = &d->next;
375 		}
376 		spin_unlock(&d->lock);
377 		if (doomed)
378 			kfree(doomed->targets);
379 		kfree(doomed);
380 	}
381 	spin_unlock_irqrestore(&devlist_lock, flags);
382 
383 	return 0;
384 }
385 
386 int
387 aoedev_flush(const char __user *str, size_t cnt)
388 {
389 	return flush(str, cnt, NOT_EXITING);
390 }
391 
392 /* This has been confirmed to occur once with Tms=3*1000 due to the
393  * driver changing link and not processing its transmit ring.  The
394  * problem is hard enough to solve by returning an error that I'm
395  * still punting on "solving" this.
396  */
397 static void
398 skbfree(struct sk_buff *skb)
399 {
400 	enum { Sms = 250, Tms = 30 * 1000};
401 	int i = Tms / Sms;
402 
403 	if (skb == NULL)
404 		return;
405 	while (atomic_read(&skb_shinfo(skb)->dataref) != 1 && i-- > 0)
406 		msleep(Sms);
407 	if (i < 0) {
408 		printk(KERN_ERR
409 			"aoe: %s holds ref: %s\n",
410 			skb->dev ? skb->dev->name : "netif",
411 			"cannot free skb -- memory leaked.");
412 		return;
413 	}
414 	skb->truesize -= skb->data_len;
415 	skb_shinfo(skb)->nr_frags = skb->data_len = 0;
416 	skb_trim(skb, 0);
417 	dev_kfree_skb(skb);
418 }
419 
420 static void
421 skbpoolfree(struct aoedev *d)
422 {
423 	struct sk_buff *skb, *tmp;
424 
425 	skb_queue_walk_safe(&d->skbpool, skb, tmp)
426 		skbfree(skb);
427 
428 	__skb_queue_head_init(&d->skbpool);
429 }
430 
431 /* find it or allocate it */
432 struct aoedev *
433 aoedev_by_aoeaddr(ulong maj, int min, int do_alloc)
434 {
435 	struct aoedev *d;
436 	int i;
437 	ulong flags;
438 	ulong sysminor = 0;
439 
440 	spin_lock_irqsave(&devlist_lock, flags);
441 
442 	for (d=devlist; d; d=d->next)
443 		if (d->aoemajor == maj && d->aoeminor == min) {
444 			spin_lock(&d->lock);
445 			if (d->flags & DEVFL_TKILL) {
446 				spin_unlock(&d->lock);
447 				d = NULL;
448 				goto out;
449 			}
450 			d->ref++;
451 			spin_unlock(&d->lock);
452 			break;
453 		}
454 	if (d || !do_alloc || minor_get(&sysminor, maj, min) < 0)
455 		goto out;
456 	d = kcalloc(1, sizeof *d, GFP_ATOMIC);
457 	if (!d)
458 		goto out;
459 	d->targets = kcalloc(NTARGETS, sizeof(*d->targets), GFP_ATOMIC);
460 	if (!d->targets) {
461 		kfree(d);
462 		d = NULL;
463 		goto out;
464 	}
465 	d->ntargets = NTARGETS;
466 	INIT_WORK(&d->work, aoecmd_sleepwork);
467 	spin_lock_init(&d->lock);
468 	skb_queue_head_init(&d->skbpool);
469 	init_timer(&d->timer);
470 	d->timer.data = (ulong) d;
471 	d->timer.function = dummy_timer;
472 	d->timer.expires = jiffies + HZ;
473 	add_timer(&d->timer);
474 	d->bufpool = NULL;	/* defer to aoeblk_gdalloc */
475 	d->tgt = d->targets;
476 	d->ref = 1;
477 	for (i = 0; i < NFACTIVE; i++)
478 		INIT_LIST_HEAD(&d->factive[i]);
479 	INIT_LIST_HEAD(&d->rexmitq);
480 	d->sysminor = sysminor;
481 	d->aoemajor = maj;
482 	d->aoeminor = min;
483 	d->rttavg = RTTAVG_INIT;
484 	d->rttdev = RTTDEV_INIT;
485 	d->next = devlist;
486 	devlist = d;
487  out:
488 	spin_unlock_irqrestore(&devlist_lock, flags);
489 	return d;
490 }
491 
492 static void
493 freetgt(struct aoedev *d, struct aoetgt *t)
494 {
495 	struct frame *f;
496 	struct list_head *pos, *nx, *head;
497 	struct aoeif *ifp;
498 
499 	for (ifp = t->ifs; ifp < &t->ifs[NAOEIFS]; ++ifp) {
500 		if (!ifp->nd)
501 			break;
502 		dev_put(ifp->nd);
503 	}
504 
505 	head = &t->ffree;
506 	list_for_each_safe(pos, nx, head) {
507 		list_del(pos);
508 		f = list_entry(pos, struct frame, head);
509 		skbfree(f->skb);
510 		kfree(f);
511 	}
512 	kfree(t);
513 }
514 
515 void
516 aoedev_exit(void)
517 {
518 	flush_scheduled_work();
519 	flush(NULL, 0, EXITING);
520 }
521 
522 int __init
523 aoedev_init(void)
524 {
525 	return 0;
526 }
527