lib/libcr/rpc/svc_simple.c

   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
  30 #if defined(LIBC_SCCS) && !defined(lint)
  31 /*static char *sccsid = "from: @(#)svc_simple.c 1.18 87/08/11 Copyr 1984 Sun Micro";*/
  32 /*static char *sccsid = "from: @(#)svc_simple.c 2.2 88/08/01 4.0 RPCSRC";*/
  33 static char *rcsid = "$FreeBSD: src/lib/libc/rpc/svc_simple.c,v 1.9 1999/08/28 00:00:50 peter Exp $";
  34 #endif
  35
  36 /*
  37  * svc_simple.c
  38  * Simplified front end to rpc.
  39  *
  40  * Copyright (C) 1984, Sun Microsystems, Inc.
  41  */
  42
  43 #include <stdio.h>
  44 #include <stdlib.h>
  45 #include <string.h>
  46 #include <rpc/rpc.h>
  47 #include <rpc/pmap_clnt.h>
  48 #include <sys/socket.h>
  49 #include <netdb.h>
  50
  51 static struct proglst {
  52         char *(*p_progname)();
  53         int  p_prognum;
  54         int  p_procnum;
  55         xdrproc_t p_inproc, p_outproc;
  56         struct proglst *p_nxt;
  57 } *proglst;
  58 static void universal();
  59 static SVCXPRT *transp;
  60 struct proglst *pl;
  61
  62 int
  63 registerrpc(prognum, versnum, procnum, progname, inproc, outproc)
  64         int prognum, versnum, procnum;
  65         char *(*progname)();
  66         xdrproc_t inproc, outproc;
  67 {
  68
  69         if (procnum == NULLPROC) {
  70                 (void) fprintf(stderr,
  71                     "can't reassign procedure number %ld\n", NULLPROC);
  72                 return (-1);
  73         }
  74         if (transp == 0) {
  75                 transp = svcudp_create(RPC_ANYSOCK);
  76                 if (transp == NULL) {
  77                         (void) fprintf(stderr, "couldn't create an rpc server\n");
  78                         return (-1);
  79                 }
  80         }
  81         (void) pmap_unset((u_long)prognum, (u_long)versnum);
  82         if (!svc_register(transp, (u_long)prognum, (u_long)versnum,
  83             universal, IPPROTO_UDP)) {
  84                 (void) fprintf(stderr, "couldn't register prog %d vers %d\n",
  85                     prognum, versnum);
  86                 return (-1);
  87         }
  88         pl = (struct proglst *)malloc(sizeof(struct proglst));
  89         if (pl == NULL) {
  90                 (void) fprintf(stderr, "registerrpc: out of memory\n");
  91                 return (-1);
  92         }
  93         pl->p_progname = progname;
  94         pl->p_prognum = prognum;
  95         pl->p_procnum = procnum;
  96         pl->p_inproc = inproc;
  97         pl->p_outproc = outproc;
  98         pl->p_nxt = proglst;
  99         proglst = pl;
 100         return (0);
 101 }
 102
 103 static void
 104 universal(rqstp, transp)
 105         struct svc_req *rqstp;
 106         SVCXPRT *transp;
 107 {
 108         int prog, proc;
 109         char *outdata;
 110         char xdrbuf[UDPMSGSIZE];
 111         struct proglst *pl;
 112
 113         /*
 114          * enforce "procnum 0 is echo" convention
 115          */
 116         if (rqstp->rq_proc == NULLPROC) {
 117                 if (svc_sendreply(transp, xdr_void, NULL) == FALSE) {
 118                         (void) fprintf(stderr, "xxx\n");
 119                         exit(1);
 120                 }
 121                 return;
 122         }
 123         prog = rqstp->rq_prog;
 124         proc = rqstp->rq_proc;
 125         for (pl = proglst; pl != NULL; pl = pl->p_nxt)
 126                 if (pl->p_prognum == prog && pl->p_procnum == proc) {
 127                         /* decode arguments into a CLEAN buffer */
 128                         memset(xdrbuf, 0, sizeof(xdrbuf)); /* required ! */
 129                         if (!svc_getargs(transp, pl->p_inproc, xdrbuf)) {
 130                                 svcerr_decode(transp);
 131                                 return;
 132                         }
 133                         outdata = (*(pl->p_progname))(xdrbuf);
 134                         if (outdata == NULL && pl->p_outproc != xdr_void)
 135                                 /* there was an error */
 136                                 return;
 137                         if (!svc_sendreply(transp, pl->p_outproc, outdata)) {
 138                                 (void) fprintf(stderr,
 139                                     "trouble replying to prog %d\n",
 140                                     pl->p_prognum);
 141                                 exit(1);
 142                         }
 143                         /* free the decoded arguments */
 144                         (void)svc_freeargs(transp, pl->p_inproc, xdrbuf);
 145                         return;
 146                 }
 147         (void) fprintf(stderr, "never registered prog %d\n", prog);
 148         exit(1);
 149 }
 150