1 2277c5d7 2004-03-21 devnull #include "std.h"
2 2277c5d7 2004-03-21 devnull #include "dat.h"
4 2277c5d7 2004-03-21 devnull Conv *conv;
6 2277c5d7 2004-03-21 devnull ulong taggen = 1;
9 2277c5d7 2004-03-21 devnull convalloc(char *sysuser)
13 2277c5d7 2004-03-21 devnull c = mallocz(sizeof(Conv), 1);
14 2277c5d7 2004-03-21 devnull if(c == nil)
15 2277c5d7 2004-03-21 devnull return nil;
16 2277c5d7 2004-03-21 devnull c->ref = 1;
17 2277c5d7 2004-03-21 devnull c->tag = taggen++;
18 2277c5d7 2004-03-21 devnull c->next = conv;
19 2277c5d7 2004-03-21 devnull c->sysuser = estrdup(sysuser);
20 2277c5d7 2004-03-21 devnull c->state = "nascent";
21 2277c5d7 2004-03-21 devnull c->rpcwait = chancreate(sizeof(void*), 0);
22 2277c5d7 2004-03-21 devnull c->keywait = chancreate(sizeof(void*), 0);
23 2277c5d7 2004-03-21 devnull strcpy(c->err, "protocol has not started");
24 2277c5d7 2004-03-21 devnull conv = c;
25 2277c5d7 2004-03-21 devnull convreset(c);
26 2277c5d7 2004-03-21 devnull return c;
30 2277c5d7 2004-03-21 devnull convreset(Conv *c)
32 2277c5d7 2004-03-21 devnull if(c->ref != 1){
33 2277c5d7 2004-03-21 devnull c->hangup = 1;
34 2277c5d7 2004-03-21 devnull nbsendp(c->rpcwait, 0);
35 2277c5d7 2004-03-21 devnull while(c->ref > 1)
37 2277c5d7 2004-03-21 devnull c->hangup = 0;
39 2277c5d7 2004-03-21 devnull c->state = "nascent";
40 2277c5d7 2004-03-21 devnull c->err[0] = '\0';
41 2277c5d7 2004-03-21 devnull freeattr(c->attr);
42 2277c5d7 2004-03-21 devnull c->attr = nil;
43 2277c5d7 2004-03-21 devnull c->proto = nil;
44 2277c5d7 2004-03-21 devnull c->rpc.op = 0;
45 2277c5d7 2004-03-21 devnull c->active = 0;
46 2277c5d7 2004-03-21 devnull c->done = 0;
47 2277c5d7 2004-03-21 devnull c->hangup = 0;
51 2277c5d7 2004-03-21 devnull convhangup(Conv *c)
53 2277c5d7 2004-03-21 devnull c->hangup = 1;
54 2277c5d7 2004-03-21 devnull c->rpc.op = 0;
55 2277c5d7 2004-03-21 devnull (*c->kickreply)(c);
56 2277c5d7 2004-03-21 devnull nbsendp(c->rpcwait, 0);
60 2277c5d7 2004-03-21 devnull convclose(Conv *c)
64 2277c5d7 2004-03-21 devnull if(c == nil)
67 2277c5d7 2004-03-21 devnull if(--c->ref > 0)
70 2277c5d7 2004-03-21 devnull if(c == conv){
71 2277c5d7 2004-03-21 devnull conv = c->next;
72 2277c5d7 2004-03-21 devnull goto free;
74 2277c5d7 2004-03-21 devnull for(p=conv; p && p->next!=c; p=p->next)
76 2277c5d7 2004-03-21 devnull if(p == nil){
77 2277c5d7 2004-03-21 devnull print("cannot find conv in list\n");
80 2277c5d7 2004-03-21 devnull p->next = c->next;
83 2277c5d7 2004-03-21 devnull c->next = nil;
87 2277c5d7 2004-03-21 devnull static Rpc*
88 2277c5d7 2004-03-21 devnull convgetrpc(Conv *c, int want)
91 2277c5d7 2004-03-21 devnull if(c->hangup){
92 2277c5d7 2004-03-21 devnull werrstr("hangup");
93 2277c5d7 2004-03-21 devnull return nil;
95 2277c5d7 2004-03-21 devnull if(c->rpc.op == RpcUnknown){
96 2277c5d7 2004-03-21 devnull recvp(c->rpcwait);
97 2277c5d7 2004-03-21 devnull if(c->hangup){
98 2277c5d7 2004-03-21 devnull werrstr("hangup");
99 2277c5d7 2004-03-21 devnull return nil;
101 2277c5d7 2004-03-21 devnull if(c->rpc.op == RpcUnknown)
102 2277c5d7 2004-03-21 devnull continue;
104 2277c5d7 2004-03-21 devnull if(want < 0 || c->rpc.op == want)
105 2277c5d7 2004-03-21 devnull return &c->rpc;
106 2277c5d7 2004-03-21 devnull rpcrespond(c, "phase in state '%s' want '%s'", c->state, rpcname[want]);
108 2277c5d7 2004-03-21 devnull return nil; /* not reached */
111 2277c5d7 2004-03-21 devnull /* read until the done function tells us that's enough */
113 2277c5d7 2004-03-21 devnull convreadfn(Conv *c, int (*done)(void*, int), char **ps)
117 2277c5d7 2004-03-21 devnull char *s;
119 2277c5d7 2004-03-21 devnull for(;;){
120 2277c5d7 2004-03-21 devnull r = convgetrpc(c, RpcWrite);
121 2277c5d7 2004-03-21 devnull if(r == nil)
122 2277c5d7 2004-03-21 devnull return -1;
123 2277c5d7 2004-03-21 devnull n = (*done)(r->data, r->count);
124 2277c5d7 2004-03-21 devnull if(n == r->count)
126 2277c5d7 2004-03-21 devnull rpcrespond(c, "toosmall %d", n);
129 2277c5d7 2004-03-21 devnull s = emalloc(r->count+1);
130 2277c5d7 2004-03-21 devnull memmove(s, r->data, r->count);
131 2277c5d7 2004-03-21 devnull s[r->count] = 0;
132 2277c5d7 2004-03-21 devnull *ps = s;
133 2277c5d7 2004-03-21 devnull rpcrespond(c, "ok");
134 2277c5d7 2004-03-21 devnull return r->count;
138 2277c5d7 2004-03-21 devnull * read until we get a non-zero write. assumes remote side
139 2277c5d7 2004-03-21 devnull * knows something about the protocol (is not auth_proxy).
140 2277c5d7 2004-03-21 devnull * the remote side typically won't bother with the zero-length
141 2277c5d7 2004-03-21 devnull * write to find out the length -- the loop is there only so the
142 2277c5d7 2004-03-21 devnull * test program can call auth_proxy on both sides of a pipe
143 2277c5d7 2004-03-21 devnull * to play a conversation.
146 2277c5d7 2004-03-21 devnull convreadm(Conv *c, char **ps)
148 2277c5d7 2004-03-21 devnull char *s;
151 2277c5d7 2004-03-21 devnull for(;;){
152 2277c5d7 2004-03-21 devnull r = convgetrpc(c, RpcWrite);
153 2277c5d7 2004-03-21 devnull if(r == nil)
154 2277c5d7 2004-03-21 devnull return -1;
155 2277c5d7 2004-03-21 devnull if(r->count > 0)
157 2277c5d7 2004-03-21 devnull rpcrespond(c, "toosmall %d", AuthRpcMax);
159 2277c5d7 2004-03-21 devnull s = emalloc(r->count+1);
160 2277c5d7 2004-03-21 devnull memmove(s, r->data, r->count);
161 2277c5d7 2004-03-21 devnull s[r->count] = 0;
162 2277c5d7 2004-03-21 devnull *ps = s;
163 2277c5d7 2004-03-21 devnull rpcrespond(c, "ok");
164 2277c5d7 2004-03-21 devnull return r->count;
167 2277c5d7 2004-03-21 devnull /* read exactly count bytes */
169 2277c5d7 2004-03-21 devnull convread(Conv *c, void *data, int count)
173 2277c5d7 2004-03-21 devnull for(;;){
174 2277c5d7 2004-03-21 devnull r = convgetrpc(c, RpcWrite);
175 2277c5d7 2004-03-21 devnull if(r == nil)
176 2277c5d7 2004-03-21 devnull return -1;
177 2277c5d7 2004-03-21 devnull if(r->count == count)
179 2277c5d7 2004-03-21 devnull if(r->count < count)
180 2277c5d7 2004-03-21 devnull rpcrespond(c, "toosmall %d", count);
182 2277c5d7 2004-03-21 devnull rpcrespond(c, "error too much data; want %d got %d", count, r->count);
184 2277c5d7 2004-03-21 devnull memmove(data, r->data, count);
185 2277c5d7 2004-03-21 devnull rpcrespond(c, "ok");
186 2277c5d7 2004-03-21 devnull return 0;
189 2277c5d7 2004-03-21 devnull /* write exactly count bytes */
191 2277c5d7 2004-03-21 devnull convwrite(Conv *c, void *data, int count)
195 2277c5d7 2004-03-21 devnull for(;;){
196 2277c5d7 2004-03-21 devnull r = convgetrpc(c, RpcRead);
197 2277c5d7 2004-03-21 devnull if(r == nil)
198 2277c5d7 2004-03-21 devnull return -1;
201 2277c5d7 2004-03-21 devnull rpcrespondn(c, "ok", data, count);
202 2277c5d7 2004-03-21 devnull return 0;
205 2277c5d7 2004-03-21 devnull /* print to the conversation */
207 2277c5d7 2004-03-21 devnull convprint(Conv *c, char *fmt, ...)
209 2277c5d7 2004-03-21 devnull char *s;
210 2277c5d7 2004-03-21 devnull va_list arg;
211 2277c5d7 2004-03-21 devnull int ret;
213 2277c5d7 2004-03-21 devnull va_start(arg, fmt);
214 2277c5d7 2004-03-21 devnull s = vsmprint(fmt, arg);
215 2277c5d7 2004-03-21 devnull va_end(arg);
216 2277c5d7 2004-03-21 devnull if(s == nil)
217 2277c5d7 2004-03-21 devnull return -1;
218 2277c5d7 2004-03-21 devnull ret = convwrite(c, s, strlen(s));
219 2277c5d7 2004-03-21 devnull free(s);
220 2277c5d7 2004-03-21 devnull return ret;
223 2277c5d7 2004-03-21 devnull /* ask for a key */
225 2277c5d7 2004-03-21 devnull convneedkey(Conv *c, Attr *a)
228 2277c5d7 2004-03-21 devnull * Piggyback key requests in the usual RPC channel.
229 2277c5d7 2004-03-21 devnull * Wait for the next RPC and then send a key request
230 2277c5d7 2004-03-21 devnull * in response. The keys get added out-of-band (via the
231 2277c5d7 2004-03-21 devnull * ctl file), so assume the key has been added when the
232 2277c5d7 2004-03-21 devnull * next request comes in.
234 2277c5d7 2004-03-21 devnull if(convgetrpc(c, -1) == nil)
235 2277c5d7 2004-03-21 devnull return -1;
236 2277c5d7 2004-03-21 devnull rpcrespond(c, "needkey %A", a);
237 2277c5d7 2004-03-21 devnull if(convgetrpc(c, -1) == nil)
238 2277c5d7 2004-03-21 devnull return -1;
239 2277c5d7 2004-03-21 devnull return 0;
242 2277c5d7 2004-03-21 devnull /* ask for a replacement for a bad key*/
244 2277c5d7 2004-03-21 devnull convbadkey(Conv *c, Key *k, char *msg, Attr *a)
246 2277c5d7 2004-03-21 devnull if(convgetrpc(c, -1) == nil)
247 2277c5d7 2004-03-21 devnull return -1;
248 2277c5d7 2004-03-21 devnull rpcrespond(c, "badkey %A %N\n%s\n%A",
249 2277c5d7 2004-03-21 devnull k->attr, k->privattr, msg, a);
250 2277c5d7 2004-03-21 devnull if(convgetrpc(c, -1) == nil)
251 2277c5d7 2004-03-21 devnull return -1;
252 2277c5d7 2004-03-21 devnull return 0;