t_rpc.c revision 313535
1/* $NetBSD: t_rpc.c,v 1.10 2016/08/27 14:36:22 christos Exp $ */ 2 3#include <sys/cdefs.h> 4__RCSID("$NetBSD: t_rpc.c,v 1.10 2016/08/27 14:36:22 christos Exp $"); 5 6#include <sys/types.h> 7#include <sys/socket.h> 8#include <rpc/rpc.h> 9#include <stdlib.h> 10#include <string.h> 11#include <err.h> 12#include <netdb.h> 13#include <stdio.h> 14#include <errno.h> 15#include <unistd.h> 16 17#ifndef TEST 18#include <atf-c.h> 19 20#define ERRX(ev, msg, ...) ATF_REQUIRE_MSG(0, msg, __VA_ARGS__) 21 22#define SKIPX(ev, msg, ...) do { \ 23 atf_tc_skip(msg, __VA_ARGS__); \ 24 return ev; \ 25} while(/*CONSTCOND*/0) 26 27#ifdef __FreeBSD__ 28#define SKIPXI(ev, msg, ...) do { \ 29 atf_tc_skip(msg, __VA_ARGS__); \ 30 return ev; \ 31} while(/*CONSTCOND*/0) 32#endif 33 34#else 35#define ERRX(ev, msg, ...) errx(EXIT_FAILURE, msg, __VA_ARGS__) 36#define SKIPX(ev, msg, ...) errx(EXIT_FAILURE, msg, __VA_ARGS__) 37#endif 38 39#ifdef DEBUG 40#define DPRINTF(...) printf(__VA_ARGS__) 41#else 42#define DPRINTF(...) 43#endif 44 45 46#define RPCBPROC_NULL 0 47 48/* XXX (ngie): for clarity on what needs to be reverted later. */ 49#define __FreeBSD_bug_216954__ 50#ifdef __FreeBSD_bug_216954__ 51#include <signal.h> 52#endif 53 54static int 55reply(caddr_t replyp, struct netbuf * raddrp, struct netconfig * nconf) 56{ 57 char host[NI_MAXHOST]; 58 struct sockaddr *sock = raddrp->buf; 59 int error; 60 61 62 error = getnameinfo(sock, sock->sa_len, host, sizeof(host), NULL, 0, 0); 63 if (error) 64 warnx("Cannot resolve address (%s)", gai_strerror(error)); 65 else 66 printf("response from: %s\n", host); 67 return 0; 68} 69 70#ifdef __FreeBSD__ 71#define __rpc_control rpc_control 72#endif 73 74extern bool_t __rpc_control(int, void *); 75 76static void 77onehost(const char *host, const char *transp) 78{ 79 CLIENT *clnt; 80 struct netbuf addr; 81 struct timeval tv; 82 83 /* 84 * Magic! 85 */ 86 tv.tv_sec = 0; 87 tv.tv_usec = 500000; 88#define CLCR_SET_RPCB_TIMEOUT 2 89 __rpc_control(CLCR_SET_RPCB_TIMEOUT, &tv); 90 91 if ((clnt = clnt_create(host, RPCBPROG, RPCBVERS, transp)) == NULL) 92 SKIPX(, "clnt_create (%s)", clnt_spcreateerror("")); 93 94 tv.tv_sec = 1; 95 tv.tv_usec = 0; 96#ifdef __FreeBSD__ 97 if (clnt_call(clnt, RPCBPROC_NULL, (xdrproc_t)xdr_void, NULL, 98 (xdrproc_t)xdr_void, NULL, tv) 99 != RPC_SUCCESS) 100#else 101 if (clnt_call(clnt, RPCBPROC_NULL, xdr_void, NULL, xdr_void, NULL, tv) 102 != RPC_SUCCESS) 103#endif 104 ERRX(, "clnt_call (%s)", clnt_sperror(clnt, "")); 105 clnt_control(clnt, CLGET_SVC_ADDR, (char *) &addr); 106 reply(NULL, &addr, NULL); 107} 108 109#define PROGNUM 0x81 110#define VERSNUM 0x01 111#define PLUSONE 1 112#define DESTROY 2 113 114static struct timeval tout = {1, 0}; 115 116static void 117server(struct svc_req *rqstp, SVCXPRT *transp) 118{ 119 int num; 120 121 DPRINTF("Starting server\n"); 122 123 switch (rqstp->rq_proc) { 124 case NULLPROC: 125 if (!svc_sendreply(transp, (xdrproc_t)xdr_void, NULL)) 126 ERRX(, "svc_sendreply failed %d", 0); 127 return; 128 case PLUSONE: 129 break; 130 case DESTROY: 131 if (!svc_sendreply(transp, (xdrproc_t)xdr_void, NULL)) 132 ERRX(, "svc_sendreply failed %d", 0); 133 svc_destroy(transp); 134 exit(0); 135 default: 136 svcerr_noproc(transp); 137 return; 138 } 139 140 if (!svc_getargs(transp, (xdrproc_t)xdr_int, (void *)&num)) { 141 svcerr_decode(transp); 142 return; 143 } 144 DPRINTF("About to increment\n"); 145 num++; 146 if (!svc_sendreply(transp, (xdrproc_t)xdr_int, (void *)&num)) 147 ERRX(, "svc_sendreply failed %d", 1); 148 DPRINTF("Leaving server procedure.\n"); 149} 150 151static int 152rawtest(const char *arg) 153{ 154 CLIENT *clnt; 155 SVCXPRT *svc; 156 int num, resp; 157 enum clnt_stat rv; 158 159 if (arg) 160 num = atoi(arg); 161 else 162 num = 0; 163 164 svc = svc_raw_create(); 165 if (svc == NULL) 166 ERRX(EXIT_FAILURE, "Cannot create server %d", num); 167 if (!svc_reg(svc, PROGNUM, VERSNUM, server, NULL)) 168 ERRX(EXIT_FAILURE, "Cannot register server %d", num); 169 170 clnt = clnt_raw_create(PROGNUM, VERSNUM); 171 if (clnt == NULL) 172 ERRX(EXIT_FAILURE, "%s", 173 clnt_spcreateerror("clnt_raw_create")); 174 rv = clnt_call(clnt, PLUSONE, (xdrproc_t)xdr_int, (void *)&num, 175 (xdrproc_t)xdr_int, (void *)&resp, tout); 176 if (rv != RPC_SUCCESS) 177 ERRX(EXIT_FAILURE, "clnt_call: %s", clnt_sperrno(rv)); 178 DPRINTF("Got %d\n", resp); 179 clnt_destroy(clnt); 180 svc_destroy(svc); 181 if (++num != resp) 182 ERRX(EXIT_FAILURE, "expected %d got %d", num, resp); 183 184 return EXIT_SUCCESS; 185} 186 187static int 188regtest(const char *hostname, const char *transp, const char *arg, int p) 189{ 190 CLIENT *clnt; 191 int num, resp; 192 enum clnt_stat rv; 193 pid_t pid; 194 195 if (arg) 196 num = atoi(arg); 197 else 198 num = 0; 199 200#ifdef __NetBSD__ 201 svc_fdset_init(p ? SVC_FDSET_POLL : 0); 202#endif 203 if (!svc_create(server, PROGNUM, VERSNUM, transp)) 204 { 205 SKIPXI(EXIT_FAILURE, "Cannot create server %d", num); 206 } 207 208 switch ((pid = fork())) { 209 case 0: 210 DPRINTF("Calling svc_run\n"); 211 svc_run(); 212 ERRX(EXIT_FAILURE, "svc_run returned %d!", num); 213 case -1: 214 ERRX(EXIT_FAILURE, "Fork failed (%s)", strerror(errno)); 215 default: 216 sleep(1); 217 break; 218 } 219 220 DPRINTF("Initializing client\n"); 221 clnt = clnt_create(hostname, PROGNUM, VERSNUM, transp); 222 if (clnt == NULL) 223 ERRX(EXIT_FAILURE, "%s", 224 clnt_spcreateerror("clnt_raw_create")); 225 rv = clnt_call(clnt, PLUSONE, (xdrproc_t)xdr_int, (void *)&num, 226 (xdrproc_t)xdr_int, (void *)&resp, tout); 227 if (rv != RPC_SUCCESS) 228 ERRX(EXIT_FAILURE, "clnt_call: %s", clnt_sperrno(rv)); 229 DPRINTF("Got %d\n", resp); 230 if (++num != resp) 231 ERRX(EXIT_FAILURE, "expected %d got %d", num, resp); 232 rv = clnt_call(clnt, DESTROY, (xdrproc_t)xdr_void, NULL, 233 (xdrproc_t)xdr_void, NULL, tout); 234 if (rv != RPC_SUCCESS) 235 ERRX(EXIT_FAILURE, "clnt_call: %s", clnt_sperrno(rv)); 236 clnt_destroy(clnt); 237 238 return EXIT_SUCCESS; 239} 240 241 242#ifdef TEST 243static void 244allhosts(const char *transp) 245{ 246 enum clnt_stat clnt_stat; 247 248 clnt_stat = rpc_broadcast(RPCBPROG, RPCBVERS, RPCBPROC_NULL, 249 (xdrproc_t)xdr_void, NULL, (xdrproc_t)xdr_void, 250 NULL, (resultproc_t)reply, transp); 251 if (clnt_stat != RPC_SUCCESS && clnt_stat != RPC_TIMEDOUT) 252 ERRX(EXIT_FAILURE, "%s", clnt_sperrno(clnt_stat)); 253} 254 255int 256main(int argc, char *argv[]) 257{ 258 int ch; 259 int s, p; 260 const char *transp = "udp"; 261 262 p = s = 0; 263 while ((ch = getopt(argc, argv, "prstu")) != -1) 264 switch (ch) { 265 case 'p': 266 p = 1; 267 break; 268 case 's': 269 s = 1; 270 break; 271 case 't': 272 transp = "tcp"; 273 break; 274 case 'u': 275 transp = "udp"; 276 break; 277 case 'r': 278 transp = NULL; 279 break; 280 default: 281 fprintf(stderr, 282 "Usage: %s -[r|s|t|u] [<hostname>...]\n", 283 getprogname()); 284 return EXIT_FAILURE; 285 } 286 287 if (argc == optind) { 288 if (transp) 289 allhosts(transp); 290 else 291 rawtest(NULL); 292 } else { 293 for (; optind < argc; optind++) { 294 if (transp) 295 s == 0 ? 296 onehost(argv[optind], transp) : 297 regtest(argv[optind], transp, "1", p); 298 else 299 rawtest(argv[optind]); 300 } 301 } 302 303 return EXIT_SUCCESS; 304} 305 306#else 307 308ATF_TC(get_svc_addr_tcp); 309ATF_TC_HEAD(get_svc_addr_tcp, tc) 310{ 311 atf_tc_set_md_var(tc, "descr", "Checks CLGET_SVC_ADDR for tcp"); 312 313} 314 315ATF_TC_BODY(get_svc_addr_tcp, tc) 316{ 317 onehost("localhost", "tcp"); 318 319} 320 321ATF_TC(get_svc_addr_udp); 322ATF_TC_HEAD(get_svc_addr_udp, tc) 323{ 324 atf_tc_set_md_var(tc, "descr", "Checks CLGET_SVC_ADDR for udp"); 325} 326 327ATF_TC_BODY(get_svc_addr_udp, tc) 328{ 329 onehost("localhost", "udp"); 330 331} 332 333ATF_TC(raw); 334ATF_TC_HEAD(raw, tc) 335{ 336 atf_tc_set_md_var(tc, "descr", "Checks svc raw"); 337} 338 339ATF_TC_BODY(raw, tc) 340{ 341#ifdef __FreeBSD__ 342#ifdef __FreeBSD_bug_216954__ 343 atf_tc_expect_signal(SIGSEGV, 344 "fails with SIGSEGV only on ^/stable/10 -- bug # 216954"); 345#else 346 atf_tc_expect_fail("fails with: clnt_call: " 347 "RPC: Can't decode result -- PR # 211804"); 348#endif 349#endif 350 rawtest(NULL); 351 352} 353 354ATF_TC(tcp); 355ATF_TC_HEAD(tcp, tc) 356{ 357 atf_tc_set_md_var(tc, "descr", "Checks svc tcp (select)"); 358#ifdef __FreeBSD__ 359 atf_tc_set_md_var(tc, "require.user", "root"); 360#endif 361} 362 363ATF_TC_BODY(tcp, tc) 364{ 365 regtest("localhost", "tcp", "1", 0); 366 367} 368 369ATF_TC(udp); 370ATF_TC_HEAD(udp, tc) 371{ 372 atf_tc_set_md_var(tc, "descr", "Checks svc udp (select)"); 373#ifdef __FreeBSD__ 374 atf_tc_set_md_var(tc, "require.user", "root"); 375#endif 376} 377 378ATF_TC_BODY(udp, tc) 379{ 380 regtest("localhost", "udp", "1", 0); 381 382} 383 384ATF_TC(tcp_poll); 385ATF_TC_HEAD(tcp_poll, tc) 386{ 387 atf_tc_set_md_var(tc, "descr", "Checks svc tcp (poll)"); 388#ifdef __FreeBSD__ 389 atf_tc_set_md_var(tc, "require.user", "root"); 390#endif 391} 392 393ATF_TC_BODY(tcp_poll, tc) 394{ 395 regtest("localhost", "tcp", "1", 1); 396 397} 398 399ATF_TC(udp_poll); 400ATF_TC_HEAD(udp_poll, tc) 401{ 402 atf_tc_set_md_var(tc, "descr", "Checks svc udp (poll)"); 403#ifdef __FreeBSD__ 404 atf_tc_set_md_var(tc, "require.user", "root"); 405#endif 406} 407 408ATF_TC_BODY(udp_poll, tc) 409{ 410 regtest("localhost", "udp", "1", 1); 411 412} 413 414ATF_TP_ADD_TCS(tp) 415{ 416 ATF_TP_ADD_TC(tp, get_svc_addr_udp); 417 ATF_TP_ADD_TC(tp, get_svc_addr_tcp); 418 ATF_TP_ADD_TC(tp, raw); 419 ATF_TP_ADD_TC(tp, tcp); 420 ATF_TP_ADD_TC(tp, udp); 421 ATF_TP_ADD_TC(tp, tcp_poll); 422 ATF_TP_ADD_TC(tp, udp_poll); 423 424 return atf_no_error(); 425} 426 427#endif 428