Contents

# frozen_string_literal: true

require "http"
require "parallel"

module RangeScan
  class Scanner
    attr_reader :context
    attr_reader :host
    attr_reader :port
    attr_reader :scheme
    attr_reader :ssl_context
    attr_reader :timeout
    attr_reader :user_agent

    def initialize(host: nil, port: nil, scheme: "http", verify_ssl: true, timeout: 5, user_agent: nil)
      @host = host
      @port = port || (scheme == "http" ? 80 : 443)
      @timeout = timeout
      @scheme = scheme
      @user_agent = user_agent

      @ssl_context = OpenSSL::SSL::SSLContext.new
      @ssl_context.verify_mode = OpenSSL::SSL::VERIFY_NONE unless verify_ssl
    end

    def scan(ipv4s)
      Parallel.map(ipv4s) do |ipv4|
        get ipv4
      end.compact
    end

    private

    def default_headers
      { host: host, user_agent: user_agent }.compact
    end

    def ssl_options
      scheme == "http" ? {} : { ssl_context: ssl_context }
    end

    def get(ipv4)
      url = "#{scheme}://#{ipv4}:#{port}"

      begin
        res = HTTP.timeout(timeout).headers(default_headers).get(url, ssl_options)
        { ipv4: ipv4, code: res.code, body: res.body.to_s }
      rescue StandardError
        nil
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
rangescan-0.1.0	lib/rangescan/scanner.rb