metasploit-framework/lib/rex/parser/ip360_xml.rb

98 lines
1.8 KiB
Ruby

require 'rexml/document'
require 'rex/ui'
module Rex
module Parser
class IP360XMLStreamParser
attr_accessor :on_found_host
def initialize(&block)
reset_state
on_found_host = block if block
end
def reset_state
@host = {'hname' => nil, 'hid' => nil, 'addr' => nil, 'mac' => nil, 'os' => nil,
'vulns' => ['vuln' => {'vulnid' => nil, 'port' => nil, 'proto' => nil} ],
'apps' => ['app' => {'appid' => nil, 'svcid' => nil, 'port' => nil, 'proto' => nil } ],
}
@state = :generic_state
end
def tag_start(name, attributes)
case name
when "host"
@host['hid'] = attributes['persistent_id']
when "ip"
@state = :is_ip
when "dnsName"
@state = :is_fqdn
when "macAddress"
@state = :is_mac
when "os"
@host['os'] = attributes['id']
when "vulnerability"
@x = Hash.new
@x['vulnid'] = attributes['id']
when "port"
@state = :is_port
when "protocol"
@state = :is_proto
when "application"
@y = Hash.new
@y['appid'] = attributes['application_id']
@y['svcid'] = attributes['svcid']
@y['port'] = attributes['port']
@y['proto'] = attributes['protocol']
@host['apps'].push @y
end
end
def text(str)
case @state
when :is_fqdn
@host['hname'] = str
when :is_ip
@host['addr'] = str
when :is_mac
@host['mac'] = str
when :is_port
@x['port'] = str
when :is_proto
@x['proto'] = str
end
end
def tag_end(name)
case name
when "host"
on_found_host.call(@host) if on_found_host
reset_state
when "vulnerability"
@host['vulns'].push @x
end
@state = :generic_state
end
def cdata(d)
#do nothing
end
# We don't need these methods, but they're necessary to keep REXML happy
#
def xmldecl(version, encoding, standalone) # :nodoc:
end
def comment(str) # :nodoc:
end
def instruction(name, instruction) # :nodoc:
end
def attlist # :nodoc:
end
end
end
end