metasploit-framework/lib/rex/service.rb

48 lines
839 B
Ruby

require 'rex'
require 'rex/proto'
module Rex
###
#
# The service module is used to extend classes that are passed into the
# service manager start routine. It provides extra methods, such as reference
# counting, that are used to track the service instances more uniformly.
#
###
module Service
include Ref
require 'rex/services/local_relay'
#
# Returns the hardcore, as in porno, alias for this service. This is used
# by the service manager to manage singleton services.
#
def self.hardcore_alias(*args)
return "__#{args}"
end
def deref
rv = super
# If there's only one reference, then it's the service managers.
if @_references == 1
Rex::ServiceManager.stop_service(self)
end
rv
end
#
# Calls stop on the service once the ref count drops.
#
def cleanup
stop
end
attr_accessor :alias
end
end