fix odd indentation and style issues

bug/bundler_fix
Brent Cook 2016-05-13 22:06:18 -05:00
parent 901b793406
commit 7cfc4d4523
1 changed files with 51 additions and 53 deletions

View File

@ -5,8 +5,8 @@ while File.symlink?(msfbase)
msfbase = File.expand_path(File.readlink(msfbase), File.dirname(msfbase)) msfbase = File.expand_path(File.readlink(msfbase), File.dirname(msfbase))
end end
$:.unshift(File.expand_path(File.join(File.dirname(msfbase), '..', '..', 'lib'))) $LOAD_PATH.unshift(File.expand_path(File.join(File.dirname(msfbase), '..', '..', 'lib')))
$:.unshift(ENV['MSF_LOCAL_LIB']) if ENV['MSF_LOCAL_LIB'] $LOAD_PATH.unshift(ENV['MSF_LOCAL_LIB']) if ENV['MSF_LOCAL_LIB']
require 'msfenv' require 'msfenv'
require 'msf/core' require 'msf/core'
@ -22,6 +22,7 @@ module PatternOffset
opt.banner = "Usage: #{__FILE__} [options]\nExample: #{__FILE__} -q Aa3A|0x39634138|0xFFFF4138\n[*] Exact match at offset 9" opt.banner = "Usage: #{__FILE__} [options]\nExample: #{__FILE__} -q Aa3A|0x39634138|0xFFFF4138\n[*] Exact match at offset 9"
opt.separator '' opt.separator ''
opt.separator 'Options:' opt.separator 'Options:'
opt.on('-q', '--query Aa0A', String, "Query to Locate") do |query| opt.on('-q', '--query Aa0A', String, "Query to Locate") do |query|
options[:query] = query options[:query] = query
end end
@ -69,13 +70,11 @@ class Driver
def run def run
query = (@opts[:query]) query = (@opts[:query])
pattern = Rex::Text.pattern_create(@opts[:length], @opts[:sets])
if query.length >= 8 && query.hex > 0
if (query.length >= 8 and query.hex > 0)
query = query.hex query = query.hex
# However, you can also specify a four-byte string # However, you can also specify a four-byte string
elsif (query.length == 4) elsif query.length == 4
query = query.unpack("V").first query = query.unpack("V").first
else else
# Or even a hex query that isn't 8 bytes long # Or even a hex query that isn't 8 bytes long
@ -131,7 +130,6 @@ class Driver
puts "[*] Exact match at offset #{offset}" puts "[*] Exact match at offset #{offset}"
offset = Rex::Text.pattern_offset(buffer, query, offset + 1) offset = Rex::Text.pattern_offset(buffer, query, offset + 1)
end end
end end
end end
end end
@ -140,7 +138,7 @@ if __FILE__ == $PROGRAM_NAME
driver = PatternOffset::Driver.new driver = PatternOffset::Driver.new
begin begin
driver.run driver.run
rescue ::Exception => e rescue ::StandardError => e
elog("#{e.class}: #{e.message}\n#{e.backtrace * "\n"}") elog("#{e.class}: #{e.message}\n#{e.backtrace * "\n"}")
$stderr.puts "[x] #{e.class}: #{e.message}" $stderr.puts "[x] #{e.class}: #{e.message}"
$stderr.puts "[*] If necessary, please refer to framework.log for more details." $stderr.puts "[*] If necessary, please refer to framework.log for more details."