comparison hgext/largefiles/lfutil.py @ 44062:2d49482d0dd4

hgext: replace references to hashlib.sha1 with hashutil.sha1

When in a non-pure build of Mercurial, this will provide protections
against SHA1 collision attacks.

Differential Revision: https://phab.mercurial-scm.org/D7851

44061:cbc5755df6bf

'''largefiles utility code: must not import other modules in this package.'''
from __future__ import absolute_import

import contextlib
import copy
import hashlib
import os
import stat

from mercurial.i18n import _
from mercurial.node import hex

    scmutil,
    sparse,
    util,
    vfs as vfsmod,
)

shortname = b'.hglf'
shortnameslash = shortname + b'/'
longname = b'largefiles'

def copyandhash(instream, outfile):
    '''Read bytes from instream (iterable) and write them to outfile,
    computing the SHA-1 hash of the data along the way. Return the hash.'''
    hasher = hashlib.sha1(b'')
    for data in instream:
        hasher.update(data)
        outfile.write(data)
    return hex(hasher.digest())

def hexsha1(fileobj):
    """hexsha1 returns the hex-encoded sha1 sum of the data in the file-like
    object data"""
    h = hashlib.sha1()
    for chunk in util.filechunkiter(fileobj):
        h.update(chunk)
    return hex(h.digest())

44062:2d49482d0dd4

'''largefiles utility code: must not import other modules in this package.'''
from __future__ import absolute_import

import contextlib
import copy
import os
import stat

from mercurial.i18n import _
from mercurial.node import hex

    scmutil,
    sparse,
    util,
    vfs as vfsmod,
)
from mercurial.utils import hashutil

shortname = b'.hglf'
shortnameslash = shortname + b'/'
longname = b'largefiles'

def copyandhash(instream, outfile):
    '''Read bytes from instream (iterable) and write them to outfile,
    computing the SHA-1 hash of the data along the way. Return the hash.'''
    hasher = hashutil.sha1(b'')
    for data in instream:
        hasher.update(data)
        outfile.write(data)
    return hex(hasher.digest())

def hexsha1(fileobj):
    """hexsha1 returns the hex-encoded sha1 sum of the data in the file-like
    object data"""
    h = hashutil.sha1()
    for chunk in util.filechunkiter(fileobj):
        h.update(chunk)
    return hex(h.digest())