net: Add network_offset to blocks
[akaros.git] / kern / src / net / ip.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 <net/ip.h>
42
43 typedef struct IP IP;
44 typedef struct Fragment4 Fragment4;
45 typedef struct Fragment6 Fragment6;
46 typedef struct Ipfrag Ipfrag;
47
48 enum {
49         IP4HDR = 20,                            /* sizeof(Ip4hdr) */
50         IP6HDR = 40,    /* sizeof(Ip6hdr) */
51         IP_HLEN4 = 0x05,        /* Header length in words */
52         IP_DF = 0x4000, /* Don't fragment */
53         IP_MF = 0x2000, /* More fragments */
54         IP6FHDR = 8,    /* sizeof(Fraghdr6) */
55         IP_MAX = 64 * 1024,     /* Maximum Internet packet size */
56 };
57
58 #define BLKIPVER(xp)    (((struct Ip4hdr*)((xp)->rp))->vihl&0xF0)
59 #define NEXT_ID(x) (__sync_add_and_fetch(&(x), 1))
60
61 /* MIB II counters */
62 enum {
63         Forwarding,
64         DefaultTTL,
65         InReceives,
66         InHdrErrors,
67         InAddrErrors,
68         ForwDatagrams,
69         InUnknownProtos,
70         InDiscards,
71         InDelivers,
72         OutRequests,
73         OutDiscards,
74         OutNoRoutes,
75         ReasmTimeout,
76         ReasmReqds,
77         ReasmOKs,
78         ReasmFails,
79         FragOKs,
80         FragFails,
81         FragCreates,
82
83         Nstats,
84 };
85
86 struct fragment4 {
87         struct block *blist;
88         struct fragment4 *next;
89         uint32_t src;
90         uint32_t dst;
91         uint16_t id;
92         uint64_t age;
93 };
94
95 struct fragment6 {
96         struct block *blist;
97         struct fragment6 *next;
98         uint8_t src[IPaddrlen];
99         uint8_t dst[IPaddrlen];
100         unsigned int id;
101         uint64_t age;
102 };
103
104 struct Ipfrag {
105         uint16_t foff;
106         uint16_t flen;
107 };
108
109 /* an instance of IP */
110 struct IP {
111         uint32_t stats[Nstats];
112
113         qlock_t fraglock4;
114         struct fragment4 *flisthead4;
115         struct fragment4 *fragfree4;
116         int id4;
117
118         qlock_t fraglock6;
119         struct fragment6 *flisthead6;
120         struct fragment6 *fragfree6;
121         int id6;
122
123         int iprouting;                          /* true if we route like a gateway */
124 };
125
126 static char *statnames[] = {
127         [Forwarding] "Forwarding",
128         [DefaultTTL] "DefaultTTL",
129         [InReceives] "InReceives",
130         [InHdrErrors] "InHdrErrors",
131         [InAddrErrors] "InAddrErrors",
132         [ForwDatagrams] "ForwDatagrams",
133         [InUnknownProtos] "InUnknownProtos",
134         [InDiscards] "InDiscards",
135         [InDelivers] "InDelivers",
136         [OutRequests] "OutRequests",
137         [OutDiscards] "OutDiscards",
138         [OutNoRoutes] "OutNoRoutes",
139         [ReasmTimeout] "ReasmTimeout",
140         [ReasmReqds] "ReasmReqds",
141         [ReasmOKs] "ReasmOKs",
142         [ReasmFails] "ReasmFails",
143         [FragOKs] "FragOKs",
144         [FragFails] "FragFails",
145         [FragCreates] "FragCreates",
146 };
147
148 #define BLKIP(xp)       ((struct Ip4hdr*)((xp)->rp))
149 /*
150  * This sleazy macro relies on the media header size being
151  * larger than sizeof(Ipfrag). ipreassemble checks this is true
152  */
153 #define BKFG(xp)        ((struct Ipfrag*)((xp)->base))
154
155 uint16_t ipcsum(uint8_t * unused_uint8_p_t);
156 struct block *ip4reassemble(struct IP *, int unused_int,
157                                                         struct block *, struct Ip4hdr *);
158 void ipfragfree4(struct IP *, struct fragment4 *);
159 struct fragment4 *ipfragallo4(struct IP *);
160
161 void ip_init_6(struct Fs *f)
162 {
163         struct V6params *v6p;
164
165         v6p = kzmalloc(sizeof(struct V6params), 0);
166
167         v6p->rp.mflag = 0;      // default not managed
168         v6p->rp.oflag = 0;
169         v6p->rp.maxraint = 600000;      // millisecs
170         v6p->rp.minraint = 200000;
171         v6p->rp.linkmtu = 0;    // no mtu sent
172         v6p->rp.reachtime = 0;
173         v6p->rp.rxmitra = 0;
174         v6p->rp.ttl = MAXTTL;
175         v6p->rp.routerlt = 3 * (v6p->rp.maxraint);
176
177         v6p->hp.rxmithost = 1000;       // v6 RETRANS_TIMER
178
179         v6p->cdrouter = -1;
180
181         f->v6p = v6p;
182
183 }
184
185 void initfrag(struct IP *ip, int size)
186 {
187         struct fragment4 *fq4, *eq4;
188         struct fragment6 *fq6, *eq6;
189
190         ip->fragfree4 =
191                 (struct fragment4 *)kzmalloc(sizeof(struct fragment4) * size, 0);
192         if (ip->fragfree4 == NULL)
193                 panic("initfrag");
194
195         eq4 = &ip->fragfree4[size];
196         for (fq4 = ip->fragfree4; fq4 < eq4; fq4++)
197                 fq4->next = fq4 + 1;
198
199         ip->fragfree4[size - 1].next = NULL;
200
201         ip->fragfree6 =
202                 (struct fragment6 *)kzmalloc(sizeof(struct fragment6) * size, 0);
203         if (ip->fragfree6 == NULL)
204                 panic("initfrag");
205
206         eq6 = &ip->fragfree6[size];
207         for (fq6 = ip->fragfree6; fq6 < eq6; fq6++)
208                 fq6->next = fq6 + 1;
209
210         ip->fragfree6[size - 1].next = NULL;
211 }
212
213 void ip_init(struct Fs *f)
214 {
215         struct IP *ip;
216
217         ip = kzmalloc(sizeof(struct IP), 0);
218         qlock_init(&ip->fraglock4);
219         qlock_init(&ip->fraglock6);
220         initfrag(ip, 100);
221         f->ip = ip;
222
223         ip_init_6(f);
224 }
225
226 void iprouting(struct Fs *f, int on)
227 {
228         f->ip->iprouting = on;
229         if (f->ip->iprouting == 0)
230                 f->ip->stats[Forwarding] = 2;
231         else
232                 f->ip->stats[Forwarding] = 1;
233 }
234
235 int
236 ipoput4(struct Fs *f,
237                 struct block *bp, int gating, int ttl, int tos, struct conv *c)
238 {
239         ERRSTACK(1);
240         struct Ipifc *ifc;
241         uint8_t *gate;
242         uint32_t fragoff;
243         struct block *xp, *nb;
244         struct Ip4hdr *eh, *feh;
245         int lid, len, seglen, chunk, dlen, blklen, offset, medialen;
246         struct route *r, *sr;
247         struct IP *ip;
248         int rv = 0;
249
250         ip = f->ip;
251
252         /* Fill out the ip header */
253         eh = (struct Ip4hdr *)(bp->rp);
254
255         ip->stats[OutRequests]++;
256
257         /* Number of uint8_ts in data and ip header to write */
258         len = blocklen(bp);
259
260         if (gating) {
261                 chunk = nhgets(eh->length);
262                 if (chunk > len) {
263                         ip->stats[OutDiscards]++;
264                         netlog(f, Logip, "short gated packet\n");
265                         goto free;
266                 }
267                 if (chunk < len)
268                         len = chunk;
269         }
270         if (len >= IP_MAX) {
271                 ip->stats[OutDiscards]++;
272                 netlog(f, Logip, "exceeded ip max size %V\n", eh->dst);
273                 goto free;
274         }
275
276         r = v4lookup(f, eh->dst, c);
277         if (r == NULL) {
278                 ip->stats[OutNoRoutes]++;
279                 netlog(f, Logip, "no interface %V\n", eh->dst);
280                 rv = -1;
281                 goto free;
282         }
283
284         ifc = r->rt.ifc;
285         if (r->rt.type & (Rifc | Runi))
286                 gate = eh->dst;
287         else if (r->rt.type & (Rbcast | Rmulti)) {
288                 gate = eh->dst;
289                 sr = v4lookup(f, eh->src, NULL);
290                 if (sr != NULL && (sr->rt.type & Runi))
291                         ifc = sr->rt.ifc;
292         } else
293                 gate = r->v4.gate;
294
295         if (!gating)
296                 eh->vihl = IP_VER4 | IP_HLEN4;
297         eh->ttl = ttl;
298         if (!gating)
299                 eh->tos = tos;
300
301         if (!canrlock(&ifc->rwlock))
302                 goto free;
303         if (waserror()) {
304                 runlock(&ifc->rwlock);
305                 nexterror();
306         }
307         if (ifc->m == NULL)
308                 goto raise;
309
310         /* If we dont need to fragment just send it */
311         medialen = ifc->maxtu - ifc->m->hsize;
312         if (bp->flag & Btso || len <= medialen) {
313                 if (!gating)
314                         hnputs(eh->id, NEXT_ID(ip->id4));
315                 hnputs(eh->length, len);
316                 if (!gating) {
317                         eh->frag[0] = 0x40;
318                         eh->frag[1] = 0;
319                 }
320                 eh->cksum[0] = 0;
321                 eh->cksum[1] = 0;
322                 hnputs(eh->cksum, ipcsum(&eh->vihl));
323                 ifc->m->bwrite(ifc, bp, V4, gate);
324                 runlock(&ifc->rwlock);
325                 poperror();
326                 return 0;
327         }
328
329         if ((eh->frag[0] & (IP_DF >> 8)) && !gating)
330                 printd("%V: DF set\n", eh->dst);
331
332         if (eh->frag[0] & (IP_DF >> 8)) {
333                 ip->stats[FragFails]++;
334                 ip->stats[OutDiscards]++;
335                 icmpcantfrag(f, bp, medialen);
336                 netlog(f, Logip, "%V: eh->frag[0] & (IP_DF>>8)\n", eh->dst);
337                 goto raise;
338         }
339
340         seglen = (medialen - IP4HDR) & ~7;
341         if (seglen < 8) {
342                 ip->stats[FragFails]++;
343                 ip->stats[OutDiscards]++;
344                 netlog(f, Logip, "%V seglen < 8\n", eh->dst);
345                 goto raise;
346         }
347
348         /* compute tcp/udp checksum in software before fragmenting */
349         ptclcsum_finalize(bp, 0);
350
351         dlen = len - IP4HDR;
352         xp = bp;
353         if (gating)
354                 lid = nhgets(eh->id);
355         else
356                 lid = NEXT_ID(ip->id4);
357
358         /* advance through the blist enough to drop IP4HDR size.  this should
359          * usually just be the first block. */
360         offset = IP4HDR;
361         while (xp != NULL && offset && offset >= BLEN(xp)) {
362                 offset -= BLEN(xp);
363                 xp = xp->next;
364         }
365         xp->rp += offset;
366
367         if (gating)
368                 fragoff = nhgets(eh->frag) << 3;
369         else
370                 fragoff = 0;
371         dlen += fragoff;
372         for (; fragoff < dlen; fragoff += seglen) {
373                 nb = blist_clone(xp, IP4HDR, seglen, fragoff);
374                 feh = (struct Ip4hdr *)(nb->rp);
375
376                 memmove(nb->wp, eh, IP4HDR);
377                 nb->wp += IP4HDR;
378
379                 if ((fragoff + seglen) >= dlen) {
380                         seglen = dlen - fragoff;
381                         hnputs(feh->frag, fragoff >> 3);
382                 } else
383                         hnputs(feh->frag, (fragoff >> 3) | IP_MF);
384
385                 hnputs(feh->length, seglen + IP4HDR);
386                 hnputs(feh->id, lid);
387
388                 feh->cksum[0] = 0;
389                 feh->cksum[1] = 0;
390                 hnputs(feh->cksum, ipcsum(&feh->vihl));
391                 ifc->m->bwrite(ifc, nb, V4, gate);
392                 ip->stats[FragCreates]++;
393         }
394         ip->stats[FragOKs]++;
395 raise:
396         runlock(&ifc->rwlock);
397         poperror();
398 free:
399         freeblist(bp);
400         return rv;
401 }
402
403 void ipiput4(struct Fs *f, struct Ipifc *ifc, struct block *bp)
404 {
405         int hl;
406         int hop, tos, proto, olen;
407         struct Ip4hdr *h;
408         struct Proto *p;
409         uint16_t frag;
410         int notforme;
411         uint8_t *dp, v6dst[IPaddrlen];
412         struct IP *ip;
413         struct route *r;
414
415         bp = pullupblock(bp, 1);
416         if (BLKIPVER(bp) != IP_VER4) {
417                 ipiput6(f, ifc, bp);
418                 return;
419         }
420
421         ip = f->ip;
422         ip->stats[InReceives]++;
423
424         /*
425          *  Ensure we have all the header info in the first
426          *  block.  Make life easier for other protocols by
427          *  collecting up to the first 64 bytes in the first block.
428          */
429         if (BLEN(bp) < 64) {
430                 hl = blocklen(bp);
431                 if (hl < IP4HDR)
432                         hl = IP4HDR;
433                 if (hl > 64)
434                         hl = 64;
435                 bp = pullupblock(bp, hl);
436                 if (bp == NULL)
437                         return;
438         }
439
440         h = (struct Ip4hdr *)(bp->rp);
441
442         /* dump anything that whose header doesn't checksum */
443         if ((bp->flag & Bipck) == 0 && ipcsum(&h->vihl)) {
444                 ip->stats[InHdrErrors]++;
445                 netlog(f, Logip, "ip: checksum error %V\n", h->src);
446                 freeblist(bp);
447                 return;
448         }
449         v4tov6(v6dst, h->dst);
450         notforme = ipforme(f, v6dst) == 0;
451
452         /* Check header length and version */
453         if ((h->vihl & 0x0F) != IP_HLEN4) {
454                 hl = (h->vihl & 0xF) << 2;
455                 if (hl < (IP_HLEN4 << 2)) {
456                         ip->stats[InHdrErrors]++;
457                         netlog(f, Logip, "ip: %V bad hivl 0x%x\n", h->src, h->vihl);
458                         freeblist(bp);
459                         return;
460                 }
461                 /* If this is not routed strip off the options */
462                 if (notforme == 0) {
463                         olen = nhgets(h->length);
464                         dp = bp->rp + (hl - (IP_HLEN4 << 2));
465                         memmove(dp, h, IP_HLEN4 << 2);
466                         bp->rp = dp;
467                         h = (struct Ip4hdr *)(bp->rp);
468                         h->vihl = (IP_VER4 | IP_HLEN4);
469                         hnputs(h->length, olen - hl + (IP_HLEN4 << 2));
470                 }
471         }
472
473         /* route */
474         if (notforme) {
475                 struct conv conv;
476
477                 if (!ip->iprouting) {
478                         freeb(bp);
479                         return;
480                 }
481
482                 /* don't forward to source's network */
483                 conv.r = NULL;
484                 r = v4lookup(f, h->dst, &conv);
485                 if (r == NULL || r->rt.ifc == ifc) {
486                         ip->stats[OutDiscards]++;
487                         freeblist(bp);
488                         return;
489                 }
490
491                 /* don't forward if packet has timed out */
492                 hop = h->ttl;
493                 if (hop < 1) {
494                         ip->stats[InHdrErrors]++;
495                         icmpttlexceeded(f, ifc->lifc->local, bp);
496                         freeblist(bp);
497                         return;
498                 }
499
500                 /* reassemble if the interface expects it */
501                 if (r->rt.ifc == NULL)
502                         panic("NULL route rfc");
503                 if (r->rt.ifc->reassemble) {
504                         frag = nhgets(h->frag);
505                         if (frag) {
506                                 h->tos = 0;
507                                 if (frag & IP_MF)
508                                         h->tos = 1;
509                                 bp = ip4reassemble(ip, frag, bp, h);
510                                 if (bp == NULL)
511                                         return;
512                                 h = (struct Ip4hdr *)(bp->rp);
513                         }
514                 }
515
516                 ip->stats[ForwDatagrams]++;
517                 tos = h->tos;
518                 hop = h->ttl;
519                 ipoput4(f, bp, 1, hop - 1, tos, &conv);
520                 return;
521         }
522
523         frag = nhgets(h->frag);
524         if (frag && frag != IP_DF) {
525                 h->tos = 0;
526                 if (frag & IP_MF)
527                         h->tos = 1;
528                 bp = ip4reassemble(ip, frag, bp, h);
529                 if (bp == NULL)
530                         return;
531                 h = (struct Ip4hdr *)(bp->rp);
532         }
533
534         /* don't let any frag info go up the stack */
535         h->frag[0] = 0;
536         h->frag[1] = 0;
537
538         proto = h->proto;
539         p = Fsrcvpcol(f, proto);
540         if (p != NULL && p->rcv != NULL) {
541                 ip->stats[InDelivers]++;
542                 (*p->rcv) (p, ifc, bp);
543                 return;
544         }
545         ip->stats[InDiscards]++;
546         ip->stats[InUnknownProtos]++;
547         freeblist(bp);
548 }
549
550 int ipstats(struct Fs *f, char *buf, int len)
551 {
552         struct IP *ip;
553         char *p, *e;
554         int i;
555
556         ip = f->ip;
557         ip->stats[DefaultTTL] = MAXTTL;
558
559         p = buf;
560         e = p + len;
561         for (i = 0; i < Nstats; i++)
562                 p = seprintf(p, e, "%s: %u\n", statnames[i], ip->stats[i]);
563         return p - buf;
564 }
565
566 struct block *ip4reassemble(struct IP *ip, int offset, struct block *bp,
567                                                         struct Ip4hdr *ih)
568 {
569         int fend;
570         uint16_t id;
571         struct fragment4 *f, *fnext;
572         uint32_t src, dst;
573         struct block *bl, **l, *last, *prev;
574         int ovlap, len, fragsize, pktposn;
575
576         src = nhgetl(ih->src);
577         dst = nhgetl(ih->dst);
578         id = nhgets(ih->id);
579
580         /*
581          *  block lists are too hard, pullupblock into a single block
582          */
583         if (bp->next) {
584                 bp = pullupblock(bp, blocklen(bp));
585                 ih = (struct Ip4hdr *)(bp->rp);
586         }
587
588         qlock(&ip->fraglock4);
589
590         /*
591          *  find a reassembly queue for this fragment
592          */
593         for (f = ip->flisthead4; f; f = fnext) {
594                 fnext = f->next;        /* because ipfragfree4 changes the list */
595                 if (f->src == src && f->dst == dst && f->id == id)
596                         break;
597                 if (f->age < NOW) {
598                         ip->stats[ReasmTimeout]++;
599                         ipfragfree4(ip, f);
600                 }
601         }
602
603         /*
604          *  if this isn't a fragmented packet, accept it
605          *  and get rid of any fragments that might go
606          *  with it.
607          */
608         if (!ih->tos && (offset & ~(IP_MF | IP_DF)) == 0) {
609                 if (f != NULL) {
610                         ipfragfree4(ip, f);
611                         ip->stats[ReasmFails]++;
612                 }
613                 qunlock(&ip->fraglock4);
614                 return bp;
615         }
616
617         if (bp->base + sizeof(struct Ipfrag) >= bp->rp) {
618                 bp = padblock(bp, sizeof(struct Ipfrag));
619                 bp->rp += sizeof(struct Ipfrag);
620         }
621
622         BKFG(bp)->foff = offset << 3;
623         BKFG(bp)->flen = nhgets(ih->length) - IP4HDR;
624
625         /* First fragment allocates a reassembly queue */
626         if (f == NULL) {
627                 f = ipfragallo4(ip);
628                 f->id = id;
629                 f->src = src;
630                 f->dst = dst;
631
632                 f->blist = bp;
633
634                 qunlock(&ip->fraglock4);
635                 ip->stats[ReasmReqds]++;
636                 return NULL;
637         }
638
639         /*
640          *  find the new fragment's position in the queue
641          */
642         prev = NULL;
643         l = &f->blist;
644         bl = f->blist;
645         while (bl != NULL && BKFG(bp)->foff > BKFG(bl)->foff) {
646                 prev = bl;
647                 l = &bl->next;
648                 bl = bl->next;
649         }
650
651         /* Check overlap of a previous fragment - trim away as necessary */
652         if (prev) {
653                 ovlap = BKFG(prev)->foff + BKFG(prev)->flen - BKFG(bp)->foff;
654                 if (ovlap > 0) {
655                         if (ovlap >= BKFG(bp)->flen) {
656                                 freeblist(bp);
657                                 qunlock(&ip->fraglock4);
658                                 return NULL;
659                         }
660                         BKFG(prev)->flen -= ovlap;
661                 }
662         }
663
664         /* Link onto assembly queue */
665         bp->next = *l;
666         *l = bp;
667
668         /* Check to see if succeeding segments overlap */
669         if (bp->next) {
670                 l = &bp->next;
671                 fend = BKFG(bp)->foff + BKFG(bp)->flen;
672                 /* Take completely covered segments out */
673                 while (*l) {
674                         ovlap = fend - BKFG(*l)->foff;
675                         if (ovlap <= 0)
676                                 break;
677                         if (ovlap < BKFG(*l)->flen) {
678                                 BKFG(*l)->flen -= ovlap;
679                                 BKFG(*l)->foff += ovlap;
680                                 /* move up ih hdrs */
681                                 memmove((*l)->rp + ovlap, (*l)->rp, IP4HDR);
682                                 (*l)->rp += ovlap;
683                                 break;
684                         }
685                         last = (*l)->next;
686                         (*l)->next = NULL;
687                         freeblist(*l);
688                         *l = last;
689                 }
690         }
691
692         /*
693          *  look for a complete packet.  if we get to a fragment
694          *  without IP_MF set, we're done.
695          */
696         pktposn = 0;
697         for (bl = f->blist; bl; bl = bl->next) {
698                 if (BKFG(bl)->foff != pktposn)
699                         break;
700                 if ((BLKIP(bl)->frag[0] & (IP_MF >> 8)) == 0) {
701                         bl = f->blist;
702                         len = nhgets(BLKIP(bl)->length);
703                         bl->wp = bl->rp + len;
704
705                         /* Pullup all the fragment headers and
706                          * return a complete packet
707                          */
708                         for (bl = bl->next; bl; bl = bl->next) {
709                                 fragsize = BKFG(bl)->flen;
710                                 len += fragsize;
711                                 bl->rp += IP4HDR;
712                                 bl->wp = bl->rp + fragsize;
713                         }
714
715                         bl = f->blist;
716                         f->blist = NULL;
717                         ipfragfree4(ip, f);
718                         ih = BLKIP(bl);
719                         hnputs(ih->length, len);
720                         qunlock(&ip->fraglock4);
721                         ip->stats[ReasmOKs]++;
722                         return bl;
723                 }
724                 pktposn += BKFG(bl)->flen;
725         }
726         qunlock(&ip->fraglock4);
727         return NULL;
728 }
729
730 /*
731  * ipfragfree4 - Free a list of fragments - assume hold fraglock4
732  */
733 void ipfragfree4(struct IP *ip, struct fragment4 *frag)
734 {
735         struct fragment4 *fl, **l;
736
737         if (frag->blist)
738                 freeblist(frag->blist);
739
740         frag->src = 0;
741         frag->id = 0;
742         frag->blist = NULL;
743
744         l = &ip->flisthead4;
745         for (fl = *l; fl; fl = fl->next) {
746                 if (fl == frag) {
747                         *l = frag->next;
748                         break;
749                 }
750                 l = &fl->next;
751         }
752
753         frag->next = ip->fragfree4;
754         ip->fragfree4 = frag;
755
756 }
757
758 /*
759  * ipfragallo4 - allocate a reassembly queue - assume hold fraglock4
760  */
761 struct fragment4 *ipfragallo4(struct IP *ip)
762 {
763         struct fragment4 *f;
764
765         while (ip->fragfree4 == NULL) {
766                 /* free last entry on fraglist */
767                 for (f = ip->flisthead4; f->next; f = f->next) ;
768                 ipfragfree4(ip, f);
769         }
770         f = ip->fragfree4;
771         ip->fragfree4 = f->next;
772         f->next = ip->flisthead4;
773         ip->flisthead4 = f;
774         f->age = NOW + 30000;
775
776         return f;
777 }
778
779 /* coreboot.c among other things needs this
780  * type of checksum.
781  */
782 uint16_t ipchecksum(uint8_t *addr, int len)
783 {
784         uint16_t sum = 0;
785
786         while (len > 0) {
787                 sum += addr[0] << 8 | addr[1];
788                 len -= 2;
789                 addr += 2;
790         }
791
792         sum = (sum & 0xffff) + (sum >> 16);
793         sum = (sum & 0xffff) + (sum >> 16);
794
795         return (sum ^ 0xffff);
796
797 }
798
799 /* change this to call ipchecksum later.
800  * but we have to be sure we're not doing something bad
801  * that violates some ip stack assumption (such as
802  * boundaries etc.)
803  */
804 uint16_t ipcsum(uint8_t * addr)
805 {
806         int len;
807         uint32_t sum;
808
809         sum = 0;
810         len = (addr[0] & 0xf) << 2;
811
812         while (len > 0) {
813                 sum += addr[0] << 8 | addr[1];
814                 len -= 2;
815                 addr += 2;
816         }
817
818         sum = (sum & 0xffff) + (sum >> 16);
819         sum = (sum & 0xffff) + (sum >> 16);
820
821         return (sum ^ 0xffff);
822 }