svc_simple.c 5.0 KB

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