30 lines
507 B
Ruby
30 lines
507 B
Ruby
#!/usr/bin/env ruby
|
|
|
|
$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
|
|
|
|
require 'test/unit'
|
|
require 'rex'
|
|
require 'msf/core'
|
|
require 'msf/core/exploit/seh'
|
|
|
|
module Msf
|
|
|
|
class Exploit::Seh::UnitTest < Test::Unit::TestCase
|
|
|
|
class Stub < Msf::Exploit::Remote
|
|
include Msf::Exploit::Seh
|
|
end
|
|
|
|
def test_seh
|
|
e = Stub.new
|
|
r = e.generate_seh_record(0x41414141,
|
|
'EvasionLevel' => EVASION_NORMAL)
|
|
|
|
assert_equal("\xeb\x06", r[0, 2])
|
|
assert_equal("\x41\x41\x41\x41", r[4, 4])
|
|
end
|
|
|
|
end
|
|
|
|
end
|