Check for CF-Connecting-IP and decreased timeout (snapdrop pr SnapDro…

…p#544)
dxlliv · May 28, 2023 · c6c0f24 · c6c0f24
1 parent f40a806
commit c6c0f24
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/server/index.js b/server/index.js
@@ -138,7 +138,7 @@ class SnapdropServer {
 
     _keepAlive(peer) {
         this._cancelKeepAlive(peer);
-        var timeout = 30000;
+        var timeout = 500;
         if (!peer.lastBeat) {
             peer.lastBeat = Date.now();
         }
@@ -183,7 +183,9 @@ class Peer {
     }
 
     _setIP(request) {
-        if (request.headers['x-forwarded-for']) {
+        if (request.headers['cf-connecting-ip']) {
+            this.ip = request.headers['cf-connecting-ip'].split(/\s*,\s*/)[0];
+        } else if (request.headers['x-forwarded-for']) {
             this.ip = request.headers['x-forwarded-for'].split(/\s*,\s*/)[0];
         } else {
             this.ip = request.connection.remoteAddress;