Add length-prefixed-stream message framing to the peersockets

Author: pfrazee

hyperswarm/peersocket.js

@@ -1,6 +1,7 @@
 const EventEmitter = require('events')
 const crypto = require('crypto')
 const createSwarm = require('@hyperswarm/network')
+const lpstream = require('length-prefixed-stream')
 const schemas = require('./peersocket-schemas')
 const {extractOrigin} = require('../../../lib/strings')
 
@@ -147,6 +148,8 @@ function handleConnection (swarm, socket, details) {
   if (lobby) {
     // create the connection
     var id = ++lobby.connIdCounter
+    var encoder = lpstream.encode()
+    var decoder = lpstream.decode()
     var conn = {
       id,
       socket,
@@ -155,9 +158,10 @@ function handleConnection (swarm, socket, details) {
     }
     lobby.connections.add(conn)
     lobby.emit('connection', {socketInfo: {id}})
+    encoder.pipe(socket).pipe(decoder)
 
     // wire up events
-    socket.on('data', message => {
+    decoder.on('data', message => {
       try {
         message = decodeMsg(message)
         conn.events.emit('message', {message})

package-lock.json

@@ -48,6 +48,7 @@
     "hyperdrive-to-zip-stream": "^2.1.1",
     "identify-filetype": "^1.0.0",
     "into-stream": "^3.1.0",
+    "length-prefixed-stream": "^1.6.0",
     "lodash.debounce": "^4.0.8",
     "lodash.get": "^4.4.2",
     "lodash.isequal": "^4.5.0",

package.json

@@ -50,7 +50,7 @@ module.exports = {
     var conn = getLobbyConnection(this.sender, tabIdentity, lobbyType, lobbyName, socketId)
     if (conn) {
       return new Promise((resolve, reject) => {
-        conn.socket.write(PeerSocket.encodeMsg(data), err => {
+        conn.encoder.write(PeerSocket.encodeMsg(data), err => {
           if (err) {
             console.error('Error writing to PeerSocket', err)
             reject(new Error('Failed to send message'))