45 lines
1.8 KiB
Ruby
45 lines
1.8 KiB
Ruby
|
#!/usr/bin/env ruby
|
||
|
|
|
||
|
|
$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
|
||
|
|
|
||
|
|
require 'rex/test'
|
||
|
|
require 'rex/exceptions'
|
||
|
|
require 'rex/encoder/ndr'
|
||
|
|
|
||
|
|
class Rex::Encoder::NDR::UnitTest < Test::Unit::TestCase
|
||
|
|
|
||
|
|
Klass = Rex::Encoder::NDR
|
||
|
|
|
||
|
|
def test_align
|
||
|
|
assert_equal(0, Klass.align('').length, 'align 0')
|
||
|
|
assert_equal(3, Klass.align('f').length, 'align 1')
|
||
|
|
assert_equal(2, Klass.align('fo').length, 'align 2')
|
||
|
|
assert_equal(1, Klass.align('foo').length, 'align 3')
|
||
|
|
assert_equal(0, Klass.align('fooo').length, 'align 4')
|
||
|
|
assert_equal(3, Klass.align('foooo').length, 'align 5')
|
||
|
|
end
|
||
|
|
|
||
|
|
def test_numbers
|
||
|
|
assert_equal("\x0a\x00\x00\x00", Klass.long(10), 'long')
|
||
|
|
assert_equal("\x0a\x00", Klass.short(10), 'short')
|
||
|
|
assert_equal("\x0a", Klass.byte(10), 'byte')
|
||
|
|
end
|
||
|
|
|
||
|
|
def test_conformant_array
|
||
|
|
assert_equal("\x05\x00\x00\x00aaaaa", Klass.UniConformantArray('aaaaa').slice(0,9), 'UniConformantArray')
|
||
|
|
assert_equal(12, Klass.UniConformantArray('aaaaa').length, 'UniConformantArray length')
|
||
|
|
end
|
||
|
|
|
||
|
|
def test_string
|
||
|
|
assert_equal("\x06\x00\x00\x00" + "\x00\x00\x00\x00" + "\x06\x00\x00\x00" "aaaaa\x00", Klass.string('aaaaa').slice(0,4+4+4+6), 'string')
|
||
|
|
assert_equal(20, Klass.string('aaaaa').length, 'string length')
|
||
|
|
|
||
|
|
assert_equal("\x06\x00\x00\x00" + "\x00\x00\x00\x00" + "\x06\x00\x00\x00" "a\x00a\x00a\x00a\x00a\x00\x00\x00", Klass.wstring('aaaaa').slice(0,4+4+4+12), 'wstring')
|
||
|
|
assert_equal(24, Klass.wstring('aaaaa').length, 'wstring length')
|
||
|
|
|
||
|
|
assert_equal("\x02\x00\x00\x00" + "\x00\x00\x00\x00" + "\x02\x00\x00\x00" "aa\x00\x00", Klass.wstring_prebuilt('aa' + "\x00\x00"), 'wstring_prebuilt')
|
||
|
|
assert_equal(16, Klass.wstring_prebuilt('aa' + "\x00\x00").length, 'wstring_prebuilt length')
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|