xen/netback: shutdown the ring if it contains garbage.
[~shefty/rdma-dev.git] / drivers / net / xen-netback / netback.c
1 /*
2  * Back-end of the driver for virtual network devices. This portion of the
3  * driver exports a 'unified' network-device interface that can be accessed
4  * by any operating system that implements a compatible front end. A
5  * reference front-end implementation can be found in:
6  *  drivers/net/xen-netfront.c
7  *
8  * Copyright (c) 2002-2005, K A Fraser
9  *
10  * This program is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU General Public License version 2
12  * as published by the Free Software Foundation; or, when distributed
13  * separately from the Linux kernel or incorporated into other
14  * software packages, subject to the following license:
15  *
16  * Permission is hereby granted, free of charge, to any person obtaining a copy
17  * of this source file (the "Software"), to deal in the Software without
18  * restriction, including without limitation the rights to use, copy, modify,
19  * merge, publish, distribute, sublicense, and/or sell copies of the Software,
20  * and to permit persons to whom the Software is furnished to do so, subject to
21  * the following conditions:
22  *
23  * The above copyright notice and this permission notice shall be included in
24  * all copies or substantial portions of the Software.
25  *
26  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
27  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
28  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
29  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
30  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
31  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
32  * IN THE SOFTWARE.
33  */
34
35 #include "common.h"
36
37 #include <linux/kthread.h>
38 #include <linux/if_vlan.h>
39 #include <linux/udp.h>
40
41 #include <net/tcp.h>
42
43 #include <xen/xen.h>
44 #include <xen/events.h>
45 #include <xen/interface/memory.h>
46
47 #include <asm/xen/hypercall.h>
48 #include <asm/xen/page.h>
49
50 struct pending_tx_info {
51         struct xen_netif_tx_request req;
52         struct xenvif *vif;
53 };
54 typedef unsigned int pending_ring_idx_t;
55
56 struct netbk_rx_meta {
57         int id;
58         int size;
59         int gso_size;
60 };
61
62 #define MAX_PENDING_REQS 256
63
64 /* Discriminate from any valid pending_idx value. */
65 #define INVALID_PENDING_IDX 0xFFFF
66
67 #define MAX_BUFFER_OFFSET PAGE_SIZE
68
69 /* extra field used in struct page */
70 union page_ext {
71         struct {
72 #if BITS_PER_LONG < 64
73 #define IDX_WIDTH   8
74 #define GROUP_WIDTH (BITS_PER_LONG - IDX_WIDTH)
75                 unsigned int group:GROUP_WIDTH;
76                 unsigned int idx:IDX_WIDTH;
77 #else
78                 unsigned int group, idx;
79 #endif
80         } e;
81         void *mapping;
82 };
83
84 struct xen_netbk {
85         wait_queue_head_t wq;
86         struct task_struct *task;
87
88         struct sk_buff_head rx_queue;
89         struct sk_buff_head tx_queue;
90
91         struct timer_list net_timer;
92
93         struct page *mmap_pages[MAX_PENDING_REQS];
94
95         pending_ring_idx_t pending_prod;
96         pending_ring_idx_t pending_cons;
97         struct list_head net_schedule_list;
98
99         /* Protect the net_schedule_list in netif. */
100         spinlock_t net_schedule_list_lock;
101
102         atomic_t netfront_count;
103
104         struct pending_tx_info pending_tx_info[MAX_PENDING_REQS];
105         struct gnttab_copy tx_copy_ops[MAX_PENDING_REQS];
106
107         u16 pending_ring[MAX_PENDING_REQS];
108
109         /*
110          * Given MAX_BUFFER_OFFSET of 4096 the worst case is that each
111          * head/fragment page uses 2 copy operations because it
112          * straddles two buffers in the frontend.
113          */
114         struct gnttab_copy grant_copy_op[2*XEN_NETIF_RX_RING_SIZE];
115         struct netbk_rx_meta meta[2*XEN_NETIF_RX_RING_SIZE];
116 };
117
118 static struct xen_netbk *xen_netbk;
119 static int xen_netbk_group_nr;
120
121 void xen_netbk_add_xenvif(struct xenvif *vif)
122 {
123         int i;
124         int min_netfront_count;
125         int min_group = 0;
126         struct xen_netbk *netbk;
127
128         min_netfront_count = atomic_read(&xen_netbk[0].netfront_count);
129         for (i = 0; i < xen_netbk_group_nr; i++) {
130                 int netfront_count = atomic_read(&xen_netbk[i].netfront_count);
131                 if (netfront_count < min_netfront_count) {
132                         min_group = i;
133                         min_netfront_count = netfront_count;
134                 }
135         }
136
137         netbk = &xen_netbk[min_group];
138
139         vif->netbk = netbk;
140         atomic_inc(&netbk->netfront_count);
141 }
142
143 void xen_netbk_remove_xenvif(struct xenvif *vif)
144 {
145         struct xen_netbk *netbk = vif->netbk;
146         vif->netbk = NULL;
147         atomic_dec(&netbk->netfront_count);
148 }
149
150 static void xen_netbk_idx_release(struct xen_netbk *netbk, u16 pending_idx);
151 static void make_tx_response(struct xenvif *vif,
152                              struct xen_netif_tx_request *txp,
153                              s8       st);
154 static struct xen_netif_rx_response *make_rx_response(struct xenvif *vif,
155                                              u16      id,
156                                              s8       st,
157                                              u16      offset,
158                                              u16      size,
159                                              u16      flags);
160
161 static inline unsigned long idx_to_pfn(struct xen_netbk *netbk,
162                                        u16 idx)
163 {
164         return page_to_pfn(netbk->mmap_pages[idx]);
165 }
166
167 static inline unsigned long idx_to_kaddr(struct xen_netbk *netbk,
168                                          u16 idx)
169 {
170         return (unsigned long)pfn_to_kaddr(idx_to_pfn(netbk, idx));
171 }
172
173 /* extra field used in struct page */
174 static inline void set_page_ext(struct page *pg, struct xen_netbk *netbk,
175                                 unsigned int idx)
176 {
177         unsigned int group = netbk - xen_netbk;
178         union page_ext ext = { .e = { .group = group + 1, .idx = idx } };
179
180         BUILD_BUG_ON(sizeof(ext) > sizeof(ext.mapping));
181         pg->mapping = ext.mapping;
182 }
183
184 static int get_page_ext(struct page *pg,
185                         unsigned int *pgroup, unsigned int *pidx)
186 {
187         union page_ext ext = { .mapping = pg->mapping };
188         struct xen_netbk *netbk;
189         unsigned int group, idx;
190
191         group = ext.e.group - 1;
192
193         if (group < 0 || group >= xen_netbk_group_nr)
194                 return 0;
195
196         netbk = &xen_netbk[group];
197
198         idx = ext.e.idx;
199
200         if ((idx < 0) || (idx >= MAX_PENDING_REQS))
201                 return 0;
202
203         if (netbk->mmap_pages[idx] != pg)
204                 return 0;
205
206         *pgroup = group;
207         *pidx = idx;
208
209         return 1;
210 }
211
212 /*
213  * This is the amount of packet we copy rather than map, so that the
214  * guest can't fiddle with the contents of the headers while we do
215  * packet processing on them (netfilter, routing, etc).
216  */
217 #define PKT_PROT_LEN    (ETH_HLEN + \
218                          VLAN_HLEN + \
219                          sizeof(struct iphdr) + MAX_IPOPTLEN + \
220                          sizeof(struct tcphdr) + MAX_TCP_OPTION_SPACE)
221
222 static u16 frag_get_pending_idx(skb_frag_t *frag)
223 {
224         return (u16)frag->page_offset;
225 }
226
227 static void frag_set_pending_idx(skb_frag_t *frag, u16 pending_idx)
228 {
229         frag->page_offset = pending_idx;
230 }
231
232 static inline pending_ring_idx_t pending_index(unsigned i)
233 {
234         return i & (MAX_PENDING_REQS-1);
235 }
236
237 static inline pending_ring_idx_t nr_pending_reqs(struct xen_netbk *netbk)
238 {
239         return MAX_PENDING_REQS -
240                 netbk->pending_prod + netbk->pending_cons;
241 }
242
243 static void xen_netbk_kick_thread(struct xen_netbk *netbk)
244 {
245         wake_up(&netbk->wq);
246 }
247
248 static int max_required_rx_slots(struct xenvif *vif)
249 {
250         int max = DIV_ROUND_UP(vif->dev->mtu, PAGE_SIZE);
251
252         if (vif->can_sg || vif->gso || vif->gso_prefix)
253                 max += MAX_SKB_FRAGS + 1; /* extra_info + frags */
254
255         return max;
256 }
257
258 int xen_netbk_rx_ring_full(struct xenvif *vif)
259 {
260         RING_IDX peek   = vif->rx_req_cons_peek;
261         RING_IDX needed = max_required_rx_slots(vif);
262
263         return ((vif->rx.sring->req_prod - peek) < needed) ||
264                ((vif->rx.rsp_prod_pvt + XEN_NETIF_RX_RING_SIZE - peek) < needed);
265 }
266
267 int xen_netbk_must_stop_queue(struct xenvif *vif)
268 {
269         if (!xen_netbk_rx_ring_full(vif))
270                 return 0;
271
272         vif->rx.sring->req_event = vif->rx_req_cons_peek +
273                 max_required_rx_slots(vif);
274         mb(); /* request notification /then/ check the queue */
275
276         return xen_netbk_rx_ring_full(vif);
277 }
278
279 /*
280  * Returns true if we should start a new receive buffer instead of
281  * adding 'size' bytes to a buffer which currently contains 'offset'
282  * bytes.
283  */
284 static bool start_new_rx_buffer(int offset, unsigned long size, int head)
285 {
286         /* simple case: we have completely filled the current buffer. */
287         if (offset == MAX_BUFFER_OFFSET)
288                 return true;
289
290         /*
291          * complex case: start a fresh buffer if the current frag
292          * would overflow the current buffer but only if:
293          *     (i)   this frag would fit completely in the next buffer
294          * and (ii)  there is already some data in the current buffer
295          * and (iii) this is not the head buffer.
296          *
297          * Where:
298          * - (i) stops us splitting a frag into two copies
299          *   unless the frag is too large for a single buffer.
300          * - (ii) stops us from leaving a buffer pointlessly empty.
301          * - (iii) stops us leaving the first buffer
302          *   empty. Strictly speaking this is already covered
303          *   by (ii) but is explicitly checked because
304          *   netfront relies on the first buffer being
305          *   non-empty and can crash otherwise.
306          *
307          * This means we will effectively linearise small
308          * frags but do not needlessly split large buffers
309          * into multiple copies tend to give large frags their
310          * own buffers as before.
311          */
312         if ((offset + size > MAX_BUFFER_OFFSET) &&
313             (size <= MAX_BUFFER_OFFSET) && offset && !head)
314                 return true;
315
316         return false;
317 }
318
319 /*
320  * Figure out how many ring slots we're going to need to send @skb to
321  * the guest. This function is essentially a dry run of
322  * netbk_gop_frag_copy.
323  */
324 unsigned int xen_netbk_count_skb_slots(struct xenvif *vif, struct sk_buff *skb)
325 {
326         unsigned int count;
327         int i, copy_off;
328
329         count = DIV_ROUND_UP(skb_headlen(skb), PAGE_SIZE);
330
331         copy_off = skb_headlen(skb) % PAGE_SIZE;
332
333         if (skb_shinfo(skb)->gso_size)
334                 count++;
335
336         for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
337                 unsigned long size = skb_frag_size(&skb_shinfo(skb)->frags[i]);
338                 unsigned long offset = skb_shinfo(skb)->frags[i].page_offset;
339                 unsigned long bytes;
340
341                 offset &= ~PAGE_MASK;
342
343                 while (size > 0) {
344                         BUG_ON(offset >= PAGE_SIZE);
345                         BUG_ON(copy_off > MAX_BUFFER_OFFSET);
346
347                         bytes = PAGE_SIZE - offset;
348
349                         if (bytes > size)
350                                 bytes = size;
351
352                         if (start_new_rx_buffer(copy_off, bytes, 0)) {
353                                 count++;
354                                 copy_off = 0;
355                         }
356
357                         if (copy_off + bytes > MAX_BUFFER_OFFSET)
358                                 bytes = MAX_BUFFER_OFFSET - copy_off;
359
360                         copy_off += bytes;
361
362                         offset += bytes;
363                         size -= bytes;
364
365                         if (offset == PAGE_SIZE)
366                                 offset = 0;
367                 }
368         }
369         return count;
370 }
371
372 struct netrx_pending_operations {
373         unsigned copy_prod, copy_cons;
374         unsigned meta_prod, meta_cons;
375         struct gnttab_copy *copy;
376         struct netbk_rx_meta *meta;
377         int copy_off;
378         grant_ref_t copy_gref;
379 };
380
381 static struct netbk_rx_meta *get_next_rx_buffer(struct xenvif *vif,
382                                                 struct netrx_pending_operations *npo)
383 {
384         struct netbk_rx_meta *meta;
385         struct xen_netif_rx_request *req;
386
387         req = RING_GET_REQUEST(&vif->rx, vif->rx.req_cons++);
388
389         meta = npo->meta + npo->meta_prod++;
390         meta->gso_size = 0;
391         meta->size = 0;
392         meta->id = req->id;
393
394         npo->copy_off = 0;
395         npo->copy_gref = req->gref;
396
397         return meta;
398 }
399
400 /*
401  * Set up the grant operations for this fragment. If it's a flipping
402  * interface, we also set up the unmap request from here.
403  */
404 static void netbk_gop_frag_copy(struct xenvif *vif, struct sk_buff *skb,
405                                 struct netrx_pending_operations *npo,
406                                 struct page *page, unsigned long size,
407                                 unsigned long offset, int *head)
408 {
409         struct gnttab_copy *copy_gop;
410         struct netbk_rx_meta *meta;
411         /*
412          * These variables are used iff get_page_ext returns true,
413          * in which case they are guaranteed to be initialized.
414          */
415         unsigned int uninitialized_var(group), uninitialized_var(idx);
416         int foreign = get_page_ext(page, &group, &idx);
417         unsigned long bytes;
418
419         /* Data must not cross a page boundary. */
420         BUG_ON(size + offset > PAGE_SIZE<<compound_order(page));
421
422         meta = npo->meta + npo->meta_prod - 1;
423
424         /* Skip unused frames from start of page */
425         page += offset >> PAGE_SHIFT;
426         offset &= ~PAGE_MASK;
427
428         while (size > 0) {
429                 BUG_ON(offset >= PAGE_SIZE);
430                 BUG_ON(npo->copy_off > MAX_BUFFER_OFFSET);
431
432                 bytes = PAGE_SIZE - offset;
433
434                 if (bytes > size)
435                         bytes = size;
436
437                 if (start_new_rx_buffer(npo->copy_off, bytes, *head)) {
438                         /*
439                          * Netfront requires there to be some data in the head
440                          * buffer.
441                          */
442                         BUG_ON(*head);
443
444                         meta = get_next_rx_buffer(vif, npo);
445                 }
446
447                 if (npo->copy_off + bytes > MAX_BUFFER_OFFSET)
448                         bytes = MAX_BUFFER_OFFSET - npo->copy_off;
449
450                 copy_gop = npo->copy + npo->copy_prod++;
451                 copy_gop->flags = GNTCOPY_dest_gref;
452                 if (foreign) {
453                         struct xen_netbk *netbk = &xen_netbk[group];
454                         struct pending_tx_info *src_pend;
455
456                         src_pend = &netbk->pending_tx_info[idx];
457
458                         copy_gop->source.domid = src_pend->vif->domid;
459                         copy_gop->source.u.ref = src_pend->req.gref;
460                         copy_gop->flags |= GNTCOPY_source_gref;
461                 } else {
462                         void *vaddr = page_address(page);
463                         copy_gop->source.domid = DOMID_SELF;
464                         copy_gop->source.u.gmfn = virt_to_mfn(vaddr);
465                 }
466                 copy_gop->source.offset = offset;
467                 copy_gop->dest.domid = vif->domid;
468
469                 copy_gop->dest.offset = npo->copy_off;
470                 copy_gop->dest.u.ref = npo->copy_gref;
471                 copy_gop->len = bytes;
472
473                 npo->copy_off += bytes;
474                 meta->size += bytes;
475
476                 offset += bytes;
477                 size -= bytes;
478
479                 /* Next frame */
480                 if (offset == PAGE_SIZE && size) {
481                         BUG_ON(!PageCompound(page));
482                         page++;
483                         offset = 0;
484                 }
485
486                 /* Leave a gap for the GSO descriptor. */
487                 if (*head && skb_shinfo(skb)->gso_size && !vif->gso_prefix)
488                         vif->rx.req_cons++;
489
490                 *head = 0; /* There must be something in this buffer now. */
491
492         }
493 }
494
495 /*
496  * Prepare an SKB to be transmitted to the frontend.
497  *
498  * This function is responsible for allocating grant operations, meta
499  * structures, etc.
500  *
501  * It returns the number of meta structures consumed. The number of
502  * ring slots used is always equal to the number of meta slots used
503  * plus the number of GSO descriptors used. Currently, we use either
504  * zero GSO descriptors (for non-GSO packets) or one descriptor (for
505  * frontend-side LRO).
506  */
507 static int netbk_gop_skb(struct sk_buff *skb,
508                          struct netrx_pending_operations *npo)
509 {
510         struct xenvif *vif = netdev_priv(skb->dev);
511         int nr_frags = skb_shinfo(skb)->nr_frags;
512         int i;
513         struct xen_netif_rx_request *req;
514         struct netbk_rx_meta *meta;
515         unsigned char *data;
516         int head = 1;
517         int old_meta_prod;
518
519         old_meta_prod = npo->meta_prod;
520
521         /* Set up a GSO prefix descriptor, if necessary */
522         if (skb_shinfo(skb)->gso_size && vif->gso_prefix) {
523                 req = RING_GET_REQUEST(&vif->rx, vif->rx.req_cons++);
524                 meta = npo->meta + npo->meta_prod++;
525                 meta->gso_size = skb_shinfo(skb)->gso_size;
526                 meta->size = 0;
527                 meta->id = req->id;
528         }
529
530         req = RING_GET_REQUEST(&vif->rx, vif->rx.req_cons++);
531         meta = npo->meta + npo->meta_prod++;
532
533         if (!vif->gso_prefix)
534                 meta->gso_size = skb_shinfo(skb)->gso_size;
535         else
536                 meta->gso_size = 0;
537
538         meta->size = 0;
539         meta->id = req->id;
540         npo->copy_off = 0;
541         npo->copy_gref = req->gref;
542
543         data = skb->data;
544         while (data < skb_tail_pointer(skb)) {
545                 unsigned int offset = offset_in_page(data);
546                 unsigned int len = PAGE_SIZE - offset;
547
548                 if (data + len > skb_tail_pointer(skb))
549                         len = skb_tail_pointer(skb) - data;
550
551                 netbk_gop_frag_copy(vif, skb, npo,
552                                     virt_to_page(data), len, offset, &head);
553                 data += len;
554         }
555
556         for (i = 0; i < nr_frags; i++) {
557                 netbk_gop_frag_copy(vif, skb, npo,
558                                     skb_frag_page(&skb_shinfo(skb)->frags[i]),
559                                     skb_frag_size(&skb_shinfo(skb)->frags[i]),
560                                     skb_shinfo(skb)->frags[i].page_offset,
561                                     &head);
562         }
563
564         return npo->meta_prod - old_meta_prod;
565 }
566
567 /*
568  * This is a twin to netbk_gop_skb.  Assume that netbk_gop_skb was
569  * used to set up the operations on the top of
570  * netrx_pending_operations, which have since been done.  Check that
571  * they didn't give any errors and advance over them.
572  */
573 static int netbk_check_gop(struct xenvif *vif, int nr_meta_slots,
574                            struct netrx_pending_operations *npo)
575 {
576         struct gnttab_copy     *copy_op;
577         int status = XEN_NETIF_RSP_OKAY;
578         int i;
579
580         for (i = 0; i < nr_meta_slots; i++) {
581                 copy_op = npo->copy + npo->copy_cons++;
582                 if (copy_op->status != GNTST_okay) {
583                         netdev_dbg(vif->dev,
584                                    "Bad status %d from copy to DOM%d.\n",
585                                    copy_op->status, vif->domid);
586                         status = XEN_NETIF_RSP_ERROR;
587                 }
588         }
589
590         return status;
591 }
592
593 static void netbk_add_frag_responses(struct xenvif *vif, int status,
594                                      struct netbk_rx_meta *meta,
595                                      int nr_meta_slots)
596 {
597         int i;
598         unsigned long offset;
599
600         /* No fragments used */
601         if (nr_meta_slots <= 1)
602                 return;
603
604         nr_meta_slots--;
605
606         for (i = 0; i < nr_meta_slots; i++) {
607                 int flags;
608                 if (i == nr_meta_slots - 1)
609                         flags = 0;
610                 else
611                         flags = XEN_NETRXF_more_data;
612
613                 offset = 0;
614                 make_rx_response(vif, meta[i].id, status, offset,
615                                  meta[i].size, flags);
616         }
617 }
618
619 struct skb_cb_overlay {
620         int meta_slots_used;
621 };
622
623 static void xen_netbk_rx_action(struct xen_netbk *netbk)
624 {
625         struct xenvif *vif = NULL, *tmp;
626         s8 status;
627         u16 irq, flags;
628         struct xen_netif_rx_response *resp;
629         struct sk_buff_head rxq;
630         struct sk_buff *skb;
631         LIST_HEAD(notify);
632         int ret;
633         int nr_frags;
634         int count;
635         unsigned long offset;
636         struct skb_cb_overlay *sco;
637
638         struct netrx_pending_operations npo = {
639                 .copy  = netbk->grant_copy_op,
640                 .meta  = netbk->meta,
641         };
642
643         skb_queue_head_init(&rxq);
644
645         count = 0;
646
647         while ((skb = skb_dequeue(&netbk->rx_queue)) != NULL) {
648                 vif = netdev_priv(skb->dev);
649                 nr_frags = skb_shinfo(skb)->nr_frags;
650
651                 sco = (struct skb_cb_overlay *)skb->cb;
652                 sco->meta_slots_used = netbk_gop_skb(skb, &npo);
653
654                 count += nr_frags + 1;
655
656                 __skb_queue_tail(&rxq, skb);
657
658                 /* Filled the batch queue? */
659                 if (count + MAX_SKB_FRAGS >= XEN_NETIF_RX_RING_SIZE)
660                         break;
661         }
662
663         BUG_ON(npo.meta_prod > ARRAY_SIZE(netbk->meta));
664
665         if (!npo.copy_prod)
666                 return;
667
668         BUG_ON(npo.copy_prod > ARRAY_SIZE(netbk->grant_copy_op));
669         gnttab_batch_copy(netbk->grant_copy_op, npo.copy_prod);
670
671         while ((skb = __skb_dequeue(&rxq)) != NULL) {
672                 sco = (struct skb_cb_overlay *)skb->cb;
673
674                 vif = netdev_priv(skb->dev);
675
676                 if (netbk->meta[npo.meta_cons].gso_size && vif->gso_prefix) {
677                         resp = RING_GET_RESPONSE(&vif->rx,
678                                                 vif->rx.rsp_prod_pvt++);
679
680                         resp->flags = XEN_NETRXF_gso_prefix | XEN_NETRXF_more_data;
681
682                         resp->offset = netbk->meta[npo.meta_cons].gso_size;
683                         resp->id = netbk->meta[npo.meta_cons].id;
684                         resp->status = sco->meta_slots_used;
685
686                         npo.meta_cons++;
687                         sco->meta_slots_used--;
688                 }
689
690
691                 vif->dev->stats.tx_bytes += skb->len;
692                 vif->dev->stats.tx_packets++;
693
694                 status = netbk_check_gop(vif, sco->meta_slots_used, &npo);
695
696                 if (sco->meta_slots_used == 1)
697                         flags = 0;
698                 else
699                         flags = XEN_NETRXF_more_data;
700
701                 if (skb->ip_summed == CHECKSUM_PARTIAL) /* local packet? */
702                         flags |= XEN_NETRXF_csum_blank | XEN_NETRXF_data_validated;
703                 else if (skb->ip_summed == CHECKSUM_UNNECESSARY)
704                         /* remote but checksummed. */
705                         flags |= XEN_NETRXF_data_validated;
706
707                 offset = 0;
708                 resp = make_rx_response(vif, netbk->meta[npo.meta_cons].id,
709                                         status, offset,
710                                         netbk->meta[npo.meta_cons].size,
711                                         flags);
712
713                 if (netbk->meta[npo.meta_cons].gso_size && !vif->gso_prefix) {
714                         struct xen_netif_extra_info *gso =
715                                 (struct xen_netif_extra_info *)
716                                 RING_GET_RESPONSE(&vif->rx,
717                                                   vif->rx.rsp_prod_pvt++);
718
719                         resp->flags |= XEN_NETRXF_extra_info;
720
721                         gso->u.gso.size = netbk->meta[npo.meta_cons].gso_size;
722                         gso->u.gso.type = XEN_NETIF_GSO_TYPE_TCPV4;
723                         gso->u.gso.pad = 0;
724                         gso->u.gso.features = 0;
725
726                         gso->type = XEN_NETIF_EXTRA_TYPE_GSO;
727                         gso->flags = 0;
728                 }
729
730                 netbk_add_frag_responses(vif, status,
731                                          netbk->meta + npo.meta_cons + 1,
732                                          sco->meta_slots_used);
733
734                 RING_PUSH_RESPONSES_AND_CHECK_NOTIFY(&vif->rx, ret);
735                 irq = vif->irq;
736                 if (ret && list_empty(&vif->notify_list))
737                         list_add_tail(&vif->notify_list, &notify);
738
739                 xenvif_notify_tx_completion(vif);
740
741                 xenvif_put(vif);
742                 npo.meta_cons += sco->meta_slots_used;
743                 dev_kfree_skb(skb);
744         }
745
746         list_for_each_entry_safe(vif, tmp, &notify, notify_list) {
747                 notify_remote_via_irq(vif->irq);
748                 list_del_init(&vif->notify_list);
749         }
750
751         /* More work to do? */
752         if (!skb_queue_empty(&netbk->rx_queue) &&
753                         !timer_pending(&netbk->net_timer))
754                 xen_netbk_kick_thread(netbk);
755 }
756
757 void xen_netbk_queue_tx_skb(struct xenvif *vif, struct sk_buff *skb)
758 {
759         struct xen_netbk *netbk = vif->netbk;
760
761         skb_queue_tail(&netbk->rx_queue, skb);
762
763         xen_netbk_kick_thread(netbk);
764 }
765
766 static void xen_netbk_alarm(unsigned long data)
767 {
768         struct xen_netbk *netbk = (struct xen_netbk *)data;
769         xen_netbk_kick_thread(netbk);
770 }
771
772 static int __on_net_schedule_list(struct xenvif *vif)
773 {
774         return !list_empty(&vif->schedule_list);
775 }
776
777 /* Must be called with net_schedule_list_lock held */
778 static void remove_from_net_schedule_list(struct xenvif *vif)
779 {
780         if (likely(__on_net_schedule_list(vif))) {
781                 list_del_init(&vif->schedule_list);
782                 xenvif_put(vif);
783         }
784 }
785
786 static struct xenvif *poll_net_schedule_list(struct xen_netbk *netbk)
787 {
788         struct xenvif *vif = NULL;
789
790         spin_lock_irq(&netbk->net_schedule_list_lock);
791         if (list_empty(&netbk->net_schedule_list))
792                 goto out;
793
794         vif = list_first_entry(&netbk->net_schedule_list,
795                                struct xenvif, schedule_list);
796         if (!vif)
797                 goto out;
798
799         xenvif_get(vif);
800
801         remove_from_net_schedule_list(vif);
802 out:
803         spin_unlock_irq(&netbk->net_schedule_list_lock);
804         return vif;
805 }
806
807 void xen_netbk_schedule_xenvif(struct xenvif *vif)
808 {
809         unsigned long flags;
810         struct xen_netbk *netbk = vif->netbk;
811
812         if (__on_net_schedule_list(vif))
813                 goto kick;
814
815         spin_lock_irqsave(&netbk->net_schedule_list_lock, flags);
816         if (!__on_net_schedule_list(vif) &&
817             likely(xenvif_schedulable(vif))) {
818                 list_add_tail(&vif->schedule_list, &netbk->net_schedule_list);
819                 xenvif_get(vif);
820         }
821         spin_unlock_irqrestore(&netbk->net_schedule_list_lock, flags);
822
823 kick:
824         smp_mb();
825         if ((nr_pending_reqs(netbk) < (MAX_PENDING_REQS/2)) &&
826             !list_empty(&netbk->net_schedule_list))
827                 xen_netbk_kick_thread(netbk);
828 }
829
830 void xen_netbk_deschedule_xenvif(struct xenvif *vif)
831 {
832         struct xen_netbk *netbk = vif->netbk;
833         spin_lock_irq(&netbk->net_schedule_list_lock);
834         remove_from_net_schedule_list(vif);
835         spin_unlock_irq(&netbk->net_schedule_list_lock);
836 }
837
838 void xen_netbk_check_rx_xenvif(struct xenvif *vif)
839 {
840         int more_to_do;
841
842         RING_FINAL_CHECK_FOR_REQUESTS(&vif->tx, more_to_do);
843
844         if (more_to_do)
845                 xen_netbk_schedule_xenvif(vif);
846 }
847
848 static void tx_add_credit(struct xenvif *vif)
849 {
850         unsigned long max_burst, max_credit;
851
852         /*
853          * Allow a burst big enough to transmit a jumbo packet of up to 128kB.
854          * Otherwise the interface can seize up due to insufficient credit.
855          */
856         max_burst = RING_GET_REQUEST(&vif->tx, vif->tx.req_cons)->size;
857         max_burst = min(max_burst, 131072UL);
858         max_burst = max(max_burst, vif->credit_bytes);
859
860         /* Take care that adding a new chunk of credit doesn't wrap to zero. */
861         max_credit = vif->remaining_credit + vif->credit_bytes;
862         if (max_credit < vif->remaining_credit)
863                 max_credit = ULONG_MAX; /* wrapped: clamp to ULONG_MAX */
864
865         vif->remaining_credit = min(max_credit, max_burst);
866 }
867
868 static void tx_credit_callback(unsigned long data)
869 {
870         struct xenvif *vif = (struct xenvif *)data;
871         tx_add_credit(vif);
872         xen_netbk_check_rx_xenvif(vif);
873 }
874
875 static void netbk_tx_err(struct xenvif *vif,
876                          struct xen_netif_tx_request *txp, RING_IDX end)
877 {
878         RING_IDX cons = vif->tx.req_cons;
879
880         do {
881                 make_tx_response(vif, txp, XEN_NETIF_RSP_ERROR);
882                 if (cons >= end)
883                         break;
884                 txp = RING_GET_REQUEST(&vif->tx, cons++);
885         } while (1);
886         vif->tx.req_cons = cons;
887         xen_netbk_check_rx_xenvif(vif);
888         xenvif_put(vif);
889 }
890
891 static void netbk_fatal_tx_err(struct xenvif *vif)
892 {
893         netdev_err(vif->dev, "fatal error; disabling device\n");
894         xenvif_carrier_off(vif);
895         xenvif_put(vif);
896 }
897
898 static int netbk_count_requests(struct xenvif *vif,
899                                 struct xen_netif_tx_request *first,
900                                 struct xen_netif_tx_request *txp,
901                                 int work_to_do)
902 {
903         RING_IDX cons = vif->tx.req_cons;
904         int frags = 0;
905
906         if (!(first->flags & XEN_NETTXF_more_data))
907                 return 0;
908
909         do {
910                 if (frags >= work_to_do) {
911                         netdev_err(vif->dev, "Need more frags\n");
912                         netbk_fatal_tx_err(vif);
913                         return -frags;
914                 }
915
916                 if (unlikely(frags >= MAX_SKB_FRAGS)) {
917                         netdev_err(vif->dev, "Too many frags\n");
918                         netbk_fatal_tx_err(vif);
919                         return -frags;
920                 }
921
922                 memcpy(txp, RING_GET_REQUEST(&vif->tx, cons + frags),
923                        sizeof(*txp));
924                 if (txp->size > first->size) {
925                         netdev_err(vif->dev, "Frag is bigger than frame.\n");
926                         netbk_fatal_tx_err(vif);
927                         return -frags;
928                 }
929
930                 first->size -= txp->size;
931                 frags++;
932
933                 if (unlikely((txp->offset + txp->size) > PAGE_SIZE)) {
934                         netdev_err(vif->dev, "txp->offset: %x, size: %u\n",
935                                  txp->offset, txp->size);
936                         netbk_fatal_tx_err(vif);
937                         return -frags;
938                 }
939         } while ((txp++)->flags & XEN_NETTXF_more_data);
940         return frags;
941 }
942
943 static struct page *xen_netbk_alloc_page(struct xen_netbk *netbk,
944                                          struct sk_buff *skb,
945                                          u16 pending_idx)
946 {
947         struct page *page;
948         page = alloc_page(GFP_KERNEL|__GFP_COLD);
949         if (!page)
950                 return NULL;
951         set_page_ext(page, netbk, pending_idx);
952         netbk->mmap_pages[pending_idx] = page;
953         return page;
954 }
955
956 static struct gnttab_copy *xen_netbk_get_requests(struct xen_netbk *netbk,
957                                                   struct xenvif *vif,
958                                                   struct sk_buff *skb,
959                                                   struct xen_netif_tx_request *txp,
960                                                   struct gnttab_copy *gop)
961 {
962         struct skb_shared_info *shinfo = skb_shinfo(skb);
963         skb_frag_t *frags = shinfo->frags;
964         u16 pending_idx = *((u16 *)skb->data);
965         int i, start;
966
967         /* Skip first skb fragment if it is on same page as header fragment. */
968         start = (frag_get_pending_idx(&shinfo->frags[0]) == pending_idx);
969
970         for (i = start; i < shinfo->nr_frags; i++, txp++) {
971                 struct page *page;
972                 pending_ring_idx_t index;
973                 struct pending_tx_info *pending_tx_info =
974                         netbk->pending_tx_info;
975
976                 index = pending_index(netbk->pending_cons++);
977                 pending_idx = netbk->pending_ring[index];
978                 page = xen_netbk_alloc_page(netbk, skb, pending_idx);
979                 if (!page)
980                         return NULL;
981
982                 gop->source.u.ref = txp->gref;
983                 gop->source.domid = vif->domid;
984                 gop->source.offset = txp->offset;
985
986                 gop->dest.u.gmfn = virt_to_mfn(page_address(page));
987                 gop->dest.domid = DOMID_SELF;
988                 gop->dest.offset = txp->offset;
989
990                 gop->len = txp->size;
991                 gop->flags = GNTCOPY_source_gref;
992
993                 gop++;
994
995                 memcpy(&pending_tx_info[pending_idx].req, txp, sizeof(*txp));
996                 xenvif_get(vif);
997                 pending_tx_info[pending_idx].vif = vif;
998                 frag_set_pending_idx(&frags[i], pending_idx);
999         }
1000
1001         return gop;
1002 }
1003
1004 static int xen_netbk_tx_check_gop(struct xen_netbk *netbk,
1005                                   struct sk_buff *skb,
1006                                   struct gnttab_copy **gopp)
1007 {
1008         struct gnttab_copy *gop = *gopp;
1009         u16 pending_idx = *((u16 *)skb->data);
1010         struct pending_tx_info *pending_tx_info = netbk->pending_tx_info;
1011         struct xenvif *vif = pending_tx_info[pending_idx].vif;
1012         struct xen_netif_tx_request *txp;
1013         struct skb_shared_info *shinfo = skb_shinfo(skb);
1014         int nr_frags = shinfo->nr_frags;
1015         int i, err, start;
1016
1017         /* Check status of header. */
1018         err = gop->status;
1019         if (unlikely(err)) {
1020                 pending_ring_idx_t index;
1021                 index = pending_index(netbk->pending_prod++);
1022                 txp = &pending_tx_info[pending_idx].req;
1023                 make_tx_response(vif, txp, XEN_NETIF_RSP_ERROR);
1024                 netbk->pending_ring[index] = pending_idx;
1025                 xenvif_put(vif);
1026         }
1027
1028         /* Skip first skb fragment if it is on same page as header fragment. */
1029         start = (frag_get_pending_idx(&shinfo->frags[0]) == pending_idx);
1030
1031         for (i = start; i < nr_frags; i++) {
1032                 int j, newerr;
1033                 pending_ring_idx_t index;
1034
1035                 pending_idx = frag_get_pending_idx(&shinfo->frags[i]);
1036
1037                 /* Check error status: if okay then remember grant handle. */
1038                 newerr = (++gop)->status;
1039                 if (likely(!newerr)) {
1040                         /* Had a previous error? Invalidate this fragment. */
1041                         if (unlikely(err))
1042                                 xen_netbk_idx_release(netbk, pending_idx);
1043                         continue;
1044                 }
1045
1046                 /* Error on this fragment: respond to client with an error. */
1047                 txp = &netbk->pending_tx_info[pending_idx].req;
1048                 make_tx_response(vif, txp, XEN_NETIF_RSP_ERROR);
1049                 index = pending_index(netbk->pending_prod++);
1050                 netbk->pending_ring[index] = pending_idx;
1051                 xenvif_put(vif);
1052
1053                 /* Not the first error? Preceding frags already invalidated. */
1054                 if (err)
1055                         continue;
1056
1057                 /* First error: invalidate header and preceding fragments. */
1058                 pending_idx = *((u16 *)skb->data);
1059                 xen_netbk_idx_release(netbk, pending_idx);
1060                 for (j = start; j < i; j++) {
1061                         pending_idx = frag_get_pending_idx(&shinfo->frags[j]);
1062                         xen_netbk_idx_release(netbk, pending_idx);
1063                 }
1064
1065                 /* Remember the error: invalidate all subsequent fragments. */
1066                 err = newerr;
1067         }
1068
1069         *gopp = gop + 1;
1070         return err;
1071 }
1072
1073 static void xen_netbk_fill_frags(struct xen_netbk *netbk, struct sk_buff *skb)
1074 {
1075         struct skb_shared_info *shinfo = skb_shinfo(skb);
1076         int nr_frags = shinfo->nr_frags;
1077         int i;
1078
1079         for (i = 0; i < nr_frags; i++) {
1080                 skb_frag_t *frag = shinfo->frags + i;
1081                 struct xen_netif_tx_request *txp;
1082                 struct page *page;
1083                 u16 pending_idx;
1084
1085                 pending_idx = frag_get_pending_idx(frag);
1086
1087                 txp = &netbk->pending_tx_info[pending_idx].req;
1088                 page = virt_to_page(idx_to_kaddr(netbk, pending_idx));
1089                 __skb_fill_page_desc(skb, i, page, txp->offset, txp->size);
1090                 skb->len += txp->size;
1091                 skb->data_len += txp->size;
1092                 skb->truesize += txp->size;
1093
1094                 /* Take an extra reference to offset xen_netbk_idx_release */
1095                 get_page(netbk->mmap_pages[pending_idx]);
1096                 xen_netbk_idx_release(netbk, pending_idx);
1097         }
1098 }
1099
1100 static int xen_netbk_get_extras(struct xenvif *vif,
1101                                 struct xen_netif_extra_info *extras,
1102                                 int work_to_do)
1103 {
1104         struct xen_netif_extra_info extra;
1105         RING_IDX cons = vif->tx.req_cons;
1106
1107         do {
1108                 if (unlikely(work_to_do-- <= 0)) {
1109                         netdev_err(vif->dev, "Missing extra info\n");
1110                         netbk_fatal_tx_err(vif);
1111                         return -EBADR;
1112                 }
1113
1114                 memcpy(&extra, RING_GET_REQUEST(&vif->tx, cons),
1115                        sizeof(extra));
1116                 if (unlikely(!extra.type ||
1117                              extra.type >= XEN_NETIF_EXTRA_TYPE_MAX)) {
1118                         vif->tx.req_cons = ++cons;
1119                         netdev_err(vif->dev,
1120                                    "Invalid extra type: %d\n", extra.type);
1121                         netbk_fatal_tx_err(vif);
1122                         return -EINVAL;
1123                 }
1124
1125                 memcpy(&extras[extra.type - 1], &extra, sizeof(extra));
1126                 vif->tx.req_cons = ++cons;
1127         } while (extra.flags & XEN_NETIF_EXTRA_FLAG_MORE);
1128
1129         return work_to_do;
1130 }
1131
1132 static int netbk_set_skb_gso(struct xenvif *vif,
1133                              struct sk_buff *skb,
1134                              struct xen_netif_extra_info *gso)
1135 {
1136         if (!gso->u.gso.size) {
1137                 netdev_err(vif->dev, "GSO size must not be zero.\n");
1138                 netbk_fatal_tx_err(vif);
1139                 return -EINVAL;
1140         }
1141
1142         /* Currently only TCPv4 S.O. is supported. */
1143         if (gso->u.gso.type != XEN_NETIF_GSO_TYPE_TCPV4) {
1144                 netdev_err(vif->dev, "Bad GSO type %d.\n", gso->u.gso.type);
1145                 netbk_fatal_tx_err(vif);
1146                 return -EINVAL;
1147         }
1148
1149         skb_shinfo(skb)->gso_size = gso->u.gso.size;
1150         skb_shinfo(skb)->gso_type = SKB_GSO_TCPV4;
1151
1152         /* Header must be checked, and gso_segs computed. */
1153         skb_shinfo(skb)->gso_type |= SKB_GSO_DODGY;
1154         skb_shinfo(skb)->gso_segs = 0;
1155
1156         return 0;
1157 }
1158
1159 static int checksum_setup(struct xenvif *vif, struct sk_buff *skb)
1160 {
1161         struct iphdr *iph;
1162         unsigned char *th;
1163         int err = -EPROTO;
1164         int recalculate_partial_csum = 0;
1165
1166         /*
1167          * A GSO SKB must be CHECKSUM_PARTIAL. However some buggy
1168          * peers can fail to set NETRXF_csum_blank when sending a GSO
1169          * frame. In this case force the SKB to CHECKSUM_PARTIAL and
1170          * recalculate the partial checksum.
1171          */
1172         if (skb->ip_summed != CHECKSUM_PARTIAL && skb_is_gso(skb)) {
1173                 vif->rx_gso_checksum_fixup++;
1174                 skb->ip_summed = CHECKSUM_PARTIAL;
1175                 recalculate_partial_csum = 1;
1176         }
1177
1178         /* A non-CHECKSUM_PARTIAL SKB does not require setup. */
1179         if (skb->ip_summed != CHECKSUM_PARTIAL)
1180                 return 0;
1181
1182         if (skb->protocol != htons(ETH_P_IP))
1183                 goto out;
1184
1185         iph = (void *)skb->data;
1186         th = skb->data + 4 * iph->ihl;
1187         if (th >= skb_tail_pointer(skb))
1188                 goto out;
1189
1190         skb->csum_start = th - skb->head;
1191         switch (iph->protocol) {
1192         case IPPROTO_TCP:
1193                 skb->csum_offset = offsetof(struct tcphdr, check);
1194
1195                 if (recalculate_partial_csum) {
1196                         struct tcphdr *tcph = (struct tcphdr *)th;
1197                         tcph->check = ~csum_tcpudp_magic(iph->saddr, iph->daddr,
1198                                                          skb->len - iph->ihl*4,
1199                                                          IPPROTO_TCP, 0);
1200                 }
1201                 break;
1202         case IPPROTO_UDP:
1203                 skb->csum_offset = offsetof(struct udphdr, check);
1204
1205                 if (recalculate_partial_csum) {
1206                         struct udphdr *udph = (struct udphdr *)th;
1207                         udph->check = ~csum_tcpudp_magic(iph->saddr, iph->daddr,
1208                                                          skb->len - iph->ihl*4,
1209                                                          IPPROTO_UDP, 0);
1210                 }
1211                 break;
1212         default:
1213                 if (net_ratelimit())
1214                         netdev_err(vif->dev,
1215                                    "Attempting to checksum a non-TCP/UDP packet, dropping a protocol %d packet\n",
1216                                    iph->protocol);
1217                 goto out;
1218         }
1219
1220         if ((th + skb->csum_offset + 2) > skb_tail_pointer(skb))
1221                 goto out;
1222
1223         err = 0;
1224
1225 out:
1226         return err;
1227 }
1228
1229 static bool tx_credit_exceeded(struct xenvif *vif, unsigned size)
1230 {
1231         unsigned long now = jiffies;
1232         unsigned long next_credit =
1233                 vif->credit_timeout.expires +
1234                 msecs_to_jiffies(vif->credit_usec / 1000);
1235
1236         /* Timer could already be pending in rare cases. */
1237         if (timer_pending(&vif->credit_timeout))
1238                 return true;
1239
1240         /* Passed the point where we can replenish credit? */
1241         if (time_after_eq(now, next_credit)) {
1242                 vif->credit_timeout.expires = now;
1243                 tx_add_credit(vif);
1244         }
1245
1246         /* Still too big to send right now? Set a callback. */
1247         if (size > vif->remaining_credit) {
1248                 vif->credit_timeout.data     =
1249                         (unsigned long)vif;
1250                 vif->credit_timeout.function =
1251                         tx_credit_callback;
1252                 mod_timer(&vif->credit_timeout,
1253                           next_credit);
1254
1255                 return true;
1256         }
1257
1258         return false;
1259 }
1260
1261 static unsigned xen_netbk_tx_build_gops(struct xen_netbk *netbk)
1262 {
1263         struct gnttab_copy *gop = netbk->tx_copy_ops, *request_gop;
1264         struct sk_buff *skb;
1265         int ret;
1266
1267         while (((nr_pending_reqs(netbk) + MAX_SKB_FRAGS) < MAX_PENDING_REQS) &&
1268                 !list_empty(&netbk->net_schedule_list)) {
1269                 struct xenvif *vif;
1270                 struct xen_netif_tx_request txreq;
1271                 struct xen_netif_tx_request txfrags[MAX_SKB_FRAGS];
1272                 struct page *page;
1273                 struct xen_netif_extra_info extras[XEN_NETIF_EXTRA_TYPE_MAX-1];
1274                 u16 pending_idx;
1275                 RING_IDX idx;
1276                 int work_to_do;
1277                 unsigned int data_len;
1278                 pending_ring_idx_t index;
1279
1280                 /* Get a netif from the list with work to do. */
1281                 vif = poll_net_schedule_list(netbk);
1282                 /* This can sometimes happen because the test of
1283                  * list_empty(net_schedule_list) at the top of the
1284                  * loop is unlocked.  Just go back and have another
1285                  * look.
1286                  */
1287                 if (!vif)
1288                         continue;
1289
1290                 if (vif->tx.sring->req_prod - vif->tx.req_cons >
1291                     XEN_NETIF_TX_RING_SIZE) {
1292                         netdev_err(vif->dev,
1293                                    "Impossible number of requests. "
1294                                    "req_prod %d, req_cons %d, size %ld\n",
1295                                    vif->tx.sring->req_prod, vif->tx.req_cons,
1296                                    XEN_NETIF_TX_RING_SIZE);
1297                         netbk_fatal_tx_err(vif);
1298                         continue;
1299                 }
1300
1301                 RING_FINAL_CHECK_FOR_REQUESTS(&vif->tx, work_to_do);
1302                 if (!work_to_do) {
1303                         xenvif_put(vif);
1304                         continue;
1305                 }
1306
1307                 idx = vif->tx.req_cons;
1308                 rmb(); /* Ensure that we see the request before we copy it. */
1309                 memcpy(&txreq, RING_GET_REQUEST(&vif->tx, idx), sizeof(txreq));
1310
1311                 /* Credit-based scheduling. */
1312                 if (txreq.size > vif->remaining_credit &&
1313                     tx_credit_exceeded(vif, txreq.size)) {
1314                         xenvif_put(vif);
1315                         continue;
1316                 }
1317
1318                 vif->remaining_credit -= txreq.size;
1319
1320                 work_to_do--;
1321                 vif->tx.req_cons = ++idx;
1322
1323                 memset(extras, 0, sizeof(extras));
1324                 if (txreq.flags & XEN_NETTXF_extra_info) {
1325                         work_to_do = xen_netbk_get_extras(vif, extras,
1326                                                           work_to_do);
1327                         idx = vif->tx.req_cons;
1328                         if (unlikely(work_to_do < 0))
1329                                 continue;
1330                 }
1331
1332                 ret = netbk_count_requests(vif, &txreq, txfrags, work_to_do);
1333                 if (unlikely(ret < 0))
1334                         continue;
1335
1336                 idx += ret;
1337
1338                 if (unlikely(txreq.size < ETH_HLEN)) {
1339                         netdev_dbg(vif->dev,
1340                                    "Bad packet size: %d\n", txreq.size);
1341                         netbk_tx_err(vif, &txreq, idx);
1342                         continue;
1343                 }
1344
1345                 /* No crossing a page as the payload mustn't fragment. */
1346                 if (unlikely((txreq.offset + txreq.size) > PAGE_SIZE)) {
1347                         netdev_err(vif->dev,
1348                                    "txreq.offset: %x, size: %u, end: %lu\n",
1349                                    txreq.offset, txreq.size,
1350                                    (txreq.offset&~PAGE_MASK) + txreq.size);
1351                         netbk_fatal_tx_err(vif);
1352                         continue;
1353                 }
1354
1355                 index = pending_index(netbk->pending_cons);
1356                 pending_idx = netbk->pending_ring[index];
1357
1358                 data_len = (txreq.size > PKT_PROT_LEN &&
1359                             ret < MAX_SKB_FRAGS) ?
1360                         PKT_PROT_LEN : txreq.size;
1361
1362                 skb = alloc_skb(data_len + NET_SKB_PAD + NET_IP_ALIGN,
1363                                 GFP_ATOMIC | __GFP_NOWARN);
1364                 if (unlikely(skb == NULL)) {
1365                         netdev_dbg(vif->dev,
1366                                    "Can't allocate a skb in start_xmit.\n");
1367                         netbk_tx_err(vif, &txreq, idx);
1368                         break;
1369                 }
1370
1371                 /* Packets passed to netif_rx() must have some headroom. */
1372                 skb_reserve(skb, NET_SKB_PAD + NET_IP_ALIGN);
1373
1374                 if (extras[XEN_NETIF_EXTRA_TYPE_GSO - 1].type) {
1375                         struct xen_netif_extra_info *gso;
1376                         gso = &extras[XEN_NETIF_EXTRA_TYPE_GSO - 1];
1377
1378                         if (netbk_set_skb_gso(vif, skb, gso)) {
1379                                 /* Failure in netbk_set_skb_gso is fatal. */
1380                                 kfree_skb(skb);
1381                                 continue;
1382                         }
1383                 }
1384
1385                 /* XXX could copy straight to head */
1386                 page = xen_netbk_alloc_page(netbk, skb, pending_idx);
1387                 if (!page) {
1388                         kfree_skb(skb);
1389                         netbk_tx_err(vif, &txreq, idx);
1390                         continue;
1391                 }
1392
1393                 gop->source.u.ref = txreq.gref;
1394                 gop->source.domid = vif->domid;
1395                 gop->source.offset = txreq.offset;
1396
1397                 gop->dest.u.gmfn = virt_to_mfn(page_address(page));
1398                 gop->dest.domid = DOMID_SELF;
1399                 gop->dest.offset = txreq.offset;
1400
1401                 gop->len = txreq.size;
1402                 gop->flags = GNTCOPY_source_gref;
1403
1404                 gop++;
1405
1406                 memcpy(&netbk->pending_tx_info[pending_idx].req,
1407                        &txreq, sizeof(txreq));
1408                 netbk->pending_tx_info[pending_idx].vif = vif;
1409                 *((u16 *)skb->data) = pending_idx;
1410
1411                 __skb_put(skb, data_len);
1412
1413                 skb_shinfo(skb)->nr_frags = ret;
1414                 if (data_len < txreq.size) {
1415                         skb_shinfo(skb)->nr_frags++;
1416                         frag_set_pending_idx(&skb_shinfo(skb)->frags[0],
1417                                              pending_idx);
1418                 } else {
1419                         frag_set_pending_idx(&skb_shinfo(skb)->frags[0],
1420                                              INVALID_PENDING_IDX);
1421                 }
1422
1423                 netbk->pending_cons++;
1424
1425                 request_gop = xen_netbk_get_requests(netbk, vif,
1426                                                      skb, txfrags, gop);
1427                 if (request_gop == NULL) {
1428                         kfree_skb(skb);
1429                         netbk_tx_err(vif, &txreq, idx);
1430                         continue;
1431                 }
1432                 gop = request_gop;
1433
1434                 __skb_queue_tail(&netbk->tx_queue, skb);
1435
1436                 vif->tx.req_cons = idx;
1437                 xen_netbk_check_rx_xenvif(vif);
1438
1439                 if ((gop-netbk->tx_copy_ops) >= ARRAY_SIZE(netbk->tx_copy_ops))
1440                         break;
1441         }
1442
1443         return gop - netbk->tx_copy_ops;
1444 }
1445
1446 static void xen_netbk_tx_submit(struct xen_netbk *netbk)
1447 {
1448         struct gnttab_copy *gop = netbk->tx_copy_ops;
1449         struct sk_buff *skb;
1450
1451         while ((skb = __skb_dequeue(&netbk->tx_queue)) != NULL) {
1452                 struct xen_netif_tx_request *txp;
1453                 struct xenvif *vif;
1454                 u16 pending_idx;
1455                 unsigned data_len;
1456
1457                 pending_idx = *((u16 *)skb->data);
1458                 vif = netbk->pending_tx_info[pending_idx].vif;
1459                 txp = &netbk->pending_tx_info[pending_idx].req;
1460
1461                 /* Check the remap error code. */
1462                 if (unlikely(xen_netbk_tx_check_gop(netbk, skb, &gop))) {
1463                         netdev_dbg(vif->dev, "netback grant failed.\n");
1464                         skb_shinfo(skb)->nr_frags = 0;
1465                         kfree_skb(skb);
1466                         continue;
1467                 }
1468
1469                 data_len = skb->len;
1470                 memcpy(skb->data,
1471                        (void *)(idx_to_kaddr(netbk, pending_idx)|txp->offset),
1472                        data_len);
1473                 if (data_len < txp->size) {
1474                         /* Append the packet payload as a fragment. */
1475                         txp->offset += data_len;
1476                         txp->size -= data_len;
1477                 } else {
1478                         /* Schedule a response immediately. */
1479                         xen_netbk_idx_release(netbk, pending_idx);
1480                 }
1481
1482                 if (txp->flags & XEN_NETTXF_csum_blank)
1483                         skb->ip_summed = CHECKSUM_PARTIAL;
1484                 else if (txp->flags & XEN_NETTXF_data_validated)
1485                         skb->ip_summed = CHECKSUM_UNNECESSARY;
1486
1487                 xen_netbk_fill_frags(netbk, skb);
1488
1489                 /*
1490                  * If the initial fragment was < PKT_PROT_LEN then
1491                  * pull through some bytes from the other fragments to
1492                  * increase the linear region to PKT_PROT_LEN bytes.
1493                  */
1494                 if (skb_headlen(skb) < PKT_PROT_LEN && skb_is_nonlinear(skb)) {
1495                         int target = min_t(int, skb->len, PKT_PROT_LEN);
1496                         __pskb_pull_tail(skb, target - skb_headlen(skb));
1497                 }
1498
1499                 skb->dev      = vif->dev;
1500                 skb->protocol = eth_type_trans(skb, skb->dev);
1501
1502                 if (checksum_setup(vif, skb)) {
1503                         netdev_dbg(vif->dev,
1504                                    "Can't setup checksum in net_tx_action\n");
1505                         kfree_skb(skb);
1506                         continue;
1507                 }
1508
1509                 vif->dev->stats.rx_bytes += skb->len;
1510                 vif->dev->stats.rx_packets++;
1511
1512                 xenvif_receive_skb(vif, skb);
1513         }
1514 }
1515
1516 /* Called after netfront has transmitted */
1517 static void xen_netbk_tx_action(struct xen_netbk *netbk)
1518 {
1519         unsigned nr_gops;
1520
1521         nr_gops = xen_netbk_tx_build_gops(netbk);
1522
1523         if (nr_gops == 0)
1524                 return;
1525
1526         gnttab_batch_copy(netbk->tx_copy_ops, nr_gops);
1527
1528         xen_netbk_tx_submit(netbk);
1529 }
1530
1531 static void xen_netbk_idx_release(struct xen_netbk *netbk, u16 pending_idx)
1532 {
1533         struct xenvif *vif;
1534         struct pending_tx_info *pending_tx_info;
1535         pending_ring_idx_t index;
1536
1537         /* Already complete? */
1538         if (netbk->mmap_pages[pending_idx] == NULL)
1539                 return;
1540
1541         pending_tx_info = &netbk->pending_tx_info[pending_idx];
1542
1543         vif = pending_tx_info->vif;
1544
1545         make_tx_response(vif, &pending_tx_info->req, XEN_NETIF_RSP_OKAY);
1546
1547         index = pending_index(netbk->pending_prod++);
1548         netbk->pending_ring[index] = pending_idx;
1549
1550         xenvif_put(vif);
1551
1552         netbk->mmap_pages[pending_idx]->mapping = 0;
1553         put_page(netbk->mmap_pages[pending_idx]);
1554         netbk->mmap_pages[pending_idx] = NULL;
1555 }
1556
1557 static void make_tx_response(struct xenvif *vif,
1558                              struct xen_netif_tx_request *txp,
1559                              s8       st)
1560 {
1561         RING_IDX i = vif->tx.rsp_prod_pvt;
1562         struct xen_netif_tx_response *resp;
1563         int notify;
1564
1565         resp = RING_GET_RESPONSE(&vif->tx, i);
1566         resp->id     = txp->id;
1567         resp->status = st;
1568
1569         if (txp->flags & XEN_NETTXF_extra_info)
1570                 RING_GET_RESPONSE(&vif->tx, ++i)->status = XEN_NETIF_RSP_NULL;
1571
1572         vif->tx.rsp_prod_pvt = ++i;
1573         RING_PUSH_RESPONSES_AND_CHECK_NOTIFY(&vif->tx, notify);
1574         if (notify)
1575                 notify_remote_via_irq(vif->irq);
1576 }
1577
1578 static struct xen_netif_rx_response *make_rx_response(struct xenvif *vif,
1579                                              u16      id,
1580                                              s8       st,
1581                                              u16      offset,
1582                                              u16      size,
1583                                              u16      flags)
1584 {
1585         RING_IDX i = vif->rx.rsp_prod_pvt;
1586         struct xen_netif_rx_response *resp;
1587
1588         resp = RING_GET_RESPONSE(&vif->rx, i);
1589         resp->offset     = offset;
1590         resp->flags      = flags;
1591         resp->id         = id;
1592         resp->status     = (s16)size;
1593         if (st < 0)
1594                 resp->status = (s16)st;
1595
1596         vif->rx.rsp_prod_pvt = ++i;
1597
1598         return resp;
1599 }
1600
1601 static inline int rx_work_todo(struct xen_netbk *netbk)
1602 {
1603         return !skb_queue_empty(&netbk->rx_queue);
1604 }
1605
1606 static inline int tx_work_todo(struct xen_netbk *netbk)
1607 {
1608
1609         if (((nr_pending_reqs(netbk) + MAX_SKB_FRAGS) < MAX_PENDING_REQS) &&
1610                         !list_empty(&netbk->net_schedule_list))
1611                 return 1;
1612
1613         return 0;
1614 }
1615
1616 static int xen_netbk_kthread(void *data)
1617 {
1618         struct xen_netbk *netbk = data;
1619         while (!kthread_should_stop()) {
1620                 wait_event_interruptible(netbk->wq,
1621                                 rx_work_todo(netbk) ||
1622                                 tx_work_todo(netbk) ||
1623                                 kthread_should_stop());
1624                 cond_resched();
1625
1626                 if (kthread_should_stop())
1627                         break;
1628
1629                 if (rx_work_todo(netbk))
1630                         xen_netbk_rx_action(netbk);
1631
1632                 if (tx_work_todo(netbk))
1633                         xen_netbk_tx_action(netbk);
1634         }
1635
1636         return 0;
1637 }
1638
1639 void xen_netbk_unmap_frontend_rings(struct xenvif *vif)
1640 {
1641         if (vif->tx.sring)
1642                 xenbus_unmap_ring_vfree(xenvif_to_xenbus_device(vif),
1643                                         vif->tx.sring);
1644         if (vif->rx.sring)
1645                 xenbus_unmap_ring_vfree(xenvif_to_xenbus_device(vif),
1646                                         vif->rx.sring);
1647 }
1648
1649 int xen_netbk_map_frontend_rings(struct xenvif *vif,
1650                                  grant_ref_t tx_ring_ref,
1651                                  grant_ref_t rx_ring_ref)
1652 {
1653         void *addr;
1654         struct xen_netif_tx_sring *txs;
1655         struct xen_netif_rx_sring *rxs;
1656
1657         int err = -ENOMEM;
1658
1659         err = xenbus_map_ring_valloc(xenvif_to_xenbus_device(vif),
1660                                      tx_ring_ref, &addr);
1661         if (err)
1662                 goto err;
1663
1664         txs = (struct xen_netif_tx_sring *)addr;
1665         BACK_RING_INIT(&vif->tx, txs, PAGE_SIZE);
1666
1667         err = xenbus_map_ring_valloc(xenvif_to_xenbus_device(vif),
1668                                      rx_ring_ref, &addr);
1669         if (err)
1670                 goto err;
1671
1672         rxs = (struct xen_netif_rx_sring *)addr;
1673         BACK_RING_INIT(&vif->rx, rxs, PAGE_SIZE);
1674
1675         vif->rx_req_cons_peek = 0;
1676
1677         return 0;
1678
1679 err:
1680         xen_netbk_unmap_frontend_rings(vif);
1681         return err;
1682 }
1683
1684 static int __init netback_init(void)
1685 {
1686         int i;
1687         int rc = 0;
1688         int group;
1689
1690         if (!xen_domain())
1691                 return -ENODEV;
1692
1693         xen_netbk_group_nr = num_online_cpus();
1694         xen_netbk = vzalloc(sizeof(struct xen_netbk) * xen_netbk_group_nr);
1695         if (!xen_netbk)
1696                 return -ENOMEM;
1697
1698         for (group = 0; group < xen_netbk_group_nr; group++) {
1699                 struct xen_netbk *netbk = &xen_netbk[group];
1700                 skb_queue_head_init(&netbk->rx_queue);
1701                 skb_queue_head_init(&netbk->tx_queue);
1702
1703                 init_timer(&netbk->net_timer);
1704                 netbk->net_timer.data = (unsigned long)netbk;
1705                 netbk->net_timer.function = xen_netbk_alarm;
1706
1707                 netbk->pending_cons = 0;
1708                 netbk->pending_prod = MAX_PENDING_REQS;
1709                 for (i = 0; i < MAX_PENDING_REQS; i++)
1710                         netbk->pending_ring[i] = i;
1711
1712                 init_waitqueue_head(&netbk->wq);
1713                 netbk->task = kthread_create(xen_netbk_kthread,
1714                                              (void *)netbk,
1715                                              "netback/%u", group);
1716
1717                 if (IS_ERR(netbk->task)) {
1718                         printk(KERN_ALERT "kthread_create() fails at netback\n");
1719                         del_timer(&netbk->net_timer);
1720                         rc = PTR_ERR(netbk->task);
1721                         goto failed_init;
1722                 }
1723
1724                 kthread_bind(netbk->task, group);
1725
1726                 INIT_LIST_HEAD(&netbk->net_schedule_list);
1727
1728                 spin_lock_init(&netbk->net_schedule_list_lock);
1729
1730                 atomic_set(&netbk->netfront_count, 0);
1731
1732                 wake_up_process(netbk->task);
1733         }
1734
1735         rc = xenvif_xenbus_init();
1736         if (rc)
1737                 goto failed_init;
1738
1739         return 0;
1740
1741 failed_init:
1742         while (--group >= 0) {
1743                 struct xen_netbk *netbk = &xen_netbk[group];
1744                 for (i = 0; i < MAX_PENDING_REQS; i++) {
1745                         if (netbk->mmap_pages[i])
1746                                 __free_page(netbk->mmap_pages[i]);
1747                 }
1748                 del_timer(&netbk->net_timer);
1749                 kthread_stop(netbk->task);
1750         }
1751         vfree(xen_netbk);
1752         return rc;
1753
1754 }
1755
1756 module_init(netback_init);
1757
1758 MODULE_LICENSE("Dual BSD/GPL");
1759 MODULE_ALIAS("xen-backend:vif");