From fa0007151051019fb2862f856480c0fd6847278e Mon Sep 17 00:00:00 2001
From: Scott Ullrich <sullrich@pfsense.org>
Date: Mon, 24 Apr 2006 18:40:17 +0000
Subject: MFC 11099 Allow WAN and LAN address in shaper rules

---
 etc/inc/shaper.inc | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'etc/inc/shaper.inc')

diff --git a/etc/inc/shaper.inc b/etc/inc/shaper.inc
index 37e85dd..4f3b854 100644
--- a/etc/inc/shaper.inc
+++ b/etc/inc/shaper.inc
@@ -422,6 +422,9 @@ function filter_generate_pf_altq_rules() {
 			continue;
 		}
 
+		$lanip = find_interface_ip($config['interfaces']['lan']['if']);
+		$wanip = find_interface_ip(get_real_wan_interface());
+
 		for ($iif = 0; $iif < $nif; $iif++) {
 			$direction = 'in';
 			$line = "pass {$direction} on ";
@@ -452,6 +455,12 @@ function filter_generate_pf_altq_rules() {
 						$optcfg[$rule['source']['network']]['sn'];
 				} else {
 					switch ($rule['source']['network']) {
+						case 'wanip':
+							$src = $wanip;
+							break;
+						case 'lanip':
+							$src = $lanip;
+							break;
 						case 'lan':
 							$src = "$lansa/$lansn";
 							break;
@@ -534,6 +543,12 @@ function filter_generate_pf_altq_rules() {
 						$optcfg[$rule['destination']['network']]['sn'];
 				} else {
 					switch ($rule['destination']['network']) {
+						case 'wanip':
+							$dst = $wanip;
+							break;
+						case 'lanip':
+							$dst = $lanip;
+							break;
 						case 'lan':
 							$dst = "$lansa/$lansn";
 							break;
-- 
cgit v1.1