home *** CD-ROM | disk | FTP | other *** search
/ Borland Programmer's Resource / Borland_Programmers_Resource_CD_1995.iso / utils / sossntr3 / src / svc.c < prev    next >
Encoding:
C/C++ Source or Header  |  1995-05-19  |  11.6 KB  |  485 lines
  1. /*
  2.  * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
  3.  * unrestricted use provided that this legend is included on all tape
  4.  * media and as a part of the software program in whole or part.  Users
  5.  * may copy or modify Sun RPC without charge, but are not authorized
  6.  * to license or distribute it to anyone else except as part of a product or
  7.  * program developed by the user.
  8.  * 
  9.  * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
  10.  * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
  11.  * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
  12.  * 
  13.  * Sun RPC is provided with no support and without any obligation on the
  14.  * part of Sun Microsystems, Inc. to assist in its use, correction,
  15.  * modification or enhancement.
  16.  * 
  17.  * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
  18.  * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
  19.  * OR ANY PART THEREOF.
  20.  * 
  21.  * In no event will Sun Microsystems, Inc. be liable for any lost revenue
  22.  * or profits or other special, indirect and consequential damages, even if
  23.  * Sun has been advised of the possibility of such damages.
  24.  * 
  25.  * Sun Microsystems, Inc.
  26.  * 2550 Garcia Avenue
  27.  * Mountain View, California  94043
  28.  */
  29. #ifndef lint
  30. static char sccsid[] = "@(#)svc.c 1.1 86/02/03 Copyr 1984 Sun Micro";
  31. #endif
  32.  
  33. /*
  34.  * svc.c, Server-side remote procedure call interface.
  35.  *
  36.  * There are two sets of procedures here.  The xprt routines are
  37.  * for handling transport handles.  The svc routines handle the
  38.  * list of service routines.
  39.  *
  40.  * Copyright (C) 1984, Sun Microsystems, Inc.
  41.  */
  42. #include <winsock.h>
  43. #include <malloc.h>
  44. #include <errno.h>
  45. #include <stdio.h>
  46. #include <stdlib.h>
  47. #include "sosserr.h"
  48. #include "types.h"
  49. /* #include "in.h" */
  50. #include "xdr.h"
  51. #include "auth.h"
  52. #include "clnt.h"
  53. #include "rpc_msg.h"
  54. #include "svc.h"
  55. #include "svc_auth.h"
  56. #include "p_clnt.h"
  57. #include <conio.h>
  58.  
  59. static SVCXPRT *xports[FD_SETSIZE];
  60. fd_set svc_fds;
  61. extern errno;
  62.  
  63. #define NULL_SVC ((struct svc_callout *)0)
  64. #define RQCRED_SIZE 400
  65.  
  66. /*
  67.  * The services list
  68.  * Each entry represents a set of procedures (an rpc program).
  69.  * The dispatch routine takes request structs and runs the
  70.  * apropriate procedure.
  71.  */
  72. static struct svc_callout {
  73.     struct svc_callout *sc_next;
  74.     u_long          sc_prog;
  75.     u_long          sc_vers;
  76.     void            (*sc_dispatch)();
  77. } *svc_head;
  78.  
  79. static struct svc_callout *svc_find();
  80.  
  81. /* ***************  SVCXPRT related stuff **************** */
  82.  
  83. /*
  84.  * Activate a transport handle.
  85.  */
  86. void
  87. xprt_register(xprt)
  88.     SVCXPRT *xprt;
  89. {
  90.     SOCKET sock = xprt->xp_sock;
  91.  
  92.     if (sock < FD_SETSIZE) {
  93.         xports[sock] = xprt;
  94.         FD_SET(sock,&svc_fds);
  95.         /* svc_fds |= (1 << sock); */
  96.     }
  97. }
  98.  
  99. /*
  100.  * De-activate a transport handle. 
  101.  */
  102. void
  103. xprt_unregister(xprt) 
  104.     SVCXPRT *xprt;
  105. { 
  106.     SOCKET sock = xprt->xp_sock;
  107.  
  108.     if ((sock < FD_SETSIZE) && (xports[sock] == xprt)) {
  109.         xports[sock] = (SVCXPRT *)0;
  110.         FD_CLR(sock,&svc_fds);
  111.         /* svc_fds &= ~(1L << sock); */
  112.     }
  113. } 
  114.  
  115.  
  116. /* ********************** CALLOUT list related stuff ************* */
  117.  
  118. /*
  119.  * Add a service program to the callout list.
  120.  * The dispatch routine will be called when a rpc request for this
  121.  * program number comes in.
  122.  */
  123. bool_t
  124. svc_register(xprt, prog, vers, dispatch, protocol)
  125.     SVCXPRT *xprt;
  126.     u_long prog;
  127.     u_long vers;
  128.     void (*dispatch)();
  129.     int protocol;
  130. {
  131.     struct svc_callout *prev;
  132.     register struct svc_callout *s;
  133.  
  134.     if ((s = svc_find(prog, vers, &prev)) != NULL_SVC) {
  135.         if (s->sc_dispatch == dispatch)
  136.             goto pmap_it;  /* he is registering another xptr */
  137.         return (FALSE);
  138.     }
  139.     s = (struct svc_callout *)mem_alloc(sizeof(struct svc_callout));
  140.     if (s == (struct svc_callout *)0) {
  141.         return (FALSE);
  142.     }
  143.     s->sc_prog = prog;
  144.     s->sc_vers = vers;
  145.     s->sc_dispatch = dispatch;
  146.     s->sc_next = svc_head;
  147.     svc_head = s;
  148. pmap_it:
  149.     /* now register the information with the local binder service */
  150.     /* port map service not available on the PC */
  151.         /* See-Mong Tan, 6/14/88 */
  152. /*  if (protocol) {
  153.         return (pmap_set(prog, vers, protocol, xprt->xp_port));
  154.     }
  155. */
  156.     return (TRUE);
  157. }
  158.  
  159. /*
  160.  * Remove a service program from the callout list.
  161.  */
  162. void
  163. svc_unregister(prog, vers)
  164.     u_long prog;
  165.     u_long vers;
  166. {
  167.     struct svc_callout *prev;
  168.     register struct svc_callout *s;
  169.  
  170.     if ((s = svc_find(prog, vers, &prev)) == NULL_SVC)
  171.         return;
  172.     if (prev == NULL_SVC) {
  173.         svc_head = s->sc_next;
  174.     } else {
  175.         prev->sc_next = s->sc_next;
  176.     }
  177.     s->sc_next = NULL_SVC;
  178.     mem_free((char *) s, (u_int) sizeof(struct svc_callout));
  179.     /* now unregister the information with the local binder service */
  180. /*  (void)pmap_unset(prog, vers);   */
  181. }
  182.  
  183. /*
  184.  * Search the callout list for a program number, return the callout
  185.  * struct.
  186.  */
  187. static struct svc_callout *
  188. svc_find(prog, vers, prev)
  189.     u_long prog;
  190.     u_long vers;
  191.     struct svc_callout **prev;
  192. {
  193.     register struct svc_callout *s, *p;
  194.  
  195.     p = NULL_SVC;
  196.     for (s = svc_head; s != NULL_SVC; s = s->sc_next) {
  197.         if ((s->sc_prog == prog) && (s->sc_vers == vers))
  198.             goto done;
  199.         p = s;
  200.     }
  201. done:
  202.     *prev = p;
  203.     return (s);
  204. }
  205.  
  206. /* ******************* REPLY GENERATION ROUTINES  ************ */
  207.  
  208. /*
  209.  * Send a reply to an rpc request
  210.  */
  211. bool_t
  212. svc_sendreply(xprt, xdr_results, xdr_location)
  213.     register SVCXPRT *xprt;
  214.     xdrproc_t xdr_results;
  215.     caddr_t xdr_location;
  216. {
  217.     struct rpc_msg rply; 
  218.  
  219.     rply.rm_direction = REPLY;  
  220.     rply.rm_reply.rp_stat = MSG_ACCEPTED; 
  221.     rply.acpted_rply.ar_verf = xprt->xp_verf; 
  222.     rply.acpted_rply.ar_stat = SUCCESS1;
  223.     rply.acpted_rply.ar_results.where = xdr_location;
  224.     rply.acpted_rply.ar_results.proc = xdr_results;
  225.     return (SVC_REPLY(xprt, &rply)); 
  226. }
  227.  
  228. /*
  229.  * No procedure error reply
  230.  */
  231. void
  232. svcerr_noproc(xprt)
  233.     register SVCXPRT *xprt;
  234. {
  235.     struct rpc_msg rply;
  236.  
  237.     rply.rm_direction = REPLY;
  238.     rply.rm_reply.rp_stat = MSG_ACCEPTED;
  239.     rply.acpted_rply.ar_verf = xprt->xp_verf;
  240.     rply.acpted_rply.ar_stat = PROC_UNAVAIL;
  241.     SVC_REPLY(xprt, &rply);
  242. }
  243.  
  244. /*
  245.  * Can't decode args error reply
  246.  */
  247. void
  248. svcerr_decode(xprt)
  249.     register SVCXPRT *xprt;
  250. {
  251.     struct rpc_msg rply; 
  252.  
  253.     rply.rm_direction = REPLY; 
  254.     rply.rm_reply.rp_stat = MSG_ACCEPTED; 
  255.     rply.acpted_rply.ar_verf = xprt->xp_verf;
  256.     rply.acpted_rply.ar_stat = GARBAGE_ARGS;
  257.     SVC_REPLY(xprt, &rply); 
  258. }
  259.  
  260. /*
  261.  * Some system error
  262.  */
  263. void
  264. svcerr_systemerr(xprt)
  265.     register SVCXPRT *xprt;
  266. {
  267.     struct rpc_msg rply; 
  268.  
  269.     rply.rm_direction = REPLY; 
  270.     rply.rm_reply.rp_stat = MSG_ACCEPTED; 
  271.     rply.acpted_rply.ar_verf = xprt->xp_verf;
  272.     rply.acpted_rply.ar_stat = SYSTEM_ERR;
  273.     SVC_REPLY(xprt, &rply); 
  274. }
  275.  
  276. /*
  277.  * Authentication error reply
  278.  */
  279. void
  280. svcerr_auth(xprt, why)
  281.     SVCXPRT *xprt;
  282.     enum auth_stat why;
  283. {
  284.     struct rpc_msg rply;
  285.  
  286.     rply.rm_direction = REPLY;
  287.     rply.rm_reply.rp_stat = MSG_DENIED;
  288.     rply.rjcted_rply.rj_stat = AUTH_ERROR;
  289.     rply.rjcted_rply.rj_why = why;
  290.     SVC_REPLY(xprt, &rply);
  291. }
  292.  
  293. /*
  294.  * Auth too weak error reply
  295.  */
  296. void
  297. svcerr_weakauth(xprt)
  298.     SVCXPRT *xprt;
  299. {
  300.  
  301.     svcerr_auth(xprt, AUTH_TOOWEAK);
  302. }
  303.  
  304. /*
  305.  * Program unavailable error reply
  306.  */
  307. void 
  308. svcerr_noprog(xprt)
  309.     register SVCXPRT *xprt;
  310. {
  311.     struct rpc_msg rply;  
  312.  
  313.     rply.rm_direction = REPLY;   
  314.     rply.rm_reply.rp_stat = MSG_ACCEPTED;  
  315.     rply.acpted_rply.ar_verf = xprt->xp_verf;  
  316.     rply.acpted_rply.ar_stat = PROG_UNAVAIL;
  317.     SVC_REPLY(xprt, &rply);
  318. }
  319.  
  320. /*
  321.  * Program version mismatch error reply
  322.  */
  323. void  
  324. svcerr_progvers(xprt, low_vers, high_vers)
  325.     register SVCXPRT *xprt; 
  326.     u_long low_vers;
  327.     u_long high_vers;
  328. {
  329.     struct rpc_msg rply;
  330.  
  331.     rply.rm_direction = REPLY;
  332.     rply.rm_reply.rp_stat = MSG_ACCEPTED;
  333.     rply.acpted_rply.ar_verf = xprt->xp_verf;
  334.     rply.acpted_rply.ar_stat = PROG_MISMATCH;
  335.     rply.acpted_rply.ar_vers.low = low_vers;
  336.     rply.acpted_rply.ar_vers.high = high_vers;
  337.     SVC_REPLY(xprt, &rply);
  338. }
  339.  
  340. /* ******************* SERVER INPUT STUFF ******************* */
  341.  
  342. /*
  343.  * Get server side input from some transport.
  344.  *
  345.  * Statement of authentication parameters management:
  346.  * This function owns and manages all authentication parameters, specifically
  347.  * the "raw" parameters (msg.rm_call.cb_cred and msg.rm_call.cb_verf) and
  348.  * the "cooked" credentials (rqst->rq_clntcred).  However, this function
  349.  * does not know the structure of the cooked credentials, so it make the
  350.  * following two assumptions: a) the structure is contiguous (no pointers), and
  351.  * b) the structure size does not exceed RQCRED_SIZE bytes. 
  352.  * In all events, all three parameters are freed upon exit from this routine.
  353.  * The storage is trivially management on the call stack in user land, but
  354.  * is mallocated in kernel land.
  355.  */
  356. void
  357. svc_getreq(rdfds)
  358. fd_set rdfds;
  359. {
  360.     register enum xprt_stat stat;
  361.     struct rpc_msg msg;
  362.     int prog_found;
  363.     u_long low_vers;
  364.     u_long high_vers;
  365.     struct svc_req r;
  366.     register int sock;
  367.     register SVCXPRT *xprt;
  368.     char cred_area[2*MAX_AUTH_BYTES + RQCRED_SIZE];
  369.     u_int i;
  370.  
  371.     msg.rm_call.cb_cred.oa_base = cred_area;
  372.     msg.rm_call.cb_verf.oa_base = &(cred_area[MAX_AUTH_BYTES]);
  373.     r.rq_clntcred = &(cred_area[2*MAX_AUTH_BYTES]);
  374.  
  375.     /* Loop through the sockets that have input ready */
  376.     for ( i=0; i<rdfds.fd_count; i++ ) {
  377.         sock = rdfds.fd_array[i];
  378.         /* sock has input waiting */
  379.         xprt = xports[sock];
  380.         /* now receive msgs from xprt (support batch calls) */
  381.         do {
  382.             if (SVC_RECV(xprt, &msg)) {
  383.                 /* now find the exported program and call it */
  384.                 register struct svc_callout *s;
  385.                 enum auth_stat why;
  386.  
  387.                 r.rq_xprt = xprt;
  388.                 r.rq_prog = msg.rm_call.cb_prog;
  389.                 r.rq_vers = msg.rm_call.cb_vers;
  390.                 r.rq_proc = msg.rm_call.cb_proc;
  391.                 r.rq_cred = msg.rm_call.cb_cred;
  392.                 /* first authenticate the message */
  393.                 if ((why= _authenticate(&r, &msg)) != AUTH_OK) {
  394.                     svcerr_auth(xprt, why);
  395.                     goto call_done;
  396.                 }
  397.                 /* now match message with a registered service*/
  398.                 prog_found = FALSE;
  399.                 low_vers = (unsigned)(0 - 1);
  400.                 high_vers = 0;
  401.                 for (s = svc_head; s != NULL_SVC; s = s->sc_next) {
  402.                     if (s->sc_prog == r.rq_prog) {
  403.                         if (s->sc_vers == r.rq_vers) {
  404.                             (*s->sc_dispatch)(&r, xprt);
  405.                             goto call_done;
  406.                         }  /* found correct version */
  407.                         prog_found = TRUE;
  408.                         if (s->sc_vers < low_vers)
  409.                             low_vers = s->sc_vers;
  410.                         if (s->sc_vers > high_vers)
  411.                             high_vers = s->sc_vers;
  412.                     }   /* found correct program */
  413.                 }
  414.                 /*
  415.                  * if we got here, the program or version
  416.                  * is not served ...
  417.                  */
  418.                 if (prog_found)
  419.                     svcerr_progvers(xprt,
  420.                     low_vers, high_vers);
  421.                 else
  422.                      svcerr_noprog(xprt);
  423.                 /* Fall through to ... */
  424.             }
  425.         call_done:
  426.             if ((stat = SVC_STAT(xprt)) == XPRT_DIED){
  427.                 SVC_DESTROY(xprt);
  428.                 break;
  429.             }
  430.         } while (stat == XPRT_MOREREQS);
  431.     }
  432. }
  433.  
  434. void
  435. svc_init()
  436. {
  437.     FD_ZERO(&svc_fds);
  438. }
  439.  
  440. /*
  441.  * This is the rpc server side idle loop
  442.  * Wait for input, call server program.
  443.  */
  444. void
  445. svc_run()
  446. {
  447.     fd_set readfds;
  448.     long n;
  449.     int i = 0;
  450.  
  451.     while (TRUE) {
  452. printf("svc_run() after while(%d).\n", ++i);
  453. #ifdef OLDSTUFF
  454.         if(kbhit()) {
  455.             if(getch() == 'q') {
  456.             printf("\nTermination requested.\n");
  457.             exit(0);
  458.             }
  459.         }
  460. #endif
  461.         /* make a copy of svc_fds, put it in readfds */
  462.         readfds = svc_fds;
  463.  
  464.         /* switch (sock_select(&readfds, (long *)0)) { */
  465. printf("svc_run() before select(%d).\n", i);
  466.         n = select(0,&readfds,(fd_set*)0,
  467.                 (fd_set*)0,(struct timeval *)0);
  468.         switch (n) {
  469.         case -1:
  470. printf("svc_run() error in transmission(%d).\n", i);
  471.             if (errno == EINTR || errno == EWOULDBLOCK)
  472.                 continue;
  473.             else {
  474.                 perror("svc.c: - Select failed");
  475.                 return;
  476.             }
  477.         case 0:
  478.             continue;
  479.         default:
  480. printf("svc_run() before svc_getreq(%d).\n", i);
  481.             svc_getreq(readfds);
  482.         }
  483.     }
  484. }
  485.