sunrpc/svc_simple.c

   1 /* @(#)svc_simple.c     2.2 88/08/01 4.0 RPCSRC */
   2 /*
   3  * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
   4  * unrestricted use provided that this legend is included on all tape
   5  * media and as a part of the software program in whole or part.  Users
   6  * may copy or modify Sun RPC without charge, but are not authorized
   7  * to license or distribute it to anyone else except as part of a product or
   8  * program developed by the user.
   9  *
  10  * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
  11  * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
  12  * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
  13  *
  14  * Sun RPC is provided with no support and without any obligation on the
  15  * part of Sun Microsystems, Inc. to assist in its use, correction,
  16  * modification or enhancement.
  17  *
  18  * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
  19  * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
  20  * OR ANY PART THEREOF.
  21  *
  22  * In no event will Sun Microsystems, Inc. be liable for any lost revenue
  23  * or profits or other special, indirect and consequential damages, even if
  24  * Sun has been advised of the possibility of such damages.
  25  *
  26  * Sun Microsystems, Inc.
  27  * 2550 Garcia Avenue
  28  * Mountain View, California  94043
  29  */
  30 #if !defined(lint) && defined(SCCSIDS)
  31 static char sccsid[] = "@(#)svc_simple.c 1.18 87/08/11 Copyr 1984 Sun Micro";
  32 #endif
  33
  34 /*
  35  * svc_simple.c
  36  * Simplified front end to rpc.
  37  *
  38  * Copyright (C) 1984, Sun Microsystems, Inc.
  39  */
  40
  41 #include <stdio.h>
  42 #include <string.h>
  43 #include <rpc/rpc.h>
  44 #include <rpc/pmap_clnt.h>
  45 #include <sys/socket.h>
  46 #include <netdb.h>
  47
  48 #ifdef USE_IN_LIBIO
  49 # define fputs(s, f) _IO_fputs (s, f)
  50 #endif
  51
  52 static struct proglst
  53   {
  54     char *(*p_progname) (char *);
  55     int p_prognum;
  56     int p_procnum;
  57     xdrproc_t p_inproc, p_outproc;
  58     struct proglst *p_nxt;
  59   }
  60  *proglst;
  61
  62 static void universal (struct svc_req *rqstp, SVCXPRT *transp);
  63 static SVCXPRT *transp;
  64
  65 int
  66 registerrpc (u_long prognum, u_long versnum, u_long procnum,
  67              char *(*progname) (char *), xdrproc_t inproc, xdrproc_t outproc)
  68 {
  69   struct proglst *pl;
  70
  71   if (procnum == NULLPROC)
  72     {
  73       (void) fprintf (stderr,
  74                       _("can't reassign procedure number %d\n"), NULLPROC);
  75       return -1;
  76     }
  77   if (transp == 0)
  78     {
  79       transp = svcudp_create (RPC_ANYSOCK);
  80       if (transp == NULL)
  81         {
  82           (void) fputs (_("couldn't create an rpc server\n"), stderr);
  83           return -1;
  84         }
  85     }
  86   (void) pmap_unset ((u_long) prognum, (u_long) versnum);
  87   if (!svc_register (transp, (u_long) prognum, (u_long) versnum,
  88                      universal, IPPROTO_UDP))
  89     {
  90       (void) fprintf (stderr, _("couldn't register prog %d vers %d\n"),
  91                       prognum, versnum);
  92       return -1;
  93     }
  94   pl = (struct proglst *) malloc (sizeof (struct proglst));
  95   if (pl == NULL)
  96     {
  97       (void) fprintf (stderr, _("registerrpc: out of memory\n"));
  98       return -1;
  99     }
 100   pl->p_progname = progname;
 101   pl->p_prognum = prognum;
 102   pl->p_procnum = procnum;
 103   pl->p_inproc = inproc;
 104   pl->p_outproc = outproc;
 105   pl->p_nxt = proglst;
 106   proglst = pl;
 107   return 0;
 108 }
 109
 110 static void
 111 universal (struct svc_req *rqstp, SVCXPRT *transp)
 112 {
 113   int prog, proc;
 114   char *outdata;
 115   char xdrbuf[UDPMSGSIZE];
 116   struct proglst *pl;
 117
 118   /*
 119    * enforce "procnum 0 is echo" convention
 120    */
 121   if (rqstp->rq_proc == NULLPROC)
 122     {
 123       if (svc_sendreply (transp, (xdrproc_t)xdr_void, (char *) NULL) == FALSE)
 124         {
 125           (void) fprintf (stderr, "xxx\n");
 126           exit (1);
 127         }
 128       return;
 129     }
 130   prog = rqstp->rq_prog;
 131   proc = rqstp->rq_proc;
 132   for (pl = proglst; pl != NULL; pl = pl->p_nxt)
 133     if (pl->p_prognum == prog && pl->p_procnum == proc)
 134       {
 135         /* decode arguments into a CLEAN buffer */
 136         __bzero (xdrbuf, sizeof (xdrbuf));      /* required ! */
 137         if (!svc_getargs (transp, pl->p_inproc, xdrbuf))
 138           {
 139             svcerr_decode (transp);
 140             return;
 141           }
 142         outdata = (*(pl->p_progname)) (xdrbuf);
 143         if (outdata == NULL && pl->p_outproc != (xdrproc_t)xdr_void)
 144           /* there was an error */
 145           return;
 146         if (!svc_sendreply (transp, pl->p_outproc, outdata))
 147           {
 148             (void) fprintf (stderr,
 149                             _ ("trouble replying to prog %d\n"),
 150                             pl->p_prognum);
 151             exit (1);
 152           }
 153         /* free the decoded arguments */
 154         (void) svc_freeargs (transp, pl->p_inproc, xdrbuf);
 155         return;
 156       }
 157   (void) fprintf (stderr, _ ("never registered prog %d\n"), prog);
 158   exit (1);
 159 }