misc_rbtools/fake_tunnel_connector/fake_tunnel_connector.rb

require 'socket'
require './http.rb'
require 'time'

class FakeTunnelConnector
  attr_accessor :port, :server, :ssl_context
  def initialize(port, ssl_context = nil)
    @port = port
    @server = TCPServer.open(port)
    @ssl_context = ssl_context
  end

  def listen_and_respond
    if ssl_context
      tls_server = ssl_server
      tcp_socket = tls_server.accept
    else
      tcp_socket = server.accept
    end
    response = read_socket(tcp_socket)
    location = response[:start_line][:location]
    write_response(tcp_socket, location)
    tcp_socket.close
  end

  def ssl_server
    OpenSSL::SSL::SSLServer.new server, ssl_context if ssl_context
  end

  def read_socket(tcp_socket)
    start_line = []
    headers = []
    puts 'Request Incoming:'
    puts '-------------------'
    # read lines from socket
    while (line = tcp_socket.gets) && (line.chomp.length > 0)
      # check for a valid http verb sent
      start_line = parse_http_start_request_line(line) if Http::METHODS.include?(line.split.first)
      header_line = parse_http_header_request_line(line)
      headers << header_line unless header_line.nil?
    end
    puts start_line
    puts 'Request Headers:'
    puts '-------------------'
    puts headers
    puts "\r\n"
    { start_line: start_line, headers: headers }
  end

  def parse_http_start_request_line(line)
    split_request = line.split
    verb = split_request.first
    location = split_request.select { |l| l if l.start_with?('/') }.first
    protocol = split_request.last
    {
      verb: verb,
      location: location,
      protocol: protocol
    }
  end

  def parse_http_header_request_line(line)
    if line.split.first.end_with?(':')
      split_request = line.split(':')
      key = split_request[0]
      value = if split_request.count >= 3
                split_request[1] + ':' + split_request[2]
              else
                split_request[1]
              end
      Hash[key, value]
    end
  end

  def write_response(tcp_socket, location)
    route_response_string = route_request(location)
    tcp_socket.print(ok_headers)
    tcp_socket.print(ok(route_response_string))
  end

  def ok(body = 'Success')
    body + "\r"
  end

  def ok_headers
    "HTTP/1.1 200 OK\r\n" +
      "Date: #{Time.now.utc}\r\n" +
      "\r\n"
  end

  def route_request(location)
    if location == '/scans'
      scans
    elsif location == '/login'
      login
    else
      ok
    end
  end

  def scans
    body = '{"folders":[{"unread_count":0,"custom":0,"default_tag":0,"type":"trash","name":"Trash","id":7},{"unread_count":0,"custom":0,"default_tag":1,"type":"main","name":"My
    Scans","id":8}],"scans":[{"legacy":false,"permissions":128,"type":"remote","read":true,"last_modification_date":1535393942,"creation_date":1535393789,"status":"completed","uuid":"71e19d73-6308-42b0-adee-b26c9ddabfda","shared":false,"user_permissions":128,"owner":"edbellis","schedule_uuid":"template-7f64b006-dfaf-feee-85bd-ea9dd47280cf3a9573b1fc8089e7","timezone":null,"rrules":null,"starttime":null,"enabled":false,"control":true,"name":"Kenna
    Security: Nessus API (ON PREM YO)","id":29}],"timestamp":1535395026}'
    ok(body)
  end

  def login
    body = '{"token":"797118d801342a0c5c5be3ed5420782becbea2e3bceea9275543dff4ee62dfc4"}'
    ok(body)
  end
end