nagios-gearman-downtime 1.0.0

data/bin/downtime_client

@@ -0,0 +1,90 @@
+#!/usr/bin/env ruby
+require 'gearman'
+require 'base64'
+require "openssl"
+require "thor"
+require "../lib/nagios-gearman-downtime"
+
+class Downtime < Thor
+
+  default_task 'show_help'
+
+  desc "show_help", 'Display help for downtime client'
+
+  def show_help
+    puts <<-HELP
+
+Usage:
+  downtime_client set --duration=duration in minutes --gearman-server=Gearman server address
+
+  --object_type [host, 
+                 hostgroup_host, 
+                 hostgroup_svc, 
+                 servicegroup_host, 
+                 servicegroup_svc,
+                 service,
+                 enable_servicegroup,
+                 disable_servicegroup], default => host
+
+  --object_name  default => 'hostname -f', the name of the object type
+
+
+  --encryption true/false - default => false
+  --key Key to encrypt request - must be identical to the gearman_server key
+  --queue gearman server queue default => downtime
+
+
+  --start_time  default => now
+  --end_time    default => now + duration
+  --duration    time in minutes, default => 1 minute
+
+  --author    Author of downtime (default hostname) 
+  --comment   default => 'Downtime recieved by gearman server'
+    HELP
+  end
+
+  desc "set", "Send Downtime request to Gearman server - downtime queue"
+
+  method_option :gearman_server, :banner => 'Gearman server address', :desc => 'Gearman server address', :required => true
+  method_option :gearman_port,   :desc => 'Gearman server port',                                                     :default => '4730'
+  method_option :encryption,     :desc => 'Enable encryption of request( true / false)', :type => 'boolean',         :default => false
+  method_option :key,            :desc => 'Key to encrypt request - must be identical to the gearman_server key',    :default => ''
+  method_option :object_type,    :desc => 'Object type in nagios',:enum => %w{ host hostgroup_host hostgroup_svc servicegroup_host servicegroup_svc service enable_servicegroup disable_servicegroup}, :default => 'host'
+  method_option :object_name,    :desc => 'Name of the object in nagios - default to hostname -f ',                  :default => `hostname -f`.chomp
+  method_option :start_time,     :desc => 'Downtime start_time in epoch - default now',                              :default => Time.now.to_i
+  method_option :end_time,       :desc => 'Downtime end_time in epoch - default now'                                
+  method_option :duration, :banner => 'duration in minutes',      :desc => 'duration of downtime in minutes', :required => true                                         
+  method_option :fixed,          :desc => 'downtime start and end is fixed',                                         :default => 1
+  method_option :author,         :desc => 'Author of downtime (default hostname)',                                   :default => `hostname -f`.chomp
+  method_option :comment,        :desc => 'Comment for downtime',                                                    :default => 'Downtime recieved by gearman server'
+  method_option :trigger_id,     :desc => 'triggered by the ID of another scheduled downtime entry',                 :default => 0
+
+  def set   
+    duration = options[:duration].to_i * 60
+    puts "duration: #{duration}"
+    end_time = options[:end_time] || Time.now.to_i + duration
+
+    args = {
+      :gearman_job_server => ["#{options[:gearman_server]}:#{options[:gearman_port] || 4730}"],
+      :encryption         => options[:encryption],
+      :key                => options[:key],
+      :queue              => 'downtime',
+      :object_type        => options[:object_type],
+      :object_name        => options[:object_name],
+      :start_time         => options[:start_time],
+      :end_time           => end_time,
+      :duration           => duration,
+      :fixed              => options[:fixed],
+      :author             => options[:author],
+      :comment            => options[:comment],
+      :trigger_id         => options[:trigger_id]
+    }
+
+    
+    args.merge!(:job => ::Nagios::Gearman::Downtime.build_payload(args))
+
+    ::Nagios::Gearman::Downtime.send_external_cmd(args)
+  end 
+end
+
+Downtime.start

data/bin/downtime_worker

@@ -0,0 +1,54 @@
+#!/usr/bin/env ruby
+require 'gearman'
+require 'base64'
+require 'openssl'
+require 'logger'
+
+logger = Logger.new(STDOUT)
+
+key = `grep key /etc/mod-gearman/worker.conf`.chomp.split('=').last.strip
+logger.info "key: #{key}"
+
+servers = []
+servers << `grep server /etc/mod-gearman/worker.conf`.chomp.split('=').last.strip
+logger.info "servers: #{servers}"
+
+def aes256_decrypt(key, data)
+  key = null_padding(key)
+  aes = OpenSSL::Cipher.new('AES-256-ECB')
+  aes.decrypt
+  aes.key = key
+  aes.update(data) + aes.final
+end
+
+def null_padding(key)
+  padding = (32 - key.bytesize) if(key.kind_of?(String) && 32 != key.bytesize)
+  key += "\0" * padding
+end
+
+
+def process_external_cmd(data)
+  path = '/var/lib/icinga/rw/icinga.cmd'
+
+  File.open(path, 'a') do |pipe|
+    puts "Sending downtime: #{data}"
+    pipe.puts data
+    pipe.close
+  end
+end
+
+w = Gearman::Worker.new(['0.0.0.0:4730'])
+logger.info w.status
+
+w.add_ability('downtime') do |data,job|
+  logger.info "Data recieved:\n#{data}"
+  decoded_data  = Base64.strict_decode64(data.gsub("\n", ''))
+
+  decrypt = aes256_decrypt(key, decoded_data)
+  logger.info "decrypted data:\n#{decrypt}"
+
+  process_external_cmd(decrypt)
+  logger.info "Finished processing job"
+end
+
+loop { w.work }

metadata

@@ -0,0 +1,83 @@
+--- !ruby/object:Gem::Specification
+name: nagios-gearman-downtime
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Ami Mahloof
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-03-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: gearman-ruby
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Gearman Client & Worker for sending downtime and enable / disable notifications
+  for service groups
+email: ami.mahloof@gmail.com
+executables:
+- downtime_client
+- downtime_worker
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/downtime_client
+- bin/downtime_worker
+homepage: https://github.com/gtforge/nagios-gearman-downtime
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: Send external commands to mod-gearman
+test_files: []
+has_rdoc: