| 
									
										
										
										
											2023-01-05 13:25:55 +01:00
										 |  |  | /**************************************************************************/ | 
					
						
							|  |  |  | /*  udp_server.h                                                          */ | 
					
						
							|  |  |  | /**************************************************************************/ | 
					
						
							|  |  |  | /*                         This file is part of:                          */ | 
					
						
							|  |  |  | /*                             GODOT ENGINE                               */ | 
					
						
							|  |  |  | /*                        https://godotengine.org                         */ | 
					
						
							|  |  |  | /**************************************************************************/ | 
					
						
							|  |  |  | /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */ | 
					
						
							|  |  |  | /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur.                  */ | 
					
						
							|  |  |  | /*                                                                        */ | 
					
						
							|  |  |  | /* Permission is hereby granted, free of charge, to any person obtaining  */ | 
					
						
							|  |  |  | /* a copy of this software and associated documentation files (the        */ | 
					
						
							|  |  |  | /* "Software"), to deal in the Software without restriction, including    */ | 
					
						
							|  |  |  | /* without limitation the rights to use, copy, modify, merge, publish,    */ | 
					
						
							|  |  |  | /* distribute, sublicense, and/or sell copies of the Software, and to     */ | 
					
						
							|  |  |  | /* permit persons to whom the Software is furnished to do so, subject to  */ | 
					
						
							|  |  |  | /* the following conditions:                                              */ | 
					
						
							|  |  |  | /*                                                                        */ | 
					
						
							|  |  |  | /* The above copyright notice and this permission notice shall be         */ | 
					
						
							|  |  |  | /* included in all copies or substantial portions of the Software.        */ | 
					
						
							|  |  |  | /*                                                                        */ | 
					
						
							|  |  |  | /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,        */ | 
					
						
							|  |  |  | /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF     */ | 
					
						
							|  |  |  | /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */ | 
					
						
							|  |  |  | /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY   */ | 
					
						
							|  |  |  | /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,   */ | 
					
						
							|  |  |  | /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE      */ | 
					
						
							|  |  |  | /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                 */ | 
					
						
							|  |  |  | /**************************************************************************/ | 
					
						
							| 
									
										
										
										
											2019-08-22 19:03:35 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  | #ifndef UDP_SERVER_H
 | 
					
						
							|  |  |  | #define UDP_SERVER_H
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #include "core/io/net_socket.h"
 | 
					
						
							|  |  |  | #include "core/io/packet_peer_udp.h"
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-06-04 18:03:15 +02:00
										 |  |  | class UDPServer : public RefCounted { | 
					
						
							|  |  |  | 	GDCLASS(UDPServer, RefCounted); | 
					
						
							| 
									
										
										
										
											2019-08-22 19:03:35 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  | protected: | 
					
						
							| 
									
										
										
										
											2020-06-25 15:32:50 +02:00
										 |  |  | 	enum { | 
					
						
							|  |  |  | 		PACKET_BUFFER_SIZE = 65536 | 
					
						
							|  |  |  | 	}; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	struct Peer { | 
					
						
							| 
									
										
										
										
											2022-04-04 15:06:57 +02:00
										 |  |  | 		PacketPeerUDP *peer = nullptr; | 
					
						
							| 
									
										
										
										
											2021-05-06 02:48:18 +02:00
										 |  |  | 		IPAddress ip; | 
					
						
							| 
									
										
										
										
											2020-06-25 15:32:50 +02:00
										 |  |  | 		uint16_t port = 0; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		bool operator==(const Peer &p_other) const { | 
					
						
							|  |  |  | 			return (ip == p_other.ip && port == p_other.port); | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	}; | 
					
						
							|  |  |  | 	uint8_t recv_buffer[PACKET_BUFFER_SIZE]; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	List<Peer> peers; | 
					
						
							|  |  |  | 	List<Peer> pending; | 
					
						
							|  |  |  | 	int max_pending_connections = 16; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-08-22 19:03:35 +02:00
										 |  |  | 	Ref<NetSocket> _sock; | 
					
						
							| 
									
										
										
										
											2020-06-25 15:32:50 +02:00
										 |  |  | 	static void _bind_methods(); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-08-22 19:03:35 +02:00
										 |  |  | public: | 
					
						
							| 
									
										
										
										
											2021-05-06 02:48:18 +02:00
										 |  |  | 	void remove_peer(IPAddress p_ip, int p_port); | 
					
						
							|  |  |  | 	Error listen(uint16_t p_port, const IPAddress &p_bind_address = IPAddress("*")); | 
					
						
							| 
									
										
										
										
											2020-06-25 15:32:50 +02:00
										 |  |  | 	Error poll(); | 
					
						
							| 
									
										
										
										
											2021-03-21 10:15:30 +00:00
										 |  |  | 	int get_local_port() const; | 
					
						
							| 
									
										
										
										
											2019-08-22 19:03:35 +02:00
										 |  |  | 	bool is_listening() const; | 
					
						
							|  |  |  | 	bool is_connection_available() const; | 
					
						
							| 
									
										
										
										
											2020-06-25 15:32:50 +02:00
										 |  |  | 	void set_max_pending_connections(int p_max); | 
					
						
							|  |  |  | 	int get_max_pending_connections() const; | 
					
						
							| 
									
										
										
										
											2019-08-22 19:03:35 +02:00
										 |  |  | 	Ref<PacketPeerUDP> take_connection(); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	void stop(); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	UDPServer(); | 
					
						
							|  |  |  | 	~UDPServer(); | 
					
						
							|  |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #endif // UDP_SERVER_H
 |