From 6334b9cdfc47bd85b2ce236572e08406324d25cd Mon Sep 17 00:00:00 2001 From: Steve Kemp Date: Mon, 12 Nov 2012 21:00:16 +0000 Subject: Initial dump of code. --- parser/parser.rb | 306 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 306 insertions(+) create mode 100755 parser/parser.rb (limited to 'parser/parser.rb') 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 -- cgit v1.2.1