Bump to version 1.22.1
[platform/upstream/busybox.git] / util-linux / rdate.c
index bb53927..6e35cd5 100644 (file)
 /* vi: set sw=4 ts=4: */
 /*
  * The Rdate command will ask a time server for the RFC 868 time
- *  and optionally set the system time.
+ * and optionally set the system time.
  *
  * 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 GPLv2 or later, see file LICENSE in this source tree.
 */
 
-#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>
+//usage:#define rdate_trivial_usage
+//usage:       "[-sp] HOST"
+//usage:#define rdate_full_usage "\n\n"
+//usage:       "Get and possibly set the system date/time from a remote HOST\n"
+//usage:     "\n       -s      Set the system date/time (default)"
+//usage:     "\n       -p      Print the date/time"
 
+#include "libbb.h"
 
-#define RFC_868_BIAS   2208988800UL
+enum { RFC_868_BIAS = 2208988800UL };
 
-int setdate= 0;
-int printdate= 0;
+static void socket_timeout(int sig UNUSED_PARAM)
+{
+       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;
+       uint32_t nett;
        int fd;
 
-       if (!(h = gethostbyname(host))) {       /* get the IP addr */
-               perror_msg("%s", host);
-               return(-1);
-       }
-       if ((tserv = getservbyname("time", "tcp")) == NULL) { /* find port # */
-               perror_msg("%s", "time");
-               return(-1);
-       }
-       if ((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {  /* get net connection */
-               perror_msg("%s", "socket");
-               return(-1);
-       }
+       /* Add a timeout for dead or inaccessible servers */
+       alarm(10);
+       signal(SIGALRM, socket_timeout);
 
-       memcpy(&sin.sin_addr, h->h_addr, sizeof(sin.sin_addr));
-       sin.sin_port= tserv->s_port;
-       sin.sin_family = AF_INET;
+       fd = create_and_connect_stream_or_die(host, bb_lookup_port("time", "tcp", 37));
 
-       if (connect(fd, (struct sockaddr *)&sin, sizeof(sin)) < 0) {    /* connect to time server */
-               perror_msg("%s", host);
+       if (safe_read(fd, &nett, 4) != 4)    /* read time from server */
+               bb_error_msg_and_die("%s did not send the complete time", host);
+       if (ENABLE_FEATURE_CLEAN_UP)
                close(fd);
-               return(-1);
-       }
-       if (read(fd, (void *)&nett, 4) != 4) {  /* read time from server */
-               close(fd);
-               error_msg("%s did not send the complete time\n", host);
-       }
-       close(fd);
 
-       /* convert from network byte order to local byte order.
+       /* Convert from network byte order to local byte order.
         * RFC 868 time is the number of seconds
-        *  since 00:00 (midnight) 1 January 1900 GMT
-        *  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
+        * since 00:00 (midnight) 1 January 1900 GMT
+        * 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)
+int rdate_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int rdate_main(int argc UNUSED_PARAM, 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 flags;
 
-       /* the default action is to set the date */
-       if (printdate==0 && setdate==0) setdate++;
+       opt_complementary = "-1";
+       flags = getopt32(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)
-                       perror_msg_and_die("Could not set time of day");
-       }
-       if (printdate) {
-               printf("%s", ctime(&time));
+       if (!(flags & 2)) { /* no -p (-s may be present) */
+               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("can't set time of day");
        }
 
+       if (flags != 1) /* not lone -s */
+               printf("%s", ctime(&remote_time));
+
        return EXIT_SUCCESS;
 }