module BeEF
module Core
module Models
class BrowserDetails < BeEF::Core::Model
def self.get(session_id, key)
browserdetail = where(session_id: session_id, detail_key: key).first
return nil if browserdetail.nil?
return nil if browserdetail.detail_value.nil?
browserdetail.detail_value
end
def self.set(session_id, detail_key, detail_value)
browserdetails = BeEF::Core::Models::BrowserDetails.where(
session_id: session_id,
detail_key: detail_key
).first
if browserdetails.nil?
browserdetails = BeEF::Core::Models::BrowserDetails.new(
session_id: session_id,
detail_key: detail_key,
detail_value: detail_value || ''
)
result = browserdetails.save!
else
browserdetails.detail_value = detail_value || ''
result = browserdetails.save!
print_debug "Browser has updated '#{detail_key}' to '#{detail_value}'"
end
print_error "Failed to save browser details: #{detail_key}=#{detail_value}" if result.nil?
browserdetails
end
end
end
end
end