Get rid of a cast which increases the required alignment.

[nagiosplug.git] / plugins-root / check_icmp.c

diff --git a/plugins-root/check_icmp.c b/plugins-root/check_icmp.c
index f0ba9d0ce7313d78e995d8035ee7e65cbcd88a73..c0b29f25cd4b60f4b733fb4b515ca463c9246744 100644 (file)
--- a/plugins-root/check_icmp.c
+++ b/plugins-root/check_icmp.c
@@ -54,6 +54,10 @@ const char *email = "nagiosplug-devel@lists.sourceforge.net";
 #include "netutils.h"
 #include "utils.h"
 
+#if HAVE_SYS_SOCKIO_H
+#include <sys/sockio.h>
+#endif
+#include <sys/ioctl.h>
 #include <sys/time.h>
 #include <sys/types.h>
 #include <stdio.h>
@@ -66,6 +70,7 @@ const char *email = "nagiosplug-devel@lists.sourceforge.net";
 #include <ctype.h>
 #include <netdb.h>
 #include <sys/socket.h>
+#include <net/if.h>
 #include <netinet/in_systm.h>
 #include <netinet/in.h>
 #include <netinet/ip.h>
@@ -79,7 +84,7 @@ const char *email = "nagiosplug-devel@lists.sourceforge.net";
 # define MAXTTL        255
 #endif
 #ifndef INADDR_NONE
-# define INADDR_NONE 0xffffffU
+# define INADDR_NONE (in_addr_t)(-1)
 #endif
 
 #ifndef SOL_IP
@@ -178,11 +183,13 @@ void print_help (void);
 void print_usage (void);
 static u_int get_timevar(const char *);
 static u_int get_timevaldiff(struct timeval *, struct timeval *);
+static in_addr_t get_ip_address(const char *);
 static int wait_for_reply(int, u_int);
 static int recvfrom_wto(int, char *, unsigned int, struct sockaddr *, u_int *);
 static int send_icmp_ping(int, struct rta_host *);
 static int get_threshold(char *str, threshold *th);
 static void run_checks(void);
+static void set_source_ip(char *);
 static int add_target(char *);
 static int add_target_ip(char *, struct in_addr *);
 static int handle_random_icmp(struct icmp *, struct sockaddr_in *);
@@ -235,10 +242,10 @@ crash(const char *fmt, ...)
 }
 
 
-static char *
+static const char *
 get_icmp_error_msg(unsigned char icmp_type, unsigned char icmp_code)
 {
-       char *msg = "unreachable";
+       const char *msg = "unreachable";
 
        if(debug > 1) printf("get_icmp_error_msg(%u, %u)\n", icmp_type, icmp_code);
        switch(icmp_type) {
@@ -371,6 +378,10 @@ main(int argc, char **argv)
        int icmp_sockerrno, udp_sockerrno, tcp_sockerrno;
        int result;
        struct rta_host *host;
+
+       setlocale (LC_ALL, "");
+       bindtextdomain (PACKAGE, LOCALEDIR);
+       textdomain (PACKAGE);
        
        /* print a helpful error message if geteuid != 0 */
        np_warn_if_not_root();
@@ -442,7 +453,7 @@ main(int argc, char **argv)
 
        /* parse the arguments */
        for(i = 1; i < argc; i++) {
-               while((arg = getopt(argc, argv, "vhVw:c:n:p:t:H:i:b:I:l:m:")) != EOF) {
+               while((arg = getopt(argc, argv, "vhVw:c:n:p:t:H:s:i:b:I:l:m:")) != EOF) {
                        switch(arg) {
                        case 'v':
                                debug++;
@@ -485,6 +496,9 @@ main(int argc, char **argv)
                                        crit_down = (unsigned char)strtoul(ptr + 1, NULL, 0);
                                }
                                break;
+                       case 's': /* specify source IP address */
+                               set_source_ip(optarg);
+                               break;
       case 'V':                 /* version */
         /*print_revision (progname, revision);*/ /* FIXME: Why? */
         exit (STATE_OK);
@@ -883,7 +897,7 @@ finish(int sig)
        unsigned char pl;
        double rta;
        struct rta_host *host;
-       char *status_string[] =
+       const char *status_string[] =
        {"OK", "WARNING", "CRITICAL", "UNKNOWN", "DEPENDENT"};
        int hosts_ok = 0;
        int hosts_warn = 0;
@@ -1105,6 +1119,40 @@ add_target(char *arg)
 
        return 0;
 }
+
+static void
+set_source_ip(char *arg)
+{
+       struct sockaddr_in src;
+
+       memset(&src, 0, sizeof(src));
+       src.sin_family = AF_INET;
+       if((src.sin_addr.s_addr = inet_addr(arg)) == INADDR_NONE)
+               src.sin_addr.s_addr = get_ip_address(arg);
+       if(bind(icmp_sock, (struct sockaddr *)&src, sizeof(src)) == -1)
+               crash("Cannot bind to IP address %s", arg);
+}
+
+/* TODO: Move this to netutils.c and also change check_dhcp to use that. */
+static in_addr_t
+get_ip_address(const char *ifname)
+{
+#if defined(SIOCGIFADDR)
+       struct ifreq ifr;
+       struct sockaddr_in ip;
+
+       strncpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name) - 1);
+       ifr.ifr_name[sizeof(ifr.ifr_name) - 1] = '\0';
+       if(ioctl(icmp_sock, SIOCGIFADDR, &ifr) == -1)
+               crash("Cannot determine IP address of interface %s", ifname);
+       memcpy(&ip, &ifr.ifr_addr, sizeof(ip));
+       return ip.sin_addr.s_addr;
+#else
+       errno = 0;
+       crash("Cannot get interface IP address on this platform.");
+#endif
+}
+
 /*
  * u = micro
  * m = milli
@@ -1227,6 +1275,8 @@ print_help(void)
   printf (" %s\n", "-c");
   printf ("    %s", _("critical threshold (currently "));
   printf ("%0.3fms,%u%%)\n", (float)crit.rta, crit.pl);
+  printf (" %s\n", "-s");
+  printf ("    %s\n", _("specify a source IP address or device name"));
   printf (" %s\n", "-n");
   printf ("    %s", _("number of packets to send (currently "));
   printf ("%u)\n",packets);