chore: fix connectedness

dht.go

@@ -737,7 +737,8 @@ func (dht *IpfsDHT) FindLocal(ctx context.Context, id peer.ID) peer.AddrInfo {
 	_, span := internal.StartSpan(ctx, "IpfsDHT.FindLocal", trace.WithAttributes(attribute.Stringer("PeerID", id)))
 	defer span.End()
 
-	if dht.host.Network().Connectedness(id) == network.Connected {
+	connectedness := dht.host.Network().Connectedness(id)
+	if connectedness == network.Connected || connectedness == network.Limited {
 		return dht.peerstore.PeerInfo(id)
 	}
 	return peer.AddrInfo{}
@@ -926,7 +927,8 @@ func (dht *IpfsDHT) newContextWithLocalTags(ctx context.Context, extraTags ...ta
 
 func (dht *IpfsDHT) maybeAddAddrs(p peer.ID, addrs []ma.Multiaddr, ttl time.Duration) {
 	// Don't add addresses for self or our connected peers. We have better ones.
-	if p == dht.self || dht.host.Network().Connectedness(p) == network.Connected {
+	connectedness := dht.host.Network().Connectedness(p)
+	if p == dht.self || connectedness == network.Connected || connectedness == network.Limited {
 		return
 	}
 	dht.peerstore.AddAddrs(p, dht.filterAddrs(addrs), ttl)

fullrt/dht.go

@@ -1460,7 +1460,7 @@
 	// Return peer information if we tried to dial the peer during the query or we are (or recently were) connected
 	// to the peer.
 	connectedness := dht.h.Network().Connectedness(id)
-	if connectedness == network.Connected {
+	if connectedness == network.Connected || connectedness == network.Limited {
 		return dht.h.Peerstore().PeerInfo(id), nil
 	}
 
@@ -1538,15 +1538,17 @@
 
 // FindLocal looks for a peer with a given ID connected to this dht and returns the peer and the table it was found in.
 func (dht *FullRT) FindLocal(id peer.ID) peer.AddrInfo {
-	if dht.h.Network().Connectedness(id) == network.Connected {
+	connectedness := dht.h.Network().Connectedness(id)
+	if connectedness == network.Connected || connectedness == network.Limited {
 		return dht.h.Peerstore().PeerInfo(id)
 	}
 	return peer.AddrInfo{}
 }
 
 func (dht *FullRT) maybeAddAddrs(p peer.ID, addrs []multiaddr.Multiaddr, ttl time.Duration) {
 	// Don't add addresses for self or our connected peers. We have better ones.
-	if p == dht.h.ID() || dht.h.Network().Connectedness(p) == network.Connected {
+	connectedness := dht.h.Network().Connectedness(p)
+	if p == dht.h.ID() || connectedness == network.Connected || connectedness == network.Limited {
 		return
 	}
 	dht.h.Peerstore().AddAddrs(p, addrs, ttl)

query_test.go

@@ -34,18 +34,18 @@ func TestRTEvictionOnFailedQuery(t *testing.T) {
 	// peers should be in the RT because of fixLowPeers
 	require.NoError(t, tu.WaitFor(ctx, func() error {
 		if !checkRoutingTable(d1, d2) {
-			return fmt.Errorf("should have routes 0")
+			return fmt.Errorf("should have routes")
 		}
 		return nil
 	}))
 
-	// close both hosts so query fails
-	require.NoError(t, d1.host.Close())
-	require.NoError(t, d2.host.Close())
-	// peers will still be in the RT because we have decoupled membership from connectivity
+	// clear the addresses of the peers so that the next queries fail
+	d1.host.Peerstore().ClearAddrs(d2.self)
+	d2.host.Peerstore().ClearAddrs(d1.self)
+	// peers will still be in the RT because RT is decoupled with the host and peerstore
 	require.NoError(t, tu.WaitFor(ctx, func() error {
 		if !checkRoutingTable(d1, d2) {
-			return fmt.Errorf("should have routes 1")
+			return fmt.Errorf("should have routes")
 		}
 		return nil
 	}))
@@ -59,7 +59,7 @@ func TestRTEvictionOnFailedQuery(t *testing.T) {
 
 	require.NoError(t, tu.WaitFor(ctx, func() error {
 		if checkRoutingTable(d1, d2) {
-			return fmt.Errorf("should not have routes 2")
+			return fmt.Errorf("should not have routes")
 		}
 		return nil
 	}))

routing.go

@@ -664,7 +664,8 @@ func (dht *IpfsDHT) FindPeer(ctx context.Context, id peer.ID) (pi peer.AddrInfo,
 			return peers, err
 		},
 		func(*qpeerset.QueryPeerset) bool {
-			return dht.host.Network().Connectedness(id) == network.Connected
+			connectedness := dht.host.Network().Connectedness(id)
+			return connectedness == network.Connected || connectedness == network.Limited
 		},
 	)
 
@@ -686,7 +687,7 @@ func (dht *IpfsDHT) FindPeer(ctx context.Context, id peer.ID) (pi peer.AddrInfo,
 	// Return peer information if we tried to dial the peer during the query or we are (or recently were) connected
 	// to the peer.
 	connectedness := dht.host.Network().Connectedness(id)
-	if dialedPeerDuringQuery || connectedness == network.Connected {
+	if dialedPeerDuringQuery || connectedness == network.Connected || connectedness == network.Limited {
 		return dht.peerstore.PeerInfo(id), nil
 	}