Add firewall support

Add firewall support + testcases