-
Notifications
You must be signed in to change notification settings - Fork 2
/
socket-to-me.js
125 lines (108 loc) · 3.28 KB
/
socket-to-me.js
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
'use strict';
var Socket = require('ws'),
connections = {};
// Get the function that is used to generate the data.
var generator = require(process.argv[2]);
function byteReadDelta(socket){
var delta = socket._socket.bytesRead - socket.lastBytesRead;
socket.lastBytesRead = socket._socket.bytesRead;
return delta;
}
function byteWrittenDelta(socket){
var delta = socket._socket.bytesWritten - socket.lastBytesWritten;
socket.lastBytesWritten = socket._socket.bytesWritten;
return delta;
}
process.on('message', function message(task) {
// Shut down every single socket.
if (task.shutdown) {
Object.keys(connections).forEach(function shutdown(id) {
connections[id].close();
connections[id].stop = true;
});
}
// End of the line, we are gonna start generating new connections.
if (!task.url) return;
var socket = new Socket(task.url, task.wsoptions);
socket.lastBytesWritten = 0;
socket.lastBytesRead = 0;
socket.generateMsg = generator.generateMsg();
var initTime = Date.now();
socket.on('open', function open() {
process.send({ type: 'open', duration: Date.now() - initTime, id: task.id });
write(socket, task, task.id);
});
socket.on('message', function message(data) {
process.send({
type: 'message',
read: byteReadDelta(socket),
id: task.id
});
});
socket.on('close', function close(code, msg) {
var internal = socket._socket || {};
if (code !== 1000) {
process.send({
type: 'disconnect',
message: msg,
id: task.id
});
}
process.send({
type: 'close', id: task.id,
read: internal.bytesRead || 0,
send: internal.bytesWritten || 0
});
});
socket.on('error', function error(err) {
process.send({ type: 'error', message: err.message, id: task.id });
socket.close();
delete connections[task.id];
});
// Adding a new socket to our socket collection.
connections[task.id] = socket;
});
function getSize(task){
var maxPayload = Math.max(task.payload[0], task.payload[1]);
var minPayload = Math.min(task.payload[0], task.payload[1]);
if(task.buffer){
return task.buffer;
} else {
return Math.round(Math.random() * (maxPayload - minPayload) + minPayload);
}
}
function getFrequency(task){
var maxDuration = Math.max(task.frequency[0], task.frequency[1]);
var minDuration = Math.min(task.frequency[0], task.frequency[1]);
return Math.round(Math.random() * (maxDuration - minDuration) + minDuration);
}
/**
* Helper function from writing messages to the socket.
*
* @param {WebSocket} socket WebSocket connection we should write to
* @param {Object} task The given task
* @param {String} id
* @api private
*/
function write(socket, task, id) {
var sz = getSize(task);
socket.generateMsg(sz, function message(err, data) {
if (socket.readyState !== socket.OPEN) {
return;
}
socket.send(data, function(err){
if (err) {
process.send({ type: 'error', message: err.message });
socket.close();
delete connections[id];
}else {
process.send({ type: 'sent', send: byteWrittenDelta(socket)});
}
});
});
if(!socket.stop && !(socket.readyState === socket.CLOSED)){
setTimeout(function(){
write(socket, task, id);
}, getFrequency(task));
}
}