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
|
# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
# For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE
# Copyright (c) https://github.com/pylint-dev/astroid/blob/main/CONTRIBUTORS.txt
from astroid.brain.helpers import register_module_extender
from astroid.builder import parse
from astroid.const import PY39_PLUS
from astroid.manager import AstroidManager
def _hashlib_transform():
maybe_usedforsecurity = ", usedforsecurity=True" if PY39_PLUS else ""
init_signature = f"value=''{maybe_usedforsecurity}"
digest_signature = "self"
shake_digest_signature = "self, length"
template = """
class %(name)s:
def __init__(self, %(init_signature)s): pass
def digest(%(digest_signature)s):
return %(digest)s
def copy(self):
return self
def update(self, value): pass
def hexdigest(%(digest_signature)s):
return ''
@property
def name(self):
return %(name)r
@property
def block_size(self):
return 1
@property
def digest_size(self):
return 1
"""
algorithms_with_signature = dict.fromkeys(
[
"md5",
"sha1",
"sha224",
"sha256",
"sha384",
"sha512",
"sha3_224",
"sha3_256",
"sha3_384",
"sha3_512",
],
(init_signature, digest_signature),
)
blake2b_signature = (
"data=b'', *, digest_size=64, key=b'', salt=b'', "
"person=b'', fanout=1, depth=1, leaf_size=0, node_offset=0, "
f"node_depth=0, inner_size=0, last_node=False{maybe_usedforsecurity}"
)
blake2s_signature = (
"data=b'', *, digest_size=32, key=b'', salt=b'', "
"person=b'', fanout=1, depth=1, leaf_size=0, node_offset=0, "
f"node_depth=0, inner_size=0, last_node=False{maybe_usedforsecurity}"
)
shake_algorithms = dict.fromkeys(
["shake_128", "shake_256"],
(init_signature, shake_digest_signature),
)
algorithms_with_signature.update(shake_algorithms)
algorithms_with_signature.update(
{
"blake2b": (blake2b_signature, digest_signature),
"blake2s": (blake2s_signature, digest_signature),
}
)
classes = "".join(
template
% {
"name": hashfunc,
"digest": 'b""',
"init_signature": init_signature,
"digest_signature": digest_signature,
}
for hashfunc, (
init_signature,
digest_signature,
) in algorithms_with_signature.items()
)
return parse(classes)
register_module_extender(AstroidManager(), "hashlib", _hashlib_transform)
|
