1 6e527fbc 2005-02-13 devnull #include "std.h"
  2 6e527fbc 2005-02-13 devnull #include "dat.h"
  3 6e527fbc 2005-02-13 devnull 
  4 6e527fbc 2005-02-13 devnull Conv *conv;
  5 6e527fbc 2005-02-13 devnull 
  6 6e527fbc 2005-02-13 devnull ulong taggen = 1;
  7 6e527fbc 2005-02-13 devnull 
  8 6e527fbc 2005-02-13 devnull Conv*
  9 6e527fbc 2005-02-13 devnull convalloc(char *sysuser)
 10 6e527fbc 2005-02-13 devnull {
 11 6e527fbc 2005-02-13 devnull 	Conv *c;
 12 6e527fbc 2005-02-13 devnull 
 13 6e527fbc 2005-02-13 devnull 	c = mallocz(sizeof(Conv), 1);
 14 6e527fbc 2005-02-13 devnull 	if(c == nil)
 15 6e527fbc 2005-02-13 devnull 		return nil;
 16 6e527fbc 2005-02-13 devnull 	c->ref = 1;
 17 6e527fbc 2005-02-13 devnull 	c->tag = taggen++;
 18 6e527fbc 2005-02-13 devnull 	c->next = conv;
 19 6e527fbc 2005-02-13 devnull 	c->sysuser = estrdup(sysuser);
 20 6e527fbc 2005-02-13 devnull 	c->state = "nascent";
 21 6e527fbc 2005-02-13 devnull 	c->rpcwait = chancreate(sizeof(void*), 0);
 22 6e527fbc 2005-02-13 devnull 	c->keywait = chancreate(sizeof(void*), 0);
 23 6e527fbc 2005-02-13 devnull 	strcpy(c->err, "protocol has not started");
 24 6e527fbc 2005-02-13 devnull 	conv = c;
 25 6e527fbc 2005-02-13 devnull 	convreset(c);
 26 6e527fbc 2005-02-13 devnull 	return c;
 27 6e527fbc 2005-02-13 devnull }
 28 6e527fbc 2005-02-13 devnull 
 29 6e527fbc 2005-02-13 devnull void
 30 6e527fbc 2005-02-13 devnull convreset(Conv *c)
 31 6e527fbc 2005-02-13 devnull {
 32 6e527fbc 2005-02-13 devnull 	if(c->ref != 1){
 33 6e527fbc 2005-02-13 devnull 		c->hangup = 1;
 34 6e527fbc 2005-02-13 devnull 		nbsendp(c->rpcwait, 0);
 35 6e527fbc 2005-02-13 devnull 		while(c->ref > 1)
 36 6e527fbc 2005-02-13 devnull 			yield();
 37 6e527fbc 2005-02-13 devnull 		c->hangup = 0;
 38 6e527fbc 2005-02-13 devnull 	}
 39 6e527fbc 2005-02-13 devnull 	c->state = "nascent";
 40 6e527fbc 2005-02-13 devnull 	c->err[0] = '\0';
 41 6e527fbc 2005-02-13 devnull 	freeattr(c->attr);
 42 6e527fbc 2005-02-13 devnull 	c->attr = nil;
 43 6e527fbc 2005-02-13 devnull 	c->proto = nil;
 44 6e527fbc 2005-02-13 devnull 	c->rpc.op = 0;
 45 6e527fbc 2005-02-13 devnull 	c->active = 0;
 46 6e527fbc 2005-02-13 devnull 	c->done = 0;
 47 6e527fbc 2005-02-13 devnull 	c->hangup = 0;
 48 6e527fbc 2005-02-13 devnull }
 49 6e527fbc 2005-02-13 devnull 
 50 6e527fbc 2005-02-13 devnull void
 51 6e527fbc 2005-02-13 devnull convhangup(Conv *c)
 52 6e527fbc 2005-02-13 devnull {
 53 6e527fbc 2005-02-13 devnull 	c->hangup = 1;
 54 6e527fbc 2005-02-13 devnull 	c->rpc.op = 0;
 55 6e527fbc 2005-02-13 devnull 	(*c->kickreply)(c);
 56 6e527fbc 2005-02-13 devnull 	nbsendp(c->rpcwait, 0);
 57 6e527fbc 2005-02-13 devnull }
 58 6e527fbc 2005-02-13 devnull 
 59 6e527fbc 2005-02-13 devnull void
 60 6e527fbc 2005-02-13 devnull convclose(Conv *c)
 61 6e527fbc 2005-02-13 devnull {
 62 6e527fbc 2005-02-13 devnull 	Conv *p;
 63 6e527fbc 2005-02-13 devnull 
 64 6e527fbc 2005-02-13 devnull 	if(c == nil)
 65 6e527fbc 2005-02-13 devnull 		return;
 66 6e527fbc 2005-02-13 devnull 
 67 6e527fbc 2005-02-13 devnull 	if(--c->ref > 0)
 68 6e527fbc 2005-02-13 devnull 		return;
 69 6e527fbc 2005-02-13 devnull 
 70 6e527fbc 2005-02-13 devnull 	if(c == conv){
 71 6e527fbc 2005-02-13 devnull 		conv = c->next;
 72 6e527fbc 2005-02-13 devnull 		goto free;
 73 6e527fbc 2005-02-13 devnull 	}
 74 6e527fbc 2005-02-13 devnull 	for(p=conv; p && p->next!=c; p=p->next)
 75 6e527fbc 2005-02-13 devnull 		;
 76 6e527fbc 2005-02-13 devnull 	if(p == nil){
 77 6e527fbc 2005-02-13 devnull 		print("cannot find conv in list\n");
 78 6e527fbc 2005-02-13 devnull 		return;
 79 6e527fbc 2005-02-13 devnull 	}
 80 6e527fbc 2005-02-13 devnull 	p->next = c->next;
 81 6e527fbc 2005-02-13 devnull 
 82 6e527fbc 2005-02-13 devnull free:
 83 6e527fbc 2005-02-13 devnull 	c->next = nil;
 84 6e527fbc 2005-02-13 devnull 	free(c);
 85 6e527fbc 2005-02-13 devnull }
 86 6e527fbc 2005-02-13 devnull 
 87 6e527fbc 2005-02-13 devnull static Rpc*
 88 6e527fbc 2005-02-13 devnull convgetrpc(Conv *c, int want)
 89 6e527fbc 2005-02-13 devnull {
 90 6e527fbc 2005-02-13 devnull 	for(;;){
 91 6e527fbc 2005-02-13 devnull 		if(c->hangup){
 92 6e527fbc 2005-02-13 devnull 			werrstr("hangup");
 93 6e527fbc 2005-02-13 devnull 			return nil;
 94 6e527fbc 2005-02-13 devnull 		}
 95 6e527fbc 2005-02-13 devnull 		if(c->rpc.op == RpcUnknown){
 96 6e527fbc 2005-02-13 devnull 			recvp(c->rpcwait);
 97 6e527fbc 2005-02-13 devnull 			if(c->hangup){
 98 6e527fbc 2005-02-13 devnull 				werrstr("hangup");
 99 6e527fbc 2005-02-13 devnull 				return nil;
100 6e527fbc 2005-02-13 devnull 			}
101 6e527fbc 2005-02-13 devnull 			if(c->rpc.op == RpcUnknown)
102 6e527fbc 2005-02-13 devnull 				continue;
103 6e527fbc 2005-02-13 devnull 		}
104 6e527fbc 2005-02-13 devnull 		if(want < 0 || c->rpc.op == want)
105 6e527fbc 2005-02-13 devnull 			return &c->rpc;
106 6e527fbc 2005-02-13 devnull 		rpcrespond(c, "phase in state '%s' want '%s'", c->state, rpcname[want]);
107 6e527fbc 2005-02-13 devnull 	}
108 bcac59d8 2005-08-11 devnull 	/* not reached */
109 6e527fbc 2005-02-13 devnull }
110 6e527fbc 2005-02-13 devnull 
111 6e527fbc 2005-02-13 devnull /* read until the done function tells us that's enough */
112 6e527fbc 2005-02-13 devnull int
113 6e527fbc 2005-02-13 devnull convreadfn(Conv *c, int (*done)(void*, int), char **ps)
114 6e527fbc 2005-02-13 devnull {
115 6e527fbc 2005-02-13 devnull 	int n;
116 6e527fbc 2005-02-13 devnull 	Rpc *r;
117 6e527fbc 2005-02-13 devnull 	char *s;
118 6e527fbc 2005-02-13 devnull 
119 6e527fbc 2005-02-13 devnull 	for(;;){
120 6e527fbc 2005-02-13 devnull 		r = convgetrpc(c, RpcWrite);
121 6e527fbc 2005-02-13 devnull 		if(r == nil)
122 6e527fbc 2005-02-13 devnull 			return -1;
123 6e527fbc 2005-02-13 devnull 		n = (*done)(r->data, r->count);
124 6e527fbc 2005-02-13 devnull 		if(n == r->count)
125 6e527fbc 2005-02-13 devnull 			break;
126 6e527fbc 2005-02-13 devnull 		rpcrespond(c, "toosmall %d", n);
127 6e527fbc 2005-02-13 devnull 	}
128 6e527fbc 2005-02-13 devnull 
129 6e527fbc 2005-02-13 devnull 	s = emalloc(r->count+1);
130 6e527fbc 2005-02-13 devnull 	memmove(s, r->data, r->count);
131 6e527fbc 2005-02-13 devnull 	s[r->count] = 0;
132 6e527fbc 2005-02-13 devnull 	*ps = s;
133 6e527fbc 2005-02-13 devnull 	rpcrespond(c, "ok");
134 6e527fbc 2005-02-13 devnull 	return r->count;
135 6e527fbc 2005-02-13 devnull }
136 6e527fbc 2005-02-13 devnull 
137 6e527fbc 2005-02-13 devnull /*
138 6e527fbc 2005-02-13 devnull  * read until we get a non-zero write.  assumes remote side
139 6e527fbc 2005-02-13 devnull  * knows something about the protocol (is not auth_proxy).
140 6e527fbc 2005-02-13 devnull  * the remote side typically won't bother with the zero-length
141 6e527fbc 2005-02-13 devnull  * write to find out the length -- the loop is there only so the
142 6e527fbc 2005-02-13 devnull  * test program can call auth_proxy on both sides of a pipe
143 6e527fbc 2005-02-13 devnull  * to play a conversation.
144 6e527fbc 2005-02-13 devnull  */
145 6e527fbc 2005-02-13 devnull int
146 6e527fbc 2005-02-13 devnull convreadm(Conv *c, char **ps)
147 6e527fbc 2005-02-13 devnull {
148 6e527fbc 2005-02-13 devnull 	char *s;
149 6e527fbc 2005-02-13 devnull 	Rpc *r;
150 6e527fbc 2005-02-13 devnull 
151 6e527fbc 2005-02-13 devnull 	for(;;){
152 6e527fbc 2005-02-13 devnull 		r = convgetrpc(c, RpcWrite);
153 6e527fbc 2005-02-13 devnull 		if(r == nil)
154 6e527fbc 2005-02-13 devnull 			return -1;
155 6e527fbc 2005-02-13 devnull 		if(r->count > 0)
156 6e527fbc 2005-02-13 devnull 			break;
157 6e527fbc 2005-02-13 devnull 		rpcrespond(c, "toosmall %d", AuthRpcMax);
158 6e527fbc 2005-02-13 devnull 	}
159 6e527fbc 2005-02-13 devnull 	s = emalloc(r->count+1);
160 6e527fbc 2005-02-13 devnull 	memmove(s, r->data, r->count);
161 6e527fbc 2005-02-13 devnull 	s[r->count] = 0;
162 6e527fbc 2005-02-13 devnull 	*ps = s;
163 6e527fbc 2005-02-13 devnull 	rpcrespond(c, "ok");
164 6e527fbc 2005-02-13 devnull 	return r->count;
165 6e527fbc 2005-02-13 devnull }
166 6e527fbc 2005-02-13 devnull 
167 6e527fbc 2005-02-13 devnull /* read exactly count bytes */
168 6e527fbc 2005-02-13 devnull int
169 6e527fbc 2005-02-13 devnull convread(Conv *c, void *data, int count)
170 6e527fbc 2005-02-13 devnull {
171 6e527fbc 2005-02-13 devnull 	Rpc *r;
172 6e527fbc 2005-02-13 devnull 
173 6e527fbc 2005-02-13 devnull 	for(;;){
174 6e527fbc 2005-02-13 devnull 		r = convgetrpc(c, RpcWrite);
175 6e527fbc 2005-02-13 devnull 		if(r == nil)
176 6e527fbc 2005-02-13 devnull 			return -1;
177 6e527fbc 2005-02-13 devnull 		if(r->count == count)
178 6e527fbc 2005-02-13 devnull 			break;
179 6e527fbc 2005-02-13 devnull 		if(r->count < count)
180 6e527fbc 2005-02-13 devnull 			rpcrespond(c, "toosmall %d", count);
181 6e527fbc 2005-02-13 devnull 		else
182 6e527fbc 2005-02-13 devnull 			rpcrespond(c, "error too much data; want %d got %d", count, r->count);
183 6e527fbc 2005-02-13 devnull 	}
184 6e527fbc 2005-02-13 devnull 	memmove(data, r->data, count);
185 6e527fbc 2005-02-13 devnull 	rpcrespond(c, "ok");
186 6e527fbc 2005-02-13 devnull 	return 0;
187 6e527fbc 2005-02-13 devnull }
188 6e527fbc 2005-02-13 devnull 
189 6e527fbc 2005-02-13 devnull /* write exactly count bytes */
190 6e527fbc 2005-02-13 devnull int
191 6e527fbc 2005-02-13 devnull convwrite(Conv *c, void *data, int count)
192 6e527fbc 2005-02-13 devnull {
193 6e527fbc 2005-02-13 devnull 	Rpc *r;
194 6e527fbc 2005-02-13 devnull 
195 1f8a8072 2005-03-15 devnull 	r = convgetrpc(c, RpcRead);
196 1f8a8072 2005-03-15 devnull 	if(r == nil)
197 1f8a8072 2005-03-15 devnull 		return -1;
198 6e527fbc 2005-02-13 devnull 	rpcrespondn(c, "ok", data, count);
199 6e527fbc 2005-02-13 devnull 	return 0;
200 6e527fbc 2005-02-13 devnull }
201 6e527fbc 2005-02-13 devnull 
202 6e527fbc 2005-02-13 devnull /* print to the conversation */
203 6e527fbc 2005-02-13 devnull int
204 6e527fbc 2005-02-13 devnull convprint(Conv *c, char *fmt, ...)
205 6e527fbc 2005-02-13 devnull {
206 6e527fbc 2005-02-13 devnull 	char *s;
207 6e527fbc 2005-02-13 devnull 	va_list arg;
208 6e527fbc 2005-02-13 devnull 	int ret;
209 6e527fbc 2005-02-13 devnull 
210 6e527fbc 2005-02-13 devnull 	va_start(arg, fmt);
211 6e527fbc 2005-02-13 devnull 	s = vsmprint(fmt, arg);
212 6e527fbc 2005-02-13 devnull 	va_end(arg);
213 6e527fbc 2005-02-13 devnull 	if(s == nil)
214 6e527fbc 2005-02-13 devnull 		return -1;
215 6e527fbc 2005-02-13 devnull 	ret = convwrite(c, s, strlen(s));
216 6e527fbc 2005-02-13 devnull 	free(s);
217 6e527fbc 2005-02-13 devnull 	return ret;
218 6e527fbc 2005-02-13 devnull }
219 6e527fbc 2005-02-13 devnull 
220 6e527fbc 2005-02-13 devnull /* ask for a key */
221 6e527fbc 2005-02-13 devnull int
222 6e527fbc 2005-02-13 devnull convneedkey(Conv *c, Attr *a)
223 6e527fbc 2005-02-13 devnull {
224 6e527fbc 2005-02-13 devnull 	/*
225 6e527fbc 2005-02-13 devnull 	 * Piggyback key requests in the usual RPC channel.
226 6e527fbc 2005-02-13 devnull 	 * Wait for the next RPC and then send a key request
227 6e527fbc 2005-02-13 devnull 	 * in response.  The keys get added out-of-band (via the
228 6e527fbc 2005-02-13 devnull 	 * ctl file), so assume the key has been added when the
229 6e527fbc 2005-02-13 devnull 	 * next request comes in.
230 6e527fbc 2005-02-13 devnull 	 */
231 6e527fbc 2005-02-13 devnull 	if(convgetrpc(c, -1) == nil)
232 6e527fbc 2005-02-13 devnull 		return -1;
233 6e527fbc 2005-02-13 devnull 	rpcrespond(c, "needkey %A", a);
234 6e527fbc 2005-02-13 devnull 	if(convgetrpc(c, -1) == nil)
235 6e527fbc 2005-02-13 devnull 		return -1;
236 6e527fbc 2005-02-13 devnull 	return 0;
237 6e527fbc 2005-02-13 devnull }
238 6e527fbc 2005-02-13 devnull 
239 6e527fbc 2005-02-13 devnull /* ask for a replacement for a bad key*/
240 6e527fbc 2005-02-13 devnull int
241 6e527fbc 2005-02-13 devnull convbadkey(Conv *c, Key *k, char *msg, Attr *a)
242 6e527fbc 2005-02-13 devnull {
243 6e527fbc 2005-02-13 devnull 	if(convgetrpc(c, -1) == nil)
244 6e527fbc 2005-02-13 devnull 		return -1;
245 6e527fbc 2005-02-13 devnull 	rpcrespond(c, "badkey %A %N\n%s\n%A",
246 6e527fbc 2005-02-13 devnull 		k->attr, k->privattr, msg, a);
247 6e527fbc 2005-02-13 devnull 	if(convgetrpc(c, -1) == nil)
248 6e527fbc 2005-02-13 devnull 		return -1;
249 6e527fbc 2005-02-13 devnull 	return 0;
250 6e527fbc 2005-02-13 devnull }
251 6e527fbc 2005-02-13 devnull