Skip to content

fix: prevent duplicate connections to multiaddr #2734

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 8 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 29 additions & 1 deletion packages/libp2p/src/connection-manager/dial-queue.ts
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@
interface DialQueueJobOptions extends PriorityQueueJobOptions, ProgressOptions<OpenConnectionProgressEvents> {
peerId?: PeerId
multiaddrs: Set<string>
force?: boolean
}

interface DialerInit {
Expand Down Expand Up @@ -135,11 +136,12 @@
*/
async dial (peerIdOrMultiaddr: PeerId | Multiaddr | Multiaddr[], options: OpenConnectionOptions = {}): Promise<Connection> {
const { peerId, multiaddrs } = getPeerAddress(peerIdOrMultiaddr)
const { force } = options

// make sure we don't have an existing connection to any of the addresses we
// are about to dial
const existingConnection = Array.from(this.connections.values()).flat().find(conn => {
if (options.force === true) {
if (force === true) {
return false
}

Expand Down Expand Up @@ -220,6 +222,7 @@
peerId,
priority: options.priority ?? DEFAULT_DIAL_PRIORITY,
multiaddrs: new Set(multiaddrs.map(ma => ma.toString())),
force,
signal: options.signal ?? AbortSignal.timeout(this.dialTimeout),
onProgress: options.onProgress
})
Expand Down Expand Up @@ -313,6 +316,31 @@
this.log.error('could not update last dial failure key for %p', peerId, err)
}

const { remotePeer } = conn

// make sure we don't have an existing connection to the address we dialed
const existingConnection = Array.from(this.connections.values()).flat().find(_conn => {
if (options.force === true) {
return false
}

Check warning on line 325 in packages/libp2p/src/connection-manager/dial-queue.ts

View check run for this annotation

Codecov / codecov/patch

packages/libp2p/src/connection-manager/dial-queue.ts#L324-L325

Added lines #L324 - L325 were not covered by tests

if (_conn.remotePeer.equals(remotePeer) && _conn !== conn) {
return true
}

return false

Check warning on line 331 in packages/libp2p/src/connection-manager/dial-queue.ts

View check run for this annotation

Codecov / codecov/patch

packages/libp2p/src/connection-manager/dial-queue.ts#L330-L331

Added lines #L330 - L331 were not covered by tests
})

if (existingConnection?.status === 'open') {
this.log('already connected to %a', existingConnection.remoteAddr)
options?.onProgress?.(new CustomProgressEvent('dial-queue:already-connected'))

this.log('closing duplicate connection to %p', remotePeer)
await conn.close()

return existingConnection
}

// dial successful, return the connection
return conn
} catch (err: any) {
Expand Down
37 changes: 37 additions & 0 deletions packages/libp2p/test/connection-manager/dial-queue.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import { generateKeyPair } from '@libp2p/crypto/keys'
import { NotFoundError } from '@libp2p/interface'
import { peerLogger } from '@libp2p/logger'
import { PeerMap } from '@libp2p/peer-collections'
import { peerIdFromPrivateKey } from '@libp2p/peer-id'
import { multiaddr } from '@multiformats/multiaddr'
import { TCP, WebRTC } from '@multiformats/multiaddr-matcher'
Expand Down Expand Up @@ -381,6 +382,42 @@ describe('dial queue', () => {
await expect(dialer.dial(remotePeer)).to.eventually.equal(connection)
})

it('should return existing connection when dialing a multiaddr without a peer id', async () => {
const remotePeer = peerIdFromPrivateKey(await generateKeyPair('Ed25519'))
const ip = multiaddr('/ip4/123.123.123.123')
const addr1 = ip.encapsulate('/tcp/123')
const addr2 = ip.encapsulate('/tcp/321')

const existingConnection = stubInterface<Connection>({
limits: {
bytes: 100n
},
remotePeer,
remoteAddr: addr1.encapsulate(`/p2p/${remotePeer}`),
status: 'open'
})

const newConnection = stubInterface<Connection>({
limits: {
bytes: 100n
},
remotePeer,
remoteAddr: addr2.encapsulate(`/p2p/${remotePeer}`),
status: 'open'
})

const connections = new PeerMap<Connection[]>()
connections.set(remotePeer, [existingConnection])

components.transportManager.dialTransportForMultiaddr.callsFake(ma => {
return stubInterface<Transport>()
})
components.transportManager.dial.callsFake(async (ma, opts = {}) => newConnection)
dialer = new DialQueue(components, { connections })

await expect(dialer.dial(addr2)).to.eventually.equal(existingConnection)
})

it('should respect user dial signal over default timeout if it is passed', async () => {
const dialTimeout = 10
const userTimeout = 500
Expand Down