usr.sbin/ypbind/yp_ping.c

   1 /*
   2  * Copyright (c) 1996, 1997
   3  *      Bill Paul <wpaul@ctr.columbia.edu>.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by Bill Paul.
  16  * 4. Neither the name of the author nor the names of any co-contributors
  17  *    may be used to endorse or promote products derived from this software
  18  *    without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY Bill Paul AND CONTRIBUTORS ``AS IS'' AND
  21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  23  * ARE DISCLAIMED.  IN NO EVENT SHALL Bill Paul OR CONTRIBUTORS BE LIABLE
  24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  30  * SUCH DAMAGE.
  31  *
  32  * @(#)from: clnt_udp.c 1.39 87/08/11 Copyr 1984 Sun Micro
  33  * @(#)from: clnt_udp.c 2.2 88/08/01 4.0 RPCSRC
  34  * $FreeBSD: src/usr.sbin/ypbind/yp_ping.c,v 1.17 2008/09/15 14:01:40 dfr Exp $
  35  * $DragonFly: src/usr.sbin/ypbind/yp_ping.c,v 1.7 2005/11/24 22:23:02 swildner Exp $
  36  */
  37
  38 /*
  39  * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
  40  * unrestricted use provided that this legend is included on all tape
  41  * media and as a part of the software program in whole or part.  Users
  42  * may copy or modify Sun RPC without charge, but are not authorized
  43  * to license or distribute it to anyone else except as part of a product or
  44  * program developed by the user.
  45  *
  46  * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
  47  * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
  48  * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
  49  *
  50  * Sun RPC is provided with no support and without any obligation on the
  51  * part of Sun Microsystems, Inc. to assist in its use, correction,
  52  * modification or enhancement.
  53  *
  54  * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
  55  * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
  56  * OR ANY PART THEREOF.
  57  *
  58  * In no event will Sun Microsystems, Inc. be liable for any lost revenue
  59  * or profits or other special, indirect and consequential damages, even if
  60  * Sun has been advised of the possibility of such damages.
  61  *
  62  * Sun Microsystems, Inc.
  63  * 2550 Garcia Avenue
  64  * Mountain View, California  94043
  65  */
  66
  67 /*
  68  * clnt_udp.c, Implements a UDP/IP based, client side RPC.
  69  *
  70  * Copyright (C) 1984, Sun Microsystems, Inc.
  71  */
  72
  73 #include <errno.h>
  74 #include <netdb.h>
  75 #include <stdio.h>
  76 #include <stdlib.h>
  77 #include <string.h>
  78 #include <unistd.h>
  79 #include <pthread.h>
  80 #include <rpc/rpc.h>
  81 #include <rpc/pmap_clnt.h>
  82 #include <rpc/pmap_prot.h>
  83 #include <rpcsvc/yp.h>
  84 #include <sys/types.h>
  85 #include <sys/poll.h>
  86 #include <sys/socket.h>
  87 #include <sys/signal.h>
  88 #include <sys/ioctl.h>
  89 #include <arpa/inet.h>
  90 #include <net/if.h>
  91
  92 #include "yp_ping.h"
  93
  94 /*
  95  * pmap_getport.c
  96  * Client interface to pmap rpc service.
  97  *
  98  * Copyright (C) 1984, Sun Microsystems, Inc.
  99  */
 100
 101
 102 static struct timeval timeout = { 1, 0 };
 103 static struct timeval tottimeout = { 1, 0 };
 104
 105 /*
 106  * Find the mapped port for program,version.
 107  * Calls the pmap service remotely to do the lookup.
 108  * Returns 0 if no map exists.
 109  */
 110 static u_short
 111 __pmap_getport(struct sockaddr_in *address, u_long program, u_long version,
 112                u_int protocol)
 113 {
 114         u_short port = 0;
 115         int sock = -1;
 116         CLIENT *client;
 117         struct pmap parms;
 118
 119         address->sin_port = htons(PMAPPORT);
 120
 121         client = clntudp_bufcreate(address, PMAPPROG,
 122             PMAPVERS, timeout, &sock, RPCSMALLMSGSIZE, RPCSMALLMSGSIZE);
 123         if (client != NULL) {
 124                 parms.pm_prog = program;
 125                 parms.pm_vers = version;
 126                 parms.pm_prot = protocol;
 127                 parms.pm_port = 0;  /* not needed or used */
 128                 if (CLNT_CALL(client, PMAPPROC_GETPORT,
 129                         (xdrproc_t)xdr_pmap, &parms,
 130                         (xdrproc_t)xdr_u_short, &port,
 131                         tottimeout) != RPC_SUCCESS){
 132                         rpc_createerr.cf_stat = RPC_PMAPFAILURE;
 133                         clnt_geterr(client, &rpc_createerr.cf_error);
 134                 } else if (port == 0) {
 135                         rpc_createerr.cf_stat = RPC_PROGNOTREGISTERED;
 136                 }
 137                 CLNT_DESTROY(client);
 138         }
 139         if (sock != -1)
 140                 close(sock);
 141         address->sin_port = 0;
 142         return (port);
 143 }
 144
 145 /*
 146  * Transmit to YPPROC_DOMAIN_NONACK, return immediately.
 147  */
 148 static bool_t *
 149 ypproc_domain_nonack_2_send(domainname *argp, CLIENT *clnt)
 150 {
 151         static bool_t clnt_res;
 152         struct timeval TIMEOUT = { 0, 0 };
 153
 154         memset((char *)&clnt_res, 0, sizeof (clnt_res));
 155         if (clnt_call(clnt, YPPROC_DOMAIN_NONACK,
 156                 (xdrproc_t) xdr_domainname, (caddr_t) argp,
 157                 (xdrproc_t) xdr_bool, (caddr_t) &clnt_res,
 158                 TIMEOUT) != RPC_SUCCESS) {
 159                 return (NULL);
 160         }
 161         return (&clnt_res);
 162 }
 163
 164 /*
 165  * Receive response from YPPROC_DOMAIN_NONACK asynchronously.
 166  */
 167 static bool_t *
 168 ypproc_domain_nonack_2_recv(domainname *argp, CLIENT *clnt)
 169 {
 170         static bool_t clnt_res;
 171         struct timeval TIMEOUT = { 0, 0 };
 172
 173         memset((char *)&clnt_res, 0, sizeof (clnt_res));
 174         if (clnt_call(clnt, YPPROC_DOMAIN_NONACK,
 175                 (xdrproc_t) NULL, (caddr_t) argp,
 176                 (xdrproc_t) xdr_bool, (caddr_t) &clnt_res,
 177                 TIMEOUT) != RPC_SUCCESS) {
 178                 return (NULL);
 179         }
 180         return (&clnt_res);
 181 }
 182
 183 /*
 184  * "We have the machine that goes 'ping!'" -- Monty Python
 185  *
 186  * This function blasts packets at the YPPROC_DOMAIN_NONACK procedures
 187  * of the NIS servers listed in restricted_addrs structure.
 188  * Whoever replies the fastest becomes our chosen server.
 189  *
 190  * Note: THIS IS NOT A BROADCAST OPERATION! We could use clnt_broadcast()
 191  * for this, but that has the following problems:
 192  * - We only get the address of the machine that replied in the
 193  *   'eachresult' callback, and on multi-homed machines this can
 194  *   lead to confusion.
 195  * - clnt_broadcast() only transmits to local networks, whereas with
 196  *   NIS+ you can have a perfectly good server located anywhere on or
 197  *   off the local network.
 198  * - clnt_broadcast() blocks for an arbitrary amount of time which the
 199  *   caller can't control -- we want to avoid that.
 200  *
 201  * Also note that this has nothing to do with the NIS_PING procedure used
 202  * for replica updates.
 203  */
 204
 205 struct ping_req {
 206         struct sockaddr_in      sin;
 207         u_int32_t               xid;
 208 };
 209
 210 int
 211 __yp_ping(struct in_addr *restricted_addrs, int cnt, char *dom, short *port)
 212 {
 213         struct timeval          tv = { 5, 0 };
 214         struct ping_req         **reqs;
 215         unsigned long           i;
 216         int                     async;
 217         struct sockaddr_in      sin, *any = NULL;
 218         struct netbuf           addr;
 219         int                     winner = -1;
 220         u_int32_t               xid_seed, xid_lookup;
 221         int                     sock, dontblock = 1;
 222         CLIENT                  *clnt;
 223         char                    *foo = dom;
 224         int                     validsrvs = 0;
 225
 226         /* Set up handles. */
 227         reqs = calloc(1, sizeof(struct ping_req *) * cnt);
 228         xid_seed = time(NULL) ^ getpid();
 229
 230         for (i = 0; i < cnt; i++) {
 231                 bzero((char *)&sin, sizeof(sin));
 232                 sin.sin_family = AF_INET;
 233                 bcopy((char *)&restricted_addrs[i],
 234                         (char *)&sin.sin_addr, sizeof(struct in_addr));
 235                 sin.sin_port = htons(__pmap_getport(&sin, YPPROG,
 236                                         YPVERS, IPPROTO_UDP));
 237                 if (sin.sin_port == 0)
 238                         continue;
 239                 reqs[i] = calloc(1, sizeof(struct ping_req));
 240                 bcopy((char *)&sin, (char *)&reqs[i]->sin, sizeof(sin));
 241                 any = &reqs[i]->sin;
 242                 reqs[i]->xid = xid_seed;
 243                 xid_seed++;
 244                 validsrvs++;
 245         }
 246
 247         /* Make sure at least one server was assigned */
 248         if (!validsrvs) {
 249                 free(reqs);
 250                 return(-1);
 251         }
 252
 253         /* Create RPC handle */
 254         sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
 255         clnt = clntudp_create(any, YPPROG, YPVERS, tv, &sock);
 256         if (clnt == NULL) {
 257                 close(sock);
 258                 for (i = 0; i < cnt; i++)
 259                         if (reqs[i] != NULL)
 260                                 free(reqs[i]);
 261                 free(reqs);
 262                 return(-1);
 263         }
 264         clnt->cl_auth = authunix_create_default();
 265         tv.tv_sec = 0;
 266
 267         clnt_control(clnt, CLSET_TIMEOUT, (char *)&tv);
 268         async = TRUE;
 269         clnt_control(clnt, CLSET_ASYNC, (char *)&async);
 270         ioctl(sock, FIONBIO, &dontblock);
 271
 272         /* Transmit */
 273         for (i = 0; i < cnt; i++) {
 274                 if (reqs[i] != NULL) {
 275                         clnt_control(clnt, CLSET_XID, (char *)&reqs[i]->xid);
 276                         addr.len = sizeof(reqs[i]->sin);
 277                         addr.buf = (char *) &reqs[i]->sin;
 278                         clnt_control(clnt, CLSET_SVC_ADDR, &addr);
 279                         ypproc_domain_nonack_2_send(&foo, clnt);
 280                 }
 281         }
 282
 283         /* Receive reply */
 284         ypproc_domain_nonack_2_recv(&foo, clnt);
 285
 286         /* Got a winner -- look him up. */
 287         clnt_control(clnt, CLGET_XID, (char *)&xid_lookup);
 288         for (i = 0; i < cnt; i++) {
 289                 if (reqs[i] != NULL && reqs[i]->xid == xid_lookup) {
 290                         winner = i;
 291                         *port = reqs[i]->sin.sin_port;
 292                 }
 293         }
 294
 295         /* Shut everything down */
 296         auth_destroy(clnt->cl_auth);
 297         clnt_destroy(clnt);
 298         close(sock);
 299
 300         for (i = 0; i < cnt; i++)
 301                 if (reqs[i] != NULL)
 302                         free(reqs[i]);
 303         free(reqs);
 304
 305         return(winner);
 306 }