123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293 |
- ###
- #
- # This plugin hooks all session creation and db events
- # and send desktop notifications using notify-send command.
- #
- ###
- module Msf
- class Plugin::EventLibnotify < Msf::Plugin
- include Msf::SessionEvent
- include Msf::DatabaseEvent
- def initialize(framework, opts)
- super
- @bin = opts[:bin] || opts['bin'] || `which notify-send`.chomp
- @bin_opts = opts[:opts] || opts['opts'] || '--app-name=Metasploit'
- raise 'libnotify not found' if @bin.empty?
- self.framework.events.add_session_subscriber(self)
- self.framework.events.add_db_subscriber(self)
- end
- def notify_send(urgency, title, message)
- system(@bin, @bin_opts, '-u', urgency, title, message)
- rescue StandarError => e
- puts "Error running #{@bin}: #{e.message}"
- end
- def on_session_open(session)
- notify_send('normal', 'Got Shell!',
- "New Session: #{session.sid}\nIP: #{session.session_host}\nPeer: #{session.tunnel_peer}\n"\
- "Platform: #{session.platform}\nType: #{session.type}")
- end
- def on_session_close(session, reason = '')
- notify_send('normal', 'Connection closed',
- "Session:#{session.sid} Type:#{session.type} closed.\n#{reason}")
- end
- def on_session_fail(reason = '')
- notify_send('critical', 'Session Failure!', reason)
- end
- def on_db_host(host)
- notify_send('normal', 'New host',
- "Address: #{host.address}\nOS: #{host.os_name}")
- end
- def on_db_host_state(host, _ostate)
- notify_send('normal', "Host #{host.address} changed",
- "OS: #{host.os_name}\nNb Services: #{host.service_count}\nNb vulns: #{host.vuln_count}\n")
- end
- def on_db_service(service)
- notify_send('normal', 'New service',
- "New service: #{service.host.address}:#{service.port}")
- end
- def on_db_service_state(service, _port, _ostate)
- notify_send('normal', "Service #{service.host.address}:#{service.port} changed",
- "Name: #{service.name}\nState: #{service.state}\nProto: #{service.proto}\nInfo: #{service.info}")
- end
- def on_db_vuln(vuln)
- notify_send('critical', "New vulnerability on #{vuln.host.address}:#{vuln.service ? vuln.service.port : '0'}",
- "Vuln: #{vuln.name}\nInfos: #{vuln.info}")
- end
- def on_db_ref(ref)
- notify_send('normal', 'New ref', "Reference #{ref.name} added in database.")
- end
- def on_db_client(client)
- notify_send('critical', 'New client', "New client connected: #{client.ua_string}")
- end
- def cleanup
- framework.events.remove_session_subscriber(self)
- framework.events.remove_db_subscriber(self)
- end
- def name
- 'libnotify'
- end
- def desc
- 'Send desktop notification with libnotify on sessions and db events'
- end
- end
- end
|