[thirdparty/glibc.git] / nis / nis_findserv.c

/* Copyright (C) 1997, 1998, 2000, 2001 Free Software Foundation, Inc.
   This file is part of the GNU C Library.
   Contributed by Thorsten Kukuk <kukuk@vt.uni-paderborn.de>, 1997.

   The GNU C Library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 2.1 of the License, or (at your option) any later version.

   The GNU C Library is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public
   License along with the GNU C Library; if not, write to the Free
   Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
   02111-1307 USA.  */

#include <string.h>
#include <time.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <rpc/pmap_prot.h>
#include <rpc/pmap_clnt.h>
#include <rpcsvc/nis.h>

#include "nis_intern.h"

/* Private data kept per client handle, from sunrpc/clnt_udp.c */
struct cu_data
  {
    int cu_sock;
    bool_t cu_closeit;
    struct sockaddr_in cu_raddr;
    int cu_rlen;
    struct timeval cu_wait;
    struct timeval cu_total;
    struct rpc_err cu_error;
    XDR cu_outxdrs;
    u_int cu_xdrpos;
    u_int cu_sendsz;
    char *cu_outbuf;
    u_int cu_recvsz;
    char cu_inbuf[1];
  };


/* The following is the original routine from sunrpc/pm_getport.c.
   The only change is the much shorter timeout. */
/*
 * pmap_getport.c
 * Client interface to pmap rpc service.
 *
 * Copyright (C) 1984, Sun Microsystems, Inc.
 */

/*
 * Find the mapped port for program,version.
 * Calls the pmap service remotely to do the lookup.
 * Returns 0 if no map exists.
 */
u_short
__pmap_getnisport (struct sockaddr_in *address, u_long program,
		   u_long version, u_int protocol)
{
  const struct timeval timeout = {1, 0};
  const struct timeval tottimeout = {1, 0};
  u_short port = 0;
  int socket = -1;
  CLIENT *client;
  struct pmap parms;

  address->sin_port = htons (PMAPPORT);
  client = clntudp_bufcreate (address, PMAPPROG, PMAPVERS, timeout, &socket,
			      RPCSMALLMSGSIZE, RPCSMALLMSGSIZE);
  if (client != (CLIENT *) NULL)
    {
      parms.pm_prog = program;
      parms.pm_vers = version;
      parms.pm_prot = protocol;
      parms.pm_port = 0;	/* not needed or used */
      if (CLNT_CALL (client, PMAPPROC_GETPORT, (xdrproc_t) xdr_pmap,
		     (caddr_t) & parms, (xdrproc_t) xdr_u_short,
		     (caddr_t) & port, tottimeout) != RPC_SUCCESS)
	{
	  rpc_createerr.cf_stat = RPC_PMAPFAILURE;
	  clnt_geterr (client, &rpc_createerr.cf_error);
	}
      else
	{
	  if (port == 0)
	    rpc_createerr.cf_stat = RPC_PROGNOTREGISTERED;
	}
      CLNT_DESTROY (client);
    }
  /* (void)close(socket); CLNT_DESTROY already closed it */
  address->sin_port = 0;
  return port;
}

/* This is now the public function, which should find the fastest server */

struct findserv_req
{
  struct sockaddr_in sin;
  u_int32_t xid;
  u_int server_nr;
  u_int server_ep;
};


static long int
__nis_findfastest_with_timeout (dir_binding *bind,
				const struct timeval *timeout)
{
  static const struct timeval TIMEOUT00 = { 0, 0 };
  struct findserv_req *pings;
  struct sockaddr_in sin, saved_sin;
  int found = -1;
  u_int32_t xid_seed;
  int sock, dontblock = 1;
  CLIENT *clnt;
  u_long i, j, pings_count, pings_max, fastest = -1;
  struct cu_data *cu;

  pings_max = bind->server_len * 2;	/* Reserve a little bit more memory
					   for multihomed hosts */
  pings_count = 0;
  pings = malloc (sizeof (struct findserv_req) * pings_max);
  xid_seed = (u_int32_t) (time (NULL) ^ getpid ());

  if (__builtin_expect (pings == NULL, 0))
    return -1;

  memset (&sin, '\0', sizeof (sin));
  sin.sin_family = AF_INET;
  for (i = 0; i < bind->server_len; i++)
    for (j = 0; j < bind->server_val[i].ep.ep_len; ++j)
      if (strcmp (bind->server_val[i].ep.ep_val[j].family, "inet") == 0)
	if ((bind->server_val[i].ep.ep_val[j].proto == NULL) ||
	    (bind->server_val[i].ep.ep_val[j].proto[0] == '-') ||
	    (bind->server_val[i].ep.ep_val[j].proto[0] == '\0'))
	  {
	    sin.sin_addr.s_addr =
	      inetstr2int (bind->server_val[i].ep.ep_val[j].uaddr);
	    if (sin.sin_addr.s_addr == 0)
	      continue;
	    sin.sin_port = htons (__pmap_getnisport (&sin, NIS_PROG,
						     NIS_VERSION,
						     IPPROTO_UDP));
	    if (sin.sin_port == 0)
	      continue;

	    if (pings_count >= pings_max)
	      {
		struct findserv_req *new_pings;

		pings_max += 10;
		new_pings = realloc (pings, sizeof (struct findserv_req) *
				     pings_max);
		if (__builtin_expect (new_pings == NULL, 0))
		  {
		    free (pings);
		    return -1;
		  }
		pings = new_pings;
	      }
	    memcpy ((char *) &pings[pings_count].sin, (char *) &sin,
		    sizeof (sin));
	    memcpy ((char *)&saved_sin, (char *)&sin, sizeof(sin));
	    pings[pings_count].xid = xid_seed + pings_count;
	    pings[pings_count].server_nr = i;
	    pings[pings_count].server_ep = j;
	    ++pings_count;
	  }

  /* Make sure at least one server was assigned */
  if (pings_count == 0)
    {
      free (pings);
      return -1;
    }

  /* Create RPC handle */
  sock = socket (AF_INET, SOCK_DGRAM, IPPROTO_UDP);
  clnt = clntudp_create (&saved_sin, NIS_PROG, NIS_VERSION, *timeout, &sock);
  if (clnt == NULL)
    {
      close (sock);
      free (pings);
      return -1;
    }
  auth_destroy (clnt->cl_auth);
  clnt->cl_auth = authunix_create_default ();
  cu = (struct cu_data *) clnt->cl_private;
  ioctl (sock, FIONBIO, &dontblock);
  /* Send to all servers the NULLPROC */
  for (i = 0; i < pings_count; ++i)
    {
      /* clntudp_call() will increment, subtract one */
      *((u_int32_t *) (cu->cu_outbuf)) = pings[i].xid - 1;
      memcpy ((char *) &cu->cu_raddr, (char *) &pings[i].sin,
	      sizeof (struct sockaddr_in));
      /* Transmit to NULLPROC, return immediately. */
      clnt_call (clnt, NULLPROC,
		 (xdrproc_t) xdr_void, (caddr_t) 0,
		 (xdrproc_t) xdr_void, (caddr_t) 0, TIMEOUT00);
    }

  while (found == -1) {
    /* Receive reply from NULLPROC asynchronously. Note null inproc. */
    int rc = clnt_call (clnt, NULLPROC,
			(xdrproc_t) NULL, (caddr_t) 0,
			(xdrproc_t) xdr_void, (caddr_t) 0,
			*timeout);
    if (RPC_SUCCESS == rc) {
      fastest = *((u_int32_t *) (cu->cu_inbuf)) - xid_seed;
      if (fastest < pings_count) {
	bind->server_used = pings[fastest].server_nr;
	bind->current_ep = pings[fastest].server_ep;
	found = 1;
      }
    } else {
      /*      clnt_perror(clnt, "__nis_findfastest"); */
      break;
    }
  }


  auth_destroy (clnt->cl_auth);
  clnt_destroy (clnt);
  close (sock);

  free (pings);

  return found;
}


long int
__nis_findfastest (dir_binding *bind)
{
  struct timeval timeout = { __NIS_PING_TIMEOUT_START, 0 };
  long int found = -1;
  long int retry = __NIS_PING_RETRY + 1;

  while (retry--)
    {
      found = __nis_findfastest_with_timeout (bind, &timeout);
      if (found != -1)
	break;
      timeout.tv_sec += __NIS_PING_TIMEOUT_INCREMENT;
    }

  return found;
}
Commit	Line	Data
306eeae5	1	/* Copyright (C) 1997, 1998, 2000, 2001 Free Software Foundation, Inc.
3996f34b UD	2	This file is part of the GNU C Library.
	3	Contributed by Thorsten Kukuk <kukuk@vt.uni-paderborn.de>, 1997.
	4
	5	The GNU C Library is free software; you can redistribute it and/or
41bdb6e2 AJ	6	modify it under the terms of the GNU Lesser General Public
	7	License as published by the Free Software Foundation; either
	8	version 2.1 of the License, or (at your option) any later version.
3996f34b UD	9
	10	The GNU C Library is distributed in the hope that it will be useful,
	11	but WITHOUT ANY WARRANTY; without even the implied warranty of
	12	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
41bdb6e2	13	Lesser General Public License for more details.
3996f34b	14
41bdb6e2 AJ	15	You should have received a copy of the GNU Lesser General Public
	16	License along with the GNU C Library; if not, write to the Free
	17	Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
	18	02111-1307 USA. */
3996f34b UD	19
3996f34b UD	20	#include <string.h>
306eeae5	21	#include <time.h>
3996f34b UD	22	#include <unistd.h>
3996f34b UD	23	#include <sys/ioctl.h>
18de8c73	24	#include <sys/socket.h>
3996f34b UD	25	#include <rpc/pmap_prot.h>
	26	#include <rpc/pmap_clnt.h>
	27	#include <rpcsvc/nis.h>
	28
	29	#include "nis_intern.h"
	30
	31	/* Private data kept per client handle, from sunrpc/clnt_udp.c */
	32	struct cu_data
	33	{
	34	int cu_sock;
	35	bool_t cu_closeit;
	36	struct sockaddr_in cu_raddr;
	37	int cu_rlen;
	38	struct timeval cu_wait;
	39	struct timeval cu_total;
	40	struct rpc_err cu_error;
	41	XDR cu_outxdrs;
	42	u_int cu_xdrpos;
	43	u_int cu_sendsz;
	44	char *cu_outbuf;
	45	u_int cu_recvsz;
	46	char cu_inbuf[1];
	47	};
	48
	49
	50	/* The following is the original routine from sunrpc/pm_getport.c.
	51	The only change is the much shorter timeout. */
	52	/*
	53	* pmap_getport.c
	54	* Client interface to pmap rpc service.
	55	*
	56	* Copyright (C) 1984, Sun Microsystems, Inc.
	57	*/
	58
	59	/*
	60	* Find the mapped port for program,version.
	61	* Calls the pmap service remotely to do the lookup.
	62	* Returns 0 if no map exists.
	63	*/
26a60f90 UD	64	u_short
	65	__pmap_getnisport (struct sockaddr_in *address, u_long program,
	66	u_long version, u_int protocol)
3996f34b UD	67	{
	68	const struct timeval timeout = {1, 0};
	69	const struct timeval tottimeout = {1, 0};
	70	u_short port = 0;
	71	int socket = -1;
	72	CLIENT *client;
	73	struct pmap parms;
	74
	75	address->sin_port = htons (PMAPPORT);
	76	client = clntudp_bufcreate (address, PMAPPROG, PMAPVERS, timeout, &socket,
	77	RPCSMALLMSGSIZE, RPCSMALLMSGSIZE);
	78	if (client != (CLIENT *) NULL)
	79	{
	80	parms.pm_prog = program;
	81	parms.pm_vers = version;
	82	parms.pm_prot = protocol;
	83	parms.pm_port = 0; /* not needed or used */
	84	if (CLNT_CALL (client, PMAPPROC_GETPORT, (xdrproc_t) xdr_pmap,
	85	(caddr_t) & parms, (xdrproc_t) xdr_u_short,
	86	(caddr_t) & port, tottimeout) != RPC_SUCCESS)
	87	{
	88	rpc_createerr.cf_stat = RPC_PMAPFAILURE;
	89	clnt_geterr (client, &rpc_createerr.cf_error);
	90	}
650425ce	91	else
3996f34b	92	{
650425ce UD	93	if (port == 0)
650425ce UD	94	rpc_createerr.cf_stat = RPC_PROGNOTREGISTERED;
3996f34b UD	95	}
	96	CLNT_DESTROY (client);
	97	}
	98	/* (void)close(socket); CLNT_DESTROY already closed it */
	99	address->sin_port = 0;
	100	return port;
	101	}
	102
650425ce	103	/* This is now the public function, which should find the fastest server */
3996f34b UD	104
	105	struct findserv_req
	106	{
	107	struct sockaddr_in sin;
55c14926	108	u_int32_t xid;
3996f34b UD	109	u_int server_nr;
	110	u_int server_ep;
	111	};
	112
e852e889	113
495818ce UD	114	static long int
	115	__nis_findfastest_with_timeout (dir_binding *bind,
	116	const struct timeval *timeout)
	117	{
	118	static const struct timeval TIMEOUT00 = { 0, 0 };
54eb84d0	119	struct findserv_req *pings;
650425ce	120	struct sockaddr_in sin, saved_sin;
3996f34b	121	int found = -1;
6c3ebebd	122	u_int32_t xid_seed;
3996f34b UD	123	int sock, dontblock = 1;
3996f34b UD	124	CLIENT *clnt;
6c3ebebd	125	u_long i, j, pings_count, pings_max, fastest = -1;
3996f34b UD	126	struct cu_data *cu;
	127
	128	pings_max = bind->server_len * 2; /* Reserve a little bit more memory
	129	for multihomed hosts */
	130	pings_count = 0;
54eb84d0	131	pings = malloc (sizeof (struct findserv_req) * pings_max);
650425ce	132	xid_seed = (u_int32_t) (time (NULL) ^ getpid ());
3996f34b	133
54eb84d0 UD	134	if (__builtin_expect (pings == NULL, 0))
	135	return -1;
	136
3996f34b UD	137	memset (&sin, '\0', sizeof (sin));
	138	sin.sin_family = AF_INET;
	139	for (i = 0; i < bind->server_len; i++)
	140	for (j = 0; j < bind->server_val[i].ep.ep_len; ++j)
	141	if (strcmp (bind->server_val[i].ep.ep_val[j].family, "inet") == 0)
650425ce	142	if ((bind->server_val[i].ep.ep_val[j].proto == NULL) \|\|
e852e889	143	(bind->server_val[i].ep.ep_val[j].proto[0] == '-') \|\|
a53bad16	144	(bind->server_val[i].ep.ep_val[j].proto[0] == '\0'))
3996f34b UD	145	{
	146	sin.sin_addr.s_addr =
	147	inetstr2int (bind->server_val[i].ep.ep_val[j].uaddr);
	148	if (sin.sin_addr.s_addr == 0)
	149	continue;
26a60f90 UD	150	sin.sin_port = htons (__pmap_getnisport (&sin, NIS_PROG,
	151	NIS_VERSION,
	152	IPPROTO_UDP));
3996f34b UD	153	if (sin.sin_port == 0)
	154	continue;
	155
	156	if (pings_count >= pings_max)
	157	{
54eb84d0 UD	158	struct findserv_req *new_pings;
54eb84d0 UD	159
3996f34b	160	pings_max += 10;
54eb84d0 UD	161	new_pings = realloc (pings, sizeof (struct findserv_req) *
	162	pings_max);
	163	if (__builtin_expect (new_pings == NULL, 0))
	164	{
	165	free (pings);
	166	return -1;
	167	}
	168	pings = new_pings;
3996f34b	169	}
54eb84d0	170	memcpy ((char ) &pings[pings_count].sin, (char ) &sin,
3996f34b	171	sizeof (sin));
650425ce	172	memcpy ((char )&saved_sin, (char )&sin, sizeof(sin));
6c3ebebd	173	pings[pings_count].xid = xid_seed + pings_count;
54eb84d0 UD	174	pings[pings_count].server_nr = i;
54eb84d0 UD	175	pings[pings_count].server_ep = j;
3996f34b UD	176	++pings_count;
	177	}
	178
	179	/* Make sure at least one server was assigned */
	180	if (pings_count == 0)
	181	{
	182	free (pings);
	183	return -1;
	184	}
	185
	186	/* Create RPC handle */
	187	sock = socket (AF_INET, SOCK_DGRAM, IPPROTO_UDP);
495818ce	188	clnt = clntudp_create (&saved_sin, NIS_PROG, NIS_VERSION, *timeout, &sock);
3996f34b UD	189	if (clnt == NULL)
	190	{
	191	close (sock);
3996f34b UD	192	free (pings);
	193	return -1;
	194	}
26a60f90	195	auth_destroy (clnt->cl_auth);
3996f34b UD	196	clnt->cl_auth = authunix_create_default ();
3996f34b UD	197	cu = (struct cu_data *) clnt->cl_private;
3996f34b	198	ioctl (sock, FIONBIO, &dontblock);
3996f34b UD	199	/* Send to all servers the NULLPROC */
	200	for (i = 0; i < pings_count; ++i)
	201	{
	202	/* clntudp_call() will increment, subtract one */
54eb84d0 UD	203	((u_int32_t ) (cu->cu_outbuf)) = pings[i].xid - 1;
54eb84d0 UD	204	memcpy ((char ) &cu->cu_raddr, (char ) &pings[i].sin,
650425ce UD	205	sizeof (struct sockaddr_in));
650425ce UD	206	/* Transmit to NULLPROC, return immediately. */
495818ce	207	clnt_call (clnt, NULLPROC,
6c3ebebd UD	208	(xdrproc_t) xdr_void, (caddr_t) 0,
6c3ebebd UD	209	(xdrproc_t) xdr_void, (caddr_t) 0, TIMEOUT00);
3996f34b	210	}
495818ce UD	211
	212	while (found == -1) {
	213	/* Receive reply from NULLPROC asynchronously. Note null inproc. */
	214	int rc = clnt_call (clnt, NULLPROC,
	215	(xdrproc_t) NULL, (caddr_t) 0,
	216	(xdrproc_t) xdr_void, (caddr_t) 0,
	217	*timeout);
	218	if (RPC_SUCCESS == rc) {
6c3ebebd UD	219	fastest = ((u_int32_t ) (cu->cu_inbuf)) - xid_seed;
6c3ebebd UD	220	if (fastest < pings_count) {
495818ce UD	221	bind->server_used = pings[fastest].server_nr;
	222	bind->current_ep = pings[fastest].server_ep;
	223	found = 1;
6c3ebebd	224	}
495818ce UD	225	} else {
	226	/* clnt_perror(clnt, "__nis_findfastest"); */
	227	break;
3996f34b	228	}
495818ce UD	229	}
	230
	231
3996f34b UD	232	auth_destroy (clnt->cl_auth);
	233	clnt_destroy (clnt);
	234	close (sock);
	235
3996f34b UD	236	free (pings);
	237
	238	return found;
495818ce UD	239	}
	240
	241
	242	long int
	243	__nis_findfastest (dir_binding *bind)
	244	{
	245	struct timeval timeout = { __NIS_PING_TIMEOUT_START, 0 };
	246	long int found = -1;
	247	long int retry = __NIS_PING_RETRY + 1;
	248
	249	while (retry--)
	250	{
	251	found = __nis_findfastest_with_timeout (bind, &timeout);
	252	if (found != -1)
	253	break;
	254	timeout.tv_sec += __NIS_PING_TIMEOUT_INCREMENT;
	255	}
	256
	257	return found;
3996f34b	258	}