transp_sockets.c revision 42629
1/* 2 * Copyright (c) 1997-1998 Erez Zadok 3 * Copyright (c) 1990 Jan-Simon Pendry 4 * Copyright (c) 1990 Imperial College of Science, Technology & Medicine 5 * Copyright (c) 1990 The Regents of the University of California. 6 * All rights reserved. 7 * 8 * This code is derived from software contributed to Berkeley by 9 * Jan-Simon Pendry at Imperial College, London. 10 * 11 * Redistribution and use in source and binary forms, with or without 12 * modification, are permitted provided that the following conditions 13 * are met: 14 * 1. Redistributions of source code must retain the above copyright 15 * notice, this list of conditions and the following disclaimer. 16 * 2. Redistributions in binary form must reproduce the above copyright 17 * notice, this list of conditions and the following disclaimer in the 18 * documentation and/or other materials provided with the distribution. 19 * 3. All advertising materials mentioning features or use of this software 20 * must display the following acknowledgment: 21 * This product includes software developed by the University of 22 * California, Berkeley and its contributors. 23 * 4. Neither the name of the University nor the names of its contributors 24 * may be used to endorse or promote products derived from this software 25 * without specific prior written permission. 26 * 27 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 28 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 29 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 30 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 31 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 32 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 33 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 34 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 35 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 36 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 37 * SUCH DAMAGE. 38 * 39 * %W% (Berkeley) %G% 40 * 41 * $Id: transp_sockets.c,v 1.2 1998/11/10 16:23:41 ezk Exp $ 42 * 43 * Socket specific utilities. 44 * -Erez Zadok <ezk@cs.columbia.edu> 45 */ 46 47#ifdef HAVE_CONFIG_H 48# include <config.h> 49#endif /* HAVE_CONFIG_H */ 50#include <am_defs.h> 51#include <amu.h> 52 53 54/* 55 * find the IP address that can be used to connect to the local host 56 */ 57void 58amu_get_myaddress(struct in_addr *iap) 59{ 60 struct sockaddr_in sin; 61 62 memset((char *) &sin, 0, sizeof(sin)); 63 get_myaddress(&sin); 64 iap->s_addr = sin.sin_addr.s_addr; 65} 66 67 68/* 69 * How to bind to reserved ports. 70 */ 71int 72bind_resv_port(int so, u_short *pp) 73{ 74 struct sockaddr_in sin; 75 int rc; 76 u_short port; 77 78 memset((voidp) &sin, 0, sizeof(sin)); 79 sin.sin_family = AF_INET; 80 81 port = IPPORT_RESERVED; 82 83 do { 84 --port; 85 sin.sin_port = htons(port); 86 rc = bind(so, (struct sockaddr *) &sin, sizeof(sin)); 87 } while (rc < 0 && (int) port > IPPORT_RESERVED / 2); 88 89 if (pp && rc == 0) 90 *pp = port; 91 92 return rc; 93} 94 95 96/* 97 * close a descriptor, Sockets style 98 */ 99int 100amu_close(int fd) 101{ 102 return close(fd); 103} 104 105 106/* 107 * Create an rpc client attached to the mount daemon. 108 */ 109CLIENT * 110get_mount_client(char *unused_host, struct sockaddr_in *sin, struct timeval *tv, int *sock, u_long mnt_version) 111{ 112 CLIENT *client; 113 114 /* 115 * First try a TCP socket 116 */ 117 if ((*sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) > 0) { 118 /* 119 * Bind to a privileged port 120 */ 121 if (bind_resv_port(*sock, (u_short *) 0) < 0) 122 plog(XLOG_ERROR, "can't bind privileged port"); 123 124 /* 125 * Find mountd port to connect to. 126 * Connect to mountd. 127 * Create a tcp client. 128 */ 129 if ((sin->sin_port = htons(pmap_getport(sin, MOUNTPROG, mnt_version, IPPROTO_TCP))) != 0) { 130 if (connect(*sock, (struct sockaddr *) sin, sizeof(*sin)) >= 0 131 && ((client = clnttcp_create(sin, MOUNTPROG, mnt_version, sock, 0, 0)) != NULL)) 132 return client; 133 } 134 /* 135 * Failed so close socket 136 */ 137 (void) close(*sock); 138 } /* tcp socket opened */ 139 /* TCP failed so try UDP */ 140 if ((*sock = socket(AF_INET, SOCK_DGRAM, 0)) < 0) { 141 plog(XLOG_ERROR, "Can't create socket to connect to mountd: %m"); 142 *sock = RPC_ANYSOCK; 143 return NULL; 144 } 145 /* 146 * Bind to a privileged port 147 */ 148 if (bind_resv_port(*sock, (u_short *) 0) < 0) 149 plog(XLOG_ERROR, "can't bind privileged port"); 150 151 /* 152 * Zero out the port - make sure we recompute 153 */ 154 sin->sin_port = 0; 155 156 /* 157 * Make a UDP client 158 */ 159 if ((client = clntudp_create(sin, MOUNTPROG, mnt_version, *tv, sock)) == NULL) { 160 (void) close(*sock); 161 *sock = RPC_ANYSOCK; 162 return NULL; 163 } 164#ifdef DEBUG 165 dlog("get_mount_client: Using udp, port %d", sin->sin_port); 166#endif /* DEBUG */ 167 return client; 168} 169 170 171/* 172 * find the address of the caller of an RPC procedure. 173 */ 174struct sockaddr_in * 175amu_svc_getcaller(SVCXPRT *xprt) 176{ 177 return svc_getcaller(xprt); 178} 179 180 181/* 182 * Bind NFS to a reserved port. 183 */ 184static int 185bindnfs_port(int so, u_short *nfs_portp) 186{ 187 u_short port; 188 int error = bind_resv_port(so, &port); 189 190 if (error == 0) 191 *nfs_portp = port; 192 return error; 193} 194 195 196/* 197 * Create the nfs service for amd 198 */ 199int 200create_nfs_service(int *soNFSp, u_short *nfs_portp, SVCXPRT **nfs_xprtp, void (*dispatch_fxn)(struct svc_req *rqstp, SVCXPRT *transp)) 201{ 202 203 *soNFSp = socket(AF_INET, SOCK_DGRAM, 0); 204 205 if (*soNFSp < 0 || bindnfs_port(*soNFSp, nfs_portp) < 0) { 206 plog(XLOG_FATAL, "Can't create privileged nfs port"); 207 return 1; 208 } 209 if ((*nfs_xprtp = svcudp_create(*soNFSp)) == NULL) { 210 plog(XLOG_FATAL, "cannot create rpc/udp service"); 211 return 2; 212 } 213 if (!svc_register(*nfs_xprtp, NFS_PROGRAM, NFS_VERSION, dispatch_fxn, 0)) { 214 plog(XLOG_FATAL, "unable to register (NFS_PROGRAM, NFS_VERSION, 0)"); 215 return 3; 216 } 217 218 return 0; /* all is well */ 219} 220 221 222/* 223 * Create the amq service for amd (both TCP and UDP) 224 */ 225int 226create_amq_service(int *udp_soAMQp, SVCXPRT **udp_amqpp, int *tcp_soAMQp, SVCXPRT **tcp_amqpp) 227{ 228 /* first create TCP service */ 229 if (tcp_soAMQp) { 230 *tcp_soAMQp = socket(AF_INET, SOCK_STREAM, 0); 231 if (*tcp_soAMQp < 0) { 232 plog(XLOG_FATAL, "cannot create tcp socket for amq service: %m"); 233 return 1; 234 } 235 236 /* now create RPC service handle for amq */ 237 if (tcp_amqpp && 238 (*tcp_amqpp = svctcp_create(*tcp_soAMQp, AMQ_SIZE, AMQ_SIZE)) == NULL) { 239 plog(XLOG_FATAL, "cannot create tcp service for amq: soAMQp=%d", *tcp_soAMQp); 240 return 2; 241 } 242 } 243 244 /* next create UDP service */ 245 if (udp_soAMQp) { 246 *udp_soAMQp = socket(AF_INET, SOCK_DGRAM, 0); 247 if (*udp_soAMQp < 0) { 248 plog(XLOG_FATAL, "cannot create udp socket for amq service: %m"); 249 return 3; 250 } 251 252 /* now create RPC service handle for amq */ 253 if (udp_amqpp && 254 (*udp_amqpp = svcudp_bufcreate(*udp_soAMQp, AMQ_SIZE, AMQ_SIZE)) == NULL) { 255 plog(XLOG_FATAL, "cannot create udp service for amq: soAMQp=%d", *udp_soAMQp); 256 return 4; 257 } 258 } 259 260 return 0; /* all is well */ 261} 262 263 264/* 265 * Ping the portmapper on a remote system by calling the nullproc 266 */ 267enum clnt_stat 268pmap_ping(struct sockaddr_in *address) 269{ 270 CLIENT *client; 271 enum clnt_stat clnt_stat = RPC_TIMEDOUT; /* assume failure */ 272 int socket = RPC_ANYSOCK; 273 struct timeval timeout; 274 275 timeout.tv_sec = 3; 276 timeout.tv_usec = 0; 277 address->sin_port = htons(PMAPPORT); 278 client = clntudp_create(address, PMAPPROG, PMAPVERS, timeout, &socket); 279 if (client != (CLIENT *) NULL) { 280 clnt_stat = clnt_call(client, 281 PMAPPROC_NULL, 282 (XDRPROC_T_TYPE) xdr_void, 283 NULL, 284 (XDRPROC_T_TYPE) xdr_void, 285 NULL, 286 timeout); 287 clnt_destroy(client); 288 } 289 close(socket); 290 address->sin_port = 0; 291 292 return clnt_stat; 293} 294 295 296/* 297 * Find the best NFS version for a host and protocol. 298 */ 299u_long 300get_nfs_version(char *host, struct sockaddr_in *sin, u_long nfs_version, const char *proto) 301{ 302 CLIENT *clnt; 303 int again = 0; 304 enum clnt_stat clnt_stat; 305 struct timeval tv; 306 int sock; 307 308 /* 309 * If not set or set wrong, then try from NFS_VERS_MAX on down. If 310 * set, then try from nfs_version on down. 311 */ 312 if (nfs_version <= 0 || nfs_version > NFS_VERS_MAX) { 313 nfs_version = NFS_VERS_MAX; 314 again = 1; 315 } 316 tv.tv_sec = 3; /* retry every 3 seconds, but also timeout */ 317 tv.tv_usec = 0; 318 319 /* 320 * First check if remote portmapper is up (verify if remote host is up). 321 */ 322 clnt_stat = pmap_ping(sin); 323 if (clnt_stat == RPC_TIMEDOUT) { 324 plog(XLOG_ERROR, "get_nfs_version: failed to contact portmapper on host \"%s\": %s", host, clnt_sperrno(clnt_stat)); 325 return 0; 326 } 327 328#ifdef HAVE_FS_NFS3 329try_again: 330#endif /* HAVE_FS_NFS3 */ 331 332 sock = RPC_ANYSOCK; 333 if (STREQ(proto, "tcp")) 334 clnt = clnttcp_create(sin, NFS_PROGRAM, nfs_version, &sock, 0, 0); 335 else if (STREQ(proto, "udp")) 336 clnt = clntudp_create(sin, NFS_PROGRAM, nfs_version, tv, &sock); 337 else 338 clnt = NULL; 339 340 if (clnt == NULL) { 341#ifdef HAVE_CLNT_SPCREATEERROR 342 plog(XLOG_INFO, "get_nfs_version NFS(%d,%s) failed for %s :%s", 343 nfs_version, proto, host, clnt_spcreateerror("")); 344#else /* not HAVE_CLNT_SPCREATEERROR */ 345 plog(XLOG_INFO, "get_nfs_version NFS(%d,%s) failed for %s", 346 nfs_version, proto, host); 347#endif /* not HAVE_CLNT_SPCREATEERROR */ 348 return 0; 349 } 350 351 /* Try a couple times to verify the CLIENT handle. */ 352 tv.tv_sec = 6; 353 clnt_stat = clnt_call(clnt, 354 NFSPROC_NULL, 355 (XDRPROC_T_TYPE) xdr_void, 356 0, 357 (XDRPROC_T_TYPE) xdr_void, 358 0, 359 tv); 360 close(sock); 361 clnt_destroy(clnt); 362 if (clnt_stat != RPC_SUCCESS) { 363 if (again) { 364#ifdef HAVE_FS_NFS3 365 if (nfs_version == NFS_VERSION3) { 366 plog(XLOG_INFO, "get_nfs_version trying a lower version"); 367 nfs_version = NFS_VERSION; 368 again = 0; 369 } 370 goto try_again; 371#endif /* HAVE_FS_NFS3 */ 372 } 373 plog(XLOG_INFO, "get_nfs_version NFS(%d,%s) failed for %s", 374 nfs_version, proto, host); 375 return 0; 376 } 377 378 plog(XLOG_INFO, "get_nfs_version: returning (%d,%s) on host %s", 379 nfs_version, proto, host); 380 return nfs_version; 381} 382 383 384/* 385 * AUTOFS FUNCTIONS FOR SOCKETS: 386 */ 387#ifdef HAVE_FS_AUTOFS 388/* 389 * Create the nfs service for amd 390 */ 391int 392create_autofs_service(int *soAUTOFSp, u_short *autofs_portp, SVCXPRT **autofs_xprtp, void (*dispatch_fxn)(struct svc_req *rqstp, SVCXPRT *transp)) 393{ 394 /* NOT IMPLEMENTED! */ 395 return -1; 396} 397#endif /* HAVE_FS_AUTOFS */ 398