qio: Track the amount of bytes read
[akaros.git] / kern / src / net / loopbackmedium.c
1 /* Copyright © 1994-1999 Lucent Technologies Inc.  All rights reserved.
2  * Portions Copyright © 1997-1999 Vita Nuova Limited
3  * Portions Copyright © 2000-2007 Vita Nuova Holdings Limited
4  *                                (www.vitanuova.com)
5  * Revisions Copyright © 2000-2007 Lucent Technologies Inc. and others
6  *
7  * Modified for the Akaros operating system:
8  * Copyright (c) 2013-2014 The Regents of the University of California
9  * Copyright (c) 2013-2015 Google Inc.
10  *
11  * Permission is hereby granted, free of charge, to any person obtaining a copy
12  * of this software and associated documentation files (the "Software"), to deal
13  * in the Software without restriction, including without limitation the rights
14  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
15  * copies of the Software, and to permit persons to whom the Software is
16  * furnished to do so, subject to the following conditions:
17  *
18  * The above copyright notice and this permission notice shall be included in
19  * all copies or substantial portions of the Software.
20  *
21  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
22  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
23  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
24  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
25  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
26  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
27  * SOFTWARE. */
28
29 #include <vfs.h>
30 #include <kfs.h>
31 #include <slab.h>
32 #include <kmalloc.h>
33 #include <kref.h>
34 #include <string.h>
35 #include <stdio.h>
36 #include <assert.h>
37 #include <error.h>
38 #include <cpio.h>
39 #include <pmap.h>
40 #include <smp.h>
41 #include <ip.h>
42
43 enum {
44         Maxtu = 16 * 1024,
45 };
46
47 typedef struct LB LB;
48 struct LB {
49         struct proc *readp;
50         struct queue *q;
51         struct Fs *f;
52 };
53
54 static void loopbackread(void *a);
55
56 static void
57 loopbackbind(struct Ipifc *ifc, int unused_int, char **unused_char_pp_t)
58 {
59         LB *lb;
60
61         lb = kzmalloc(sizeof(*lb), 0);
62         lb->f = ifc->conv->p->f;
63         /* TO DO: make queue size a function of kernel memory */
64         lb->q = qopen(128 * 1024, Qmsg, NULL, NULL);
65         ifc->arg = lb;
66         ifc->mbps = 1000;
67
68         ktask("loopbackread", loopbackread, ifc);
69
70 }
71
72 static void loopbackunbind(struct Ipifc *ifc)
73 {
74         LB *lb = ifc->arg;
75
76         printk("%s is messed up, shouldn't track procs\n", __FUNCTION__);
77
78         /*if(lb->readp)
79            postnote(lb->readp, 1, "unbind", 0);
80          */
81
82         /* wait for reader to die */
83         while (lb->readp != 0)
84                 kthread_usleep(300 * 1000);
85
86         /* clean up */
87         qfree(lb->q);
88         kfree(lb);
89 }
90
91 static void
92 loopbackbwrite(struct Ipifc *ifc, struct block *bp, int unused_int,
93                            uint8_t * unused_uint8_p_t)
94 {
95         LB *lb;
96
97         ptclcsum_finalize(bp, 0);
98         lb = ifc->arg;
99         if (qpass(lb->q, bp) < 0)
100                 ifc->outerr++;
101         ifc->out++;
102 }
103
104 static void loopbackread(void *a)
105 {
106         ERRSTACK(2);
107         struct Ipifc *ifc;
108         struct block *bp;
109         LB *lb;
110
111         ifc = a;
112         lb = ifc->arg;
113         lb->readp = current;    /* hide identity under a rock for unbind */
114         if (waserror()) {
115                 lb->readp = 0;
116                 warn("loopbackread exits unexpectedly");
117                 return;
118                 poperror();
119         }
120         for (;;) {
121                 bp = qbread(lb->q, Maxtu);
122                 if (bp == NULL)
123                         continue;
124                 ifc->in++;
125                 if (!canrlock(&ifc->rwlock)) {
126                         freeb(bp);
127                         continue;
128                 }
129                 if (waserror()) {
130                         runlock(&ifc->rwlock);
131                         nexterror();
132                 }
133                 if (ifc->lifc == NULL) {
134                         freeb(bp);
135                 } else {
136                         ipifc_trace_block(ifc, bp);
137                         ipiput4(lb->f, ifc, bp);
138                 }
139                 runlock(&ifc->rwlock);
140                 poperror();
141         }
142         poperror();
143 }
144
145 struct medium loopbackmedium = {
146         .hsize = 0,
147         .mintu = 0,
148         .maxtu = Maxtu,
149         .maclen = 0,
150         .name = "loopback",
151         .bind = loopbackbind,
152         .unbind = loopbackunbind,
153         .bwrite = loopbackbwrite,
154 };
155
156 linker_func_4(loopbackmediumlink)
157 {
158         addipmedium(&loopbackmedium);
159 }