policy: T3624: add extended community bandwith support to route-map

3 files changed, 44 insertions, 0 deletions

diff --git a/data/templates/frr/policy.frr.tmpl b/data/templates/frr/policy.frr.tmpl
index a16eacdef..b5649b44e 100644
--- a/data/templates/frr/policy.frr.tmpl
+++ b/data/templates/frr/policy.frr.tmpl
@@ -247,6 +247,9 @@ route-map {{ route_map }} {{ rule_config.action }} {{ rule }}
 {%           if rule_config.set.distance is defined and rule_config.set.distance is not none %}
  set distance {{ rule_config.set.distance }}
 {%           endif %}
+{%           if rule_config.set.extcommunity is defined and rule_config.set.extcommunity.bandwidth is defined and rule_config.set.extcommunity.bandwidth is not none %}
+ set extcommunity bandwidth {{ rule_config.set.extcommunity.bandwidth }}
+{%           endif %}
 {%           if rule_config.set.extcommunity is defined and rule_config.set.extcommunity.rt is defined and rule_config.set.extcommunity.rt is not none %}
  set extcommunity rt {{ rule_config.set.extcommunity.rt }}
 {%           endif %}
diff --git a/interface-definitions/policy.xml.in b/interface-definitions/policy.xml.in
index 1b37c93b0..7bee8d635 100644
--- a/interface-definitions/policy.xml.in
+++ b/interface-definitions/policy.xml.in
@@ -1012,6 +1012,30 @@
                       <help>BGP extended community attribute</help>
                     </properties>
                     <children>
+                      <leafNode name="bandwidth">
+                        <properties>
+                          <help>Bandwidth value in Mbps</help>
+                          <completionHelp>
+                            <list>cumulative num-multipaths</list>
+                          </completionHelp>
+                          <valueHelp>
+                            <format>u32:1-25600</format>
+                            <description>Bandwidth value in Mbps</description>
+                          </valueHelp>
+                          <valueHelp>
+                            <format>cumulative</format>
+                            <description>Cumulative bandwidth of all multipaths (outbound-only)</description>
+                          </valueHelp>
+                          <valueHelp>
+                            <format>num-multipaths</format>
+                            <description>Internally computed bandwidth based on number of multipaths (outbound-only)</description>
+                          </valueHelp>
+                          <constraint>
+                            <validator name="numeric" argument="--range 1-25600"/>
+                            <regex>^(cumulative|num-multipaths)$</regex>
+                          </constraint>
+                        </properties>
+                      </leafNode>
                       <leafNode name="rt">
                         <properties>
                           <help>Set route target value</help>
diff --git a/smoketest/scripts/cli/test_policy.py b/smoketest/scripts/cli/test_policy.py
index bfa44088e..2d7b78048 100755
--- a/smoketest/scripts/cli/test_policy.py
+++ b/smoketest/scripts/cli/test_policy.py
@@ -773,6 +773,7 @@ class TestPolicy(VyOSUnitTestSHIM.TestCase):
                             'as-path-prepend'     : '1234567890 987654321',
                             'atomic-aggregate'    : '',
                             'distance'            : '110',
+                            'extcommunity-bw'     : '20000',
                             'extcommunity-rt'     : '123:456',
                             'extcommunity-soo'    : '456:789',
                             'ipv6-next-hop-global': '2001::1',
@@ -791,6 +792,18 @@ class TestPolicy(VyOSUnitTestSHIM.TestCase):
                     },
                 },
             },
+            'bandwidth-configuration' : {
+                'rule' : {
+                    '10' : {
+                        'action' : 'deny',
+                        'set' : {
+                            'as-path-prepend'     : '100 100',
+                            'distance'            : '200',
+                            'extcommunity-bw'     : 'num-multipaths',
+                        },
+                    },
+                },
+            },
         }
 
         self.cli_set(['policy', 'access-list', access_list, 'rule', '10', 'action', 'permit'])
@@ -898,6 +911,8 @@ class TestPolicy(VyOSUnitTestSHIM.TestCase):
                         self.cli_set(path + ['rule', rule, 'set', 'atomic-aggregate'])
                     if 'distance' in rule_config['set']:
                         self.cli_set(path + ['rule', rule, 'set', 'distance', rule_config['set']['distance']])
+                    if 'extcommunity-bw' in rule_config['set']:
+                        self.cli_set(path + ['rule', rule, 'set', 'extcommunity', 'bandwidth', rule_config['set']['extcommunity-bw']])
                     if 'extcommunity-rt' in rule_config['set']:
                         self.cli_set(path + ['rule', rule, 'set', 'extcommunity', 'rt', rule_config['set']['extcommunity-rt']])
                     if 'extcommunity-soo' in rule_config['set']:
@@ -1041,6 +1056,8 @@ class TestPolicy(VyOSUnitTestSHIM.TestCase):
                         tmp += 'atomic-aggregate'
                     elif 'distance' in rule_config['set']:
                         tmp += 'distance ' + rule_config['set']['distance']
+                    elif 'extcommunity-bw' in rule_config['set']:
+                        tmp += 'extcommunity bandwidth' + rule_config['set']['extcommunity-bw']
                     elif 'extcommunity-rt' in rule_config['set']:
                         tmp += 'extcommunity rt' + rule_config['set']['extcommunity-rt']
                     elif 'extcommunity-soo' in rule_config['set']: