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