summaryrefslogtreecommitdiff
path: root/port_forward
blob: 4842b3cb775c3e5448e571e7de4b517311cf3c06 (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
#!/bin/sh
set -e


usage() {
    cat <<EOF
    $(basename $0) [options] <in-interface> <out-interface> <port> <destination>

    --clear           Clear the previous rules before inserting new
                      ones

    --protocol        Protocol for the rules to use.

    in-interface      Interface on which incoming traffic is expected
    out-interface     Interface to which incoming traffic is to be
                      forwarded.

    port              Port to be forwarded. Can be integer or string
                      from /etc/services.
    destination       IP and port of the destination system to which
                      traffic needs to be forwarded. This should be in
                      form <destination_ip:port>

(C) 2015 Vasudev Kamath - This program comes with ABSOLUTELY NO
WARRANTY. This is free software, and you are welcome to redistribute
it under the GNU GPL Version 3 (or later) License

EOF
}

setup_portforwarding () {
    local protocol="$1"
    iptables -t nat -A PREROUTING -i $IN_INTERFACE -p "$protocol" --dport $PORT \
	     -j DNAT --to $DESTINATION
    iptables -A FORWARD -p "$protocol" -d ${DESTINATION%%\:*} --dport $PORT -j ACCEPT

    # Returning packet should have gateway IP
    for ip in ${IN_IP} ; do
	iptables -t nat -A POSTROUTING -s ${DESTINATION%%\:*} -o \
		 $IN_INTERFACE -j SNAT --to ${ip%%\/*}
    done

}

if [ $(id -u) -ne 0 ]; then
    echo "You need to be root to run this script"
    exit 1
fi

while true; do
    case "$1" in
	--clear)
	    CLEAR_RULES=1
	    shift
	    ;;
	--protocol|--protocol=*?)
	if [ "$1" = "--protocol" -a -n "$2" ];then
	    PROTOCOL="$2"
	    shift 2
	elif [ "${1#--protocol=}" != "$1" ]; then
	    PROTOCOL="${1$--protocol=}"
	    shift 1
	else
	    echo "You need to specify protocl (tcp|udp)"
	    exit 2
	fi
	;;

	*)
	    break
	    ;;
    esac
done

if [ $# -ne 4 ]; then
    usage $0
    exit 2
fi

IN_INTERFACE="$1"
OUT_INTERFACE="$2"
PORT="$3"
DESTINATION="$4"

# Get the incoming interface IP. This is used for SNAT.
IN_IP=$(ip addr show $IN_INTERFACE|\
	       perl -nE '/inet\s(.*)\sscope/ and say $1' | tr '\n' ' ')


if [ -n "$CLEAR_RULES" ]; then
    iptables -t nat -X
    iptables -t nat -F
    iptables -F
fi

if [ -n "$PROTOCOL" ]; then
    setup_portforwarding $PROTOCOL
else
    setup_portforwarding tcp
    setup_portforwarding udp
fi