mirror of
https://github.com/node-red/node-red.git
synced 2023-10-10 13:36:53 +02:00
314 lines
13 KiB
JavaScript
314 lines
13 KiB
JavaScript
/**
|
|
* Copyright 2013 IBM Corp.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
**/
|
|
|
|
module.exports = function(RED) {
|
|
var reconnectTime = RED.settings.socketReconnectTime||10000;
|
|
var socketTimeout = RED.settings.socketTimeout||null;
|
|
var net = require('net');
|
|
|
|
var connectionPool = {};
|
|
|
|
function TcpIn(n) {
|
|
RED.nodes.createNode(this,n);
|
|
this.host = n.host;
|
|
this.port = n.port * 1;
|
|
this.topic = n.topic;
|
|
this.stream = (!n.datamode||n.datamode=='stream'); /* stream,single*/
|
|
this.datatype = n.datatype||'buffer'; /* buffer,utf8,base64 */
|
|
this.newline = (n.newline||"").replace("\\n","\n").replace("\\r","\r");
|
|
this.base64 = n.base64;
|
|
this.server = (typeof n.server == 'boolean')?n.server:(n.server == "server");
|
|
this.closing = false;
|
|
var node = this;
|
|
|
|
if (!node.server) {
|
|
var buffer = null;
|
|
var client;
|
|
var reconnectTimeout;
|
|
function setupTcpClient() {
|
|
node.log("connecting to "+node.host+":"+node.port);
|
|
node.status({fill:"grey",shape:"dot",text:"connecting"},true);
|
|
var id = (1+Math.random()*4294967295).toString(16);
|
|
client = net.connect(node.port, node.host, function() {
|
|
buffer = (node.datatype == 'buffer')? new Buffer(0):"";
|
|
node.log("connected to "+node.host+":"+node.port);
|
|
node.status({fill:"green",shape:"dot",text:"connected"},true);
|
|
});
|
|
connectionPool[id] = client;
|
|
|
|
client.on('data', function (data) {
|
|
if (node.datatype != 'buffer') {
|
|
data = data.toString(node.datatype);
|
|
}
|
|
if (node.stream) {
|
|
if ((node.datatype) === "utf8" && node.newline != "") {
|
|
buffer = buffer+data;
|
|
var parts = buffer.split(node.newline);
|
|
for (var i = 0;i<parts.length-1;i+=1) {
|
|
var msg = {topic:node.topic, payload:parts[i]};
|
|
msg._session = {type:"tcp",id:id};
|
|
node.send(msg);
|
|
}
|
|
buffer = parts[parts.length-1];
|
|
} else {
|
|
var msg = {topic:node.topic, payload:data};
|
|
msg._session = {type:"tcp",id:id};
|
|
node.send(msg);
|
|
}
|
|
} else {
|
|
if ((typeof data) === "string") {
|
|
buffer = buffer+data;
|
|
} else {
|
|
buffer = Buffer.concat([buffer,data],buffer.length+data.length);
|
|
}
|
|
}
|
|
});
|
|
client.on('end', function() {
|
|
if (!node.stream || (node.datatype == "utf8" && node.newline != "" && buffer.length > 0)) {
|
|
var msg = {topic:node.topic,payload:buffer};
|
|
msg._session = {type:"tcp",id:id};
|
|
node.send(msg);
|
|
buffer = null;
|
|
}
|
|
});
|
|
client.on('close', function() {
|
|
delete connectionPool[id];
|
|
node.log("connection lost to "+node.host+":"+node.port);
|
|
node.status({fill:"red",shape:"ring",text:"disconnected"});
|
|
if (!node.closing) {
|
|
reconnectTimeout = setTimeout(setupTcpClient, reconnectTime);
|
|
}
|
|
});
|
|
client.on('error', function(err) {
|
|
node.log(err);
|
|
});
|
|
}
|
|
setupTcpClient();
|
|
|
|
this.on('close', function() {
|
|
this.closing = true;
|
|
client.end();
|
|
clearTimeout(reconnectTimeout);
|
|
});
|
|
} else {
|
|
var server = net.createServer(function (socket) {
|
|
if (socketTimeout !== null) { socket.setTimeout(socketTimeout); }
|
|
var id = (1+Math.random()*4294967295).toString(16);
|
|
connectionPool[id] = socket;
|
|
|
|
var buffer = (node.datatype == 'buffer')? new Buffer(0):"";
|
|
socket.on('data', function (data) {
|
|
if (node.datatype != 'buffer') {
|
|
data = data.toString(node.datatype);
|
|
}
|
|
|
|
if (node.stream) {
|
|
if ((typeof data) === "string" && node.newline != "") {
|
|
buffer = buffer+data;
|
|
var parts = buffer.split(node.newline);
|
|
for (var i = 0;i<parts.length-1;i+=1) {
|
|
var msg = {topic:node.topic, payload:parts[i],ip:socket.remoteAddress,port:socket.remotePort};
|
|
msg._session = {type:"tcp",id:id};
|
|
node.send(msg);
|
|
}
|
|
buffer = parts[parts.length-1];
|
|
} else {
|
|
var msg = {topic:node.topic, payload:data};
|
|
msg._session = {type:"tcp",id:id};
|
|
node.send(msg);
|
|
}
|
|
} else {
|
|
if ((typeof data) === "string") {
|
|
buffer = buffer+data;
|
|
} else {
|
|
buffer = Buffer.concat([buffer,data],buffer.length+data.length);
|
|
}
|
|
}
|
|
});
|
|
socket.on('end', function() {
|
|
if (!node.stream || (node.datatype == "utf8" && node.newline != "" && buffer.length > 0)) {
|
|
var msg = {topic:node.topic,payload:buffer};
|
|
msg._session = {type:"tcp",id:id};
|
|
node.send(msg);
|
|
buffer = null;
|
|
}
|
|
});
|
|
socket.on('timeout', function() {
|
|
node.log('timeout closed socket port '+node.port);
|
|
socket.end();
|
|
});
|
|
socket.on('close', function() {
|
|
delete connectionPool[id];
|
|
});
|
|
socket.on('error',function(err) {
|
|
node.log(err);
|
|
});
|
|
});
|
|
server.on('error', function(err) {
|
|
if (err) {
|
|
node.error('unable to listen on port '+node.port+' : '+err);
|
|
}
|
|
});
|
|
server.listen(node.port, function(err) {
|
|
if (err) {
|
|
node.error('unable to listen on port '+node.port+' : '+err);
|
|
} else {
|
|
node.log('listening on port '+node.port);
|
|
|
|
node.on('close', function() {
|
|
node.closing = true;
|
|
server.close();
|
|
node.log('stopped listening on port '+node.port);
|
|
});
|
|
}
|
|
});
|
|
}
|
|
|
|
}
|
|
RED.nodes.registerType("tcp in",TcpIn);
|
|
|
|
function TcpOut(n) {
|
|
RED.nodes.createNode(this,n);
|
|
this.host = n.host;
|
|
this.port = n.port * 1;
|
|
this.base64 = n.base64;
|
|
this.beserver = n.beserver;
|
|
this.name = n.name;
|
|
this.closing = false;
|
|
var node = this;
|
|
|
|
if (!node.beserver||node.beserver=="client") {
|
|
var reconnectTimeout;
|
|
var client = null;
|
|
var connected = false;
|
|
|
|
function setupTcpClient() {
|
|
node.log("connecting to "+node.host+":"+node.port);
|
|
node.status({fill:"grey",shape:"dot",text:"connecting"},true);
|
|
client = net.connect(node.port, node.host, function() {
|
|
connected = true;
|
|
node.log("connected to "+node.host+":"+node.port);
|
|
node.status({fill:"green",shape:"dot",text:"connected"},true);
|
|
});
|
|
client.on('error', function (err) {
|
|
node.log('error : '+err);
|
|
});
|
|
client.on('end', function (err) {
|
|
});
|
|
client.on('close', function() {
|
|
node.log("connection lost to "+node.host+":"+node.port);
|
|
node.status({fill:"red",shape:"ring",text:"disconnected"},true);
|
|
connected = false;
|
|
client.destroy();
|
|
if (!node.closing) {
|
|
reconnectTimeout = setTimeout(setupTcpClient,reconnectTime);
|
|
}
|
|
});
|
|
}
|
|
setupTcpClient();
|
|
|
|
node.on("input", function(msg) {
|
|
if (connected && msg.payload != null) {
|
|
if (Buffer.isBuffer(msg.payload)) {
|
|
client.write(msg.payload);
|
|
} else if (typeof msg.payload === "string" && node.base64) {
|
|
client.write(new Buffer(msg.payload,'base64'));
|
|
} else {
|
|
client.write(new Buffer(""+msg.payload));
|
|
}
|
|
}
|
|
});
|
|
|
|
node.on("close", function() {
|
|
this.closing = true;
|
|
client.end();
|
|
clearTimeout(reconnectTimeout);
|
|
});
|
|
|
|
} else if (node.beserver == "reply") {
|
|
node.on("input",function(msg) {
|
|
if (msg._session && msg._session.type == "tcp") {
|
|
var client = connectionPool[msg._session.id];
|
|
if (client) {
|
|
if (Buffer.isBuffer(msg.payload)) {
|
|
client.write(msg.payload);
|
|
} else if (typeof msg.payload === "string" && node.base64) {
|
|
client.write(new Buffer(msg.payload,'base64'));
|
|
} else {
|
|
client.write(new Buffer(""+msg.payload));
|
|
}
|
|
}
|
|
}
|
|
});
|
|
} else {
|
|
var connectedSockets = [];
|
|
var server = net.createServer(function (socket) {
|
|
if (socketTimeout !== null) { socket.setTimeout(socketTimeout); }
|
|
var remoteDetails = socket.remoteAddress+":"+socket.remotePort;
|
|
node.log("connection from "+remoteDetails);
|
|
connectedSockets.push(socket);
|
|
socket.on('timeout', function() {
|
|
node.log('timeout closed socket port '+node.port);
|
|
socket.end();
|
|
});
|
|
socket.on('close',function() {
|
|
node.log("connection closed from "+remoteDetails);
|
|
connectedSockets.splice(connectedSockets.indexOf(socket),1);
|
|
});
|
|
socket.on('error',function() {
|
|
node.log("socket error from "+remoteDetails);
|
|
connectedSockets.splice(connectedSockets.indexOf(socket),1);
|
|
});
|
|
});
|
|
node.on("input", function(msg) {
|
|
if (msg.payload != null) {
|
|
var buffer;
|
|
if (Buffer.isBuffer(msg.payload)) {
|
|
buffer = msg.payload;
|
|
} else if (typeof msg.payload === "string" && node.base64) {
|
|
buffer = new Buffer(msg.payload,'base64');
|
|
} else {
|
|
buffer = new Buffer(""+msg.payload);
|
|
}
|
|
for (var i = 0; i<connectedSockets.length;i+=1) {
|
|
connectedSockets[i].write(buffer);
|
|
}
|
|
}
|
|
});
|
|
|
|
server.on('error', function(err) {
|
|
if (err) {
|
|
node.error('unable to listen on port '+node.port+' : '+err);
|
|
}
|
|
});
|
|
|
|
server.listen(node.port, function(err) {
|
|
if (err) {
|
|
node.error('unable to listen on port '+node.port+' : '+err);
|
|
} else {
|
|
node.log('listening on port '+node.port);
|
|
node.on('close', function() {
|
|
server.close();
|
|
node.log('stopped listening on port '+node.port);
|
|
});
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
RED.nodes.registerType("tcp out",TcpOut);
|
|
}
|