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 <libintl.h>
  44 #include <unistd.h>
  45 #include <rpc/rpc.h>
  46 #include <rpc/pmap_clnt.h>
  47 #include <sys/socket.h>
  48 #include <netdb.h>
  49
  50 #ifdef USE_IN_LIBIO
  51 # include <wchar.h>
  52 # include <libio/iolibio.h>
  53 #endif
  54
  55 struct proglst_
  56   {
  57     char *(*p_progname) (char *);
  58     int p_prognum;
  59     int p_procnum;
  60     xdrproc_t p_inproc, p_outproc;
  61     struct proglst_ *p_nxt;
  62   };
  63 #ifdef _RPC_THREAD_SAFE_
  64 #define proglst RPC_THREAD_VARIABLE(svcsimple_proglst_s)
  65 #else
  66 static struct proglst_ *proglst;
  67 #endif
  68
  69
  70 static void universal (struct svc_req *rqstp, SVCXPRT *transp_s);
  71 #ifdef _RPC_THREAD_SAFE_
  72 #define transp RPC_THREAD_VARIABLE(svcsimple_transp_s)
  73 #else
  74 static SVCXPRT *transp;
  75 #endif
  76
  77 int
  78 registerrpc (u_long prognum, u_long versnum, u_long procnum,
  79              char *(*progname) (char *), xdrproc_t inproc, xdrproc_t outproc)
  80 {
  81   struct proglst_ *pl;
  82   char *buf;
  83
  84   if (procnum == NULLPROC)
  85     {
  86
  87       if (__asprintf (&buf, _("can't reassign procedure number %ld\n"),
  88                       NULLPROC) < 0)
  89         buf = NULL;
  90       goto err_out;
  91     }
  92   if (transp == 0)
  93     {
  94       transp = INTUSE(svcudp_create) (RPC_ANYSOCK);
  95       if (transp == NULL)
  96         {
  97           buf = strdup (_("couldn't create an rpc server\n"));
  98           goto err_out;
  99         }
 100     }
 101   (void) pmap_unset ((u_long) prognum, (u_long) versnum);
 102   if (!svc_register (transp, (u_long) prognum, (u_long) versnum,
 103                      universal, IPPROTO_UDP))
 104     {
 105       if (__asprintf (&buf, _("couldn't register prog %ld vers %ld\n"),
 106                       prognum, versnum) < 0)
 107         buf = NULL;
 108       goto err_out;
 109     }
 110   pl = (struct proglst_ *) malloc (sizeof (struct proglst_));
 111   if (pl == NULL)
 112     {
 113       buf = strdup (_("registerrpc: out of memory\n"));
 114       goto err_out;
 115     }
 116   pl->p_progname = progname;
 117   pl->p_prognum = prognum;
 118   pl->p_procnum = procnum;
 119   pl->p_inproc = inproc;
 120   pl->p_outproc = outproc;
 121   pl->p_nxt = proglst;
 122   proglst = pl;
 123   return 0;
 124
 125  err_out:
 126   if (buf == NULL)
 127     return -1;
 128 #ifdef USE_IN_LIBIO
 129   if (_IO_fwide (stderr, 0) > 0)
 130     (void) __fwprintf (stderr, L"%s", buf);
 131   else
 132 #endif
 133     (void) fputs (buf, stderr);
 134   free (buf);
 135   return -1;
 136 }
 137
 138 static void
 139 universal (struct svc_req *rqstp, SVCXPRT *transp_l)
 140 {
 141   int prog, proc;
 142   char *outdata;
 143   char xdrbuf[UDPMSGSIZE];
 144   struct proglst_ *pl;
 145   char *buf = NULL;
 146
 147   /*
 148    * enforce "procnum 0 is echo" convention
 149    */
 150   if (rqstp->rq_proc == NULLPROC)
 151     {
 152       if (INTUSE(svc_sendreply) (transp_l, (xdrproc_t)INTUSE(xdr_void),
 153                                  (char *) NULL) == FALSE)
 154         {
 155           __write (STDERR_FILENO, "xxx\n", 4);
 156           exit (1);
 157         }
 158       return;
 159     }
 160   prog = rqstp->rq_prog;
 161   proc = rqstp->rq_proc;
 162   for (pl = proglst; pl != NULL; pl = pl->p_nxt)
 163     if (pl->p_prognum == prog && pl->p_procnum == proc)
 164       {
 165         /* decode arguments into a CLEAN buffer */
 166         __bzero (xdrbuf, sizeof (xdrbuf));      /* required ! */
 167         if (!svc_getargs (transp_l, pl->p_inproc, xdrbuf))
 168           {
 169             INTUSE(svcerr_decode) (transp_l);
 170             return;
 171           }
 172         outdata = (*(pl->p_progname)) (xdrbuf);
 173         if (outdata == NULL && pl->p_outproc != (xdrproc_t)INTUSE(xdr_void))
 174           /* there was an error */
 175           return;
 176         if (!INTUSE(svc_sendreply) (transp_l, pl->p_outproc, outdata))
 177           {
 178             if (__asprintf (&buf, _("trouble replying to prog %d\n"),
 179                             pl->p_prognum) < 0)
 180               buf = NULL;
 181             goto err_out2;
 182           }
 183         /* free the decoded arguments */
 184         (void) svc_freeargs (transp_l, pl->p_inproc, xdrbuf);
 185         return;
 186       }
 187   if (__asprintf (&buf, _("never registered prog %d\n"), prog) < 0)
 188     buf = NULL;
 189  err_out2:
 190   if (buf == NULL)
 191     exit (1);
 192 #ifdef USE_IN_LIBIO
 193   if (_IO_fwide (stderr, 0) > 0)
 194     __fwprintf (stderr, L"%s", buf);
 195   else
 196 #endif
 197     fputs (buf, stderr);
 198   free (buf);
 199   exit (1);
 200 }