1368c31abSDavid du Colombier #include "stdinc.h"
2368c31abSDavid du Colombier #include "dat.h"
3368c31abSDavid du Colombier #include "fns.h"
4368c31abSDavid du Colombier
5368c31abSDavid du Colombier static int verbose;
6368c31abSDavid du Colombier
7368c31abSDavid du Colombier static void
checkarena(Arena * arena,int scan,int fix)8368c31abSDavid du Colombier checkarena(Arena *arena, int scan, int fix)
9368c31abSDavid du Colombier {
10368c31abSDavid du Colombier ATailStats old;
11368c31abSDavid du Colombier int err, e;
12368c31abSDavid du Colombier
13368c31abSDavid du Colombier if(verbose && arena->memstats.clumps)
14368c31abSDavid du Colombier printarena(2, arena);
15368c31abSDavid du Colombier
16368c31abSDavid du Colombier old = arena->memstats;
17368c31abSDavid du Colombier
18368c31abSDavid du Colombier if(scan){
19368c31abSDavid du Colombier arena->memstats.used = 0;
20368c31abSDavid du Colombier arena->memstats.clumps = 0;
21368c31abSDavid du Colombier arena->memstats.cclumps = 0;
22368c31abSDavid du Colombier arena->memstats.uncsize = 0;
23368c31abSDavid du Colombier }
24368c31abSDavid du Colombier
25368c31abSDavid du Colombier err = 0;
26368c31abSDavid du Colombier for(;;){
27f9e1cf08SDavid du Colombier e = syncarena(arena, 1000, 0, fix);
28368c31abSDavid du Colombier err |= e;
29368c31abSDavid du Colombier if(!(e & SyncHeader))
30368c31abSDavid du Colombier break;
31368c31abSDavid du Colombier if(verbose && arena->memstats.clumps)
32368c31abSDavid du Colombier fprint(2, ".");
33368c31abSDavid du Colombier }
34368c31abSDavid du Colombier if(verbose && arena->memstats.clumps)
35368c31abSDavid du Colombier fprint(2, "\n");
36368c31abSDavid du Colombier
37368c31abSDavid du Colombier err &= ~SyncHeader;
38368c31abSDavid du Colombier if(arena->memstats.used != old.used
39368c31abSDavid du Colombier || arena->memstats.clumps != old.clumps
40368c31abSDavid du Colombier || arena->memstats.cclumps != old.cclumps
41368c31abSDavid du Colombier || arena->memstats.uncsize != old.uncsize){
42368c31abSDavid du Colombier fprint(2, "%s: incorrect arena header fields\n", arena->name);
43368c31abSDavid du Colombier printarena(2, arena);
44368c31abSDavid du Colombier err |= SyncHeader;
45368c31abSDavid du Colombier }
46368c31abSDavid du Colombier
47368c31abSDavid du Colombier if(!err || !fix)
48368c31abSDavid du Colombier return;
49368c31abSDavid du Colombier
50368c31abSDavid du Colombier fprint(2, "%s: writing fixed arena header fields\n", arena->name);
51368c31abSDavid du Colombier arena->diskstats = arena->memstats;
52368c31abSDavid du Colombier if(wbarena(arena) < 0)
53368c31abSDavid du Colombier fprint(2, "arena header write failed: %r\n");
54368c31abSDavid du Colombier flushdcache();
55368c31abSDavid du Colombier }
56368c31abSDavid du Colombier
57368c31abSDavid du Colombier void
usage(void)58368c31abSDavid du Colombier usage(void)
59368c31abSDavid du Colombier {
60368c31abSDavid du Colombier fprint(2, "usage: checkarenas [-afv] file [arenaname...]\n");
61368c31abSDavid du Colombier threadexitsall(0);
62368c31abSDavid du Colombier }
63368c31abSDavid du Colombier
64368c31abSDavid du Colombier int
should(char * name,int argc,char ** argv)65368c31abSDavid du Colombier should(char *name, int argc, char **argv)
66368c31abSDavid du Colombier {
67368c31abSDavid du Colombier int i;
68368c31abSDavid du Colombier
69368c31abSDavid du Colombier if(argc == 0)
70368c31abSDavid du Colombier return 1;
71368c31abSDavid du Colombier for(i=0; i<argc; i++)
72368c31abSDavid du Colombier if(strcmp(name, argv[i]) == 0)
73368c31abSDavid du Colombier return 1;
74368c31abSDavid du Colombier return 0;
75368c31abSDavid du Colombier }
76368c31abSDavid du Colombier
77368c31abSDavid du Colombier void
threadmain(int argc,char * argv[])78368c31abSDavid du Colombier threadmain(int argc, char *argv[])
79368c31abSDavid du Colombier {
80368c31abSDavid du Colombier ArenaPart *ap;
81368c31abSDavid du Colombier Part *part;
82368c31abSDavid du Colombier char *file;
83368c31abSDavid du Colombier int i, fix, scan;
84368c31abSDavid du Colombier
85368c31abSDavid du Colombier ventifmtinstall();
86368c31abSDavid du Colombier statsinit();
87368c31abSDavid du Colombier
88368c31abSDavid du Colombier fix = 0;
89368c31abSDavid du Colombier scan = 0;
90368c31abSDavid du Colombier ARGBEGIN{
91368c31abSDavid du Colombier case 'f':
92368c31abSDavid du Colombier fix++;
93368c31abSDavid du Colombier break;
94368c31abSDavid du Colombier case 'a':
95368c31abSDavid du Colombier scan = 1;
96368c31abSDavid du Colombier break;
97368c31abSDavid du Colombier case 'v':
98368c31abSDavid du Colombier verbose++;
99368c31abSDavid du Colombier break;
100368c31abSDavid du Colombier default:
101368c31abSDavid du Colombier usage();
102368c31abSDavid du Colombier break;
103368c31abSDavid du Colombier }ARGEND
104368c31abSDavid du Colombier
105368c31abSDavid du Colombier if(!fix)
106368c31abSDavid du Colombier readonly = 1;
107368c31abSDavid du Colombier
108368c31abSDavid du Colombier if(argc < 1)
109368c31abSDavid du Colombier usage();
110368c31abSDavid du Colombier
111368c31abSDavid du Colombier file = argv[0];
112368c31abSDavid du Colombier argc--;
113368c31abSDavid du Colombier argv++;
114368c31abSDavid du Colombier
115368c31abSDavid du Colombier part = initpart(file, (fix ? ORDWR : OREAD)|ODIRECT);
116368c31abSDavid du Colombier if(part == nil)
117368c31abSDavid du Colombier sysfatal("can't open partition %s: %r", file);
118368c31abSDavid du Colombier
119368c31abSDavid du Colombier ap = initarenapart(part);
120368c31abSDavid du Colombier if(ap == nil)
121368c31abSDavid du Colombier sysfatal("can't initialize arena partition in %s: %r", file);
122368c31abSDavid du Colombier
123368c31abSDavid du Colombier if(verbose > 1){
124368c31abSDavid du Colombier printarenapart(2, ap);
125368c31abSDavid du Colombier fprint(2, "\n");
126368c31abSDavid du Colombier }
127368c31abSDavid du Colombier
128368c31abSDavid du Colombier initdcache(8 * MaxDiskBlock);
129368c31abSDavid du Colombier
130368c31abSDavid du Colombier for(i = 0; i < ap->narenas; i++)
131*b8a11165SDavid du Colombier if(should(ap->arenas[i]->name, argc, argv)) {
132*b8a11165SDavid du Colombier debugarena = i;
133368c31abSDavid du Colombier checkarena(ap->arenas[i], scan, fix);
134*b8a11165SDavid du Colombier }
135368c31abSDavid du Colombier
136368c31abSDavid du Colombier if(verbose > 1)
137368c31abSDavid du Colombier printstats();
138368c31abSDavid du Colombier threadexitsall(0);
139368c31abSDavid du Colombier }
140