1*8ccd4a63SDavid du Colombier #include <u.h>
2*8ccd4a63SDavid du Colombier #include <libc.h>
3*8ccd4a63SDavid du Colombier #include <auth.h>
4*8ccd4a63SDavid du Colombier #include <authsrv.h>
5*8ccd4a63SDavid du Colombier #include "authlocal.h"
6*8ccd4a63SDavid du Colombier
7*8ccd4a63SDavid du Colombier Chalstate*
auth_challenge(char * fmt,...)8*8ccd4a63SDavid du Colombier auth_challenge(char *fmt, ...)
9*8ccd4a63SDavid du Colombier {
10*8ccd4a63SDavid du Colombier char *p;
11*8ccd4a63SDavid du Colombier va_list arg;
12*8ccd4a63SDavid du Colombier Chalstate *c;
13*8ccd4a63SDavid du Colombier
14*8ccd4a63SDavid du Colombier quotefmtinstall(); /* just in case */
15*8ccd4a63SDavid du Colombier va_start(arg, fmt);
16*8ccd4a63SDavid du Colombier p = vsmprint(fmt, arg);
17*8ccd4a63SDavid du Colombier va_end(arg);
18*8ccd4a63SDavid du Colombier if(p == nil)
19*8ccd4a63SDavid du Colombier return nil;
20*8ccd4a63SDavid du Colombier
21*8ccd4a63SDavid du Colombier c = mallocz(sizeof(*c), 1);
22*8ccd4a63SDavid du Colombier if(c == nil){
23*8ccd4a63SDavid du Colombier free(p);
24*8ccd4a63SDavid du Colombier return nil;
25*8ccd4a63SDavid du Colombier }
26*8ccd4a63SDavid du Colombier
27*8ccd4a63SDavid du Colombier if((c->afd = open("/mnt/factotum/rpc", ORDWR)) < 0){
28*8ccd4a63SDavid du Colombier Error:
29*8ccd4a63SDavid du Colombier auth_freechal(c);
30*8ccd4a63SDavid du Colombier free(p);
31*8ccd4a63SDavid du Colombier return nil;
32*8ccd4a63SDavid du Colombier }
33*8ccd4a63SDavid du Colombier
34*8ccd4a63SDavid du Colombier if((c->rpc=auth_allocrpc(c->afd)) == nil
35*8ccd4a63SDavid du Colombier || auth_rpc(c->rpc, "start", p, strlen(p)) != ARok
36*8ccd4a63SDavid du Colombier || auth_rpc(c->rpc, "read", nil, 0) != ARok)
37*8ccd4a63SDavid du Colombier goto Error;
38*8ccd4a63SDavid du Colombier
39*8ccd4a63SDavid du Colombier if(c->rpc->narg > sizeof(c->chal)-1){
40*8ccd4a63SDavid du Colombier werrstr("buffer too small for challenge");
41*8ccd4a63SDavid du Colombier goto Error;
42*8ccd4a63SDavid du Colombier }
43*8ccd4a63SDavid du Colombier memmove(c->chal, c->rpc->arg, c->rpc->narg);
44*8ccd4a63SDavid du Colombier c->nchal = c->rpc->narg;
45*8ccd4a63SDavid du Colombier free(p);
46*8ccd4a63SDavid du Colombier return c;
47*8ccd4a63SDavid du Colombier }
48*8ccd4a63SDavid du Colombier
49*8ccd4a63SDavid du Colombier AuthInfo*
auth_response(Chalstate * c)50*8ccd4a63SDavid du Colombier auth_response(Chalstate *c)
51*8ccd4a63SDavid du Colombier {
52*8ccd4a63SDavid du Colombier int ret;
53*8ccd4a63SDavid du Colombier AuthInfo *ai;
54*8ccd4a63SDavid du Colombier
55*8ccd4a63SDavid du Colombier ai = nil;
56*8ccd4a63SDavid du Colombier if(c->afd < 0){
57*8ccd4a63SDavid du Colombier werrstr("auth_response: connection not open");
58*8ccd4a63SDavid du Colombier return nil;
59*8ccd4a63SDavid du Colombier }
60*8ccd4a63SDavid du Colombier if(c->resp == nil){
61*8ccd4a63SDavid du Colombier werrstr("auth_response: nil response");
62*8ccd4a63SDavid du Colombier return nil;
63*8ccd4a63SDavid du Colombier }
64*8ccd4a63SDavid du Colombier if(c->nresp == 0){
65*8ccd4a63SDavid du Colombier werrstr("auth_response: unspecified response length");
66*8ccd4a63SDavid du Colombier return nil;
67*8ccd4a63SDavid du Colombier }
68*8ccd4a63SDavid du Colombier
69*8ccd4a63SDavid du Colombier if(c->user){
70*8ccd4a63SDavid du Colombier if(auth_rpc(c->rpc, "write", c->user, strlen(c->user)) != ARok){
71*8ccd4a63SDavid du Colombier /*
72*8ccd4a63SDavid du Colombier * if this fails we're out of phase with factotum.
73*8ccd4a63SDavid du Colombier * give up.
74*8ccd4a63SDavid du Colombier */
75*8ccd4a63SDavid du Colombier goto Out;
76*8ccd4a63SDavid du Colombier }
77*8ccd4a63SDavid du Colombier }
78*8ccd4a63SDavid du Colombier
79*8ccd4a63SDavid du Colombier if(auth_rpc(c->rpc, "write", c->resp, c->nresp) != ARok){
80*8ccd4a63SDavid du Colombier /*
81*8ccd4a63SDavid du Colombier * don't close the connection -- maybe we'll try again.
82*8ccd4a63SDavid du Colombier */
83*8ccd4a63SDavid du Colombier return nil;
84*8ccd4a63SDavid du Colombier }
85*8ccd4a63SDavid du Colombier
86*8ccd4a63SDavid du Colombier switch(ret = auth_rpc(c->rpc, "read", nil, 0)){
87*8ccd4a63SDavid du Colombier case ARok:
88*8ccd4a63SDavid du Colombier default:
89*8ccd4a63SDavid du Colombier werrstr("factotum protocol botch %d %s", ret, c->rpc->ibuf);
90*8ccd4a63SDavid du Colombier break;
91*8ccd4a63SDavid du Colombier case ARdone:
92*8ccd4a63SDavid du Colombier ai = auth_getinfo(c->rpc);
93*8ccd4a63SDavid du Colombier break;
94*8ccd4a63SDavid du Colombier }
95*8ccd4a63SDavid du Colombier
96*8ccd4a63SDavid du Colombier Out:
97*8ccd4a63SDavid du Colombier close(c->afd);
98*8ccd4a63SDavid du Colombier auth_freerpc(c->rpc);
99*8ccd4a63SDavid du Colombier c->afd = -1;
100*8ccd4a63SDavid du Colombier c->rpc = nil;
101*8ccd4a63SDavid du Colombier return ai;
102*8ccd4a63SDavid du Colombier }
103*8ccd4a63SDavid du Colombier
104*8ccd4a63SDavid du Colombier void
auth_freechal(Chalstate * c)105*8ccd4a63SDavid du Colombier auth_freechal(Chalstate *c)
106*8ccd4a63SDavid du Colombier {
107*8ccd4a63SDavid du Colombier if(c == nil)
108*8ccd4a63SDavid du Colombier return;
109*8ccd4a63SDavid du Colombier
110*8ccd4a63SDavid du Colombier if(c->afd >= 0)
111*8ccd4a63SDavid du Colombier close(c->afd);
112*8ccd4a63SDavid du Colombier if(c->rpc != nil)
113*8ccd4a63SDavid du Colombier auth_freerpc(c->rpc);
114*8ccd4a63SDavid du Colombier
115*8ccd4a63SDavid du Colombier memset(c, 0xBB, sizeof(*c));
116*8ccd4a63SDavid du Colombier free(c);
117*8ccd4a63SDavid du Colombier }
118