1 files changed, 29 insertions, 3 deletions

diff --git a/plugins/check_time.c b/plugins/check_time.c
index 49e7c87..e6c3c71 100644
--- a/plugins/check_time.c
+++ b/plugins/check_time.c
@@ -43,6 +43,7 @@ unsigned long critical_diff = 0;
 int check_critical_diff = FALSE;
 int server_port = TIME_PORT;
 char *server_address = NULL;
+int use_udp = FALSE;
 
 int process_arguments (int, char **);
 void print_help (void);
@@ -65,7 +66,13 @@ main (int argc, char **argv)
         time (&start_time);
 
         /* try to connect to the host at the given port number */
-        if (my_tcp_connect (server_address, server_port, &sd) != STATE_OK) {
+        if (use_udp) {
+                result = my_udp_connect (server_address, server_port, &sd);
+        } else {
+                result = my_tcp_connect (server_address, server_port, &sd);
+        }
+
+        if (result != STATE_OK) {
                 if (check_critical_time == TRUE)
                         result = STATE_CRITICAL;
                 else if (check_warning_time == TRUE)
@@ -77,6 +84,20 @@ main (int argc, char **argv)
                            server_address, server_port);
         }
 
+        if (use_udp) {
+                if (send (sd, "", 0, 0) < 0) {
+                        if (check_critical_time == TRUE)
+                                result = STATE_CRITICAL;
+                        else if (check_warning_time == TRUE)
+                                result = STATE_WARNING;
+                        else
+                                result = STATE_UNKNOWN;
+                        die (result, 
+                          _("TIME UNKNOWN - could not send UDP request to server %s, port %d\n"),
+                          server_address, server_port);
+                }
+        }
+
         /* watch for the connection string */
         result = recv (sd, (void *)&raw_server_time, sizeof (raw_server_time), 0);
 
@@ -146,6 +167,7 @@ process_arguments (int argc, char **argv)
                 {"warning-connect", required_argument, 0, 'W'},
                 {"critical-connect", required_argument, 0, 'C'},
                 {"port", required_argument, 0, 'p'},
+                {"udp", no_argument, 0, 'u'},
                 {"timeout", required_argument, 0, 't'},
                 {"version", no_argument, 0, 'V'},
                 {"help", no_argument, 0, 'h'},
@@ -169,7 +191,7 @@ process_arguments (int argc, char **argv)
         }
 
         while (1) {
-                c = getopt_long (argc, argv, "hVH:w:c:W:C:p:t:", longopts,
+                c = getopt_long (argc, argv, "hVH:w:c:W:C:p:t:u", longopts,
                                                                          &option);
 
                 if (c == -1 || c == EOF)
@@ -252,6 +274,8 @@ process_arguments (int argc, char **argv)
                         else
                                 socket_timeout = atoi (optarg);
                         break;
+                case 'u':                                                                       /* udp */
+                        use_udp = TRUE;
                 }
         }
 
@@ -296,6 +320,8 @@ This plugin will check the time on the specified host.\n\n"));
         printf (_(UT_HOST_PORT), 'p', myport);
 
         printf (_("\
+ -u, --udp\n\
+    Use UDP to connect, not TCP\n\
  -w, --warning-variance=INTEGER\n\
     Time difference (sec.) necessary to result in a warning status\n\
  -c, --critical-variance=INTEGER\n\
@@ -317,7 +343,7 @@ void
 print_usage (void)
 {
         printf (_("\
-Usage: %s -H <host_address> [-p port] [-w variance] [-c variance]\n\
+Usage: %s -H <host_address> [-p port] [-u] [-w variance] [-c variance]\n\
     [-W connect_time] [-C connect_time] [-t timeout]\n"), progname);
         printf (_(UT_HLP_VRS), progname, progname);
 }