infosecn1nja
/
metasploit-framework
mirror of https://github.com/infosecn1nja/metasploit-framework.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Support RPC API

bug/bundler_fix
wchen-r7 2016-02-18 00:39:12 -06:00
parent 089d6985b6
commit a5f3bddfc8
3 changed files with 34 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
lib/msf/core/rpc/v10/rpc_module.rb
View File

@ -1,5 +1,7 @@
# -*- coding: binary -*- # -*- coding: binary -*-
require 'msf/util/document_generator'
module Msf module Msf
module RPC module RPC
class RPC_Module < RPC_Base class RPC_Module < RPC_Base
@ -70,6 +72,17 @@ class RPC_Module < RPC_Base
end end
# Returns detailed information about a module in HTML.
#
# @return [String] HTML file.
# @example Here's how you would use this from the client:
# rpc.call('module.info_html', 'exploit', 'windows/smb/ms08_067_netapi')
def rpc_info_html(mtype, mname)
m = _find_module(mtype, mname)
Msf::Util::DocumentGenerator.get_module_document(m)
end
# Returns the metadata for a module. # Returns the metadata for a module.
# #
# @param [String] mtype Module type. Supported types include (case-sensitive): # @param [String] mtype Module type. Supported types include (case-sensitive):

2
lib/msf/ui/console/command_dispatcher/core.rb
View File

@ -774,7 +774,7 @@ class Core
if dump_json if dump_json
print(Serializer::Json.dump_module(active_module) + "\n") print(Serializer::Json.dump_module(active_module) + "\n")
elsif show_doc elsif show_doc
Msf::Util::DocumentGenerator.get_module_document(active_module) Msf::Util::DocumentGenerator.spawn_module_document(active_module)
else else
print(Serializer::ReadableText.dump_module(active_module)) print(Serializer::ReadableText.dump_module(active_module))
end end

26
lib/msf/util/document_generator.rb
View File

@ -50,12 +50,22 @@ module Msf
private private
def load_css
@css ||= lambda {
data = ''
File.open(CSS_BASE_PATH, 'rb') { |f| data = f.read }
return data
}.call
end
def md_to_html(md) def md_to_html(md)
r = Redcarpet::Markdown.new(Redcarpet::Render::MsfMdHTML, fenced_code_blocks: true) r = Redcarpet::Markdown.new(Redcarpet::Render::MsfMdHTML, fenced_code_blocks: true)
%Q| %Q|
<html> <html>
<head> <head>
<link rel="stylesheet" href="file://#{CSS_BASE_PATH}"> <style>
#{load_css}
</style>
</head> </head>
<body> <body>
#{r.render(md)} #{r.render(md)}
@ -247,6 +257,14 @@ module Msf
end end
def self.spawn_module_document(mod)
md = get_module_document(mod)
f = Rex::Quickfile.new(["#{mod.shortname}_doc", '.html'])
f.write(md)
f.close
Rex::Compat.open_webrtc_browser("file://#{f.path}")
end
def self.get_module_document(mod) def self.get_module_document(mod)
manual_path = File.join(PullRequestFinder::MANUAL_BASE_PATH, "#{mod.fullname}.md") manual_path = File.join(PullRequestFinder::MANUAL_BASE_PATH, "#{mod.fullname}.md")
@ -279,11 +297,7 @@ module Msf
items[:mod_targets] = mod.targets items[:mod_targets] = mod.targets
end end
md = n.get_md_content(items) n.get_md_content(items)
f = Rex::Quickfile.new(["#{mod.shortname}_doc", '.html'])
f.write(md)
f.close
Rex::Compat.open_webrtc_browser("file://#{f.path}")
end end
end end