1 #include "stdinc.h"
2 #include "dat.h"
3 #include "fns.h"
4
5 static int verbose;
6
7 static void
checkarena(Arena * arena,int scan,int fix)8 checkarena(Arena *arena, int scan, int fix)
9 {
10 ATailStats old;
11 int err, e;
12
13 if(verbose && arena->memstats.clumps)
14 printarena(2, arena);
15
16 old = arena->memstats;
17
18 if(scan){
19 arena->memstats.used = 0;
20 arena->memstats.clumps = 0;
21 arena->memstats.cclumps = 0;
22 arena->memstats.uncsize = 0;
23 }
24
25 err = 0;
26 for(;;){
27 e = syncarena(arena, 1000, 0, fix);
28 err |= e;
29 if(!(e & SyncHeader))
30 break;
31 if(verbose && arena->memstats.clumps)
32 fprint(2, ".");
33 }
34 if(verbose && arena->memstats.clumps)
35 fprint(2, "\n");
36
37 err &= ~SyncHeader;
38 if(arena->memstats.used != old.used
39 || arena->memstats.clumps != old.clumps
40 || arena->memstats.cclumps != old.cclumps
41 || arena->memstats.uncsize != old.uncsize){
42 fprint(2, "%s: incorrect arena header fields\n", arena->name);
43 printarena(2, arena);
44 err |= SyncHeader;
45 }
46
47 if(!err || !fix)
48 return;
49
50 fprint(2, "%s: writing fixed arena header fields\n", arena->name);
51 arena->diskstats = arena->memstats;
52 if(wbarena(arena) < 0)
53 fprint(2, "arena header write failed: %r\n");
54 flushdcache();
55 }
56
57 void
usage(void)58 usage(void)
59 {
60 fprint(2, "usage: checkarenas [-afv] file [arenaname...]\n");
61 threadexitsall(0);
62 }
63
64 int
should(char * name,int argc,char ** argv)65 should(char *name, int argc, char **argv)
66 {
67 int i;
68
69 if(argc == 0)
70 return 1;
71 for(i=0; i<argc; i++)
72 if(strcmp(name, argv[i]) == 0)
73 return 1;
74 return 0;
75 }
76
77 void
threadmain(int argc,char * argv[])78 threadmain(int argc, char *argv[])
79 {
80 ArenaPart *ap;
81 Part *part;
82 char *file;
83 int i, fix, scan;
84
85 ventifmtinstall();
86 statsinit();
87
88 fix = 0;
89 scan = 0;
90 ARGBEGIN{
91 case 'f':
92 fix++;
93 break;
94 case 'a':
95 scan = 1;
96 break;
97 case 'v':
98 verbose++;
99 break;
100 default:
101 usage();
102 break;
103 }ARGEND
104
105 if(!fix)
106 readonly = 1;
107
108 if(argc < 1)
109 usage();
110
111 file = argv[0];
112 argc--;
113 argv++;
114
115 part = initpart(file, (fix ? ORDWR : OREAD)|ODIRECT);
116 if(part == nil)
117 sysfatal("can't open partition %s: %r", file);
118
119 ap = initarenapart(part);
120 if(ap == nil)
121 sysfatal("can't initialize arena partition in %s: %r", file);
122
123 if(verbose > 1){
124 printarenapart(2, ap);
125 fprint(2, "\n");
126 }
127
128 initdcache(8 * MaxDiskBlock);
129
130 for(i = 0; i < ap->narenas; i++)
131 if(should(ap->arenas[i]->name, argc, argv)) {
132 debugarena = i;
133 checkarena(ap->arenas[i], scan, fix);
134 }
135
136 if(verbose > 1)
137 printstats();
138 threadexitsall(0);
139 }
140