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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
#!/usr/bin/python3
import unittest
import warnings
import apt_pkg
import testcommon
class testHashes(testcommon.TestCase):
"test the hashsum functions against strings and files"
DATA_PATH = "data/hashsums/hashsum_test.data"
DATA_WITH_ZERO_PATH = "data/hashsums/hashsum_test_with_zero.data"
def setUp(self):
testcommon.TestCase.setUp(self)
warnings.filterwarnings("ignore", category=DeprecationWarning)
def tearDown(self):
testcommon.TestCase.tearDown(self)
warnings.resetwarnings()
def testMD5(self):
# simple
s = b"foo"
s_md5 = "acbd18db4cc2f85cedef654fccc4a4d8"
res = apt_pkg.md5sum(s)
self.assertEqual(res, s_md5)
# file
with open(self.DATA_PATH) as fobj:
self.assertEqual(apt_pkg.md5sum(fobj), s_md5)
# with zero (\0) in the string
s = b"foo\0bar"
s_md5 = "f6f5f8cd0cb63668898ba29025ae824e"
res = apt_pkg.md5sum(s)
self.assertEqual(res, s_md5)
# file
with open(self.DATA_WITH_ZERO_PATH) as fobj:
self.assertEqual(apt_pkg.md5sum(fobj), s_md5)
def testSHA1(self):
# simple
s = b"foo"
s_hash = "0beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a33"
res = apt_pkg.sha1sum(s)
self.assertEqual(res, s_hash)
# file
with open(self.DATA_PATH) as fobj:
self.assertEqual(apt_pkg.sha1sum(fobj), s_hash)
# with zero (\0) in the string
s = b"foo\0bar"
s_hash = "e2c300a39311a2dfcaff799528415cb74c19317f"
res = apt_pkg.sha1sum(s)
self.assertEqual(res, s_hash)
# file
with open(self.DATA_WITH_ZERO_PATH) as fobj:
self.assertEqual(apt_pkg.sha1sum(fobj), s_hash)
def testSHA256(self):
# simple
s = b"foo"
s_hash = "2c26b46b68ffc68ff99b453c1d30413413422d706483bfa0f98a5e886266e7ae"
res = apt_pkg.sha256sum(s)
self.assertEqual(res, s_hash)
# file
with open(self.DATA_PATH) as fobj:
self.assertEqual(apt_pkg.sha256sum(fobj), s_hash)
# with zero (\0) in the string
s = b"foo\0bar"
s_hash = "d6b681bfce7155d44721afb79c296ef4f0fa80a9dd6b43c5cf74dd0f64c85512"
res = apt_pkg.sha256sum(s)
self.assertEqual(res, s_hash)
# file
with open(self.DATA_WITH_ZERO_PATH) as fobj:
self.assertEqual(apt_pkg.sha256sum(fobj), s_hash)
def testSHA512(self):
# simple
s = b"foo"
s_hash = (
"f7fbba6e0636f890e56fbbf3283e524c6fa3204ae298382d624741d"
"0dc6638326e282c41be5e4254d8820772c5518a2c5a8c0c7f7eda19"
"594a7eb539453e1ed7"
)
res = apt_pkg.sha512sum(s)
self.assertEqual(res, s_hash)
# file
with open(self.DATA_PATH) as fobj:
self.assertEqual(apt_pkg.sha512sum(fobj), s_hash)
# with zero (\0) in the string
s = b"foo\0bar"
s_hash = (
"8c5e791db8f6bfb40eba884f70c9ac52231f01a393e4e55b4576d45"
"9a827f34f77e41e7fac806724517b9e96bb42387c5f9bbf325d2f99"
"ed52a4aa6abebc3350"
)
res = apt_pkg.sha512sum(s)
self.assertEqual(res, s_hash)
# file
with open(self.DATA_WITH_ZERO_PATH) as fobj:
self.assertEqual(apt_pkg.sha512sum(fobj), s_hash)
if __name__ == "__main__":
unittest.main()
|
