hg: obtain lock when creating share from pooled repo (issue5104) There are race conditions between clients performing a shared clone to pooled storage: 1) Clients race to create the new shared repo in the pool directory 2) 1 client is seeding the repo in the pool directory and another goes to share it before it is fully cloned We prevent these race conditions by obtaining a lock in the pool directory that is derived from the name of the repo we will be accessing. To test this, a simple generic "lockdelay" extension has been added. The extension inserts an optional, configurable delay before or after lock acquisition. In the test, we delay 2 seconds after lock acquisition in the first process and 1 second before lock acquisition in the 2nd process. This means the first process has 1s to obtain the lock. There is a race condition here. If we encounter it in the wild, we could change the dummy extension to wait on the lock file to appear instead of relying on timing. But that's more complicated. Let's see what happens first.

     1

import os, __builtin__

     2

from mercurial import util

     3

     4

def lowerwrap(scope, funcname):

     5

    f = getattr(scope, funcname)

     6

    def wrap(fname, *args, **kwargs):

     7

        d, base = os.path.split(fname)

     8

        try:

     9

            files = os.listdir(d or '.')

    10

        except OSError:

    11

            files = []

    12

        if base in files:

    13

            return f(fname, *args, **kwargs)

    14

        for fn in files:

    15

            if fn.lower() == base.lower():

    16

                return f(os.path.join(d, fn), *args, **kwargs)

    17

        return f(fname, *args, **kwargs)

    18

    scope.__dict__[funcname] = wrap

    19

    20

def normcase(path):

    21

    return path.lower()

    22

    23

os.path.normcase = normcase

    24

    25

for f in 'file open'.split():

    26

    lowerwrap(__builtin__, f)

    27

    28

for f in "chmod chown open lstat stat remove unlink".split():

    29

    lowerwrap(os, f)

    30

    31

for f in "exists lexists".split():

    32

    lowerwrap(os.path, f)

    33

    34

lowerwrap(util, 'posixfile')