#
# Copyright (c) 2006-2015 Wade Alcorn - wade@bindshell.net
# Browser Exploitation Framework (BeEF) - http://beefproject.com
# See the file 'doc/COPYING' for copying permission
#
module BeEF
  module Core
    module Models
      #
      # Table stores each host identified on the zombie browser's network(s)
      #
      class NetworkHost

        include DataMapper::Resource
        storage_names[:default] = 'network_host'

        property :id, Serial

        property :hooked_browser_id, Text, :lazy => false
        property :ip, Text, :lazy => false
        property :hostname, String, :lazy => false
        property :type, String, :lazy => false # proxy, router, gateway, dns, etc
        property :os, String, :lazy => false
        property :mac, String, :lazy => false
        property :cid, String, :lazy => false # command id or 'init'

        #
        # Stores a network host in the data store
        #
        def self.add(host={})
          (print_error "Invalid hooked browser session"; return) unless BeEF::Filters.is_valid_hook_session_id?(host[:hooked_browser_id])
          (print_error "Invalid IP address"; return) unless BeEF::Filters.is_valid_ip?(host[:ip])

          # prevent duplicates
          return unless BeEF::Core::Models::NetworkHost.all(
            :hooked_browser_id => host[:hooked_browser_id],
            :ip => host[:ip],
            :hostname => host[:hostname],
            :type => host[:type],
            :os => host[:os],
            :mac => host[:mac]).empty?

          if host[:hostname].nil? && host[:type].nil? && host[:os].nil? && host[:mac].nil?
            return unless BeEF::Core::Models::NetworkHost.all(
              :hooked_browser_id => host[:hooked_browser_id],
              :ip => host[:ip]).empty?
          end

          # store the returned network host details
          network_host = BeEF::Core::Models::NetworkHost.new(
            :hooked_browser_id => host[:hooked_browser_id],
            :ip => host[:ip],
            :hostname => host[:hostname],
            :type => host[:type],
            :os => host[:os],
            :mac => host[:mac],
            :cid => host[:cid])
          result = network_host.save
          (print_error "Failed to save network host"; return) if result.nil?

          network_host
        end

      end

    end
  end
end