Bläddra i källkod

ws spoof issue

Michelle Bu 12 år sedan
förälder
incheckning
728b21278d
1 ändrade filer med 3 tillägg och 1 borttagningar
  1. 3 1
      lib/server.js

+ 3 - 1
lib/server.js

@@ -96,7 +96,7 @@ PeerServer.prototype._initializeWSS = function() {
           case 'LEAVE':
           case 'LEAVE':
             // Clean up if a Peer sends a LEAVE.
             // Clean up if a Peer sends a LEAVE.
             if (!message.dst) {
             if (!message.dst) {
-              self._removePeer(message.src);
+              self._removePeer(id);
               break;
               break;
             }
             }
           // ICE candidates
           // ICE candidates
@@ -106,6 +106,8 @@ PeerServer.prototype._initializeWSS = function() {
           case 'ANSWER':
           case 'ANSWER':
           // Firefoxism (connectDataConnection ports)
           // Firefoxism (connectDataConnection ports)
           case 'PORT':
           case 'PORT':
+            // Use the ID we know to be correct to prevent spoofing.
+            message.src = id;
             self._handleTransmission(message);
             self._handleTransmission(message);
             break;
             break;
           default:
           default: