121 lines
1.9 KiB
C
121 lines
1.9 KiB
C
#include <u.h>
|
|
#include <libc.h>
|
|
#include <auth.h>
|
|
|
|
typedef struct Test Test;
|
|
|
|
struct Test
|
|
{
|
|
char *name;
|
|
int (*server)(Test*, AuthRpc*, int);
|
|
int (*client)(Test*, int);
|
|
};
|
|
|
|
int
|
|
ai2status(AuthInfo *ai)
|
|
{
|
|
if(ai == nil)
|
|
return -1;
|
|
auth_freeAI(ai);
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
proxyserver(Test *t, AuthRpc *rpc, int fd)
|
|
{
|
|
char buf[1024];
|
|
|
|
sprint(buf, "proto=%q role=server", t->name);
|
|
return ai2status(fauth_proxy(fd, rpc, nil, buf));
|
|
}
|
|
|
|
int
|
|
proxyclient(Test *t, int fd)
|
|
{
|
|
return ai2status(auth_proxy(fd, auth_getkey, "proto=%q role=client", t->name));
|
|
}
|
|
|
|
Test test[] =
|
|
{
|
|
"apop", proxyserver, proxyclient,
|
|
"cram", proxyserver, proxyclient,
|
|
"p9sk1", proxyserver, proxyclient,
|
|
"p9sk2", proxyserver, proxyclient,
|
|
"p9any", proxyserver, proxyclient,
|
|
};
|
|
|
|
void
|
|
usage(void)
|
|
{
|
|
fprint(2, "usage: test [name]...\n");
|
|
exits("usage");
|
|
}
|
|
|
|
void
|
|
runtest(AuthRpc *srpc, Test *t)
|
|
{
|
|
int p[2], bad;
|
|
Waitmsg *w;
|
|
|
|
if(pipe(p) < 0)
|
|
sysfatal("pipe: %r");
|
|
|
|
print("%s...", t->name);
|
|
|
|
switch(fork()){
|
|
case -1:
|
|
sysfatal("fork: %r");
|
|
|
|
case 0:
|
|
close(p[0]);
|
|
if((*t->server)(t, srpc, p[1]) < 0){
|
|
print("\n\tserver: %r");
|
|
_exits("oops");
|
|
}
|
|
close(p[1]);
|
|
_exits(nil);
|
|
default:
|
|
close(p[1]);
|
|
if((*t->client)(t, p[0]) < 0){
|
|
print("\n\tclient: %r");
|
|
bad = 1;
|
|
}
|
|
close(p[0]);
|
|
break;
|
|
}
|
|
w = wait();
|
|
if(w->msg[0])
|
|
bad = 1;
|
|
print("\n");
|
|
}
|
|
|
|
void
|
|
main(int argc, char **argv)
|
|
{
|
|
int i, j;
|
|
int afd;
|
|
AuthRpc *srpc;
|
|
|
|
ARGBEGIN{
|
|
default:
|
|
usage();
|
|
}ARGEND
|
|
|
|
quotefmtinstall();
|
|
afd = open("/n/kremvax/factotum/rpc", ORDWR);
|
|
if(afd < 0)
|
|
sysfatal("open /n/kremvax/factotum/rpc: %r");
|
|
srpc = auth_allocrpc(afd);
|
|
if(srpc == nil)
|
|
sysfatal("auth_allocrpc: %r");
|
|
|
|
if(argc == 0)
|
|
for(i=0; i<nelem(test); i++)
|
|
runtest(srpc, &test[i]);
|
|
else
|
|
for(i=0; i<argc; i++)
|
|
for(j=0; j<nelem(test); j++)
|
|
if(strcmp(argv[i], test[j].name) == 0)
|
|
runtest(srpc, &test[j]);
|
|
exits(nil);
|
|
}
|