Implement TCP Server bind address
This commit is contained in:
parent
4198291cd4
commit
b2839343ca
9 changed files with 50 additions and 30 deletions
|
@ -44,16 +44,6 @@ TCP_Server* TCP_Server::create() {
|
|||
return _create();
|
||||
}
|
||||
|
||||
Error TCP_Server::_listen(uint16_t p_port, PoolVector<String> p_accepted_hosts) {
|
||||
|
||||
List<String> hosts;
|
||||
for(int i=0;i<p_accepted_hosts.size();i++)
|
||||
hosts.push_back(p_accepted_hosts.get(i));
|
||||
|
||||
return listen(p_port, hosts.size()?&hosts:NULL);
|
||||
|
||||
}
|
||||
|
||||
void TCP_Server::set_ip_type(IP::Type p_type) {
|
||||
stop();
|
||||
ip_type = p_type;
|
||||
|
@ -62,7 +52,7 @@ void TCP_Server::set_ip_type(IP::Type p_type) {
|
|||
void TCP_Server::_bind_methods() {
|
||||
|
||||
ClassDB::bind_method(_MD("set_ip_type","ip_type"),&TCP_Server::set_ip_type);
|
||||
ClassDB::bind_method(_MD("listen","port","accepted_hosts"),&TCP_Server::_listen,DEFVAL(PoolVector<String>()));
|
||||
ClassDB::bind_method(_MD("listen","port","bind_address"),&TCP_Server::listen,DEFVAL("*"));
|
||||
ClassDB::bind_method(_MD("is_connection_available"),&TCP_Server::is_connection_available);
|
||||
ClassDB::bind_method(_MD("take_connection"),&TCP_Server::take_connection);
|
||||
ClassDB::bind_method(_MD("stop"),&TCP_Server::stop);
|
||||
|
|
|
@ -43,12 +43,11 @@ protected:
|
|||
static TCP_Server* (*_create)();
|
||||
|
||||
//bind helper
|
||||
Error _listen(uint16_t p_port, PoolVector<String> p_accepted_hosts=PoolVector<String>());
|
||||
static void _bind_methods();
|
||||
public:
|
||||
|
||||
virtual void set_ip_type(IP::Type p_type);
|
||||
virtual Error listen(uint16_t p_port, const List<String> *p_accepted_hosts=NULL)=0;
|
||||
virtual Error listen(uint16_t p_port, const IP_Address p_bind_address=IP_Address("*"))=0;
|
||||
virtual bool is_connection_available() const=0;
|
||||
virtual Ref<StreamPeerTCP> take_connection()=0;
|
||||
|
||||
|
|
|
@ -130,7 +130,7 @@ Error PacketPeerUDPPosix::listen(int p_port, int p_recv_buffer_size) {
|
|||
return ERR_CANT_CREATE;
|
||||
|
||||
sockaddr_storage addr = {0};
|
||||
size_t addr_size = _set_listen_sockaddr(&addr, p_port, ip_type, NULL);
|
||||
size_t addr_size = _set_listen_sockaddr(&addr, p_port, ip_type, IP_Address());
|
||||
|
||||
if (bind(sock, (struct sockaddr*)&addr, addr_size) == -1 ) {
|
||||
close();
|
||||
|
|
|
@ -44,21 +44,29 @@ static size_t _set_sockaddr(struct sockaddr_storage* p_addr, const IP_Address& p
|
|||
};
|
||||
};
|
||||
|
||||
static size_t _set_listen_sockaddr(struct sockaddr_storage* p_addr, int p_port, IP::Type p_sock_type, const List<String> *p_accepted_hosts) {
|
||||
static size_t _set_listen_sockaddr(struct sockaddr_storage* p_addr, int p_port, IP::Type p_sock_type, const IP_Address p_bind_address) {
|
||||
|
||||
memset(p_addr, 0, sizeof(struct sockaddr_storage));
|
||||
if (p_sock_type == IP::TYPE_IPV4) {
|
||||
struct sockaddr_in* addr4 = (struct sockaddr_in*)p_addr;
|
||||
addr4->sin_family = AF_INET;
|
||||
addr4->sin_port = htons(p_port);
|
||||
addr4->sin_addr.s_addr = INADDR_ANY; // TODO: use accepted hosts list
|
||||
if(p_bind_address.is_valid()) {
|
||||
copymem(&addr4->sin_addr.s_addr, p_bind_address.get_ipv4(), 4);
|
||||
} else {
|
||||
addr4->sin_addr.s_addr = INADDR_ANY;
|
||||
}
|
||||
return sizeof(sockaddr_in);
|
||||
} else {
|
||||
struct sockaddr_in6* addr6 = (struct sockaddr_in6*)p_addr;
|
||||
|
||||
addr6->sin6_family = AF_INET6;
|
||||
addr6->sin6_port = htons(p_port);
|
||||
addr6->sin6_addr = in6addr_any; // TODO: use accepted hosts list
|
||||
if(p_bind_address.is_valid()) {
|
||||
copymem(&addr6->sin6_addr.s6_addr, p_bind_address.get_ipv6(), 16);
|
||||
} else {
|
||||
addr6->sin6_addr = in6addr_any;
|
||||
}
|
||||
return sizeof(sockaddr_in6);
|
||||
};
|
||||
};
|
||||
|
|
|
@ -68,10 +68,19 @@ void TCPServerPosix::make_default() {
|
|||
TCP_Server::_create = TCPServerPosix::_create;
|
||||
};
|
||||
|
||||
Error TCPServerPosix::listen(uint16_t p_port,const List<String> *p_accepted_hosts) {
|
||||
Error TCPServerPosix::listen(uint16_t p_port,const IP_Address p_bind_address) {
|
||||
|
||||
ERR_FAIL_COND_V(listen_sockfd!=-1,ERR_ALREADY_IN_USE);
|
||||
ERR_FAIL_COND_V(!p_bind_address.is_valid() && !p_bind_address.is_wildcard(), ERR_INVALID_PARAMETER);
|
||||
|
||||
int sockfd;
|
||||
sockfd = _socket_create(ip_type, SOCK_STREAM, IPPROTO_TCP);
|
||||
sock_type = ip_type;
|
||||
|
||||
// If the bind address is valid use its type as the socket type
|
||||
if (p_bind_address.is_valid())
|
||||
sock_type = p_bind_address.is_ipv4() ? IP::TYPE_IPV4 : IP::TYPE_IPV6;
|
||||
|
||||
sockfd = _socket_create(sock_type, SOCK_STREAM, IPPROTO_TCP);
|
||||
|
||||
ERR_FAIL_COND_V(sockfd == -1, FAILED);
|
||||
|
||||
|
@ -88,9 +97,7 @@ Error TCPServerPosix::listen(uint16_t p_port,const List<String> *p_accepted_host
|
|||
}
|
||||
|
||||
struct sockaddr_storage addr;
|
||||
size_t addr_size = _set_listen_sockaddr(&addr, p_port, ip_type, p_accepted_hosts);
|
||||
|
||||
// automatically fill with my IP TODO: use p_accepted_hosts
|
||||
size_t addr_size = _set_listen_sockaddr(&addr, p_port, sock_type, p_bind_address);
|
||||
|
||||
if (bind(sockfd, (struct sockaddr *)&addr, addr_size) != -1) {
|
||||
|
||||
|
@ -157,7 +164,7 @@ Ref<StreamPeerTCP> TCPServerPosix::take_connection() {
|
|||
int port;
|
||||
_set_ip_addr_port(ip, port, &their_addr);
|
||||
|
||||
conn->set_socket(fd, ip, port, ip_type);
|
||||
conn->set_socket(fd, ip, port, sock_type);
|
||||
|
||||
return conn;
|
||||
};
|
||||
|
@ -170,6 +177,7 @@ void TCPServerPosix::stop() {
|
|||
};
|
||||
|
||||
listen_sockfd = -1;
|
||||
sock_type = IP::TYPE_NONE;
|
||||
};
|
||||
|
||||
|
||||
|
@ -177,6 +185,7 @@ TCPServerPosix::TCPServerPosix() {
|
|||
|
||||
listen_sockfd = -1;
|
||||
ip_type = IP::TYPE_ANY;
|
||||
sock_type = IP::TYPE_NONE;
|
||||
};
|
||||
|
||||
TCPServerPosix::~TCPServerPosix() {
|
||||
|
|
|
@ -35,12 +35,13 @@
|
|||
class TCPServerPosix : public TCP_Server {
|
||||
|
||||
int listen_sockfd;
|
||||
IP::Type sock_type;
|
||||
|
||||
static TCP_Server* _create();
|
||||
|
||||
public:
|
||||
|
||||
virtual Error listen(uint16_t p_port,const List<String> *p_accepted_hosts=NULL);
|
||||
virtual Error listen(uint16_t p_port, IP_Address p_bind_address=IP_Address("*"));
|
||||
virtual bool is_connection_available() const;
|
||||
virtual Ref<StreamPeerTCP> take_connection();
|
||||
|
||||
|
|
|
@ -122,7 +122,7 @@ Error PacketPeerUDPWinsock::listen(int p_port, int p_recv_buffer_size) {
|
|||
return ERR_CANT_CREATE;
|
||||
|
||||
struct sockaddr_storage addr = {0};
|
||||
size_t addr_size = _set_listen_sockaddr(&addr, p_port, ip_type, NULL);
|
||||
size_t addr_size = _set_listen_sockaddr(&addr, p_port, ip_type, IP_Address());
|
||||
|
||||
if (bind(sock, (struct sockaddr*)&addr, addr_size) == -1 ) {
|
||||
close();
|
||||
|
|
|
@ -63,10 +63,20 @@ void TCPServerWinsock::cleanup() {
|
|||
};
|
||||
|
||||
|
||||
Error TCPServerWinsock::listen(uint16_t p_port,const List<String> *p_accepted_hosts) {
|
||||
Error TCPServerWinsock::listen(uint16_t p_port,const IP_Address p_bind_address) {
|
||||
|
||||
ERR_FAIL_COND_V(listen_sockfd!=-1,ERR_ALREADY_IN_USE);
|
||||
ERR_FAIL_COND_V(!p_bind_address.is_valid() && !p_bind_address.is_wildcard(), ERR_INVALID_PARAMETER);
|
||||
|
||||
int sockfd;
|
||||
sockfd = _socket_create(ip_type, SOCK_STREAM, IPPROTO_TCP);
|
||||
sock_type = ip_type;
|
||||
|
||||
// If the bind address is valid use its type as the socket type
|
||||
if (p_bind_address.is_valid())
|
||||
sock_type = p_bind_address.is_ipv4() ? IP::TYPE_IPV4 : IP::TYPE_IPV6;
|
||||
|
||||
|
||||
sockfd = _socket_create(sock_type, SOCK_STREAM, IPPROTO_TCP);
|
||||
ERR_FAIL_COND_V(sockfd == INVALID_SOCKET, FAILED);
|
||||
|
||||
unsigned long par = 1;
|
||||
|
@ -77,7 +87,7 @@ Error TCPServerWinsock::listen(uint16_t p_port,const List<String> *p_accepted_ho
|
|||
};
|
||||
|
||||
struct sockaddr_storage my_addr;
|
||||
size_t addr_size = _set_listen_sockaddr(&my_addr, p_port, ip_type, p_accepted_hosts);
|
||||
size_t addr_size = _set_listen_sockaddr(&my_addr, p_port, sock_type, p_bind_address);
|
||||
|
||||
int reuse=1;
|
||||
if(setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (char *)&reuse, sizeof(reuse)) < 0) {
|
||||
|
@ -150,7 +160,7 @@ Ref<StreamPeerTCP> TCPServerWinsock::take_connection() {
|
|||
int port;
|
||||
_set_ip_addr_port(ip, port, &their_addr);
|
||||
|
||||
conn->set_socket(fd, ip, port, ip_type);
|
||||
conn->set_socket(fd, ip, port, sock_type);
|
||||
|
||||
return conn;
|
||||
};
|
||||
|
@ -162,6 +172,7 @@ void TCPServerWinsock::stop() {
|
|||
};
|
||||
|
||||
listen_sockfd = -1;
|
||||
sock_type = IP::TYPE_NONE;
|
||||
};
|
||||
|
||||
|
||||
|
@ -169,6 +180,7 @@ TCPServerWinsock::TCPServerWinsock() {
|
|||
|
||||
listen_sockfd = INVALID_SOCKET;
|
||||
ip_type = IP::TYPE_ANY;
|
||||
sock_type = IP::TYPE_NONE;
|
||||
};
|
||||
|
||||
TCPServerWinsock::~TCPServerWinsock() {
|
||||
|
|
|
@ -34,12 +34,13 @@
|
|||
class TCPServerWinsock : public TCP_Server {
|
||||
|
||||
int listen_sockfd;
|
||||
IP::Type sock_type;
|
||||
|
||||
static TCP_Server* _create();
|
||||
|
||||
public:
|
||||
|
||||
virtual Error listen(uint16_t p_port,const List<String> *p_accepted_hosts=NULL);
|
||||
virtual Error listen(uint16_t p_port, const IP_Address p_bind_address=IP_Address("*"));
|
||||
virtual bool is_connection_available() const;
|
||||
virtual Ref<StreamPeerTCP> take_connection();
|
||||
|
||||
|
|
Loading…
Reference in a new issue