summaryrefslogtreecommitdiff
path: root/parser/parser.rb
diff options
context:
space:
mode:
authorSteve Kemp <steve@steve.org.uk>2012-11-12 21:00:16 +0000
committerSteve Kemp <steve@steve.org.uk>2012-11-12 21:00:16 +0000
commit6334b9cdfc47bd85b2ce236572e08406324d25cd (patch)
treebd0bd3cc279d8377efde2affc8dc223bfb858ca2 /parser/parser.rb
Initial dump of code.
Diffstat (limited to 'parser/parser.rb')
-rwxr-xr-xparser/parser.rb306
1 files changed, 306 insertions, 0 deletions
diff --git a/parser/parser.rb b/parser/parser.rb
new file mode 100755
index 0000000..aef436b
--- /dev/null
+++ b/parser/parser.rb
@@ -0,0 +1,306 @@
+#!/usr/bin/ruby
+#
+# Notes
+#
+# Macros may be defined either literally, or as a result of a HTTP-fetch.
+# Macro names match the pattern "^[A-Z_]$"
+#
+#
+# TODO:
+# Add command line parsing.
+#
+# Add --dump flag to just dump the JSON and not insert it.
+#
+# Write manual.
+#
+# Use a real parser framework?
+#
+# Love me.
+#
+
+
+require 'beanstalk-client'
+require 'getoptlong'
+require 'json'
+
+
+
+
+
+
+
+
+#
+# Simple wrapper to parse the configuration file
+#
+class MonitorConfig
+ attr_reader :MACROS
+ attr_reader :queue
+ attr_reader :filename
+
+ #
+ # Constructors
+ #
+ def initialize( filename )
+ @MACROS = Hash.new()
+ @queue = Beanstalk::Pool.new(['localhost:11300'])
+ @file = filename
+
+ if ( @file.nil? || ( ! File.exists?( @file) ) )
+ puts "Missing configuration file"
+ exit( 0 )
+ end
+ end
+
+
+ #
+ # Define a macro, from the configuration file.
+ #
+ def define_macro( line )
+ name = nil
+ val = Array.new
+
+ #
+ # Get the name
+ #
+ name = $1.dup if ( line =~ /^([A-Z_]+)\s+/ )
+
+
+ #
+ # Get the value
+ #
+ if ( line =~ /fetched\s+from\s+(.*)[\r\n\.]*$/ )
+
+ #
+ # HTTP-fetch
+ #
+ val.push( "steve")
+ val.push("kemp")
+
+ elsif ( line =~ /\s(is|are)\s+(.*)\.+$/ )
+
+ #
+ # Literal list.
+ #
+ tmp = $2.dup.split( /\s+and\s+/ )
+ tmp.each do |entry|
+ val.push( entry )
+ end
+
+ end
+
+ @MACROS[name] = val
+
+
+ #
+ # Save it away
+ #
+# puts "Macro definition: #{name}"
+ # val.each do |value|
+ # puts "\t#{value}"
+ # end
+ end
+
+
+ def is_macro?( name )
+ !@MACROS[name].nil?
+ end
+
+ def get_macro_targets( name )
+ @MACROS[name]
+ end
+
+
+
+
+
+ #
+ # Parse the configuration file, named in our constructor.
+ #
+ def parse_file()
+ #
+ # Parse the configuration file on the command line
+ #
+ File.open( @file, "r").each_line do |line|
+
+ #
+ # A blank line, or a comment may be skipped.
+ #
+ next if ( ( line =~ /^#/ ) || ( line.length < 1 ) )
+
+ # specification of mauve-server to which we should raise our alerts to.
+ next if ( line =~ /Mauve\s+server(.*)source/ )
+
+
+ #
+ # Look for macro definitions, inline
+ #
+ if ( line =~ /^([A-Z]_+)\s+are\s+fetched\s+from\s+([^\s]+)\.?/ )
+ define_macro( line )
+
+ elsif ( line =~ /^([0-9A-Z_]+)\s+(is|are)\s+/ )
+ define_macro( line )
+
+ elsif ( line =~ /\s+must\s+ping/ )
+
+ #
+ # Target
+ #
+ targets = Array.new
+
+ #
+ # Fallback target is the first token on the line
+ #
+ target = line.split( /\s+/)[0]
+
+
+ #
+ # If the target is a macro
+ #
+ if ( is_macro?( target ) )
+ targets = get_macro_targets(target)
+ else
+ targets.push( target )
+ end
+
+ #
+ # The alert-failure message
+ #
+ alert = "Ping failed"
+ if ( line =~ /otherwise '([^']+)'/ )
+ alert=$1.dup
+ end
+
+
+ #
+ # Store the test(s)
+ #
+ targets.each do |host|
+ test = {
+ :target_host => host,
+ :test_type => "ping",
+ :test_alert => alert
+ }
+
+ if ( !ENV['DUMP'].nil? )
+ puts ( test.to_json )
+ else
+ @queue.put( test.to_json )
+ end
+ end
+
+ elsif ( line =~ /\s+must\s+run\s+([^\s]+)\s+/i )
+
+ service = $1.dup
+
+ service.chomp!(".")
+
+ #
+ # Target
+ #
+ targets = Array.new
+ target = line.split( /\s+/)[0]
+
+ #
+ # If the target is a macro
+ #
+ if ( is_macro?( target ) )
+ targets = get_macro_targets( target )
+ else
+ targets.push( target )
+ end
+
+ #
+ # Alert
+ #
+ alert = "#{service} failed"
+ if ( line =~ /otherwise '([^']+)'/ )
+ alert=$1.dup
+ end
+
+ #
+ # Does this service require a port?
+ #
+ case service
+ when /ssh/ then
+ port=22
+ when /ldap/ then
+ port=389
+ when /^https$/ then
+ port=443
+ when /^http$/ then
+ port=80
+ when /rsync/i then
+ port=873
+ when /ftp/i then
+ port=21
+ when /telnet/i then
+ port=20
+ when /smtp/i then
+ port=25
+ end
+
+ if ( line =~ /\s+on\s+([0-9]+)/ )
+ port = $1.dup
+ end
+
+ targets.each do |host|
+
+ test = {
+ :target_host => host,
+ :test_type => service,
+ :test_port => port,
+ :test_alert => alert
+ }
+
+ if ( line =~ /\s+with\s+status\s+([0-9]+)\s+/ )
+ test[:http_status]=$1.dup
+ end
+ if ( line =~ /\s+with\s+content\s+'([^']+)'/ )
+ test[:http_text]=$1.dup
+ end
+
+ if ( !ENV['DUMP'].nil? )
+ puts ( test.to_json )
+ else
+ @queue.put( test.to_json )
+ end
+ end
+
+ else
+ puts "Unknown line: #{line}" if ( line.length > 2 )
+ end
+
+ end
+
+
+ end
+end
+
+#
+# Entry-point to our code.
+#
+if __FILE__ == $0 then
+
+
+ begin
+ opts = GetoptLong.new(
+ [ "--dump", "-d", GetoptLong::NO_ARGUMENT ],
+ [ "--file", "-f", GetoptLong::REQUIRED_ARGUMENT ]
+ )
+ opts.each do |opt, arg|
+ case opt
+ when "--dump":
+ ENV["DUMP"] = "1"
+ when "--file":
+ ENV["FILE"] = arg
+ end
+ end
+ rescue StandardError => ex
+ puts "Option parsing failed: #{ex.to_s}"
+ exit
+ end
+
+ mon = MonitorConfig.new( ENV['FILE'] )
+ mon.parse_file();
+end