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

verify_service_availability()
{
    dig . @127.0.0.1 >/dev/null 2>&1
}

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)
            shift
            DEST=${1:-149.56.44.185}
            PORT=${2:-53}
            LOCAL_PORT=${3:-$PORT}
            COMMAND="socat UDP-RECVFROM:$LOCAL_PORT,fork UDP-SENDTO:$DEST:$PORT"
            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="always" \
                --property WatchdogSec="5" \
                -- \
                "$0" watchdog "$COMMAND"
            ;;
        watchdog)
	    shift
	    COMMAND="$1"
            watchdog $$ &
            exec $COMMAND
            ;;
        *)
            echo "Usage: $0 install [destination [destination port [local port]]]" >&2
            exit 1
            ;;
    esac
}

main "$@"