From 95b98b3845be287952ebdb59ece0cce194127824 Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Mon, 30 Jul 2018 11:10:11 -0700
Subject: [PATCH] Fix --addr option

---
 src/bin/bench-tps.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/bin/bench-tps.rs b/src/bin/bench-tps.rs
index 708cc79238..7d9f1a49bb 100644
--- a/src/bin/bench-tps.rs
+++ b/src/bin/bench-tps.rs
@@ -609,10 +609,15 @@ fn mk_client(r: &NodeInfo) -> ThinClient {
 fn spy_node(addr: Option<String>) -> (NodeInfo, UdpSocket) {
     let gossip_socket_pair;
     if let Some(a) = addr {
+        let gossip_socket = udp_random_bind(8000, 10000, 5).unwrap();
+        let gossip_addr = SocketAddr::new(
+            a.parse().unwrap(),
+            gossip_socket.local_addr().unwrap().port(),
+        );
         gossip_socket_pair = UdpSocketPair {
-            addr: a.parse().unwrap(),
-            receiver: UdpSocket::bind("0.0.0.0:0").unwrap(),
-            sender: UdpSocket::bind("0.0.0.0:0").unwrap(),
+            addr: gossip_addr,
+            receiver: gossip_socket.try_clone().unwrap(),
+            sender: gossip_socket,
         };
     } else {
         gossip_socket_pair = udp_public_bind("gossip", 8000, 10000);