浏览代码

Merge pull request #31482 from Faless/ws/fix_relay

WebSocketServer now sanitize destination peers.
Rémi Verschelde 6 年之前
父节点
当前提交
a1aeeb34d0
共有 1 个文件被更改,包括 4 次插入1 次删除
  1. 4 1
      modules/websocket/websocket_multiplayer_peer.cpp

+ 4 - 1
modules/websocket/websocket_multiplayer_peer.cpp

@@ -265,7 +265,10 @@ Error WebSocketMultiplayerPeer::_server_relay(int32_t p_from, int32_t p_to, cons
 
 		ERR_FAIL_COND_V(p_to == p_from, FAILED);
 
-		return get_peer(p_to)->put_packet(p_buffer, p_buffer_size); // Sending to specific peer
+		Ref<WebSocketPeer> peer_to = get_peer(p_to);
+		ERR_FAIL_COND_V(peer_to.is_null(), FAILED);
+
+		return peer_to->put_packet(p_buffer, p_buffer_size); // Sending to specific peer
 	}
 }