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