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