rpc.c 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314
  1. #include "all.h"
  2. #define SHORT(x) r->x = (p[1] | (p[0]<<8)); p += 2
  3. #define LONG(x) r->x = (p[3] | (p[2]<<8) |\
  4. (p[1]<<16) | (p[0]<<24)); p += 4
  5. #define SKIPLONG p += 4
  6. #define PTR(x, n) r->x = (void *)(p); p += ROUNDUP(n)
  7. int
  8. rpcM2S(void *ap, Rpccall *r, int n)
  9. {
  10. int k;
  11. uchar *p;
  12. Udphdr *up;
  13. /* copy IPv4 header fields from Udphdr */
  14. up = ap;
  15. p = &up->raddr[IPaddrlen - IPv4addrlen];
  16. LONG(host);
  17. USED(p);
  18. p = &up->laddr[IPaddrlen - IPv4addrlen];
  19. LONG(lhost);
  20. USED(p);
  21. /* ignore up->ifcaddr */
  22. p = up->rport;
  23. SHORT(port);
  24. SHORT(lport);
  25. LONG(xid);
  26. LONG(mtype);
  27. switch(r->mtype){
  28. case CALL:
  29. LONG(rpcvers);
  30. if(r->rpcvers != 2)
  31. break;
  32. LONG(prog);
  33. LONG(vers);
  34. LONG(proc);
  35. LONG(cred.flavor);
  36. LONG(cred.count);
  37. PTR(cred.data, r->cred.count);
  38. LONG(verf.flavor);
  39. LONG(verf.count);
  40. PTR(verf.data, r->verf.count);
  41. r->up = 0;
  42. k = n - (p - (uchar *)ap);
  43. if(k < 0)
  44. break;
  45. PTR(args, k);
  46. break;
  47. case REPLY:
  48. LONG(stat);
  49. switch(r->stat){
  50. case MSG_ACCEPTED:
  51. LONG(averf.flavor);
  52. LONG(averf.count);
  53. PTR(averf.data, r->averf.count);
  54. LONG(astat);
  55. switch(r->astat){
  56. case SUCCESS:
  57. k = n - (p - (uchar *)ap);
  58. if(k < 0)
  59. break;
  60. PTR(results, k);
  61. break;
  62. case PROG_MISMATCH:
  63. LONG(plow);
  64. LONG(phigh);
  65. break;
  66. }
  67. break;
  68. case MSG_DENIED:
  69. LONG(rstat);
  70. switch(r->rstat){
  71. case RPC_MISMATCH:
  72. LONG(rlow);
  73. LONG(rhigh);
  74. break;
  75. case AUTH_ERROR:
  76. LONG(authstat);
  77. break;
  78. }
  79. break;
  80. }
  81. break;
  82. }
  83. n -= p - (uchar *)ap;
  84. return n;
  85. }
  86. int
  87. auth2unix(Auth *arg, Authunix *r)
  88. {
  89. int i, n;
  90. uchar *p;
  91. if(arg->flavor != AUTH_UNIX)
  92. return -1;
  93. p = arg->data;
  94. LONG(stamp);
  95. LONG(mach.n);
  96. PTR(mach.s, r->mach.n);
  97. LONG(uid);
  98. LONG(gid);
  99. LONG(gidlen);
  100. n = r->gidlen;
  101. for(i=0; i<n && i < nelem(r->gids); i++){
  102. LONG(gids[i]);
  103. }
  104. for(; i<n; i++){
  105. SKIPLONG;
  106. }
  107. return arg->count - (p - (uchar *)arg->data);
  108. }
  109. int
  110. string2S(void *arg, String *r)
  111. {
  112. uchar *p;
  113. char *s;
  114. p = arg;
  115. LONG(n);
  116. PTR(s, r->n);
  117. /* must NUL terminate */
  118. s = malloc(r->n+1);
  119. if(s == nil)
  120. panic("malloc(%ld) failed in string2S\n", r->n+1);
  121. memmove(s, r->s, r->n);
  122. s[r->n] = '\0';
  123. r->s = strstore(s);
  124. free(s);
  125. return p - (uchar *)arg;
  126. }
  127. #undef SHORT
  128. #undef LONG
  129. #undef PTR
  130. #define SHORT(x) p[1] = r->x; p[0] = r->x>>8; p += 2
  131. #define LONG(x) p[3] = r->x; p[2] = r->x>>8; p[1] = r->x>>16; p[0] = r->x>>24; p += 4
  132. #define PTR(x,n) memmove(p, r->x, n); p += ROUNDUP(n)
  133. int
  134. rpcS2M(Rpccall *r, int ndata, void *ap)
  135. {
  136. uchar *p;
  137. Udphdr *up;
  138. /* copy header fields to Udphdr */
  139. up = ap;
  140. memmove(up->raddr, v4prefix, IPaddrlen);
  141. p = &up->raddr[IPaddrlen - IPv4addrlen];
  142. LONG(host);
  143. USED(p);
  144. memmove(up->laddr, v4prefix, IPaddrlen);
  145. p = &up->laddr[IPaddrlen - IPv4addrlen];
  146. LONG(lhost);
  147. USED(p);
  148. memmove(up->ifcaddr, IPnoaddr, sizeof up->ifcaddr);
  149. p = up->rport;
  150. SHORT(port);
  151. SHORT(lport);
  152. LONG(xid);
  153. LONG(mtype);
  154. switch(r->mtype){
  155. case CALL:
  156. LONG(rpcvers);
  157. LONG(prog);
  158. LONG(vers);
  159. LONG(proc);
  160. LONG(cred.flavor);
  161. LONG(cred.count);
  162. PTR(cred.data, r->cred.count);
  163. LONG(verf.flavor);
  164. LONG(verf.count);
  165. PTR(verf.data, r->verf.count);
  166. PTR(args, ndata);
  167. break;
  168. case REPLY:
  169. LONG(stat);
  170. switch(r->stat){
  171. case MSG_ACCEPTED:
  172. LONG(averf.flavor);
  173. LONG(averf.count);
  174. PTR(averf.data, r->averf.count);
  175. LONG(astat);
  176. switch(r->astat){
  177. case SUCCESS:
  178. PTR(results, ndata);
  179. break;
  180. case PROG_MISMATCH:
  181. LONG(plow);
  182. LONG(phigh);
  183. break;
  184. }
  185. break;
  186. case MSG_DENIED:
  187. LONG(rstat);
  188. switch(r->rstat){
  189. case RPC_MISMATCH:
  190. LONG(rlow);
  191. LONG(rhigh);
  192. break;
  193. case AUTH_ERROR:
  194. LONG(authstat);
  195. break;
  196. }
  197. break;
  198. }
  199. break;
  200. }
  201. return p - (uchar *)ap;
  202. }
  203. #undef SHORT
  204. #undef LONG
  205. #undef PTR
  206. #define LONG(m, x) fprint(fd, "%s = %ld\n", m, r->x)
  207. #define PTR(m, count) fprint(fd, "%s [%ld]\n", m, count)
  208. void
  209. rpcprint(int fd, Rpccall *r)
  210. {
  211. fprint(fd, "%s: host = %I, port = %ld\n",
  212. argv0, r->host, r->port);
  213. LONG("xid", xid);
  214. LONG("mtype", mtype);
  215. switch(r->mtype){
  216. case CALL:
  217. LONG("rpcvers", rpcvers);
  218. LONG("prog", prog);
  219. LONG("vers", vers);
  220. LONG("proc", proc);
  221. LONG("cred.flavor", cred.flavor);
  222. PTR("cred.data", r->cred.count);
  223. LONG("verf.flavor", verf.flavor);
  224. PTR("verf.data", r->verf.count);
  225. fprint(fd, "args...\n");
  226. break;
  227. case REPLY:
  228. LONG("stat", stat);
  229. switch(r->stat){
  230. case MSG_ACCEPTED:
  231. LONG("averf.flavor", averf.flavor);
  232. PTR("averf.data", r->averf.count);
  233. LONG("astat", astat);
  234. switch(r->astat){
  235. case SUCCESS:
  236. fprint(fd, "results...\n");
  237. break;
  238. case PROG_MISMATCH:
  239. LONG("plow", plow);
  240. LONG("phigh", phigh);
  241. break;
  242. }
  243. break;
  244. case MSG_DENIED:
  245. LONG("rstat", rstat);
  246. switch(r->rstat){
  247. case RPC_MISMATCH:
  248. LONG("rlow", rlow);
  249. LONG("rhigh", rhigh);
  250. break;
  251. case AUTH_ERROR:
  252. LONG("authstat", authstat);
  253. break;
  254. }
  255. break;
  256. }
  257. }
  258. }
  259. void
  260. showauth(Auth *ap)
  261. {
  262. Authunix au;
  263. int i;
  264. if(auth2unix(ap, &au) != 0){
  265. chat("auth flavor=%ld, count=%ld",
  266. ap->flavor, ap->count);
  267. for(i=0; i<ap->count; i++)
  268. chat(" %.2ux", ((uchar *)ap->data)[i]);
  269. }else{
  270. chat("auth: %ld %.*s u=%ld g=%ld",
  271. au.stamp, utfnlen(au.mach.s, au.mach.n), au.mach.s, au.uid, au.gid);
  272. for(i=0; i<au.gidlen; i++)
  273. chat(", %ld", au.gids[i]);
  274. }
  275. chat("...");
  276. }
  277. int
  278. garbage(Rpccall *reply, char *msg)
  279. {
  280. chat("%s\n", msg ? msg : "garbage");
  281. reply->astat = GARBAGE_ARGS;
  282. return 0;
  283. }
  284. int
  285. error(Rpccall *reply, int errno)
  286. {
  287. uchar *dataptr = reply->results;
  288. chat("error %d\n", errno);
  289. PLONG(errno);
  290. return dataptr - (uchar *)reply->results;
  291. }