lock-and-run.sh 1.47 KB
Newer Older
1
#! /bin/sh
2 3 4 5
# Shell-based mutex using mkdir.  This script is used in make to prefer
# serialized execution to avoid consuming too much RAM.  If reusing it,
# bear in mind that the lock-breaking logic is not race-free, so disable
# it in err() if concurrent execution could cause more serious problems.
6

7
self=`basename $0`
Jason Merrill committed
8 9
lockdir="$1" prog="$2"; shift 2 || exit 1

10
# Remember when we started trying to acquire the lock.
Jason Merrill committed
11 12
count=0

13 14 15
err () {
    if test -f $lockdir/lock-$1.$$; then
	rm -rf $lockdir
16 17
	echo "$self: *** (PID $$) removed stale $lockdir" >&2

18 19 20 21 22 23 24
	# Possible variant for uses where races are more problematic:
	#echo "$self: *** (PID $$) giving up, maybe rm -r $lockdir" >&2
	#exit 42
    else
	touch $lockdir/lock-$1.$$
    fi
}
Jason Merrill committed
25

26 27 28
until mkdir "$lockdir" 2>/dev/null; do
    # Say something periodically so the user knows what's up.
    if [ `expr $count % 30` = 0 ]; then
29 30 31 32 33 34
	# Check for valid lock.
	if pid=`cat $lockdir/pid 2>/dev/null` && kill -0 $pid 2>/dev/null; then
	    echo "$self: (PID $$) waiting $count sec to acquire $lockdir from PID $pid" >&2
	elif test -z "$pid"; then
	    echo "$self: (PID $$) cannot read $lockdir/pid" >&2
	    err nopid
35
	else
36 37
	    echo "$self: (PID $$) cannot signal $lockdir owner PID $pid" >&2
	    err dead
38 39 40 41 42
	fi
    fi
    sleep 1
    count=`expr $count + 1`
done
Jason Merrill committed
43

44 45 46 47
trap 'rm -rf "$lockdir"' 0
echo $$ > $lockdir/pidT && mv $lockdir/pidT $lockdir/pid
echo "$self: (PID $$) acquired $lockdir after $count seconds" >&2

48 49
echo $prog "$@"
$prog "$@"
Jason Merrill committed
50

51
# The trap runs on exit.