summaryrefslogtreecommitdiff
path: root/files/common/sbin/p2ptbl
blob: 2048e374e0253f00d8a032e8e1fdbfff23b35bfc (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
128
129
130
131
132
133
134
135
136
#!/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-
}

show () {
    cut -f1,3- "$1"
}

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 >&2 "Usage:
$0 init   table
$0 get    table key
$0 show   table
$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 >&2 "$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"
	;;
    show)
	lockTable -s "$2"
	show "$2"
	;;
    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