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