PM / sleep: Asynchronous threads for suspend_noirq
[linux/fpc-iii.git] / drivers / block / aoe / aoecmd.c
blob8184451b57c04999bd23c286080e14ca7f069031
1 /* Copyright (c) 2013 Coraid, Inc. See COPYING for GPL terms. */
2 /*
3 * aoecmd.c
4 * Filesystem request handling methods
5 */
7 #include <linux/ata.h>
8 #include <linux/slab.h>
9 #include <linux/hdreg.h>
10 #include <linux/blkdev.h>
11 #include <linux/skbuff.h>
12 #include <linux/netdevice.h>
13 #include <linux/genhd.h>
14 #include <linux/moduleparam.h>
15 #include <linux/workqueue.h>
16 #include <linux/kthread.h>
17 #include <net/net_namespace.h>
18 #include <asm/unaligned.h>
19 #include <linux/uio.h>
20 #include "aoe.h"
22 #define MAXIOC (8192) /* default meant to avoid most soft lockups */
24 static void ktcomplete(struct frame *, struct sk_buff *);
25 static int count_targets(struct aoedev *d, int *untainted);
27 static struct buf *nextbuf(struct aoedev *);
29 static int aoe_deadsecs = 60 * 3;
30 module_param(aoe_deadsecs, int, 0644);
31 MODULE_PARM_DESC(aoe_deadsecs, "After aoe_deadsecs seconds, give up and fail dev.");
33 static int aoe_maxout = 64;
34 module_param(aoe_maxout, int, 0644);
35 MODULE_PARM_DESC(aoe_maxout,
36 "Only aoe_maxout outstanding packets for every MAC on eX.Y.");
38 /* The number of online cpus during module initialization gives us a
39 * convenient heuristic cap on the parallelism used for ktio threads
40 * doing I/O completion. It is not important that the cap equal the
41 * actual number of running CPUs at any given time, but because of CPU
42 * hotplug, we take care to use ncpus instead of using
43 * num_online_cpus() after module initialization.
45 static int ncpus;
47 /* mutex lock used for synchronization while thread spawning */
48 static DEFINE_MUTEX(ktio_spawn_lock);
50 static wait_queue_head_t *ktiowq;
51 static struct ktstate *kts;
53 /* io completion queue */
54 struct iocq_ktio {
55 struct list_head head;
56 spinlock_t lock;
58 static struct iocq_ktio *iocq;
60 static struct page *empty_page;
62 static struct sk_buff *
63 new_skb(ulong len)
65 struct sk_buff *skb;
67 skb = alloc_skb(len + MAX_HEADER, GFP_ATOMIC);
68 if (skb) {
69 skb_reserve(skb, MAX_HEADER);
70 skb_reset_mac_header(skb);
71 skb_reset_network_header(skb);
72 skb->protocol = __constant_htons(ETH_P_AOE);
73 skb_checksum_none_assert(skb);
75 return skb;
78 static struct frame *
79 getframe_deferred(struct aoedev *d, u32 tag)
81 struct list_head *head, *pos, *nx;
82 struct frame *f;
84 head = &d->rexmitq;
85 list_for_each_safe(pos, nx, head) {
86 f = list_entry(pos, struct frame, head);
87 if (f->tag == tag) {
88 list_del(pos);
89 return f;
92 return NULL;
95 static struct frame *
96 getframe(struct aoedev *d, u32 tag)
98 struct frame *f;
99 struct list_head *head, *pos, *nx;
100 u32 n;
102 n = tag % NFACTIVE;
103 head = &d->factive[n];
104 list_for_each_safe(pos, nx, head) {
105 f = list_entry(pos, struct frame, head);
106 if (f->tag == tag) {
107 list_del(pos);
108 return f;
111 return NULL;
115 * Leave the top bit clear so we have tagspace for userland.
116 * The bottom 16 bits are the xmit tick for rexmit/rttavg processing.
117 * This driver reserves tag -1 to mean "unused frame."
119 static int
120 newtag(struct aoedev *d)
122 register ulong n;
124 n = jiffies & 0xffff;
125 return n |= (++d->lasttag & 0x7fff) << 16;
128 static u32
129 aoehdr_atainit(struct aoedev *d, struct aoetgt *t, struct aoe_hdr *h)
131 u32 host_tag = newtag(d);
133 memcpy(h->src, t->ifp->nd->dev_addr, sizeof h->src);
134 memcpy(h->dst, t->addr, sizeof h->dst);
135 h->type = __constant_cpu_to_be16(ETH_P_AOE);
136 h->verfl = AOE_HVER;
137 h->major = cpu_to_be16(d->aoemajor);
138 h->minor = d->aoeminor;
139 h->cmd = AOECMD_ATA;
140 h->tag = cpu_to_be32(host_tag);
142 return host_tag;
145 static inline void
146 put_lba(struct aoe_atahdr *ah, sector_t lba)
148 ah->lba0 = lba;
149 ah->lba1 = lba >>= 8;
150 ah->lba2 = lba >>= 8;
151 ah->lba3 = lba >>= 8;
152 ah->lba4 = lba >>= 8;
153 ah->lba5 = lba >>= 8;
156 static struct aoeif *
157 ifrotate(struct aoetgt *t)
159 struct aoeif *ifp;
161 ifp = t->ifp;
162 ifp++;
163 if (ifp >= &t->ifs[NAOEIFS] || ifp->nd == NULL)
164 ifp = t->ifs;
165 if (ifp->nd == NULL)
166 return NULL;
167 return t->ifp = ifp;
170 static void
171 skb_pool_put(struct aoedev *d, struct sk_buff *skb)
173 __skb_queue_tail(&d->skbpool, skb);
176 static struct sk_buff *
177 skb_pool_get(struct aoedev *d)
179 struct sk_buff *skb = skb_peek(&d->skbpool);
181 if (skb && atomic_read(&skb_shinfo(skb)->dataref) == 1) {
182 __skb_unlink(skb, &d->skbpool);
183 return skb;
185 if (skb_queue_len(&d->skbpool) < NSKBPOOLMAX &&
186 (skb = new_skb(ETH_ZLEN)))
187 return skb;
189 return NULL;
192 void
193 aoe_freetframe(struct frame *f)
195 struct aoetgt *t;
197 t = f->t;
198 f->buf = NULL;
199 memset(&f->iter, 0, sizeof(f->iter));
200 f->r_skb = NULL;
201 f->flags = 0;
202 list_add(&f->head, &t->ffree);
205 static struct frame *
206 newtframe(struct aoedev *d, struct aoetgt *t)
208 struct frame *f;
209 struct sk_buff *skb;
210 struct list_head *pos;
212 if (list_empty(&t->ffree)) {
213 if (t->falloc >= NSKBPOOLMAX*2)
214 return NULL;
215 f = kcalloc(1, sizeof(*f), GFP_ATOMIC);
216 if (f == NULL)
217 return NULL;
218 t->falloc++;
219 f->t = t;
220 } else {
221 pos = t->ffree.next;
222 list_del(pos);
223 f = list_entry(pos, struct frame, head);
226 skb = f->skb;
227 if (skb == NULL) {
228 f->skb = skb = new_skb(ETH_ZLEN);
229 if (!skb) {
230 bail: aoe_freetframe(f);
231 return NULL;
235 if (atomic_read(&skb_shinfo(skb)->dataref) != 1) {
236 skb = skb_pool_get(d);
237 if (skb == NULL)
238 goto bail;
239 skb_pool_put(d, f->skb);
240 f->skb = skb;
243 skb->truesize -= skb->data_len;
244 skb_shinfo(skb)->nr_frags = skb->data_len = 0;
245 skb_trim(skb, 0);
246 return f;
249 static struct frame *
250 newframe(struct aoedev *d)
252 struct frame *f;
253 struct aoetgt *t, **tt;
254 int totout = 0;
255 int use_tainted;
256 int has_untainted;
258 if (!d->targets || !d->targets[0]) {
259 printk(KERN_ERR "aoe: NULL TARGETS!\n");
260 return NULL;
262 tt = d->tgt; /* last used target */
263 for (use_tainted = 0, has_untainted = 0;;) {
264 tt++;
265 if (tt >= &d->targets[d->ntargets] || !*tt)
266 tt = d->targets;
267 t = *tt;
268 if (!t->taint) {
269 has_untainted = 1;
270 totout += t->nout;
272 if (t->nout < t->maxout
273 && (use_tainted || !t->taint)
274 && t->ifp->nd) {
275 f = newtframe(d, t);
276 if (f) {
277 ifrotate(t);
278 d->tgt = tt;
279 return f;
282 if (tt == d->tgt) { /* we've looped and found nada */
283 if (!use_tainted && !has_untainted)
284 use_tainted = 1;
285 else
286 break;
289 if (totout == 0) {
290 d->kicked++;
291 d->flags |= DEVFL_KICKME;
293 return NULL;
296 static void
297 skb_fillup(struct sk_buff *skb, struct bio *bio, struct bvec_iter iter)
299 int frag = 0;
300 struct bio_vec bv;
302 __bio_for_each_segment(bv, bio, iter, iter)
303 skb_fill_page_desc(skb, frag++, bv.bv_page,
304 bv.bv_offset, bv.bv_len);
307 static void
308 fhash(struct frame *f)
310 struct aoedev *d = f->t->d;
311 u32 n;
313 n = f->tag % NFACTIVE;
314 list_add_tail(&f->head, &d->factive[n]);
317 static void
318 ata_rw_frameinit(struct frame *f)
320 struct aoetgt *t;
321 struct aoe_hdr *h;
322 struct aoe_atahdr *ah;
323 struct sk_buff *skb;
324 char writebit, extbit;
326 skb = f->skb;
327 h = (struct aoe_hdr *) skb_mac_header(skb);
328 ah = (struct aoe_atahdr *) (h + 1);
329 skb_put(skb, sizeof(*h) + sizeof(*ah));
330 memset(h, 0, skb->len);
332 writebit = 0x10;
333 extbit = 0x4;
335 t = f->t;
336 f->tag = aoehdr_atainit(t->d, t, h);
337 fhash(f);
338 t->nout++;
339 f->waited = 0;
340 f->waited_total = 0;
342 /* set up ata header */
343 ah->scnt = f->iter.bi_size >> 9;
344 put_lba(ah, f->iter.bi_sector);
345 if (t->d->flags & DEVFL_EXT) {
346 ah->aflags |= AOEAFL_EXT;
347 } else {
348 extbit = 0;
349 ah->lba3 &= 0x0f;
350 ah->lba3 |= 0xe0; /* LBA bit + obsolete 0xa0 */
352 if (f->buf && bio_data_dir(f->buf->bio) == WRITE) {
353 skb_fillup(skb, f->buf->bio, f->iter);
354 ah->aflags |= AOEAFL_WRITE;
355 skb->len += f->iter.bi_size;
356 skb->data_len = f->iter.bi_size;
357 skb->truesize += f->iter.bi_size;
358 t->wpkts++;
359 } else {
360 t->rpkts++;
361 writebit = 0;
364 ah->cmdstat = ATA_CMD_PIO_READ | writebit | extbit;
365 skb->dev = t->ifp->nd;
368 static int
369 aoecmd_ata_rw(struct aoedev *d)
371 struct frame *f;
372 struct buf *buf;
373 struct sk_buff *skb;
374 struct sk_buff_head queue;
376 buf = nextbuf(d);
377 if (buf == NULL)
378 return 0;
379 f = newframe(d);
380 if (f == NULL)
381 return 0;
383 /* initialize the headers & frame */
384 f->buf = buf;
385 f->iter = buf->iter;
386 f->iter.bi_size = min_t(unsigned long,
387 d->maxbcnt ?: DEFAULTBCNT,
388 f->iter.bi_size);
389 bio_advance_iter(buf->bio, &buf->iter, f->iter.bi_size);
391 if (!buf->iter.bi_size)
392 d->ip.buf = NULL;
394 /* mark all tracking fields and load out */
395 buf->nframesout += 1;
397 ata_rw_frameinit(f);
399 skb = skb_clone(f->skb, GFP_ATOMIC);
400 if (skb) {
401 do_gettimeofday(&f->sent);
402 f->sent_jiffs = (u32) jiffies;
403 __skb_queue_head_init(&queue);
404 __skb_queue_tail(&queue, skb);
405 aoenet_xmit(&queue);
407 return 1;
410 /* some callers cannot sleep, and they can call this function,
411 * transmitting the packets later, when interrupts are on
413 static void
414 aoecmd_cfg_pkts(ushort aoemajor, unsigned char aoeminor, struct sk_buff_head *queue)
416 struct aoe_hdr *h;
417 struct aoe_cfghdr *ch;
418 struct sk_buff *skb;
419 struct net_device *ifp;
421 rcu_read_lock();
422 for_each_netdev_rcu(&init_net, ifp) {
423 dev_hold(ifp);
424 if (!is_aoe_netif(ifp))
425 goto cont;
427 skb = new_skb(sizeof *h + sizeof *ch);
428 if (skb == NULL) {
429 printk(KERN_INFO "aoe: skb alloc failure\n");
430 goto cont;
432 skb_put(skb, sizeof *h + sizeof *ch);
433 skb->dev = ifp;
434 __skb_queue_tail(queue, skb);
435 h = (struct aoe_hdr *) skb_mac_header(skb);
436 memset(h, 0, sizeof *h + sizeof *ch);
438 memset(h->dst, 0xff, sizeof h->dst);
439 memcpy(h->src, ifp->dev_addr, sizeof h->src);
440 h->type = __constant_cpu_to_be16(ETH_P_AOE);
441 h->verfl = AOE_HVER;
442 h->major = cpu_to_be16(aoemajor);
443 h->minor = aoeminor;
444 h->cmd = AOECMD_CFG;
446 cont:
447 dev_put(ifp);
449 rcu_read_unlock();
452 static void
453 resend(struct aoedev *d, struct frame *f)
455 struct sk_buff *skb;
456 struct sk_buff_head queue;
457 struct aoe_hdr *h;
458 struct aoetgt *t;
459 char buf[128];
460 u32 n;
462 t = f->t;
463 n = newtag(d);
464 skb = f->skb;
465 if (ifrotate(t) == NULL) {
466 /* probably can't happen, but set it up to fail anyway */
467 pr_info("aoe: resend: no interfaces to rotate to.\n");
468 ktcomplete(f, NULL);
469 return;
471 h = (struct aoe_hdr *) skb_mac_header(skb);
473 if (!(f->flags & FFL_PROBE)) {
474 snprintf(buf, sizeof(buf),
475 "%15s e%ld.%d oldtag=%08x@%08lx newtag=%08x s=%pm d=%pm nout=%d\n",
476 "retransmit", d->aoemajor, d->aoeminor,
477 f->tag, jiffies, n,
478 h->src, h->dst, t->nout);
479 aoechr_error(buf);
482 f->tag = n;
483 fhash(f);
484 h->tag = cpu_to_be32(n);
485 memcpy(h->dst, t->addr, sizeof h->dst);
486 memcpy(h->src, t->ifp->nd->dev_addr, sizeof h->src);
488 skb->dev = t->ifp->nd;
489 skb = skb_clone(skb, GFP_ATOMIC);
490 if (skb == NULL)
491 return;
492 do_gettimeofday(&f->sent);
493 f->sent_jiffs = (u32) jiffies;
494 __skb_queue_head_init(&queue);
495 __skb_queue_tail(&queue, skb);
496 aoenet_xmit(&queue);
499 static int
500 tsince_hr(struct frame *f)
502 struct timeval now;
503 int n;
505 do_gettimeofday(&now);
506 n = now.tv_usec - f->sent.tv_usec;
507 n += (now.tv_sec - f->sent.tv_sec) * USEC_PER_SEC;
509 if (n < 0)
510 n = -n;
512 /* For relatively long periods, use jiffies to avoid
513 * discrepancies caused by updates to the system time.
515 * On system with HZ of 1000, 32-bits is over 49 days
516 * worth of jiffies, or over 71 minutes worth of usecs.
518 * Jiffies overflow is handled by subtraction of unsigned ints:
519 * (gdb) print (unsigned) 2 - (unsigned) 0xfffffffe
520 * $3 = 4
521 * (gdb)
523 if (n > USEC_PER_SEC / 4) {
524 n = ((u32) jiffies) - f->sent_jiffs;
525 n *= USEC_PER_SEC / HZ;
528 return n;
531 static int
532 tsince(u32 tag)
534 int n;
536 n = jiffies & 0xffff;
537 n -= tag & 0xffff;
538 if (n < 0)
539 n += 1<<16;
540 return jiffies_to_usecs(n + 1);
543 static struct aoeif *
544 getif(struct aoetgt *t, struct net_device *nd)
546 struct aoeif *p, *e;
548 p = t->ifs;
549 e = p + NAOEIFS;
550 for (; p < e; p++)
551 if (p->nd == nd)
552 return p;
553 return NULL;
556 static void
557 ejectif(struct aoetgt *t, struct aoeif *ifp)
559 struct aoeif *e;
560 struct net_device *nd;
561 ulong n;
563 nd = ifp->nd;
564 e = t->ifs + NAOEIFS - 1;
565 n = (e - ifp) * sizeof *ifp;
566 memmove(ifp, ifp+1, n);
567 e->nd = NULL;
568 dev_put(nd);
571 static struct frame *
572 reassign_frame(struct frame *f)
574 struct frame *nf;
575 struct sk_buff *skb;
577 nf = newframe(f->t->d);
578 if (!nf)
579 return NULL;
580 if (nf->t == f->t) {
581 aoe_freetframe(nf);
582 return NULL;
585 skb = nf->skb;
586 nf->skb = f->skb;
587 nf->buf = f->buf;
588 nf->iter = f->iter;
589 nf->waited = 0;
590 nf->waited_total = f->waited_total;
591 nf->sent = f->sent;
592 nf->sent_jiffs = f->sent_jiffs;
593 f->skb = skb;
595 return nf;
598 static void
599 probe(struct aoetgt *t)
601 struct aoedev *d;
602 struct frame *f;
603 struct sk_buff *skb;
604 struct sk_buff_head queue;
605 size_t n, m;
606 int frag;
608 d = t->d;
609 f = newtframe(d, t);
610 if (!f) {
611 pr_err("%s %pm for e%ld.%d: %s\n",
612 "aoe: cannot probe remote address",
613 t->addr,
614 (long) d->aoemajor, d->aoeminor,
615 "no frame available");
616 return;
618 f->flags |= FFL_PROBE;
619 ifrotate(t);
620 f->iter.bi_size = t->d->maxbcnt ? t->d->maxbcnt : DEFAULTBCNT;
621 ata_rw_frameinit(f);
622 skb = f->skb;
623 for (frag = 0, n = f->iter.bi_size; n > 0; ++frag, n -= m) {
624 if (n < PAGE_SIZE)
625 m = n;
626 else
627 m = PAGE_SIZE;
628 skb_fill_page_desc(skb, frag, empty_page, 0, m);
630 skb->len += f->iter.bi_size;
631 skb->data_len = f->iter.bi_size;
632 skb->truesize += f->iter.bi_size;
634 skb = skb_clone(f->skb, GFP_ATOMIC);
635 if (skb) {
636 do_gettimeofday(&f->sent);
637 f->sent_jiffs = (u32) jiffies;
638 __skb_queue_head_init(&queue);
639 __skb_queue_tail(&queue, skb);
640 aoenet_xmit(&queue);
644 static long
645 rto(struct aoedev *d)
647 long t;
649 t = 2 * d->rttavg >> RTTSCALE;
650 t += 8 * d->rttdev >> RTTDSCALE;
651 if (t == 0)
652 t = 1;
654 return t;
657 static void
658 rexmit_deferred(struct aoedev *d)
660 struct aoetgt *t;
661 struct frame *f;
662 struct frame *nf;
663 struct list_head *pos, *nx, *head;
664 int since;
665 int untainted;
667 count_targets(d, &untainted);
669 head = &d->rexmitq;
670 list_for_each_safe(pos, nx, head) {
671 f = list_entry(pos, struct frame, head);
672 t = f->t;
673 if (t->taint) {
674 if (!(f->flags & FFL_PROBE)) {
675 nf = reassign_frame(f);
676 if (nf) {
677 if (t->nout_probes == 0
678 && untainted > 0) {
679 probe(t);
680 t->nout_probes++;
682 list_replace(&f->head, &nf->head);
683 pos = &nf->head;
684 aoe_freetframe(f);
685 f = nf;
686 t = f->t;
688 } else if (untainted < 1) {
689 /* don't probe w/o other untainted aoetgts */
690 goto stop_probe;
691 } else if (tsince_hr(f) < t->taint * rto(d)) {
692 /* reprobe slowly when taint is high */
693 continue;
695 } else if (f->flags & FFL_PROBE) {
696 stop_probe: /* don't probe untainted aoetgts */
697 list_del(pos);
698 aoe_freetframe(f);
699 /* leaving d->kicked, because this is routine */
700 f->t->d->flags |= DEVFL_KICKME;
701 continue;
703 if (t->nout >= t->maxout)
704 continue;
705 list_del(pos);
706 t->nout++;
707 if (f->flags & FFL_PROBE)
708 t->nout_probes++;
709 since = tsince_hr(f);
710 f->waited += since;
711 f->waited_total += since;
712 resend(d, f);
716 /* An aoetgt accumulates demerits quickly, and successful
717 * probing redeems the aoetgt slowly.
719 static void
720 scorn(struct aoetgt *t)
722 int n;
724 n = t->taint++;
725 t->taint += t->taint * 2;
726 if (n > t->taint)
727 t->taint = n;
728 if (t->taint > MAX_TAINT)
729 t->taint = MAX_TAINT;
732 static int
733 count_targets(struct aoedev *d, int *untainted)
735 int i, good;
737 for (i = good = 0; i < d->ntargets && d->targets[i]; ++i)
738 if (d->targets[i]->taint == 0)
739 good++;
741 if (untainted)
742 *untainted = good;
743 return i;
746 static void
747 rexmit_timer(ulong vp)
749 struct aoedev *d;
750 struct aoetgt *t;
751 struct aoeif *ifp;
752 struct frame *f;
753 struct list_head *head, *pos, *nx;
754 LIST_HEAD(flist);
755 register long timeout;
756 ulong flags, n;
757 int i;
758 int utgts; /* number of aoetgt descriptors (not slots) */
759 int since;
761 d = (struct aoedev *) vp;
763 spin_lock_irqsave(&d->lock, flags);
765 /* timeout based on observed timings and variations */
766 timeout = rto(d);
768 utgts = count_targets(d, NULL);
770 if (d->flags & DEVFL_TKILL) {
771 spin_unlock_irqrestore(&d->lock, flags);
772 return;
775 /* collect all frames to rexmit into flist */
776 for (i = 0; i < NFACTIVE; i++) {
777 head = &d->factive[i];
778 list_for_each_safe(pos, nx, head) {
779 f = list_entry(pos, struct frame, head);
780 if (tsince_hr(f) < timeout)
781 break; /* end of expired frames */
782 /* move to flist for later processing */
783 list_move_tail(pos, &flist);
787 /* process expired frames */
788 while (!list_empty(&flist)) {
789 pos = flist.next;
790 f = list_entry(pos, struct frame, head);
791 since = tsince_hr(f);
792 n = f->waited_total + since;
793 n /= USEC_PER_SEC;
794 if (aoe_deadsecs
795 && n > aoe_deadsecs
796 && !(f->flags & FFL_PROBE)) {
797 /* Waited too long. Device failure.
798 * Hang all frames on first hash bucket for downdev
799 * to clean up.
801 list_splice(&flist, &d->factive[0]);
802 aoedev_downdev(d);
803 goto out;
806 t = f->t;
807 n = f->waited + since;
808 n /= USEC_PER_SEC;
809 if (aoe_deadsecs && utgts > 0
810 && (n > aoe_deadsecs / utgts || n > HARD_SCORN_SECS))
811 scorn(t); /* avoid this target */
813 if (t->maxout != 1) {
814 t->ssthresh = t->maxout / 2;
815 t->maxout = 1;
818 if (f->flags & FFL_PROBE) {
819 t->nout_probes--;
820 } else {
821 ifp = getif(t, f->skb->dev);
822 if (ifp && ++ifp->lost > (t->nframes << 1)
823 && (ifp != t->ifs || t->ifs[1].nd)) {
824 ejectif(t, ifp);
825 ifp = NULL;
828 list_move_tail(pos, &d->rexmitq);
829 t->nout--;
831 rexmit_deferred(d);
833 out:
834 if ((d->flags & DEVFL_KICKME) && d->blkq) {
835 d->flags &= ~DEVFL_KICKME;
836 d->blkq->request_fn(d->blkq);
839 d->timer.expires = jiffies + TIMERTICK;
840 add_timer(&d->timer);
842 spin_unlock_irqrestore(&d->lock, flags);
845 static unsigned long
846 rqbiocnt(struct request *r)
848 struct bio *bio;
849 unsigned long n = 0;
851 __rq_for_each_bio(bio, r)
852 n++;
853 return n;
856 /* This can be removed if we are certain that no users of the block
857 * layer will ever use zero-count pages in bios. Otherwise we have to
858 * protect against the put_page sometimes done by the network layer.
860 * See http://oss.sgi.com/archives/xfs/2007-01/msg00594.html for
861 * discussion.
863 * We cannot use get_page in the workaround, because it insists on a
864 * positive page count as a precondition. So we use _count directly.
866 static void
867 bio_pageinc(struct bio *bio)
869 struct bio_vec bv;
870 struct page *page;
871 struct bvec_iter iter;
873 bio_for_each_segment(bv, bio, iter) {
874 /* Non-zero page count for non-head members of
875 * compound pages is no longer allowed by the kernel.
877 page = compound_trans_head(bv.bv_page);
878 atomic_inc(&page->_count);
882 static void
883 bio_pagedec(struct bio *bio)
885 struct page *page;
886 struct bio_vec bv;
887 struct bvec_iter iter;
889 bio_for_each_segment(bv, bio, iter) {
890 page = compound_trans_head(bv.bv_page);
891 atomic_dec(&page->_count);
895 static void
896 bufinit(struct buf *buf, struct request *rq, struct bio *bio)
898 memset(buf, 0, sizeof(*buf));
899 buf->rq = rq;
900 buf->bio = bio;
901 buf->iter = bio->bi_iter;
902 bio_pageinc(bio);
905 static struct buf *
906 nextbuf(struct aoedev *d)
908 struct request *rq;
909 struct request_queue *q;
910 struct buf *buf;
911 struct bio *bio;
913 q = d->blkq;
914 if (q == NULL)
915 return NULL; /* initializing */
916 if (d->ip.buf)
917 return d->ip.buf;
918 rq = d->ip.rq;
919 if (rq == NULL) {
920 rq = blk_peek_request(q);
921 if (rq == NULL)
922 return NULL;
923 blk_start_request(rq);
924 d->ip.rq = rq;
925 d->ip.nxbio = rq->bio;
926 rq->special = (void *) rqbiocnt(rq);
928 buf = mempool_alloc(d->bufpool, GFP_ATOMIC);
929 if (buf == NULL) {
930 pr_err("aoe: nextbuf: unable to mempool_alloc!\n");
931 return NULL;
933 bio = d->ip.nxbio;
934 bufinit(buf, rq, bio);
935 bio = bio->bi_next;
936 d->ip.nxbio = bio;
937 if (bio == NULL)
938 d->ip.rq = NULL;
939 return d->ip.buf = buf;
942 /* enters with d->lock held */
943 void
944 aoecmd_work(struct aoedev *d)
946 rexmit_deferred(d);
947 while (aoecmd_ata_rw(d))
951 /* this function performs work that has been deferred until sleeping is OK
953 void
954 aoecmd_sleepwork(struct work_struct *work)
956 struct aoedev *d = container_of(work, struct aoedev, work);
957 struct block_device *bd;
958 u64 ssize;
960 if (d->flags & DEVFL_GDALLOC)
961 aoeblk_gdalloc(d);
963 if (d->flags & DEVFL_NEWSIZE) {
964 ssize = get_capacity(d->gd);
965 bd = bdget_disk(d->gd, 0);
966 if (bd) {
967 mutex_lock(&bd->bd_inode->i_mutex);
968 i_size_write(bd->bd_inode, (loff_t)ssize<<9);
969 mutex_unlock(&bd->bd_inode->i_mutex);
970 bdput(bd);
972 spin_lock_irq(&d->lock);
973 d->flags |= DEVFL_UP;
974 d->flags &= ~DEVFL_NEWSIZE;
975 spin_unlock_irq(&d->lock);
979 static void
980 ata_ident_fixstring(u16 *id, int ns)
982 u16 s;
984 while (ns-- > 0) {
985 s = *id;
986 *id++ = s >> 8 | s << 8;
990 static void
991 ataid_complete(struct aoedev *d, struct aoetgt *t, unsigned char *id)
993 u64 ssize;
994 u16 n;
996 /* word 83: command set supported */
997 n = get_unaligned_le16(&id[83 << 1]);
999 /* word 86: command set/feature enabled */
1000 n |= get_unaligned_le16(&id[86 << 1]);
1002 if (n & (1<<10)) { /* bit 10: LBA 48 */
1003 d->flags |= DEVFL_EXT;
1005 /* word 100: number lba48 sectors */
1006 ssize = get_unaligned_le64(&id[100 << 1]);
1008 /* set as in ide-disk.c:init_idedisk_capacity */
1009 d->geo.cylinders = ssize;
1010 d->geo.cylinders /= (255 * 63);
1011 d->geo.heads = 255;
1012 d->geo.sectors = 63;
1013 } else {
1014 d->flags &= ~DEVFL_EXT;
1016 /* number lba28 sectors */
1017 ssize = get_unaligned_le32(&id[60 << 1]);
1019 /* NOTE: obsolete in ATA 6 */
1020 d->geo.cylinders = get_unaligned_le16(&id[54 << 1]);
1021 d->geo.heads = get_unaligned_le16(&id[55 << 1]);
1022 d->geo.sectors = get_unaligned_le16(&id[56 << 1]);
1025 ata_ident_fixstring((u16 *) &id[10<<1], 10); /* serial */
1026 ata_ident_fixstring((u16 *) &id[23<<1], 4); /* firmware */
1027 ata_ident_fixstring((u16 *) &id[27<<1], 20); /* model */
1028 memcpy(d->ident, id, sizeof(d->ident));
1030 if (d->ssize != ssize)
1031 printk(KERN_INFO
1032 "aoe: %pm e%ld.%d v%04x has %llu sectors\n",
1033 t->addr,
1034 d->aoemajor, d->aoeminor,
1035 d->fw_ver, (long long)ssize);
1036 d->ssize = ssize;
1037 d->geo.start = 0;
1038 if (d->flags & (DEVFL_GDALLOC|DEVFL_NEWSIZE))
1039 return;
1040 if (d->gd != NULL) {
1041 set_capacity(d->gd, ssize);
1042 d->flags |= DEVFL_NEWSIZE;
1043 } else
1044 d->flags |= DEVFL_GDALLOC;
1045 schedule_work(&d->work);
1048 static void
1049 calc_rttavg(struct aoedev *d, struct aoetgt *t, int rtt)
1051 register long n;
1053 n = rtt;
1055 /* cf. Congestion Avoidance and Control, Jacobson & Karels, 1988 */
1056 n -= d->rttavg >> RTTSCALE;
1057 d->rttavg += n;
1058 if (n < 0)
1059 n = -n;
1060 n -= d->rttdev >> RTTDSCALE;
1061 d->rttdev += n;
1063 if (!t || t->maxout >= t->nframes)
1064 return;
1065 if (t->maxout < t->ssthresh)
1066 t->maxout += 1;
1067 else if (t->nout == t->maxout && t->next_cwnd-- == 0) {
1068 t->maxout += 1;
1069 t->next_cwnd = t->maxout;
1073 static struct aoetgt *
1074 gettgt(struct aoedev *d, char *addr)
1076 struct aoetgt **t, **e;
1078 t = d->targets;
1079 e = t + d->ntargets;
1080 for (; t < e && *t; t++)
1081 if (memcmp((*t)->addr, addr, sizeof((*t)->addr)) == 0)
1082 return *t;
1083 return NULL;
1086 static void
1087 bvcpy(struct sk_buff *skb, struct bio *bio, struct bvec_iter iter, long cnt)
1089 int soff = 0;
1090 struct bio_vec bv;
1092 iter.bi_size = cnt;
1094 __bio_for_each_segment(bv, bio, iter, iter) {
1095 char *p = page_address(bv.bv_page) + bv.bv_offset;
1096 skb_copy_bits(skb, soff, p, bv.bv_len);
1097 soff += bv.bv_len;
1101 void
1102 aoe_end_request(struct aoedev *d, struct request *rq, int fastfail)
1104 struct bio *bio;
1105 int bok;
1106 struct request_queue *q;
1108 q = d->blkq;
1109 if (rq == d->ip.rq)
1110 d->ip.rq = NULL;
1111 do {
1112 bio = rq->bio;
1113 bok = !fastfail && test_bit(BIO_UPTODATE, &bio->bi_flags);
1114 } while (__blk_end_request(rq, bok ? 0 : -EIO, bio->bi_iter.bi_size));
1116 /* cf. http://lkml.org/lkml/2006/10/31/28 */
1117 if (!fastfail)
1118 __blk_run_queue(q);
1121 static void
1122 aoe_end_buf(struct aoedev *d, struct buf *buf)
1124 struct request *rq;
1125 unsigned long n;
1127 if (buf == d->ip.buf)
1128 d->ip.buf = NULL;
1129 rq = buf->rq;
1130 bio_pagedec(buf->bio);
1131 mempool_free(buf, d->bufpool);
1132 n = (unsigned long) rq->special;
1133 rq->special = (void *) --n;
1134 if (n == 0)
1135 aoe_end_request(d, rq, 0);
1138 static void
1139 ktiocomplete(struct frame *f)
1141 struct aoe_hdr *hin, *hout;
1142 struct aoe_atahdr *ahin, *ahout;
1143 struct buf *buf;
1144 struct sk_buff *skb;
1145 struct aoetgt *t;
1146 struct aoeif *ifp;
1147 struct aoedev *d;
1148 long n;
1149 int untainted;
1151 if (f == NULL)
1152 return;
1154 t = f->t;
1155 d = t->d;
1156 skb = f->r_skb;
1157 buf = f->buf;
1158 if (f->flags & FFL_PROBE)
1159 goto out;
1160 if (!skb) /* just fail the buf. */
1161 goto noskb;
1163 hout = (struct aoe_hdr *) skb_mac_header(f->skb);
1164 ahout = (struct aoe_atahdr *) (hout+1);
1166 hin = (struct aoe_hdr *) skb->data;
1167 skb_pull(skb, sizeof(*hin));
1168 ahin = (struct aoe_atahdr *) skb->data;
1169 skb_pull(skb, sizeof(*ahin));
1170 if (ahin->cmdstat & 0xa9) { /* these bits cleared on success */
1171 pr_err("aoe: ata error cmd=%2.2Xh stat=%2.2Xh from e%ld.%d\n",
1172 ahout->cmdstat, ahin->cmdstat,
1173 d->aoemajor, d->aoeminor);
1174 noskb: if (buf)
1175 clear_bit(BIO_UPTODATE, &buf->bio->bi_flags);
1176 goto out;
1179 n = ahout->scnt << 9;
1180 switch (ahout->cmdstat) {
1181 case ATA_CMD_PIO_READ:
1182 case ATA_CMD_PIO_READ_EXT:
1183 if (skb->len < n) {
1184 pr_err("%s e%ld.%d. skb->len=%d need=%ld\n",
1185 "aoe: runt data size in read from",
1186 (long) d->aoemajor, d->aoeminor,
1187 skb->len, n);
1188 clear_bit(BIO_UPTODATE, &buf->bio->bi_flags);
1189 break;
1191 if (n > f->iter.bi_size) {
1192 pr_err_ratelimited("%s e%ld.%d. bytes=%ld need=%u\n",
1193 "aoe: too-large data size in read from",
1194 (long) d->aoemajor, d->aoeminor,
1195 n, f->iter.bi_size);
1196 clear_bit(BIO_UPTODATE, &buf->bio->bi_flags);
1197 break;
1199 bvcpy(skb, f->buf->bio, f->iter, n);
1200 case ATA_CMD_PIO_WRITE:
1201 case ATA_CMD_PIO_WRITE_EXT:
1202 spin_lock_irq(&d->lock);
1203 ifp = getif(t, skb->dev);
1204 if (ifp)
1205 ifp->lost = 0;
1206 spin_unlock_irq(&d->lock);
1207 break;
1208 case ATA_CMD_ID_ATA:
1209 if (skb->len < 512) {
1210 pr_info("%s e%ld.%d. skb->len=%d need=512\n",
1211 "aoe: runt data size in ataid from",
1212 (long) d->aoemajor, d->aoeminor,
1213 skb->len);
1214 break;
1216 if (skb_linearize(skb))
1217 break;
1218 spin_lock_irq(&d->lock);
1219 ataid_complete(d, t, skb->data);
1220 spin_unlock_irq(&d->lock);
1221 break;
1222 default:
1223 pr_info("aoe: unrecognized ata command %2.2Xh for %d.%d\n",
1224 ahout->cmdstat,
1225 be16_to_cpu(get_unaligned(&hin->major)),
1226 hin->minor);
1228 out:
1229 spin_lock_irq(&d->lock);
1230 if (t->taint > 0
1231 && --t->taint > 0
1232 && t->nout_probes == 0) {
1233 count_targets(d, &untainted);
1234 if (untainted > 0) {
1235 probe(t);
1236 t->nout_probes++;
1240 aoe_freetframe(f);
1242 if (buf && --buf->nframesout == 0 && buf->iter.bi_size == 0)
1243 aoe_end_buf(d, buf);
1245 spin_unlock_irq(&d->lock);
1246 aoedev_put(d);
1247 dev_kfree_skb(skb);
1250 /* Enters with iocq.lock held.
1251 * Returns true iff responses needing processing remain.
1253 static int
1254 ktio(int id)
1256 struct frame *f;
1257 struct list_head *pos;
1258 int i;
1259 int actual_id;
1261 for (i = 0; ; ++i) {
1262 if (i == MAXIOC)
1263 return 1;
1264 if (list_empty(&iocq[id].head))
1265 return 0;
1266 pos = iocq[id].head.next;
1267 list_del(pos);
1268 f = list_entry(pos, struct frame, head);
1269 spin_unlock_irq(&iocq[id].lock);
1270 ktiocomplete(f);
1272 /* Figure out if extra threads are required. */
1273 actual_id = f->t->d->aoeminor % ncpus;
1275 if (!kts[actual_id].active) {
1276 BUG_ON(id != 0);
1277 mutex_lock(&ktio_spawn_lock);
1278 if (!kts[actual_id].active
1279 && aoe_ktstart(&kts[actual_id]) == 0)
1280 kts[actual_id].active = 1;
1281 mutex_unlock(&ktio_spawn_lock);
1283 spin_lock_irq(&iocq[id].lock);
1287 static int
1288 kthread(void *vp)
1290 struct ktstate *k;
1291 DECLARE_WAITQUEUE(wait, current);
1292 int more;
1294 k = vp;
1295 current->flags |= PF_NOFREEZE;
1296 set_user_nice(current, -10);
1297 complete(&k->rendez); /* tell spawner we're running */
1298 do {
1299 spin_lock_irq(k->lock);
1300 more = k->fn(k->id);
1301 if (!more) {
1302 add_wait_queue(k->waitq, &wait);
1303 __set_current_state(TASK_INTERRUPTIBLE);
1305 spin_unlock_irq(k->lock);
1306 if (!more) {
1307 schedule();
1308 remove_wait_queue(k->waitq, &wait);
1309 } else
1310 cond_resched();
1311 } while (!kthread_should_stop());
1312 complete(&k->rendez); /* tell spawner we're stopping */
1313 return 0;
1316 void
1317 aoe_ktstop(struct ktstate *k)
1319 kthread_stop(k->task);
1320 wait_for_completion(&k->rendez);
1324 aoe_ktstart(struct ktstate *k)
1326 struct task_struct *task;
1328 init_completion(&k->rendez);
1329 task = kthread_run(kthread, k, "%s", k->name);
1330 if (task == NULL || IS_ERR(task))
1331 return -ENOMEM;
1332 k->task = task;
1333 wait_for_completion(&k->rendez); /* allow kthread to start */
1334 init_completion(&k->rendez); /* for waiting for exit later */
1335 return 0;
1338 /* pass it off to kthreads for processing */
1339 static void
1340 ktcomplete(struct frame *f, struct sk_buff *skb)
1342 int id;
1343 ulong flags;
1345 f->r_skb = skb;
1346 id = f->t->d->aoeminor % ncpus;
1347 spin_lock_irqsave(&iocq[id].lock, flags);
1348 if (!kts[id].active) {
1349 spin_unlock_irqrestore(&iocq[id].lock, flags);
1350 /* The thread with id has not been spawned yet,
1351 * so delegate the work to the main thread and
1352 * try spawning a new thread.
1354 id = 0;
1355 spin_lock_irqsave(&iocq[id].lock, flags);
1357 list_add_tail(&f->head, &iocq[id].head);
1358 spin_unlock_irqrestore(&iocq[id].lock, flags);
1359 wake_up(&ktiowq[id]);
1362 struct sk_buff *
1363 aoecmd_ata_rsp(struct sk_buff *skb)
1365 struct aoedev *d;
1366 struct aoe_hdr *h;
1367 struct frame *f;
1368 u32 n;
1369 ulong flags;
1370 char ebuf[128];
1371 u16 aoemajor;
1373 h = (struct aoe_hdr *) skb->data;
1374 aoemajor = be16_to_cpu(get_unaligned(&h->major));
1375 d = aoedev_by_aoeaddr(aoemajor, h->minor, 0);
1376 if (d == NULL) {
1377 snprintf(ebuf, sizeof ebuf, "aoecmd_ata_rsp: ata response "
1378 "for unknown device %d.%d\n",
1379 aoemajor, h->minor);
1380 aoechr_error(ebuf);
1381 return skb;
1384 spin_lock_irqsave(&d->lock, flags);
1386 n = be32_to_cpu(get_unaligned(&h->tag));
1387 f = getframe(d, n);
1388 if (f) {
1389 calc_rttavg(d, f->t, tsince_hr(f));
1390 f->t->nout--;
1391 if (f->flags & FFL_PROBE)
1392 f->t->nout_probes--;
1393 } else {
1394 f = getframe_deferred(d, n);
1395 if (f) {
1396 calc_rttavg(d, NULL, tsince_hr(f));
1397 } else {
1398 calc_rttavg(d, NULL, tsince(n));
1399 spin_unlock_irqrestore(&d->lock, flags);
1400 aoedev_put(d);
1401 snprintf(ebuf, sizeof(ebuf),
1402 "%15s e%d.%d tag=%08x@%08lx s=%pm d=%pm\n",
1403 "unexpected rsp",
1404 get_unaligned_be16(&h->major),
1405 h->minor,
1406 get_unaligned_be32(&h->tag),
1407 jiffies,
1408 h->src,
1409 h->dst);
1410 aoechr_error(ebuf);
1411 return skb;
1414 aoecmd_work(d);
1416 spin_unlock_irqrestore(&d->lock, flags);
1418 ktcomplete(f, skb);
1421 * Note here that we do not perform an aoedev_put, as we are
1422 * leaving this reference for the ktio to release.
1424 return NULL;
1427 void
1428 aoecmd_cfg(ushort aoemajor, unsigned char aoeminor)
1430 struct sk_buff_head queue;
1432 __skb_queue_head_init(&queue);
1433 aoecmd_cfg_pkts(aoemajor, aoeminor, &queue);
1434 aoenet_xmit(&queue);
1437 struct sk_buff *
1438 aoecmd_ata_id(struct aoedev *d)
1440 struct aoe_hdr *h;
1441 struct aoe_atahdr *ah;
1442 struct frame *f;
1443 struct sk_buff *skb;
1444 struct aoetgt *t;
1446 f = newframe(d);
1447 if (f == NULL)
1448 return NULL;
1450 t = *d->tgt;
1452 /* initialize the headers & frame */
1453 skb = f->skb;
1454 h = (struct aoe_hdr *) skb_mac_header(skb);
1455 ah = (struct aoe_atahdr *) (h+1);
1456 skb_put(skb, sizeof *h + sizeof *ah);
1457 memset(h, 0, skb->len);
1458 f->tag = aoehdr_atainit(d, t, h);
1459 fhash(f);
1460 t->nout++;
1461 f->waited = 0;
1462 f->waited_total = 0;
1464 /* set up ata header */
1465 ah->scnt = 1;
1466 ah->cmdstat = ATA_CMD_ID_ATA;
1467 ah->lba3 = 0xa0;
1469 skb->dev = t->ifp->nd;
1471 d->rttavg = RTTAVG_INIT;
1472 d->rttdev = RTTDEV_INIT;
1473 d->timer.function = rexmit_timer;
1475 skb = skb_clone(skb, GFP_ATOMIC);
1476 if (skb) {
1477 do_gettimeofday(&f->sent);
1478 f->sent_jiffs = (u32) jiffies;
1481 return skb;
1484 static struct aoetgt **
1485 grow_targets(struct aoedev *d)
1487 ulong oldn, newn;
1488 struct aoetgt **tt;
1490 oldn = d->ntargets;
1491 newn = oldn * 2;
1492 tt = kcalloc(newn, sizeof(*d->targets), GFP_ATOMIC);
1493 if (!tt)
1494 return NULL;
1495 memmove(tt, d->targets, sizeof(*d->targets) * oldn);
1496 d->tgt = tt + (d->tgt - d->targets);
1497 kfree(d->targets);
1498 d->targets = tt;
1499 d->ntargets = newn;
1501 return &d->targets[oldn];
1504 static struct aoetgt *
1505 addtgt(struct aoedev *d, char *addr, ulong nframes)
1507 struct aoetgt *t, **tt, **te;
1509 tt = d->targets;
1510 te = tt + d->ntargets;
1511 for (; tt < te && *tt; tt++)
1514 if (tt == te) {
1515 tt = grow_targets(d);
1516 if (!tt)
1517 goto nomem;
1519 t = kzalloc(sizeof(*t), GFP_ATOMIC);
1520 if (!t)
1521 goto nomem;
1522 t->nframes = nframes;
1523 t->d = d;
1524 memcpy(t->addr, addr, sizeof t->addr);
1525 t->ifp = t->ifs;
1526 aoecmd_wreset(t);
1527 t->maxout = t->nframes / 2;
1528 INIT_LIST_HEAD(&t->ffree);
1529 return *tt = t;
1531 nomem:
1532 pr_info("aoe: cannot allocate memory to add target\n");
1533 return NULL;
1536 static void
1537 setdbcnt(struct aoedev *d)
1539 struct aoetgt **t, **e;
1540 int bcnt = 0;
1542 t = d->targets;
1543 e = t + d->ntargets;
1544 for (; t < e && *t; t++)
1545 if (bcnt == 0 || bcnt > (*t)->minbcnt)
1546 bcnt = (*t)->minbcnt;
1547 if (bcnt != d->maxbcnt) {
1548 d->maxbcnt = bcnt;
1549 pr_info("aoe: e%ld.%d: setting %d byte data frames\n",
1550 d->aoemajor, d->aoeminor, bcnt);
1554 static void
1555 setifbcnt(struct aoetgt *t, struct net_device *nd, int bcnt)
1557 struct aoedev *d;
1558 struct aoeif *p, *e;
1559 int minbcnt;
1561 d = t->d;
1562 minbcnt = bcnt;
1563 p = t->ifs;
1564 e = p + NAOEIFS;
1565 for (; p < e; p++) {
1566 if (p->nd == NULL)
1567 break; /* end of the valid interfaces */
1568 if (p->nd == nd) {
1569 p->bcnt = bcnt; /* we're updating */
1570 nd = NULL;
1571 } else if (minbcnt > p->bcnt)
1572 minbcnt = p->bcnt; /* find the min interface */
1574 if (nd) {
1575 if (p == e) {
1576 pr_err("aoe: device setifbcnt failure; too many interfaces.\n");
1577 return;
1579 dev_hold(nd);
1580 p->nd = nd;
1581 p->bcnt = bcnt;
1583 t->minbcnt = minbcnt;
1584 setdbcnt(d);
1587 void
1588 aoecmd_cfg_rsp(struct sk_buff *skb)
1590 struct aoedev *d;
1591 struct aoe_hdr *h;
1592 struct aoe_cfghdr *ch;
1593 struct aoetgt *t;
1594 ulong flags, aoemajor;
1595 struct sk_buff *sl;
1596 struct sk_buff_head queue;
1597 u16 n;
1599 sl = NULL;
1600 h = (struct aoe_hdr *) skb_mac_header(skb);
1601 ch = (struct aoe_cfghdr *) (h+1);
1604 * Enough people have their dip switches set backwards to
1605 * warrant a loud message for this special case.
1607 aoemajor = get_unaligned_be16(&h->major);
1608 if (aoemajor == 0xfff) {
1609 printk(KERN_ERR "aoe: Warning: shelf address is all ones. "
1610 "Check shelf dip switches.\n");
1611 return;
1613 if (aoemajor == 0xffff) {
1614 pr_info("aoe: e%ld.%d: broadcast shelf number invalid\n",
1615 aoemajor, (int) h->minor);
1616 return;
1618 if (h->minor == 0xff) {
1619 pr_info("aoe: e%ld.%d: broadcast slot number invalid\n",
1620 aoemajor, (int) h->minor);
1621 return;
1624 n = be16_to_cpu(ch->bufcnt);
1625 if (n > aoe_maxout) /* keep it reasonable */
1626 n = aoe_maxout;
1628 d = aoedev_by_aoeaddr(aoemajor, h->minor, 1);
1629 if (d == NULL) {
1630 pr_info("aoe: device allocation failure\n");
1631 return;
1634 spin_lock_irqsave(&d->lock, flags);
1636 t = gettgt(d, h->src);
1637 if (t) {
1638 t->nframes = n;
1639 if (n < t->maxout)
1640 aoecmd_wreset(t);
1641 } else {
1642 t = addtgt(d, h->src, n);
1643 if (!t)
1644 goto bail;
1646 n = skb->dev->mtu;
1647 n -= sizeof(struct aoe_hdr) + sizeof(struct aoe_atahdr);
1648 n /= 512;
1649 if (n > ch->scnt)
1650 n = ch->scnt;
1651 n = n ? n * 512 : DEFAULTBCNT;
1652 setifbcnt(t, skb->dev, n);
1654 /* don't change users' perspective */
1655 if (d->nopen == 0) {
1656 d->fw_ver = be16_to_cpu(ch->fwver);
1657 sl = aoecmd_ata_id(d);
1659 bail:
1660 spin_unlock_irqrestore(&d->lock, flags);
1661 aoedev_put(d);
1662 if (sl) {
1663 __skb_queue_head_init(&queue);
1664 __skb_queue_tail(&queue, sl);
1665 aoenet_xmit(&queue);
1669 void
1670 aoecmd_wreset(struct aoetgt *t)
1672 t->maxout = 1;
1673 t->ssthresh = t->nframes / 2;
1674 t->next_cwnd = t->nframes;
1677 void
1678 aoecmd_cleanslate(struct aoedev *d)
1680 struct aoetgt **t, **te;
1682 d->rttavg = RTTAVG_INIT;
1683 d->rttdev = RTTDEV_INIT;
1684 d->maxbcnt = 0;
1686 t = d->targets;
1687 te = t + d->ntargets;
1688 for (; t < te && *t; t++)
1689 aoecmd_wreset(*t);
1692 void
1693 aoe_failbuf(struct aoedev *d, struct buf *buf)
1695 if (buf == NULL)
1696 return;
1697 buf->iter.bi_size = 0;
1698 clear_bit(BIO_UPTODATE, &buf->bio->bi_flags);
1699 if (buf->nframesout == 0)
1700 aoe_end_buf(d, buf);
1703 void
1704 aoe_flush_iocq(void)
1706 int i;
1708 for (i = 0; i < ncpus; i++) {
1709 if (kts[i].active)
1710 aoe_flush_iocq_by_index(i);
1714 void
1715 aoe_flush_iocq_by_index(int id)
1717 struct frame *f;
1718 struct aoedev *d;
1719 LIST_HEAD(flist);
1720 struct list_head *pos;
1721 struct sk_buff *skb;
1722 ulong flags;
1724 spin_lock_irqsave(&iocq[id].lock, flags);
1725 list_splice_init(&iocq[id].head, &flist);
1726 spin_unlock_irqrestore(&iocq[id].lock, flags);
1727 while (!list_empty(&flist)) {
1728 pos = flist.next;
1729 list_del(pos);
1730 f = list_entry(pos, struct frame, head);
1731 d = f->t->d;
1732 skb = f->r_skb;
1733 spin_lock_irqsave(&d->lock, flags);
1734 if (f->buf) {
1735 f->buf->nframesout--;
1736 aoe_failbuf(d, f->buf);
1738 aoe_freetframe(f);
1739 spin_unlock_irqrestore(&d->lock, flags);
1740 dev_kfree_skb(skb);
1741 aoedev_put(d);
1745 int __init
1746 aoecmd_init(void)
1748 void *p;
1749 int i;
1750 int ret;
1752 /* get_zeroed_page returns page with ref count 1 */
1753 p = (void *) get_zeroed_page(GFP_KERNEL | __GFP_REPEAT);
1754 if (!p)
1755 return -ENOMEM;
1756 empty_page = virt_to_page(p);
1758 ncpus = num_online_cpus();
1760 iocq = kcalloc(ncpus, sizeof(struct iocq_ktio), GFP_KERNEL);
1761 if (!iocq)
1762 return -ENOMEM;
1764 kts = kcalloc(ncpus, sizeof(struct ktstate), GFP_KERNEL);
1765 if (!kts) {
1766 ret = -ENOMEM;
1767 goto kts_fail;
1770 ktiowq = kcalloc(ncpus, sizeof(wait_queue_head_t), GFP_KERNEL);
1771 if (!ktiowq) {
1772 ret = -ENOMEM;
1773 goto ktiowq_fail;
1776 mutex_init(&ktio_spawn_lock);
1778 for (i = 0; i < ncpus; i++) {
1779 INIT_LIST_HEAD(&iocq[i].head);
1780 spin_lock_init(&iocq[i].lock);
1781 init_waitqueue_head(&ktiowq[i]);
1782 snprintf(kts[i].name, sizeof(kts[i].name), "aoe_ktio%d", i);
1783 kts[i].fn = ktio;
1784 kts[i].waitq = &ktiowq[i];
1785 kts[i].lock = &iocq[i].lock;
1786 kts[i].id = i;
1787 kts[i].active = 0;
1789 kts[0].active = 1;
1790 if (aoe_ktstart(&kts[0])) {
1791 ret = -ENOMEM;
1792 goto ktstart_fail;
1794 return 0;
1796 ktstart_fail:
1797 kfree(ktiowq);
1798 ktiowq_fail:
1799 kfree(kts);
1800 kts_fail:
1801 kfree(iocq);
1803 return ret;
1806 void
1807 aoecmd_exit(void)
1809 int i;
1811 for (i = 0; i < ncpus; i++)
1812 if (kts[i].active)
1813 aoe_ktstop(&kts[i]);
1815 aoe_flush_iocq();
1817 /* Free up the iocq and thread speicific configuration
1818 * allocated during startup.
1820 kfree(iocq);
1821 kfree(kts);
1822 kfree(ktiowq);
1824 free_page((unsigned long) page_address(empty_page));
1825 empty_page = NULL;