Initial dump of code.

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