X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=util-linux%2Frdate.c;h=645b0b9291ae12af1be793e431ab02c431b30e52;hb=1119972f917db59ccb7c74daebf5a0688e9d0e24;hp=7c8d54117efe2667a7e5657d778f143386dfad76;hpb=3e856ce428cabaf6c8d99a2374a1f9a4a05db5f0;p=oweals%2Fbusybox.git

diff --git a/util-linux/rdate.c b/util-linux/rdate.c
index 7c8d54117..645b0b929 100644
--- a/util-linux/rdate.c
+++ b/util-linux/rdate.c
@@ -5,73 +5,47 @@
  *
  * by Sterling Huxley <sterling@europa.com>
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program 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
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- *
+ * Licensed under GPL v2 or later, see file License for details.
 */
 
-#include "busybox.h"
-#define BB_DECLARE_EXTERN
-#include "messages.c"
-#include <sys/time.h>
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <netdb.h>
 #include <stdio.h>
-#include <getopt.h>
+#include <string.h>
+#include <time.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <signal.h>
+
+#include "busybox.h"
 
 
-#define RFC_868_BIAS	2208988800UL
+static const int RFC_868_BIAS = 2208988800UL;
 
-int setdate= 0;
-int printdate= 0;
+static void socket_timeout(int sig)
+{
+	bb_error_msg_and_die("timeout connecting to time server");
+}
 
-time_t askremotedate(char *host)
+static time_t askremotedate(const char *host)
 {
-	struct hostent *h;
-	struct sockaddr_in sin;
-	struct servent *tserv;
-	unsigned long int nett, localt;
+	unsigned long nett;
+	struct sockaddr_in s_in;
 	int fd;
 
-	if (!(h = gethostbyname(host))) {	/* get the IP addr */
-		errorMsg("%s: %s\n", host, strerror(errno));
-		return(-1);
-	}
-	if ((tserv = getservbyname("time", "tcp")) == NULL) { /* find port # */
-		errorMsg("%s: %s\n", "time", strerror(errno));
-		return(-1);
-	}
-	if ((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {  /* get net connection */
-		errorMsg("%s: %s\n", "socket", strerror(errno));
-		return(-1);
-	}
+	bb_lookup_host(&s_in, host);
+	s_in.sin_port = bb_lookup_port("time", "tcp", 37);
 
-	memcpy(&sin.sin_addr, h->h_addr, sizeof(sin.sin_addr));
-	sin.sin_port= tserv->s_port;
-	sin.sin_family = AF_INET;
+	/* Add a timeout for dead or inaccessible servers */
+	alarm(10);
+	signal(SIGALRM, socket_timeout);
 
-	if (connect(fd, (struct sockaddr *)&sin, sizeof(sin)) < 0) {	/* connect to time server */
-		errorMsg("%s: %s\n", host, strerror(errno));
-		close(fd);
-		return(-1);
-	}
-	if (read(fd, (void *)&nett, 4) != 4) {	/* read time from server */
-		close(fd);
-		errorMsg("%s did not send the complete time\n", host);
-	}
+	fd = xconnect_tcp_v4(&s_in);
+
+	if (safe_read(fd, (void *)&nett, 4) != 4)    /* read time from server */
+		bb_error_msg_and_die("%s did not send the complete time", host);
 	close(fd);
 
 	/* convert from network byte order to local byte order.
@@ -80,54 +54,33 @@ time_t askremotedate(char *host)
 	 *  the RFC 868 time 2,208,988,800 corresponds to 00:00  1 Jan 1970 GMT
 	 * Subtract the RFC 868 time  to get Linux epoch
 	 */
-	localt= ntohl(nett) - RFC_868_BIAS;
 
-	return(localt);
+	return(ntohl(nett) - RFC_868_BIAS);
 }
 
 int rdate_main(int argc, char **argv)
 {
-	time_t time;
-	int opt;
-
-	/* Interpret command line args */
-	/* do special-case option parsing */
-	if (argv[1] && (strcmp(argv[1], "--help") == 0))
-		usage(rdate_usage);
-
-	/* do normal option parsing */
-	while ((opt = getopt(argc, argv, "Hsp")) > 0) {
-		switch (opt) {
-			default:
-			case 'H':
-				usage(rdate_usage);
-				break;
-			case 's':
-				setdate++;
-				break;
-			case 'p':
-				printdate++;
-				break;
-		}
-	}
+	time_t remote_time;
+	unsigned long flags;
 
-	/* the default action is to set the date */
-	if (printdate==0 && setdate==0) setdate++;
+	opt_complementary = "-1";
+	flags = getopt32(argc, argv, "sp");
 
-	if (optind == argc) {
-		usage(rdate_usage);
-	}
+	remote_time = askremotedate(argv[optind]);
 
-	if ((time= askremotedate(argv[optind])) == (time_t)-1) {
-		return EXIT_FAILURE;
-	}
-	if (setdate) {
-		if (stime(&time) < 0)
-			fatalError("Could not set time of day: %s\n", strerror(errno));
-	}
-	if (printdate) {
-		fprintf(stdout, "%s", ctime(&time));
+	if ((flags & 2) == 0) {
+		time_t current_time;
+
+		time(&current_time);
+		if (current_time == remote_time)
+			bb_error_msg("current time matches remote time");
+		else
+			if (stime(&remote_time) < 0)
+				bb_perror_msg_and_die("cannot set time of day");
 	}
 
+	if ((flags & 1) == 0)
+		printf("%s", ctime(&remote_time));
+
 	return EXIT_SUCCESS;
 }