diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/cryptography/hazmat/primitives/twofactor/hotp.py | 6 | ||||
-rw-r--r-- | src/cryptography/hazmat/primitives/twofactor/totp.py | 6 | ||||
-rw-r--r-- | src/cryptography/hazmat/primitives/twofactor/utils.py | 4 |
3 files changed, 8 insertions, 8 deletions
diff --git a/src/cryptography/hazmat/primitives/twofactor/hotp.py b/src/cryptography/hazmat/primitives/twofactor/hotp.py index f59f551c..8c0cec14 100644 --- a/src/cryptography/hazmat/primitives/twofactor/hotp.py +++ b/src/cryptography/hazmat/primitives/twofactor/hotp.py @@ -15,7 +15,7 @@ from cryptography.hazmat.backends.interfaces import HMACBackend from cryptography.hazmat.primitives import constant_time, hmac from cryptography.hazmat.primitives.hashes import SHA1, SHA256, SHA512 from cryptography.hazmat.primitives.twofactor import InvalidToken -from cryptography.hazmat.primitives.twofactor.utils import generate_uri +from cryptography.hazmat.primitives.twofactor.utils import _generate_uri class HOTP(object): @@ -61,7 +61,7 @@ class HOTP(object): p = hmac_value[offset:offset + 4] return struct.unpack(">I", p)[0] & 0x7fffffff - def get_provisioning_uri(self, account_name, counter, issuer=None): - return generate_uri(self, 'hotp', account_name, issuer, [ + def get_provisioning_uri(self, account_name, counter, issuer): + return _generate_uri(self, 'hotp', account_name, issuer, [ ('counter', int(counter)), ]) diff --git a/src/cryptography/hazmat/primitives/twofactor/totp.py b/src/cryptography/hazmat/primitives/twofactor/totp.py index 9c8eddad..98493b6d 100644 --- a/src/cryptography/hazmat/primitives/twofactor/totp.py +++ b/src/cryptography/hazmat/primitives/twofactor/totp.py @@ -11,7 +11,7 @@ from cryptography.hazmat.backends.interfaces import HMACBackend from cryptography.hazmat.primitives import constant_time from cryptography.hazmat.primitives.twofactor import InvalidToken from cryptography.hazmat.primitives.twofactor.hotp import HOTP -from cryptography.hazmat.primitives.twofactor.utils import generate_uri +from cryptography.hazmat.primitives.twofactor.utils import _generate_uri class TOTP(object): @@ -33,7 +33,7 @@ class TOTP(object): if not constant_time.bytes_eq(self.generate(time), totp): raise InvalidToken("Supplied TOTP value does not match.") - def get_provisioning_uri(self, account_name, issuer=None): - return generate_uri(self._hotp, 'totp', account_name, issuer, [ + def get_provisioning_uri(self, account_name, issuer): + return _generate_uri(self._hotp, 'totp', account_name, issuer, [ ('period', int(self._time_step)), ]) diff --git a/src/cryptography/hazmat/primitives/twofactor/utils.py b/src/cryptography/hazmat/primitives/twofactor/utils.py index 89d38ff2..ae32058b 100644 --- a/src/cryptography/hazmat/primitives/twofactor/utils.py +++ b/src/cryptography/hazmat/primitives/twofactor/utils.py @@ -1,11 +1,11 @@ -from __future__ import unicode_literals +from __future__ import absolute_import, division, print_function import base64 from six.moves.urllib.parse import quote, urlencode -def generate_uri(hotp, type_name, account_name, issuer, extra_parameters): +def _generate_uri(hotp, type_name, account_name, issuer, extra_parameters): parameters = [ ('digits', hotp._length), ('secret', base64.b32encode(hotp._key)), |