123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183 |
- /*
- * This file is part of the UCB release of Plan 9. It is subject to the license
- * terms in the LICENSE file found in the top-level directory of this
- * distribution and at http://akaros.cs.berkeley.edu/files/Plan9License. No
- * part of the UCB release of Plan 9, including this file, may be copied,
- * modified, propagated, or distributed except according to the terms contained
- * in the LICENSE file.
- */
- #include "u.h"
- #include "../port/lib.h"
- #include "mem.h"
- #include "dat.h"
- #include "fns.h"
- enum
- {
- Hdrspc = 64, /* leave room for high-level headers */
- Bdead = 0x51494F42, /* "QIOB" */
- };
- struct
- {
- Lock Lock;
- uint32_t bytes;
- } ialloc;
- static Block*
- _allocb(int size)
- {
- Block *b;
- uint8_t *p;
- int n;
- n = BLOCKALIGN + ROUNDUP(size+Hdrspc, BLOCKALIGN) + sizeof(Block);
- if((p = malloc(n)) == nil)
- return nil;
- b = (Block*)(p + n - sizeof(Block)); /* block at end of allocated space */
- b->base = p;
- b->next = nil;
- b->list = nil;
- b->free = 0;
- b->flag = 0;
- /* align base and bounds of data */
- b->lim = (uint8_t*)(PTR2UINT(b) & ~(BLOCKALIGN-1));
- /* align start of writable data, leaving space below for added headers */
- b->rp = b->lim - ROUNDUP(size, BLOCKALIGN);
- b->wp = b->rp;
- if(b->rp < b->base || b->lim - b->rp < size)
- panic("_allocb");
- return b;
- }
- Block*
- allocb(int size)
- {
- Proc *up = externup();
- Block *b;
- /*
- * Check in a process and wait until successful.
- * Can still error out of here, though.
- */
- if(up == nil)
- panic("allocb without up: %#p\n", getcallerpc(&size));
- if((b = _allocb(size)) == nil){
- mallocsummary();
- panic("allocb: no memory for %d bytes\n", size);
- }
- return b;
- }
- Block*
- iallocb(int size)
- {
- Block *b;
- static int m1, m2, mp;
- if(ialloc.bytes > conf.ialloc){
- if((m1++%10000)==0){
- if(mp++ > 1000){
- active.exiting = 1;
- exit(0);
- }
- iprint("iallocb: limited %lu/%lu\n",
- ialloc.bytes, conf.ialloc);
- }
- return nil;
- }
- if((b = _allocb(size)) == nil){
- if((m2++%10000)==0){
- if(mp++ > 1000){
- active.exiting = 1;
- exit(0);
- }
- iprint("iallocb: no memory %lu/%lu\n",
- ialloc.bytes, conf.ialloc);
- }
- return nil;
- }
- b->flag = BINTR;
- ilock(&ialloc.Lock);
- ialloc.bytes += b->lim - b->base;
- iunlock(&ialloc.Lock);
- return b;
- }
- void
- freeb(Block *b)
- {
- void *dead = (void*)Bdead;
- uint8_t *p;
- if(b == nil)
- return;
- /*
- * drivers which perform non cache coherent DMA manage their own buffer
- * pool of uncached buffers and provide their own free routine.
- */
- if(b->free) {
- b->free(b);
- return;
- }
- if(b->flag & BINTR) {
- ilock(&ialloc.Lock);
- ialloc.bytes -= b->lim - b->base;
- iunlock(&ialloc.Lock);
- }
- p = b->base;
- /* poison the block in case someone is still holding onto it */
- b->next = dead;
- b->rp = dead;
- b->wp = dead;
- b->lim = dead;
- b->base = dead;
- free(p);
- }
- void
- checkb(Block *b, char *msg)
- {
- void *dead = (void*)Bdead;
- if(b == dead)
- panic("checkb b %s %#p", msg, b);
- if(b->base == dead || b->lim == dead || b->next == dead
- || b->rp == dead || b->wp == dead){
- print("checkb: base %#p lim %#p next %#p\n",
- b->base, b->lim, b->next);
- print("checkb: rp %#p wp %#p\n", b->rp, b->wp);
- panic("checkb dead: %s\n", msg);
- }
- if(b->base > b->lim)
- panic("checkb 0 %s %#p %#p", msg, b->base, b->lim);
- if(b->rp < b->base)
- panic("checkb 1 %s %#p %#p", msg, b->base, b->rp);
- if(b->wp < b->base)
- panic("checkb 2 %s %#p %#p", msg, b->base, b->wp);
- if(b->rp > b->lim)
- panic("checkb 3 %s %#p %#p", msg, b->rp, b->lim);
- if(b->wp > b->lim)
- panic("checkb 4 %s %#p %#p", msg, b->wp, b->lim);
- }
- void
- iallocsummary(void)
- {
- print("ialloc %lu/%lu\n", ialloc.bytes, conf.ialloc);
- }
|