diff --git a/lib/hector.rb b/lib/hector.rb index a266431..8351d2a 100644 --- a/lib/hector.rb +++ b/lib/hector.rb @@ -11,8 +11,10 @@ require "hector/session" module Hector - def self.start_server(address = "0.0.0.0", port = 6767) + def self.start_server(address = "0.0.0.0", port = 6767, ssl_port = 6868) EventMachine.start_server(address, port, Connection) + EventMachine.start_server(address, ssl_port, SSLConnection) logger.info("Hector running on #{address}:#{port}") + logger.info("Secure Hector running on #{address}:#{ssl_port}") end end diff --git a/lib/hector/connection.rb b/lib/hector/connection.rb index 9b511c2..5237d47 100644 --- a/lib/hector/connection.rb +++ b/lib/hector/connection.rb @@ -98,4 +98,15 @@ def log_tag end end end + + class SSLConnection < Connection + def post_init + log(:info, "opened SSL connection") + start_tls + end + + def ssl_handshake_completed + log(:info, "SSL handshake completed") + end + end end