1 056fe1ba 2003-11-23 devnull #include <u.h>
2 056fe1ba 2003-11-23 devnull #include <libc.h>
3 056fe1ba 2003-11-23 devnull #include <venti.h>
4 056fe1ba 2003-11-23 devnull #include <libsec.h>
6 056fe1ba 2003-11-23 devnull typedef struct Mem Mem;
7 056fe1ba 2003-11-23 devnull typedef struct Frag Frag;
10 056fe1ba 2003-11-23 devnull BigMemSize = MaxFragSize,
11 056fe1ba 2003-11-23 devnull SmallMemSize = BigMemSize/8,
12 056fe1ba 2003-11-23 devnull NLocalFrag = 2,
15 056fe1ba 2003-11-23 devnull /* position to carve out of a Mem */
22 056fe1ba 2003-11-23 devnull struct Mem
26 056fe1ba 2003-11-23 devnull uchar *bp;
27 056fe1ba 2003-11-23 devnull uchar *ep;
28 056fe1ba 2003-11-23 devnull uchar *rp;
29 056fe1ba 2003-11-23 devnull uchar *wp;
30 056fe1ba 2003-11-23 devnull Mem *next;
34 056fe1ba 2003-11-23 devnull FragLocalFree,
35 056fe1ba 2003-11-23 devnull FragLocalAlloc,
36 056fe1ba 2003-11-23 devnull FragGlobal,
39 056fe1ba 2003-11-23 devnull struct Frag
41 056fe1ba 2003-11-23 devnull int state;
42 056fe1ba 2003-11-23 devnull Mem *mem;
43 056fe1ba 2003-11-23 devnull uchar *rp;
44 056fe1ba 2003-11-23 devnull uchar *wp;
45 056fe1ba 2003-11-23 devnull Frag *next;
46 056fe1ba 2003-11-23 devnull void (*free)(void*);
50 056fe1ba 2003-11-23 devnull struct Packet
52 056fe1ba 2003-11-23 devnull int size;
53 056fe1ba 2003-11-23 devnull int asize; /* allocated memory - greater than size unless foreign frags */
55 056fe1ba 2003-11-23 devnull Packet *next;
57 056fe1ba 2003-11-23 devnull Frag *first;
58 056fe1ba 2003-11-23 devnull Frag *last;
60 056fe1ba 2003-11-23 devnull Frag local[NLocalFrag];
63 056fe1ba 2003-11-23 devnull static Frag *fragalloc(Packet*, int n, int pos, Frag *next);
64 056fe1ba 2003-11-23 devnull static Frag *fragdup(Packet*, Frag*);
65 056fe1ba 2003-11-23 devnull static void fragfree(Frag*);
67 056fe1ba 2003-11-23 devnull static Mem *memalloc(int, int);
68 056fe1ba 2003-11-23 devnull static void memfree(Mem*);
69 056fe1ba 2003-11-23 devnull static int memhead(Mem *m, uchar *rp, int n);
70 056fe1ba 2003-11-23 devnull static int memtail(Mem *m, uchar *wp, int n);
72 056fe1ba 2003-11-23 devnull static char EPacketSize[] = "bad packet size";
73 056fe1ba 2003-11-23 devnull static char EPacketOffset[] = "bad packet offset";
74 056fe1ba 2003-11-23 devnull static char EBadSize[] = "bad size";
76 056fe1ba 2003-11-23 devnull static struct {
78 056fe1ba 2003-11-23 devnull Packet *packet;
79 056fe1ba 2003-11-23 devnull int npacket;
80 056fe1ba 2003-11-23 devnull Frag *frag;
81 056fe1ba 2003-11-23 devnull int nfrag;
82 056fe1ba 2003-11-23 devnull Mem *bigmem;
83 056fe1ba 2003-11-23 devnull int nbigmem;
84 056fe1ba 2003-11-23 devnull Mem *smallmem;
85 056fe1ba 2003-11-23 devnull int nsmallmem;
86 056fe1ba 2003-11-23 devnull } freelist;
88 056fe1ba 2003-11-23 devnull #define FRAGSIZE(f) ((f)->wp - (f)->rp)
89 056fe1ba 2003-11-23 devnull #define FRAGASIZE(f) ((f)->mem ? (f)->mem->ep - (f)->mem->bp : 0)
91 056fe1ba 2003-11-23 devnull #define NOTFREE(p) assert((p)->size>=0)
94 056fe1ba 2003-11-23 devnull packetalloc(void)
96 056fe1ba 2003-11-23 devnull Packet *p;
98 056fe1ba 2003-11-23 devnull lock(&freelist.lk);
99 056fe1ba 2003-11-23 devnull p = freelist.packet;
100 056fe1ba 2003-11-23 devnull if(p != nil)
101 056fe1ba 2003-11-23 devnull freelist.packet = p->next;
103 056fe1ba 2003-11-23 devnull freelist.npacket++;
104 056fe1ba 2003-11-23 devnull unlock(&freelist.lk);
106 056fe1ba 2003-11-23 devnull if(p == nil)
107 056fe1ba 2003-11-23 devnull p = vtbrk(sizeof(Packet));
109 056fe1ba 2003-11-23 devnull assert(p->size == -1);
110 056fe1ba 2003-11-23 devnull p->size = 0;
111 056fe1ba 2003-11-23 devnull p->asize = 0;
112 056fe1ba 2003-11-23 devnull p->first = nil;
113 056fe1ba 2003-11-23 devnull p->last = nil;
114 056fe1ba 2003-11-23 devnull p->next = nil;
116 d23a617a 2004-03-15 devnull if(0)fprint(2, "packetalloc %p from %08lux %08lux %08lux\n", p, *((uint*)&p+2), *((uint*)&p+3), *((uint*)&p+4));
118 056fe1ba 2003-11-23 devnull return p;
122 056fe1ba 2003-11-23 devnull packetfree(Packet *p)
124 056fe1ba 2003-11-23 devnull Frag *f, *ff;
126 d23a617a 2004-03-15 devnull if(0)fprint(2, "packetfree %p from %08lux\n", p, getcallerpc(&p));
128 056fe1ba 2003-11-23 devnull if(p == nil)
131 056fe1ba 2003-11-23 devnull NOTFREE(p);
132 056fe1ba 2003-11-23 devnull p->size = -1;
134 056fe1ba 2003-11-23 devnull for(f=p->first; f!=nil; f=ff) {
135 056fe1ba 2003-11-23 devnull ff = f->next;
136 056fe1ba 2003-11-23 devnull fragfree(f);
138 056fe1ba 2003-11-23 devnull p->first = nil;
139 056fe1ba 2003-11-23 devnull p->last = nil;
141 056fe1ba 2003-11-23 devnull lock(&freelist.lk);
142 056fe1ba 2003-11-23 devnull p->next = freelist.packet;
143 056fe1ba 2003-11-23 devnull freelist.packet = p;
144 056fe1ba 2003-11-23 devnull unlock(&freelist.lk);
147 056fe1ba 2003-11-23 devnull Packet *
148 056fe1ba 2003-11-23 devnull packetdup(Packet *p, int offset, int n)
150 056fe1ba 2003-11-23 devnull Frag *f, *ff;
151 056fe1ba 2003-11-23 devnull Packet *pp;
153 056fe1ba 2003-11-23 devnull NOTFREE(p);
154 056fe1ba 2003-11-23 devnull if(offset < 0 || n < 0 || offset+n > p->size) {
155 056fe1ba 2003-11-23 devnull werrstr(EBadSize);
156 056fe1ba 2003-11-23 devnull return nil;
159 056fe1ba 2003-11-23 devnull pp = packetalloc();
160 056fe1ba 2003-11-23 devnull if(n == 0)
161 056fe1ba 2003-11-23 devnull return pp;
163 056fe1ba 2003-11-23 devnull pp->size = n;
165 056fe1ba 2003-11-23 devnull /* skip offset */
166 056fe1ba 2003-11-23 devnull for(f=p->first; offset >= FRAGSIZE(f); f=f->next)
167 056fe1ba 2003-11-23 devnull offset -= FRAGSIZE(f);
169 056fe1ba 2003-11-23 devnull /* first frag */
170 056fe1ba 2003-11-23 devnull ff = fragdup(pp, f);
171 056fe1ba 2003-11-23 devnull ff->rp += offset;
172 056fe1ba 2003-11-23 devnull pp->first = ff;
173 056fe1ba 2003-11-23 devnull n -= FRAGSIZE(ff);
174 056fe1ba 2003-11-23 devnull pp->asize += FRAGASIZE(ff);
176 056fe1ba 2003-11-23 devnull /* the remaining */
177 056fe1ba 2003-11-23 devnull while(n > 0) {
178 056fe1ba 2003-11-23 devnull f = f->next;
179 056fe1ba 2003-11-23 devnull ff->next = fragdup(pp, f);
180 056fe1ba 2003-11-23 devnull ff = ff->next;
181 056fe1ba 2003-11-23 devnull n -= FRAGSIZE(ff);
182 056fe1ba 2003-11-23 devnull pp->asize += FRAGASIZE(ff);
185 056fe1ba 2003-11-23 devnull /* fix up last frag: note n <= 0 */
186 056fe1ba 2003-11-23 devnull ff->wp += n;
187 056fe1ba 2003-11-23 devnull ff->next = nil;
188 056fe1ba 2003-11-23 devnull pp->last = ff;
190 056fe1ba 2003-11-23 devnull return pp;
193 056fe1ba 2003-11-23 devnull Packet *
194 056fe1ba 2003-11-23 devnull packetsplit(Packet *p, int n)
196 056fe1ba 2003-11-23 devnull Packet *pp;
197 056fe1ba 2003-11-23 devnull Frag *f, *ff;
199 056fe1ba 2003-11-23 devnull NOTFREE(p);
200 056fe1ba 2003-11-23 devnull if(n < 0 || n > p->size) {
201 056fe1ba 2003-11-23 devnull werrstr(EPacketSize);
202 056fe1ba 2003-11-23 devnull return nil;
205 056fe1ba 2003-11-23 devnull pp = packetalloc();
206 056fe1ba 2003-11-23 devnull if(n == 0)
207 056fe1ba 2003-11-23 devnull return pp;
209 056fe1ba 2003-11-23 devnull pp->size = n;
210 056fe1ba 2003-11-23 devnull p->size -= n;
211 056fe1ba 2003-11-23 devnull ff = nil;
212 056fe1ba 2003-11-23 devnull for(f=p->first; n > 0 && n >= FRAGSIZE(f); f=f->next) {
213 056fe1ba 2003-11-23 devnull n -= FRAGSIZE(f);
214 056fe1ba 2003-11-23 devnull p->asize -= FRAGASIZE(f);
215 056fe1ba 2003-11-23 devnull pp->asize += FRAGASIZE(f);
219 056fe1ba 2003-11-23 devnull /* split shared frag */
220 056fe1ba 2003-11-23 devnull if(n > 0) {
222 056fe1ba 2003-11-23 devnull f = fragdup(pp, ff);
223 056fe1ba 2003-11-23 devnull pp->asize += FRAGASIZE(ff);
224 056fe1ba 2003-11-23 devnull ff->next = nil;
225 056fe1ba 2003-11-23 devnull ff->wp = ff->rp + n;
226 056fe1ba 2003-11-23 devnull f->rp += n;
229 056fe1ba 2003-11-23 devnull pp->first = p->first;
230 056fe1ba 2003-11-23 devnull pp->last = ff;
231 056fe1ba 2003-11-23 devnull p->first = f;
232 056fe1ba 2003-11-23 devnull return pp;
236 056fe1ba 2003-11-23 devnull packetconsume(Packet *p, uchar *buf, int n)
238 056fe1ba 2003-11-23 devnull NOTFREE(p);
239 056fe1ba 2003-11-23 devnull if(buf && packetcopy(p, buf, 0, n) < 0)
240 d23a617a 2004-03-15 devnull return -1;
241 056fe1ba 2003-11-23 devnull return packettrim(p, n, p->size-n);
245 056fe1ba 2003-11-23 devnull packettrim(Packet *p, int offset, int n)
247 056fe1ba 2003-11-23 devnull Frag *f, *ff;
249 056fe1ba 2003-11-23 devnull NOTFREE(p);
250 056fe1ba 2003-11-23 devnull if(offset < 0 || offset > p->size) {
251 056fe1ba 2003-11-23 devnull werrstr(EPacketOffset);
252 056fe1ba 2003-11-23 devnull return -1;
255 056fe1ba 2003-11-23 devnull if(n < 0 || offset + n > p->size) {
256 056fe1ba 2003-11-23 devnull werrstr(EPacketOffset);
257 056fe1ba 2003-11-23 devnull return -1;
260 056fe1ba 2003-11-23 devnull p->size = n;
262 056fe1ba 2003-11-23 devnull /* easy case */
263 056fe1ba 2003-11-23 devnull if(n == 0) {
264 056fe1ba 2003-11-23 devnull for(f=p->first; f != nil; f=ff) {
265 056fe1ba 2003-11-23 devnull ff = f->next;
266 056fe1ba 2003-11-23 devnull fragfree(f);
268 056fe1ba 2003-11-23 devnull p->first = p->last = nil;
269 056fe1ba 2003-11-23 devnull p->asize = 0;
270 056fe1ba 2003-11-23 devnull return 0;
273 056fe1ba 2003-11-23 devnull /* free before offset */
274 056fe1ba 2003-11-23 devnull for(f=p->first; offset >= FRAGSIZE(f); f=ff) {
275 056fe1ba 2003-11-23 devnull p->asize -= FRAGASIZE(f);
276 056fe1ba 2003-11-23 devnull offset -= FRAGSIZE(f);
277 056fe1ba 2003-11-23 devnull ff = f->next;
278 056fe1ba 2003-11-23 devnull fragfree(f);
281 056fe1ba 2003-11-23 devnull /* adjust frag */
282 056fe1ba 2003-11-23 devnull f->rp += offset;
283 056fe1ba 2003-11-23 devnull p->first = f;
285 056fe1ba 2003-11-23 devnull /* skip middle */
286 056fe1ba 2003-11-23 devnull for(; n > 0 && n > FRAGSIZE(f); f=f->next)
287 056fe1ba 2003-11-23 devnull n -= FRAGSIZE(f);
289 056fe1ba 2003-11-23 devnull /* adjust end */
290 056fe1ba 2003-11-23 devnull f->wp = f->rp + n;
291 056fe1ba 2003-11-23 devnull p->last = f;
292 056fe1ba 2003-11-23 devnull ff = f->next;
293 056fe1ba 2003-11-23 devnull f->next = nil;
295 056fe1ba 2003-11-23 devnull /* free after */
296 056fe1ba 2003-11-23 devnull for(f=ff; f != nil; f=ff) {
297 056fe1ba 2003-11-23 devnull p->asize -= FRAGASIZE(f);
298 056fe1ba 2003-11-23 devnull ff = f->next;
299 056fe1ba 2003-11-23 devnull fragfree(f);
301 056fe1ba 2003-11-23 devnull return 0;
305 056fe1ba 2003-11-23 devnull packetheader(Packet *p, int n)
307 056fe1ba 2003-11-23 devnull Frag *f;
310 056fe1ba 2003-11-23 devnull NOTFREE(p);
311 056fe1ba 2003-11-23 devnull if(n <= 0 || n > MaxFragSize) {
312 056fe1ba 2003-11-23 devnull werrstr(EPacketSize);
313 056fe1ba 2003-11-23 devnull return nil;
316 056fe1ba 2003-11-23 devnull p->size += n;
318 056fe1ba 2003-11-23 devnull /* try and fix in current frag */
319 056fe1ba 2003-11-23 devnull f = p->first;
320 056fe1ba 2003-11-23 devnull if(f != nil) {
321 056fe1ba 2003-11-23 devnull m = f->mem;
322 056fe1ba 2003-11-23 devnull if(n <= f->rp - m->bp)
323 056fe1ba 2003-11-23 devnull if(m->ref == 1 || memhead(m, f->rp, n) >= 0) {
324 056fe1ba 2003-11-23 devnull f->rp -= n;
325 056fe1ba 2003-11-23 devnull return f->rp;
329 056fe1ba 2003-11-23 devnull /* add frag to front */
330 056fe1ba 2003-11-23 devnull f = fragalloc(p, n, PEnd, p->first);
331 056fe1ba 2003-11-23 devnull p->asize += FRAGASIZE(f);
332 056fe1ba 2003-11-23 devnull if(p->first == nil)
333 056fe1ba 2003-11-23 devnull p->last = f;
334 056fe1ba 2003-11-23 devnull p->first = f;
335 056fe1ba 2003-11-23 devnull return f->rp;
339 056fe1ba 2003-11-23 devnull packettrailer(Packet *p, int n)
342 056fe1ba 2003-11-23 devnull Frag *f;
344 056fe1ba 2003-11-23 devnull NOTFREE(p);
345 056fe1ba 2003-11-23 devnull if(n <= 0 || n > MaxFragSize) {
346 056fe1ba 2003-11-23 devnull werrstr(EPacketSize);
347 056fe1ba 2003-11-23 devnull return nil;
350 056fe1ba 2003-11-23 devnull p->size += n;
352 056fe1ba 2003-11-23 devnull /* try and fix in current frag */
353 056fe1ba 2003-11-23 devnull if(p->first != nil) {
354 056fe1ba 2003-11-23 devnull f = p->last;
355 056fe1ba 2003-11-23 devnull m = f->mem;
356 056fe1ba 2003-11-23 devnull if(n <= m->ep - f->wp)
357 056fe1ba 2003-11-23 devnull if(m->ref == 1 || memtail(m, f->wp, n) >= 0) {
358 056fe1ba 2003-11-23 devnull f->wp += n;
359 056fe1ba 2003-11-23 devnull return f->wp - n;
363 056fe1ba 2003-11-23 devnull /* add frag to end */
364 056fe1ba 2003-11-23 devnull f = fragalloc(p, n, (p->first == nil)?PMiddle:PFront, nil);
365 056fe1ba 2003-11-23 devnull p->asize += FRAGASIZE(f);
366 056fe1ba 2003-11-23 devnull if(p->first == nil)
367 056fe1ba 2003-11-23 devnull p->first = f;
369 056fe1ba 2003-11-23 devnull p->last->next = f;
370 056fe1ba 2003-11-23 devnull p->last = f;
371 056fe1ba 2003-11-23 devnull return f->rp;
375 056fe1ba 2003-11-23 devnull packetprefix(Packet *p, uchar *buf, int n)
377 056fe1ba 2003-11-23 devnull Frag *f;
381 056fe1ba 2003-11-23 devnull NOTFREE(p);
382 056fe1ba 2003-11-23 devnull if(n <= 0)
385 056fe1ba 2003-11-23 devnull p->size += n;
387 056fe1ba 2003-11-23 devnull /* try and fix in current frag */
388 056fe1ba 2003-11-23 devnull f = p->first;
389 056fe1ba 2003-11-23 devnull if(f != nil) {
390 056fe1ba 2003-11-23 devnull m = f->mem;
391 056fe1ba 2003-11-23 devnull nn = f->rp - m->bp;
392 056fe1ba 2003-11-23 devnull if(nn > n)
394 056fe1ba 2003-11-23 devnull if(m->ref == 1 || memhead(m, f->rp, nn) >= 0) {
395 056fe1ba 2003-11-23 devnull f->rp -= nn;
396 056fe1ba 2003-11-23 devnull n -= nn;
397 056fe1ba 2003-11-23 devnull memmove(f->rp, buf+n, nn);
401 056fe1ba 2003-11-23 devnull while(n > 0) {
403 056fe1ba 2003-11-23 devnull if(nn > MaxFragSize)
404 056fe1ba 2003-11-23 devnull nn = MaxFragSize;
405 056fe1ba 2003-11-23 devnull f = fragalloc(p, nn, PEnd, p->first);
406 056fe1ba 2003-11-23 devnull p->asize += FRAGASIZE(f);
407 056fe1ba 2003-11-23 devnull if(p->first == nil)
408 056fe1ba 2003-11-23 devnull p->last = f;
409 056fe1ba 2003-11-23 devnull p->first = f;
410 056fe1ba 2003-11-23 devnull n -= nn;
411 056fe1ba 2003-11-23 devnull memmove(f->rp, buf+n, nn);
416 056fe1ba 2003-11-23 devnull packetappend(Packet *p, uchar *buf, int n)
418 056fe1ba 2003-11-23 devnull Frag *f;
422 056fe1ba 2003-11-23 devnull NOTFREE(p);
423 056fe1ba 2003-11-23 devnull if(n <= 0)
426 056fe1ba 2003-11-23 devnull p->size += n;
427 056fe1ba 2003-11-23 devnull /* try and fix in current frag */
428 056fe1ba 2003-11-23 devnull if(p->first != nil) {
429 056fe1ba 2003-11-23 devnull f = p->last;
430 056fe1ba 2003-11-23 devnull m = f->mem;
431 056fe1ba 2003-11-23 devnull nn = m->ep - f->wp;
432 056fe1ba 2003-11-23 devnull if(nn > n)
434 056fe1ba 2003-11-23 devnull if(m->ref == 1 || memtail(m, f->wp, nn) >= 0) {
435 056fe1ba 2003-11-23 devnull memmove(f->wp, buf, nn);
436 056fe1ba 2003-11-23 devnull f->wp += nn;
437 056fe1ba 2003-11-23 devnull buf += nn;
438 056fe1ba 2003-11-23 devnull n -= nn;
442 056fe1ba 2003-11-23 devnull while(n > 0) {
444 056fe1ba 2003-11-23 devnull if(nn > MaxFragSize)
445 056fe1ba 2003-11-23 devnull nn = MaxFragSize;
446 056fe1ba 2003-11-23 devnull f = fragalloc(p, nn, (p->first == nil)?PMiddle:PFront, nil);
447 056fe1ba 2003-11-23 devnull p->asize += FRAGASIZE(f);
448 056fe1ba 2003-11-23 devnull if(p->first == nil)
449 056fe1ba 2003-11-23 devnull p->first = f;
451 056fe1ba 2003-11-23 devnull p->last->next = f;
452 056fe1ba 2003-11-23 devnull p->last = f;
453 056fe1ba 2003-11-23 devnull memmove(f->rp, buf, nn);
454 056fe1ba 2003-11-23 devnull buf += nn;
455 056fe1ba 2003-11-23 devnull n -= nn;
461 056fe1ba 2003-11-23 devnull packetconcat(Packet *p, Packet *pp)
463 056fe1ba 2003-11-23 devnull NOTFREE(p);
464 056fe1ba 2003-11-23 devnull NOTFREE(pp);
465 056fe1ba 2003-11-23 devnull if(pp->size == 0)
467 056fe1ba 2003-11-23 devnull p->size += pp->size;
468 056fe1ba 2003-11-23 devnull p->asize += pp->asize;
470 056fe1ba 2003-11-23 devnull if(p->first != nil)
471 056fe1ba 2003-11-23 devnull p->last->next = pp->first;
473 056fe1ba 2003-11-23 devnull p->first = pp->first;
474 056fe1ba 2003-11-23 devnull p->last = pp->last;
475 056fe1ba 2003-11-23 devnull pp->size = 0;
476 056fe1ba 2003-11-23 devnull pp->asize = 0;
477 056fe1ba 2003-11-23 devnull pp->first = nil;
478 056fe1ba 2003-11-23 devnull pp->last = nil;
482 056fe1ba 2003-11-23 devnull packetpeek(Packet *p, uchar *buf, int offset, int n)
484 056fe1ba 2003-11-23 devnull Frag *f;
486 056fe1ba 2003-11-23 devnull uchar *b;
488 056fe1ba 2003-11-23 devnull NOTFREE(p);
489 056fe1ba 2003-11-23 devnull if(n == 0)
490 056fe1ba 2003-11-23 devnull return buf;
492 056fe1ba 2003-11-23 devnull if(offset < 0 || offset >= p->size) {
493 056fe1ba 2003-11-23 devnull werrstr(EPacketOffset);
494 056fe1ba 2003-11-23 devnull return nil;
497 056fe1ba 2003-11-23 devnull if(n < 0 || offset + n > p->size) {
498 056fe1ba 2003-11-23 devnull werrstr(EPacketSize);
499 056fe1ba 2003-11-23 devnull return nil;
502 056fe1ba 2003-11-23 devnull /* skip up to offset */
503 056fe1ba 2003-11-23 devnull for(f=p->first; offset >= FRAGSIZE(f); f=f->next)
504 056fe1ba 2003-11-23 devnull offset -= FRAGSIZE(f);
506 056fe1ba 2003-11-23 devnull /* easy case */
507 056fe1ba 2003-11-23 devnull if(offset + n <= FRAGSIZE(f))
508 056fe1ba 2003-11-23 devnull return f->rp + offset;
510 056fe1ba 2003-11-23 devnull for(b=buf; n>0; n -= nn) {
511 056fe1ba 2003-11-23 devnull nn = FRAGSIZE(f) - offset;
512 056fe1ba 2003-11-23 devnull if(nn > n)
514 056fe1ba 2003-11-23 devnull memmove(b, f->rp+offset, nn);
515 056fe1ba 2003-11-23 devnull offset = 0;
516 056fe1ba 2003-11-23 devnull f = f->next;
517 056fe1ba 2003-11-23 devnull b += nn;
520 056fe1ba 2003-11-23 devnull return buf;
524 056fe1ba 2003-11-23 devnull packetcopy(Packet *p, uchar *buf, int offset, int n)
526 056fe1ba 2003-11-23 devnull uchar *b;
528 056fe1ba 2003-11-23 devnull NOTFREE(p);
529 056fe1ba 2003-11-23 devnull b = packetpeek(p, buf, offset, n);
530 056fe1ba 2003-11-23 devnull if(b == nil)
531 056fe1ba 2003-11-23 devnull return -1;
532 056fe1ba 2003-11-23 devnull if(b != buf)
533 056fe1ba 2003-11-23 devnull memmove(buf, b, n);
534 056fe1ba 2003-11-23 devnull return 0;
538 056fe1ba 2003-11-23 devnull packetfragments(Packet *p, IOchunk *io, int nio, int offset)
540 056fe1ba 2003-11-23 devnull Frag *f;
541 056fe1ba 2003-11-23 devnull int size;
542 056fe1ba 2003-11-23 devnull IOchunk *eio;
544 056fe1ba 2003-11-23 devnull NOTFREE(p);
545 056fe1ba 2003-11-23 devnull if(p->size == 0 || nio <= 0)
546 056fe1ba 2003-11-23 devnull return 0;
548 056fe1ba 2003-11-23 devnull if(offset < 0 || offset > p->size) {
549 056fe1ba 2003-11-23 devnull werrstr(EPacketOffset);
550 056fe1ba 2003-11-23 devnull return -1;
553 056fe1ba 2003-11-23 devnull for(f=p->first; offset >= FRAGSIZE(f); f=f->next)
554 056fe1ba 2003-11-23 devnull offset -= FRAGSIZE(f);
556 056fe1ba 2003-11-23 devnull size = 0;
557 056fe1ba 2003-11-23 devnull eio = io + nio;
558 056fe1ba 2003-11-23 devnull for(; f != nil && io < eio; f=f->next) {
559 056fe1ba 2003-11-23 devnull io->addr = f->rp + offset;
560 056fe1ba 2003-11-23 devnull io->len = f->wp - (f->rp + offset);
561 056fe1ba 2003-11-23 devnull offset = 0;
562 056fe1ba 2003-11-23 devnull size += io->len;
566 056fe1ba 2003-11-23 devnull return size;
570 056fe1ba 2003-11-23 devnull packetstats(void)
572 056fe1ba 2003-11-23 devnull Packet *p;
573 056fe1ba 2003-11-23 devnull Frag *f;
576 056fe1ba 2003-11-23 devnull int np, nf, nsm, nbm;
578 056fe1ba 2003-11-23 devnull lock(&freelist.lk);
580 056fe1ba 2003-11-23 devnull for(p=freelist.packet; p; p=p->next)
583 056fe1ba 2003-11-23 devnull for(f=freelist.frag; f; f=f->next)
585 056fe1ba 2003-11-23 devnull nsm = 0;
586 056fe1ba 2003-11-23 devnull for(m=freelist.smallmem; m; m=m->next)
588 056fe1ba 2003-11-23 devnull nbm = 0;
589 056fe1ba 2003-11-23 devnull for(m=freelist.bigmem; m; m=m->next)
592 056fe1ba 2003-11-23 devnull fprint(2, "packet: %d/%d frag: %d/%d small mem: %d/%d big mem: %d/%d\n",
593 056fe1ba 2003-11-23 devnull np, freelist.npacket,
594 056fe1ba 2003-11-23 devnull nf, freelist.nfrag,
595 056fe1ba 2003-11-23 devnull nsm, freelist.nsmallmem,
596 056fe1ba 2003-11-23 devnull nbm, freelist.nbigmem);
598 056fe1ba 2003-11-23 devnull unlock(&freelist.lk);
603 056fe1ba 2003-11-23 devnull packetsize(Packet *p)
605 056fe1ba 2003-11-23 devnull NOTFREE(p);
607 056fe1ba 2003-11-23 devnull Frag *f;
608 056fe1ba 2003-11-23 devnull int size = 0;
610 056fe1ba 2003-11-23 devnull for(f=p->first; f; f=f->next)
611 056fe1ba 2003-11-23 devnull size += FRAGSIZE(f);
612 056fe1ba 2003-11-23 devnull if(size != p->size)
613 056fe1ba 2003-11-23 devnull fprint(2, "packetsize %d %d\n", size, p->size);
614 056fe1ba 2003-11-23 devnull assert(size == p->size);
616 056fe1ba 2003-11-23 devnull return p->size;
620 056fe1ba 2003-11-23 devnull packetasize(Packet *p)
622 056fe1ba 2003-11-23 devnull NOTFREE(p);
624 056fe1ba 2003-11-23 devnull Frag *f;
625 056fe1ba 2003-11-23 devnull int asize = 0;
627 056fe1ba 2003-11-23 devnull for(f=p->first; f; f=f->next)
628 056fe1ba 2003-11-23 devnull asize += FRAGASIZE(f);
629 056fe1ba 2003-11-23 devnull if(asize != p->asize)
630 056fe1ba 2003-11-23 devnull fprint(2, "packetasize %d %d\n", asize, p->asize);
631 056fe1ba 2003-11-23 devnull assert(asize == p->asize);
633 056fe1ba 2003-11-23 devnull return p->asize;
637 056fe1ba 2003-11-23 devnull packetsha1(Packet *p, uchar digest[VtScoreSize])
639 056fe1ba 2003-11-23 devnull DigestState ds;
640 056fe1ba 2003-11-23 devnull Frag *f;
641 056fe1ba 2003-11-23 devnull int size;
643 056fe1ba 2003-11-23 devnull NOTFREE(p);
644 056fe1ba 2003-11-23 devnull memset(&ds, 0, sizeof ds);
645 056fe1ba 2003-11-23 devnull size = p->size;
646 056fe1ba 2003-11-23 devnull for(f=p->first; f; f=f->next) {
647 056fe1ba 2003-11-23 devnull sha1(f->rp, FRAGSIZE(f), nil, &ds);
648 056fe1ba 2003-11-23 devnull size -= FRAGSIZE(f);
650 056fe1ba 2003-11-23 devnull assert(size == 0);
651 056fe1ba 2003-11-23 devnull sha1(nil, 0, digest, &ds);
655 056fe1ba 2003-11-23 devnull packetcmp(Packet *pkt0, Packet *pkt1)
657 056fe1ba 2003-11-23 devnull Frag *f0, *f1;
658 056fe1ba 2003-11-23 devnull int n0, n1, x;
660 056fe1ba 2003-11-23 devnull NOTFREE(pkt0);
661 056fe1ba 2003-11-23 devnull NOTFREE(pkt1);
662 056fe1ba 2003-11-23 devnull f0 = pkt0->first;
663 056fe1ba 2003-11-23 devnull f1 = pkt1->first;
665 056fe1ba 2003-11-23 devnull if(f0 == nil)
666 056fe1ba 2003-11-23 devnull return (f1 == nil)?0:-1;
667 056fe1ba 2003-11-23 devnull if(f1 == nil)
668 056fe1ba 2003-11-23 devnull return 1;
669 056fe1ba 2003-11-23 devnull n0 = FRAGSIZE(f0);
670 056fe1ba 2003-11-23 devnull n1 = FRAGSIZE(f1);
672 056fe1ba 2003-11-23 devnull for(;;) {
673 056fe1ba 2003-11-23 devnull if(n0 < n1) {
674 056fe1ba 2003-11-23 devnull x = memcmp(f0->wp - n0, f1->wp - n1, n0);
675 056fe1ba 2003-11-23 devnull if(x != 0)
676 056fe1ba 2003-11-23 devnull return x;
677 056fe1ba 2003-11-23 devnull n1 -= n0;
678 056fe1ba 2003-11-23 devnull f0 = f0->next;
679 056fe1ba 2003-11-23 devnull if(f0 == nil)
680 056fe1ba 2003-11-23 devnull return -1;
681 056fe1ba 2003-11-23 devnull n0 = FRAGSIZE(f0);
682 056fe1ba 2003-11-23 devnull } else if (n0 > n1) {
683 056fe1ba 2003-11-23 devnull x = memcmp(f0->wp - n0, f1->wp - n1, n1);
684 056fe1ba 2003-11-23 devnull if(x != 0)
685 056fe1ba 2003-11-23 devnull return x;
686 056fe1ba 2003-11-23 devnull n0 -= n1;
687 056fe1ba 2003-11-23 devnull f1 = f1->next;
688 056fe1ba 2003-11-23 devnull if(f1 == nil)
689 056fe1ba 2003-11-23 devnull return 1;
690 056fe1ba 2003-11-23 devnull n1 = FRAGSIZE(f1);
691 056fe1ba 2003-11-23 devnull } else { /* n0 == n1 */
692 056fe1ba 2003-11-23 devnull x = memcmp(f0->wp - n0, f1->wp - n1, n0);
693 056fe1ba 2003-11-23 devnull if(x != 0)
694 056fe1ba 2003-11-23 devnull return x;
695 056fe1ba 2003-11-23 devnull f0 = f0->next;
696 056fe1ba 2003-11-23 devnull f1 = f1->next;
697 056fe1ba 2003-11-23 devnull if(f0 == nil)
698 056fe1ba 2003-11-23 devnull return (f1 == nil)?0:-1;
699 056fe1ba 2003-11-23 devnull if(f1 == nil)
700 056fe1ba 2003-11-23 devnull return 1;
701 056fe1ba 2003-11-23 devnull n0 = FRAGSIZE(f0);
702 056fe1ba 2003-11-23 devnull n1 = FRAGSIZE(f1);
705 056fe1ba 2003-11-23 devnull return 0; /* for ken */
709 056fe1ba 2003-11-23 devnull static Frag *
710 056fe1ba 2003-11-23 devnull fragalloc(Packet *p, int n, int pos, Frag *next)
712 056fe1ba 2003-11-23 devnull Frag *f, *ef;
715 056fe1ba 2003-11-23 devnull /* look for local frag */
716 056fe1ba 2003-11-23 devnull f = &p->local[0];
717 056fe1ba 2003-11-23 devnull ef = &p->local[NLocalFrag];
718 056fe1ba 2003-11-23 devnull for(;f<ef; f++) {
719 056fe1ba 2003-11-23 devnull if(f->state == FragLocalFree) {
720 056fe1ba 2003-11-23 devnull f->state = FragLocalAlloc;
721 056fe1ba 2003-11-23 devnull goto Found;
724 056fe1ba 2003-11-23 devnull lock(&freelist.lk);
725 056fe1ba 2003-11-23 devnull f = freelist.frag;
726 056fe1ba 2003-11-23 devnull if(f != nil)
727 056fe1ba 2003-11-23 devnull freelist.frag = f->next;
729 056fe1ba 2003-11-23 devnull freelist.nfrag++;
730 056fe1ba 2003-11-23 devnull unlock(&freelist.lk);
732 056fe1ba 2003-11-23 devnull if(f == nil) {
733 056fe1ba 2003-11-23 devnull f = vtbrk(sizeof(Frag));
734 056fe1ba 2003-11-23 devnull f->state = FragGlobal;
738 056fe1ba 2003-11-23 devnull f->next = next;
740 056fe1ba 2003-11-23 devnull if(n == 0){
741 056fe1ba 2003-11-23 devnull f->mem = 0;
742 056fe1ba 2003-11-23 devnull f->rp = 0;
743 056fe1ba 2003-11-23 devnull f->wp = 0;
744 056fe1ba 2003-11-23 devnull return f;
747 056fe1ba 2003-11-23 devnull if(pos == PEnd && next == nil)
748 056fe1ba 2003-11-23 devnull pos = PMiddle;
749 056fe1ba 2003-11-23 devnull m = memalloc(n, pos);
750 056fe1ba 2003-11-23 devnull f->mem = m;
751 056fe1ba 2003-11-23 devnull f->rp = m->rp;
752 056fe1ba 2003-11-23 devnull f->wp = m->wp;
753 056fe1ba 2003-11-23 devnull return f;
757 056fe1ba 2003-11-23 devnull packetforeign(uchar *buf, int n, void (*free)(void *a), void *a)
759 056fe1ba 2003-11-23 devnull Packet *p;
760 056fe1ba 2003-11-23 devnull Frag *f;
762 056fe1ba 2003-11-23 devnull p = packetalloc();
763 056fe1ba 2003-11-23 devnull f = fragalloc(p, 0, 0, nil);
764 056fe1ba 2003-11-23 devnull f->free = free;
765 056fe1ba 2003-11-23 devnull f->a = a;
766 056fe1ba 2003-11-23 devnull f->next = nil;
767 056fe1ba 2003-11-23 devnull f->rp = buf;
768 056fe1ba 2003-11-23 devnull f->wp = buf+n;
770 056fe1ba 2003-11-23 devnull p->first = f;
771 056fe1ba 2003-11-23 devnull p->size = n;
772 056fe1ba 2003-11-23 devnull return p;
775 056fe1ba 2003-11-23 devnull static Frag *
776 056fe1ba 2003-11-23 devnull fragdup(Packet *p, Frag *f)
778 056fe1ba 2003-11-23 devnull Frag *ff;
781 056fe1ba 2003-11-23 devnull m = f->mem;
784 056fe1ba 2003-11-23 devnull * m->rp && m->wp can be out of date when ref == 1
785 056fe1ba 2003-11-23 devnull * also, potentially reclaims space from previous frags
787 056fe1ba 2003-11-23 devnull if(m && m->ref == 1) {
788 056fe1ba 2003-11-23 devnull m->rp = f->rp;
789 056fe1ba 2003-11-23 devnull m->wp = f->wp;
792 056fe1ba 2003-11-23 devnull ff = fragalloc(p, 0, 0, nil);
793 056fe1ba 2003-11-23 devnull *ff = *f;
795 056fe1ba 2003-11-23 devnull lock(&m->lk);
796 056fe1ba 2003-11-23 devnull m->ref++;
797 056fe1ba 2003-11-23 devnull unlock(&m->lk);
799 056fe1ba 2003-11-23 devnull return ff;
803 056fe1ba 2003-11-23 devnull static void
804 056fe1ba 2003-11-23 devnull fragfree(Frag *f)
806 056fe1ba 2003-11-23 devnull if(f->mem == nil){
807 056fe1ba 2003-11-23 devnull if(f->free)
808 056fe1ba 2003-11-23 devnull (*f->free)(f->a);
810 056fe1ba 2003-11-23 devnull memfree(f->mem);
811 056fe1ba 2003-11-23 devnull f->mem = 0;
814 056fe1ba 2003-11-23 devnull if(f->state == FragLocalAlloc) {
815 056fe1ba 2003-11-23 devnull f->state = FragLocalFree;
819 056fe1ba 2003-11-23 devnull lock(&freelist.lk);
820 056fe1ba 2003-11-23 devnull f->next = freelist.frag;
821 056fe1ba 2003-11-23 devnull freelist.frag = f;
822 056fe1ba 2003-11-23 devnull unlock(&freelist.lk);
825 056fe1ba 2003-11-23 devnull static Mem *
826 056fe1ba 2003-11-23 devnull memalloc(int n, int pos)
831 056fe1ba 2003-11-23 devnull if(n < 0 || n > MaxFragSize) {
832 056fe1ba 2003-11-23 devnull werrstr(EPacketSize);
833 056fe1ba 2003-11-23 devnull return 0;
835 056fe1ba 2003-11-23 devnull if(n <= SmallMemSize) {
836 056fe1ba 2003-11-23 devnull lock(&freelist.lk);
837 056fe1ba 2003-11-23 devnull m = freelist.smallmem;
838 056fe1ba 2003-11-23 devnull if(m != nil)
839 056fe1ba 2003-11-23 devnull freelist.smallmem = m->next;
841 056fe1ba 2003-11-23 devnull freelist.nsmallmem++;
842 056fe1ba 2003-11-23 devnull unlock(&freelist.lk);
843 056fe1ba 2003-11-23 devnull nn = SmallMemSize;
844 056fe1ba 2003-11-23 devnull } else {
845 056fe1ba 2003-11-23 devnull lock(&freelist.lk);
846 056fe1ba 2003-11-23 devnull m = freelist.bigmem;
847 056fe1ba 2003-11-23 devnull if(m != nil)
848 056fe1ba 2003-11-23 devnull freelist.bigmem = m->next;
850 056fe1ba 2003-11-23 devnull freelist.nbigmem++;
851 056fe1ba 2003-11-23 devnull unlock(&freelist.lk);
852 056fe1ba 2003-11-23 devnull nn = BigMemSize;
855 056fe1ba 2003-11-23 devnull if(m == nil) {
856 056fe1ba 2003-11-23 devnull m = vtbrk(sizeof(Mem));
857 056fe1ba 2003-11-23 devnull m->bp = vtbrk(nn);
858 056fe1ba 2003-11-23 devnull m->ep = m->bp + nn;
860 056fe1ba 2003-11-23 devnull assert(m->ref == 0);
861 056fe1ba 2003-11-23 devnull m->ref = 1;
863 056fe1ba 2003-11-23 devnull switch(pos) {
864 056fe1ba 2003-11-23 devnull default:
865 056fe1ba 2003-11-23 devnull assert(0);
866 056fe1ba 2003-11-23 devnull case PFront:
867 056fe1ba 2003-11-23 devnull m->rp = m->bp;
869 056fe1ba 2003-11-23 devnull case PMiddle:
870 056fe1ba 2003-11-23 devnull /* leave a little bit at end */
871 056fe1ba 2003-11-23 devnull m->rp = m->ep - n - 32;
873 056fe1ba 2003-11-23 devnull case PEnd:
874 056fe1ba 2003-11-23 devnull m->rp = m->ep - n;
877 056fe1ba 2003-11-23 devnull /* check we did not blow it */
878 056fe1ba 2003-11-23 devnull if(m->rp < m->bp)
879 056fe1ba 2003-11-23 devnull m->rp = m->bp;
880 056fe1ba 2003-11-23 devnull m->wp = m->rp + n;
881 056fe1ba 2003-11-23 devnull assert(m->rp >= m->bp && m->wp <= m->ep);
882 056fe1ba 2003-11-23 devnull return m;
885 056fe1ba 2003-11-23 devnull static void
886 056fe1ba 2003-11-23 devnull memfree(Mem *m)
888 056fe1ba 2003-11-23 devnull lock(&m->lk);
889 056fe1ba 2003-11-23 devnull m->ref--;
890 056fe1ba 2003-11-23 devnull if(m->ref > 0) {
891 056fe1ba 2003-11-23 devnull unlock(&m->lk);
894 056fe1ba 2003-11-23 devnull unlock(&m->lk);
895 056fe1ba 2003-11-23 devnull assert(m->ref == 0);
897 056fe1ba 2003-11-23 devnull switch(m->ep - m->bp) {
898 056fe1ba 2003-11-23 devnull default:
899 056fe1ba 2003-11-23 devnull assert(0);
900 056fe1ba 2003-11-23 devnull case SmallMemSize:
901 056fe1ba 2003-11-23 devnull lock(&freelist.lk);
902 056fe1ba 2003-11-23 devnull m->next = freelist.smallmem;
903 056fe1ba 2003-11-23 devnull freelist.smallmem = m;
904 056fe1ba 2003-11-23 devnull unlock(&freelist.lk);
906 056fe1ba 2003-11-23 devnull case BigMemSize:
907 056fe1ba 2003-11-23 devnull lock(&freelist.lk);
908 056fe1ba 2003-11-23 devnull m->next = freelist.bigmem;
909 056fe1ba 2003-11-23 devnull freelist.bigmem = m;
910 056fe1ba 2003-11-23 devnull unlock(&freelist.lk);
915 056fe1ba 2003-11-23 devnull static int
916 056fe1ba 2003-11-23 devnull memhead(Mem *m, uchar *rp, int n)
918 056fe1ba 2003-11-23 devnull lock(&m->lk);
919 056fe1ba 2003-11-23 devnull if(m->rp != rp) {
920 056fe1ba 2003-11-23 devnull unlock(&m->lk);
921 056fe1ba 2003-11-23 devnull return -1;
923 056fe1ba 2003-11-23 devnull m->rp -= n;
924 056fe1ba 2003-11-23 devnull unlock(&m->lk);
925 056fe1ba 2003-11-23 devnull return 0;
928 056fe1ba 2003-11-23 devnull static int
929 056fe1ba 2003-11-23 devnull memtail(Mem *m, uchar *wp, int n)
931 056fe1ba 2003-11-23 devnull lock(&m->lk);
932 056fe1ba 2003-11-23 devnull if(m->wp != wp) {
933 056fe1ba 2003-11-23 devnull unlock(&m->lk);
934 056fe1ba 2003-11-23 devnull return -1;
936 056fe1ba 2003-11-23 devnull m->wp += n;
937 056fe1ba 2003-11-23 devnull unlock(&m->lk);
938 056fe1ba 2003-11-23 devnull return 0;