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