mirror of
https://github.com/node-red/node-red.git
synced 2023-10-10 13:36:53 +02:00
Buffer fix
This commit is contained in:
bartbutenaers
Nick O'Leary
parent
e3b052bc38
commit
9bd4598c6a
@ -25,7 +25,6 @@ module.exports = function(RED) {
|
||||
function HTTPRequest(n) {
|
||||
RED.nodes.createNode(this,n);
|
||||
var node = this;
|
||||
var req;
|
||||
var nodeUrl = n.url;
|
||||
var isTemplatedUrl = (nodeUrl||"").indexOf("{{") != -1;
|
||||
var nodeMethod = n.method || "GET";
|
||||
@ -42,32 +41,7 @@ module.exports = function(RED) {
|
||||
if (process.env.no_proxy != null) { noprox = process.env.no_proxy.split(","); }
|
||||
if (process.env.NO_PROXY != null) { noprox = process.env.NO_PROXY.split(","); }
|
||||
|
||||
function handleMsg(msg) {
|
||||
if (node.metric()) {
|
||||
// Calculate request time
|
||||
var diff = process.hrtime(preRequestTimestamp);
|
||||
var ms = diff[0] * 1e3 + diff[1] * 1e-6;
|
||||
var metricRequestDurationMillis = ms.toFixed(3);
|
||||
node.metric("duration.millis", msg, metricRequestDurationMillis);
|
||||
if (res.client && res.client.bytesRead) {
|
||||
node.metric("size.bytes", msg, res.client.bytesRead);
|
||||
}
|
||||
}
|
||||
if (node.ret === "txt") {
|
||||
msg.payload = msg.payload.toString();
|
||||
}
|
||||
else if (node.ret === "obj") {
|
||||
try { msg.payload = JSON.parse(msg.payload); }
|
||||
catch(e) { node.warn(RED._("httpin.errors.json-error")); }
|
||||
}
|
||||
node.send(msg);
|
||||
node.status({});
|
||||
}
|
||||
|
||||
this.on("input",function(msg) {
|
||||
var boundary = "";
|
||||
var chunkBuffer = Buffer.from('');
|
||||
var headerBodySeparator = "";
|
||||
var preRequestTimestamp = process.hrtime();
|
||||
node.status({fill:"blue",shape:"dot",text:"httpin.status.requesting"});
|
||||
var url = nodeUrl || msg.url;
|
||||
@ -105,7 +79,6 @@ module.exports = function(RED) {
|
||||
var opts = urllib.parse(url);
|
||||
opts.method = method;
|
||||
opts.headers = {};
|
||||
opts.encoding = null; // response body should be buffer, not string
|
||||
var ctSet = "Content-Type"; // set default camel case
|
||||
var clSet = "Content-Length";
|
||||
if (msg.headers) {
|
||||
@ -187,111 +160,52 @@ module.exports = function(RED) {
|
||||
if (tlsNode) {
|
||||
tlsNode.addTLSOptions(opts);
|
||||
}
|
||||
if (req) {
|
||||
req.abort();
|
||||
}
|
||||
req = ((/^https/.test(urltotest))?https:http).request(opts,function(res) {
|
||||
var req = ((/^https/.test(urltotest))?https:http).request(opts,function(res) {
|
||||
//(node.ret === "bin") ? res.setEncoding('binary') : res.setEncoding('utf8');
|
||||
if (node.ret === "bin") {
|
||||
// Force NodeJs to return a Buffer instead of a string
|
||||
// See https://github.com/nodejs/node/issues/6038
|
||||
res.setEncoding(null);
|
||||
delete res._readableState.decoder;
|
||||
}
|
||||
msg.statusCode = res.statusCode;
|
||||
msg.headers = res.headers;
|
||||
msg.responseUrl = res.responseUrl;
|
||||
msg.payload = Buffer.from('');
|
||||
if (node.ret === "bin") {
|
||||
msg.payload = [];
|
||||
}
|
||||
else {
|
||||
msg.payload = "";
|
||||
}
|
||||
// msg.url = url; // revert when warning above finally removed
|
||||
|
||||
res.setEncoding(null);
|
||||
delete res._readableState.decoder;
|
||||
|
||||
res.on('data',function(chunk) {
|
||||
var nextPart = 0;
|
||||
|
||||
if (!boundary) {
|
||||
var contentType = this.headers['content-type'];
|
||||
if (contentType) {
|
||||
// Automatically check whether multipart streaming is required
|
||||
if (/multipart/.test(contentType)) {
|
||||
// Automatically detect the required boundary (that will be used between parts of the stream)
|
||||
boundary = (contentType.match(/.*;\sboundary=(.*)/) || [null, null])[1];
|
||||
|
||||
if(!boundary) {
|
||||
node.error(RED._("httpin.errors.no-boundary"),msg);
|
||||
return;
|
||||
}
|
||||
|
||||
// A boundary needs to start with -- (even if not specified in the http header variable)
|
||||
if (!boundary.startsWith('--')) {
|
||||
boundary = '--' + boundary;
|
||||
}
|
||||
|
||||
// Every part contains of headers and a body (content) separated by two EOL (end of line) symbols.
|
||||
// The end of line can be LF (linefeed \n), CR (carriage return \r), CRLF (carriage return linefeed \r\n).
|
||||
// When the stream starts, the EOL should be determined.
|
||||
var eolSymbols = (chunk.toString().match(/(?:\r\r|\n\n|\r\n\r\n)/g) || []);
|
||||
|
||||
if (eolSymbols.indexOf('\r\n\r\n') >= 0) {
|
||||
headerBodySeparator = '\r\n\r\n';
|
||||
}
|
||||
else if (eolSymbols.indexOf('\r\r') >= 0) {
|
||||
headerBodySeparator = '\r\r';
|
||||
}
|
||||
else if (eolSymbols.indexOf('\n\n') >= 0) {
|
||||
headerBodySeparator = '\n\n';
|
||||
}
|
||||
|
||||
if(!headerBodySeparator) {
|
||||
node.error(RED._("httpin.errors.no-separator"),msg);
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Append the chunk to other (non-processed) chunk data
|
||||
chunkBuffer = Buffer.concat([chunkBuffer, chunk]);
|
||||
chunk = null;
|
||||
|
||||
if (boundary) {
|
||||
while(true) {
|
||||
// Parts are separated by boundaries, so try to isolate parts in the received chunks.
|
||||
var bodyEnd = chunkBuffer.indexOf(boundary, nextPart);
|
||||
|
||||
if (bodyEnd == -1) {
|
||||
// Store the remaining (incomplete) part in the chunk buffer, to be processed when the next chunk arrives
|
||||
chunkBuffer = chunkBuffer.slice(nextPart, chunkBuffer.length);
|
||||
break;
|
||||
}
|
||||
|
||||
nextPart = bodyEnd + boundary.length;
|
||||
|
||||
// Find the part body (that arrives after the part header)
|
||||
// The header 'Content length' is optional, so it cannot be used here
|
||||
var bodyStart = chunkBuffer.indexOf(headerBodySeparator) + headerBodySeparator.length;
|
||||
|
||||
// Trim optional CR or LF characters at the start of the body
|
||||
for (var i = bodyStart; i <= bodyEnd; i++) {
|
||||
if (chunkBuffer[i] !== '\n' && chunkBuffer[i] !== '\r') {
|
||||
break;
|
||||
}
|
||||
bodyStart++;
|
||||
}
|
||||
|
||||
// Trim optional CR or LF characters at the end of the body
|
||||
for (var i = bodyEnd - 1; i >= bodyStart; i--) {
|
||||
if (chunkBuffer[i] !== '\n' && chunkBuffer[i] !== '\r') {
|
||||
break;
|
||||
}
|
||||
bodyEnd--;
|
||||
}
|
||||
|
||||
if (bodyEnd - bodyStart > 0) {
|
||||
// Send the body to the output port of this node
|
||||
msg.payload = chunkBuffer.slice(bodyStart, bodyEnd);
|
||||
handleMsg(msg);
|
||||
}
|
||||
if (node.ret === "bin") {
|
||||
msg.payload.push(chunk);
|
||||
}
|
||||
else {
|
||||
msg.payload += chunk;
|
||||
}
|
||||
});
|
||||
res.on('end',function() {
|
||||
handleMsg(msg);
|
||||
if (node.metric()) {
|
||||
// Calculate request time
|
||||
var diff = process.hrtime(preRequestTimestamp);
|
||||
var ms = diff[0] * 1e3 + diff[1] * 1e-6;
|
||||
var metricRequestDurationMillis = ms.toFixed(3);
|
||||
node.metric("duration.millis", msg, metricRequestDurationMillis);
|
||||
if (res.client && res.client.bytesRead) {
|
||||
node.metric("size.bytes", msg, res.client.bytesRead);
|
||||
}
|
||||
}
|
||||
if (node.ret === "bin") {
|
||||
msg.payload = Buffer.concat(msg.payload);
|
||||
}
|
||||
else if (node.ret === "obj") {
|
||||
try { msg.payload = JSON.parse(msg.payload); }
|
||||
catch(e) { node.warn(RED._("httpin.errors.json-error")); }
|
||||
}
|
||||
node.send(msg);
|
||||
node.status({});
|
||||
});
|
||||
});
|
||||
req.setTimeout(node.reqTimeout, function() {
|
||||
|
||||
Loading…
Reference in New Issue
Block a user