123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439 |
- /*
- * 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 "stdinc.h"
- #include "dat.h"
- #include "fns.h"
- typedef struct AHash AHash;
- /*
- * hash table for finding arena's based on their names.
- */
- struct AHash
- {
- AHash *next;
- Arena *arena;
- };
- enum
- {
- AHashSize = 512,
- Emergency = 0, /* flag: performing emergency surgery */
- };
- static AHash *ahash[AHashSize];
- static uint32_t
- hashstr(char *s)
- {
- uint32_t h;
- int c;
- h = 0;
- for(; c = *s; s++){
- c ^= c << 6;
- h += (c << 11) ^ (c >> 1);
- c = *s;
- h ^= (c << 14) + (c << 7) + (c << 4) + c;
- }
- return h;
- }
- int
- addarena(Arena *arena)
- {
- AHash *a;
- uint32_t h;
- h = hashstr(arena->name) & (AHashSize - 1);
- a = MK(AHash);
- if(a == nil)
- return -1;
- a->arena = arena;
- a->next = ahash[h];
- ahash[h] = a;
- return 0;
- }
- Arena*
- findarena(char *name)
- {
- AHash *a;
- uint32_t h;
- h = hashstr(name) & (AHashSize - 1);
- for(a = ahash[h]; a != nil; a = a->next)
- if(strcmp(a->arena->name, name) == 0)
- return a->arena;
- return nil;
- }
- int
- delarena(Arena *arena)
- {
- AHash *a, *last;
- uint32_t h;
- h = hashstr(arena->name) & (AHashSize - 1);
- last = nil;
- for(a = ahash[h]; a != nil; a = a->next){
- if(a->arena == arena){
- if(last != nil)
- last->next = a->next;
- else
- ahash[h] = a->next;
- free(a);
- return 0;
- }
- last = a;
- }
- return -1;
- }
- ArenaPart*
- initarenapart(Part *part)
- {
- AMapN amn;
- ArenaPart *ap;
- ZBlock *b;
- uint32_t i;
- int ok;
- b = alloczblock(HeadSize, 0, 0);
- if(b == nil || readpart(part, PartBlank, b->data, HeadSize) < 0){
- seterr(EAdmin, "can't read arena partition header: %r");
- return nil;
- }
- ap = MKZ(ArenaPart);
- if(ap == nil){
- freezblock(b);
- return nil;
- }
- ap->part = part;
- ok = unpackarenapart(ap, b->data);
- freezblock(b);
- if(ok < 0){
- freearenapart(ap, 0);
- return nil;
- }
- ap->tabbase = (PartBlank + HeadSize + ap->blocksize - 1) & ~(ap->blocksize - 1);
- if(ap->version != ArenaPartVersion){
- seterr(ECorrupt, "unknown arena partition version %d", ap->version);
- freearenapart(ap, 0);
- return nil;
- }
- if(ap->blocksize & (ap->blocksize - 1)){
- seterr(ECorrupt, "illegal non-power-of-2 block size %d\n", ap->blocksize);
- freearenapart(ap, 0);
- return nil;
- }
- if(ap->tabbase >= ap->arenabase){
- seterr(ECorrupt, "arena partition table overlaps with arena storage");
- freearenapart(ap, 0);
- return nil;
- }
- ap->tabsize = ap->arenabase - ap->tabbase;
- partblocksize(part, ap->blocksize);
- ap->size = ap->part->size & ~(uint64_t)(ap->blocksize - 1);
- if(readarenamap(&amn, part, ap->tabbase, ap->tabsize) < 0){
- freearenapart(ap, 0);
- return nil;
- }
- ap->narenas = amn.n;
- ap->map = amn.map;
- if(okamap(ap->map, ap->narenas, ap->arenabase, ap->size, "arena table") < 0){
- if(!Emergency){
- freearenapart(ap, 0);
- return nil;
- }
- /* else keep on, for emergency use */
- }
- ap->arenas = MKNZ(Arena*, ap->narenas);
- for(i = 0; i < ap->narenas; i++){
- debugarena = i;
- ap->arenas[i] = initarena(part, ap->map[i].start, ap->map[i].stop - ap->map[i].start, ap->blocksize);
- if(ap->arenas[i] == nil){
- seterr(ECorrupt, "%s: %r", ap->map[i].name);
- if(!Emergency){
- freearenapart(ap, 1);
- return nil;
- }else{
- /* keep on, for emergency use */
- ap->narenas = i;
- break;
- }
- }
- if(namecmp(ap->map[i].name, ap->arenas[i]->name) != 0){
- seterr(ECorrupt, "arena name mismatches with expected name: %s vs. %s",
- ap->map[i].name, ap->arenas[i]->name);
- freearenapart(ap, 1);
- return nil;
- }
- if(findarena(ap->arenas[i]->name)){
- seterr(ECorrupt, "duplicate arena name %s in %s",
- ap->map[i].name, ap->part->name);
- freearenapart(ap, 1);
- return nil;
- }
- }
- for(i = 0; i < ap->narenas; i++) {
- debugarena = i;
- addarena(ap->arenas[i]);
- }
- debugarena = -1;
- return ap;
- }
- ArenaPart*
- newarenapart(Part *part, uint32_t blocksize, uint32_t tabsize)
- {
- ArenaPart *ap;
- if(blocksize & (blocksize - 1)){
- seterr(ECorrupt, "illegal non-power-of-2 block size %d\n", blocksize);
- return nil;
- }
- ap = MKZ(ArenaPart);
- if(ap == nil)
- return nil;
- ap->version = ArenaPartVersion;
- ap->part = part;
- ap->blocksize = blocksize;
- partblocksize(part, blocksize);
- ap->size = part->size & ~(uint64_t)(blocksize - 1);
- ap->tabbase = (PartBlank + HeadSize + blocksize - 1) & ~(blocksize - 1);
- ap->arenabase = (ap->tabbase + tabsize + blocksize - 1) & ~(blocksize - 1);
- ap->tabsize = ap->arenabase - ap->tabbase;
- ap->narenas = 0;
- if(wbarenapart(ap) < 0){
- freearenapart(ap, 0);
- return nil;
- }
- return ap;
- }
- int
- wbarenapart(ArenaPart *ap)
- {
- ZBlock *b;
- if(okamap(ap->map, ap->narenas, ap->arenabase, ap->size, "arena table") < 0)
- return -1;
- b = alloczblock(HeadSize, 1, 0);
- if(b == nil)
- /* ZZZ set error message? */
- return -1;
- if(packarenapart(ap, b->data) < 0){
- seterr(ECorrupt, "can't make arena partition header: %r");
- freezblock(b);
- return -1;
- }
- if(writepart(ap->part, PartBlank, b->data, HeadSize) < 0 ||
- flushpart(ap->part) < 0){
- seterr(EAdmin, "can't write arena partition header: %r");
- freezblock(b);
- return -1;
- }
- freezblock(b);
- return wbarenamap(ap->map, ap->narenas, ap->part, ap->tabbase, ap->tabsize);
- }
- void
- freearenapart(ArenaPart *ap, int freearenas)
- {
- int i;
- if(ap == nil)
- return;
- if(freearenas){
- for(i = 0; i < ap->narenas; i++){
- if(ap->arenas[i] == nil)
- continue;
- delarena(ap->arenas[i]);
- freearena(ap->arenas[i]);
- }
- }
- free(ap->map);
- free(ap->arenas);
- free(ap);
- }
- int
- okamap(AMap *am, int n, uint64_t start, uint64_t stop, char *what)
- {
- uint64_t last;
- uint32_t i;
- last = start;
- for(i = 0; i < n; i++){
- if(am[i].start < last){
- if(i == 0)
- seterr(ECorrupt, "invalid start address in %s", what);
- else
- seterr(ECorrupt, "overlapping ranges in %s", what);
- return -1;
- }
- if(am[i].stop < am[i].start){
- seterr(ECorrupt, "invalid range in %s", what);
- return -1;
- }
- last = am[i].stop;
- }
- if(last > stop){
- seterr(ECorrupt, "invalid ending address in %s", what);
- return -1;
- }
- return 0;
- }
- int
- maparenas(AMap *am, Arena **arenas, int n, char *what)
- {
- uint32_t i;
- for(i = 0; i < n; i++){
- arenas[i] = findarena(am[i].name);
- if(arenas[i] == nil){
- seterr(EAdmin, "can't find arena '%s' for '%s'\n", am[i].name, what);
- return -1;
- }
- }
- return 0;
- }
- int
- readarenamap(AMapN *amn, Part *part, uint64_t base, uint32_t size)
- {
- IFile f;
- uint32_t ok;
- if(partifile(&f, part, base, size) < 0)
- return -1;
- ok = parseamap(&f, amn);
- freeifile(&f);
- return ok;
- }
- int
- wbarenamap(AMap *am, int n, Part *part, uint64_t base, uint64_t size)
- {
- Fmt f;
- ZBlock *b;
- b = alloczblock(size, 1, part->blocksize);
- if(b == nil)
- return -1;
- fmtzbinit(&f, b);
- if(outputamap(&f, am, n) < 0){
- seterr(ECorrupt, "arena set size too small");
- freezblock(b);
- return -1;
- }
- if(writepart(part, base, b->data, size) < 0 || flushpart(part) < 0){
- seterr(EAdmin, "can't write arena set: %r");
- freezblock(b);
- return -1;
- }
- freezblock(b);
- return 0;
- }
- /*
- * amap: n '\n' amapelem * n
- * n: uint32_t
- * amapelem: name '\t' astart '\t' astop '\n'
- * astart, astop: uint64_t
- */
- int
- parseamap(IFile *f, AMapN *amn)
- {
- AMap *am;
- uint64_t v64;
- uint32_t v;
- char *s, *t, *flds[4];
- int i, n;
- /*
- * arenas
- */
- if(ifileuint32_t(f, &v) < 0){
- seterr(ECorrupt, "syntax error: bad number of elements in %s", f->name);
- return -1;
- }
- n = v;
- if(n > MaxAMap){
- seterr(ECorrupt, "illegal number of elements %d in %s",
- n, f->name);
- return -1;
- }
- am = MKNZ(AMap, n);
- if(am == nil){
- fprint(2, "out of memory\n");
- return -1;
- }
- for(i = 0; i < n; i++){
- s = ifileline(f);
- if(s)
- t = estrdup(s);
- else
- t = nil;
- if(s == nil || getfields(s, flds, 4, 0, "\t") != 3){
- fprint(2, "early eof after %d of %d, %s:#%d: %s\n", i, n, f->name, f->pos, t);
- free(t);
- return -1;
- }
- free(t);
- if(nameok(flds[0]) < 0)
- return -1;
- namecp(am[i].name, flds[0]);
- if(struint64_t(flds[1], &v64) < 0){
- seterr(ECorrupt, "syntax error: bad arena base address in %s", f->name);
- free(am);
- return -1;
- }
- am[i].start = v64;
- if(struint64_t(flds[2], &v64) < 0){
- seterr(ECorrupt, "syntax error: bad arena size in %s", f->name);
- free(am);
- return -1;
- }
- am[i].stop = v64;
- }
- amn->map = am;
- amn->n = n;
- return 0;
- }
- int
- outputamap(Fmt *f, AMap *am, int n)
- {
- int i;
- if(fmtprint(f, "%u\n", n) < 0)
- return -1;
- for(i = 0; i < n; i++)
- if(fmtprint(f, "%s\t%llu\t%llu\n", am[i].name, am[i].start, am[i].stop) < 0)
- return -1;
- return 0;
- }
|