#
# $Id$
# $Revision$
#

module Msf

class Plugin::TokenHunter < Msf::Plugin

	class TokenCommandDispatcher
		include Msf::Ui::Console::CommandDispatcher

		def name
			"Token Hunter"
		end

		def commands
			{
				'token_hunt_user'        => "Scan all connected meterpreter sessions for active tokens corresponding to one or more users"
			}
		end

		def cmd_token_hunt_user(*args)

			opts = Rex::Parser::Arguments.new(
				"-h"   => [ false,  "This help menu"],
				"-f"   => [ true,   "A file containing a list of users to search for (one per line)"]
			)

			opt_userfile  = nil
			opt_users     = []

			opts.parse(args) do |opt, idx, val|
				case opt
				when "-h"
					print_line("Usage: token_hunt_user [options] <username> [username] .. [username]")
					print_line(opts.usage)
					return
				when "-f"
					opt_userfile = val
				else
					opt_users << val
				end
			end

			if(opt_userfile)
				::File.open(opt_userfile, "rb") do |fd|
					fd.each_line do |line|
						line.strip!
						next if line.empty?
						next if line =~ /^#/
						opt_users << line
					end
				end
			end

			opt_users.uniq!

			tokens_del = {}
			tokens_imp = {}

			framework.sessions.each_key do |sid|
				session = framework.sessions[sid]
				next if session.type != "meterpreter"

				print_status(">> Scanning session #{session.sid} / #{session.session_host}")

				if(! session.incognito)
					session.core.use("incognito")
				end

				if(! session.incognito)
					print_status("!! Failed to load incognito on #{session.sid} / #{session.session_host}")
					next
				end

				res = session.incognito.incognito_list_tokens(0)
				if(res)
					res["delegation"].split("\n").each do |user|

						opt_users.each do |needle|

							ndom,nusr = needle.split("\\")
							if(not nusr)
								nusr = ndom
								ndom = nil
							end

							if(not user.nil? and ndom and user.strip.downcase == needle.strip.downcase)
								print_status("FOUND: #{session.sid} - #{session.session_host} - #{user} (delegation)")
								next
							end

							fdom,fusr = user.split("\\")

							if (not fusr.nil? and ! ndom and fusr.strip.downcase == nusr.strip.downcase)
								print_status("FOUND: #{session.sid} - #{session.session_host} - #{user} (delegation)")
							end
						end

						tokens_del[user] ||= []
						tokens_del[user] << session.sid
					end


					res["impersonation"].split("\n").each do |user|

						opt_users.each do |needle|
							ndom,nusr = needle.split("\\")
							if(not nusr)
								nusr = ndom
								ndom = nil
							end

							if(not user.nil? and ndom and user.strip.downcase == needle.strip.downcase)
								print_status(">> Found #{session.sid} - #{session.session_host} - #{user} (impersonation)")
								next
							end

							fdom,fusr = user.split("\\")
							if (not fusr.nil? and ! ndom and fusr.strip.downcase == nusr.strip.downcase)
								print_status(">> Found #{session.sid} - #{session.session_host} - #{user} (impersonation)")
							end
						end

						tokens_imp[user] ||= []
						tokens_imp[user] << session.sid
					end
				end
			end
		end
	end


	def initialize(framework, opts)
		super
		add_console_dispatcher(TokenCommandDispatcher)
	end

	def cleanup
		remove_console_dispatcher('Token Hunter')
	end

	def name
		"token_hunter"
	end

	def desc
		"Search all active meterpreter sessions for specific tokens"
	end
end
end

