Perl Nmap报告处理+入库

#!/usr/bin/perl
use Nmap::Parser; use DBI; use Config::IniFiles; main: { if ($#ARGV == -1){
 usage(); exit(1); }elsif($ARGV[1] eq undef || $ARGV[1] == m/^w$/){ usage(); exit(1); } my $xmlfile=$ARGV[0]; my $taskid=$ARGV[1]; nmap_info($xmlfile,$taskid); print "搞定!n"; } sub usage{ print "--------------------n"; print "Usage: perl nmap.xml taskidn"; print "--------------------n"; exit; } sub nmap_info{ local($file_name)=shift; local($taskid)=shift; my $cfg=new Config::IniFiles(-file=>"config.ini"); $drive=$cfg->val('DBI','db_drive'); $db_name=$cfg->val('DBI','db_database'); $db_host=$cfg->val('DBI','db_host'); $db_username=$cfg->val('DBI','db_username'); $db_password=$cfg->val('DBI','db_password'); my $np=new Nmap::Parser; $np->parsefile("$file_name"); my $in=DBI->connect("DBI:$drive:database=$db_name:host=$db_host","$db_username","$db_password") or die "Error:$!n"; my $session=$np->get_session(); for my $host ($np->all_hosts()){ if($host->addr eq undef){ $host_addr='null'; }else{ $host_addr=$host->addr; } if($host->status eq undef){ $host_status='null'; }elsif($host->status eq 'down'){ $host_status=1; }elsif($host->status eq 'up'){ $host_status=0; } if($host->hostname eq undef){ $host_hostname='null'; }else{ $host_hostname=$host->hostname; } if($host->mac_addr eq undef){ $host_mac_addr='null'; }else{ $host_mac_addr=$host->mac_addr(); } if($host->tcp_port_count eq undef){ $tcp_port_count='null'; }else{ $tcp_port_count=$host->tcp_port_count; } if($host->os_sig->osfamily eq undef){ $osfamily='null'; }else{ $osfamily=$host->os_sig->osfamily; } if($host->os_sig->osgen eq undef){ $osgen='null'; }else{ $osgen=$host->os_sig->osgen; } if($host->os_sig->name eq undef){ $os_name='null'; }else{ $os_name=$host->os_sig->name; } $start_time=$session->start_str(); $end_time=$session->time_str(); for my $tcp ($host->tcp_ports()){ my $service=$host->tcp_service($tcp); if($host->tcp_port_state($tcp) eq 'open'){ $tcp_port=1; }elsif($host->tcp_port_state($tcp) eq 'filtered'){ $tcp_port=2; }elsif($host->tcp_port_state($tcp) eq 'unfiltered'){ $tcp_port=3; } $service_name=$service->name(); $service_port=$service->port(); $service_confidence=$service->confidence(); my $insert=$in->do("insert into services(hostid,create_at,port,proto,state,name,update_at,info)values('$taskid','$start_time','$service_port','tcp','$tcp_port','$service_name','$end_time','$service_confidence')"); } for my $udp ($host->udp_ports()){ my $service=$host->udp_service($udp); if($host->udp_port_state($udp) eq 'open'){ $udp_port=1; }elsif($host->udp_port_state($udp) eq 'filtered'){ $udp_port=2; }elsif($host->udp_port_state($udp) eq 'unfiltered'){ $udp_port=3; } $service_name=$service->name(); $service_port=$service->port(); $service_confidence=$service->confidence(); my $insert=$in->do("insert into services(host_id,'udp','$udp_port','$service_confidence')"); } my $insert=$in->do("insert into hosts(taskid,address,mac,os_name,os_flavor,os_sp,os_lang,vuln_count,service_count,exploit_attemp_count)values('$taskid','$host_addr','$host_mac_addr','$host_hostname','$host_status','$osfamily','$osgen','$os_name','null','$tcp_port_count','null')"); } $in->disconnect(); }