#!/bin/sh
. /etc/functions.sh
WAN=$(nvram get wan_ifname)
LAN=$(nvram get lan_ifname)
WIFI=$(nvram get wifi_ifname)

VPN_SERVER="10.10.1.1"
WWW_SERVER="127.0.0.1"

## CLEAR TABLES
for T in filter nat mangle; do
  iptables -t $T -F
  iptables -t $T -X
done

iptables -N input_rule
iptables -N output_rule
iptables -N forwarding_rule

iptables -t nat -N prerouting_rule
iptables -t nat -N postrouting_rule

#SSH on all interfaces
iptables        -A input_rule                -p tcp --dport 22 -j ACCEPT

#DNS requests on WIFI
iptables        -A input_rule      -i $WIFI  -p tcp --dport 53 -j ACCEPT 
iptables        -A input_rule      -i $WIFI  -p udp --dport 53 -j ACCEPT

#DHCP Server on WIFI
iptables        -A input_rule      -i $WIFI  -p udp --dport 67 -j ACCEPT 

#DHCP Client on LAN
iptables        -A input_rule      -i $LAN   -p udp --dport 68 -j ACCEPT


### Port forwarding
#PPTP on port 1723 
iptables -t nat -A prerouting_rule -i $WIFI -p tcp --dport 1723 -j DNAT --to $VPN_SERVER 
iptables        -A forwarding_rule -i $WIFI -p tcp --dport 1723 -d $VPN_SERVER -j ACCEPT 

#PPTP GRE forwarding
iptables -t nat -A prerouting_rule -i $WIFI -p 47 -j DNAT --to $VPN_SERVER 
iptables        -A forwarding_rule -i $WIFI -p 47 -d $VPN_SERVER -j ACCEPT 

#Forward port 80 requests to localhost 
iptables -A PREROUTING -t nat -p tcp -i $WIFI --dport 80 -j DNAT --to-destination 127.0.0.1:80
iptables -I FORWARD -t filter -i $WIFI -p tcp --dport 80 -j ACCEPT


### INPUT
###  (connections with the router as destination)

  # base case               
  iptables -P INPUT DROP
  iptables -A INPUT -m state --state INVALID -j DROP
  iptables -A INPUT -m state --state RELATED,ESTABLISHED -j ACCEPT
  iptables -A INPUT -p tcp --syn --tcp-option \! 2 -j  DROP

  # allow
  iptables -A INPUT -p icmp     -j ACCEPT       # allow ICMP
  #
  # insert accept rule or to jump to new accept-check table here
  #
  iptables -A INPUT -j input_rule

  # reject (what to do with anything not allowed earlier)
  iptables -A INPUT -p tcp -j REJECT --reject-with tcp-reset
  iptables -A INPUT -j REJECT --reject-with icmp-port-unreachable

### OUTPUT
### (connections with the router as source)

  # base case
  iptables -P OUTPUT DROP
  iptables -A OUTPUT -m state --state INVALID -j DROP
  iptables -A OUTPUT -m state --state RELATED,ESTABLISHED -j ACCEPT

  # allow
  iptables -A OUTPUT -j ACCEPT          #allow everything out
  #                         
  # insert accept rule or to jump to new accept-check table here
  #
  iptables -A OUTPUT -j output_rule

  # reject (what to do with anything not allowed earlier)
  iptables -A OUTPUT -p tcp -j REJECT --reject-with tcp-reset
  iptables -A OUTPUT -j REJECT --reject-with icmp-port-unreachable

### FORWARDING
### (connections routed through the router)

  # base case
  iptables -P FORWARD DROP 
  iptables -A FORWARD -m state --state INVALID -j DROP
  iptables -A FORWARD -m state --state RELATED,ESTABLISHED -j ACCEPT
  iptables -A FORWARD -p tcp --tcp-flags SYN,RST SYN -j TCPMSS --clamp-mss-to-pmtu

  # allow
  iptables -A FORWARD -i br0 -o br0 -j ACCEPT
  iptables -A FORWARD -i $LAN -o $WAN -j ACCEPT
  #
  # insert accept rule or to jump to new accept-check table here
  #
  iptables -A FORWARD -j forwarding_rule

  # reject (what to do with anything not allowed earlier)
  # uses the default -P DROP

### MASQ
  iptables -t nat -A PREROUTING -j prerouting_rule
  iptables -t nat -A POSTROUTING -j postrouting_rule
  iptables -t nat -A postrouting_rule -p tcp --dport 1723 -d $VPN_SERVER -j MASQUERADE
  iptables -t nat -A postrouting_rule -p 47 -d $VPN_SERVER -j MASQUERADE