return of venti
This commit is contained in:
rsc
parent
88bb285e3d
commit
a0d146edd7
68 changed files with 14443 additions and 2 deletions
334
src/cmd/venti/randtest.c
Normal file
334
src/cmd/venti/randtest.c
Normal file
|
|
@ -0,0 +1,334 @@
|
|||
#include <u.h>
|
||||
#include <libc.h>
|
||||
#include <venti.h>
|
||||
#include <libsec.h>
|
||||
#include <thread.h>
|
||||
|
||||
|
||||
enum { STACK = 32768 };
|
||||
void xxxsrand(long);
|
||||
long xxxlrand(void);
|
||||
|
||||
Channel *cw;
|
||||
Channel *cr;
|
||||
char *host;
|
||||
int blocksize, seed, randpct;
|
||||
int doread, dowrite, packets, permute;
|
||||
vlong totalbytes, cur;
|
||||
VtConn *z;
|
||||
int multi;
|
||||
int maxpackets;
|
||||
int sequence;
|
||||
int doublecheck = 1;
|
||||
uint *order;
|
||||
|
||||
void
|
||||
usage(void)
|
||||
{
|
||||
fprint(2, "usage: randtest [-q] [-h host] [-s seed] [-b blocksize] [-p randpct] [-n totalbytes] [-M maxblocks] [-P] [-r] [-w]\n");
|
||||
threadexitsall("usage");
|
||||
}
|
||||
|
||||
void
|
||||
wr(char *buf, char *buf2)
|
||||
{
|
||||
uchar score[VtScoreSize], score2[VtScoreSize];
|
||||
DigestState ds;
|
||||
|
||||
memset(&ds, 0, sizeof ds);
|
||||
if(doublecheck)
|
||||
sha1((uchar*)buf, blocksize, score, &ds);
|
||||
if(vtwrite(z, score2, VtDataType, (uchar*)buf, blocksize) < 0)
|
||||
sysfatal("vtwrite %V at %,lld: %r", score, cur);
|
||||
if(doublecheck && memcmp(score, score2, VtScoreSize) != 0)
|
||||
sysfatal("score mismatch! %V %V", score, score2);
|
||||
}
|
||||
|
||||
void
|
||||
wrthread(void *v)
|
||||
{
|
||||
char *p;
|
||||
|
||||
USED(v);
|
||||
while((p = recvp(cw)) != nil){
|
||||
wr(p, nil);
|
||||
free(p);
|
||||
}
|
||||
}
|
||||
|
||||
void
|
||||
rd(char *buf, char *buf2)
|
||||
{
|
||||
uchar score[VtScoreSize];
|
||||
DigestState ds;
|
||||
|
||||
memset(&ds, 0, sizeof ds);
|
||||
sha1((uchar*)buf, blocksize, score, &ds);
|
||||
if(vtread(z, score, VtDataType, (uchar*)buf2, blocksize) < 0)
|
||||
sysfatal("vtread %V at %,lld: %r", score, cur);
|
||||
if(memcmp(buf, buf2, blocksize) != 0)
|
||||
sysfatal("bad data read! %V", score);
|
||||
}
|
||||
|
||||
void
|
||||
rdthread(void *v)
|
||||
{
|
||||
char *p, *buf2;
|
||||
|
||||
buf2 = vtmalloc(blocksize);
|
||||
USED(v);
|
||||
while((p = recvp(cr)) != nil){
|
||||
rd(p, buf2);
|
||||
free(p);
|
||||
}
|
||||
}
|
||||
|
||||
char *template;
|
||||
|
||||
void
|
||||
run(void (*fn)(char*, char*), Channel *c)
|
||||
{
|
||||
int i, t, j, packets;
|
||||
char *buf2, *buf;
|
||||
|
||||
buf2 = vtmalloc(blocksize);
|
||||
buf = vtmalloc(blocksize);
|
||||
cur = 0;
|
||||
packets = totalbytes/blocksize;
|
||||
if(maxpackets == 0)
|
||||
maxpackets = packets;
|
||||
order = vtmalloc(packets*sizeof order[0]);
|
||||
for(i=0; i<packets; i++)
|
||||
order[i] = i;
|
||||
if(permute){
|
||||
for(i=1; i<packets; i++){
|
||||
j = nrand(i+1);
|
||||
t = order[i];
|
||||
order[i] = order[j];
|
||||
order[j] = t;
|
||||
}
|
||||
}
|
||||
for(i=0; i<packets && i<maxpackets; i++){
|
||||
memmove(buf, template, blocksize);
|
||||
*(uint*)buf = order[i];
|
||||
if(c){
|
||||
sendp(c, buf);
|
||||
buf = vtmalloc(blocksize);
|
||||
}else
|
||||
(*fn)(buf, buf2);
|
||||
cur += blocksize;
|
||||
}
|
||||
free(order);
|
||||
}
|
||||
|
||||
#define TWID64 ((u64int)~(u64int)0)
|
||||
|
||||
u64int
|
||||
unittoull(char *s)
|
||||
{
|
||||
char *es;
|
||||
u64int n;
|
||||
|
||||
if(s == nil)
|
||||
return TWID64;
|
||||
n = strtoul(s, &es, 0);
|
||||
if(*es == 'k' || *es == 'K'){
|
||||
n *= 1024;
|
||||
es++;
|
||||
}else if(*es == 'm' || *es == 'M'){
|
||||
n *= 1024*1024;
|
||||
es++;
|
||||
}else if(*es == 'g' || *es == 'G'){
|
||||
n *= 1024*1024*1024;
|
||||
es++;
|
||||
}else if(*es == 't' || *es == 'T'){
|
||||
n *= 1024*1024;
|
||||
n *= 1024*1024;
|
||||
}
|
||||
if(*es != '\0')
|
||||
return TWID64;
|
||||
return n;
|
||||
}
|
||||
|
||||
void
|
||||
threadmain(int argc, char *argv[])
|
||||
{
|
||||
int i, max;
|
||||
vlong t0;
|
||||
double t;
|
||||
|
||||
blocksize = 8192;
|
||||
seed = 0;
|
||||
randpct = 50;
|
||||
host = nil;
|
||||
doread = 0;
|
||||
dowrite = 0;
|
||||
totalbytes = 1*1024*1024*1024;
|
||||
fmtinstall('V', vtscorefmt);
|
||||
fmtinstall('F', vtfcallfmt);
|
||||
|
||||
ARGBEGIN{
|
||||
case 'b':
|
||||
blocksize = unittoull(EARGF(usage()));
|
||||
break;
|
||||
case 'h':
|
||||
host = EARGF(usage());
|
||||
break;
|
||||
case 'M':
|
||||
maxpackets = unittoull(EARGF(usage()));
|
||||
break;
|
||||
case 'm':
|
||||
multi = atoi(EARGF(usage()));
|
||||
break;
|
||||
case 'n':
|
||||
totalbytes = unittoull(EARGF(usage()));
|
||||
break;
|
||||
case 'p':
|
||||
randpct = atoi(EARGF(usage()));
|
||||
break;
|
||||
case 'P':
|
||||
permute = 1;
|
||||
break;
|
||||
case 'S':
|
||||
doublecheck = 0;
|
||||
ventidoublechecksha1 = 0;
|
||||
break;
|
||||
case 's':
|
||||
seed = atoi(EARGF(usage()));
|
||||
break;
|
||||
case 'r':
|
||||
doread = 1;
|
||||
break;
|
||||
case 'w':
|
||||
dowrite = 1;
|
||||
break;
|
||||
case 'V':
|
||||
chattyventi++;
|
||||
break;
|
||||
default:
|
||||
usage();
|
||||
}ARGEND
|
||||
|
||||
if(doread==0 && dowrite==0){
|
||||
doread = 1;
|
||||
dowrite = 1;
|
||||
}
|
||||
|
||||
z = vtdial(host);
|
||||
if(z == nil)
|
||||
sysfatal("could not connect to server: %r");
|
||||
if(vtconnect(z) < 0)
|
||||
sysfatal("vtconnect: %r");
|
||||
|
||||
if(multi){
|
||||
cr = chancreate(sizeof(void*), 0);
|
||||
cw = chancreate(sizeof(void*), 0);
|
||||
for(i=0; i<multi; i++){
|
||||
proccreate(wrthread, nil, STACK);
|
||||
proccreate(rdthread, nil, STACK);
|
||||
}
|
||||
}
|
||||
|
||||
template = vtmalloc(blocksize);
|
||||
xxxsrand(seed);
|
||||
max = (256*randpct)/100;
|
||||
if(max == 0)
|
||||
max = 1;
|
||||
for(i=0; i<blocksize; i++)
|
||||
template[i] = xxxlrand()%max;
|
||||
if(dowrite){
|
||||
t0 = nsec();
|
||||
run(wr, cw);
|
||||
for(i=0; i<multi; i++)
|
||||
sendp(cw, nil);
|
||||
t = (nsec() - t0)/1.e9;
|
||||
print("write: %lld bytes / %.3f seconds = %.6f MB/s\n",
|
||||
totalbytes, t, (double)totalbytes/1e6/t);
|
||||
}
|
||||
if(doread){
|
||||
t0 = nsec();
|
||||
run(rd, cr);
|
||||
for(i=0; i<multi; i++)
|
||||
sendp(cr, nil);
|
||||
t = (nsec() - t0)/1.e9;
|
||||
print("read: %lld bytes / %.3f seconds = %.6f MB/s\n",
|
||||
totalbytes, t, (double)totalbytes/1e6/t);
|
||||
}
|
||||
threadexitsall(nil);
|
||||
}
|
||||
|
||||
|
||||
/*
|
||||
* algorithm by
|
||||
* D. P. Mitchell & J. A. Reeds
|
||||
*/
|
||||
|
||||
#define LEN 607
|
||||
#define TAP 273
|
||||
#define MASK 0x7fffffffL
|
||||
#define A 48271
|
||||
#define M 2147483647
|
||||
#define Q 44488
|
||||
#define R 3399
|
||||
#define NORM (1.0/(1.0+MASK))
|
||||
|
||||
static ulong rng_vec[LEN];
|
||||
static ulong* rng_tap = rng_vec;
|
||||
static ulong* rng_feed = 0;
|
||||
|
||||
static void
|
||||
isrand(long seed)
|
||||
{
|
||||
long lo, hi, x;
|
||||
int i;
|
||||
|
||||
rng_tap = rng_vec;
|
||||
rng_feed = rng_vec+LEN-TAP;
|
||||
seed = seed%M;
|
||||
if(seed < 0)
|
||||
seed += M;
|
||||
if(seed == 0)
|
||||
seed = 89482311;
|
||||
x = seed;
|
||||
/*
|
||||
* Initialize by x[n+1] = 48271 * x[n] mod (2**31 - 1)
|
||||
*/
|
||||
for(i = -20; i < LEN; i++) {
|
||||
hi = x / Q;
|
||||
lo = x % Q;
|
||||
x = A*lo - R*hi;
|
||||
if(x < 0)
|
||||
x += M;
|
||||
if(i >= 0)
|
||||
rng_vec[i] = x;
|
||||
}
|
||||
}
|
||||
|
||||
void
|
||||
xxxsrand(long seed)
|
||||
{
|
||||
isrand(seed);
|
||||
}
|
||||
|
||||
long
|
||||
xxxlrand(void)
|
||||
{
|
||||
ulong x;
|
||||
|
||||
rng_tap--;
|
||||
if(rng_tap < rng_vec) {
|
||||
if(rng_feed == 0) {
|
||||
isrand(1);
|
||||
rng_tap--;
|
||||
}
|
||||
rng_tap += LEN;
|
||||
}
|
||||
rng_feed--;
|
||||
if(rng_feed < rng_vec)
|
||||
rng_feed += LEN;
|
||||
x = (*rng_feed + *rng_tap) & MASK;
|
||||
*rng_feed = x;
|
||||
|
||||
return x;
|
||||
}
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue