Server factory integration

README.md

@@ -14,10 +14,10 @@
 
 Protocol decoder for Aedes MQTT Broker
 
-The purpose of this module is to be used inside [aedes](https://github.com/moscajs/aedes) `decodeProtocol` hook, which is called when aedes instance receives a first valid buffer from client ( before CONNECT packet). The client object state is in default and its connected state is false. 
-The function extract socket details and if aedes `trustProxy` option is set to true, it will first parse http headers (x-real-ip | x-forwarded-for) and proxy protocol (v1 and v2) to retrieve information in client.connDetails.
+The purpose of this module is to be used inside [aedes-server-factory](https://github.com/moscajs/aedes-server-factory) `bindConnection` function, which is called when the server receives a connection from client (before CONNECT packet). The client object state is in default and its connected state is false. 
+The function extract socket details and if `aedes-server-factory` `trustProxy` option is set to true, it will first parse http headers (x-real-ip | x-forwarded-for) and/or proxy protocol (v1 and v2), then passing the informations to `aedes` that will assign them to `client.connDetails`.
 
-The function `protocolDecoder` returns [ConnectionDetails](./types/index.d.ts), if the object contains data property, it will be parsed as an mqtt-packet.
+The function `protocolDecoder` and `extractSocketDetails` returns [ConnectionDetails](./types/index.d.ts), if the object contains `data` property, it will be parsed as an [mqtt-packet](https://github.com/mqttjs/mqtt-packet).
 
 ## Install
 
@@ -30,30 +30,22 @@ npm install aedes-protocol-decoder --save
 ```js
 var aedes = require('aedes')
 var { protocolDecoder } = require('aedes-protocol-decoder')
-var net = require('net')
+var { createServer } = require('aedes-server-factory')
 var port = 1883
 
 var broker = aedes({
-	decodeProtocol: function (client, buffer) {
-	  var proto = protocolDecoder(client, buffer)
-	  return proto
-	},
-	preConnect: function (client, done) {
+	preConnect: function (client, packet, done) {
 	  if (client.connDetails && client.connDetails.ipAddress) {
 	    client.ip = client.connDetails.ipAddress
 	  }
 	  return done(null, true)
 	},
-	trustProxy: true
 })
 
-var server = net.createServer(broker.handle)
-
+var server = createServer(broker, { trustProxy: true, protocolDecoder })
 server.listen(port, function () {
   console.log('server listening on port', port)
 })
-
-
 ```
 
 ## License

example.js

@@ -4,7 +4,8 @@ var aedes = require('aedes')
 var mqttPacket = require('mqtt-packet')
 var net = require('net')
 var proxyProtocol = require('proxy-protocol-js')
-var protocolDecoder = require('./lib/protocol-decoder')
+var { createServer } = require('aedes-server-factory')
+var { extractSocketDetails, protocolDecoder } = require('./index')
 
 var brokerPort = 4883
 
@@ -119,22 +120,17 @@ function sendProxyPacket (version = 1, ipFamily = 4) {
 
 function startAedes () {
   var broker = aedes({
-    decodeProtocol: function (client, buffer) {
-      var proto = protocolDecoder(client, buffer)
-      return proto
-    },
-    preConnect: function (client, done) {
+    preConnect: function (client, packet, done) {
       console.log('Aedes preConnect check client ip:', client.connDetails)
       if (client.connDetails && client.connDetails.ipAddress) {
         client.ip = client.connDetails.ipAddress
       }
       client.close()
-      return done(null, true)
-    },
-    trustProxy: true
+      done(null, true)
+    }
   })
 
-  var server = require('net').createServer(broker.handle)
+  var server = createServer(broker, { trustProxy: true, extractSocketDetails, protocolDecoder })
 
   server.listen(brokerPort, function () {
     console.log('Aedes listening on :', server.address())

index.js

@@ -1,5 +1,151 @@
-var protocolDecoder = require('./lib/protocol-decoder')
+'use strict'
+
+var proxyProtocol = require('proxy-protocol-js')
+var forwarded = require('forwarded')
+
+var v1ProxyProtocolSignature = Buffer.from('PROXY ', 'utf8')
+var v2ProxyProtocolSignature = Buffer.from([
+  0x0d,
+  0x0a,
+  0x0d,
+  0x0a,
+  0x00,
+  0x0d,
+  0x0a,
+  0x51,
+  0x55,
+  0x49,
+  0x54,
+  0x0a
+])
+
+function isValidV1ProxyProtocol (buffer) {
+  for (var i = 0; i < v1ProxyProtocolSignature.length; i++) {
+    if (buffer[i] !== v1ProxyProtocolSignature[i]) {
+      return false
+    }
+  }
+  return true
+}
+
+function isValidV2ProxyProtocol (buffer) {
+  for (var i = 0; i < v2ProxyProtocolSignature.length; i++) {
+    if (buffer[i] !== v2ProxyProtocolSignature[i]) {
+      return false
+    }
+  }
+  return true
+}
+
+// from https://stackoverflow.com/questions/57077161/how-do-i-convert-hex-buffer-to-ipv6-in-javascript
+function parseIpV6Array (ip) {
+  var ipHex = Buffer.from(ip).toString('hex')
+  return ipHex.match(/.{1,4}/g)
+    .map((val) => val.replace(/^0+/, ''))
+    .join(':')
+    .replace(/0000:/g, ':')
+    .replace(/:{2,}/g, '::')
+}
+
+function getProtoIpFamily (ipFamily) {
+  if (ipFamily && ipFamily.endsWith('4')) {
+    return 4
+  } else if (ipFamily && ipFamily.endsWith('6')) {
+    return 6
+  }
+  return 0
+}
+
+function extractHttpDetails (req, socket) {
+  const details = {}
+  var headers = req && req.headers ? req.headers : null
+  if (headers) {
+    if (headers['x-forwarded-for']) {
+      var addresses = forwarded(req)
+      details.ipAddress = headers['x-real-ip'] ? headers['x-real-ip'] : addresses[addresses.length - 1]
+      details.serverIpAddress = addresses[0]
+    }
+    if (headers['x-real-ip']) {
+      details.ipAddress = headers['x-real-ip']
+    }
+    details.port = socket._socket.remotePort
+    details.ipFamily = getProtoIpFamily(socket._socket.remoteFamily)
+    details.isWebsocket = true
+  }
+  return details
+}
+
+function extractProxyDetails (buffer) {
+  const details = {}
+  var proxyProto
+  if (isValidV1ProxyProtocol(buffer)) {
+    proxyProto = proxyProtocol.V1BinaryProxyProtocol.parse(buffer)
+    if (proxyProto && proxyProto.source && proxyProto.data) {
+      details.ipFamily = getProtoIpFamily(proxyProto.inetProtocol)
+      details.ipAddress = proxyProto.source.ipAddress
+      details.port = proxyProto.source.port
+      details.serverIpAddress = proxyProto.destination.ipAddress
+      details.data = proxyProto.data
+      details.isProxy = 1
+    }
+  } else if (isValidV2ProxyProtocol(buffer)) {
+    proxyProto = proxyProtocol.V2ProxyProtocol.parse(buffer)
+    if (proxyProto && proxyProto.proxyAddress && proxyProto.data) {
+      if (proxyProto.proxyAddress instanceof proxyProtocol.IPv4ProxyAddress) {
+        details.ipAddress = proxyProto.proxyAddress.sourceAddress.address.join('.')
+        details.port = proxyProto.proxyAddress.sourcePort
+        details.serverIpAddress = proxyProto.proxyAddress.destinationAddress.address.join('.')
+        details.ipFamily = 4
+      } else if (proxyProto.proxyAddress instanceof proxyProtocol.IPv6ProxyAddress) {
+        details.ipAddress = parseIpV6Array(proxyProto.proxyAddress.sourceAddress.address)
+        details.port = proxyProto.proxyAddress.sourcePort
+        details.serverIpAddress = parseIpV6Array(proxyProto.proxyAddress.destinationAddress.address)
+        details.ipFamily = 6
+      }
+      details.isProxy = 2
+      if (Buffer.isBuffer(proxyProto.data)) {
+        details.data = proxyProto.data
+      } else {
+        details.data = Buffer.from(proxyProto.data)
+      }
+    }
+  }
+  return details
+}
+
+function extractSocketDetails (socket) {
+  const details = {}
+  if (socket._socket && socket._socket.address) {
+    details.isWebsocket = true
+    details.ipAddress = socket._socket.remoteAddress
+    details.port = socket._socket.remotePort
+    details.serverIpAddress = socket._socket.address().address
+    details.ipFamily = getProtoIpFamily(socket._socket.remoteFamily)
+  } else if (socket.address) {
+    details.ipAddress = socket.remoteAddress
+    details.port = socket.remotePort
+    details.serverIpAddress = socket.address().address
+    details.ipFamily = getProtoIpFamily(socket.remoteFamily)
+  }
+  return details
+}
+
+function protocolDecoder (conn, buffer, req) {
+  var proto = {}
+  if (!buffer) return proto
+  var socket = conn.socket || conn
+  proto.isProxy = 0
+  proto.isWebsocket = false
+  proto = { ...proto, ...extractHttpDetails(req, socket), ...extractProxyDetails(buffer) }
+
+  if (!proto.ipAddress) {
+    proto = { ...proto, ...extractSocketDetails(socket) }
+  }
+
+  return proto
+}
 
 module.exports = {
+  extractSocketDetails,
   protocolDecoder
 }