sunrpc/thrsvc.c

   1 #include <pthread.h>
   2 #include <stdio.h>
   3 #include <string.h>
   4 #include <unistd.h>
   5 #include <rpc/rpc.h>
   6 #include <arpa/inet.h>
   7
   8 #define PROGNUM 1234
   9 #define VERSNUM 1
  10 #define PROCNUM 1
  11 #define PROCQUIT 2
  12
  13 static int exitcode;
  14
  15 struct rpc_arg
  16 {
  17   CLIENT *client;
  18   u_long proc;
  19 };
  20
  21 static void
  22 dispatch(struct svc_req *request, SVCXPRT *xprt)
  23 {
  24   svc_sendreply(xprt, (xdrproc_t)xdr_void, 0);
  25   if (request->rq_proc == PROCQUIT)
  26     exit (0);
  27 }
  28
  29 static void
  30 test_one_call (struct rpc_arg *a)
  31 {
  32   struct timeval tout = { 60, 0 };
  33   enum clnt_stat result;
  34
  35   printf ("test_one_call: ");
  36   result = clnt_call (a->client, a->proc,
  37                       (xdrproc_t) xdr_void, 0,
  38                       (xdrproc_t) xdr_void, 0, tout);
  39   if (result == RPC_SUCCESS)
  40     puts ("success");
  41   else
  42     {
  43       clnt_perrno (result);
  44       putchar ('\n');
  45       exitcode = 1;
  46     }
  47 }
  48
  49 static void *
  50 thread_wrapper (void *arg)
  51 {
  52   struct rpc_arg a;
  53
  54   a.client = (CLIENT *)arg;
  55   a.proc = PROCNUM;
  56   test_one_call (&a);
  57   a.client = (CLIENT *)arg;
  58   a.proc = PROCQUIT;
  59   test_one_call (&a);
  60   return 0;
  61 }
  62
  63 int
  64 main (void)
  65 {
  66   pthread_t tid;
  67   pid_t pid;
  68   int err;
  69   SVCXPRT *svx;
  70   CLIENT *clnt;
  71   struct sockaddr_in sin;
  72   struct timeval wait = { 5, 0 };
  73   int sock = RPC_ANYSOCK;
  74   struct rpc_arg a;
  75
  76   svx = svcudp_create (RPC_ANYSOCK);
  77   svc_register (svx, PROGNUM, VERSNUM, dispatch, 0);
  78
  79   pid = fork ();
  80   if (pid == -1)
  81     {
  82       perror ("fork");
  83       return 1;
  84     }
  85   if (pid == 0)
  86     svc_run ();
  87
  88   inet_aton ("127.0.0.1", &sin.sin_addr);
  89   sin.sin_port = htons (svx->xp_port);
  90   sin.sin_family = AF_INET;
  91
  92   clnt = clntudp_create (&sin, PROGNUM, VERSNUM, wait, &sock);
  93
  94   a.client = clnt;
  95   a.proc = PROCNUM;
  96
  97   /* Test in this thread */
  98   test_one_call (&a);
  99
 100   /* Test in a child thread */
 101   err = pthread_create (&tid, 0, thread_wrapper, (void *) clnt);
 102   if (err)
 103     fprintf (stderr, "pthread_create: %s\n", strerror (err));
 104   err = pthread_join (tid, 0);
 105   if (err)
 106     fprintf (stderr, "pthread_join: %s\n", strerror (err));
 107
 108   return exitcode;
 109 }