76 lines
1.2 KiB
Ruby
76 lines
1.2 KiB
Ruby
require "xmlrpc/server"
|
|
require "rex"
|
|
|
|
module Msf
|
|
module RPC
|
|
class Service < ::XMLRPC::BasicServer
|
|
|
|
attr_accessor :service, :state
|
|
|
|
def initialize(srvhost, srvport, ssl=false, cert=nil, ckey=nil)
|
|
self.service = Rex::Socket::TcpServer.create(
|
|
'LocalHost' => srvhost,
|
|
'LocalPort' => srvport,
|
|
'SSL' => ssl
|
|
)
|
|
|
|
self.service.on_client_connect_proc = Proc.new { |client|
|
|
on_client_connect(client)
|
|
}
|
|
self.service.on_client_data_proc = Proc.new { |client|
|
|
on_client_data(client)
|
|
}
|
|
self.service.on_client_close_proc = Proc.new { |client|
|
|
on_client_close(client)
|
|
}
|
|
|
|
self.state = {}
|
|
super()
|
|
end
|
|
|
|
def start
|
|
self.state = {}
|
|
self.service.start
|
|
end
|
|
|
|
def stop
|
|
self.state = {}
|
|
self.service.stop
|
|
end
|
|
|
|
def wait
|
|
self.service.wait
|
|
end
|
|
|
|
def on_client_close(c)
|
|
self.state.delete(c)
|
|
end
|
|
|
|
def on_client_connect(c)
|
|
self.state[c] = ""
|
|
end
|
|
|
|
def on_client_data(c)
|
|
data = c.get_once(-1)
|
|
self.state[c] << data if data
|
|
|
|
procxml(c)
|
|
end
|
|
|
|
def procxml(c)
|
|
while(self.state[c].index("\x00"))
|
|
mesg,left = self.state[c].split("\x00", 2)
|
|
self.state[c] = left
|
|
begin
|
|
res = process(mesg)
|
|
rescue ::Exception => e
|
|
$stderr.puts "ERROR: #{e.class} #{e}"
|
|
end
|
|
c.put(res+"\x00")
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|