1 files changed, 6 insertions, 9 deletions

diff --git a/core/io/udp_server.h b/core/io/udp_server.h
index bbd2f951c9..e49a559c51 100644
--- a/core/io/udp_server.h
+++ b/core/io/udp_server.h
@@ -34,8 +34,8 @@
 #include "core/io/net_socket.h"
 #include "core/io/packet_peer_udp.h"
 
-class UDPServer : public Reference {
-	GDCLASS(UDPServer, Reference);
+class UDPServer : public RefCounted {
+	GDCLASS(UDPServer, RefCounted);
 
 protected:
 	enum {
@@ -44,7 +44,7 @@ protected:
 
 	struct Peer {
 		PacketPeerUDP *peer;
-		IP_Address ip;
+		IPAddress ip;
 		uint16_t port = 0;
 
 		bool operator==(const Peer &p_other) const {
@@ -53,21 +53,18 @@ protected:
 	};
 	uint8_t recv_buffer[PACKET_BUFFER_SIZE];
 
-	int bind_port = 0;
-	IP_Address bind_address;
-
 	List<Peer> peers;
 	List<Peer> pending;
 	int max_pending_connections = 16;
 
 	Ref<NetSocket> _sock;
-
 	static void _bind_methods();
 
 public:
-	void remove_peer(IP_Address p_ip, int p_port);
-	Error listen(uint16_t p_port, const IP_Address &p_bind_address = IP_Address("*"));
+	void remove_peer(IPAddress p_ip, int p_port);
+	Error listen(uint16_t p_port, const IPAddress &p_bind_address = IPAddress("*"));
 	Error poll();
+	int get_local_port() const;
 	bool is_listening() const;
 	bool is_connection_available() const;
 	void set_max_pending_connections(int p_max);