upstream/mercurial-mirror Files · tests/test-context.py

dispatch: protect against malicious 'hg serve --stdio' invocations (sec)...

dispatch: protect against malicious 'hg serve --stdio' invocations (sec) Some shared-ssh installations assume that 'hg serve --stdio' is a safe command to run for minimally trusted users. Unfortunately, the messy implementation of argument parsing here meant that trying to access a repo named '--debugger' would give the user a pdb prompt, thereby sidestepping any hoped-for sandboxing. Serving repositories over HTTP(S) is unaffected. We're not currently hardening any subcommands other than 'serve'. If your service exposes other commands to users with arbitrary repository names, it is imperative that you defend against repository names of '--debugger' and anything starting with '--config'. The read-only mode of hg-ssh stopped working because it provided its hook configuration to "hg serve --stdio" via --config parameter. This is banned for security reasons now. This patch switches it to directly call ui.setconfig(). If your custom hosting infrastructure relies on passing --config to "hg serve --stdio", you'll need to find a different way to get that configuration into Mercurial, either by using ui.setconfig() as hg-ssh does in this patch, or by placing an hgrc file someplace where Mercurial will read it. mitrandir@fb.com provided some extra fixes for the dispatch code and for hg-ssh in places that I overlooked.

Denis Laxalde - - Load All Authors

File last commit:

r31270:59aec562 default


                r32050:77eaf953

4.1.3 stable

Download file

             test-context.py
        
                    148 lines
            
             | 4.3 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / tests / test-context.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      from __future__ import absolute_import, print_function

      import os

      from mercurial import (

          context,

          encoding,

          hg,

          ui as uimod,

      )

      u = uimod.ui.load()

      repo = hg.repository(u, 'test1', create=1)

      os.chdir('test1')

      # create 'foo' with fixed time stamp

      f = open('foo', 'wb')

      f.write(b'foo\n')

      f.close()

      os.utime('foo', (1000, 1000))

      # add+commit 'foo'

      repo[None].add(['foo'])

      repo.commit(text='commit1', date="0 0")

      if os.name == 'nt':

          d = repo[None]['foo'].date()

          print("workingfilectx.date = (%d, %d)" % (d[0], d[1]))

      else:

          print("workingfilectx.date =", repo[None]['foo'].date())

      # test memctx with non-ASCII commit message

      def filectxfn(repo, memctx, path):

          return context.memfilectx(repo, "foo", "")

      ctx = context.memctx(repo, ['tip', None],

                           encoding.tolocal("Gr\xc3\xbcezi!"),

                           ["foo"], filectxfn)

      ctx.commit()

      for enc in "ASCII", "Latin-1", "UTF-8":

          encoding.encoding = enc

          print("%-8s: %s" % (enc, repo["tip"].description()))

      # test performing a status

      def getfilectx(repo, memctx, f):

          fctx = memctx.parents()[0][f]

          data, flags = fctx.data(), fctx.flags()

          if f == 'foo':

              data += 'bar\n'

          return context.memfilectx(repo, f, data, 'l' in flags, 'x' in flags)

      ctxa = repo.changectx(0)

      ctxb = context.memctx(repo, [ctxa.node(), None], "test diff", ["foo"],

                            getfilectx, ctxa.user(), ctxa.date())

      print(ctxb.status(ctxa))

      # test performing a diff on a memctx

      for d in ctxb.diff(ctxa, git=True):

          print(d)

      # test safeness and correctness of "ctx.status()"

      print('= checking context.status():')

      # ancestor "wcctx ~ 2"

      actx2 = repo['.']

      repo.wwrite('bar-m', 'bar-m\n', '')

      repo.wwrite('bar-r', 'bar-r\n', '')

      repo[None].add(['bar-m', 'bar-r'])

      repo.commit(text='add bar-m, bar-r', date="0 0")

      # ancestor "wcctx ~ 1"

      actx1 = repo['.']

      repo.wwrite('bar-m', 'bar-m bar-m\n', '')

      repo.wwrite('bar-a', 'bar-a\n', '')

      repo[None].add(['bar-a'])

      repo[None].forget(['bar-r'])

      # status at this point:

      #   M bar-m

      #   A bar-a

      #   R bar-r

      #   C foo

      from mercurial import scmutil

      print('== checking workingctx.status:')

      wctx = repo[None]

      print('wctx._status=%s' % (str(wctx._status)))

      print('=== with "pattern match":')

      print(actx1.status(other=wctx,

                         match=scmutil.matchfiles(repo, ['bar-m', 'foo'])))

      print('wctx._status=%s' % (str(wctx._status)))

      print(actx2.status(other=wctx,

                         match=scmutil.matchfiles(repo, ['bar-m', 'foo'])))

      print('wctx._status=%s' % (str(wctx._status)))

      print('=== with "always match" and "listclean=True":')

      print(actx1.status(other=wctx, listclean=True))

      print('wctx._status=%s' % (str(wctx._status)))

      print(actx2.status(other=wctx, listclean=True))

      print('wctx._status=%s' % (str(wctx._status)))

      print("== checking workingcommitctx.status:")

      wcctx = context.workingcommitctx(repo,

                                       scmutil.status(['bar-m'],

                                                      ['bar-a'],

                                                      [],

                                                      [], [], [], []),

                                       text='', date='0 0')

      print('wcctx._status=%s' % (str(wcctx._status)))

      print('=== with "always match":')

      print(actx1.status(other=wcctx))

      print('wcctx._status=%s' % (str(wcctx._status)))

      print(actx2.status(other=wcctx))

      print('wcctx._status=%s' % (str(wcctx._status)))

      print('=== with "always match" and "listclean=True":')

      print(actx1.status(other=wcctx, listclean=True))

      print('wcctx._status=%s' % (str(wcctx._status)))

      print(actx2.status(other=wcctx, listclean=True))

      print('wcctx._status=%s' % (str(wcctx._status)))

      print('=== with "pattern match":')

      print(actx1.status(other=wcctx,

                         match=scmutil.matchfiles(repo, ['bar-m', 'foo'])))

      print('wcctx._status=%s' % (str(wcctx._status)))

      print(actx2.status(other=wcctx,

                         match=scmutil.matchfiles(repo, ['bar-m', 'foo'])))

      print('wcctx._status=%s' % (str(wcctx._status)))

      print('=== with "pattern match" and "listclean=True":')

      print(actx1.status(other=wcctx,

                         match=scmutil.matchfiles(repo, ['bar-r', 'foo']),

                         listclean=True))

      print('wcctx._status=%s' % (str(wcctx._status)))

      print(actx2.status(other=wcctx,

                         match=scmutil.matchfiles(repo, ['bar-r', 'foo']),

                         listclean=True))

      print('wcctx._status=%s' % (str(wcctx._status)))

	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

				from __future__ import absolute_import, print_function
				import os
				from mercurial import (
				context,
				encoding,
				hg,
				ui as uimod,
				)

				u = uimod.ui.load()

				repo = hg.repository(u, 'test1', create=1)
				os.chdir('test1')

				# create 'foo' with fixed time stamp
				f = open('foo', 'wb')
				f.write(b'foo\n')
				f.close()
				os.utime('foo', (1000, 1000))

				# add+commit 'foo'
				repo[None].add(['foo'])
				repo.commit(text='commit1', date="0 0")

				if os.name == 'nt':
				d = repo[None]['foo'].date()
				print("workingfilectx.date = (%d, %d)" % (d[0], d[1]))
				else:
				print("workingfilectx.date =", repo[None]['foo'].date())

				# test memctx with non-ASCII commit message

				def filectxfn(repo, memctx, path):
				return context.memfilectx(repo, "foo", "")

				ctx = context.memctx(repo, ['tip', None],
				encoding.tolocal("Gr\xc3\xbcezi!"),
				["foo"], filectxfn)
				ctx.commit()
				for enc in "ASCII", "Latin-1", "UTF-8":
				encoding.encoding = enc
				print("%-8s: %s" % (enc, repo["tip"].description()))

				# test performing a status

				def getfilectx(repo, memctx, f):
				fctx = memctx.parents()[0][f]
				data, flags = fctx.data(), fctx.flags()
				if f == 'foo':
				data += 'bar\n'
				return context.memfilectx(repo, f, data, 'l' in flags, 'x' in flags)

				ctxa = repo.changectx(0)
				ctxb = context.memctx(repo, [ctxa.node(), None], "test diff", ["foo"],
				getfilectx, ctxa.user(), ctxa.date())

				print(ctxb.status(ctxa))

				# test performing a diff on a memctx

				for d in ctxb.diff(ctxa, git=True):
				print(d)

				# test safeness and correctness of "ctx.status()"
				print('= checking context.status():')

				# ancestor "wcctx ~ 2"
				actx2 = repo['.']

				repo.wwrite('bar-m', 'bar-m\n', '')
				repo.wwrite('bar-r', 'bar-r\n', '')
				repo[None].add(['bar-m', 'bar-r'])
				repo.commit(text='add bar-m, bar-r', date="0 0")

				# ancestor "wcctx ~ 1"
				actx1 = repo['.']

				repo.wwrite('bar-m', 'bar-m bar-m\n', '')
				repo.wwrite('bar-a', 'bar-a\n', '')
				repo[None].add(['bar-a'])
				repo[None].forget(['bar-r'])

				# status at this point:
				# M bar-m
				# A bar-a
				# R bar-r
				# C foo

				from mercurial import scmutil

				print('== checking workingctx.status:')

				wctx = repo[None]
				print('wctx._status=%s' % (str(wctx._status)))

				print('=== with "pattern match":')
				print(actx1.status(other=wctx,
				match=scmutil.matchfiles(repo, ['bar-m', 'foo'])))
				print('wctx._status=%s' % (str(wctx._status)))
				print(actx2.status(other=wctx,
				match=scmutil.matchfiles(repo, ['bar-m', 'foo'])))
				print('wctx._status=%s' % (str(wctx._status)))

				print('=== with "always match" and "listclean=True":')
				print(actx1.status(other=wctx, listclean=True))
				print('wctx._status=%s' % (str(wctx._status)))
				print(actx2.status(other=wctx, listclean=True))
				print('wctx._status=%s' % (str(wctx._status)))

				print("== checking workingcommitctx.status:")

				wcctx = context.workingcommitctx(repo,
				scmutil.status(['bar-m'],
				['bar-a'],
				[],
				[], [], [], []),
				text='', date='0 0')
				print('wcctx._status=%s' % (str(wcctx._status)))

				print('=== with "always match":')
				print(actx1.status(other=wcctx))
				print('wcctx._status=%s' % (str(wcctx._status)))
				print(actx2.status(other=wcctx))
				print('wcctx._status=%s' % (str(wcctx._status)))

				print('=== with "always match" and "listclean=True":')
				print(actx1.status(other=wcctx, listclean=True))
				print('wcctx._status=%s' % (str(wcctx._status)))
				print(actx2.status(other=wcctx, listclean=True))
				print('wcctx._status=%s' % (str(wcctx._status)))

				print('=== with "pattern match":')
				print(actx1.status(other=wcctx,
				match=scmutil.matchfiles(repo, ['bar-m', 'foo'])))
				print('wcctx._status=%s' % (str(wcctx._status)))
				print(actx2.status(other=wcctx,
				match=scmutil.matchfiles(repo, ['bar-m', 'foo'])))
				print('wcctx._status=%s' % (str(wcctx._status)))

				print('=== with "pattern match" and "listclean=True":')
				print(actx1.status(other=wcctx,
				match=scmutil.matchfiles(repo, ['bar-r', 'foo']),
				listclean=True))
				print('wcctx._status=%s' % (str(wcctx._status)))
				print(actx2.status(other=wcctx,
				match=scmutil.matchfiles(repo, ['bar-r', 'foo']),
				listclean=True))
				print('wcctx._status=%s' % (str(wcctx._status)))