44 lines
912 B
Ruby
44 lines
912 B
Ruby
#!/usr/bin/env ruby
|
|
|
|
$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..', 'lib'))
|
|
$:.unshift(File.dirname(__FILE__))
|
|
|
|
require 'test/unit'
|
|
require 'rex/ui'
|
|
require 'msf/core'
|
|
require 'countdown'
|
|
|
|
class Msf::Encoders::X86::Countdown::UnitTest < Test::Unit::TestCase
|
|
|
|
Klass = Msf::Encoders::X86::Countdown
|
|
|
|
def test_encoder
|
|
|
|
k = Klass.new
|
|
|
|
{
|
|
"\xcc\xcc\xcc\xcc" =>
|
|
[
|
|
"\x6a\x03\x59\xe8\xff\xff\xff\xff\xc1\x5e\x30\x4c\x0e\x07" +
|
|
"\xe2\xfa\xcd\xce\xcf\xc8",
|
|
4
|
|
],
|
|
"ABCDEFGHIJKLMNOPQRSTUVWXYZ" =>
|
|
[
|
|
"\x6a\x19\x59\xe8\xff\xff\xff\xff\xc1\x5e\x30\x4c\x0e\x07" +
|
|
"\xe2\xfa\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40" +
|
|
"\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40",
|
|
4
|
|
]
|
|
}.each_pair { |raw, real|
|
|
offset = real[1] || 0
|
|
|
|
encoded = k.encode(raw, '')
|
|
|
|
assert_equal(real[0][offset, -1], encoded[offset, -1])
|
|
}
|
|
|
|
end
|
|
|
|
end
|