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 (format: "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 (s: "success" ); |
41 | else |
42 | { |
43 | clnt_perrno (result); |
44 | putchar (c: '\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: &a); |
57 | a.client = (CLIENT *)arg; |
58 | a.proc = PROCQUIT; |
59 | test_one_call (a: &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 (cp: "127.0.0.1" , inp: &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: &a); |
99 | |
100 | /* Test in a child thread */ |
101 | err = pthread_create (newthread: &tid, attr: 0, start_routine: thread_wrapper, arg: (void *) clnt); |
102 | if (err) |
103 | fprintf (stderr, "pthread_create: %s\n" , strerror (errnum: err)); |
104 | err = pthread_join (th: tid, thread_return: 0); |
105 | if (err) |
106 | fprintf (stderr, "pthread_join: %s\n" , strerror (errnum: err)); |
107 | |
108 | return exitcode; |
109 | } |
110 | |