From: Tim Beale <timbeale@catalyst.net.nz>
Date: Thu, 13 Jun 2019 04:04:46 +0000 (+1200)
Subject: traffic_replay: Store total conversations on the replay context
X-Git-Tag: tdb-1.4.2~517
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=9f504fd5a320edcbd3d763d4e0464f3cf7d270e1;p=thirdparty%2Fsamba.git

traffic_replay: Store total conversations on the replay context

This is useful info to know, and will be used in subsequent commits.

Signed-off-by: Tim Beale <timbeale@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
---

diff --git a/python/samba/emulate/traffic.py b/python/samba/emulate/traffic.py
index 7f720c98671..c511fef6557 100644
--- a/python/samba/emulate/traffic.py
+++ b/python/samba/emulate/traffic.py
@@ -355,6 +355,7 @@ class ReplayContext(object):
                  server=None,
                  lp=None,
                  creds=None,
+                 total_conversations=None,
                  badpassword_frequency=None,
                  prefer_kerberos=None,
                  tempdir=None,
@@ -389,6 +390,7 @@ class ReplayContext(object):
         self.last_drsuapi_bad         = False
         self.last_netlogon_bad        = False
         self.last_samlogon_bad        = False
+        self.total_conversations      = total_conversations
         self.generate_ldap_search_tables()
 
     def generate_ldap_search_tables(self):
@@ -1607,6 +1609,7 @@ def replay(conversation_seq,
     context = ReplayContext(server=host,
                             creds=creds,
                             lp=lp,
+                            total_conversations=len(conversation_seq),
                             **kwargs)
 
     if len(accounts) < len(conversation_seq):