add seed per stream

4 files changed, 37 insertions, 3 deletions

diff --git a/src/stateless/cp/trex_stream.h b/src/stateless/cp/trex_stream.h
index 36f9d407..161e9592 100644
--- a/src/stateless/cp/trex_stream.h
+++ b/src/stateless/cp/trex_stream.h
@@ -429,6 +429,7 @@ public:
         dp->m_ibg_usec              =   m_ibg_usec;
         dp->m_flags                 =   m_flags;
         dp->m_action_count          =   m_action_count;
+        dp->m_random_seed           =   m_random_seed;
 
         dp->m_rate                  =   m_rate;
 
@@ -485,6 +486,7 @@ public:
     uint16_t      m_flags;
     uint32_t      m_stream_id;              /* id from RPC can be anything */
     uint16_t      m_action_count;       
+    uint32_t      m_random_seed;
     
 
     /* config fields */
diff --git a/src/stateless/cp/trex_stream_vm.h b/src/stateless/cp/trex_stream_vm.h
index c1db090d..4a0b1d59 100644
--- a/src/stateless/cp/trex_stream_vm.h
+++ b/src/stateless/cp/trex_stream_vm.h
@@ -1276,7 +1276,8 @@ public:
                 uint16_t prog_size,
                 uint16_t max_pkt_offset,
                 uint16_t prefix_size,
-                bool a_is_pkt_size_var
+                bool a_is_pkt_size_var,
+                bool a_is_random_seed
                 ){
 
         if (bss) {
@@ -1303,6 +1304,7 @@ public:
         m_max_pkt_offset_change = max_pkt_offset;
         m_prefix_size = prefix_size;
         m_is_pkt_size_var=a_is_pkt_size_var;
+        m_is_random_seed=a_is_random_seed;
     }
 
     ~StreamVmDp(){
@@ -1325,7 +1327,8 @@ public:
                                          m_program_size,
                                          m_max_pkt_offset_change,
                                          m_prefix_size,
-                                         m_is_pkt_size_var
+                                         m_is_pkt_size_var,
+                                         m_is_random_seed
                                          );
         assert(lp);
         return (lp);
@@ -1373,6 +1376,9 @@ public:
     bool is_pkt_size_var(){
         return (m_is_pkt_size_var);
     }
+    bool is_random_seed(){
+        return (m_is_random_seed);
+    }
 
 
 private:
@@ -1383,6 +1389,7 @@ private:
     uint16_t   m_max_pkt_offset_change;
     uint16_t   m_prefix_size;
     bool       m_is_pkt_size_var;
+    bool       m_is_random_seed;
 
 };
 
@@ -1445,7 +1452,8 @@ public:
                                         get_dp_instruction_buffer()->get_program_size(),
                                         get_max_packet_update_offset(),
                                         get_prefix_size(),
-                                        is_var_pkt_size()
+                                        is_var_pkt_size(),
+                                        m_is_random_var
                                         );
         assert(lp);
         return (lp);
diff --git a/src/stateless/dp/trex_stateless_dp_core.cpp b/src/stateless/dp/trex_stateless_dp_core.cpp
index 39080453..549f923f 100644
--- a/src/stateless/dp/trex_stateless_dp_core.cpp
+++ b/src/stateless/dp/trex_stateless_dp_core.cpp
@@ -75,6 +75,13 @@ void CGenNodeStateless::refresh_vm_bss(){
         StreamVmDp  * vm_s=m_ref_stream_info->m_vm_dp;
         assert(vm_s);
         memcpy(m_vm_flow_var,vm_s->get_bss(),vm_s->get_bss_size());
+
+        if ( vm_s->is_random_seed() ){
+            /* if we have random seed for this program */
+            if (m_ref_stream_info->m_random_seed) {
+                set_random_seed(m_ref_stream_info->m_random_seed);
+            }
+        }
     }
 }
 
@@ -670,6 +677,14 @@ TrexStatelessDpCore::add_stream(TrexStatelessDpPerPort * lp_port,
         node->m_vm_program_size  = lpDpVm->get_program_size();
 
 
+        /* set the random seed if was set */
+        if ( lpDpVm->is_random_seed() ){
+            /* if we have random seed for this program */
+            if (stream->m_random_seed) {
+                node->set_random_seed(stream->m_random_seed);
+            }
+        }
+
         /* we need to copy the object */
         if ( pkt_size > lpDpVm->get_prefix_size() ) {
             /* we need const packet */
diff --git a/src/stateless/dp/trex_stream_node.h b/src/stateless/dp/trex_stream_node.h
index ab43defe..b366a770 100644
--- a/src/stateless/dp/trex_stream_node.h
+++ b/src/stateless/dp/trex_stream_node.h
@@ -116,6 +116,15 @@ private:
 
 public:
 
+    void set_random_seed(uint32_t seed){
+        uint32_t *p=get_random_bss_seed_memory();
+        *p=seed;
+    }
+
+    uint32_t* get_random_bss_seed_memory(){
+        return (uint32_t*)m_vm_flow_var;/* always the first 4 bytes */
+    }
+
     uint8_t             get_port_id(){
         return (m_port_id);
     }