From 502a52452cec9251dc297b386bfea8cb9fd0f4a8 Mon Sep 17 00:00:00 2001 From: Ben Passmore Date: Sat, 30 Nov 2019 15:21:50 +1000 Subject: [PATCH] Updated social_engineering for AR --- extensions/social_engineering/models/interceptor.rb | 5 +---- extensions/social_engineering/models/mass_mailer.rb | 5 +---- extensions/social_engineering/models/web_cloner.rb | 7 ++----- 3 files changed, 4 insertions(+), 13 deletions(-) diff --git a/extensions/social_engineering/models/interceptor.rb b/extensions/social_engineering/models/interceptor.rb index b8df088db..e6a70bd9a 100644 --- a/extensions/social_engineering/models/interceptor.rb +++ b/extensions/social_engineering/models/interceptor.rb @@ -6,10 +6,7 @@ module BeEF module Core module Models - class Interceptor < ActiveRecord::Base - attribute :id, :Serial - attribute :ip, :Text, :lazy => false - attribute :post_data, :Text, :lazy => false + class Interceptor < BeEF::Core::Model belongs_to :webcloner diff --git a/extensions/social_engineering/models/mass_mailer.rb b/extensions/social_engineering/models/mass_mailer.rb index 317571093..adbddc272 100644 --- a/extensions/social_engineering/models/mass_mailer.rb +++ b/extensions/social_engineering/models/mass_mailer.rb @@ -7,11 +7,8 @@ module BeEF module Core module Models - class Massmailer < ActiveRecord::Base + class Massmailer < BeEF::Core::Model - attribute :id, :Serial - - #todo fields end end diff --git a/extensions/social_engineering/models/web_cloner.rb b/extensions/social_engineering/models/web_cloner.rb index fa1977147..e3aedbad4 100644 --- a/extensions/social_engineering/models/web_cloner.rb +++ b/extensions/social_engineering/models/web_cloner.rb @@ -6,13 +6,10 @@ module BeEF module Core module Models - class Webcloner < ActiveRecord::Base - attribute :id, :Serial + class Webcloner < BeEF::Core::Model - attribute :uri, :Text, :lazy => false - attribute :mount, :Text, :lazy => false + has_many :interceptors - belongs_tos :extension_seng_interceptor, 'Interceptor' end end