TSG-13837 decrypted traffic steering适配mrzcpd新增MR_BUFF_USER_0类型

2 files changed, 13 insertions, 1 deletions

diff --git a/test/marsio.cpp b/test/marsio.cpp
index 1a6c974..bda26b9 100644
--- a/test/marsio.cpp
+++ b/test/marsio.cpp
@@ -55,6 +55,7 @@ struct mrb_metadata
     uint16_t port_egress;
 
     uint16_t link_db_index;
+    uint16_t user_data_0;
 };
 
 struct mock_marsio_buff_t
@@ -171,6 +172,9 @@ int marsio_buff_set_metadata(marsio_buff_t *m, enum mr_buff_metadata_type type,
     case MR_BUFF_PAYLOAD_OFFSET:
         mrb_metadata->payload_offset = *(uint16_t *)data;
         return 0;
+    case MR_BUFF_USER_0:
+        mrb_metadata->user_data_0 = *(uint16_t *)data;
+        return 0;
     default:
         return -1;
     }
@@ -224,6 +228,13 @@ int marsio_buff_get_metadata(marsio_buff_t *m, enum mr_buff_metadata_type type,
         }
         *(uint16_t *)(data) = (uint16_t)mrb_metadata->payload_offset;
         return sizeof(uint16_t);
+    case MR_BUFF_USER_0:
+        if (sz_data < sizeof(uint16_t))
+        {
+            return -1;
+        }
+        *(uint16_t *)(data) = (uint16_t)mrb_metadata->user_data_0;
+        return sizeof(uint16_t);
     default:
         return -1;
     }
diff --git a/test/marsio.h b/test/marsio.h
index 3ea0dc2..516c824 100644
--- a/test/marsio.h
+++ b/test/marsio.h
@@ -38,7 +38,8 @@ enum mr_buff_metadata_type
     MR_BUFF_ROUTE_CTX = 2,
     MR_BUFF_SESSION_ID = 3,
     MR_BUFF_DIR = 4,
-    MR_BUFF_PAYLOAD_OFFSET = 5
+    MR_BUFF_PAYLOAD_OFFSET = 5,
+    MR_BUFF_USER_0 = 254,
 };
 
 struct mr_instance *marsio_create();