mirror of
https://github.com/fluencelabs/js-libp2p-tcp
synced 2025-07-03 07:01:37 +00:00
Compare commits
12 Commits
feat/async
...
test/travi
Author | SHA1 | Date | |
---|---|---|---|
b38b57bdca | |||
fde9cd40d3 | |||
09e5752cb4 | |||
1eca9d656e | |||
330853f470 | |||
9d5ccd17a2 | |||
6bc505d73a | |||
431219db93 | |||
1ffd549dab | |||
776a5246b5 | |||
65c1888f54 | |||
6cf8f71d10 |
1
.gitignore
vendored
1
.gitignore
vendored
@ -12,7 +12,6 @@ logs
|
|||||||
*.log
|
*.log
|
||||||
|
|
||||||
coverage
|
coverage
|
||||||
.nyc_output
|
|
||||||
|
|
||||||
# Runtime data
|
# Runtime data
|
||||||
pids
|
pids
|
||||||
|
20
.travis.yml
20
.travis.yml
@ -1,6 +1,5 @@
|
|||||||
language: node_js
|
language: node_js
|
||||||
cache: npm
|
cache: npm
|
||||||
sudo: false
|
|
||||||
|
|
||||||
stages:
|
stages:
|
||||||
- check
|
- check
|
||||||
@ -9,24 +8,27 @@ stages:
|
|||||||
|
|
||||||
node_js:
|
node_js:
|
||||||
- '10'
|
- '10'
|
||||||
- '12'
|
|
||||||
|
|
||||||
os:
|
|
||||||
- linux
|
|
||||||
- osx
|
|
||||||
- windows
|
|
||||||
|
|
||||||
before_script:
|
|
||||||
- if [ "$TRAVIS_OS_NAME" = "linux" ]; then sudo sh -c 'echo 0 > /proc/sys/net/ipv6/conf/all/disable_ipv6'; fi
|
|
||||||
script: npx nyc -s npm run test:node -- --bail
|
script: npx nyc -s npm run test:node -- --bail
|
||||||
after_success: npx nyc report --reporter=text-lcov > coverage.lcov && npx codecov
|
after_success: npx nyc report --reporter=text-lcov > coverage.lcov && npx codecov
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
include:
|
include:
|
||||||
|
- os: linux
|
||||||
|
sudo: false
|
||||||
|
before_script: sudo sh -c 'echo 0 > /proc/sys/net/ipv6/conf/all/disable_ipv6'
|
||||||
|
|
||||||
|
- os: windows
|
||||||
|
filter_secrets: false
|
||||||
|
cache: false
|
||||||
|
|
||||||
|
- os: osx
|
||||||
|
|
||||||
- stage: check
|
- stage: check
|
||||||
os: linux
|
os: linux
|
||||||
script:
|
script:
|
||||||
- npx aegir build --bundlesize
|
- npx aegir build --bundlesize
|
||||||
|
- npx aegir commitlint --travis
|
||||||
- npx aegir dep-check
|
- npx aegir dep-check
|
||||||
- npm run lint
|
- npm run lint
|
||||||
|
|
||||||
|
@ -1,8 +1,3 @@
|
|||||||
<a name="0.13.1"></a>
|
|
||||||
## [0.13.1](https://github.com/libp2p/js-libp2p-tcp/compare/v0.13.0...v0.13.1) (2019-08-08)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
<a name="0.13.0"></a>
|
<a name="0.13.0"></a>
|
||||||
# [0.13.0](https://github.com/libp2p/js-libp2p-tcp/compare/v0.12.1...v0.13.0) (2018-09-12)
|
# [0.13.0](https://github.com/libp2p/js-libp2p-tcp/compare/v0.12.1...v0.13.0) (2018-09-12)
|
||||||
|
|
||||||
|
14
README.md
14
README.md
@ -1,13 +1,13 @@
|
|||||||
# js-libp2p-tcp
|
# js-libp2p-tcp
|
||||||
|
|
||||||
[](http://protocol.ai)
|
[](http://ipn.io)
|
||||||
[](http://libp2p.io/)
|
[](http://webchat.freenode.net/?channels=%23ipfs)
|
||||||
[](http://webchat.freenode.net/?channels=%23libp2p)
|
[](https://travis-ci.org/libp2p/js-libp2p-tcp)
|
||||||
[](https://discuss.libp2p.io)
|
[](https://coveralls.io/github/libp2p/js-libp2p-tcp?branch=master)
|
||||||
[](https://codecov.io/gh/libp2p/js-libp2p-tcp)
|
|
||||||
[](https://travis-ci.com/libp2p/js-libp2p-tcp)
|
|
||||||
[](https://david-dm.org/libp2p/js-libp2p-tcp)
|
[](https://david-dm.org/libp2p/js-libp2p-tcp)
|
||||||
[](https://github.com/feross/standard)
|
[](https://github.com/feross/standard)
|
||||||
|

|
||||||
|

|
||||||
|
|
||||||
[](https://github.com/libp2p/interface-transport)
|
[](https://github.com/libp2p/interface-transport)
|
||||||
[](https://github.com/libp2p/interface-connection)
|
[](https://github.com/libp2p/interface-connection)
|
||||||
@ -56,7 +56,7 @@ const listener = tcp.createListener((socket) => {
|
|||||||
)
|
)
|
||||||
})
|
})
|
||||||
|
|
||||||
await listener.listen(addr)
|
await listener.listen([addr])
|
||||||
console.log('listening')
|
console.log('listening')
|
||||||
|
|
||||||
const socket = await tcp.dial(addr)
|
const socket = await tcp.dial(addr)
|
||||||
|
29
appveyor.yml
Normal file
29
appveyor.yml
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
# Warning: This file is automatically synced from https://github.com/ipfs/ci-sync so if you want to change it, please change it there and ask someone to sync all repositories.
|
||||||
|
version: "{build}"
|
||||||
|
|
||||||
|
environment:
|
||||||
|
matrix:
|
||||||
|
- nodejs_version: "6"
|
||||||
|
- nodejs_version: "8"
|
||||||
|
|
||||||
|
matrix:
|
||||||
|
fast_finish: true
|
||||||
|
|
||||||
|
install:
|
||||||
|
# Install Node.js
|
||||||
|
- ps: Install-Product node $env:nodejs_version
|
||||||
|
|
||||||
|
# Upgrade npm
|
||||||
|
- npm install -g npm
|
||||||
|
|
||||||
|
# Output our current versions for debugging
|
||||||
|
- node --version
|
||||||
|
- npm --version
|
||||||
|
|
||||||
|
# Install our package dependencies
|
||||||
|
- npm install
|
||||||
|
|
||||||
|
test_script:
|
||||||
|
- npm run test:node
|
||||||
|
|
||||||
|
build: off
|
2
ci/Jenkinsfile
vendored
Normal file
2
ci/Jenkinsfile
vendored
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
// Warning: This file is automatically synced from https://github.com/ipfs/ci-sync so if you want to change it, please change it there and ask someone to sync all repositories.
|
||||||
|
javascript()
|
15
circle.yml
Normal file
15
circle.yml
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
# Warning: This file is automatically synced from https://github.com/ipfs/ci-sync so if you want to change it, please change it there and ask someone to sync all repositories.
|
||||||
|
machine:
|
||||||
|
node:
|
||||||
|
version: stable
|
||||||
|
|
||||||
|
dependencies:
|
||||||
|
pre:
|
||||||
|
- google-chrome --version
|
||||||
|
- curl -L -o google-chrome.deb https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb
|
||||||
|
- sudo dpkg -i google-chrome.deb || true
|
||||||
|
- sudo apt-get update
|
||||||
|
- sudo apt-get install -f
|
||||||
|
- sudo apt-get install --only-upgrade lsb-base
|
||||||
|
- sudo dpkg -i google-chrome.deb
|
||||||
|
- google-chrome --version
|
37
package.json
37
package.json
@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "libp2p-tcp",
|
"name": "libp2p-tcp",
|
||||||
"version": "0.13.1",
|
"version": "0.13.0",
|
||||||
"description": "Node.js implementation of the TCP module that libp2p uses, which implements the interface-connection and interface-transport interfaces",
|
"description": "Node.js implementation of the TCP module that libp2p uses, which implements the interface-connection and interface-transport interfaces",
|
||||||
"leadMaintainer": "Jacob Heun <jacobheun@gmail.com>",
|
"leadMaintainer": "Jacob Heun <jacobheun@gmail.com>",
|
||||||
"main": "src/index.js",
|
"main": "src/index.js",
|
||||||
@ -10,13 +10,15 @@
|
|||||||
"test:node": "aegir test -t node",
|
"test:node": "aegir test -t node",
|
||||||
"build": "aegir build",
|
"build": "aegir build",
|
||||||
"docs": "aegir docs",
|
"docs": "aegir docs",
|
||||||
"release": "aegir release -t node --no-build",
|
"release": "aegir release --docs -t node",
|
||||||
"release-minor": "aegir release -t node --type minor --no-build",
|
"release-minor": "aegir release --type minor --docs -t node",
|
||||||
"release-major": "aegir-release -t node --type major --no-build",
|
"release-major": "aegir release --type major --docs -t node",
|
||||||
"coverage": "nyc --reporter=text --reporter=lcov npm run test:node"
|
"coverage": "aegir coverage",
|
||||||
|
"coverage-publish": "aegir-coverage publish"
|
||||||
},
|
},
|
||||||
"pre-push": [
|
"pre-push": [
|
||||||
"lint"
|
"lint",
|
||||||
|
"test"
|
||||||
],
|
],
|
||||||
"repository": {
|
"repository": {
|
||||||
"type": "git",
|
"type": "git",
|
||||||
@ -35,25 +37,26 @@
|
|||||||
"npm": ">=3.0.0"
|
"npm": ">=3.0.0"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"aegir": "^20.0.0",
|
"aegir": "^18.1.1",
|
||||||
"chai": "^4.2.0",
|
"chai": "^4.1.2",
|
||||||
"dirty-chai": "^2.0.1",
|
"dirty-chai": "^2.0.1",
|
||||||
|
"lodash.isfunction": "^3.0.9",
|
||||||
"pull-stream": "^3.6.9",
|
"pull-stream": "^3.6.9",
|
||||||
"sinon": "^7.4.1"
|
"sinon": "^7.3.1"
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"abortable-iterator": "^2.1.0",
|
"abort-controller": "^3.0.0",
|
||||||
"class-is": "^1.1.0",
|
"class-is": "^1.1.0",
|
||||||
"debug": "^4.1.1",
|
"debug": "^4.1.1",
|
||||||
"err-code": "^1.1.2",
|
"err-code": "^1.1.2",
|
||||||
"interface-connection": "~0.3.3",
|
"interface-connection": "~0.3.3",
|
||||||
"interface-transport": "~0.5.2",
|
"interface-transport": "~0.4.0",
|
||||||
"ip-address": "^6.1.0",
|
"ip-address": "^5.8.9",
|
||||||
"it-pipe": "^1.0.1",
|
"it-pipe": "^1.0.0",
|
||||||
"lodash.includes": "^4.3.0",
|
"lodash.includes": "^4.3.0",
|
||||||
"lodash.isfunction": "^3.0.9",
|
"lodash.isfunction": "^3.0.9",
|
||||||
"mafmt": "^6.0.8",
|
"mafmt": "^6.0.7",
|
||||||
"multiaddr": "^6.1.0",
|
"multiaddr": "^6.0.6",
|
||||||
"streaming-iterables": "^4.1.0"
|
"streaming-iterables": "^4.1.0"
|
||||||
},
|
},
|
||||||
"contributors": [
|
"contributors": [
|
||||||
@ -68,12 +71,10 @@
|
|||||||
"Jacob Heun <jacobheun@gmail.com>",
|
"Jacob Heun <jacobheun@gmail.com>",
|
||||||
"Jacob Heun <jake@andyet.net>",
|
"Jacob Heun <jake@andyet.net>",
|
||||||
"João Antunes <j.goncalo.antunes@gmail.com>",
|
"João Antunes <j.goncalo.antunes@gmail.com>",
|
||||||
"Linus Unnebäck <linus@folkdatorn.se>",
|
|
||||||
"Pedro Teixeira <i@pgte.me>",
|
"Pedro Teixeira <i@pgte.me>",
|
||||||
"Prashanth Chandra <coolshanth94@gmail.com>",
|
"Prashanth Chandra <coolshanth94@gmail.com>",
|
||||||
"Richard Littauer <richard.littauer@gmail.com>",
|
"Richard Littauer <richard.littauer@gmail.com>",
|
||||||
"Stephen Whitmore <stephen.whitmore@gmail.com>",
|
"Stephen Whitmore <stephen.whitmore@gmail.com>",
|
||||||
"TomCoded <tomtinkerer@gmail.com>",
|
"TomCoded <tomtinkerer@gmail.com>"
|
||||||
"Vasco Santos <vasco.santos@moxy.studio>"
|
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
14
src/index.js
14
src/index.js
@ -18,7 +18,7 @@ function noop () {}
|
|||||||
class TCP {
|
class TCP {
|
||||||
async dial (ma, options) {
|
async dial (ma, options) {
|
||||||
const cOpts = ma.toOptions()
|
const cOpts = ma.toOptions()
|
||||||
log('Dialing %s:%s', cOpts.host, cOpts.port)
|
log('Connecting to %s:%s', cOpts.host, cOpts.port)
|
||||||
|
|
||||||
const rawSocket = await this._connect(cOpts, options)
|
const rawSocket = await this._connect(cOpts, options)
|
||||||
return new Libp2pSocket(rawSocket, ma, options)
|
return new Libp2pSocket(rawSocket, ma, options)
|
||||||
@ -34,14 +34,13 @@ class TCP {
|
|||||||
const rawSocket = net.connect(cOpts)
|
const rawSocket = net.connect(cOpts)
|
||||||
|
|
||||||
const onError = (err) => {
|
const onError = (err) => {
|
||||||
const msg = `Error dialing ${cOpts.host}:${cOpts.port}: ${err.message}`
|
const msg = `Error connecting to ${cOpts.host}:${cOpts.port}: ${err.message}`
|
||||||
done(errcode(msg, err.code))
|
done(errcode(msg, err.code))
|
||||||
}
|
}
|
||||||
|
|
||||||
const onTimeout = () => {
|
const onTimeout = () => {
|
||||||
log('Timeout dialing %s:%s', cOpts.host, cOpts.port)
|
log('Timeout connecting to %s:%s', cOpts.host, cOpts.port)
|
||||||
const err = errcode(`Timeout after ${Date.now() - start}ms`, 'ETIMEDOUT')
|
const err = errcode(`Timeout after ${Date.now() - start}ms`, 'ETIMEDOUT')
|
||||||
// Note: this will result in onError() being called
|
|
||||||
rawSocket.emit('error', err)
|
rawSocket.emit('error', err)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -51,7 +50,7 @@ class TCP {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const onAbort = () => {
|
const onAbort = () => {
|
||||||
log('Dial to %s:%s aborted', cOpts.host, cOpts.port)
|
log('Connect to %s:%s aborted', cOpts.host, cOpts.port)
|
||||||
rawSocket.destroy()
|
rawSocket.destroy()
|
||||||
done(new AbortError())
|
done(new AbortError())
|
||||||
}
|
}
|
||||||
@ -60,8 +59,7 @@ class TCP {
|
|||||||
rawSocket.removeListener('error', onError)
|
rawSocket.removeListener('error', onError)
|
||||||
rawSocket.removeListener('timeout', onTimeout)
|
rawSocket.removeListener('timeout', onTimeout)
|
||||||
rawSocket.removeListener('connect', onConnect)
|
rawSocket.removeListener('connect', onConnect)
|
||||||
|
options.signal && options.signal.removeEventListener(onAbort)
|
||||||
options.signal && options.signal.removeEventListener('abort', onAbort)
|
|
||||||
|
|
||||||
err ? reject(err) : resolve(res)
|
err ? reject(err) : resolve(res)
|
||||||
}
|
}
|
||||||
@ -80,7 +78,7 @@ class TCP {
|
|||||||
}
|
}
|
||||||
|
|
||||||
handler = handler || noop
|
handler = handler || noop
|
||||||
return createListener(handler)
|
return createListener(options, handler)
|
||||||
}
|
}
|
||||||
|
|
||||||
filter (multiaddrs) {
|
filter (multiaddrs) {
|
||||||
|
302
src/listener.js
302
src/listener.js
@ -2,26 +2,174 @@
|
|||||||
|
|
||||||
const multiaddr = require('multiaddr')
|
const multiaddr = require('multiaddr')
|
||||||
const os = require('os')
|
const os = require('os')
|
||||||
const includes = require('lodash.includes')
|
|
||||||
const net = require('net')
|
const net = require('net')
|
||||||
const EventEmitter = require('events').EventEmitter
|
const EventEmitter = require('events').EventEmitter
|
||||||
|
const { AllListenersFailedError } = require('interface-transport')
|
||||||
const debug = require('debug')
|
const debug = require('debug')
|
||||||
const log = debug('libp2p:tcp:listen')
|
const log = debug('libp2p:tcp:listen')
|
||||||
const logError = debug('libp2p:tcp:listen:error')
|
log.error = debug('libp2p:tcp:listen:error')
|
||||||
|
|
||||||
const Libp2pSocket = require('./socket')
|
const Libp2pSocket = require('./socket')
|
||||||
const getMultiaddr = require('./get-multiaddr')
|
const getMultiaddr = require('./get-multiaddr')
|
||||||
const c = require('./constants')
|
const c = require('./constants')
|
||||||
|
|
||||||
module.exports = (handler) => {
|
function noop () {}
|
||||||
const listener = new EventEmitter()
|
|
||||||
|
|
||||||
const server = net.createServer((socket) => {
|
class Listener extends EventEmitter {
|
||||||
// Avoid uncaught errors caused by unstable connections
|
/**
|
||||||
socket.on('error', (err) => {
|
*
|
||||||
logError('Error emitted by server handler socket: ' + err.message)
|
* @param {object} options
|
||||||
|
* @param {function(Connection)} handler
|
||||||
|
*/
|
||||||
|
constructor (options, handler) {
|
||||||
|
super()
|
||||||
|
this._options = options
|
||||||
|
this._connectionHandler = handler
|
||||||
|
this._servers = new Set()
|
||||||
|
this.__connections = new Map()
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Closes all open servers
|
||||||
|
* @param {object} options
|
||||||
|
* @param {number} options.timeout how long before closure is forced, defaults to 2000 ms
|
||||||
|
* @returns {Promise}
|
||||||
|
*/
|
||||||
|
close (options = {}) {
|
||||||
|
if ([...this._servers].filter(server => server.listening).length === 0) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close all running servers in parallel
|
||||||
|
return Promise.all(
|
||||||
|
[...this._servers].map(server => {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const start = Date.now()
|
||||||
|
|
||||||
|
// Attempt to stop the server. If it takes longer than the timeout,
|
||||||
|
// destroy all the underlying sockets manually.
|
||||||
|
const timeout = setTimeout(() => {
|
||||||
|
log('Timeout closing server after %dms, destroying connections manually', Date.now() - start)
|
||||||
|
resolve()
|
||||||
|
}, options.timeout || c.CLOSE_TIMEOUT)
|
||||||
|
|
||||||
|
server.once('close', () => {
|
||||||
|
clearTimeout(timeout)
|
||||||
|
this._servers.delete(server)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
server.close((err) => err ? reject(err) : resolve())
|
||||||
|
})
|
||||||
|
})
|
||||||
|
).then(() => {
|
||||||
|
this.__connections.forEach((connection, key) => {
|
||||||
|
log('destroying %s', key)
|
||||||
|
connection.destroy()
|
||||||
|
})
|
||||||
|
this.__connections.clear()
|
||||||
|
this._servers.clear()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates servers listening on the given `addrs`
|
||||||
|
* @async
|
||||||
|
* @param {Array<Multiaddr>} addrs
|
||||||
|
*/
|
||||||
|
async listen (addrs) {
|
||||||
|
addrs = Array.isArray(addrs) ? addrs : [addrs]
|
||||||
|
|
||||||
|
let listeners = []
|
||||||
|
let errors = []
|
||||||
|
|
||||||
|
// Filter out duplicate ports, unless it's port 0
|
||||||
|
addrs = uniqueBy(addrs, (addr) => {
|
||||||
|
const port = Number(addr.toOptions().port)
|
||||||
|
return isNaN(port) || port === 0 ? addr.toString() : port
|
||||||
|
})
|
||||||
|
|
||||||
|
for (const ma of addrs) {
|
||||||
|
const lOpts = ma.toOptions()
|
||||||
|
|
||||||
|
listeners.push(
|
||||||
|
new Promise((resolve) => {
|
||||||
|
const server = net.createServer(this._onSocket.bind(this))
|
||||||
|
this._servers.add(server)
|
||||||
|
// TODO: clean these up
|
||||||
|
server.on('listening', () => this.emit('listening'))
|
||||||
|
server.on('close', () => this.emit('close'))
|
||||||
|
server.on('error', (err) => this.emit('error', err))
|
||||||
|
|
||||||
|
server.listen(lOpts.port, lOpts.host, (err) => {
|
||||||
|
if (err) {
|
||||||
|
errors.push(err)
|
||||||
|
return resolve()
|
||||||
|
}
|
||||||
|
|
||||||
|
log('Listening on %s %s', lOpts.port, lOpts.host)
|
||||||
|
resolve()
|
||||||
|
})
|
||||||
|
})
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
return Promise.all(listeners)
|
||||||
|
.then(() => {
|
||||||
|
errors.forEach((err) => {
|
||||||
|
log.error('received an error while attempting to listen', err)
|
||||||
|
})
|
||||||
|
|
||||||
|
// All servers failed to listen, throw an error
|
||||||
|
if (errors.length === listeners.length) {
|
||||||
|
throw new AllListenersFailedError()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Return the addresses we are listening on
|
||||||
|
* @returns {Array<Multiaddr>}
|
||||||
|
*/
|
||||||
|
getAddrs () {
|
||||||
|
const multiaddrs = []
|
||||||
|
this._servers.forEach(server => {
|
||||||
|
const address = server.address()
|
||||||
|
|
||||||
|
if (address.address === '0.0.0.0') {
|
||||||
|
const netInterfaces = os.networkInterfaces()
|
||||||
|
Object.keys(netInterfaces).forEach((niKey) => {
|
||||||
|
netInterfaces[niKey].forEach((ni) => {
|
||||||
|
if (ni.internal === false && ni.family === address.family) {
|
||||||
|
multiaddrs.push(
|
||||||
|
multiaddr.fromNodeAddress({
|
||||||
|
...address,
|
||||||
|
address: ni.address
|
||||||
|
}, 'tcp')
|
||||||
|
)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
})
|
||||||
|
// TODO: handle IPv6 wildcard
|
||||||
|
} else {
|
||||||
|
multiaddrs.push(multiaddr.fromNodeAddress(address, 'tcp'))
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
if (multiaddrs.length === 0) {
|
||||||
|
throw new Error('Listener is not ready yet')
|
||||||
|
}
|
||||||
|
|
||||||
|
return multiaddrs
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handler for new sockets from `net.createServer`
|
||||||
|
* @param {net.Socket} socket
|
||||||
|
*/
|
||||||
|
_onSocket (socket) {
|
||||||
|
// Avoid uncaught errors caused by unstable connections
|
||||||
|
socket.on('error', noop)
|
||||||
|
|
||||||
const addr = getMultiaddr(socket)
|
const addr = getMultiaddr(socket)
|
||||||
if (!addr) {
|
if (!addr) {
|
||||||
if (socket.remoteAddress === undefined) {
|
if (socket.remoteAddress === undefined) {
|
||||||
@ -35,124 +183,30 @@ module.exports = (handler) => {
|
|||||||
log('new connection', addr.toString())
|
log('new connection', addr.toString())
|
||||||
|
|
||||||
const s = new Libp2pSocket(socket, addr)
|
const s = new Libp2pSocket(socket, addr)
|
||||||
trackSocket(server, socket)
|
|
||||||
|
|
||||||
handler && handler(s)
|
// Track the connection
|
||||||
listener.emit('connection', s)
|
|
||||||
})
|
|
||||||
|
|
||||||
server.on('listening', () => listener.emit('listening'))
|
|
||||||
server.on('error', (err) => listener.emit('error', err))
|
|
||||||
server.on('close', () => listener.emit('close'))
|
|
||||||
|
|
||||||
// Keep track of open connections to destroy in case of timeout
|
|
||||||
server.__connections = {}
|
|
||||||
|
|
||||||
listener.close = (options = {}) => {
|
|
||||||
if (!server.listening) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
return new Promise((resolve, reject) => {
|
|
||||||
const start = Date.now()
|
|
||||||
|
|
||||||
// Attempt to stop the server. If it takes longer than the timeout,
|
|
||||||
// destroy all the underlying sockets manually.
|
|
||||||
const timeout = setTimeout(() => {
|
|
||||||
log('Timeout closing server after %dms, destroying connections manually', Date.now() - start)
|
|
||||||
Object.keys(server.__connections).forEach((key) => {
|
|
||||||
log('destroying %s', key)
|
|
||||||
server.__connections[key].destroy()
|
|
||||||
})
|
|
||||||
resolve()
|
|
||||||
}, options.timeout || c.CLOSE_TIMEOUT)
|
|
||||||
|
|
||||||
server.once('close', () => clearTimeout(timeout))
|
|
||||||
|
|
||||||
server.close((err) => err ? reject(err) : resolve())
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
let ipfsId
|
|
||||||
let listeningAddr
|
|
||||||
|
|
||||||
listener.listen = (ma) => {
|
|
||||||
listeningAddr = ma
|
|
||||||
if (includes(ma.protoNames(), 'ipfs')) {
|
|
||||||
ipfsId = getIpfsId(ma)
|
|
||||||
listeningAddr = ma.decapsulate('ipfs')
|
|
||||||
}
|
|
||||||
|
|
||||||
const lOpts = listeningAddr.toOptions()
|
|
||||||
return new Promise((resolve, reject) => {
|
|
||||||
server.listen(lOpts.port, lOpts.host, (err) => {
|
|
||||||
if (err) {
|
|
||||||
return reject(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
log('Listening on %s %s', lOpts.port, lOpts.host)
|
|
||||||
resolve()
|
|
||||||
})
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
listener.getAddrs = () => {
|
|
||||||
const multiaddrs = []
|
|
||||||
const address = server.address()
|
|
||||||
|
|
||||||
if (!address) {
|
|
||||||
throw new Error('Listener is not ready yet')
|
|
||||||
}
|
|
||||||
|
|
||||||
// Because TCP will only return the IPv6 version
|
|
||||||
// we need to capture from the passed multiaddr
|
|
||||||
if (listeningAddr.toString().indexOf('ip4') !== -1) {
|
|
||||||
let m = listeningAddr.decapsulate('tcp')
|
|
||||||
m = m.encapsulate('/tcp/' + address.port)
|
|
||||||
if (ipfsId) {
|
|
||||||
m = m.encapsulate('/ipfs/' + ipfsId)
|
|
||||||
}
|
|
||||||
|
|
||||||
if (m.toString().indexOf('0.0.0.0') !== -1) {
|
|
||||||
const netInterfaces = os.networkInterfaces()
|
|
||||||
Object.keys(netInterfaces).forEach((niKey) => {
|
|
||||||
netInterfaces[niKey].forEach((ni) => {
|
|
||||||
if (ni.family === 'IPv4') {
|
|
||||||
multiaddrs.push(multiaddr(m.toString().replace('0.0.0.0', ni.address)))
|
|
||||||
}
|
|
||||||
})
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
multiaddrs.push(m)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (address.family === 'IPv6') {
|
|
||||||
let ma = multiaddr('/ip6/' + address.address + '/tcp/' + address.port)
|
|
||||||
if (ipfsId) {
|
|
||||||
ma = ma.encapsulate('/ipfs/' + ipfsId)
|
|
||||||
}
|
|
||||||
|
|
||||||
multiaddrs.push(ma)
|
|
||||||
}
|
|
||||||
|
|
||||||
return multiaddrs
|
|
||||||
}
|
|
||||||
|
|
||||||
return listener
|
|
||||||
}
|
|
||||||
|
|
||||||
function getIpfsId (ma) {
|
|
||||||
return ma.stringTuples().filter((tuple) => {
|
|
||||||
return tuple[0] === c.IPFS_MA_CODE
|
|
||||||
})[0][1]
|
|
||||||
}
|
|
||||||
|
|
||||||
function trackSocket (server, socket) {
|
|
||||||
const key = `${socket.remoteAddress}:${socket.remotePort}`
|
const key = `${socket.remoteAddress}:${socket.remotePort}`
|
||||||
server.__connections[key] = socket
|
this.__connections.set(key, socket)
|
||||||
|
|
||||||
socket.once('close', () => {
|
socket.once('close', () => {
|
||||||
delete server.__connections[key]
|
this.__connections.delete(key)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
this._connectionHandler(s)
|
||||||
|
this.emit('connection', s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = (options, handler) => {
|
||||||
|
return new Listener(options, handler)
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Get unique values from `arr` using `getValue` to determine
|
||||||
|
* what is used for uniqueness
|
||||||
|
* @param {Array} arr The array to get unique values for
|
||||||
|
* @param {function(value)} getValue The function to determine what is compared
|
||||||
|
* @returns {Array}
|
||||||
|
*/
|
||||||
|
function uniqueBy (arr, getValue) {
|
||||||
|
return [...new Map(arr.map((i) => [getValue(i), i])).values()]
|
||||||
}
|
}
|
||||||
|
@ -1,47 +1,27 @@
|
|||||||
'use strict'
|
'use strict'
|
||||||
|
|
||||||
const abortable = require('abortable-iterator')
|
|
||||||
const debug = require('debug')
|
const debug = require('debug')
|
||||||
const log = debug('libp2p:tcp:socket')
|
const log = debug('libp2p:tcp:socket')
|
||||||
|
|
||||||
const c = require('./constants')
|
const c = require('./constants')
|
||||||
|
|
||||||
class Libp2pSocket {
|
class Libp2pSocket {
|
||||||
constructor (rawSocket, ma, opts = {}) {
|
constructor (rawSocket, ma, opts) {
|
||||||
this._rawSocket = rawSocket
|
this._rawSocket = rawSocket
|
||||||
this._ma = ma
|
this._ma = ma
|
||||||
|
|
||||||
this.sink = this._sink(opts)
|
this.sink = this._sink(opts)
|
||||||
this.source = opts.signal ? abortable(rawSocket, opts.signal) : rawSocket
|
this.source = rawSocket
|
||||||
}
|
}
|
||||||
|
|
||||||
_sink (opts) {
|
_sink (opts = {}) {
|
||||||
// By default, close when the source is exhausted
|
// By default, close when the source is exhausted
|
||||||
const closeOnEnd = opts.closeOnEnd !== false
|
const closeOnEnd = opts.closeOnEnd !== false
|
||||||
|
return (source) => this._write(source, closeOnEnd)
|
||||||
return async (source) => {
|
|
||||||
try {
|
|
||||||
const src = opts.signal ? abortable(source, opts.signal) : source
|
|
||||||
await this._write(src, closeOnEnd)
|
|
||||||
} catch (err) {
|
|
||||||
// If the connection is aborted just close the socket
|
|
||||||
if (err.type === 'aborted') {
|
|
||||||
return this.close()
|
|
||||||
}
|
|
||||||
|
|
||||||
throw err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
async _write (source, closeOnEnd) {
|
async _write (source, closeOnEnd) {
|
||||||
for await (const data of source) {
|
for await (const data of source) {
|
||||||
if (this._rawSocket.destroyed) {
|
|
||||||
const cOpts = this._ma.toOptions()
|
|
||||||
throw new Error('Cannot write %d bytes to destroyed socket %s:%s',
|
|
||||||
data.length, cOpts.host, cOpts.port)
|
|
||||||
}
|
|
||||||
|
|
||||||
const flushed = this._rawSocket.write(data)
|
const flushed = this._rawSocket.write(data)
|
||||||
if (!flushed) {
|
if (!flushed) {
|
||||||
await new Promise((resolve) => this._rawSocket.once('drain', resolve))
|
await new Promise((resolve) => this._rawSocket.once('drain', resolve))
|
||||||
|
@ -114,14 +114,14 @@ describe('listen', () => {
|
|||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
it('getAddrs preserves IPFS Id', (done) => {
|
it('getAddrs does not preserve IPFS Id', (done) => {
|
||||||
const mh = multiaddr('/ip4/127.0.0.1/tcp/9090/ipfs/Qmb6owHp6eaWArVbcJJbQSyifyJBttMMjYV76N2hMbf5Vw')
|
const mh = multiaddr('/ip4/127.0.0.1/tcp/9090/ipfs/Qmb6owHp6eaWArVbcJJbQSyifyJBttMMjYV76N2hMbf5Vw')
|
||||||
const listener = tcp.createListener((conn) => {})
|
const listener = tcp.createListener((conn) => {})
|
||||||
listener.listen(mh, () => {
|
listener.listen(mh, () => {
|
||||||
listener.getAddrs((err, multiaddrs) => {
|
listener.getAddrs((err, multiaddrs) => {
|
||||||
expect(err).to.not.exist()
|
expect(err).to.not.exist()
|
||||||
expect(multiaddrs.length).to.equal(1)
|
expect(multiaddrs.length).to.equal(1)
|
||||||
expect(multiaddrs[0]).to.deep.equal(mh)
|
expect(multiaddrs[0]).to.deep.equal(mh.decapsulate('ipfs'))
|
||||||
listener.close(done)
|
listener.close(done)
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
@ -38,7 +38,7 @@ describe('listen', () => {
|
|||||||
|
|
||||||
await new Promise((resolve) => {
|
await new Promise((resolve) => {
|
||||||
socket1.on('connect', async () => {
|
socket1.on('connect', async () => {
|
||||||
await listener.close({ timeout: 100 })
|
await listener.close()
|
||||||
resolve()
|
resolve()
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
@ -115,14 +115,14 @@ describe('listen', () => {
|
|||||||
await listener.close()
|
await listener.close()
|
||||||
})
|
})
|
||||||
|
|
||||||
it('getAddrs preserves IPFS Id', async () => {
|
it('getAddrs does not preserve IPFS Id', async () => {
|
||||||
const mh = multiaddr('/ip4/127.0.0.1/tcp/9090/ipfs/Qmb6owHp6eaWArVbcJJbQSyifyJBttMMjYV76N2hMbf5Vw')
|
const mh = multiaddr('/ip4/127.0.0.1/tcp/9090/ipfs/Qmb6owHp6eaWArVbcJJbQSyifyJBttMMjYV76N2hMbf5Vw')
|
||||||
const listener = tcp.createListener((conn) => {})
|
const listener = tcp.createListener((conn) => {})
|
||||||
await listener.listen(mh)
|
await listener.listen(mh)
|
||||||
|
|
||||||
const multiaddrs = listener.getAddrs()
|
const multiaddrs = listener.getAddrs()
|
||||||
expect(multiaddrs.length).to.equal(1)
|
expect(multiaddrs.length).to.equal(1)
|
||||||
expect(multiaddrs[0]).to.deep.equal(mh)
|
expect(multiaddrs[0]).to.deep.equal(mh.decapsulate('ipfs'))
|
||||||
|
|
||||||
await listener.close()
|
await listener.close()
|
||||||
})
|
})
|
||||||
@ -221,7 +221,7 @@ describe('dial', () => {
|
|||||||
|
|
||||||
await listener.listen(ma)
|
await listener.listen(ma)
|
||||||
const addrs = listener.getAddrs()
|
const addrs = listener.getAddrs()
|
||||||
await pipe(await tcp.dial(addrs[0]))
|
await pipe([], await tcp.dial(addrs[0]))
|
||||||
|
|
||||||
await handledPromise
|
await handledPromise
|
||||||
await listener.close()
|
await listener.close()
|
||||||
|
Reference in New Issue
Block a user