gnms 2.1.0.rc1

data/lib/node_network.rb

@@ -0,0 +1,269 @@
+class Network < Node
+  attr_accessor :name, :dns_domain, :netbios_domain, :host, :network, :node_view, :map_view, :node_size, :map_bg_type, :map_bg_data
+
+  def initialize (ip="", mask=24, ipv6="", mask6=64, mp=nil, p=nil, s=UNKNOWN, xd=0 ,yd=0, i="", d=NODESCR_PROPERTY, t="network", o="unknown", n=NONAME_PROPERTY, dns_domain=NONAME_PROPERTY, netbios_domain=NONAME_PROPERTY, uuid=create_uuid())
+
+    super(ip, mask, ipv6, mask6, mp, p, s, xd, yd, i, d, t, o, uuid)
+    @name=n
+    @dns_domain=NONAME_PROPERTY
+    @netbios_domain=NONAME_PROPERTY
+    @host=Hash.new
+    @network=Hash.new
+    @node_view=nil
+    @node_size=nil
+    @map_bg_type=nil
+    @map_bg_data=nil
+
+    if mp != nil and $network[mp]
+      if $network[mp].status == UNMANAGED
+        self.status=UNMANAGED
+      end
+    end
+    #create gui view
+    create_node()
+    #store node in db
+    #check from which the call is coming, if it's from db load fct we do nothing
+    caller_func = caller[1][/`.*'/][1..-2]
+    write_db() unless caller_func == "db_read_node"
+  end
+
+  #
+  # h is a node
+  #
+  def add_host(h)
+    if $treenode_win.visible?
+      $treenode_win.add_row(h)
+    end
+    @host[h.ip]=h
+  end
+
+  def add_network(h)
+    if $treenode_win.visible?
+      $treenode_win.add_row(h)
+    end
+    @network[h.ip]=h
+  end
+
+  def del_host(hip)
+    if $treenode_win.visible?
+      $treenode_win.del_row(@host[hip])
+    end
+    @host.delete hip
+  end
+
+  def del_net(hip)
+    if hip == @ip
+      $log.error("Can't delete itself ip=#{ip}")
+      return
+    end
+    if $treenode_win.visible?
+      $treenode_win.del_row(@network[hip])
+    end
+    @network.delete hip
+  end
+
+  def contain_host(hip)
+    @host.each_value {|node|
+      if node.ip == hip
+        return true
+      end
+    }
+    return false
+  end
+
+  def contain_network(hip)
+    #should be recursive
+    @network.each_value {|node|
+      if node.ip == hip
+        return true
+      else
+        return node.contain_network(hip)
+      end
+    }
+    return false
+  end
+
+  def get_node()
+    return @host.merge(@network)
+  end
+
+  def nb_node()
+    return nb_network() + nb_host()
+  end
+
+  def nb_host()
+    return @host.size
+  end
+
+  def nb_network()
+    return @network.size
+  end
+
+
+  def dns_domain?()
+    return dns_domain != NONAME_PROPERTY
+  end
+
+  def netbios_domain?()
+    return netbios_domain != NONAME_PROPERTY
+  end
+
+  def name?()
+    return (@name != nil) && (@name != NONAME_PROPERTY) && (@name != "")
+  end
+  #
+  # method for changing status (color) of the node
+  #
+  def change_status(st)
+    if @status != st
+      #status has changed
+      @status=st
+      @node_view.icon_color($status_color[$status_value.index(st)])
+      @links.each {|l|
+         l.update_status()
+      }
+      
+      if $treenode_win.visible?
+        $treenode_win.update_row(self)
+      end
+
+      #propagate status UNMANAGED  or UNKNOWN to all nodes includes in this network
+      if (st == UNMANAGED) || (st == UNKNOWN)
+        get_node().each_value {|node|
+          node.change_status(st)
+        }
+      end
+    end
+  end
+
+  #
+  # update status depending on node host
+  #
+  def update_status()
+    if @node_view != nil
+      total_up=0
+      total_other=0
+      get_node().each_value{|n|
+        if n.status == NORMAL
+          total_up+=1
+        elsif n.status != UNMANAGED
+          total_other+=1
+        end
+      }
+      total_host=total_up+total_other
+      if total_host > 0
+        percent_status_not_up=(total_other*100)/total_host
+        if percent_status_not_up >= 50
+          @status=CRITICAL
+          @node_view.icon_color($status_color[$status_value.index(CRITICAL)])
+        elsif percent_status_not_up == 0
+          @status=NORMAL
+          @node_view.icon_color($status_color[$status_value.index(NORMAL)])
+        elsif	percent_status_not_up >= 25
+          @status=MAJOR
+          @node_view.icon_color($status_color[$status_value.index(MAJOR)])
+        else
+          @status=MINOR
+          @node_view.icon_color($status_color[$status_value.index(MINOR)])
+        end
+        if $treenode_win.visible?
+          $treenode_win.update_row(self)
+        end
+        #need to propagate to the network upstair
+        $network[@map].update_status() unless @map == nil
+      else
+        @status=UNKNOWN
+        @node_view.icon_color($status_color[$status_value.index(UNKNOWN)])
+      end
+    end
+  end
+
+  #
+  # method for changing label of the node
+  # lb is nil when we just want a node label refresh
+  def change_label(lb=nil)
+    if $config.show_label
+      if (@old_label != lb) or lb==nil
+        @old_label=lb unless lb==nil
+        lb=@old_label unless lb!=nil
+        if node_view
+          node_view.set_text("#{eval lb}") unless ("#{eval lb}" == NODESCR_PROPERTY) || ("#{eval lb}" == NONAME_PROPERTY)
+        end
+      end
+    end
+  end
+
+  def create_node()
+    @old_label=""
+    #set default label
+    label="#{dns_domain}"
+    if !dns_domain? && ip != ""
+      label="#{ip}"
+      if name?
+        label="#{@name}"
+      end
+    end
+    @map_view=Goo::CanvasGroup.new($canvas_root,{:x => 0.0,:y => 0.0})
+    @map_view.visibility = Goo::CanvasItem::HIDDEN
+    @node_view=nil
+    if @map == nil
+      #its the root network
+      @node_view=NodeNetworkView.new($canvas_root, self, $status_color[$status_value.index(status)], "#{PIXMAP_PATH}/#{icon}", label)
+    else
+      #puts "create_node #{$network[@map]}"
+      if $network[@map]
+        @node_view=NodeNetworkView.new($network[@map].map_view, self, $status_color[$status_value.index(status)], "#{PIXMAP_PATH}/#{icon}", label)
+      end
+    end
+    change_label($config.network_label_view)
+    if @node_view
+       if ($config.show_label &&((@map==nil)||(get_map()==@map)))
+	 @node_view.show_text()
+       else
+	 @node_view.hide_text() unless !@node_view
+       end
+    end
+  end #create_node
+
+  def delete()
+    #first need to destroy the links
+    del_all_links()
+    if @node_view
+      #destroy the tooltips
+      @node_view.hide_tooltips()
+      #destroy the view
+      @node_view.unselect()
+      @node_view.destroy()
+    end
+    @node_view=nil
+
+    #remove from db
+    db_delete_node(NETWORKNODE, ip, netmask)
+
+    del_network(@ip)
+  end
+
+  #reload pixmap icon to apply changes
+  def reload()
+    if ($config.network_icon_view=="os")
+      @icon="#{find_extension("os/#{@os}")}"
+    else
+      @icon="#{find_extension("type/#{@type}")}"
+    end
+    @node_view.set_image(@icon) unless !@node_view
+  end
+
+  #
+  # Return a table for the find tool
+  #
+  def to_t
+    tmp=[]
+    tmp.push(@ip,$status[$status_value.index(@status)], @name, @dns_domain, @netbios_domain, @description, @type, @os)
+    return tmp
+  end
+
+  def write_db()
+    return db_insert_node_prp(uuid,NETWORKNODE,ip,netmask,ipv6,netmask6,nil,map,status,avail_sev,snmp_sev,custom_sev,trap_sev, service_sev, wmi_sev, syslog_sev, jmx_sev, x,y,icon,description, snmp_version, snmp_port, snmp_timeout, snmp_retry,type,os,"",nil,dns_domain,nil,netbios_domain,nil,nil,nil)
+  end
+
+end #class

data/lib/node_property.rb

@@ -0,0 +1,39 @@
+=begin header
+
+Property of a node
+
+David Maciejak
+20/01/2003
+
+=end
+
+require "#{GNMSLIB}"+'/node_listener'
+require "#{GNMSLIB}"+'/monitor/client/snmp/snmp'
+require "#{GNMSLIB}"+'/node'
+require "#{GNMSLIB}"+'/gui/node_host_property'
+require "#{GNMSLIB}"+'/gui/node_network_property'
+require "#{GNMSLIB}"+'/gui/color_macro'
+
+if $defined_lib_gruff
+  begin
+	  require 'gruff'
+	  require "#{GNMSLIB}"+'/gui/charts/gruff'
+  rescue LoadError
+	  $log.error("gruff ruby or rmagick library to display charts is missing, you will not be able to see them")
+	  $defined_lib_gruff=false
+  rescue RuntimeError => $msg
+	$log.error($msg)
+ 	$defined_lib_gruff=false
+  end
+end
+
+def node_property(node)
+  if node.status == UNMANAGED
+    return
+  end
+  if node.class == Host
+	  NodeHostProperty.new(node)
+  else
+	  NodeNetworkProperty.new(node)
+  end
+end

data/lib/notify/alarm.rb

@@ -0,0 +1,247 @@
+require "#{GNMSLIB}"+'/notify/xmpp_bot'
+require "#{GNMSLIB}"+'/notify/irc_bot'
+
+#
+# alarm module to warn admins
+#
+
+require 'net/smtp'
+
+def alarm_handle(src_module, category, ip_str, summary, datetime)
+  $log.debug("in alarm_handle")
+  if $config.level != nil && ($LEVEL.index(category)>=$LEVEL.index($config.level))
+    i=0
+    level_alert_type=$level_alert_type
+    if $host[ip_str] && (!$network[$host[ip_str].map].inherit_from_global)
+      level_alert_type=$network[$host[ip_str].map].alert_type_by_level
+    end
+    level_alert_type[category].each do |alert|
+      if alert == true
+        case i
+        when 0 then #MAIL
+          alarm_mail("#{datetime} #{ip_str} #{summary}", ip_str, "#{src_module}", category)
+        when 1 then #POPUP
+          alarm_popup("#{src_module} #{datetime} #{ip_str} #{summary}")
+        when 2 then #BEEP
+          if $host[ip_str] && (!$network[$host[ip_str].map].inherit_from_global)
+            if $network[$host[ip_str].map].sound_file_path_by_level[category] != nil
+              if !$sound_is_played
+                Thread.start { alarm_sound($network[$host[ip_str].map].sound_file_path_by_level[category]) }
+              end
+            else
+              alarm_beep()
+            end
+          else
+            if $config.sound_file_path_by_level[category] != nil
+              if !$sound_is_played
+                Thread.start { alarm_sound($config.sound_file_path_by_level[category]) }
+              end
+            else
+              alarm_beep()
+            end
+          end
+        when 3 then #SYSLOG
+          text=sprintf("module='%s', severity='%s', time='%s', src='%s', descr='%s'", src_module, category, datetime, ip_str, summary)
+          alarm_syslog(text)
+        when 4 then #SMS / PAGER
+          if ($config.sms_device != "") && ($config.sms_device != nil)
+            if $host[ip_str] && (!$network[$host[ip_str].map].inherit_from_global) && ($network[$host[ip_str].map].sms_num_by_level[category] != "") && ($network[$host[ip_str].map].sms_num_by_level[category] != nil)
+              alarm_sms($config.sms_device,$network[$host[ip_str].map].sms_num_by_level[category],"#{src_module} #{datetime} #{ip_str} #{summary}")
+            elsif ($config.sms_num_by_level[category] != "") && ($config.sms_num_by_level[category] != nil)
+              alarm_sms($config.sms_device,$config.sms_num_by_level[category],"#{src_module} #{datetime} #{ip_str} #{summary}")
+            end
+          end
+        when 5 then #IM
+          if $host[ip_str] && (!$network[$host[ip_str].map].inherit_from_global) && ($network[$host[ip_str].map].im_dest_by_level[category] != "") && ($network[$host[ip_str].map].im_dest_by_level[category] != nil)
+            alarm_xmpp($network[$host[ip_str].map].im_dest_by_level[category],"#{src_module} #{datetime} #{ip_str} #{summary}")
+          elsif ($config.im_dest_by_level[category] != "") && ($config.im_dest_by_level[category] != nil)
+            alarm_xmpp($config.im_dest_by_level[category],"#{src_module} #{datetime} #{ip_str} #{summary}")
+          end
+        when 6 then #SCRIPT EXECUTION
+          if $host[ip_str] && (!$network[$host[ip_str].map].inherit_from_global) && ($network[$host[ip_str].map].script_name_by_level[category] != "") && ($network[$host[ip_str].map].script_name_by_level[category] != nil)
+            alarm_script($network[$host[ip_str].map].script_name_by_level[category], ip_str)
+          elsif ($config.script_name_by_level[category] != "") && ($config.script_name_by_level[category] != nil)
+            alarm_script($config.script_name_by_level[category], ip_str)
+          else
+            $log.error("Invalid script name for #{category}")
+          end
+        when 7 then #IRC
+          alarm_irc("date=#{datetime} src=#{ip_str} descr=#{summary}")
+        end
+      end
+      i+=1
+    end
+  end
+end
+
+def alarm_beep()
+  printf "\a"
+end
+
+def alarm_sound(sound_file)
+  if $sound_is_played
+    return
+  end
+
+  if !FileTest.exist?("#{SOUND_DIR}/#{sound_file}")
+    $log.error("Sound file not found at #{SOUND_DIR}/#{sound_file}")
+    alarm_beep()
+    return
+  end
+
+  if $defined_lib_gst
+    $sound_is_played = true
+      #create the pipeline
+      pipeline = Gst::Pipeline.new
+      # create a disk reader
+      filesrc = Gst::ElementFactory.make("filesrc")
+      filesrc.location = "#{SOUND_DIR}/#{sound_file}"
+
+      decoder = Gst::ElementFactory.make("decodebin")
+
+      if decoder == nil
+	$log.error("gstreamer issue when initializing sound env")
+	return
+      end
+
+      audio_convert = Gst::ElementFactory.make("audioconvert")
+      audio_resample = Gst::ElementFactory.make("audioresample")
+      audio_sink = Gst::ElementFactory.make("autoaudiosink")
+      pipeline.add(filesrc, decoder, audio_convert, audio_resample, audio_sink)
+
+      # link elements
+      filesrc >> decoder
+      audio_convert >> audio_resample >> audio_sink
+
+      decoder.signal_connect("new-decoded-pad") do |element, pad|
+	sink_pad = audio_convert["sink"]
+	pad.link(sink_pad)
+      end
+            
+      reading = true
+      bus = pipeline.bus
+      bus.add_watch do |bus, message|
+	case message.type
+	when Gst::Message::EOS
+	  $sound_is_played = false
+	  reading = false
+	when Gst::Message::ERROR
+	  #p message.parse
+	  $sound_is_played = false
+	  reading = false
+	end
+	true
+      end
+
+      # start playing
+      pipeline.play
+      while(reading) do 
+        sleep(1)
+      end
+      pipeline.stop
+      pipeline.set_state(Gst::State::NULL)
+  else
+    $log.error("Sound can't be played as sound lib gstreamer is not available")   
+  end
+end
+
+def alarm_syslog(text)
+  Syslog.log(Syslog::LOG_WARNING, text)
+end
+
+def alarm_popup(text)
+  if $win.visible?
+    dialog = Gtk::MessageDialog.new($win, Gtk::Dialog::MODAL,
+    Gtk::MessageDialog::INFO,
+    Gtk::MessageDialog::BUTTONS_OK,
+    text)
+    dialog.signal_connect('response') { dialog.destroy }
+    dialog.show
+  elsif $icon.embedded?
+    #window is hidden and trayicon exists so we just blink the icon
+    $icon.set_blinking(true)
+  end
+end
+
+def alarm_mail(msg, ip, src_module=nil, sev=INFO)
+  if src_module != "core"
+
+    mail_server_addr=$config.mail_server_addr
+    mail_server_port=$config.mail_server_port
+
+    tmail_receipt=$config.mail_by_level[sev]
+    if $host[ip] && (!$network[$host[ip].map].inherit_from_global)
+      tmail_receipt=$network[$host[ip].map].mail_by_level[sev]
+    end
+
+    Thread.start {
+
+      tmail_receipt.each{|contact|
+        mail_dst=Contact::get_contact_email(contact.firstname, contact.lastname)
+        if mail_dst != ""
+          begin
+            Net::SMTP.start( mail_server_addr, mail_server_port ) {|smtp|
+              smtp.send_mail <<EndOfMail, "root@localhost", "#{mail_dst}"
+              From: gnms admin <#{ENV["USER"]}@localhost>
+              To: gnms contact <#{Contact::get_contact_email(contact.firstname, contact.lastname)}>
+              Subject: gnms notification
+
+              #{msg}
+EndOfMail
+
+            }
+          rescue Exception => msg
+            $log.error("Error #{sev} severity level #{msg} when sending mail to '#{mail_dst}' on '#{mail_server_addr}'/'#{mail_server_port}'")
+          end
+
+        else
+          $log.error("Mail contact empty for #{contact.firstname} #{contact.lastname}")
+        end
+      }
+    }
+
+  end
+end
+
+def alarm_script(command, ip)
+  if !FileTest.exist?("#{SCRIPT_DIR}/#{command}")
+    $log.error("Invalid script path name #{SCRIPT_DIR}/#{command}")
+  else
+    `#{SCRIPT_DIR}/#{command} #{ip}`
+  end
+end
+
+def alarm_sms(device, tnumber, txt)
+  if $defined_lib_sms == false
+    $log.error("Can't send SMS, notification type is disabled cause a library is missing")
+  else
+    Thread.start {
+      tnumber.each{|contact|
+        sms=Contact::get_contact_mobile(contact.firstname, contact.lastname)
+        if sms != ""
+          mySMS = SMSController.new(device, sms, txt)
+          mySMS.sendSMS
+        else
+          $log.error("SMS contact empty for #{contact.firstname} #{contact.lastname}")
+        end
+      }
+    }
+  end
+end
+
+def alarm_xmpp(im_dest_addr, txt)
+  if $im_daemon
+    im_dest_addr.each{|contact|
+      im=Contact::get_contact_im(contact.firstname, contact.lastname)
+      if im != ""
+        $im_daemon.add_in_queue(im, txt)
+      else
+        $log.error("IM contact empty for #{contact.firstname} #{contact.lastname}")
+      end
+    }
+  end
+end
+
+def alarm_irc(txt)
+  $irc_daemon.add_in_queue(txt) if $irc_daemon
+end