module Msf

###
#
# This class hooks all session creation events and plays a sound
#
###

class Plugin::EventSounds < Msf::Plugin


	attr_accessor :theme, :base, :queue, :queue_thread

	include Msf::SessionEvent

	def play_sound(event)
		self.queue.push(event)
	end

	def on_session_open(session)
		event = 'session_open_' + session.type
		play_sound(event)
	end

	def on_session_close(session)
		sid = session.sid.to_s
		play_sound('session')
		sid.unpack("C*").each do |c|
			play_sound("num" + [c].pack("C"))
		end
		play_sound('closed')
	end

	def on_plugin_load
		play_sound('plugin_load')
	end

	def on_plugin_unload
		play_sound('plugin_unload')
	end

	def start_sound_queue
		self.queue_thread = Thread.new do
			begin
			while(true)
				while(event = self.queue.shift)
					path = ::File.join(self.base, self.theme, "#{event}.wav")
					if(::File.exists?(path))
						Rex::Compat.play_sound(path)
					else
						print_status("Warning: sound file not found: #{path}")
					end
				end
				select(nil, nil, nil, 0.25)
			end
			rescue ::Exception => e
				print_status("Sound plugin: fatal error #{e} #{e.backtrace}")
			end
		end
	end

	def stop_sound_queue
		self.queue_thread.kill if self.queue_thread
		self.queue_thread = nil
		self.queue = []
	end


	def initialize(framework, opts)
		super

		self.queue = []
		self.theme = opts['theme'] || 'default'
		self.base  = File.join(Msf::Config.install_root, "data", "sounds")
		self.framework.events.add_session_subscriber(self)
		start_sound_queue

		self.on_plugin_load
	end

	def cleanup
		self.on_plugin_unload
		self.framework.events.remove_session_subscriber(self)
		stop_sound_queue
	end

	def name
		"sounds"
	end

	def desc
		"Automatically plays a sound when various framework events occur"
	end

end
end