0bfda88a46
git-svn-id: https://beef.googlecode.com/svn/trunk@521 b87d56ec-f9c0-11de-8c8a-61c5e9addfc9
86 lines
2.4 KiB
Ruby
86 lines
2.4 KiB
Ruby
module BeEF
|
|
module UI
|
|
|
|
class Logs < BeEF::HttpController
|
|
|
|
def initialize
|
|
super({
|
|
'paths' => {
|
|
'/all.json' => method(:select_all_logs),
|
|
'/zombie.json' => method(:select_zombie_logs)
|
|
}
|
|
})
|
|
end
|
|
|
|
# Selects logs in the database and returns them in a JSON format.
|
|
def select_all_logs
|
|
|
|
# get params
|
|
start = @params['start'].to_i || 0
|
|
limit = @params['limit'].to_i || 25
|
|
raise WEBrick::HTTPStatus::BadRequest, "start less than 0" if start < 0
|
|
raise WEBrick::HTTPStatus::BadRequest, "limit less than 1" if limit < 1
|
|
raise WEBrick::HTTPStatus::BadRequest, "limit less than or equal to start" if limit <= start
|
|
|
|
# get log
|
|
log = BeEF::Models::Log.all(:offset => start, :limit => limit, :order => [:date.desc])
|
|
raise WEBrick::HTTPStatus::BadRequest, "log is nil" if log.nil?
|
|
|
|
# format log
|
|
@body = logs2json(log)
|
|
|
|
end
|
|
|
|
# Selects the logs for a zombie
|
|
def select_zombie_logs
|
|
|
|
# get params
|
|
session = @params['session'] || nil
|
|
raise WEBrick::HTTPStatus::BadRequest, "session is nil" if session.nil?
|
|
start = @params['start'].to_i || 0
|
|
limit = @params['limit'].to_i || 25
|
|
raise WEBrick::HTTPStatus::BadRequest, "start less than 0" if start < 0
|
|
raise WEBrick::HTTPStatus::BadRequest, "limit less than 1" if limit < 1
|
|
raise WEBrick::HTTPStatus::BadRequest, "limit less than or equal to start" if limit <= start
|
|
|
|
zombie = BeEF::Models::Zombie.first(:session => session)
|
|
raise WEBrick::HTTPStatus::BadRequest, "zombie is nil" if zombie.nil?
|
|
raise WEBrick::HTTPStatus::BadRequest, "zombie.id is nil" if zombie.id.nil?
|
|
zombie_id = zombie.id
|
|
|
|
# get log
|
|
log = BeEF::Models::Log.all(:offset => start, :limit => limit, :zombie_id => zombie_id, :order => [:date.desc])
|
|
raise WEBrick::HTTPStatus::BadRequest, "log is nil" if log.nil?
|
|
|
|
# format log
|
|
@body = logs2json(log)
|
|
end
|
|
|
|
private
|
|
|
|
# Returns a list of logs in JSON format.
|
|
def logs2json(logs)
|
|
logs_json = []
|
|
count = logs.length
|
|
output = '{success: false}'
|
|
|
|
logs.each do |log|
|
|
logs_json << {
|
|
'id' => log.id.to_i,
|
|
'date' => log.date.to_s,
|
|
'event' => log.event.to_s,
|
|
'type' => log.type.to_s
|
|
}
|
|
end
|
|
|
|
# format output
|
|
output = {'success' => 'true', 'count' => count, 'logs' => logs_json}.to_json if not logs_json.empty?
|
|
|
|
output
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|