diff --git a/lib/loader.rb b/lib/loader.rb index da60a22c4..f71f923cb 100644 --- a/lib/loader.rb +++ b/lib/loader.rb @@ -37,6 +37,7 @@ require 'lib/model/plugin' require 'lib/model/http' require 'lib/model/browserdetails' require 'lib/model/distributedenginerules' +require 'lib/model/dynamiccommandinfo' require 'lib/crypto' diff --git a/lib/model/commandmodule.rb b/lib/model/commandmodule.rb index 4a8829119..ab8681292 100644 --- a/lib/model/commandmodule.rb +++ b/lib/model/commandmodule.rb @@ -12,8 +12,10 @@ class CommandModule property :name, Text, :lazy => false has n, :commands + has 1, :dynamic_command_info + end end -end \ No newline at end of file +end diff --git a/lib/model/dynamiccommandinfo.rb b/lib/model/dynamiccommandinfo.rb new file mode 100644 index 000000000..4b60c016d --- /dev/null +++ b/lib/model/dynamiccommandinfo.rb @@ -0,0 +1,19 @@ +module BeEF +module Models + +class DynamicCommandInfo + + include DataMapper::Resource + + storage_names[:default] = 'dynamic_command_info' + + property :id, Serial + property :name, Text, :lazy => false + property :description, Text, :lazy => false + belongs_to :command_module + +end + +end +end +