#!/usr/bin/env ruby
# = nessus-cli.rb:  Nessus command line interface for XML-RPC
# Author:: Vlatko Kosturjak
# 
# (C) Vlatko Kosturjak, Kost. Distributed under GPL and BSD (dual licensed).

require 'nessus-xmlrpc'
require 'getoptlong'

verbose = 0
debug = 0
operation = ''
targets = ''
deletereport = false
user = ''
password = ''
scanname = ''
output = ''
output1 = ''
wait = ''
policy = ''
url = ''

def intro 
  $stderr.print $0 + ": Nessus command line interface for XML-RPC\n"
  $stderr.print "(C) Vlatko Kosturjak, Kost. Distributed under GPL.\n"
  $stderr.print "\n"
end

intro

def give_help
  puts <<-EOF
--user <user>	user for login to Nessus server
--password <p>	password for login to Nessus server
--scan <name>	start scan with name
--target <ip>	specify list of targets, separated by comma
--policy <pol>	specify policy to use (name of policy)
--url <url>	url of Nessus server (default: localhost:8834)
--wait [t]	wait scan to finish (ask in regular periods of <t> for status)
--output <f>	output report XML to file <f>
--output1 <f>	output report XML v1 to file <f>
--reportdelete	delete report after finish or delete report by id (if alone)
--stop <id>	stop scan identified by <id>
--stop-all	stop all scans
--pause <id>	pause scan identified by <id>
--pause-all	pause all scans
--resume <id>	resume scan identified by <id>
--resume-all	resume all scans
--report <id>	download report identified by <id>
--list-scans	list scans
--list-policy	list policies
--status <id>	get status of scan by <id>
--verbose	be verbose
--debug		be even more verbose
--help		this help

Examples: 
#{$0} --user john --password doe --scan scan-localhost --wait --output report.xml --target localhost
EOF
  exit 0
end

if ARGV.length < 1
  give_help
end

opt = GetoptLong.new(
  ["--help", "-h", GetoptLong::NO_ARGUMENT],
  ["--verbose", "-v", GetoptLong::OPTIONAL_ARGUMENT],
  ["--target", "-t", GetoptLong::REQUIRED_ARGUMENT],
  ["--user", "-u", GetoptLong::REQUIRED_ARGUMENT],
  ["--password", "-p", GetoptLong::REQUIRED_ARGUMENT],
  ["--policy", "-P", GetoptLong::REQUIRED_ARGUMENT],
  ["--url", "-U", GetoptLong::REQUIRED_ARGUMENT],
  ["--deletereport", "-D", GetoptLong::OPTIONAL_ARGUMENT],
  ["--wait", "-w", GetoptLong::OPTIONAL_ARGUMENT],
  ["--scan", "-s", GetoptLong::REQUIRED_ARGUMENT],
  ["--list-scans", "-l", GetoptLong::NO_ARGUMENT],
  ["--list-policy", "-L", GetoptLong::NO_ARGUMENT],
  ["--status", "-W", GetoptLong::REQUIRED_ARGUMENT],
  ["--stop", "-S", GetoptLong::REQUIRED_ARGUMENT],
  ["--stop-all", "-a", GetoptLong::NO_ARGUMENT],
  ["--pause", "-q", GetoptLong::REQUIRED_ARGUMENT],
  ["--pause-all", "-Q", GetoptLong::NO_ARGUMENT],
  ["--resume", "-e", GetoptLong::REQUIRED_ARGUMENT],
  ["--resume-all", "-E", GetoptLong::NO_ARGUMENT],
  ["--report", "-r", GetoptLong::REQUIRED_ARGUMENT],
  ["--output", "-o", GetoptLong::REQUIRED_ARGUMENT],
  ["--output1", "-1", GetoptLong::REQUIRED_ARGUMENT]
)

def give_error
  $stderr.print "You used incompatible options, probably you mixed --scan with --stop"
  $stderr.print "or something similar."
  exit 0
end

opt.each do |opt,arg|
  case opt
    when	'--help'
      give_help
    when	'--user'
      user = arg
    when	'--password'
      password = arg
    when 	'--stop'
      if operation == ''
        operation = "stop"
        scanname = arg
      else
        give_error
      end
    when 	'--pause'
      if operation == ''
        operation = "pause"
        scanname = arg
      else
        give_error
      end
    when 	'--resume'
      if operation == ''
        operation = "resume"
        scanname = arg
      else
        give_error
      end
    when 	'--stop-all'
      if operation == ''
        operation = "stop-all"
      else
        give_error
      end
    when 	'--pause-all'
      if operation == ''
        operation = "pause-all"
      else
        give_error
      end
    when 	'--resume-all'
      if operation == ''
        operation = "resume-all"
      else
        give_error
      end
    when 	'--report'
      if operation == ''
        operation = "report"
        scanname = arg
      else
        give_error
      end
    when 	'--scan'
      if operation == ''
        operation = "scan"
        scanname = arg
      else
        give_error
      end
    when	'--target'
      if arg[0..6] == 'file://'
        f = File.open(arg[7..-1], "r")
        f.each_line do |line|
          line=line.chomp
          line=line.strip
          unless line == '' or line == nil
            if targets == ''
              targets = line
            else
              targets = targets + "," + line
            end
          end
        end
        f.close
      else
        # if there's multiple target options, add comma
        if targets == ''
          targets = arg
          
        else
          targets = targets + "," + arg
        end
      end
    when	'--wait'
      if arg == ''
        wait = 15
      else
        wait = arg.to_i
      end
    when	'--reportdelete'
      if arg == ''
        deletereport=true
      else
        operation = "reportdelete"
        scanname = arg
      end

    when	'--output'
      output = arg
    when	'--output1'
      output1 = arg
    when	'--policy'
      policy = arg
    when	'--status'
      if operation == ''
        operation = "status"
        scanname = arg
      else
        give_error
      end
    when	'--url'
      url = arg
    when 	'--verbose'
      if arg == ''
        verbose += 1
      else
        verbose = arg.to_i
      end
    when 	'--debug'
      if arg == ''
        debug += 1
      else
        debug = arg.to_i
      end
    when	'--list-scans'
      if operation == ''
        operation = "list-scans"
        scanname = arg
      else
        give_error
      end
    when	'--list-policy'
      if operation == ''
        operation = "list-policy"
        scanname = arg
      else
        give_error
      end
  end
end

if (user == '') or (password == '')
  $stderr.print "User and password is required to login to Nessus server"
  $stderr.print "Try --help!"
  exit 1
end 

$stderr.print "[i] Targets: " + targets +"\n" if verbose > 0 
$stderr.print "[i] Connecting to nessus server: " if verbose > 0 
n=NessusXMLRPC::NessusXMLRPC.new(url,user,password) 
if n.logged_in 
  $stderr.print "OK!\n" if verbose > 0
else
  $stderr.print "[e] Error connecting/logging to the server!\n" 
  exit 2
end

case operation
  when "scan"
    if policy == ''
      $stderr.print "[w] Policy not defined, using first served from the server\n"
      pid,name = n.policy_get_first
      $stderr.print "[w] using policy id " + pid + " with name " + name + "\n"
    else
      pid=n.policy_get_id(policy)
      if pid == ''
        $stderr.print "[e] policy doesn't exit: " + policy + "\n"
        exit 3
      end
    end	
    if targets == ''
      $stderr.print "[w] Targets not defined, using localhost as target\n"
      targets = '127.0.0.1'
    end
    $stderr.print "[i] Initiating scan with targets: "+targets+': ' if verbose > 0
    uid=n.scan_new(pid,scanname,targets)
    $stderr.print "done\n" if verbose > 0
    unless wait == ''
      while not n.scan_finished(uid)
        $stderr.print "[v] Sleeping for " + wait.to_s() + ": " if verbose > 1			
        sleep wait
        $stderr.print "done\n" if verbose > 1
        stat = n.scan_status(uid)
        print "\r" + stat if verbose > 0
      end	
    else
      puts uid
      exit 0
    end	
    unless output == ''
      $stderr.print "[i] Output XML report to file: "+output if verbose > 0
      content=n.report_file_download(uid)	
      File.open(output, 'w') {|f| f.write(content) }	
      $stderr.print ": done\n" if verbose > 0
    end
    unless output1 == ''
      $stderr.print "[i] Output XML1 report to file: "+output1 if verbose > 0
      content=n.report_file1_download(uid)	
      File.open(output, 'w') {|f| f.write(content) }	
      $stderr.print ": done\n" if verbose > 0
    end
    if deletereport
      $stderr.print "[i] Deleting report: " if verbose > 0
      n.report_delete(uid)
      $stderr.print "done\n" if verbose > 0
    end
  when "report"
    uid=scanname
    if (output == '') and (output1 == '') 
      $stderr.print "[e] You want report, but specify filename with --output or output1\n"
    end
    unless output == ''
      $stderr.print "[i] Output XML report to file: "+output if verbose > 0
      content=n.report_file_download(uid)	
      File.open(output, 'w') {|f| f.write(content) }	
      $stderr.print ": done\n" if verbose > 0
    end
    unless output1 == ''
      $stderr.print "[i] Output XML1 report to file: "+output1 if verbose > 0
      content=n.report1_file_download(uid)	
      File.open(output, 'w') {|f| f.write(content) }	
      $stderr.print ": done\n" if verbose > 0
    end
    if deletereport
      $stderr.print "[i] Deleting report: " if verbose > 0
      n.report_delete(uid)
      $stderr.print "done\n" if verbose > 0
    end
  when "stop"
    $stderr.print "[i] Stopping scan: " + scanname if verbose > 0
    n.scan_stop(scanname)
    $stderr.print "done\n" if verbose > 0
  when "stop-all"
    $stderr.print "[i] Stopping all scans: " if verbose > 0	
    list=n.scan_stop_all
    $stderr.print "done\n" if verbose > 0
    if verbose > 1
      list.each {|uuid| puts "[v] Stop all: " + uuid }
    end
  when "pause"
    $stderr.print "[i] Pausing scan: " + scanname if verbose > 0
    n.scan_pause(scanname)
    $stderr.print "done\n" if verbose > 0
  when "pause-all"
    $stderr.print "[i] Pausing all scans: " if verbose > 0	
    list=n.scan_pause_all
    $stderr.print "done\n" if verbose > 0
    if verbose > 1
      list.each {|uuid| puts "[v] Pause all: " + uuid }
    end
  when "resume"
    $stderr.print "[i] Resuming scan: " + scanname if verbose > 0
    n.scan_resume(scanname)
    $stderr.print "done\n" if verbose > 0
  when "resume-all"
    $stderr.print "[i] Resuming all scans: " if verbose > 0	
    list=n.scan_resume_all
    $stderr.print "done\n" if verbose > 0
    if verbose > 1
      list.each {|uuid| puts "[v] Resume all: " + uuid }
    end
  when "reportdelete"
    $stderr.print "[i] Deleting report: " + scanname if verbose > 0
    n.report_delete(scanname)
    $stderr.print "done\n" if verbose > 0
  when "status"
    puts "status: " + n.scan_status(scanname)	
  when "list-scans"
    list=n.scan_list_hash
    list.each {|scan| 
      puts scan['id']+":"+scan['name']+":"+ \
        scan['current']+"/"+scan['total']
    }
  when "list-policy"
    list=n.policy_list_names
    list.each {|policy| 
      puts policy 
    }
    
end

$stderr.print "[v] End reached.\n" if verbose > 1