# HG changeset patch # User Boris Feld # Date 2018-01-24 20:44:31 # Node ID 3ad3aaeb11340cb03f213f6c17e86102bd24fa3e # Parent bbc07357b56714617a2b2313f6f7f75028d7da5e streamclone: use readexactly when reading stream v2 Yuya Nishihara pointed out that it is safer. diff --git a/mercurial/streamclone.py b/mercurial/streamclone.py --- a/mercurial/streamclone.py +++ b/mercurial/streamclone.py @@ -594,12 +594,12 @@ def consumev2(repo, fp, filecount, files for vfs in vfsmap.values()) with nested(*ctxs): for i in range(filecount): - src = fp.read(1) + src = util.readexactly(fp, 1) vfs = vfsmap[src] namelen = util.uvarintdecodestream(fp) datalen = util.uvarintdecodestream(fp) - name = fp.read(namelen) + name = util.readexactly(fp, namelen) if repo.ui.debugflag: repo.ui.debug('adding [%s] %s (%s)\n' %