httparty/spec/support/ssl_test_server.rb

require 'openssl'
require 'socket'
require 'thread'

# NOTE: This code is garbage.  It probably has deadlocks, it might leak
# threads, and otherwise cause problems in a real system.  It's really only
# intended for testing HTTParty.
class SSLTestServer
  attr_accessor :ctx    # SSLContext object
  attr_reader :port

  def initialize(options={})
    @ctx = OpenSSL::SSL::SSLContext.new
    @ctx.cert = OpenSSL::X509::Certificate.new(options[:cert])
    @ctx.key = OpenSSL::PKey::RSA.new(options[:rsa_key])
    @ctx.verify_mode = OpenSSL::SSL::VERIFY_NONE   # Don't verify client certificate
    @port = options[:port] || 0
    @thread = nil
    @stopping_mutex = Mutex.new
    @stopping = false
  end

  def start
    @raw_server = TCPServer.new(@port)
    if @port == 0
      @port = Socket::getnameinfo(@raw_server.getsockname, Socket::NI_NUMERICHOST|Socket::NI_NUMERICSERV)[1].to_i
    end
    @ssl_server = OpenSSL::SSL::SSLServer.new(@raw_server, @ctx)
    @stopping_mutex.synchronize{
      return if @stopping
      @thread = Thread.new{ thread_main }
    }
    nil
  end

  def stop
    @stopping_mutex.synchronize{
      return if @stopping
      @stopping = true
    }
    @thread.join
  end

  private

    def thread_main
      until @stopping_mutex.synchronize{ @stopping }
        (rr,ww,ee) = select([@ssl_server.to_io], nil, nil, 0.1)
        next unless rr && rr.include?(@ssl_server.to_io)
        socket = @ssl_server.accept
        Thread.new{
          header = []
          until (line = socket.readline).rstrip.empty?
            header << line
          end

          socket.write <<'EOF'.gsub(/\r\n/n, "\n").gsub(/\n/n, "\r\n")
HTTP/1.1 200 OK
Connection: close
Content-Type: application/json; charset=UTF-8

{"success":true}
EOF
          socket.close
      }
      end
      @ssl_server.close
    end
end