1 #include "stdinc.h"
2 #include "dat.h"
3 #include "fns.h"
4 
5 enum
6 {
7 	ClumpChunks	= 32*1024
8 };
9 
10 static int	verbose;
11 
12 int
13 clumpinfoeq(ClumpInfo *c, ClumpInfo *d)
14 {
15 	return c->type == d->type
16 		&& c->size == d->size
17 		&& c->uncsize == d->uncsize
18 		&& scorecmp(c->score, d->score)==0;
19 }
20 
21 int
22 findscore(Arena *arena, uchar *score)
23 {
24 	IEntry ie;
25 	ClumpInfo *ci, *cis;
26 	u64int a;
27 	u32int clump;
28 	int i, n, found;
29 
30 //ZZZ remove fprint?
31 	if(arena->clumps)
32 		fprint(2, "reading directory for arena=%s with %d entries\n", arena->name, arena->clumps);
33 
34 	cis = MKN(ClumpInfo, ClumpChunks);
35 	found = 0;
36 	a = 0;
37 	memset(&ie, 0, sizeof(IEntry));
38 	for(clump = 0; clump < arena->clumps; clump += n){
39 		n = ClumpChunks;
40 		if(n > arena->clumps - clump)
41 			n = arena->clumps - clump;
42 		if(readclumpinfos(arena, clump, cis, n) != n){
43 			seterr(EOk, "arena directory read failed: %r");
44 			break;
45 		}
46 
47 		for(i = 0; i < n; i++){
48 			ci = &cis[i];
49 			if(scorecmp(score, ci->score)==0){
50 				fprint(2, "found at clump=%d with type=%d size=%d csize=%d position=%lld\n",
51 					clump + i, ci->type, ci->uncsize, ci->size, a);
52 				found++;
53 			}
54 			a += ci->size + ClumpSize;
55 		}
56 	}
57 	free(cis);
58 	return found;
59 }
60 
61 void
62 usage(void)
63 {
64 	fprint(2, "usage: findscore [-v] arenafile score\n");
65 	threadexitsall(0);
66 }
67 
68 void
69 threadmain(int argc, char *argv[])
70 {
71 	ArenaPart *ap;
72 	Part *part;
73 	char *file;
74 	u8int score[VtScoreSize];
75 	int i, found;
76 
77 	fmtinstall('V', vtscorefmt);
78 	statsinit();
79 
80 	ARGBEGIN{
81 	case 'v':
82 		verbose++;
83 		break;
84 	default:
85 		usage();
86 		break;
87 	}ARGEND
88 
89 	readonly = 1;
90 
91 	if(argc != 2)
92 		usage();
93 
94 	file = argv[0];
95 	if(strscore(argv[1], score) < 0)
96 		sysfatal("bad score %s\n", argv[1]);
97 
98 	part = initpart(file, 0);
99 	if(part == nil)
100 		sysfatal("can't open partition %s: %r", file);
101 
102 	ap = initarenapart(part);
103 	if(ap == nil)
104 		sysfatal("can't initialize arena partition in %s: %r", file);
105 
106 	if(verbose > 1){
107 		printarenapart(2, ap);
108 		fprint(2, "\n");
109 	}
110 
111 	initdcache(8 * MaxDiskBlock);
112 
113 	found = 0;
114 	for(i = 0; i < ap->narenas; i++)
115 		found += findscore(ap->arenas[i], score);
116 
117 	print("found %d occurances of %V\n", found, score);
118 
119 	if(verbose > 1)
120 		printstats();
121 	threadexitsall(0);
122 }