9552f259b3
git-svn-id: https://beef.googlecode.com/svn/trunk@611 b87d56ec-f9c0-11de-8c8a-61c5e9addfc9
90 lines
2.4 KiB
Ruby
90 lines
2.4 KiB
Ruby
module BeEF
|
|
module UI
|
|
|
|
#
|
|
#
|
|
#
|
|
class Panel < BeEF::HttpController
|
|
|
|
def initialize
|
|
super({
|
|
'paths' => {
|
|
'/' => method(:index),
|
|
'/hooked-browser-tree-update.json' => method(:hooked_browser_tree_update)
|
|
}
|
|
})
|
|
end
|
|
|
|
# default index page
|
|
def index; end
|
|
|
|
# return a JSON object contains all the updates for the hooked browser trees
|
|
def hooked_browser_tree_update
|
|
# retrieve the hbs that are online
|
|
hooked_browsers_online = zombies2json_simple(BeEF::Models::Zombie.all(:lastseen.gte => (Time.new.to_i - 30)))
|
|
|
|
# retrieve the hbs that are offline
|
|
hooked_browsers_offline = zombies2json_simple(BeEF::Models::Zombie.all(:lastseen.lt => (Time.new.to_i - 30)))
|
|
|
|
# retrieve the distributed engine rules that are enabled
|
|
distributed_engine_rules = distributed_engine_rules_2_json_simple(BeEF::Models::DistributedEngineRules.all(:enabled => true))
|
|
|
|
# hash that gets populated with all the information for the hb trees
|
|
ret = {
|
|
'success' => true,
|
|
|
|
# the list of hb
|
|
'hooked-browsers' => {
|
|
'online' => hooked_browsers_online,
|
|
'offline' => hooked_browsers_offline
|
|
},
|
|
|
|
# the rules for the distributed engine
|
|
'ditributed-engine-rules' => distributed_engine_rules
|
|
}
|
|
|
|
@body = ret.to_json
|
|
end
|
|
|
|
# Takes a list distributed engine rules and format the results into JSON
|
|
def distributed_engine_rules_2_json_simple(rules)
|
|
#TODO
|
|
end
|
|
|
|
# TODO: we have duplicate functions here (see ui/zombies/zombies.rb), maybe we need to
|
|
# organize the code differently.
|
|
|
|
# Takes a list of zombies and format the results in a JSON array.
|
|
def zombies2json_simple(zombies)
|
|
zombies_hash = {}
|
|
i = 0
|
|
|
|
zombies.each do |zombie|
|
|
# create hash of zombie details
|
|
zombies_hash[i] = (get_simple_hooked_browser_hash(zombie))
|
|
i+=1
|
|
end
|
|
|
|
zombies_hash
|
|
end
|
|
|
|
# create a hash of simple hooked browser details
|
|
def get_simple_hooked_browser_hash(hooked_browser)
|
|
|
|
browser_icon = BeEF::Models::BrowserDetails.browser_icon(hooked_browser.session)
|
|
os_icon = BeEF::Models::BrowserDetails.os_icon(hooked_browser.session)
|
|
domain = BeEF::Models::BrowserDetails.get(hooked_browser.session, 'HostName')
|
|
|
|
return {
|
|
'session' => hooked_browser.session,
|
|
'ip' => hooked_browser.ip,
|
|
'domain' => domain,
|
|
'browser_icon' => browser_icon,
|
|
'os_icon' => os_icon
|
|
}
|
|
|
|
end
|
|
end
|
|
|
|
end
|
|
end |