View file File name : hashlib.py Content :# Copyright 2016 Donald Stufft and individual contributors # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. from __future__ import absolute_import, division, print_function import binascii import nacl.bindings from nacl.utils import bytes_as_string BYTES = nacl.bindings.crypto_generichash_BYTES BYTES_MIN = nacl.bindings.crypto_generichash_BYTES_MIN BYTES_MAX = nacl.bindings.crypto_generichash_BYTES_MAX KEYBYTES = nacl.bindings.crypto_generichash_KEYBYTES KEYBYTES_MIN = nacl.bindings.crypto_generichash_KEYBYTES_MIN KEYBYTES_MAX = nacl.bindings.crypto_generichash_KEYBYTES_MAX SALTBYTES = nacl.bindings.crypto_generichash_SALTBYTES PERSONALBYTES = nacl.bindings.crypto_generichash_PERSONALBYTES _b2b_init = nacl.bindings.crypto_generichash_blake2b_init _b2b_final = nacl.bindings.crypto_generichash_blake2b_final _b2b_copy = nacl.bindings.crypto_generichash_blake2b_state_copy _b2b_update = nacl.bindings.crypto_generichash_blake2b_update class blake2b(object): """ :py:mod:`hashlib` API compatible blake2b algorithm implementation """ MAX_DIGEST_SIZE = BYTES MAX_KEY_SIZE = KEYBYTES_MAX PERSON_SIZE = PERSONALBYTES SALT_SIZE = SALTBYTES def __init__(self, data=b'', digest_size=BYTES, key=b'', salt=b'', person=b''): """ :py:class:`.blake2b` algorithm initializer :param data: :type data: bytes :param int digest_size: the requested digest size; must be at most :py:attr:`.MAX_DIGEST_SIZE`; the default digest size is :py:data:`.BYTES` :param key: the key to be set for keyed MAC/PRF usage; if set, the key must be at most :py:data:`.KEYBYTES_MAX` long :type key: bytes :param salt: an initialization salt at most :py:attr:`.SALT_SIZE` long; it will be zero-padded if needed :type salt: bytes :param person: a personalization string at most :py:attr:`.PERSONAL_SIZE` long; it will be zero-padded if needed :type person: bytes """ self._state = _b2b_init(key=key, salt=salt, person=person, digest_size=digest_size) self._digest_size = digest_size if data: self.update(data) @property def digest_size(self): return self._digest_size @property def block_size(self): return 128 @property def name(self): return 'blake2b' def update(self, data): _b2b_update(self._state, data) def digest(self): _st = nacl.bindings.crypto_generichash_blake2b_state_copy(self._state) return _b2b_final(_st, self.digest_size) def hexdigest(self): return bytes_as_string(binascii.hexlify(self.digest())) def copy(self): _cp = type(self)(digest_size=self.digest_size) _st = _b2b_copy(self._state) _cp._state = _st return _cp def scrypt(password, salt='', n=2**20, r=8, p=1, maxmem=2**25, dklen=64): """ Derive a cryptographic key using the scrypt KDF. Implements the same signature as the ``hashlib.scrypt`` implemented in cpython version 3.6 """ return nacl.bindings.crypto_pwhash_scryptsalsa208sha256_ll( password, salt, n, r, p, maxmem=maxmem, dklen=dklen)