1 7a4ee46d 2003-11-23 devnull #include "stdinc.h"
  2 7a4ee46d 2003-11-23 devnull #include "dat.h"
  3 7a4ee46d 2003-11-23 devnull #include "fns.h"
  4 7a4ee46d 2003-11-23 devnull 
  5 7a4ee46d 2003-11-23 devnull void
  6 7a4ee46d 2003-11-23 devnull usage(void)
  7 7a4ee46d 2003-11-23 devnull {
  8 7a4ee46d 2003-11-23 devnull 	fprint(2, "usage: fmtarenas [-Z] [-b blocksize] [-a arenasize] name file\n");
  9 7a4ee46d 2003-11-23 devnull 	threadexitsall(0);
 10 7a4ee46d 2003-11-23 devnull }
 11 7a4ee46d 2003-11-23 devnull 
 12 7a4ee46d 2003-11-23 devnull void
 13 7a4ee46d 2003-11-23 devnull threadmain(int argc, char *argv[])
 14 7a4ee46d 2003-11-23 devnull {
 15 7a4ee46d 2003-11-23 devnull 	ArenaPart *ap;
 16 7a4ee46d 2003-11-23 devnull 	Part *part;
 17 7a4ee46d 2003-11-23 devnull 	Arena *arena;
 18 7a4ee46d 2003-11-23 devnull 	u64int addr, limit, asize, apsize;
 19 7a4ee46d 2003-11-23 devnull 	char *file, *name, aname[ANameSize];
 20 7a4ee46d 2003-11-23 devnull 	int i, n, blocksize, tabsize, zero;
 21 7a4ee46d 2003-11-23 devnull 
 22 7a4ee46d 2003-11-23 devnull 	fmtinstall('V', vtscorefmt);
 23 7a4ee46d 2003-11-23 devnull 
 24 7a4ee46d 2003-11-23 devnull 	statsinit();
 25 7a4ee46d 2003-11-23 devnull 
 26 7a4ee46d 2003-11-23 devnull 	blocksize = 8 * 1024;
 27 7a4ee46d 2003-11-23 devnull 	asize = 512 * 1024 *1024;
 28 7a4ee46d 2003-11-23 devnull 	tabsize = 64 * 1024;		/* BUG: should be determine from number of arenas */
 29 7a4ee46d 2003-11-23 devnull 	zero = 1;
 30 7a4ee46d 2003-11-23 devnull 	ARGBEGIN{
 31 7a4ee46d 2003-11-23 devnull 	case 'a':
 32 7a4ee46d 2003-11-23 devnull 		asize = unittoull(ARGF());
 33 7a4ee46d 2003-11-23 devnull 		if(asize == TWID64)
 34 7a4ee46d 2003-11-23 devnull 			usage();
 35 7a4ee46d 2003-11-23 devnull 		break;
 36 7a4ee46d 2003-11-23 devnull 	case 'b':
 37 7a4ee46d 2003-11-23 devnull 		blocksize = unittoull(ARGF());
 38 7a4ee46d 2003-11-23 devnull 		if(blocksize == ~0)
 39 7a4ee46d 2003-11-23 devnull 			usage();
 40 7a4ee46d 2003-11-23 devnull 		if(blocksize > MaxDiskBlock){
 41 7a4ee46d 2003-11-23 devnull 			fprint(2, "block size too large, max %d\n", MaxDiskBlock);
 42 7a4ee46d 2003-11-23 devnull 			threadexitsall("usage");
 43 7a4ee46d 2003-11-23 devnull 		}
 44 7a4ee46d 2003-11-23 devnull 		break;
 45 7a4ee46d 2003-11-23 devnull 	case 'Z':
 46 7a4ee46d 2003-11-23 devnull 		zero = 0;
 47 7a4ee46d 2003-11-23 devnull 		break;
 48 7a4ee46d 2003-11-23 devnull 	default:
 49 7a4ee46d 2003-11-23 devnull 		usage();
 50 7a4ee46d 2003-11-23 devnull 		break;
 51 7a4ee46d 2003-11-23 devnull 	}ARGEND
 52 7a4ee46d 2003-11-23 devnull 
 53 7a4ee46d 2003-11-23 devnull 	if(argc != 2)
 54 7a4ee46d 2003-11-23 devnull 		usage();
 55 7a4ee46d 2003-11-23 devnull 
 56 7a4ee46d 2003-11-23 devnull 	name = argv[0];
 57 7a4ee46d 2003-11-23 devnull 	file = argv[1];
 58 7a4ee46d 2003-11-23 devnull 
 59 7a4ee46d 2003-11-23 devnull 	if(nameok(name) < 0)
 60 7a4ee46d 2003-11-23 devnull 		sysfatal("illegal name template %s", name);
 61 7a4ee46d 2003-11-23 devnull 
 62 7a4ee46d 2003-11-23 devnull 	part = initpart(file, 1);
 63 7a4ee46d 2003-11-23 devnull 	if(part == nil)
 64 7a4ee46d 2003-11-23 devnull 		sysfatal("can't open partition %s: %r", file);
 65 7a4ee46d 2003-11-23 devnull 
 66 7a4ee46d 2003-11-23 devnull 	if(zero)
 67 7a4ee46d 2003-11-23 devnull 		zeropart(part, blocksize);
 68 7a4ee46d 2003-11-23 devnull 
 69 7a4ee46d 2003-11-23 devnull 	ap = newarenapart(part, blocksize, tabsize);
 70 7a4ee46d 2003-11-23 devnull 	if(ap == nil)
 71 7a4ee46d 2003-11-23 devnull 		sysfatal("can't initialize arena: %r");
 72 7a4ee46d 2003-11-23 devnull 
 73 7a4ee46d 2003-11-23 devnull 	apsize = ap->size - ap->arenabase;
 74 7a4ee46d 2003-11-23 devnull 	n = apsize / asize;
 75 7a4ee46d 2003-11-23 devnull 	if(apsize - (n * asize) >= MinArenaSize)
 76 7a4ee46d 2003-11-23 devnull 		n++;
 77 7a4ee46d 2003-11-23 devnull 	
 78 7a4ee46d 2003-11-23 devnull 	fprint(2, "configuring %s with arenas=%d for a total storage of bytes=%lld and directory bytes=%d\n",
 79 7a4ee46d 2003-11-23 devnull 		file, n, apsize, ap->tabsize);
 80 7a4ee46d 2003-11-23 devnull 
 81 7a4ee46d 2003-11-23 devnull 	ap->narenas = n;
 82 7a4ee46d 2003-11-23 devnull 	ap->map = MKNZ(AMap, n);
 83 7a4ee46d 2003-11-23 devnull 	ap->arenas = MKNZ(Arena*, n);
 84 7a4ee46d 2003-11-23 devnull 
 85 7a4ee46d 2003-11-23 devnull 	addr = ap->arenabase;
 86 7a4ee46d 2003-11-23 devnull 	for(i = 0; i < n; i++){
 87 7a4ee46d 2003-11-23 devnull 		limit = addr + asize;
 88 7a4ee46d 2003-11-23 devnull 		if(limit >= ap->size || ap->size - limit < MinArenaSize){
 89 7a4ee46d 2003-11-23 devnull 			limit = ap->size;
 90 7a4ee46d 2003-11-23 devnull 			if(limit - addr < MinArenaSize)
 91 7a4ee46d 2003-11-23 devnull 				sysfatal("bad arena set math: runt arena at %lld,%lld %lld\n", addr, limit, ap->size);
 92 7a4ee46d 2003-11-23 devnull 		}
 93 7a4ee46d 2003-11-23 devnull 
 94 7a4ee46d 2003-11-23 devnull 		snprint(aname, ANameSize, "%s%d", name, i);
 95 7a4ee46d 2003-11-23 devnull 
 96 7a4ee46d 2003-11-23 devnull 		fprint(2, "adding arena %s at [%lld,%lld)\n", aname, addr, limit);
 97 7a4ee46d 2003-11-23 devnull 
 98 7a4ee46d 2003-11-23 devnull 		arena = newarena(part, aname, addr, limit - addr, blocksize);
 99 7a4ee46d 2003-11-23 devnull 		if(!arena)
100 7a4ee46d 2003-11-23 devnull 			fprint(2, "can't make new arena %s: %r", aname);
101 7a4ee46d 2003-11-23 devnull 		freearena(arena);
102 7a4ee46d 2003-11-23 devnull 
103 7a4ee46d 2003-11-23 devnull 		ap->map[i].start = addr;
104 7a4ee46d 2003-11-23 devnull 		ap->map[i].stop = limit;
105 7a4ee46d 2003-11-23 devnull 		namecp(ap->map[i].name, aname);
106 7a4ee46d 2003-11-23 devnull 
107 7a4ee46d 2003-11-23 devnull 		addr = limit;
108 7a4ee46d 2003-11-23 devnull 	}
109 7a4ee46d 2003-11-23 devnull 
110 7a4ee46d 2003-11-23 devnull 	if(wbarenapart(ap) < 0)
111 7a4ee46d 2003-11-23 devnull 		fprint(2, "can't write back arena partition header for %s: %r\n", file);
112 7a4ee46d 2003-11-23 devnull 
113 7a4ee46d 2003-11-23 devnull 	threadexitsall(0);
114 7a4ee46d 2003-11-23 devnull }