Class: Ronin::UI::CommandLine::Commands::Payload

Inherits:
Command
  • Object
show all
Defined in:
lib/ronin/ui/command_line/commands/payload.rb

Instance Method Summary collapse

Instance Method Details

#default(name = nil) ⇒ Object 



49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
 
# File 'lib/ronin/ui/command_line/commands/payload.rb', line 49

def default(name=nil)
  UI::Output.silent = true if options.raw?
  Database.setup(options[:database])

  # Load the payload
  if options[:file]
    load_payload!
  else
    find_payload!(name)
  end

  unless @payload
    print_error "Could not find the specified payload"
    exit -1
  end

  params = Parameters::Parser.parse(options[:params])
  params[:host] = options[:host] if options[:host]
  params[:port] = options[:port] if options[:port]
  params[:local_host] = options[:local_host] if options[:local_host]
  params[:local_port] = options[:local_port] if options[:local_port]

  begin
    # Build the payload
    @payload.build!(params)
  rescue Parameters::MissingParam, Payloads::Exception => e
    print_error(e.message)
  end

  raw_payload = @payload.raw_payload

  if options.raw?
    # Write the raw payload
    STDOUT.write(raw_payload)
  else
    # Dump the built payload
    puts raw_payload.dump
  end
end