#
# Meterpreter script for running multiple console commands on a meterpreter session
# Provided by Carlos Perez at carlos_perez[at]darkoperator[dot]com
# Verion: 0.1
#

################## Variable Declarations ##################
@client = client

# Setting Arguments
@@exec_opts = Rex::Parser::Arguments.new(
  "-h" => [ false,"Help menu."                        ],
  "-cl" => [ true,"Commands to execute. The command must be enclosed in double quotes and separated by a comma."],
  "-rc" => [ true,"Text file with list of commands, one per line."]
)

#Setting Argument variables
commands = nil
script = []
help = 0

################## Function Declarations ##################
# Function for running a list of commands stored in a array, returs string
def list_con_exec(cmdlst)
  print_status("Running Command List ...")
  cmdout = ""
  cmdlst.each do |cmd|
    next if cmd.strip.length < 1
    next if cmd[0,1] == "#"
    begin
      print_status "\tRunning command #{cmd}"
      @client.console.run_single(cmd)
    rescue ::Exception => e
      print_status("Error Running Command #{cmd}: #{e.class} #{e}")
    end
  end
  cmdout
end


def usage
  print_line("Console Multi Command Execution Meterpreter Script ")
  print_line(@@exec_opts.usage)
  raise Rex::Script::Completed
end
################## Main ##################
@@exec_opts.parse(args) { |opt, idx, val|
  case opt

  when "-cl"
    commands = val.split(",")
  when "-rc"
    script = val
    if not ::File.exists?(script)
      raise "Command List File does not exists!"
    else
      commands = []
      ::File.open(script, "r").each_line do |line|
        commands << line.chomp
      end
    end

  when "-h"
    help = 1
  end
}

if args.length == 0 or help == 1 or commands.nil?
  usage
else
  list_con_exec(commands)
  raise Rex::Script::Completed
end