From 9beae6fc58a399d40dec36270533d04ff06e27f6 Mon Sep 17 00:00:00 2001
From: ali-aqrabawi <aaqrabaw@okdanetworks.com>
Date: Thu, 8 Aug 2024 21:34:06 +0300
Subject: [PATCH] [tc-filter]: add nat action, change action to container to
 hold both nat and gact actions

Signed-off-by: ali-aqrabawi <aaqrabaw@okdanetworks.com>
---
 tests/cases/test_tc_filter_data.xml | 33 ++++++++++++-----
 yang/iproute2-tc-filter.yang        | 55 ++++++++++++++++++++++++-----
 2 files changed, 71 insertions(+), 17 deletions(-)

diff --git a/tests/cases/test_tc_filter_data.xml b/tests/cases/test_tc_filter_data.xml
index 498e3b3..d3b299c 100644
--- a/tests/cases/test_tc_filter_data.xml
+++ b/tests/cases/test_tc_filter_data.xml
@@ -44,7 +44,10 @@
                     <src_ip>1.1.1.1/32</src_ip>
                 </match>
                 <classid>8001:20</classid>
-                <action>pass</action>
+                <actions>
+                    <gact>pass</gact>
+                </actions>
+
             </flower>
         </rule>
         <rule>
@@ -54,7 +57,9 @@
                     <src_ip>5.5.5.5/32</src_ip>
                 </match>
                 <classid>8001:30</classid>
-                <action>pass</action>
+                <actions>
+                    <gact>pass</gact>
+                </actions>
             </flower>
         </rule>
         <rule>
@@ -64,7 +69,9 @@
                     <dst_ip>9.9.9.0/24</dst_ip>
                 </match>
                 <classid>8001:40</classid>
-                <action>pass</action>
+                <actions>
+                    <gact>pass</gact>
+                </actions>
             </flower>
         </rule>
     </qdisc-filter>
@@ -95,13 +102,17 @@
                     <ip_proto>icmp</ip_proto>
                     <src_ip>1.2.1.2/32</src_ip>
                 </match>
-                <action>pass</action>
+                <actions>
+                    <gact>pass</gact>
+                </actions>
             </flower>
         </rule>
         <rule>
             <pref>20</pref>
             <flower>
-                <action>drop</action>
+                <actions>
+                    <gact>drop</gact>
+                </actions>
             </flower>
         </rule>
     </dev-filter>
@@ -116,13 +127,17 @@
                     <ip_proto>icmp</ip_proto>
                     <src_ip>11.21.11.21/32</src_ip>
                 </match>
-                <action>pass</action>
+                <actions>
+                    <gact>pass</gact>
+                </actions>
             </flower>
         </rule>
         <rule>
             <pref>201</pref>
             <flower>
-                <action>drop</action>
+                <actions>
+                    <gact>drop</gact>
+                </actions>
             </flower>
         </rule>
     </dev-filter>
@@ -132,7 +147,9 @@
         <rule>
             <pref>10</pref>
             <flower>
-                <action>pass</action>
+                <actions>
+                    <gact>pass</gact>
+                </actions>
             </flower>
         </rule>
     </shared-block-filter>
diff --git a/yang/iproute2-tc-filter.yang b/yang/iproute2-tc-filter.yang
index cc27dc0..b2d671e 100644
--- a/yang/iproute2-tc-filter.yang
+++ b/yang/iproute2-tc-filter.yang
@@ -50,16 +50,53 @@ module iproute2-tc-filter {
                 is in the form X:Y, while X and Y are interpreted as
                 numbers in hexadecimal format.";
         }
-        leaf action {
-            type enumeration {
-                enum "pass";
-                enum "drop";
-                enum "reclassify";
-                enum "continue";
-                enum "pipe";
+
+        container actions {
+            leaf gact {
+                ipr2cgen:arg-name "action";
+                type enumeration {
+                    enum "pass";
+                    enum "drop";
+                    enum "reclassify";
+                    enum "continue";
+                    enum "pipe";
+                }
+                description
+                    "specify action";
+            }
+            container nat {
+                leaf direction {
+                    ipr2cgen:arg-name "action nat ";
+                    must "../old_ip and ../new_ip"{
+                        error-message "direction, old_ip and new_ip are mandatory for nat action.";
+                    }
+                    type enumeration {
+                        enum "ingress";
+                        enum "egress";
+                    }
+                    description "NAT direction";
+                }
+                leaf old_ip {
+                    ipr2cgen:value-only;
+                    must "../direction and ../new_ip"{
+                        error-message "direction, old_ip and new_ip are mandatory for nat action.";
+                    }
+                    type union {
+                        type inet:ip-address;
+                        type inet:ip-prefix;
+                    }
+                    description "original ip";
+                }
+                leaf new_ip {
+                    ipr2cgen:value-only;
+                    must "../direction and ../old_ip"{
+                        error-message "direction, old_ip and new_ip are mandatory for nat action.";
+                    }
+                    type inet:ip-address;
+                    description "natted ip";
+                }
+
             }
-            description
-                "specify action";
         }
     }
     grouping filter-rule{