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>
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);
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;
26 a0d146ed 2005-07-12 devnull usage(void)
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");
33 a0d146ed 2005-07-12 devnull wr(char *buf, char *buf2)
35 a0d146ed 2005-07-12 devnull uchar score[VtScoreSize], score2[VtScoreSize];
36 a0d146ed 2005-07-12 devnull DigestState ds;
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);
48 a0d146ed 2005-07-12 devnull wrthread(void *v)
53 a0d146ed 2005-07-12 devnull while((p = recvp(cw)) != nil){
54 a0d146ed 2005-07-12 devnull wr(p, nil);
60 a0d146ed 2005-07-12 devnull rd(char *buf, char *buf2)
62 a0d146ed 2005-07-12 devnull uchar score[VtScoreSize];
63 a0d146ed 2005-07-12 devnull DigestState ds;
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);
74 a0d146ed 2005-07-12 devnull rdthread(void *v)
76 a0d146ed 2005-07-12 devnull char *p, *buf2;
78 a0d146ed 2005-07-12 devnull buf2 = vtmalloc(blocksize);
80 a0d146ed 2005-07-12 devnull while((p = recvp(cr)) != nil){
81 a0d146ed 2005-07-12 devnull rd(p, buf2);
86 a0d146ed 2005-07-12 devnull char *template;
89 a0d146ed 2005-07-12 devnull run(void (*fn)(char*, char*), Channel *c)
91 a0d146ed 2005-07-12 devnull int i, t, j, packets;
92 a0d146ed 2005-07-12 devnull char *buf2, *buf;
94 a0d146ed 2005-07-12 devnull buf2 = vtmalloc(blocksize);
95 a0d146ed 2005-07-12 devnull buf = vtmalloc(blocksize);
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;
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];
115 a0d146ed 2005-07-12 devnull sendp(c, buf);
116 a0d146ed 2005-07-12 devnull buf = vtmalloc(blocksize);
118 a0d146ed 2005-07-12 devnull (*fn)(buf, buf2);
119 a0d146ed 2005-07-12 devnull cur += blocksize;
121 a0d146ed 2005-07-12 devnull free(order);
124 a0d146ed 2005-07-12 devnull #define TWID64 ((u64int)~(u64int)0)
127 a0d146ed 2005-07-12 devnull unittoull(char *s)
129 a0d146ed 2005-07-12 devnull char *es;
130 a0d146ed 2005-07-12 devnull u64int n;
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;
138 a0d146ed 2005-07-12 devnull }else if(*es == 'm' || *es == 'M'){
139 a0d146ed 2005-07-12 devnull n *= 1024*1024;
141 a0d146ed 2005-07-12 devnull }else if(*es == 'g' || *es == 'G'){
142 a0d146ed 2005-07-12 devnull n *= 1024*1024*1024;
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;
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;
154 a0d146ed 2005-07-12 devnull threadmain(int argc, char *argv[])
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;
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);
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()));
174 a0d146ed 2005-07-12 devnull case 'h':
175 a0d146ed 2005-07-12 devnull host = EARGF(usage());
177 a0d146ed 2005-07-12 devnull case 'M':
178 a0d146ed 2005-07-12 devnull maxpackets = unittoull(EARGF(usage()));
180 a0d146ed 2005-07-12 devnull case 'm':
181 a0d146ed 2005-07-12 devnull multi = atoi(EARGF(usage()));
183 a0d146ed 2005-07-12 devnull case 'n':
184 a0d146ed 2005-07-12 devnull totalbytes = unittoull(EARGF(usage()));
186 a0d146ed 2005-07-12 devnull case 'p':
187 a0d146ed 2005-07-12 devnull randpct = atoi(EARGF(usage()));
189 a0d146ed 2005-07-12 devnull case 'P':
190 a0d146ed 2005-07-12 devnull permute = 1;
192 a0d146ed 2005-07-12 devnull case 'S':
193 a0d146ed 2005-07-12 devnull doublecheck = 0;
194 a0d146ed 2005-07-12 devnull ventidoublechecksha1 = 0;
196 a0d146ed 2005-07-12 devnull case 's':
197 a0d146ed 2005-07-12 devnull seed = atoi(EARGF(usage()));
199 a0d146ed 2005-07-12 devnull case 'r':
200 a0d146ed 2005-07-12 devnull doread = 1;
202 a0d146ed 2005-07-12 devnull case 'w':
203 a0d146ed 2005-07-12 devnull dowrite = 1;
205 a0d146ed 2005-07-12 devnull case 'V':
206 a0d146ed 2005-07-12 devnull chattyventi++;
208 a0d146ed 2005-07-12 devnull default:
209 a0d146ed 2005-07-12 devnull usage();
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;
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");
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);
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);
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);
257 a0d146ed 2005-07-12 devnull threadexitsall(nil);
262 a0d146ed 2005-07-12 devnull * algorithm by
263 a0d146ed 2005-07-12 devnull * D. P. Mitchell & J. A. Reeds
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))
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;
279 a0d146ed 2005-07-12 devnull static void
280 a0d146ed 2005-07-12 devnull isrand(long seed)
282 a0d146ed 2005-07-12 devnull long lo, hi, x;
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;
294 a0d146ed 2005-07-12 devnull * Initialize by x[n+1] = 48271 * x[n] mod (2**31 - 1)
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)
302 a0d146ed 2005-07-12 devnull if(i >= 0)
303 a0d146ed 2005-07-12 devnull rng_vec[i] = x;
308 a0d146ed 2005-07-12 devnull xxxsrand(long seed)
310 a0d146ed 2005-07-12 devnull isrand(seed);
314 a0d146ed 2005-07-12 devnull xxxlrand(void)
316 a0d146ed 2005-07-12 devnull ulong x;
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--;
324 a0d146ed 2005-07-12 devnull rng_tap += LEN;
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;
332 a0d146ed 2005-07-12 devnull return x;