From 3950e934a66fdecb76acbf853d7331915c55f378 Mon Sep 17 00:00:00 2001 From: Ben Passmore Date: Sat, 30 Nov 2019 15:18:19 +1000 Subject: [PATCH] Updated ipec for AR --- extensions/ipec/models/ipec_exploits.rb | 8 ++------ extensions/ipec/models/ipec_exploits_run.rb | 9 ++------- 2 files changed, 4 insertions(+), 13 deletions(-) diff --git a/extensions/ipec/models/ipec_exploits.rb b/extensions/ipec/models/ipec_exploits.rb index 4c48589e3..1c4dcc46b 100644 --- a/extensions/ipec/models/ipec_exploits.rb +++ b/extensions/ipec/models/ipec_exploits.rb @@ -6,13 +6,9 @@ module BeEF module Core module Models - class IpecExploits < ActiveRecord::Base - attribute :id, :Serial - attribute :name, :Text, :lazy => false - attribute :protocol, :String, :lazy => false - attribute :os, :String, :lazy => false + class IpecExploits < BeEF::Core::Model - belongs_to :extension_ipec_exploits_run, 'IpecExploitsRun' + has_many :ipec_exploits_run end diff --git a/extensions/ipec/models/ipec_exploits_run.rb b/extensions/ipec/models/ipec_exploits_run.rb index c87f7aed8..bc7616616 100644 --- a/extensions/ipec/models/ipec_exploits_run.rb +++ b/extensions/ipec/models/ipec_exploits_run.rb @@ -6,14 +6,9 @@ module BeEF module Core module Models - class IpecExploitsRun < ActiveRecord::Base + class IpecExploitsRun < BeEF::Core::Model - attribute :id, :Serial - attribute :launched, :Boolean, :lazy => false - attribute :http_headers, :Text, :lazy => false - attribute :junk_size, :String, :length => 3, :lazy => false - - belongs_to :ipec_exploits + belongs_to :ipec_exploit end