From 1284646730a43d1821050bd6f0ec3fa1869b8fac Mon Sep 17 00:00:00 2001
From: "sussurro@happypacket.net"
 <sussurro@happypacket.net@b87d56ec-f9c0-11de-8c8a-61c5e9addfc9>
Date: Wed, 29 Dec 2010 21:30:32 +0000
Subject: [PATCH] Add in metasploit connecitivy and base base modules to create
 generic commands

git-svn-id: https://beef.googlecode.com/svn/trunk@645 b87d56ec-f9c0-11de-8c8a-61c5e9addfc9
---
 lib/modules/msfclient.rb  |   4 ++
 lib/modules/msfcommand.rb | 121 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 125 insertions(+)
 create mode 100644 lib/modules/msfcommand.rb

diff --git a/lib/modules/msfclient.rb b/lib/modules/msfclient.rb
index 7eab5dab2..8234b473c 100644
--- a/lib/modules/msfclient.rb
+++ b/lib/modules/msfclient.rb
@@ -73,6 +73,10 @@ module BeEF
 			res = self.call('module.compatible_payloads',name)
 			res
 		end
+		def get_options(name)
+			res = self.call('module.options','exploit',name)
+			res
+		end
 	end
 	
 end
diff --git a/lib/modules/msfcommand.rb b/lib/modules/msfcommand.rb
new file mode 100644
index 000000000..e84ff5cec
--- /dev/null
+++ b/lib/modules/msfcommand.rb
@@ -0,0 +1,121 @@
+require 'pp'
+module BeEF
+module Modules
+module Commands
+
+
+class Msf < BeEF::Command
+
+  
+  def initialize
+    super({
+      'Name' => 'Generic Metasploit Exploit',
+
+      'Description' => %Q{
+        This module will launch a Metasploit exploit against the host
+        },
+      'Category' => 'Metasploit',
+      'Author' => ['sussurro'],
+      'Data' => [ ], 
+      'File' => __FILE__,
+    })
+    
+    set_target({
+      'browser_name' =>     ALL
+    })
+
+    use 'beef.dom'
+    use_template!
+  end
+  
+  def callback
+                save({'result' => @datastore['result']})
+  end
+
+	def update_info(id)
+				mod = BeEF::Models::CommandModule.first(:id => id)
+				msfinfo = nil
+
+				if mod.dynamic_command_info == nil
+					msf = BeEF::MsfClient.new
+					msf.login()
+					msfinfo = msf.get_exploit_info(mod.name)
+				  mod.dynamic_command_info = BeEF::Models::DynamicCommandInfo.new( 
+																				 :name => msfinfo['name'],
+																				 :description => msfinfo['description']);
+					mod.save
+				else
+					msfinfo = mod.dynamic_command_info
+				end
+				@info['Name'] = msfinfo['name']
+				@info['Description'] = msfinfo['description']
+				@info['MsfModName'] = mod.name
+
+	end
+	def update_data()
+				modname = @info['MsfModName']
+
+				msf = BeEF::MsfClient.new
+				msf.login()
+
+				msfoptions = msf.get_options(modname)
+			  msfoptions.keys.each { |k|
+								next if msfoptions[k]['advanced'] == true
+								next if msfoptions[k]['evasion'] == true
+							  @info['Data'] << [ 'name' => k + '_txt', 'type' => 'label', 'html' => msfoptions[k]['desc']]
+							  case msfoptions[k]['type']
+							  when "string","address","port"
+									@info['Data'] << ['name' => k , 'ui_label' => k, 'value' => msfoptions[k]['default']]
+								when "bool"
+									@info['Data'] <<  ['name' => k, 'type' => 'checkbox', 'ui_label' => k ]
+							  when "enum"
+									@info['Data'] << [ 'name' => k, 'type' => 'combobox', 'ui_label' => k, 'store_type' => 'arraystore', 'store_fields' => ['enum'], 'store_data' => msfoptions[k]['enums'], 'valueField' => 'enum', 'displayField' => 'enum' , 'autoWidth' => true, 'mode' => 'local', 'value' => msfoptions[k]['default']]
+								else
+									print "K => #{k}\n"
+									print "Status => #{msfoptions[k]['advanced']}\n"
+									p msfoptions[k]
+								end
+				}
+
+				msfpayloads = msf.get_payloads(modname)
+				payloads = msfpayloads['payloads']
+				pl = []
+				payloads.each { |p|
+						pl << [p]
+			  }	
+				
+				@info['Data'] << [ 'name' => 'Payload', 
+				  'type' => 'combobox', 
+					'ui_label' => 'Payload',
+					'store_type' => 'arraystore', 
+					'store_fields' => ['payload'], 
+					'store_data' => pl, 
+					'valueField' => 'payload', 
+					'displayField' => 'payload' , 
+					'autoWidth' => true,
+					'mode' => 'local',
+					'emptyText' => "select a payload..."]
+
+				
+	end
+
+end
+
+
+end
+end
+end
+
+#@info = info
+#@datastore = @info['Data'] || nil
+#@friendlyname = @info['Name'] || nil
+#@target = @info['Target'] || nil
+#@output = ''
+#@path = @info['File'].sub(BeEF::HttpHookServer.instance.root_dir, '')
+#@default_command_url = '/command/'+(File.basename @path, '.rb')+'.js'
+#@id = BeEF::Models::CommandModule.first(:path => @info['File']).object_id
+#@use_template = false
+#@auto_update_zombie = false
+#@results = {}
+#@beefjs_components = {}
+