开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 嵌入式/单片机编程 > VxWorks > 查看源码
portmap.c
资源名称：vxwork_src.rar　[点击查看]
上传用户：nvosite88
上传日期：2007-01-17
资源大小：4983k
文件大小：11k
源码类别：

VxWorks

开发平台：

C/C++

portmap.c：源码内容
  1. /* portmap.c - rpc portmap daemon */
  3. /* Copyright 1984-2001 Wind River Systems, Inc. */
  4. #include "copyright_wrs.h"
  6. /*
  7.  * Copyright (c) 1984 by Sun Microsystems, Inc.
  8.  */
  10. /*
  11.  * portmap.c, Implements the program,version to port number mapping for rpc.
  12.  */
  14. /*
  15.  * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
  16.  * unrestricted use provided that this legend is included on all tape
  17.  * media and as a part of the software program in whole or part.  Users
  18.  * may copy or modify Sun RPC without charge, but are not authorized
  19.  * to license or distribute it to anyone else except as part of a product or
  20.  * program developed by the user.
  21.  *
  22.  * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
  23.  * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
  24.  * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
  25.  *
  26.  * Sun RPC is provided with no support and without any obligation on the
  27.  * part of Sun Microsystems, Inc. to assist in its use, correction,
  28.  * modification or enhancement.
  29.  *
  30.  * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
  31.  * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
  32.  * OR ANY PART THEREOF.
  33.  *
  34.  * In no event will Sun Microsystems, Inc. be liable for any lost revenue
  35.  * or profits or other special, indirect and consequential damages, even if
  36.  * Sun has been advised of the possibility of such damages.
  37.  *
  38.  * Sun Microsystems, Inc.
  39.  * 2550 Garcia Avenue
  40.  * Mountain View, California  94043
  41.  */
  43. /*
  44. modification history
  45. --------------------
  46. 01q,05nov01,vvv  fixed compilation warning
  47. 01p,15oct01,rae  merge from truestack ver 01q, base 01o (AE / 5_X)
  48. 01o,15sep93,kdl  changed svcudp_create() call to only pass one param (SPR #2427)
  49. 01n,11aug93,jmm  Changed ioctl.h and socket.h to sys/ioctl.h and sys/socket.h
  50. 01m,04jun92,wmd  added forward declarations required for gcc960 v2.0.
  51. 01l,26may92,rrr  the tree shuffle
  52. 01k,04oct91,rrr  passed through the ansification filter
  53.   -changed includes to have absolute path from h/
  54.   -changed VOID to void
  55.   -changed copyright notice
  56. 01j,11jan91,shl  added keyword ENCAP_PARMS and RMTCALLARGS
  57. 01i,25oct90,dnw  removed include of utime.h.
  58. 01h,11may90,yao  added missing modification history (01g) for the last checkin.
  59. 01g,09may90,yao  typecasted malloc to (char *).
  60. 01f,19apr90,hjb  de-linted.
  61. 01e,27oct89,hjb  changed exit (1) to taskSuspend (0).
  62. 01d,05apr88,gae  changed fprintf() to printErr().
  63. 01c,24nov87,gae  removed include of netdb.h.
  64. 01b,24oct87,dnw  changed call to rpcInit() to rpcTaskInit().
  65. */
  67. #include "rpc/rpc.h"
  68. #include "rpc/pmap_prot.h"
  69. #include "sys/socket.h"
  70. #include "sys/ioctl.h"
  71. #include "vxWorks.h"
  72. #include "memLib.h"
  73. #include "memPartLib.h"
  74. #include "rpcLib.h"
  75. #include "sockLib.h"
  76. #include "stdio.h"
  77. #include "unistd.h"
  78. #include "taskLib.h"
  79. #include "stdlib.h"
  80. #include "rpc/get_myaddr.h"
  82. void reg_service();
  83. void pmapRegister ();
  84. LOCAL void  callit();
  86. static int debugging = 0;
  88. void
  89. portmapd ()
  90. {
  91. SVCXPRT *xprt;
  92. int sock;
  93. struct sockaddr_in addr;
  94. int len = sizeof(struct sockaddr_in);
  96. rpcTaskInit ();
  98. if ((sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0) {
  99. perror("portmap cannot create socket");
  100. taskSuspend (0);
  101. }
  103. addr.sin_addr.s_addr = 0;
  104. addr.sin_family = AF_INET;
  105. addr.sin_port = htons((u_short) PMAPPORT);
  106. if (bind(sock, (struct sockaddr *)&addr, len) != 0) {
  107. perror("portmap cannot bind");
  108. taskSuspend (0);
  109. }
  111. if ((xprt = svcudp_create(sock)) == (SVCXPRT *)NULL) {
  112. printErr ("couldn't do udp_createn");
  113. taskSuspend (0);
  114. }
  116. if ((sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
  117. perror("portmap cannot create socket");
  118. taskSuspend (0);
  119. }
  120. if (bind(sock, (struct sockaddr *)&addr, len) != 0) {
  121. perror("portmap cannot bind");
  122. taskSuspend (0);
  123. }
  124. if ((xprt = svctcp_create(sock, RPCSMALLMSGSIZE, RPCSMALLMSGSIZE))
  125.     == (SVCXPRT *)NULL) {
  126. printErr ("couldn't do tcp_createn");
  127. taskSuspend (0);
  128. }
  130.         (void) svc_register (xprt, PMAPPROG, PMAPVERS, reg_service, (int)NULL);
  131. svc_run();
  132. printErr ("run_svc returned unexpectedlyn");
  133. abort();
  134. }
  136. struct pmaplist *pmaplist;
  138. static struct pmaplist *
  139. find_service(prog, vers, prot)
  140. u_long prog;
  141. u_long vers;
  142. {
  143. register struct pmaplist *hit = NULL;
  144. register struct pmaplist *pml;
  146. for (pml = pmaplist; pml != NULL; pml = pml->pml_next) {
  147. if ((pml->pml_map.pm_prog != prog) ||
  148. (pml->pml_map.pm_prot != prot))
  149. continue;
  150. hit = pml;
  151. if (pml->pml_map.pm_vers == vers)
  152.     break;
  153. }
  154. return (hit);
  155. }
  157. /*
  158.  * 1 OK, 0 not
  159.  */
  160. void reg_service(rqstp, xprt)
  161. struct svc_req *rqstp;
  162. SVCXPRT *xprt;
  163. {
  164. struct pmap reg;
  165. struct pmaplist *pml, *prevpml, *fnd;
  166. int ans, port;
  167. caddr_t t;
  169. switch ((int)rqstp->rq_proc) {
  171. case PMAPPROC_NULL:
  172. /*
  173.  * Null proc call
  174.  */
  175. if ((!svc_sendreply(xprt, xdr_void, (char *)NULL)) &&
  176.     debugging) {
  177. abort();
  178. }
  179. break;
  181. case PMAPPROC_SET:
  182. /*
  183.  * Set a program,version to port mapping
  184.  */
  185. if (!svc_getargs(xprt, xdr_pmap, &reg))
  186. svcerr_decode(xprt);
  187. else {
  188. /*
  189.  * check to see if already used
  190.  * find_service returns a hit even if
  191.  * the versions don't match, so check for it
  192.  */
  193. fnd = find_service(reg.pm_prog, reg.pm_vers,
  194. (int) reg.pm_prot);
  195. if (fnd && fnd->pml_map.pm_vers == reg.pm_vers) {
  196. if (fnd->pml_map.pm_port == reg.pm_port) {
  197. ans = 1;
  198. goto done;
  199. }
  200. else {
  201. ans = 0;
  202. goto done;
  203. }
  204. } else {
  205. /*
  206.  * add to END of list
  207.  */
  208. pml = (struct pmaplist *)
  209.       KHEAP_ALLOC((u_int)sizeof(struct pmaplist));
  210. pml->pml_map = reg;
  211. pml->pml_next = 0;
  212. if (pmaplist == 0) {
  213. pmaplist = pml;
  214. } else {
  215. for (fnd= pmaplist; fnd->pml_next != 0;
  216.     fnd = fnd->pml_next);
  217. fnd->pml_next = pml;
  218. }
  219. ans = 1;
  220. }
  221. done:
  222. if ((!svc_sendreply(xprt, xdr_long, (caddr_t)&ans)) &&
  223.     debugging) {
  224. printErr ("svc_sendreplyn");
  225. abort();
  226. }
  227. }
  228. break;
  230. case PMAPPROC_UNSET:
  231. /*
  232.  * Remove a program,version to port mapping.
  233.  */
  234. if (!svc_getargs(xprt, xdr_pmap, &reg))
  235. svcerr_decode(xprt);
  236. else {
  237. ans = 0;
  238. for (prevpml = NULL, pml = pmaplist; pml != NULL; ) {
  239. if ((pml->pml_map.pm_prog != reg.pm_prog) ||
  240. (pml->pml_map.pm_vers != reg.pm_vers)) {
  241. /* both pml & prevpml move forwards */
  242. prevpml = pml;
  243. pml = pml->pml_next;
  244. continue;
  245. }
  246. /* found it; pml moves forward, prevpml stays */
  247. ans = 1;
  248. t = (caddr_t)pml;
  249. pml = pml->pml_next;
  250. if (prevpml == NULL)
  251. pmaplist = pml;
  252. else
  253. prevpml->pml_next = pml;
  254. KHEAP_FREE(t);
  255. }
  256. if ((!svc_sendreply(xprt, xdr_long, (caddr_t)&ans)) &&
  257.     debugging) {
  258. printErr ("svc_sendreplyn");
  259. abort();
  260. }
  261. }
  262. break;
  264. case PMAPPROC_GETPORT:
  265. /*
  266.  * Lookup the mapping for a program,version and return its port
  267.  */
  268. if (!svc_getargs(xprt, xdr_pmap, &reg))
  269. svcerr_decode(xprt);
  270. else {
  271. fnd = find_service(reg.pm_prog, reg.pm_vers,
  272. (int)reg.pm_prot);
  273. if (fnd)
  274. port = fnd->pml_map.pm_port;
  275. else
  276. port = 0;
  277. if ((!svc_sendreply(xprt, xdr_long, (caddr_t)&port)) &&
  278.     debugging) {
  279. printErr ("svc_sendreplyn");
  280. abort();
  281. }
  282. }
  283. break;
  285. case PMAPPROC_DUMP:
  286. /*
  287.  * Return the current set of mapped program,version
  288.  */
  289. if (!svc_getargs(xprt, xdr_void, NULL))
  290. svcerr_decode(xprt);
  291. else {
  292. if ((!svc_sendreply(xprt, xdr_pmaplist,
  293.     (caddr_t)&pmaplist)) && debugging) {
  294. printErr ("svc_sendreplyn");
  295. abort();
  296. }
  297. }
  298. break;
  300. case PMAPPROC_CALLIT:
  301. /*
  302.  * Calls a procedure on the local machine.  If the requested
  303.  * procedure is not registered this procedure does not return
  304.  * error information!!
  305.  * This procedure is only supported on rpc/udp and calls via
  306.  * rpc/udp.  It passes null authentication parameters.
  307.  */
  308. callit(rqstp, xprt);
  309. break;
  311. default:
  312. svcerr_noproc(xprt);
  313. break;
  314. }
  315. }
  318. /*
  319.  * Stuff for the rmtcall service
  320.  */
  321. #define ARGSIZE 9000
  323. typedef struct encap_parms {
  324. u_long arglen;
  325. char *args;
  326. } ENCAP_PARMS;
  328. static bool_t
  329. xdr_encap_parms(xdrs, epp)
  330. XDR *xdrs;
  331. struct encap_parms *epp;
  332. {
  334. return (xdr_bytes(xdrs, &(epp->args), (u_int *) &(epp->arglen),
  335. (u_int) ARGSIZE));
  336. }
  338. typedef struct rmtcallargs {
  339. u_long rmt_prog;
  340. u_long rmt_vers;
  341. u_long rmt_port;
  342. u_long rmt_proc;
  343. struct encap_parms rmt_args;
  344. } RMTCALLARGS;
  346. static bool_t
  347. xdr_rmtcall_args(xdrs, cap)
  348. register XDR *xdrs;
  349. register struct rmtcallargs *cap;
  350. {
  352. /* does not get a port number */
  353. if (xdr_u_long(xdrs, &(cap->rmt_prog)) &&
  354.     xdr_u_long(xdrs, &(cap->rmt_vers)) &&
  355.     xdr_u_long(xdrs, &(cap->rmt_proc))) {
  356. return (xdr_encap_parms(xdrs, &(cap->rmt_args)));
  357. }
  358. return (FALSE);
  359. }
  361. static bool_t
  362. xdr_rmtcall_result(xdrs, cap)
  363. register XDR *xdrs;
  364. register struct rmtcallargs *cap;
  365. {
  366. if (xdr_u_long(xdrs, &(cap->rmt_port)))
  367. return (xdr_encap_parms(xdrs, &(cap->rmt_args)));
  368. return (FALSE);
  369. }
  371. /*
  372.  * only worries about the struct encap_parms part of struct rmtcallargs.
  373.  * The arglen must already be set!!
  374.  */
  375. static bool_t
  376. xdr_opaque_parms(xdrs, cap)
  377. XDR *xdrs;
  378. struct rmtcallargs *cap;
  379. {
  381. return (xdr_opaque(xdrs, cap->rmt_args.args,
  382. (u_int)cap->rmt_args.arglen));
  383. }
  385. /*
  386.  * This routine finds and sets the length of incoming opaque paraters
  387.  * and then calls xdr_opaque_parms.
  388.  */
  389. static bool_t
  390. xdr_len_opaque_parms(xdrs, cap)
  391. register XDR *xdrs;
  392. struct rmtcallargs *cap;
  393. {
  394. register u_int beginpos, lowpos, highpos, currpos, pos;
  396. beginpos = lowpos = pos = xdr_getpos(xdrs);
  397. highpos = lowpos + ARGSIZE;
  398. while ((int)(highpos - lowpos) >= 0) {
  399. currpos = (lowpos + highpos) / 2;
  400. if (xdr_setpos(xdrs, currpos)) {
  401. pos = currpos;
  402. lowpos = currpos + 1;
  403. } else {
  404. highpos = currpos - 1;
  405. }
  406. }
  407. xdr_setpos(xdrs, beginpos);
  408. cap->rmt_args.arglen = pos - beginpos;
  409. return (xdr_opaque_parms(xdrs, cap));
  410. }
  412. /*
  413.  * Call a remote procedure service
  414.  * This procedure is very quiet when things go wrong.
  415.  * The proc is written to support broadcast rpc.  In the broadcast case,
  416.  * a machine should shut-up instead of complain, less the requestor be
  417.  * overrun with complaints at the expense of not hearing a valid reply ...
  418.  */
  419. static void
  420. callit(rqstp, xprt)
  421. struct svc_req *rqstp;
  422. SVCXPRT *xprt;
  423. {
  424. char buf[2000];
  425. struct rmtcallargs a;
  426. struct pmaplist *pml;
  427. u_short port;
  428. struct sockaddr_in me;
  429. int socket = -1;
  430. CLIENT *client;
  431. struct authunix_parms *au = (struct authunix_parms *)rqstp->rq_clntcred;
  432. struct timeval timeout;
  434. timeout.tv_sec = 5;
  435. timeout.tv_usec = 0;
  436. a.rmt_args.args = buf;
  437. if (!svc_getargs(xprt, xdr_rmtcall_args, &a))
  438.     return;
  439. if ((pml = find_service(a.rmt_prog, a.rmt_vers, IPPROTO_UDP)) == NULL)
  440.     return;
  441. port = pml->pml_map.pm_port;
  442. get_myaddress(&me);
  443. me.sin_port = htons(port);
  444. client = clntudp_create(&me, a.rmt_prog, a.rmt_vers, timeout, &socket);
  445. if (client != (CLIENT *)NULL) {
  446. if (rqstp->rq_cred.oa_flavor == AUTH_UNIX) {
  447. client->cl_auth = authunix_create(au->aup_machname,
  448.    au->aup_uid, au->aup_gid, (int) au->aup_len,
  449.    au->aup_gids);
  450. }
  451. a.rmt_port = (u_long)port;
  452. if (clnt_call(client, a.rmt_proc, xdr_opaque_parms, &a,
  453.     xdr_len_opaque_parms, &a, timeout) == RPC_SUCCESS) {
  454. svc_sendreply(xprt, xdr_rmtcall_result, (char *) &a);
  455. }
  456. AUTH_DESTROY(client->cl_auth);
  457. clnt_destroy(client);
  458. }
  459. (void)close(socket);
  460. }