1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
// Flags: --expose-internals
'use strict';
const common = require('../common');
const assert = require('assert');
const { internalBinding } = require('internal/test/binding');
const { TCP, constants: TCPConstants } = internalBinding('tcp_wrap');
const {
WriteWrap,
kReadBytesOrError,
kArrayBufferOffset,
streamBaseState
} = internalBinding('stream_wrap');
const server = new TCP(TCPConstants.SOCKET);
const r = server.bind('0.0.0.0', 0);
assert.strictEqual(r, 0);
let port = {};
server.getsockname(port);
port = port.port;
server.listen(128);
server.onconnection = (err, client) => {
assert.strictEqual(client.writeQueueSize, 0);
console.log('got connection');
const maybeCloseClient = () => {
if (client.pendingWrites.length === 0 && client.gotEOF) {
console.log('close client');
client.close();
}
};
client.readStart();
client.pendingWrites = [];
client.onread = common.mustCall((arrayBuffer) => {
if (arrayBuffer) {
const offset = streamBaseState[kArrayBufferOffset];
const nread = streamBaseState[kReadBytesOrError];
const buffer = Buffer.from(arrayBuffer, offset, nread);
assert.ok(buffer.length > 0);
assert.strictEqual(client.writeQueueSize, 0);
const req = new WriteWrap();
req.async = false;
const returnCode = client.writeBuffer(req, buffer);
assert.strictEqual(returnCode, 0);
client.pendingWrites.push(req);
console.log(`client.writeQueueSize: ${client.writeQueueSize}`);
// 11 bytes should flush
assert.strictEqual(client.writeQueueSize, 0);
if (req.async)
req.oncomplete = common.mustCall(done);
else
process.nextTick(done.bind(null, 0, client, req));
function done(status, client_, req_) {
assert.strictEqual(client.pendingWrites.shift(), req);
// Check parameters.
assert.strictEqual(status, 0);
assert.strictEqual(client_, client);
assert.strictEqual(req_, req);
console.log(`client.writeQueueSize: ${client.writeQueueSize}`);
assert.strictEqual(client.writeQueueSize, 0);
maybeCloseClient();
}
} else {
console.log('eof');
client.gotEOF = true;
server.close();
maybeCloseClient();
}
}, 2);
};
const net = require('net');
const c = net.createConnection(port);
c.on('connect', common.mustCall(() => { c.end('hello world'); }));
c.setEncoding('utf8');
c.on('data', common.mustCall((d) => {
assert.strictEqual(d, 'hello world');
}));
c.on('close', () => {
console.error('client closed');
});
|
