#
# Copyright (c) 2006-2024 Wade Alcorn - wade@bindshell.net
# Browser Exploitation Framework (BeEF) - https://beefproject.com
# See the file 'doc/COPYING' for copying permission
#
class Get_proxy_servers_wpad < BeEF::Core::Command
  def post_execute
    save({ 'result' => @datastore['result'] })

    configuration = BeEF::Core::Configuration.instance
    return unless configuration.get('beef.extension.network.enable') == true
    return unless @datastore['results'] =~ /^proxies=(.+)$/

    session_id = @datastore['beefhook']
    proxies = Regexp.last_match(1).to_s
    proxies.split(',').uniq.each do |proxy|
      next unless proxy =~ /^(SOCKS|PROXY)\s+([\d.]+:\d{1,5})/

      proxy_type = Regexp.last_match(1).to_s
      ip = Regexp.last_match(2).to_s.split(':')[0]
      port = Regexp.last_match(2).to_s.split(':')[1]
      proto = 'HTTP' if proxy_type =~ /PROXY/
      proto = 'SOCKS' if proxy_type =~ /SOCKS/
      if BeEF::Filters.is_valid_ip?(ip)
        print_debug("Hooked browser found #{proto} proxy [ip: #{ip}, port: #{port}]")
        BeEF::Core::Models::NetworkService.create(hooked_browser_id: session_id, proto: proto.downcase, ip: ip, port: port, type: "#{proto} Proxy")
      end
    end
  end
end