diff --git a/lib/net/pool.js b/lib/net/pool.js index cc97fa16..dcc5586b 100644 --- a/lib/net/pool.js +++ b/lib/net/pool.js @@ -1075,14 +1075,14 @@ Pool.prototype.handlePacket = co(function* handlePacket(peer, packet) { * @param {Peer} peer */ -Pool.prototype.handleConnect = function handleConnect(peer) { +Pool.prototype.handleConnect = co(function* handleConnect(peer) { this.logger.info('Connected to %s.', peer.hostname()); if (peer.outbound) this.hosts.markSuccess(peer.hostname()); this.emit('peer connect', peer); -}; +}); /** * Handle peer open event. @@ -1090,7 +1090,7 @@ Pool.prototype.handleConnect = function handleConnect(peer) { * @param {Peer} peer */ -Pool.prototype.handleOpen = function handleOpen(peer) { +Pool.prototype.handleOpen = co(function* handleOpen(peer) { // Advertise our address. if (!this.address.isNull() && !this.options.selfish @@ -1141,7 +1141,7 @@ Pool.prototype.handleOpen = function handleOpen(peer) { } this.emit('peer open', peer); -}; +}); /** * Handle peer close event. @@ -1178,10 +1178,10 @@ Pool.prototype.handleClose = co(function* handleClose(peer, connected) { * @param {Peer} peer */ -Pool.prototype.handleBan = function handleBan(peer) { +Pool.prototype.handleBan = co(function* handleBan(peer) { this.ban(peer.address); this.emit('ban', peer); -}; +}); /** * Handle peer version event.