Module: Msf::Sessions::PowerShell::Mixin

Included in:
Msf::Sessions::PowerShell
Defined in:
lib/msf/base/sessions/powershell.rb

Instance Method Summary collapse

Instance Method Details

#shell_command(cmd, timeout = 1800) ⇒ Object

Takes over the shell_command of the parent



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/msf/base/sessions/powershell.rb', line 8

def shell_command(cmd, timeout = 1800)
  # insert random marker
  strm = Rex::Text.rand_text_alpha(15)
  endm = Rex::Text.rand_text_alpha(15)

  # Send the shell channel's stdin.
  shell_write(";'#{strm}'\n" + cmd + "\n'#{endm}';\n")

  etime = ::Time.now.to_f + timeout

  buff = ''
  # Keep reading data until the marker has been received or the 30 minute timeout has occurred
  while (::Time.now.to_f < etime)
    res = shell_read(-1, timeout)
    break unless res

    timeout = etime - ::Time.now.to_f

    buff << res
    next unless buff.include?(endm)

    # if you see the end marker, read the buffer from the start marker to the end and then display back to screen
    buff = buff.split(/#{strm}\r\n/)[-1]
    buff = buff.split(endm)[0]
    buff.gsub!(/(?<=\r\n)PS [^>]*>/, '')
    return buff
  end
  buff
end