summaryrefslogtreecommitdiff
path: root/files/common/sbin/p2ptbl
blob: a728a9c6b3dd8ad8da2af791b19a3cd8d08bbd16 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#!/bin/sh

set -e

init () {
    [ -n "$1" ] || printArgs
    mkdir -p "$(dirname $1)"
    [ -f "$1" ] || cat >"$1" </dev/null
}

get () {
    [ -n "$2" ] || printArgs
    grep "^$2"$'\t' "$1" | cut -f3-
}

update () {
    [ -n "$2" -a -n "$3" ] || printArgs
    oldversion=$(grep "^$2"$'\t' "$1" | cut -f2)
    tmpfile="$1.update"
    echo -e "$2\t$(($oldversion + 1 + $RANDOM))\t$3" > "$tmpfile"
    merge "$1" "$tmpfile"
    if [ -n "$4" ]; then
	# broadcast update
	gzip <"$tmpfile" >"$tmpfile.gz"
	broadcaster $4 p2ptbl/$(basename "$1") - <"$tmpfile.gz"
	rm "$tmpfile.gz"
    fi
    rm "$tmpfile"
}

merge () {
    sort "$1" "$2" \
	| {
	oldkey=""
	oldversion=0
	oldval=""
	while read key version val; do
	    if [ "$key" == "$oldkey" ]; then
		if [ "$version" -gt "$oldversion" ]; then
		    oldversion="$version"
		    oldval="$val"
		fi
	    else
		[ -n "$oldkey" ] && echo -e "$oldkey\t$oldversion\t$oldval"
		oldkey="$key"
		oldversion="$version"
		oldval="$val"
	    fi
	done
	[ -n "$oldkey" ] && echo -e "$oldkey\t$oldversion\t$oldval"
    } > "$1~"
    mv "$1~" "$1"
}

gossip() {
    tgts="$2"
    if="$3"
    [ "$tgts" -gt 0 -a -n "$if" ] 2>/dev/null || printArgs

    # binary search over number of sent lines until size is maximized
    # and below target size
    lc=$(wc -l <"$1")
    [ $lc == 0 ] && return
    shuf -v seed=$RANDOM <"$1" >"$1.gossip"
    while [ $lc -ge 1 ]; do
	head -n$lc "$1.gossip" | gzip >"$1.gossip.gz"
	if [ $tgts -ge $(wc -c <"$1.gossip.gz") ]; then
	    broadcaster $if p2ptbl/$(basename "$1") - <"$1.gossip.gz"
	    rm "$1.gossip" "$1.gossip.gz"
	    return;
	fi
	lc=$(($lc / 2))
    done
    rm "$1.gossip" "$1.gossip.gz"
    echo "$1 has rows bigger then requested packet size"
    exit 1
}

printArgs () {
    echo -e "Usage:
$0 init   table
$0 get    table key
$0 update table key value [interface]
$0 merge  table foreign-table
$0 gossip table target-size interface"
    exit 1
}

checkTable () {
    [ -n "$1" ] || printArgs;
    [ -f "$1" ] || { 
	echo "$1 is no P2P table"; 
	printArgs;
    }
}

lockTable () {
    checkTable "$2"
    exec 666<"$2"
    flock "$1" 666
}

case "$1" in
    init) 
	init "$2"
	;;
    get)
	lockTable -s "$2"
	get "$2" "$3"
	;;
    update)
	lockTable -s "$2"
	update "$2" "$3" "$4" "$5"
	;;
    merge)
	lockTable -x "$2"
	checkTable "$3" 
	merge "$2" "$3"
	;;
    gossip)
	lockTable -s "$2"
	gossip "$2" "$3" "$4"
	;;
    *) 
	printArgs
	;;
esac
contact: Jan Huwald // Impressum