ipcpd: Wait for DHT_RUNNING state at register/query0.9.4

This will make the register and query calls at the DHT wait for a JOIN
to complete. This avoids fails when calling register immediately after
creating a data transfer connection.

Signed-off-by: Dimitri Staessens <[email protected]>
Signed-off-by: Sander Vrijders <[email protected]>

1 files changed, 10 insertions, 4 deletions

diff --git a/src/ipcpd/normal/dht.c b/src/ipcpd/normal/dht.c
index a59b7781..103ffd1c 100644
--- a/src/ipcpd/normal/dht.c
+++ b/src/ipcpd/normal/dht.c
@@ -72,9 +72,9 @@ typedef KadContactMsg kad_contact_msg_t;
 
 enum dht_state {
         DHT_INIT = 0,
+        DHT_SHUTDOWN,
         DHT_JOINING,
         DHT_RUNNING,
-        DHT_SHUTDOWN,
 };
 
 enum kad_code {
@@ -277,7 +277,7 @@ static void dht_set_state(struct dht *   dht,
 
         dht->state = state;
 
-        pthread_cond_signal(&dht->cond);
+        pthread_cond_broadcast(&dht->cond);
 
         pthread_mutex_unlock(&dht->mtx);
 }
@@ -2245,7 +2245,10 @@ int dht_reg(struct dht *    dht,
         assert(key);
         assert(dht->addr != 0);
 
-        if (dht_get_state(dht) != DHT_RUNNING)
+        if (dht_get_state(dht) < DHT_JOINING)
+                return -1;
+
+        if (dht_wait_running(dht))
                 return -1;
 
         pthread_rwlock_wrlock(&dht->lock);
@@ -2313,7 +2316,10 @@ uint64_t dht_query(struct dht *    dht,
 
         addrs[0] = 0;
 
-        if (dht_get_state(dht) != DHT_RUNNING)
+        if (dht_get_state(dht) < DHT_JOINING)
+                return 0;
+
+        if (dht_wait_running(dht))
                 return 0;
 
         pthread_rwlock_rdlock(&dht->lock);