mirror of
https://github.com/ansible-collections/community.general.git
synced 2025-05-15 05:29:11 -07:00
Hashing needs byte strings in python3 (#17221)
First try at porting this passed in string-types as that worked on python2. Python3 is more strict so be explicit about converting from text to bytes
This commit is contained in:
parent
e098c5ef82
commit
300d6482d1
1 changed files with 8 additions and 8 deletions
|
@ -38,22 +38,19 @@ except ImportError:
|
||||||
# Assume we're running in FIPS mode here
|
# Assume we're running in FIPS mode here
|
||||||
_md5 = None
|
_md5 = None
|
||||||
|
|
||||||
from ansible.compat.six import string_types
|
|
||||||
from ansible.errors import AnsibleError
|
from ansible.errors import AnsibleError
|
||||||
from ansible.utils.unicode import to_bytes
|
from ansible.utils.unicode import to_bytes
|
||||||
|
|
||||||
|
|
||||||
def secure_hash_s(data, hash_func=sha1):
|
def secure_hash_s(data, hash_func=sha1):
|
||||||
''' Return a secure hash hex digest of data. '''
|
''' Return a secure hash hex digest of data. '''
|
||||||
|
|
||||||
digest = hash_func()
|
digest = hash_func()
|
||||||
try:
|
data = to_bytes(data, errors='strict')
|
||||||
if not isinstance(data, string_types):
|
digest.update(data)
|
||||||
data = "%s" % data
|
|
||||||
digest.update(data)
|
|
||||||
except UnicodeEncodeError:
|
|
||||||
digest.update(data.encode('utf-8'))
|
|
||||||
return digest.hexdigest()
|
return digest.hexdigest()
|
||||||
|
|
||||||
|
|
||||||
def secure_hash(filename, hash_func=sha1):
|
def secure_hash(filename, hash_func=sha1):
|
||||||
''' Return a secure hash hex digest of local file, None if file is not present or a directory. '''
|
''' Return a secure hash hex digest of local file, None if file is not present or a directory. '''
|
||||||
|
|
||||||
|
@ -76,6 +73,8 @@ def secure_hash(filename, hash_func=sha1):
|
||||||
checksum = secure_hash
|
checksum = secure_hash
|
||||||
checksum_s = secure_hash_s
|
checksum_s = secure_hash_s
|
||||||
|
|
||||||
|
|
||||||
|
#
|
||||||
# Backwards compat functions. Some modules include md5s in their return values
|
# Backwards compat functions. Some modules include md5s in their return values
|
||||||
# Continue to support that for now. As of ansible-1.8, all of those modules
|
# Continue to support that for now. As of ansible-1.8, all of those modules
|
||||||
# should also return "checksum" (sha1 for now)
|
# should also return "checksum" (sha1 for now)
|
||||||
|
@ -84,14 +83,15 @@ checksum_s = secure_hash_s
|
||||||
# 2) Compliance with a third party protocol
|
# 2) Compliance with a third party protocol
|
||||||
#
|
#
|
||||||
# MD5 will not work on systems which are FIPS-140-2 compliant.
|
# MD5 will not work on systems which are FIPS-140-2 compliant.
|
||||||
|
#
|
||||||
|
|
||||||
def md5s(data):
|
def md5s(data):
|
||||||
if not _md5:
|
if not _md5:
|
||||||
raise ValueError('MD5 not available. Possibly running in FIPS mode')
|
raise ValueError('MD5 not available. Possibly running in FIPS mode')
|
||||||
return secure_hash_s(data, _md5)
|
return secure_hash_s(data, _md5)
|
||||||
|
|
||||||
|
|
||||||
def md5(filename):
|
def md5(filename):
|
||||||
if not _md5:
|
if not _md5:
|
||||||
raise ValueError('MD5 not available. Possibly running in FIPS mode')
|
raise ValueError('MD5 not available. Possibly running in FIPS mode')
|
||||||
return secure_hash(filename, _md5)
|
return secure_hash(filename, _md5)
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue