Add local assert function to avoid problems with memory allocation in
[glibc/pb-stable.git] / sunrpc / thrsvc.c
blob63cfbf9810da96ada4e64c6dc07d5a55967f8432
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>
8 #define PROGNUM 1234
9 #define VERSNUM 1
10 #define PROCNUM 1
11 #define PROCQUIT 2
13 static int exitcode;
15 struct rpc_arg
17 CLIENT *client;
18 u_long proc;
21 static void
22 dispatch(struct svc_req *request, SVCXPRT *xprt)
24 svc_sendreply(xprt, (xdrproc_t)xdr_void, 0);
25 if (request->rq_proc == PROCQUIT)
26 exit (0);
29 static void
30 test_one_call (struct rpc_arg *a)
32 struct timeval tout = { 60, 0 };
33 enum clnt_stat result;
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
43 clnt_perrno (result);
44 putchar ('\n');
45 exitcode = 1;
49 static void *
50 thread_wrapper (void *arg)
52 struct rpc_arg a;
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;
63 int
64 main (void)
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;
76 svx = svcudp_create (RPC_ANYSOCK);
77 svc_register (svx, PROGNUM, VERSNUM, dispatch, 0);
79 pid = fork ();
80 if (pid == -1)
82 perror ("fork");
83 return 1;
85 if (pid == 0)
86 svc_run ();
88 inet_aton ("127.0.0.1", &sin.sin_addr);
89 sin.sin_port = htons (svx->xp_port);
90 sin.sin_family = AF_INET;
92 clnt = clntudp_create (&sin, PROGNUM, VERSNUM, wait, &sock);
94 a.client = clnt;
95 a.proc = PROCNUM;
97 /* Test in this thread */
98 test_one_call (&a);
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));
108 return exitcode;