module Sqlmap
  class Session
    def initialize(host, port)
      @host = host
      @port = port
    end

    def get(uri, headers = nil, params = nil)
      c = Rex::Proto::Http::Client.new(@host, @port)
      args = {
        'uri' => uri
      }

      args['headers'] = headers if headers
      args['vars_get'] = params if params
      begin
        res = c.request_cgi(args)
        res = c.send_recv(res)
        return res
      rescue Rex::ConnectionRefused
        return
      end
    end

    def post(uri, headers = nil, data = nil, originator_args = nil)
      c = Rex::Proto::Http::Client.new(@host, @port)
      args = {
        'uri' => uri,
        'method' => 'POST'
      }

      args.merge!(originator_args) if originator_args
      args['headers'] = headers if headers
      args['data'] = data if data
      begin
        res = c.request_cgi(args)
        res = c.send_recv(res)
        return res
      rescue Rex::ConnectionRefused
        return
      end
    end
  end
end