def test_listshares_SMB2():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = ListSharesFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_store_long_filename_SMB1():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = False
factory = StoreFilesFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service_name = 'smbtest'
factory.filename = os.sep + 'StoreTest %d-%d.dat' % ( time.time(), random.randint(0, 10000) )
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_getAttributes_SMB1_test1():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = False
factory = GetAttributesFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.path = '/Test Folder with Long Name/'
factory.is_directory = True
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_echo():
info = getConnectionInfo()
factory = EchoFactory(info['user'],
info['password'],
info['client_name'],
info['server_name'],
use_ntlm_v2=True)
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_unicode_directory_SMB2():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = DirectoryFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service_name = 'smbtest'
factory.path = os.sep + u'文件夹创建 %d-%d' % ( time.time(), random.randint(0, 1000) )
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_listPath_SMB2():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = ListPathFactory(
info["user"], info["password"], info["client_name"], info["server_name"], use_ntlm_v2=True
)
reactor.connectTCP(info["server_ip"], info["server_port"], factory)
return factory.d
def test_listshares_SMB2():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = ListSnapshotsFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service_name = 'smbtest'
factory.path = '/rfc1001.txt'
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_listshares_SMB2():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = ListSnapshotsFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service_name = 'smbtest'
factory.path = '/rfc1001.txt'
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_getAttributes_SMB2_test3():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = GetAttributesFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.path = u'/\u6d4b\u8bd5\u6587\u4ef6\u5939'
factory.is_directory = True
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_getAttributes_SMB2_test2():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = GetAttributesFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.path = '/rfc1001.txt'
factory.is_directory = False
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_rename_unicode_directory_SMB2():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = RenameDirectoryFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service = 'smbtest'
factory.old_path = '/改名测试 %d-%d' % ( time.time(), random.randint(1000, 9999) )
factory.new_path = '/改名测试 %d-%d' % ( time.time(), random.randint(1000, 9999) )
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_NTLMv2_auth_SMB2():
def result(auth_passed):
assert auth_passed
smb_structs.SUPPORT_SMB2 = True
info = getConnectionInfo()
factory = AuthFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.d.addCallback(result)
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_rename_english_file_SMB2():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = RenameFileFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service = 'smbtest'
factory.old_path = '/RenameTest %d-%d.txt' % ( time.time(), random.randint(1000, 9999) )
factory.new_path = '/RenameTest %d-%d.txt' % ( time.time(), random.randint(1000, 9999) )
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_listshares_SMB2():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = ListSharesFactory(info['user'],
info['password'],
info['client_name'],
info['server_name'],
use_ntlm_v2=True)
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_retr_multiplereads_SMB2():
# Test file retrieval using multiple ReadAndx calls (assuming each call will not reach more than 65534 bytes)
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = RetrieveFileFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service = 'smbtest'
factory.path = '/rfc1001.txt'
factory.digest = '5367c2bbf97f521059c78eab65309ad3'
factory.filesize = 158437
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_retr_unicodefilename_SMB2():
# Test file retrieval that has a long non-English filename inside a folder with a non-English name
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = RetrieveFileFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service = 'smbtest'
factory.path = '/测试文件夹/垃圾文件.dat'
factory.digest = '8a44c1e80d55e91c92350955cdf83442'
factory.filesize = 256000
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_retr_longfilename_SMB1():
# Test file retrieval that has a long English filename
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = False
factory = RetrieveFileFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service = 'smbtest'
factory.path = '/Implementing CIFS - SMB.html'
factory.digest = '671c5700d279fcbbf958c1bba3c2639e'
factory.filesize = 421269
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_retr_offset_SMB2():
# Test file retrieval from offset to EOF
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = RetrieveFileFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service = 'smbtest'
factory.path = '/测试文件夹/垃圾文件.dat'
factory.digest = 'a141bd8024571ce7cb5c67f2b0d8ea0b'
factory.filesize = 156000
factory.offset = 100000
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_store_unicode_filename_SMB2():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = StoreFilesFactory(info['user'],
info['password'],
info['client_name'],
info['server_name'],
use_ntlm_v2=True)
factory.service_name = 'smbtest'
factory.filename = os.sep + u'上载测试 %d-%d.dat' % (time.time(),
random.randint(0, 10000))
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_retr_offset_and_smalllimit_SMB2():
# Test file retrieval from offset with a small max_length
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = RetrieveFileFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service = 'smbtest'
factory.path = '/测试文件夹/垃圾文件.dat'
factory.digest = '746f60a96b39b712a7b6e17ddde19986'
factory.filesize = 10
factory.offset = 100000
factory.max_length = 10
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_retr_offset_and_biglimit_SMB2():
# Test file retrieval from offset with a big max_length
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = RetrieveFileFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service = 'smbtest'
factory.path = '/测试文件夹/垃圾文件.dat'
factory.digest = '83b7afd7c92cdece3975338b5ca0b1c5'
factory.filesize = 100000
factory.offset = 100000
factory.max_length = 100000
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_retr_offset_and_zerolimit_SMB2():
# Test file retrieval from offset to EOF with max_length=0
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = RetrieveFileFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
factory.service = 'smbtest'
factory.path = '/测试文件夹/垃圾文件.dat'
factory.digest = 'd41d8cd98f00b204e9800998ecf8427e'
factory.filesize = 0
factory.offset = 100000
factory.max_length = 0
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_NTLMv2_auth_SMB2():
def result(auth_passed):
assert auth_passed
smb_structs.SUPPORT_SMB2 = True
info = getConnectionInfo()
factory = AuthFactory(info['user'],
info['password'],
info['client_name'],
info['server_name'],
use_ntlm_v2=True)
factory.d.addCallback(result)
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_rename_english_file_SMB1():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = False
factory = RenameFileFactory(info['user'],
info['password'],
info['client_name'],
info['server_name'],
use_ntlm_v2=True)
factory.service = 'smbtest'
factory.old_path = '/RenameTest %d-%d.txt' % (time.time(),
random.randint(1000, 9999))
factory.new_path = '/RenameTest %d-%d.txt' % (time.time(),
random.randint(1000, 9999))
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def test_rename_unicode_directory_SMB2():
info = getConnectionInfo()
smb_structs.SUPPORT_SMB2 = True
factory = RenameDirectoryFactory(info['user'],
info['password'],
info['client_name'],
info['server_name'],
use_ntlm_v2=True)
factory.service = 'smbtest'
factory.old_path = u'/改名测试 %d-%d' % (time.time(), random.randint(
1000, 9999))
factory.new_path = u'/改名测试 %d-%d' % (time.time(), random.randint(
1000, 9999))
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
def send_msg(host, port, msg, callback):
factory = TestClientFactory(msg, callback)
reactor.connectTCP(host, port, factory)
def test_echo():
info = getConnectionInfo()
factory = EchoFactory(info['user'], info['password'], info['client_name'], info['server_name'], use_ntlm_v2 = True)
reactor.connectTCP(info['server_ip'], info['server_port'], factory)
return factory.d
