1 #include <u.h>
2 #include <libc.h>
3 #include <auth.h>
4 #include "authlocal.h"
5
6 static struct {
7 char *verb;
8 int val;
9 } tab[] = {
10 "ok", ARok,
11 "done", ARdone,
12 "error", ARerror,
13 "needkey", ARneedkey,
14 "badkey", ARbadkey,
15 "phase", ARphase,
16 "toosmall", ARtoosmall,
17 "error", ARerror,
18 };
19
20 static int
classify(char * buf,uint n,AuthRpc * rpc)21 classify(char *buf, uint n, AuthRpc *rpc)
22 {
23 int i, len;
24
25 for(i=0; i<nelem(tab); i++){
26 len = strlen(tab[i].verb);
27 if(n >= len && memcmp(buf, tab[i].verb, len) == 0 && (n==len || buf[len]==' ')){
28 if(n==len){
29 rpc->narg = 0;
30 rpc->arg = "";
31 }else{
32 rpc->narg = n - (len+1);
33 rpc->arg = (char*)buf+len+1;
34 }
35 return tab[i].val;
36 }
37 }
38 werrstr("malformed rpc response: %s", buf);
39 return ARrpcfailure;
40 }
41
42 AuthRpc*
auth_allocrpc(int afd)43 auth_allocrpc(int afd)
44 {
45 AuthRpc *rpc;
46
47 rpc = mallocz(sizeof(*rpc), 1);
48 if(rpc == nil)
49 return nil;
50 rpc->afd = afd;
51 return rpc;
52 }
53
54 void
auth_freerpc(AuthRpc * rpc)55 auth_freerpc(AuthRpc *rpc)
56 {
57 free(rpc);
58 }
59
60 uint
auth_rpc(AuthRpc * rpc,char * verb,void * a,int na)61 auth_rpc(AuthRpc *rpc, char *verb, void *a, int na)
62 {
63 int l, n, type;
64 char *f[4];
65
66 l = strlen(verb);
67 if(na+l+1 > AuthRpcMax){
68 werrstr("rpc too big");
69 return ARtoobig;
70 }
71
72 memmove(rpc->obuf, verb, l);
73 rpc->obuf[l] = ' ';
74 memmove(rpc->obuf+l+1, a, na);
75 if((n=write(rpc->afd, rpc->obuf, l+1+na)) != l+1+na){
76 if(n >= 0)
77 werrstr("auth_rpc short write");
78 return ARrpcfailure;
79 }
80
81 if((n=read(rpc->afd, rpc->ibuf, AuthRpcMax)) < 0)
82 return ARrpcfailure;
83 rpc->ibuf[n] = '\0';
84
85 /*
86 * Set error string for good default behavior.
87 */
88 switch(type = classify(rpc->ibuf, n, rpc)){
89 default:
90 werrstr("unknown rpc type %d (bug in auth_rpc.c)", type);
91 break;
92 case ARok:
93 break;
94 case ARrpcfailure:
95 break;
96 case ARerror:
97 if(rpc->narg == 0)
98 werrstr("unspecified rpc error");
99 else
100 werrstr("%s", rpc->arg);
101 break;
102 case ARneedkey:
103 werrstr("needkey %s", rpc->arg);
104 break;
105 case ARbadkey:
106 if(getfields(rpc->arg, f, nelem(f), 0, "\n") < 2)
107 werrstr("badkey %s", rpc->arg);
108 else
109 werrstr("badkey %s", f[1]);
110 break;
111 case ARphase:
112 werrstr("phase error %s", rpc->arg);
113 break;
114 }
115 return type;
116 }
117