1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
#!/usr/bin/env python
# Impacket - Collection of Python classes for working with network protocols.
#
# Copyright Fortra, LLC and its affiliated companies
#
# All rights reserved.
#
# This software is provided under a slightly modified version
# of the Apache Software License. See the accompanying LICENSE file
# for more information.
#
# Description:
# Utility and helper functions for the example scripts
#
import unittest
from impacket.examples.utils import parse_target, parse_credentials
class UtilsTests(unittest.TestCase):
def test_parse_target(self):
# Parse target returns a tuple with: domain, username, password, remote_name/address
targets = {
"": ("", "", "", ""),
"HostName": ("", "", "", "HostName"),
"UserName@HostName": ("", "UserName", "", "HostName"),
"UserName:Password@HostName": ("", "UserName", "Password", "HostName"),
"UserName:Pa$$word1234@HostName": ("", "UserName", "Pa$$word1234", "HostName"),
"UserName:Password!#$@HostName": ("", "UserName", "Password!#$", "HostName"),
"UserName:Passw@rd!#$@HostName": ("", "UserName", "Passw@rd!#$", "HostName"),
"UserName:P@ssw@rd@!#$@HostName": ("", "UserName", "P@ssw@rd@!#$", "HostName"),
"DOMAIN/UserName@HostName": ("DOMAIN", "UserName", "", "HostName"),
"DOMAIN/:Password@HostName": ("DOMAIN", "", "Password", "HostName"),
"DOMAIN/UserName:Password@HostName": ("DOMAIN", "UserName", "Password", "HostName"),
"DOMAIN/UserName:Password/123@HostName": ("DOMAIN", "UserName", "Password/123", "HostName"),
}
for target, result in targets.items():
self.assertTupleEqual(parse_target(target), result)
def test_parse_credentials(self):
# Parse credentials returns a tuple with: domain, username, password
creds = {
"": ("", "", ""),
"UserName": ("", "UserName", ""),
"UserName:Password": ("", "UserName", "Password"),
"UserName:Password:123": ("", "UserName", "Password:123"),
"DOMAIN/UserName": ("DOMAIN", "UserName", ""),
"DOMAIN/UserName:Password": ("DOMAIN", "UserName", "Password"),
"DOMAIN/UserName:Password/123": ("DOMAIN", "UserName", "Password/123"),
}
for cred, result in creds.items():
self.assertTupleEqual(parse_credentials(cred), result)
if __name__ == "__main__":
unittest.main(verbosity=1)
|
