metasploit-framework/tools/makeiplist.rb

123 lines
2.6 KiB
Ruby
Executable File

#!/usr/bin/env ruby
#
# This script takes a list of ranges and converts it to a per line ip list.
# Demonstration:
# echo 192.168.100.0-50 >> rangelist.txt
# echo 192.155-156.0.1 >> rangelist.txt
# echo 192.168.200.0/25 >> rangelist.txt
# ruby tools/makeiplist.rb
#
# Author:
# mubix
#
msfbase = __FILE__
while File.symlink?(msfbase)
msfbase = File.expand_path(File.readlink(msfbase), File.dirname(msfbase))
end
$:.unshift(File.expand_path(File.join(File.dirname(msfbase), '..', 'lib')))
require 'msfenv'
require 'rex'
require 'optparse'
class OptsConsole
def self.parse(args)
options = {'output' => 'iplist.txt'}
opts = OptionParser.new do |opts|
opts.banner = %Q|This script takes a list of ranges and converts it to a per line ip list.
Usage: #{__FILE__} [options]|
opts.separator ""
opts.separator "Specific options:"
opts.on("-i", '-i <filename>', "Input file") do |v|
options['input'] = v.to_s
end
opts.on("-o", '-o <filename>', "(Optional) Output file. Default: iplist.txt") do |v|
options['output'] = v.to_s
end
opts.separator ""
opts.separator "Common options:"
opts.on_tail("-h", "--help", "Show this message") do
puts opts
exit
end
end
begin
opts.parse!(args)
if options['input'] == nil
puts opts
raise OptionParser::MissingArgument, "-i is a required option"
end
unless ::File.exists?(options['input'])
raise OptionParser::InvalidArgument, "Not found: #{options['input']}"
end
rescue OptionParser::InvalidOption
puts "[*] Invalid option, try -h for usage"
exit
rescue OptionParser::InvalidArgument => e
puts "[*] #{e.message}"
exit
end
if options.empty?
puts "[*] No options specified, try -h for usage"
exit
end
options
end
end
#
# Prints IPs
#
def make_list(in_f, out_f)
in_f.each_line do |range|
ips = Rex::Socket::RangeWalker.new(range)
ips.each do |ip|
out_f.puts ip
end
end
end
#
# Returns file handles
#
def load_files(in_f, out_f)
handle_in = ::File.open(in_f, 'r')
# Output file not found, assuming we should create one automatically
::File.open(out_f, 'w') {} unless ::File.exists?(out_f)
handle_out = ::File.open(out_f, 'a')
return handle_in, handle_out
end
options = OptsConsole.parse(ARGV)
in_f, out_f = load_files(options['input'], options['output'])
begin
puts "[*] Generating list at #{options['output']}"
make_list(in_f, out_f)
ensure
# Always makes sure the file descriptors are closed
in_f.close
out_f.close
end
puts "[*] Done."