upstream/mercurial-mirror Files · hgext/narrow/narrowrepo.py

dirs: resolve fuzzer OOM situation by disallowing deep directory hierarchies...

dirs: resolve fuzzer OOM situation by disallowing deep directory hierarchies It seems like 2048 directories ought to be enough for any reasonable use of Mercurial? A previous version of this patch scanned for slashes before any allocations occurred. That approach is slower than this in the happy path, but much faster than this in the case that too many slashes are encountered. We may want to revisit it in the future using memchr() so it'll be well-optimized by the libc we're using. .. bc: Mercurial will now defend against OOMs by refusing to operate on paths with 2048 or more components. This means that _extremely_ deep path hierarchies will be rejected, but we anticipate nobody is using hierarchies this deep. Differential Revision: https://phab.mercurial-scm.org/D7411

Augie Fackler - - Load All Authors

File last commit:

r43346:2372284d default


                r44057:0796e266

default

Download file

             narrowrepo.py
        
                    29 lines
            
             | 909 B
            
                | text/x-python
            
             |
                PythonLexer
            
             / hgext / narrow / narrowrepo.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # narrowrepo.py - repository which supports narrow revlogs, lazy loading

      #

      # Copyright 2017 Google, Inc.

      #

      # This software may be used and distributed according to the terms of the

      # GNU General Public License version 2 or any later version.

      from __future__ import absolute_import

      from mercurial import wireprototypes

      from . import narrowdirstate

      def wraprepo(repo):

          """Enables narrow clone functionality on a single local repository."""

          class narrowrepository(repo.__class__):

              def _makedirstate(self):

                  dirstate = super(narrowrepository, self)._makedirstate()

                  return narrowdirstate.wrapdirstate(self, dirstate)

              def peer(self):

                  peer = super(narrowrepository, self).peer()

                  peer._caps.add(wireprototypes.NARROWCAP)

                  peer._caps.add(wireprototypes.ELLIPSESCAP)

                  return peer

          repo.__class__ = narrowrepository

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				# narrowrepo.py - repository which supports narrow revlogs, lazy loading
				#
				# Copyright 2017 Google, Inc.
				#
				# This software may be used and distributed according to the terms of the
				# GNU General Public License version 2 or any later version.

				from __future__ import absolute_import

				from mercurial import wireprototypes

				from . import narrowdirstate


				def wraprepo(repo):
				"""Enables narrow clone functionality on a single local repository."""

				class narrowrepository(repo.__class__):
				def _makedirstate(self):
				dirstate = super(narrowrepository, self)._makedirstate()
				return narrowdirstate.wrapdirstate(self, dirstate)

				def peer(self):
				peer = super(narrowrepository, self).peer()
				peer._caps.add(wireprototypes.NARROWCAP)
				peer._caps.add(wireprototypes.ELLIPSESCAP)
				return peer

				repo.__class__ = narrowrepository