[apple/shell_cmds.git] / date / netdate.c

/*-
 * Copyright (c) 1990, 1993
 *	The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 4. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#if 0
#ifndef lint
static char sccsid[] = "@(#)netdate.c	8.1 (Berkeley) 5/31/93";
#endif /* not lint */
#endif

#include <sys/cdefs.h>
__FBSDID("$FreeBSD$");

#include <sys/param.h>
#include <sys/time.h>
#include <sys/socket.h>

#include <netinet/in.h>
#include <netdb.h>
#define TSPTYPES
#include <protocols/timed.h>

#include <err.h>
#include <errno.h>
#include <string.h>
#include <unistd.h>

#include "extern.h"

#define	WAITACK		2	/* seconds */
#define	WAITDATEACK	5	/* seconds */

/*
 * Set the date in the machines controlled by timedaemons by communicating the
 * new date to the local timedaemon.  If the timedaemon is in the master state,
 * it performs the correction on all slaves.  If it is in the slave state, it
 * notifies the master that a correction is needed.
 * Returns 0 on success.  Returns > 0 on failure, setting retval to 2;
 */
int
netsettime(time_t tval)
{
	struct timeval tout;
	struct servent *sp;
	struct tsp msg;
	struct sockaddr_in lsin, dest, from;
	fd_set ready;
	long waittime;
	int s, port, timed_ack, found, lerr;
	socklen_t length;
	char hostname[MAXHOSTNAMELEN];

	if ((sp = getservbyname("timed", "udp")) == NULL) {
		warnx("timed/udp: unknown service");
		return (retval = 2);
	}

	dest.sin_port = sp->s_port;
	dest.sin_family = AF_INET;
	dest.sin_addr.s_addr = htonl((u_long)INADDR_ANY);
	s = socket(AF_INET, SOCK_DGRAM, 0);
	if (s < 0) {
		if (errno != EAFNOSUPPORT)
			warn("timed");
		return (retval = 2);
	}

	memset(&lsin, 0, sizeof(lsin));
	lsin.sin_family = AF_INET;
	for (port = IPPORT_RESERVED - 1; port > IPPORT_RESERVED / 2; port--) {
		lsin.sin_port = htons((u_short)port);
		if (bind(s, (struct sockaddr *)&lsin, sizeof(lsin)) >= 0)
			break;
		if (errno == EADDRINUSE)
			continue;
		if (errno != EADDRNOTAVAIL)
			warn("bind");
		goto bad;
	}
	if (port == IPPORT_RESERVED / 2) {
		warnx("all ports in use");
		goto bad;
	}
	memset(&msg, 0, sizeof(msg));
	msg.tsp_type = TSP_SETDATE;
	msg.tsp_vers = TSPVERSION;
	if (gethostname(hostname, sizeof(hostname))) {
		warn("gethostname");
		goto bad;
	}
	(void)strlcpy(msg.tsp_name, hostname, sizeof(msg.tsp_name));
	msg.tsp_seq = htons((u_short)0);
	msg.tsp_time.tv_sec = htonl((u_long)tval);
	msg.tsp_time.tv_usec = htonl((u_long)0);
	length = sizeof(struct sockaddr_in);
	if (connect(s, (struct sockaddr *)&dest, length) < 0) {
		warn("connect");
		goto bad;
	}
	if (send(s, (char *)&msg, sizeof(struct tsp), 0) < 0) {
		if (errno != ECONNREFUSED)
			warn("send");
		goto bad;
	}

	timed_ack = -1;
	waittime = WAITACK;
loop:
	tout.tv_sec = waittime;
	tout.tv_usec = 0;

	FD_ZERO(&ready);
	FD_SET(s, &ready);
	found = select(FD_SETSIZE, &ready, (fd_set *)0, (fd_set *)0, &tout);

	length = sizeof(lerr);
	if (!getsockopt(s,
	    SOL_SOCKET, SO_ERROR, (char *)&lerr, &length) && lerr) {
		if (lerr != ECONNREFUSED)
			warnc(lerr, "send (delayed error)");
		goto bad;
	}

	if (found > 0 && FD_ISSET(s, &ready)) {
		length = sizeof(struct sockaddr_in);
		if (recvfrom(s, &msg, sizeof(struct tsp), 0,
		    (struct sockaddr *)&from, &length) < 0) {
			if (errno != ECONNREFUSED)
				warn("recvfrom");
			goto bad;
		}
		msg.tsp_seq = ntohs(msg.tsp_seq);
		msg.tsp_time.tv_sec = ntohl(msg.tsp_time.tv_sec);
		msg.tsp_time.tv_usec = ntohl(msg.tsp_time.tv_usec);
		switch (msg.tsp_type) {
		case TSP_ACK:
			timed_ack = TSP_ACK;
			waittime = WAITDATEACK;
			goto loop;
		case TSP_DATEACK:
			(void)close(s);
			return (0);
		default:
			warnx("wrong ack received from timed: %s",
			    tsptype[msg.tsp_type]);
			timed_ack = -1;
			break;
		}
	}
	if (timed_ack == -1)
		warnx("can't reach time daemon, time set locally");

bad:
	(void)close(s);
	return (retval = 2);
}
Commit	Line	Data
44bd5ea7 A	1	/*-
	2	* Copyright (c) 1990, 1993
	3	* The Regents of the University of California. 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.
44bd5ea7 A	13	* 4. Neither the name of the University nor the names of its contributors
	14	* may be used to endorse or promote products derived from this software
	15	* without specific prior written permission.
	16	*
	17	* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
	18	* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	19	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
	20	* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
	21	* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
	22	* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
	23	* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
	24	* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
	25	* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
	26	* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
	27	* SUCH DAMAGE.
	28	*/
	29
44bd5ea7	30	#if 0
e1a085ba A	31	#ifndef lint
e1a085ba A	32	static char sccsid[] = "@(#)netdate.c 8.1 (Berkeley) 5/31/93";
44bd5ea7	33	#endif /* not lint */
e1a085ba A	34	#endif
	35
	36	#include <sys/cdefs.h>
71aad674	37	__FBSDID("$FreeBSD$");
44bd5ea7 A	38
	39	#include <sys/param.h>
	40	#include <sys/time.h>
	41	#include <sys/socket.h>
	42
	43	#include <netinet/in.h>
	44	#include <netdb.h>
	45	#define TSPTYPES
	46	#include <protocols/timed.h>
	47
	48	#include <err.h>
	49	#include <errno.h>
44bd5ea7 A	50	#include <string.h>
	51	#include <unistd.h>
	52
	53	#include "extern.h"
	54
	55	#define WAITACK 2 /* seconds */
	56	#define WAITDATEACK 5 /* seconds */
	57
44bd5ea7 A	58	/*
	59	* Set the date in the machines controlled by timedaemons by communicating the
	60	* new date to the local timedaemon. If the timedaemon is in the master state,
	61	* it performs the correction on all slaves. If it is in the slave state, it
	62	* notifies the master that a correction is needed.
	63	* Returns 0 on success. Returns > 0 on failure, setting retval to 2;
	64	*/
	65	int
e1a085ba	66	netsettime(time_t tval)
44bd5ea7 A	67	{
	68	struct timeval tout;
	69	struct servent *sp;
	70	struct tsp msg;
e1a085ba	71	struct sockaddr_in lsin, dest, from;
44bd5ea7 A	72	fd_set ready;
44bd5ea7 A	73	long waittime;
e1a085ba A	74	int s, port, timed_ack, found, lerr;
	75	socklen_t length;
	76	char hostname[MAXHOSTNAMELEN];
44bd5ea7 A	77
44bd5ea7 A	78	if ((sp = getservbyname("timed", "udp")) == NULL) {
e1a085ba	79	warnx("timed/udp: unknown service");
44bd5ea7 A	80	return (retval = 2);
	81	}
	82
44bd5ea7	83	dest.sin_port = sp->s_port;
e1a085ba A	84	dest.sin_family = AF_INET;
e1a085ba A	85	dest.sin_addr.s_addr = htonl((u_long)INADDR_ANY);
44bd5ea7 A	86	s = socket(AF_INET, SOCK_DGRAM, 0);
44bd5ea7 A	87	if (s < 0) {
71aad674	88	if (errno != EAFNOSUPPORT)
44bd5ea7 A	89	warn("timed");
	90	return (retval = 2);
	91	}
	92
e1a085ba A	93	memset(&lsin, 0, sizeof(lsin));
	94	lsin.sin_family = AF_INET;
	95	for (port = IPPORT_RESERVED - 1; port > IPPORT_RESERVED / 2; port--) {
	96	lsin.sin_port = htons((u_short)port);
	97	if (bind(s, (struct sockaddr *)&lsin, sizeof(lsin)) >= 0)
	98	break;
	99	if (errno == EADDRINUSE)
	100	continue;
	101	if (errno != EADDRNOTAVAIL)
	102	warn("bind");
44bd5ea7 A	103	goto bad;
44bd5ea7 A	104	}
e1a085ba A	105	if (port == IPPORT_RESERVED / 2) {
e1a085ba A	106	warnx("all ports in use");
44bd5ea7 A	107	goto bad;
44bd5ea7 A	108	}
71aad674	109	memset(&msg, 0, sizeof(msg));
44bd5ea7 A	110	msg.tsp_type = TSP_SETDATE;
	111	msg.tsp_vers = TSPVERSION;
	112	if (gethostname(hostname, sizeof(hostname))) {
	113	warn("gethostname");
	114	goto bad;
	115	}
71aad674	116	(void)strlcpy(msg.tsp_name, hostname, sizeof(msg.tsp_name));
44bd5ea7 A	117	msg.tsp_seq = htons((u_short)0);
	118	msg.tsp_time.tv_sec = htonl((u_long)tval);
	119	msg.tsp_time.tv_usec = htonl((u_long)0);
	120	length = sizeof(struct sockaddr_in);
	121	if (connect(s, (struct sockaddr *)&dest, length) < 0) {
	122	warn("connect");
	123	goto bad;
	124	}
	125	if (send(s, (char *)&msg, sizeof(struct tsp), 0) < 0) {
	126	if (errno != ECONNREFUSED)
	127	warn("send");
	128	goto bad;
	129	}
	130
	131	timed_ack = -1;
	132	waittime = WAITACK;
	133	loop:
	134	tout.tv_sec = waittime;
	135	tout.tv_usec = 0;
	136
	137	FD_ZERO(&ready);
	138	FD_SET(s, &ready);
	139	found = select(FD_SETSIZE, &ready, (fd_set )0, (fd_set )0, &tout);
	140
e1a085ba	141	length = sizeof(lerr);
44bd5ea7	142	if (!getsockopt(s,
e1a085ba A	143	SOL_SOCKET, SO_ERROR, (char *)&lerr, &length) && lerr) {
	144	if (lerr != ECONNREFUSED)
	145	warnc(lerr, "send (delayed error)");
44bd5ea7 A	146	goto bad;
	147	}
	148
	149	if (found > 0 && FD_ISSET(s, &ready)) {
	150	length = sizeof(struct sockaddr_in);
	151	if (recvfrom(s, &msg, sizeof(struct tsp), 0,
	152	(struct sockaddr *)&from, &length) < 0) {
	153	if (errno != ECONNREFUSED)
	154	warn("recvfrom");
	155	goto bad;
	156	}
	157	msg.tsp_seq = ntohs(msg.tsp_seq);
	158	msg.tsp_time.tv_sec = ntohl(msg.tsp_time.tv_sec);
	159	msg.tsp_time.tv_usec = ntohl(msg.tsp_time.tv_usec);
	160	switch (msg.tsp_type) {
	161	case TSP_ACK:
	162	timed_ack = TSP_ACK;
	163	waittime = WAITDATEACK;
	164	goto loop;
	165	case TSP_DATEACK:
	166	(void)close(s);
	167	return (0);
	168	default:
e1a085ba	169	warnx("wrong ack received from timed: %s",
44bd5ea7 A	170	tsptype[msg.tsp_type]);
	171	timed_ack = -1;
	172	break;
	173	}
	174	}
	175	if (timed_ack == -1)
	176	warnx("can't reach time daemon, time set locally");
	177
	178	bad:
	179	(void)close(s);
	180	return (retval = 2);
	181	}