149 lines
4.3 KiB
Bash
149 lines
4.3 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
#
|
||
|
|
# Measurement script for the dn42 peer finder, see http://dn42.us/peers
|
||
|
|
# Dependencies: curl, sed, ping
|
||
|
|
#
|
||
|
|
# This script is designed to be run in cron every 5 minutes, like this:
|
||
|
|
#
|
||
|
|
# UUID=<Your UUID goes here>
|
||
|
|
# */5 * * * * /home/foo/cron.sh
|
||
|
|
#
|
||
|
|
|
||
|
|
# Put your UUID here, and keep it secret!
|
||
|
|
UUID=${UUID}
|
||
|
|
PEERFINDER=${PEERFINDER:-"https://dn42.us/peers"}
|
||
|
|
NB_PINGS=${NB_PINGS:-5}
|
||
|
|
LOGFILE=${LOGFILE:-/dev/stdout} # Set to /dev/null to only receive errors.
|
||
|
|
# Set to a file writable by the cron runner to record pings.
|
||
|
|
# (Errors will be sent in cron mail)
|
||
|
|
WARNLOCK=${WARNLOCK:-warn.lock} # Set this variable if you want a file written when the script updates.
|
||
|
|
LOCKFILE=${LOCKFILE:-exec.lock} # Set this variable if you want the script to not run multiple instances at once.
|
||
|
|
LOCKFD=${LOCKFD:-42}
|
||
|
|
|
||
|
|
# This avoids synchronisation (everybody fetching jobs and running
|
||
|
|
# measurements simultaneously)
|
||
|
|
RANDOM_DELAY=30
|
||
|
|
|
||
|
|
function die() {
|
||
|
|
echo "## PEERFINDER ERROR $(date) ## " \
|
||
|
|
"$*"
|
||
|
|
exit 1
|
||
|
|
}
|
||
|
|
|
||
|
|
if command -v flock >/dev/null 2>&1; then
|
||
|
|
eval "exec $LOCKFD>$LOCKFILE"
|
||
|
|
flock -n $LOCKFD || die "Unable to acquire lock."
|
||
|
|
fi
|
||
|
|
|
||
|
|
VERSION=1.0.10
|
||
|
|
ver() { printf "%03d%03d%03d%03d" $(echo "$1" | tr '.' ' '); }
|
||
|
|
|
||
|
|
[ -e $LOGFILE ] || touch $LOGFILE
|
||
|
|
exec >> $LOGFILE
|
||
|
|
|
||
|
|
echo "STARTING PEERFINDER (v. $VERSION)"
|
||
|
|
|
||
|
|
SLEEP=$((RANDOM % RANDOM_DELAY))
|
||
|
|
|
||
|
|
# check for ping binary
|
||
|
|
PING=$(which ping)
|
||
|
|
if [ -z "$PING" ]; then
|
||
|
|
die "Unable to find a suitable ping binary."
|
||
|
|
fi
|
||
|
|
|
||
|
|
CURL=$(which curl)
|
||
|
|
if [ -z "$CURL" ]; then
|
||
|
|
die "Unable to find a suitable curl binary."
|
||
|
|
fi
|
||
|
|
CURL="$CURL -A PeerFinder -sf"
|
||
|
|
|
||
|
|
case $OSTYPE in
|
||
|
|
solaris*)
|
||
|
|
GREP=$(which ggrep)
|
||
|
|
|
||
|
|
function ping_cmd() {
|
||
|
|
$PING -sn $3 56 $2
|
||
|
|
}
|
||
|
|
;;
|
||
|
|
*-gnu|*)
|
||
|
|
# check for IPv6 binary. if ping6 is missing assume 'ping -6'
|
||
|
|
PING6=$(which ping6)
|
||
|
|
[ -z "$PING6" -a -n "$PING" ] && PING6="$PING -6"
|
||
|
|
|
||
|
|
GREP=$(which grep)
|
||
|
|
|
||
|
|
function ping_cmd() {
|
||
|
|
[ "$1" -eq "1" -a -n "$PING" ] && $PING -nqc $2 $3
|
||
|
|
[ "$1" -eq "2" -a -n "$PING6" ] && $PING6 -nqc $2 $3
|
||
|
|
}
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
|
||
|
|
while true ; do
|
||
|
|
|
||
|
|
JOB=$(mktemp)
|
||
|
|
|
||
|
|
$CURL -H 'accept: text/environment' "$PEERFINDER/pending/$UUID" | tee $JOB
|
||
|
|
|
||
|
|
REQ_ID=$($GREP REQ_ID $JOB|cut -d'=' -f2|tr -d '[$`;><{}%|&!()]"/\\')
|
||
|
|
REQ_IP=$($GREP REQ_IP $JOB|cut -d'=' -f2|tr -d '[$`;><{}%|&!()]"/\\')
|
||
|
|
REQ_FAMILY=$($GREP REQ_FAMILY $JOB|cut -d'=' -f2|tr -d '[$`;><{}%|&!()]"/\\')
|
||
|
|
CUR_VERSION=$($GREP SCRIPT_VERSION $JOB|cut -d'=' -f2|tr -d '[$`;><{}%|&!()]"/\\')
|
||
|
|
|
||
|
|
rm "$JOB"
|
||
|
|
|
||
|
|
if [ $(ver "$VERSION") -lt $(ver "$CUR_VERSION") ]; then
|
||
|
|
echo "## PEERFINDER WARN $(date) ## " \
|
||
|
|
"Current script version is $CUR_VERSION. You are running $VERSION " \
|
||
|
|
"Get it here: https://dn42.us/peers/script"
|
||
|
|
|
||
|
|
[ -z "$WARNLOCK" ] && touch $WARNLOCK
|
||
|
|
else
|
||
|
|
[ -z "$WARNLOCK" -o -f "$WARNLOCK" ] && rm $WARNLOCK
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Avoid empty fields
|
||
|
|
[ -z "$REQ_ID" -a -z "$REQ_IP" ] && exit
|
||
|
|
|
||
|
|
echo "PINGING TO: $REQ_IP for $REQ_ID..."
|
||
|
|
|
||
|
|
# Parsing ping output, for Linux
|
||
|
|
if ! output=$(ping_cmd "$REQ_FAMILY" "$NB_PINGS" "$REQ_IP" 2>&1 | $GREP -A1 "packets transmitted"); then
|
||
|
|
sent=0
|
||
|
|
received=0
|
||
|
|
args="res_latency=NULL"
|
||
|
|
echo "Target $REQ_ID ($REQ_IP) is unreachable"
|
||
|
|
else
|
||
|
|
pattern='([0-9]*) packets transmitted, ([0-9]*)( packets)? received'
|
||
|
|
if [[ $output =~ $pattern ]]; then
|
||
|
|
sent=${BASH_REMATCH[1]}
|
||
|
|
received=${BASH_REMATCH[2]}
|
||
|
|
if [ "$received" -eq 0 ]
|
||
|
|
then
|
||
|
|
args="res_latency=NULL"
|
||
|
|
echo "Target $REQ_ID ($REQ_IP) is unreachable"
|
||
|
|
else
|
||
|
|
pattern='(rtt|round-trip).* min/avg/max.*= ([^/]*)/([^/]*)/([^/]*)(/(.*))?( ms)?'
|
||
|
|
if [[ $output =~ $pattern ]]; then
|
||
|
|
minrtt=${BASH_REMATCH[1]}
|
||
|
|
avgrtt=${BASH_REMATCH[2]}
|
||
|
|
maxrtt=${BASH_REMATCH[3]}
|
||
|
|
jitter=${BASH_REMATCH[4]}
|
||
|
|
[ -z "$avgrtt" ] && exit
|
||
|
|
echo "RTT to target $REQ_ID ($REQ_IP) is $avgrtt"
|
||
|
|
args="res_latency=${avgrtt}"
|
||
|
|
else
|
||
|
|
args="res_latency=NULL"
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
else
|
||
|
|
args="res_latency=NULL"
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Report results back to peerfinder
|
||
|
|
$CURL -X POST "$PEERFINDER/req/$REQ_ID" -d "peer_id=$UUID&peer_version=$VERSION&$args" -H 'accept: text/environment'
|
||
|
|
|
||
|
|
done
|
||
|
|
|