ssl_scan 0.0.1

data/lib/ssl_scan/exceptions.rb

@@ -0,0 +1,274 @@
+# -*- coding: binary -*-
+
+module SSLScan
+
+###
+#
+# Base mixin for all exceptions that can be thrown from inside SSLScan.
+#
+###
+module Exception
+end
+
+###
+#
+# This exception is raised when a timeout occurs.
+#
+###
+class TimeoutError < Interrupt
+  include Exception
+
+  def to_s
+    "Operation timed out."
+  end
+end
+
+###
+#
+# This exception is raised when a method is called or a feature is used that
+# is not implemented.
+#
+###
+class NotImplementedError < ::NotImplementedError
+  include Exception
+
+  def to_s
+    "The requested method is not implemented."
+  end
+end
+
+###
+#
+# This exception is raised when a generalized runtime error occurs.
+#
+###
+class RuntimeError < ::RuntimeError
+  include Exception
+end
+
+###
+#
+# This exception is raised when an invalid argument is supplied to a method.
+#
+###
+class ArgumentError < ::ArgumentError
+  include Exception
+
+  def initialize(message = nil)
+    @message = message
+  end
+
+  def to_s
+    str = 'An invalid argument was specified.'
+    if @message
+      str << " #{@message}"
+    end
+    str
+  end
+end
+
+###
+#
+# This exception is raised when an argument that was supplied to a method
+# could not be parsed correctly.
+#
+###
+class ArgumentParseError < ::ArgumentError
+  include Exception
+
+  def to_s
+    "The argument could not be parsed correctly."
+  end
+end
+
+###
+#
+# This exception is raised when an argument is ambiguous.
+#
+###
+class AmbiguousArgumentError < ::RuntimeError
+  include Exception
+
+  def initialize(name = nil)
+    @name = name
+  end
+
+  def to_s
+    "The name #{@name} is ambiguous."
+  end
+end
+
+###
+#
+# This error is thrown when a stream is detected as being closed.
+#
+###
+class StreamClosedError < ::IOError
+  include Exception
+
+  def initialize(stream)
+    @stream = stream
+  end
+
+  def stream
+    @stream
+  end
+
+  def to_s
+    "Stream #{@stream} is closed."
+  end
+end
+
+##
+#
+# Socket exceptions
+#
+##
+
+###
+#
+# This exception is raised when a general socket error occurs.
+#
+###
+module SocketError
+  include Exception
+
+  def to_s
+    "A socket error occurred."
+  end
+end
+
+###
+#
+# This exception is raised when there is some kind of error related to
+# communication with a host.
+#
+###
+module HostCommunicationError
+  def initialize(addr = nil, port = nil)
+    self.host = addr
+    self.port = port
+  end
+
+  #
+  # This method returns a printable address and optional port associated
+  # with the host that triggered the exception.
+  #
+  def addr_to_s
+    if host and port
+      "(#{host}:#{port})"
+    elsif host
+      "(#{host})"
+    else
+      ""
+    end
+  end
+
+  attr_accessor :host, :port
+end
+
+
+###
+#
+# This is a generic exception for errors that cause a connection to fail.
+#
+###
+class ConnectionError < ::IOError
+  include SocketError
+  include HostCommunicationError
+end
+
+###
+#
+# This exception is raised when a connection attempt fails because the remote
+# side refused the connection.
+#
+###
+class ConnectionRefused < ConnectionError
+  def to_s
+    "The connection was refused by the remote host #{addr_to_s}."
+  end
+end
+
+###
+#
+# This exception is raised when a connection attempt fails because the remote
+# side is unreachable.
+#
+###
+class HostUnreachable < ConnectionError
+  def to_s
+    "The host #{addr_to_s} was unreachable."
+  end
+end
+
+###
+#
+# This exception is raised when a connection attempt times out.
+#
+###
+class ConnectionTimeout < ConnectionError
+  def to_s
+    "The connection timed out #{addr_to_s}."
+  end
+end
+
+
+###
+#
+# This exception is raised when an attempt to use an address or port that is
+# already in use occurs, such as binding to a host on a given port that is
+# already in use.  Note that Windows raises this in some cases when attempting
+# to connect to addresses that it can't handle, e.g. "0.0.0.0".  Thus, this is
+# a ConnectionError.
+#
+###
+class AddressInUse < ConnectionError
+  include SocketError
+  include HostCommunicationError
+
+  def to_s
+    "The address is already in use #{addr_to_s}."
+  end
+end
+
+###
+#
+# This exception is raised when an unsupported internet protocol is specified.
+#
+###
+class UnsupportedProtocol < ::ArgumentError
+  include SocketError
+
+  def initialize(proto = nil)
+    self.proto = proto
+  end
+
+  def to_s
+    "The protocol #{proto} is not supported."
+  end
+
+  attr_accessor :proto
+end
+
+
+###
+#
+# This exception is raised when a proxy fails to pass a connection
+#
+###
+class ConnectionProxyError < ConnectionError
+  def initialize(host,port,ptype,reason)
+    super(host,port)
+    self.ptype = ptype
+    self.reason = reason
+  end
+
+  def to_s
+    self.ptype + ": " + self.reason
+  end
+
+  attr_accessor :ptype, :reason
+end
+
+end

data/lib/ssl_scan/io/bidirectional_pipe.rb

@@ -0,0 +1,161 @@
+# -*- coding: binary -*-
+module Rex
+module IO
+
+require 'rex/ui/text/output'
+require 'rex/ui/text/output/buffer'
+require 'rex/ui/text/input/buffer'
+
+class BidirectionalPipe < Rex::Ui::Text::Input
+
+  def initialize
+    @subscribers_out = {}
+    @subscribers_ref = {}
+    @subscribers_idx = 0
+    @pipe_input = Rex::Ui::Text::Input::Buffer.new
+
+    # We are the shell, the input, and the output
+    self.output = self
+    self.input  = self
+  end
+
+  def pipe_input
+    @pipe_input
+  end
+
+  def close
+    @pipe_input.close
+  end
+
+  def has_subscriber?(id)
+    @subscribers_out.has_key?(id)
+  end
+
+  def create_subscriber(id=nil)
+    id ||= (@subscribers_idx += 1).to_s
+    @subscribers_out[id] = Rex::Ui::Text::Output::Buffer.new
+    return id
+  end
+
+  def create_subscriber_proc(id=nil, &block)
+    id = create_subscriber(id)
+    @subscribers_ref[id] = block
+  end
+
+  def remove_subscriber(id)
+    @subscribers_out.delete(id)
+    @subscribers_ref.delete(id)
+  end
+
+  def write_input(buf)
+    @pipe_input.put(buf)
+  end
+
+  def read_subscriber(id)
+    output = @subscribers_out[id]
+
+    return '' if output.nil?
+
+    buf = output.buf
+
+    output.reset
+
+    buf
+  end
+
+  def print(msg='')
+    @subscribers_out.each_pair { |id, buf|
+      begin
+        @subscribers_ref[id] ? @subscribers_ref[id].call(msg) : buf.print(msg)
+      rescue ::Exception => e
+        # $stderr.puts "Error handling subscriber #{id}: #{e} #{e.backtrace.inspect}"
+        raise e
+      end
+    }
+    msg
+  end
+
+  def print_error(msg='')
+    print_line('[-] ' + msg)
+  end
+
+  def print_line(msg='')
+    print(msg + "\n")
+  end
+
+  def print_good(msg='')
+    print_line('[+] ' + msg)
+  end
+
+  def print_debug(msg='')
+    print_line('[!] ' + msg)
+  end
+
+  def flush
+  end
+
+  def print_status(msg='')
+    print_line('[*] ' + msg)
+  end
+
+  def print_warning(msg='')
+    print_line('[!] ' + msg)
+  end
+
+  #
+  # Wrappers for the pipe_input methods
+  #
+
+  def close
+    @pipe_input.close
+  end
+
+  def sysread(len = 1)
+    @pipe_input.sysread(len)
+  end
+
+  def put(msg)
+    @pipe_input.put(msg)
+  end
+
+  def gets
+    @pipe_input.gets
+  end
+
+  def eof?
+    @pipe_input.eof?
+  end
+
+  def fd
+    @pipe_input.fd
+  end
+
+  #
+  # Wrappers for shell methods
+  #
+
+  attr_accessor :output, :prompt, :input
+
+  def intrinsic_shell?
+    true
+  end
+
+  def supports_readline
+    false
+  end
+
+  def supports_color?
+    false
+  end
+
+  def pgets
+    gets
+  end
+
+
+protected
+
+end
+
+end
+end

data/lib/ssl_scan/io/datagram_abstraction.rb

@@ -0,0 +1,35 @@
+# -*- coding: binary -*-
+
+require 'socket'
+
+module Rex
+module IO
+
+###
+#
+# This class provides an abstraction to a datagram based
+# connection through the use of a datagram socketpair.
+#
+###
+module DatagramAbstraction
+
+  #
+  # Creates a streaming socket pair
+  #
+  def initialize_abstraction
+    self.lsock, self.rsock = Rex::Socket.udp_socket_pair()
+  end
+
+
+  # The left side of the stream (local)
+  attr_reader :lsock
+  # The right side of the stream (remote)
+  attr_reader :rsock
+
+protected
+  attr_writer :lsock
+  attr_writer :rsock
+
+end
+
+end; end