1 a0d146ed 2005-07-12 devnull #include <u.h>
  2 a0d146ed 2005-07-12 devnull #include <libc.h>
  3 a0d146ed 2005-07-12 devnull #include <venti.h>
  4 a0d146ed 2005-07-12 devnull #include <libsec.h>
  5 a0d146ed 2005-07-12 devnull #include <thread.h>
  6 a0d146ed 2005-07-12 devnull 
  7 a0d146ed 2005-07-12 devnull 
  8 a0d146ed 2005-07-12 devnull enum { STACK = 32768 };
  9 a0d146ed 2005-07-12 devnull void xxxsrand(long);
 10 a0d146ed 2005-07-12 devnull long xxxlrand(void);
 11 a0d146ed 2005-07-12 devnull 
 12 a0d146ed 2005-07-12 devnull Channel *cw;
 13 a0d146ed 2005-07-12 devnull Channel *cr;
 14 a0d146ed 2005-07-12 devnull char *host;
 15 a0d146ed 2005-07-12 devnull int blocksize, seed, randpct;
 16 a0d146ed 2005-07-12 devnull int doread, dowrite, packets, permute;
 17 a0d146ed 2005-07-12 devnull vlong totalbytes, cur;
 18 a0d146ed 2005-07-12 devnull VtConn *z;
 19 a0d146ed 2005-07-12 devnull int multi;
 20 a0d146ed 2005-07-12 devnull int maxpackets;
 21 a0d146ed 2005-07-12 devnull int sequence;
 22 a0d146ed 2005-07-12 devnull int doublecheck = 1;
 23 a0d146ed 2005-07-12 devnull uint *order;
 24 a0d146ed 2005-07-12 devnull 
 25 a0d146ed 2005-07-12 devnull void
 26 a0d146ed 2005-07-12 devnull usage(void)
 27 a0d146ed 2005-07-12 devnull {
 28 a0d146ed 2005-07-12 devnull 	fprint(2, "usage: randtest [-q] [-h host] [-s seed] [-b blocksize] [-p randpct] [-n totalbytes] [-M maxblocks] [-P] [-r] [-w]\n");
 29 a0d146ed 2005-07-12 devnull 	threadexitsall("usage");
 30 a0d146ed 2005-07-12 devnull }
 31 a0d146ed 2005-07-12 devnull 
 32 a0d146ed 2005-07-12 devnull void
 33 a0d146ed 2005-07-12 devnull wr(char *buf, char *buf2)
 34 a0d146ed 2005-07-12 devnull {
 35 a0d146ed 2005-07-12 devnull 	uchar score[VtScoreSize], score2[VtScoreSize];
 36 a0d146ed 2005-07-12 devnull 	DigestState ds;
 37 a0d146ed 2005-07-12 devnull 
 38 a0d146ed 2005-07-12 devnull 	memset(&ds, 0, sizeof ds);
 39 a0d146ed 2005-07-12 devnull 	if(doublecheck)
 40 a0d146ed 2005-07-12 devnull 		sha1((uchar*)buf, blocksize, score, &ds);
 41 a0d146ed 2005-07-12 devnull 	if(vtwrite(z, score2, VtDataType, (uchar*)buf, blocksize) < 0)
 42 a0d146ed 2005-07-12 devnull 		sysfatal("vtwrite %V at %,lld: %r", score, cur);
 43 a0d146ed 2005-07-12 devnull 	if(doublecheck && memcmp(score, score2, VtScoreSize) != 0)
 44 a0d146ed 2005-07-12 devnull 		sysfatal("score mismatch! %V %V", score, score2);
 45 a0d146ed 2005-07-12 devnull }
 46 a0d146ed 2005-07-12 devnull 
 47 a0d146ed 2005-07-12 devnull void
 48 a0d146ed 2005-07-12 devnull wrthread(void *v)
 49 a0d146ed 2005-07-12 devnull {
 50 a0d146ed 2005-07-12 devnull 	char *p;
 51 a0d146ed 2005-07-12 devnull 
 52 a0d146ed 2005-07-12 devnull 	USED(v);
 53 a0d146ed 2005-07-12 devnull 	while((p = recvp(cw)) != nil){
 54 a0d146ed 2005-07-12 devnull 		wr(p, nil);
 55 a0d146ed 2005-07-12 devnull 		free(p);
 56 a0d146ed 2005-07-12 devnull 	}
 57 a0d146ed 2005-07-12 devnull }
 58 a0d146ed 2005-07-12 devnull 
 59 a0d146ed 2005-07-12 devnull void
 60 a0d146ed 2005-07-12 devnull rd(char *buf, char *buf2)
 61 a0d146ed 2005-07-12 devnull {
 62 a0d146ed 2005-07-12 devnull 	uchar score[VtScoreSize];
 63 a0d146ed 2005-07-12 devnull 	DigestState ds;
 64 a0d146ed 2005-07-12 devnull 
 65 a0d146ed 2005-07-12 devnull 	memset(&ds, 0, sizeof ds);
 66 a0d146ed 2005-07-12 devnull 	sha1((uchar*)buf, blocksize, score, &ds);
 67 a0d146ed 2005-07-12 devnull 	if(vtread(z, score, VtDataType, (uchar*)buf2, blocksize) < 0)
 68 a0d146ed 2005-07-12 devnull 		sysfatal("vtread %V at %,lld: %r", score, cur);
 69 a0d146ed 2005-07-12 devnull 	if(memcmp(buf, buf2, blocksize) != 0)
 70 a0d146ed 2005-07-12 devnull 		sysfatal("bad data read! %V", score);
 71 a0d146ed 2005-07-12 devnull }
 72 a0d146ed 2005-07-12 devnull 
 73 a0d146ed 2005-07-12 devnull void
 74 a0d146ed 2005-07-12 devnull rdthread(void *v)
 75 a0d146ed 2005-07-12 devnull {
 76 a0d146ed 2005-07-12 devnull 	char *p, *buf2;
 77 a0d146ed 2005-07-12 devnull 
 78 a0d146ed 2005-07-12 devnull 	buf2 = vtmalloc(blocksize);
 79 a0d146ed 2005-07-12 devnull 	USED(v);
 80 a0d146ed 2005-07-12 devnull 	while((p = recvp(cr)) != nil){
 81 a0d146ed 2005-07-12 devnull 		rd(p, buf2);
 82 a0d146ed 2005-07-12 devnull 		free(p);
 83 a0d146ed 2005-07-12 devnull 	}
 84 a0d146ed 2005-07-12 devnull }
 85 a0d146ed 2005-07-12 devnull 
 86 a0d146ed 2005-07-12 devnull char *template;
 87 a0d146ed 2005-07-12 devnull 
 88 a0d146ed 2005-07-12 devnull void
 89 a0d146ed 2005-07-12 devnull run(void (*fn)(char*, char*), Channel *c)
 90 a0d146ed 2005-07-12 devnull {
 91 a0d146ed 2005-07-12 devnull 	int i, t, j, packets;
 92 a0d146ed 2005-07-12 devnull 	char *buf2, *buf;
 93 a0d146ed 2005-07-12 devnull 
 94 a0d146ed 2005-07-12 devnull 	buf2 = vtmalloc(blocksize);
 95 a0d146ed 2005-07-12 devnull 	buf = vtmalloc(blocksize);
 96 a0d146ed 2005-07-12 devnull 	cur = 0;
 97 a0d146ed 2005-07-12 devnull 	packets = totalbytes/blocksize;
 98 a0d146ed 2005-07-12 devnull 	if(maxpackets == 0)
 99 a0d146ed 2005-07-12 devnull 		maxpackets = packets;
100 a0d146ed 2005-07-12 devnull 	order = vtmalloc(packets*sizeof order[0]);
101 a0d146ed 2005-07-12 devnull 	for(i=0; i<packets; i++)
102 a0d146ed 2005-07-12 devnull 		order[i] = i;
103 a0d146ed 2005-07-12 devnull 	if(permute){
104 a0d146ed 2005-07-12 devnull 		for(i=1; i<packets; i++){
105 a0d146ed 2005-07-12 devnull 			j = nrand(i+1);
106 a0d146ed 2005-07-12 devnull 			t = order[i];
107 a0d146ed 2005-07-12 devnull 			order[i] = order[j];
108 a0d146ed 2005-07-12 devnull 			order[j] = t;
109 a0d146ed 2005-07-12 devnull 		}
110 a0d146ed 2005-07-12 devnull 	}
111 a0d146ed 2005-07-12 devnull 	for(i=0; i<packets && i<maxpackets; i++){
112 a0d146ed 2005-07-12 devnull 		memmove(buf, template, blocksize);
113 a0d146ed 2005-07-12 devnull 		*(uint*)buf = order[i];
114 a0d146ed 2005-07-12 devnull 		if(c){
115 a0d146ed 2005-07-12 devnull 			sendp(c, buf);
116 a0d146ed 2005-07-12 devnull 			buf = vtmalloc(blocksize);
117 a0d146ed 2005-07-12 devnull 		}else
118 a0d146ed 2005-07-12 devnull 			(*fn)(buf, buf2);
119 a0d146ed 2005-07-12 devnull 		cur += blocksize;
120 a0d146ed 2005-07-12 devnull 	}
121 a0d146ed 2005-07-12 devnull 	free(order);
122 a0d146ed 2005-07-12 devnull }
123 a0d146ed 2005-07-12 devnull 
124 a0d146ed 2005-07-12 devnull #define TWID64	((u64int)~(u64int)0)
125 a0d146ed 2005-07-12 devnull 
126 a0d146ed 2005-07-12 devnull u64int
127 a0d146ed 2005-07-12 devnull unittoull(char *s)
128 a0d146ed 2005-07-12 devnull {
129 a0d146ed 2005-07-12 devnull 	char *es;
130 a0d146ed 2005-07-12 devnull 	u64int n;
131 a0d146ed 2005-07-12 devnull 
132 a0d146ed 2005-07-12 devnull 	if(s == nil)
133 a0d146ed 2005-07-12 devnull 		return TWID64;
134 a0d146ed 2005-07-12 devnull 	n = strtoul(s, &es, 0);
135 a0d146ed 2005-07-12 devnull 	if(*es == 'k' || *es == 'K'){
136 a0d146ed 2005-07-12 devnull 		n *= 1024;
137 a0d146ed 2005-07-12 devnull 		es++;
138 a0d146ed 2005-07-12 devnull 	}else if(*es == 'm' || *es == 'M'){
139 a0d146ed 2005-07-12 devnull 		n *= 1024*1024;
140 a0d146ed 2005-07-12 devnull 		es++;
141 a0d146ed 2005-07-12 devnull 	}else if(*es == 'g' || *es == 'G'){
142 a0d146ed 2005-07-12 devnull 		n *= 1024*1024*1024;
143 a0d146ed 2005-07-12 devnull 		es++;
144 a0d146ed 2005-07-12 devnull 	}else if(*es == 't' || *es == 'T'){
145 a0d146ed 2005-07-12 devnull 		n *= 1024*1024;
146 a0d146ed 2005-07-12 devnull 		n *= 1024*1024;
147 a0d146ed 2005-07-12 devnull 	}
148 a0d146ed 2005-07-12 devnull 	if(*es != '\0')
149 a0d146ed 2005-07-12 devnull 		return TWID64;
150 a0d146ed 2005-07-12 devnull 	return n;
151 a0d146ed 2005-07-12 devnull }
152 a0d146ed 2005-07-12 devnull 
153 a0d146ed 2005-07-12 devnull void
154 a0d146ed 2005-07-12 devnull threadmain(int argc, char *argv[])
155 a0d146ed 2005-07-12 devnull {
156 a0d146ed 2005-07-12 devnull 	int i, max;
157 a0d146ed 2005-07-12 devnull 	vlong t0;
158 a0d146ed 2005-07-12 devnull 	double t;
159 a0d146ed 2005-07-12 devnull 
160 a0d146ed 2005-07-12 devnull 	blocksize = 8192;
161 a0d146ed 2005-07-12 devnull 	seed = 0;
162 a0d146ed 2005-07-12 devnull 	randpct = 50;
163 a0d146ed 2005-07-12 devnull 	host = nil;
164 a0d146ed 2005-07-12 devnull 	doread = 0;
165 a0d146ed 2005-07-12 devnull 	dowrite = 0;
166 a0d146ed 2005-07-12 devnull 	totalbytes = 1*1024*1024*1024;
167 a0d146ed 2005-07-12 devnull 	fmtinstall('V', vtscorefmt);
168 a0d146ed 2005-07-12 devnull 	fmtinstall('F', vtfcallfmt);
169 a0d146ed 2005-07-12 devnull 
170 a0d146ed 2005-07-12 devnull 	ARGBEGIN{
171 a0d146ed 2005-07-12 devnull 	case 'b':
172 a0d146ed 2005-07-12 devnull 		blocksize = unittoull(EARGF(usage()));
173 a0d146ed 2005-07-12 devnull 		break;
174 a0d146ed 2005-07-12 devnull 	case 'h':
175 a0d146ed 2005-07-12 devnull 		host = EARGF(usage());
176 a0d146ed 2005-07-12 devnull 		break;
177 a0d146ed 2005-07-12 devnull 	case 'M':
178 a0d146ed 2005-07-12 devnull 		maxpackets = unittoull(EARGF(usage()));
179 a0d146ed 2005-07-12 devnull 		break;
180 a0d146ed 2005-07-12 devnull 	case 'm':
181 a0d146ed 2005-07-12 devnull 		multi = atoi(EARGF(usage()));
182 a0d146ed 2005-07-12 devnull 		break;
183 a0d146ed 2005-07-12 devnull 	case 'n':
184 a0d146ed 2005-07-12 devnull 		totalbytes = unittoull(EARGF(usage()));
185 a0d146ed 2005-07-12 devnull 		break;
186 a0d146ed 2005-07-12 devnull 	case 'p':
187 a0d146ed 2005-07-12 devnull 		randpct = atoi(EARGF(usage()));
188 a0d146ed 2005-07-12 devnull 		break;
189 a0d146ed 2005-07-12 devnull 	case 'P':
190 a0d146ed 2005-07-12 devnull 		permute = 1;
191 a0d146ed 2005-07-12 devnull 		break;
192 a0d146ed 2005-07-12 devnull 	case 'S':
193 a0d146ed 2005-07-12 devnull 		doublecheck = 0;
194 a0d146ed 2005-07-12 devnull 		ventidoublechecksha1 = 0;
195 a0d146ed 2005-07-12 devnull 		break;
196 a0d146ed 2005-07-12 devnull 	case 's':
197 a0d146ed 2005-07-12 devnull 		seed = atoi(EARGF(usage()));
198 a0d146ed 2005-07-12 devnull 		break;
199 a0d146ed 2005-07-12 devnull 	case 'r':
200 a0d146ed 2005-07-12 devnull 		doread = 1;
201 a0d146ed 2005-07-12 devnull 		break;
202 a0d146ed 2005-07-12 devnull 	case 'w':
203 a0d146ed 2005-07-12 devnull 		dowrite = 1;
204 a0d146ed 2005-07-12 devnull 		break;
205 a0d146ed 2005-07-12 devnull 	case 'V':
206 a0d146ed 2005-07-12 devnull 		chattyventi++;
207 a0d146ed 2005-07-12 devnull 		break;
208 a0d146ed 2005-07-12 devnull 	default:
209 a0d146ed 2005-07-12 devnull 		usage();
210 a0d146ed 2005-07-12 devnull 	}ARGEND
211 a0d146ed 2005-07-12 devnull 
212 a0d146ed 2005-07-12 devnull 	if(doread==0 && dowrite==0){
213 a0d146ed 2005-07-12 devnull 		doread = 1;
214 a0d146ed 2005-07-12 devnull 		dowrite = 1;
215 a0d146ed 2005-07-12 devnull 	}
216 a0d146ed 2005-07-12 devnull 
217 a0d146ed 2005-07-12 devnull 	z = vtdial(host);
218 a0d146ed 2005-07-12 devnull 	if(z == nil)
219 a0d146ed 2005-07-12 devnull 		sysfatal("could not connect to server: %r");
220 a0d146ed 2005-07-12 devnull 	if(vtconnect(z) < 0)
221 a0d146ed 2005-07-12 devnull 		sysfatal("vtconnect: %r");
222 a0d146ed 2005-07-12 devnull 
223 a0d146ed 2005-07-12 devnull 	if(multi){
224 a0d146ed 2005-07-12 devnull 		cr = chancreate(sizeof(void*), 0);
225 a0d146ed 2005-07-12 devnull 		cw = chancreate(sizeof(void*), 0);
226 a0d146ed 2005-07-12 devnull 		for(i=0; i<multi; i++){
227 a0d146ed 2005-07-12 devnull 			proccreate(wrthread, nil, STACK);
228 a0d146ed 2005-07-12 devnull 			proccreate(rdthread, nil, STACK);
229 a0d146ed 2005-07-12 devnull 		}
230 a0d146ed 2005-07-12 devnull 	}
231 a0d146ed 2005-07-12 devnull 
232 a0d146ed 2005-07-12 devnull 	template = vtmalloc(blocksize);
233 a0d146ed 2005-07-12 devnull 	xxxsrand(seed);
234 a0d146ed 2005-07-12 devnull 	max = (256*randpct)/100;
235 a0d146ed 2005-07-12 devnull 	if(max == 0)
236 a0d146ed 2005-07-12 devnull 		max = 1;
237 a0d146ed 2005-07-12 devnull 	for(i=0; i<blocksize; i++)
238 a0d146ed 2005-07-12 devnull 		template[i] = xxxlrand()%max;
239 a0d146ed 2005-07-12 devnull 	if(dowrite){
240 a0d146ed 2005-07-12 devnull 		t0 = nsec();
241 a0d146ed 2005-07-12 devnull 		run(wr, cw);
242 a0d146ed 2005-07-12 devnull 		for(i=0; i<multi; i++)
243 a0d146ed 2005-07-12 devnull 			sendp(cw, nil);
244 a0d146ed 2005-07-12 devnull 		t = (nsec() - t0)/1.e9;
245 a0d146ed 2005-07-12 devnull 		print("write: %lld bytes / %.3f seconds = %.6f MB/s\n",
246 a0d146ed 2005-07-12 devnull 			totalbytes, t, (double)totalbytes/1e6/t);
247 a0d146ed 2005-07-12 devnull 	}
248 a0d146ed 2005-07-12 devnull 	if(doread){
249 a0d146ed 2005-07-12 devnull 		t0 = nsec();
250 a0d146ed 2005-07-12 devnull 		run(rd, cr);
251 a0d146ed 2005-07-12 devnull 		for(i=0; i<multi; i++)
252 a0d146ed 2005-07-12 devnull 			sendp(cr, nil);
253 a0d146ed 2005-07-12 devnull 		t = (nsec() - t0)/1.e9;
254 a0d146ed 2005-07-12 devnull 		print("read: %lld bytes / %.3f seconds = %.6f MB/s\n",
255 a0d146ed 2005-07-12 devnull 			totalbytes, t, (double)totalbytes/1e6/t);
256 a0d146ed 2005-07-12 devnull 	}
257 a0d146ed 2005-07-12 devnull 	threadexitsall(nil);
258 a0d146ed 2005-07-12 devnull }
259 a0d146ed 2005-07-12 devnull 
260 a0d146ed 2005-07-12 devnull 
261 a0d146ed 2005-07-12 devnull /*
262 a0d146ed 2005-07-12 devnull  *	algorithm by
263 a0d146ed 2005-07-12 devnull  *	D. P. Mitchell & J. A. Reeds
264 a0d146ed 2005-07-12 devnull  */
265 a0d146ed 2005-07-12 devnull 
266 a0d146ed 2005-07-12 devnull #define	LEN	607
267 a0d146ed 2005-07-12 devnull #define	TAP	273
268 a0d146ed 2005-07-12 devnull #define	MASK	0x7fffffffL
269 a0d146ed 2005-07-12 devnull #define	A	48271
270 a0d146ed 2005-07-12 devnull #define	M	2147483647
271 a0d146ed 2005-07-12 devnull #define	Q	44488
272 a0d146ed 2005-07-12 devnull #define	R	3399
273 a0d146ed 2005-07-12 devnull #define	NORM	(1.0/(1.0+MASK))
274 a0d146ed 2005-07-12 devnull 
275 a0d146ed 2005-07-12 devnull static	ulong	rng_vec[LEN];
276 a0d146ed 2005-07-12 devnull static	ulong*	rng_tap = rng_vec;
277 a0d146ed 2005-07-12 devnull static	ulong*	rng_feed = 0;
278 a0d146ed 2005-07-12 devnull 
279 a0d146ed 2005-07-12 devnull static void
280 a0d146ed 2005-07-12 devnull isrand(long seed)
281 a0d146ed 2005-07-12 devnull {
282 a0d146ed 2005-07-12 devnull 	long lo, hi, x;
283 a0d146ed 2005-07-12 devnull 	int i;
284 a0d146ed 2005-07-12 devnull 
285 a0d146ed 2005-07-12 devnull 	rng_tap = rng_vec;
286 a0d146ed 2005-07-12 devnull 	rng_feed = rng_vec+LEN-TAP;
287 a0d146ed 2005-07-12 devnull 	seed = seed%M;
288 a0d146ed 2005-07-12 devnull 	if(seed < 0)
289 a0d146ed 2005-07-12 devnull 		seed += M;
290 a0d146ed 2005-07-12 devnull 	if(seed == 0)
291 a0d146ed 2005-07-12 devnull 		seed = 89482311;
292 a0d146ed 2005-07-12 devnull 	x = seed;
293 a0d146ed 2005-07-12 devnull 	/*
294 a0d146ed 2005-07-12 devnull 	 *	Initialize by x[n+1] = 48271 * x[n] mod (2**31 - 1)
295 a0d146ed 2005-07-12 devnull 	 */
296 a0d146ed 2005-07-12 devnull 	for(i = -20; i < LEN; i++) {
297 a0d146ed 2005-07-12 devnull 		hi = x / Q;
298 a0d146ed 2005-07-12 devnull 		lo = x % Q;
299 a0d146ed 2005-07-12 devnull 		x = A*lo - R*hi;
300 a0d146ed 2005-07-12 devnull 		if(x < 0)
301 a0d146ed 2005-07-12 devnull 			x += M;
302 a0d146ed 2005-07-12 devnull 		if(i >= 0)
303 a0d146ed 2005-07-12 devnull 			rng_vec[i] = x;
304 a0d146ed 2005-07-12 devnull 	}
305 a0d146ed 2005-07-12 devnull }
306 a0d146ed 2005-07-12 devnull 
307 a0d146ed 2005-07-12 devnull void
308 a0d146ed 2005-07-12 devnull xxxsrand(long seed)
309 a0d146ed 2005-07-12 devnull {
310 a0d146ed 2005-07-12 devnull 	isrand(seed);
311 a0d146ed 2005-07-12 devnull }
312 a0d146ed 2005-07-12 devnull 
313 a0d146ed 2005-07-12 devnull long
314 a0d146ed 2005-07-12 devnull xxxlrand(void)
315 a0d146ed 2005-07-12 devnull {
316 a0d146ed 2005-07-12 devnull 	ulong x;
317 a0d146ed 2005-07-12 devnull 
318 a0d146ed 2005-07-12 devnull 	rng_tap--;
319 a0d146ed 2005-07-12 devnull 	if(rng_tap < rng_vec) {
320 a0d146ed 2005-07-12 devnull 		if(rng_feed == 0) {
321 a0d146ed 2005-07-12 devnull 			isrand(1);
322 a0d146ed 2005-07-12 devnull 			rng_tap--;
323 a0d146ed 2005-07-12 devnull 		}
324 a0d146ed 2005-07-12 devnull 		rng_tap += LEN;
325 a0d146ed 2005-07-12 devnull 	}
326 a0d146ed 2005-07-12 devnull 	rng_feed--;
327 a0d146ed 2005-07-12 devnull 	if(rng_feed < rng_vec)
328 a0d146ed 2005-07-12 devnull 		rng_feed += LEN;
329 a0d146ed 2005-07-12 devnull 	x = (*rng_feed + *rng_tap) & MASK;
330 a0d146ed 2005-07-12 devnull 	*rng_feed = x;
331 a0d146ed 2005-07-12 devnull 
332 a0d146ed 2005-07-12 devnull 	return x;
333 a0d146ed 2005-07-12 devnull }
334 a0d146ed 2005-07-12 devnull