Add support for X-Forwarded headers

main.js

@@ -6,6 +6,7 @@ var r = require('rethinkdb'),
     Cb = require('y-callback'),
     proxyProtocol = require('./main/proxyProtocol.js'),
     bindServer = require('./main/bindServer.js'),
+    injectXFH = require('./main/injectXFH.js'),
     otherSocket = Symbol(),
     connection = Symbol(),
     walker = Symbol(),
@@ -201,7 +202,9 @@ function find(backend){
 
 function* proxy(e,d,hosts,server,aliases){
   var parts,target,backend,
-      host,dest;
+      host,dest,socket;
+
+  socket = e.socket;
 
   if(hosts[e.host || '']) target = hosts[e.host || ''];
   else{
@@ -267,6 +270,16 @@ function* proxy(e,d,hosts,server,aliases){
   if(!dest.to.stripProxy) dest.socket.write(e.proxyHeader);
   if(dest.to.prependHost) dest.socket.write('host: ' + dest.to.prependHost + '\r\n');
 
+  if(dest.to.XFH){
+    try{
+      yield injectXFH(socket,e.socket,dest.socket);
+    }catch(err){
+      dest.socket.destroy();
+      e.socket.destroy();
+      return;
+    }
+  }
+
   dest.socket.pipe(e.socket).pipe(dest.socket);
   dest.socket[otherSocket] = e.socket;
   e.socket[otherSocket] = dest.socket;

main/bindServer.js

@@ -1,6 +1,6 @@
 var Emitter = require('y-emitter'),
     walk = require('y-walk'),
-    Cb = require('y-callback'),
+    read = require('./read'),
     emitter = Symbol(),
     hostsMap = Symbol(),
     processSocket,read;
@@ -193,18 +193,6 @@ processSocket = walk.wrap(function*(socket,emitter){
 
 });
 
-read = walk.wrap(function*(socket,n){
-  var cb,data;
-
-  while(true){
-    data = socket.read(n);
-    if(data) return data;
-    socket.once('readable',cb = Cb());
-    yield cb;
-  }
-
-});
-
 /*/ exports /*/
 
 module.exports = bindServer;

main/injectXFH.js

@@ -0,0 +1,30 @@
+var walk = require('y-walk'),
+    ip = require('ip'),
+    read = require('./read');
+
+module.exports = walk.wrap(function*(originalSocket,socketFrom,socketTo){
+  var sequence = [0x0d, 0x0a, 0x0d, 0x0a],
+      i,j,b;
+
+  loop: while(true){
+
+    for(i = 0;i < sequence.length;i++){
+      b = yield read(socketFrom, 1);
+
+      if(b[0] != sequence[i]){
+        socketTo.write(new Buffer(sequence.slice(0,i)));
+        socketTo.write(b);
+        continue loop;
+      }
+    }
+
+    socketTo.write(new Buffer(sequence.slice(0,2)));
+    socketTo.write(`X-Forwarded-For: ${ip.toString(ip.toBuffer(originalSocket.remoteAddress))}\r\n`);
+    if(originalSocket != socketFrom)
+      socketTo.write('X-Forwarded-Proto: https\r\n');
+    socketTo.write(new Buffer(sequence.slice(2,4)));
+    return;
+
+  }
+
+});

main/read.js

@@ -0,0 +1,14 @@
+var walk = require('y-walk'),
+    Cb = require('y-callback');
+
+module.exports = walk.wrap(function*(socket,n){
+  var cb,data;
+
+  while(true){
+    data = socket.read(n);
+    if(data) return data;
+    socket.once('readable',cb = Cb());
+    yield cb;
+  }
+
+});

test/main/init.js

@@ -171,6 +171,7 @@ exports.rules = desc => t(desc,function*(){
         }
       },
       to: {
+        XFH: true,
         port: 8001,
         host: '127.0.0.1'
       }