rapid7/metasploit-framework

# -*- coding: binary -*-
require 'rex/socket'

module Msf
module Handler
module Reverse

###
#
# Implements the reverse Rex::Socket::Comm handlng.
#
###
module Comm

  def initialize(info = {})
    super

    register_advanced_options(
      [
        OptString.new('ReverseListenerComm', [ false, 'The specific communication channel to use for this listener']),
      ], Msf::Handler::Reverse::Comm)
  end

  def select_comm
    rl_comm = datastore['ReverseListenerComm'].to_s
    case rl_comm
    when 'local'
      comm = ::Rex::Socket::Comm::Local
    when /\A-?[0-9]+\Z/
      comm = framework.sessions.get(rl_comm.to_i)
      raise(RuntimeError, "Reverse Listener Comm (Session #{rl_comm}) does not exist") unless comm
      raise(RuntimeError, "Reverse Listener Comm (Session #{rl_comm}) does not implement Rex::Socket::Comm") unless comm.is_a? ::Rex::Socket::Comm
    when nil, ''
      comm = nil
    else
      raise(RuntimeError, "Reverse Listener Comm '#{rl_comm}' is invalid")
    end

    comm
  end

  def via_string(comm)
    comm_used = comm
    comm_used ||= Rex::Socket::Comm::Local

    if comm_used.respond_to?(:type) && comm_used.respond_to?(:sid)
      via = "via the #{comm_used.type} on session #{comm_used.sid}"
    else
      via = ""
    end

    via
  end

end

end
end
end