httpx/test/support/servlets/socks5_with_none_server.rb

# frozen_string_literal: true

class Sock5WithNoneServer
  attr_reader :origin

  def initialize
    @port = 0
    @host = "localhost"

    @server = TCPServer.new(0)

    _, port, ip, _ = @server.addr
    @origin = URI::Generic.build(scheme: "socks5", host: ip, port: port)
  end

  def shutdown
    @server.close
  end

  def start
    begin
      loop do
        sock = @server.accept

        handshake = sock.readpartial(10)

        _version, _num, *_meth = handshake.unpack("C*")

        # not gonna bother verifying
        packet = [5, 0xff].pack("CC")

        sock.print(packet)
        sock.flush
      end
    rescue IOError
    end
  end
end