1 #include <u.h>
2 #include <libc.h>
3 #include <auth.h>
4 
5 typedef struct Test Test;
6 
7 struct Test
8 {
9 	char *name;
10 	int (*server)(Test*, AuthRpc*, int);
11 	int (*client)(Test*, int);
12 };
13 
14 int
15 ai2status(AuthInfo *ai)
16 {
17 	if(ai == nil)
18 		return -1;
19 	auth_freeAI(ai);
20 	return 0;
21 }
22 
23 int
24 proxyserver(Test *t, AuthRpc *rpc, int fd)
25 {
26 	char buf[1024];
27 
28 	sprint(buf, "proto=%q role=server", t->name);
29 	return ai2status(fauth_proxy(fd, rpc, nil, buf));
30 }
31 
32 int
33 proxyclient(Test *t, int fd)
34 {
35 	return ai2status(auth_proxy(fd, auth_getkey, "proto=%q role=client", t->name));
36 }
37 
38 Test test[] =
39 {
40 	"apop",		proxyserver,		proxyclient,
41 	"cram",		proxyserver,		proxyclient,
42 	"p9sk1",		proxyserver,		proxyclient,
43 	"p9sk2",		proxyserver,		proxyclient,
44 	"p9any",		proxyserver,		proxyclient
45 };
46 
47 void
48 usage(void)
49 {
50 	fprint(2, "usage: test [name]...\n");
51 	exits("usage");
52 }
53 
54 void
55 runtest(AuthRpc *srpc, Test *t)
56 {
57 	int p[2], bad;
58 	Waitmsg *w;
59 
60 	if(pipe(p) < 0)
61 		sysfatal("pipe: %r");
62 
63 	print("%s...", t->name);
64 
65 	switch(fork()){
66 	case -1:
67 		sysfatal("fork: %r");
68 
69 	case 0:
70 		close(p[0]);
71 		if((*t->server)(t, srpc, p[1]) < 0){
72 			print("\n\tserver: %r");
73 			_exits("oops");
74 		}
75 		close(p[1]);
76 		_exits(nil);
77 	default:
78 		close(p[1]);
79 		if((*t->client)(t, p[0]) < 0){
80 			print("\n\tclient: %r");
81 			bad = 1;
82 		}
83 		close(p[0]);
84 		break;
85 	}
86 	w = wait();
87 	if(w->msg[0])
88 		bad = 1;
89 	print("\n");
90 }
91 
92 void
93 main(int argc, char **argv)
94 {
95 	int i, j;
96 	int afd;
97 	AuthRpc *srpc;
98 
99 	ARGBEGIN{
100 	default:
101 		usage();
102 	}ARGEND
103 
104 	quotefmtinstall();
105 	afd = open("/n/kremvax/factotum/rpc", ORDWR);
106 	if(afd < 0)
107 		sysfatal("open /n/kremvax/factotum/rpc: %r");
108 	srpc = auth_allocrpc(afd);
109 	if(srpc == nil)
110 		sysfatal("auth_allocrpc: %r");
111 
112 	if(argc == 0)
113 		for(i=0; i<nelem(test); i++)
114 			runtest(srpc, &test[i]);
115 	else
116 		for(i=0; i<argc; i++)
117 			for(j=0; j<nelem(test); j++)
118 				if(strcmp(argv[i], test[j].name) == 0)
119 					runtest(srpc, &test[j]);
120 	exits(nil);
121 }