cloudstack-nagios 0.1.3 → 0.2.0

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    cloudstack-nagios (0.1.2)
+    cloudstack-nagios (0.1.3)
       erubis (~> 2.7.0)
       highline (~> 1.6.20)
       sshkit (~> 1.1.0)

data/lib/cloudstack-nagios/base.rb

@@ -64,6 +64,15 @@ module CloudstackNagios
       def filter_by(objects, tag_name, tag)
         objects.select {|r| r[tag_name].downcase == tag.downcase}
       end
+
+      def sshoptions
+       {
+         timeout: 5,
+         keys: %w(/var/lib/cloud/management/.ssh/id_rsa),
+         auth_methods: %w(publickey)
+       }
+      end
+
     end
   end
 end

data/lib/cloudstack-nagios/cli.rb

@@ -85,11 +85,13 @@ module CloudstackNagios
       require command
     end
 
-    desc "nagios_config SUBCOMMAND ...ARGS", "Nagios configuration commands"
+    desc ":nagios_config SUBCOMMAND ...ARGS", "Nagios configuration commands"
     subcommand :nagios_config, NagiosConfig
 
     desc "snmpd_config SUBCOMMAND ...ARGS", "snmpd configuration commands"
     subcommand :snmpd_config, SnmpdConfig
 
+    desc "check SUBCOMMAND ...ARGS", "nagios checks"
+    subcommand :check, Check
   end # class
 end # module

data/lib/cloudstack-nagios/commands/check.rb

@@ -0,0 +1,147 @@
+require 'sshkit/dsl'
+
+class Check < CloudstackNagios::Base
+
+  RETURN_CODES = {0 => 'ok', 1 => 'warning', 2 => 'critical'}
+
+  desc "check memory HOST", "check memory on host"
+  option :host,
+      desc: 'hostname or ipaddress',
+      required: true,
+      aliases: '-H'
+  option :warning,
+      desc: 'warning level',
+      type: :numeric,
+      default: 90,
+      aliases: '-w'
+  option :critical,
+      desc: 'critical level',
+      type: :numeric,
+      default: 95,
+      aliases: '-c'
+  def memory
+    host = SSHKit::Host.new("root@#{options[:host]}")
+    host.ssh_options = sshoptions
+    host.port = 3922
+    free_output = ""
+    on host do |h|
+      free_output = capture(:free)
+    end
+    values = free_output.scan(/\d+/)
+    total = values[0].to_i
+    free = values[2].to_i
+    free_b = values[7].to_i
+    data = check_data(total, free_b, options[:warning], options[:critical])
+    puts "MEMORY #{RETURN_CODES[data[0]]} - usage = #{data[1]}% | usage=#{data[1]}% total=#{total}, free=#{free}, free_wo_buffers=#{free_b}"
+    exit data[0]
+  end
+
+  desc "check cpu", "check memory on host"
+  option :host,
+     desc: 'hostname or ipaddress',
+     required: true,
+     aliases: '-H'
+  option :warning,
+      desc: 'warning level',
+      type: :numeric,
+      default: 90
+  option :critical,
+      desc: 'critical level',
+      type: :numeric,
+      default: 95
+  def cpu(host)
+    host = SSHKit::Host.new("root@#{options[:host]}")
+    host.ssh_options = sshoptions
+    host.port = 3922
+    mpstat_output = ""
+    on host do |h|
+      mpstat_output = capture(:mpstat)
+    end
+    values = mpstat_output.scan(/\d+/)
+    usage = 100 - values[-1].to_f 
+    data = check_data(100, usage, options[:warning], options[:critical])
+    puts "CPU #{RETURN_CODES[data[0]]} - usage = #{data[1]}% | usage=#{usage}"
+    exit data[0]
+  end
+
+  desc "check rootfs_rw", "check if the rootfs is read/writeable on host"
+  option :host,
+      desc: 'hostname or ipaddress',
+      required: true,
+      aliases: '-H'
+  option :warning,
+      desc: 'warning level',
+      type: :numeric,
+      default: 90,
+      aliases: '-w'
+  option :critical,
+      desc: 'critical level',
+      type: :numeric,
+      default: 95,
+      aliases: '-c'
+  def rootfs_rw
+    host = SSHKit::Host.new("root@#{options[:host]}")
+    host.ssh_options = sshoptions
+    host.port = 3922
+    mpstat_output = ""
+    on host do |h|
+      proc_out = capture(:cat, '/proc/mounts')
+    end
+    rootfs_rw = proc_out.match(/rootfs\srw\s/)
+    data = check_data(100, usage, options[:warning], options[:critical])
+    puts "ROOTFS_RW #{rootfs_rw ? 'OK - root fs writeable' : 'CRITICAL - rootfs not writeable'}"
+    exit data[0]
+  end
+
+  desc "check network", "check network usage on host"
+  option :host,
+      desc: 'hostname or ipaddress',
+      required: true,
+      aliases: '-H'
+  option :warning,
+      desc: 'warning level',
+      type: :numeric,
+      default: 90,
+      aliases: '-w'
+  option :critical,
+      desc: 'critical level',
+      type: :numeric,
+      default: 95,
+      aliases: '-c'
+  def network
+    host = SSHKit::Host.new("root@#{options[:host]}")
+    host.ssh_options = sshoptions
+    host.port = 3922
+    r1, t1, r2, t2 = ""
+    on host do |h|
+      r1 = capture(:cat, '/sys/class/net/eth0/statistics/rx_bytes').to_f
+      t1 = capture(:cat, '/sys/class/net/eth0/statistics/tx_bytes').to_f
+      sleep 1
+      r2 = capture(:cat, '/sys/class/net/eth0/statistics/rx_bytes').to_f
+      t2 = capture(:cat, '/sys/class/net/eth0/statistics/tx_bytes').to_f
+    end
+    rbps = (r2 - r1) / 1024
+    tbps = (t2 - t1) / 1024
+    data = check_data(1048576, tbps, options[:warning], options[:critical])
+    puts "NETWORK #{RETURN_CODES[data[0]]} - usage = #{data[1]}% | rxbps=#{rbps.round(0)}, txbps=#{tbps.round(0)}"
+    exit data[0]
+  end
+
+  no_commands do
+
+    def check_data(total, usage, warning, critical)
+      usage_percent = 100.0 / total * usage
+      code = 3
+      if usage_percent < warning
+        code = 0
+      elsif usage_percent < critical
+        code = 1
+      else 
+        code = 2
+      end
+      [code, usage_percent.round(0)]
+    end
+
+  end
+
+end

data/lib/cloudstack-nagios/commands/nagios_config.rb

@@ -1,13 +1,13 @@
 class NagiosConfig < CloudstackNagios::Base
 
-  desc "hosts", "generate nagios hosts configuration for virtual routers"
+  desc "nagios_config hosts", "generate nagios hosts configuration for virtual routers"
   def hosts
-  	puts load_template("cloudstack_routers_hosts.cfg.erb").result(routers: routers)
+    puts load_template("cloudstack_routers_hosts.cfg.erb").result(routers: routers)
   end
 
-  desc "services", "generate nagios services configuration for virtual routers"
+  desc "nagios_config services", "generate nagios services configuration for virtual routers"
   def services
-  	puts load_template("cloudstack_routers_services.cfg.erb").result(routers: routers)
+    puts load_template("cloudstack_routers_services.cfg.erb").result(routers: routers)
   end
 
- end
+ end

data/lib/cloudstack-nagios/commands/snmpd_config.rb

@@ -1,37 +1,70 @@
 require 'sshkit/dsl'
-require 'highline/import'
+require 'socket'
+require 'timeout'
 
 class SnmpdConfig < CloudstackNagios::Base
 
-  desc "check", "check if snpd is configured on virtual routers"
-  def check
-  	
+  desc "snmpd_config check [vms]", "check if snmpd is configured on virtual routers"
+  def check(*vms)
+    if vms.size == 0
+      say 'Get a list of all routers from cloudstack..', :yellow
+      vms = router_ips(routers)
+    end
+    vms.each do |host|
+      begin
+        Timeout::timeout(1) do
+          socket = TCPSocket.new(host, "161")
+          socket.close
+          puts "port is open on host #{host}"
+        end
+      rescue => e
+        puts "port is closed on host #{host}"
+      end
+    end
   end
 
-  desc "enable", "enable snmpd configuration on virtual routers"
-  def enable
-  	hosts = routers.map {|router| router['linklocalip']}
-  	on hosts, in: :sequence, wait: 5 do
-			puts
-			puts "On host #{host}"
-			puts "____" * 20
-			puts
-  		puts output = capture(:free, '-m')
-  		puts output =~ /Mem:\s+(\d+)\s+(\d+)/
-  		puts $1
-  		puts $2
-		end
+  desc "snmpd_config enable [vms]", "enable snmpd configuration on virtual routers"
+  option :apt,
+      desc: 'use apt-get to install snmpd packages',
+      type: :boolean
+  def enable(*vms)
+    apt = options[:apt]
+    if vms.size == 0
+      say 'Get a list of all routers from cloudstack..', :yellow
+      vms = router_ips
+    end
+    hosts = vms.map do |router|
+      host = SSHKit::Host.new("root@#{router}")
+      host.ssh_options = sshoptions
+      host.port = 3922
+      host
+    end
+    say 'Connect to router and enable snmpd...', :yellow
+    on hosts, in: :sequence, wait: 10 do
+      begin
+        execute 'apt-get', 'update'
+        execute 'apt-get', '-y', 'install', 'snmpd'
+        upload! File.join(File.dirname(__FILE__), '..', 'files', 'snmpd.conf'), '/etc/snmp/snmpd.conf'
+        execute 'service', 'snmpd', 'restart'
+        execute 'iptables', '-A INPUT -p tcp -m tcp --dport 161 -j ACCEPT'
+      rescue StandardError => e 
+        puts 'configuration failed!'
+        puts e.message
+        puts e.backtrace
+      end
+    end
   end
 
   no_commands do
 
-  	def snmp_hosts(host_names)
-  		hosts = host_names.map do |host_name|
-				host = SSHKit::Host.new("root@#{host_name}")
-				host
-			end
-		end
+    def router_ips(vrs = routers)
+      ips = []
+      vrs.each do |router|
+        ips << router['linklocalip'] unless router['linklocalip'] == nil
+      end
+      ips
+    end
 
-	end
+  end
 
-end
+end

data/lib/cloudstack-nagios/files/snmpd.conf

@@ -23,6 +23,7 @@
 #  then remember to update the other occurances in this example config file to match.
 
 
+agentaddress udp:161,udp6:161,tcp:161,tcp6:161
 
 ###############################################################################
 #
@@ -99,4 +100,4 @@ includeAllDisks  10%
 load   12 10 5
 
 #  Walk the UCD-SNMP-MIB::laTable to see the resulting output
-#  Note that this table *will* be populated, even without a "load" entry in the snmpd.conf file
+#  Note that this table *will* be populated, even without a "load" entry in the snmpd.conf file

data/lib/cloudstack-nagios/helper.rb

@@ -6,12 +6,8 @@ module CloudstackNagios
     end
 
     def routers
-    	routers = client.list_routers
-    	projects = client.list_projects
-    	projects.each do |project|
-      	routers = routers + client.list_routers({projectid: project['id']})
-    	end
-    	routers
+      routers = client.list_routers
+      routers += client.list_routers(projectid: -1)
     end
   end
 end

data/lib/cloudstack-nagios/templates/cloudstack_routers_hosts.cfg.erb

@@ -3,7 +3,7 @@
 # Host configuration file
 #
 # Created by: 
-# Date:       <%= Date.new %>
+# Date:       <%= Date.new -%>
 # Version:    Nagios 3.x config file
 #
 # --- DO NOT EDIT THIS FILE BY HAND ---
@@ -11,20 +11,16 @@
 #
 ###############################################################################
 <% routers.each do |router| -%>
-
+  <% if router['linklocalip'] -%>
 define host {
         host_name                       <%= router['name'] %>
-        alias                           <%= router['name'] %>
         display_name                    <%= router['name'] %> <%= " - #{router['project']}" if router['project'] %>
         address                         <%= router['linklocalip'] %>
-        hostgroups                      Cloudstack-Host-Group
+        hostgroups                      Cloudstack-SystemVM
         check_command                   check-host-alive
-        use                             Linux-Host,host-pnp
-        notification_period             S1
-        _COMMUNITY                      public
         register                        1
 }
-
+  <% end -%>
 <% end -%>
 
 ###############################################################################

data/lib/cloudstack-nagios/templates/cloudstack_routers_services.cfg.erb

@@ -8,22 +8,25 @@
 # Version:    Nagios 3.x config file
 #
 # --- DO NOT EDIT THIS FILE BY HAND ---
-# cloudstack-cli will overwite all manual settings during the next update
+# cloudstack-nagios will overwite all manual settings during the next update
 #
 ###############################################################################
-<% routers.each do |router| -%>
-
-define service {
-      host_name                  <%= router['name'] %>
-      service_description        <%= router['name'] %> <%= " - #{router['project']}" if router['project'] %> - Memory
-      display_name               Memory
-      use                        Generic-Service,service-pnp
-      check_command              checkcommand.rb!memory
-      register           1
-   }
 
+define command {
+        command_name                   cs-nagios_check-memory
+        command_line                   cs-nagios check memory -H $HOSTADDRESS$ -w $ARG1$ -c $ARG$
+        register                       1
+}
 
-<% end -%>
+define service {
+        hostgroup_name                  Cloudstack-SystemVM
+        service_description             Memory
+        display_name                    Memory
+        servicegroups                   Cloudstack-SystemVM
+        use                             Generic-Service,service-pnp
+        check_command                   cs-nagios_check-memory!80!90
+        register                        1
+}
 
 ###############################################################################
 #

data/lib/cloudstack-nagios/version.rb

@@ -1,3 +1,3 @@
 module CloudstackNagios
-  VERSION = "0.1.3"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cloudstack-nagios
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-11-04 00:00:00.000000000 Z
+date: 2013-11-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rdoc
@@ -150,6 +150,7 @@ files:
 - lib/cloudstack-client/version.rb
 - lib/cloudstack-nagios/base.rb
 - lib/cloudstack-nagios/cli.rb
+- lib/cloudstack-nagios/commands/check.rb
 - lib/cloudstack-nagios/commands/nagios_config.rb
 - lib/cloudstack-nagios/commands/snmpd_config.rb
 - lib/cloudstack-nagios/files/snmpd.conf