29 lines
638 B
Ruby
29 lines
638 B
Ruby
|
#!/usr/bin/ruby
|
||
|
|
|
||
|
|
$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
|
||
|
|
|
||
|
|
require 'test/unit'
|
||
|
|
require 'rex/proto/smb/utils'
|
||
|
|
|
||
|
|
class Rex::Proto::SMB::Utils::UnitTest < Test::Unit::TestCase
|
||
|
|
|
||
|
|
Klass = Rex::Proto::SMB::Utils
|
||
|
|
|
||
|
|
def test_nbname
|
||
|
|
|
||
|
|
nbdecoded = 'METASPLOITROCKS!'
|
||
|
|
nbencoded = 'ENEFFEEBFDFAEMEPEJFEFCEPEDELFDCB'
|
||
|
|
|
||
|
|
assert_equal(Klass.nbname_encode(nbdecoded), nbencoded )
|
||
|
|
assert_equal(Klass.nbname_decode(nbencoded), nbdecoded )
|
||
|
|
end
|
||
|
|
|
||
|
|
def test_unicode
|
||
|
|
plain = 'Metasploit!'
|
||
|
|
unicode = "M\x00e\x00t\x00a\x00s\x00p\x00l\x00\o\x00i\x00t\x00!\x00"
|
||
|
|
|
||
|
|
assert_equal(Klass.unicode(plain), unicode)
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|