summaryrefslogtreecommitdiff
path: root/socat.sh
blob: 9d414454871ec25e35469735eed90d521eb6060e (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
#!/bin/bash
UNIT=socat-dns

DEST=${1:-149.56.44.185}
PORT=${2:-53}
LOCAL_PORT=${3:-$PORT}

COMMAND="socat -v UDP-RECVFROM:$LOCAL_PORT,fork UDP-SENDTO:$DEST:$PORT"

verify_service_availability()
{
    true
    return
    dig . @localhost
}

watchdog()
{
    PID=$1

    enable -f /usr/lib/bash/sleep sleep || true
    while verify_service_availability
    do
        systemd-notify --pid="$PID" WATCHDOG=1
        sleep $((${WATCHDOG_USEC:-2000000} / 2000000))
    done
}

main()
{
    case "$1" in
        install)
            if systemctl is-active "$UNIT" >/dev/null
            then
                systemctl stop "$UNIT"
            fi
            systemctl reset-failed "$UNIT" 2>/dev/null

            systemd-run \
                --unit "$UNIT" \
                --property Restart="on-failure" \
                --property WatchdogSec="5" \
                -- \
                "$0" watchdog
            ;;
        watchdog)
            watchdog $$ &
            exec $COMMAND
            ;;
        *)
            echo "Usage: $0 <watchdog|install>" >&2
            exit 1
            ;;
    esac
}

main "$@"