tools: serial_throughput: set time before test

Set the remote time before running the throughput test to avoid the
remote regenerating keys on every packet sent and received.

Signed-off-by: Jordan Yates <jordan@embeint.com>

Author: JordanYates

src/infuse_iot/tools/serial_throughput.py

@@ -8,16 +8,20 @@
 import random
 import sys
 import time
+from datetime import datetime
 
 from infuse_iot.commands import InfuseCommand
 from infuse_iot.common import InfuseID, InfuseType
+from infuse_iot.definitions.rpc import time_set
 from infuse_iot.epacket.packet import Auth, PacketOutput
+from infuse_iot.rpc_client import RpcClient
 from infuse_iot.socket_comms import (
     ClientNotificationEpacketReceived,
     GatewayRequestEpacketSend,
     LocalClient,
     default_multicast_address,
 )
+from infuse_iot.time import InfuseTime
 
 
 class SubCommand(InfuseCommand):
@@ -38,6 +42,11 @@ def __init__(self, args):
         self._client = LocalClient(default_multicast_address(), 1.0)
         self._iterations = args.iterations
 
+    def run_time_set(self):
+        rpc_client = RpcClient(self._client, 128, InfuseID.GATEWAY)
+        params = time_set.request(InfuseTime.epoch_time_from_unix(datetime.now().timestamp()))
+        rpc_client.run_standard_cmd(time_set.COMMAND_ID, Auth.DEVICE, bytes(params), time_set.response.from_buffer_copy)
+
     def run_send_test(self, num, size, queue_size):
         assert size >= 4
         self._client.set_rx_timeout(0.2)
@@ -85,6 +94,9 @@ def run(self):
         if not self._client.comms_check():
             sys.exit("No communications gateway detected (infuse gateway/bt_native)")
 
+        # Set remote time to avoid key regeneration on every packet
+        self.run_time_set()
+
         # No queuing
         print(f"Averaged across {self._iterations} packets with no queuing:")
         self.run_send_test(self._iterations, 4, 1)