mirror of https://github.com/lukechilds/node.git
Browse Source
Introduce a way to wrap plain-js `stream.Duplex` streams into C++ StreamBase's child class. With such method at hand it is now possible to pass `stream.Duplex` instance as a `socket` parameter to `tls.connect()`. PR-URL: https://github.com/iojs/io.js/pull/926 Reviewed-By: Chris Dickinson <christopher.s.dickinson@gmail.com>v1.8.0-commit
Fedor Indutny
10 years ago
15 changed files with 498 additions and 52 deletions
@ -0,0 +1,118 @@ |
|||
const util = require('util'); |
|||
const Socket = require('net').Socket; |
|||
const JSStream = process.binding('js_stream').JSStream; |
|||
const uv = process.binding('uv'); |
|||
|
|||
function StreamWrap(stream) { |
|||
var handle = new JSStream(); |
|||
|
|||
this.stream = stream; |
|||
|
|||
var self = this; |
|||
handle.close = function(cb) { |
|||
cb(); |
|||
}; |
|||
handle.isAlive = function() { |
|||
return self.isAlive(); |
|||
}; |
|||
handle.isClosing = function() { |
|||
return self.isClosing(); |
|||
}; |
|||
handle.onreadstart = function() { |
|||
return self.readStart(); |
|||
}; |
|||
handle.onreadstop = function() { |
|||
return self.readStop(); |
|||
}; |
|||
handle.onshutdown = function(req) { |
|||
return self.shutdown(req); |
|||
}; |
|||
handle.onwrite = function(req, bufs) { |
|||
return self.write(req, bufs); |
|||
}; |
|||
|
|||
this.stream.pause(); |
|||
this.stream.on('data', function(chunk) { |
|||
self._handle.readBuffer(chunk); |
|||
}); |
|||
this.stream.once('end', function() { |
|||
self._handle.emitEOF(); |
|||
}); |
|||
this.stream.on('error', function(err) { |
|||
self.emit('error', err); |
|||
}); |
|||
|
|||
Socket.call(this, { |
|||
handle: handle |
|||
}); |
|||
} |
|||
util.inherits(StreamWrap, Socket); |
|||
module.exports = StreamWrap; |
|||
|
|||
// require('_stream_wrap').StreamWrap
|
|||
StreamWrap.StreamWrap = StreamWrap; |
|||
|
|||
StreamWrap.prototype.isAlive = function isAlive() { |
|||
return this.readable && this.writable; |
|||
}; |
|||
|
|||
StreamWrap.prototype.isClosing = function isClosing() { |
|||
return !this.isAlive(); |
|||
}; |
|||
|
|||
StreamWrap.prototype.readStart = function readStart() { |
|||
this.stream.resume(); |
|||
return 0; |
|||
}; |
|||
|
|||
StreamWrap.prototype.readStop = function readStop() { |
|||
this.stream.pause(); |
|||
return 0; |
|||
}; |
|||
|
|||
StreamWrap.prototype.shutdown = function shutdown(req) { |
|||
var self = this; |
|||
|
|||
this.stream.end(function() { |
|||
// Ensure that write was dispatched
|
|||
setImmediate(function() { |
|||
self._handle.finishShutdown(req, 0); |
|||
}); |
|||
}); |
|||
return 0; |
|||
}; |
|||
|
|||
StreamWrap.prototype.write = function write(req, bufs) { |
|||
var pending = bufs.length; |
|||
var self = this; |
|||
|
|||
self.stream.cork(); |
|||
bufs.forEach(function(buf) { |
|||
self.stream.write(buf, done); |
|||
}); |
|||
self.stream.uncork(); |
|||
|
|||
function done(err) { |
|||
if (!err && --pending !== 0) |
|||
return; |
|||
|
|||
// Ensure that this is called once in case of error
|
|||
pending = 0; |
|||
|
|||
// Ensure that write was dispatched
|
|||
setImmediate(function() { |
|||
var errCode = 0; |
|||
if (err) { |
|||
if (err.code && uv['UV_' + err.code]) |
|||
errCode = uv['UV_' + err.code]; |
|||
else |
|||
errCode = uv.UV_EPIPE; |
|||
} |
|||
|
|||
self._handle.doAfterWrite(req); |
|||
self._handle.finishWrite(req, errCode); |
|||
}); |
|||
} |
|||
|
|||
return 0; |
|||
}; |
|||
@ -0,0 +1,69 @@ |
|||
var assert = require('assert'); |
|||
var stream = require('stream'); |
|||
var tls = require('tls'); |
|||
var fs = require('fs'); |
|||
var net = require('net'); |
|||
|
|||
var common = require('../common'); |
|||
|
|||
var connected = { |
|||
client: 0, |
|||
server: 0 |
|||
}; |
|||
|
|||
var server = tls.createServer({ |
|||
key: fs.readFileSync(common.fixturesDir + '/keys/agent1-key.pem'), |
|||
cert: fs.readFileSync(common.fixturesDir + '/keys/agent1-cert.pem') |
|||
}, function(c) { |
|||
console.log('new client'); |
|||
connected.server++; |
|||
c.end('ohai'); |
|||
}).listen(common.PORT, function() { |
|||
var raw = net.connect(common.PORT); |
|||
|
|||
var pending = false; |
|||
raw.on('readable', function() { |
|||
if (pending) |
|||
p._read(); |
|||
}); |
|||
|
|||
var p = new stream.Duplex({ |
|||
read: function read() { |
|||
pending = false; |
|||
|
|||
var chunk = raw.read(); |
|||
if (chunk) { |
|||
console.log('read', chunk); |
|||
this.push(chunk); |
|||
} else { |
|||
pending = true; |
|||
} |
|||
}, |
|||
write: function write(data, enc, cb) { |
|||
console.log('write', data, enc); |
|||
raw.write(data, enc, cb); |
|||
} |
|||
}); |
|||
|
|||
var socket = tls.connect({ |
|||
socket: p, |
|||
rejectUnauthorized: false |
|||
}, function() { |
|||
console.log('client secure'); |
|||
|
|||
connected.client++; |
|||
|
|||
socket.end('hello'); |
|||
socket.resume(); |
|||
}); |
|||
|
|||
socket.once('close', function() { |
|||
console.log('client close'); |
|||
server.close(); |
|||
}); |
|||
}); |
|||
|
|||
process.once('exit', function() { |
|||
assert.equal(connected.client, 1); |
|||
assert.equal(connected.server, 1); |
|||
}); |
|||
Loading…
Reference in new issue