From 491556b732c47ac92d4e2a0e8e129d7c24b636fd Mon Sep 17 00:00:00 2001
From: JACK <jack9603301@163.com>
Date: Wed, 2 Jun 2021 02:58:53 +0800
Subject: op-mode: T3384: support UDP bandwidth testing

---
 op-mode-definitions/monitor-bandwidth-test.xml.in | 50 ++++++++++++++++++-----
 1 file changed, 40 insertions(+), 10 deletions(-)

(limited to 'op-mode-definitions')

diff --git a/op-mode-definitions/monitor-bandwidth-test.xml.in b/op-mode-definitions/monitor-bandwidth-test.xml.in
index 5b36b1da5..f7ee242b7 100644
--- a/op-mode-definitions/monitor-bandwidth-test.xml.in
+++ b/op-mode-definitions/monitor-bandwidth-test.xml.in
@@ -7,21 +7,51 @@
           <help>Initiate or wait for bandwidth test</help>
         </properties>
         <children>
-          <leafNode name="accept">
+          <node name="accept">
             <properties>
-              <help>Wait for bandwidth test connections (port TCP/5001)</help>
+              <help>Wait for bandwidth test connections (default: TCP)</help>
             </properties>
             <command>/usr/bin/iperf -V -s</command>
-          </leafNode>
-          <tagNode name="initiate">
+            <children>
+              <leafNode name="tcp">
+                <properties>
+                  <help>Wait for bandwidth test connections (port TCP/5001)</help>
+                </properties>
+                <command>/usr/bin/iperf -V -s</command>
+              </leafNode>
+              <leafNode name="udp">
+                <properties>
+                  <help>Wait for bandwidth test connections (port UDP/5001)</help>
+                </properties>
+                <command>/usr/bin/iperf -V -s -u</command>
+              </leafNode>
+            </children>
+          </node>
+          <node name="initiate">
             <properties>
-              <help>Initiate a bandwidth test to specified host (port TCP/5001)</help>
-              <completionHelp>
-                <list>&lt;hostname&gt; &lt;x.x.x.x&gt; &lt;h:h:h:h:h:h:h:h&gt;</list>
-              </completionHelp>
+              <help>Initiate a bandwidth test to specified host</help>
             </properties>
-            <command>${vyos_op_scripts_dir}/monitor_bandwidth_test.sh "$4"</command>
-          </tagNode>
+            <children>
+              <tagNode name="tcp">
+                <properties>
+                  <help>Initiate a bandwidth test to specified host (port TCP/5001)</help>
+                  <completionHelp>
+                    <list>&lt;hostname&gt; &lt;x.x.x.x&gt; &lt;h:h:h:h:h:h:h:h&gt;</list>
+                  </completionHelp>
+                </properties>
+                <command>${vyos_op_scripts_dir}/monitor_bandwidth_test.sh "$5"</command>
+              </tagNode>
+              <tagNode name="udp">
+                <properties>
+                  <help>Initiate a bandwidth test to specified host (port UDP/5001)</help>
+                  <completionHelp>
+                    <list>&lt;hostname&gt; &lt;x.x.x.x&gt; &lt;h:h:h:h:h:h:h:h&gt;</list>
+                  </completionHelp>
+                </properties>
+                <command>${vyos_op_scripts_dir}/monitor_bandwidth_test.sh "$5" "-u"</command>
+              </tagNode>
+            </children>
+          </node>
         </children>
       </node>
     </children>
-- 
cgit v1.2.3