diff options
Diffstat (limited to 'platform/windows/packet_peer_udp_winsock.cpp')
| -rw-r--r-- | platform/windows/packet_peer_udp_winsock.cpp | 124 |
1 files changed, 56 insertions, 68 deletions
diff --git a/platform/windows/packet_peer_udp_winsock.cpp b/platform/windows/packet_peer_udp_winsock.cpp index 276b9b663..2df8addec 100644 --- a/platform/windows/packet_peer_udp_winsock.cpp +++ b/platform/windows/packet_peer_udp_winsock.cpp @@ -35,19 +35,19 @@ int PacketPeerUDPWinsock::get_available_packet_count() const { - Error err = const_cast<PacketPeerUDPWinsock*>(this)->_poll(false); - if (err!=OK) + Error err = const_cast<PacketPeerUDPWinsock *>(this)->_poll(false); + if (err != OK) return 0; return queue_count; } -Error PacketPeerUDPWinsock::get_packet(const uint8_t **r_buffer,int &r_buffer_size) const{ +Error PacketPeerUDPWinsock::get_packet(const uint8_t **r_buffer, int &r_buffer_size) const { - Error err = const_cast<PacketPeerUDPWinsock*>(this)->_poll(false); - if (err!=OK) + Error err = const_cast<PacketPeerUDPWinsock *>(this)->_poll(false); + if (err != OK) return err; - if (queue_count==0) + if (queue_count == 0) return ERR_UNAVAILABLE; uint32_t size; @@ -55,31 +55,30 @@ Error PacketPeerUDPWinsock::get_packet(const uint8_t **r_buffer,int &r_buffer_si rb.read(&type, 1, true); if (type == IP::TYPE_IPV4) { uint8_t ip[4]; - rb.read(ip,4,true); + rb.read(ip, 4, true); packet_ip.set_ipv4(ip); } else { uint8_t ip[16]; - rb.read(ip,16,true); + rb.read(ip, 16, true); packet_ip.set_ipv6(ip); }; - rb.read((uint8_t*)&packet_port,4,true); - rb.read((uint8_t*)&size,4,true); - rb.read(packet_buffer,size,true); + rb.read((uint8_t *)&packet_port, 4, true); + rb.read((uint8_t *)&size, 4, true); + rb.read(packet_buffer, size, true); --queue_count; - *r_buffer=packet_buffer; - r_buffer_size=size; + *r_buffer = packet_buffer; + r_buffer_size = size; return OK; - } -Error PacketPeerUDPWinsock::put_packet(const uint8_t *p_buffer,int p_buffer_size){ +Error PacketPeerUDPWinsock::put_packet(const uint8_t *p_buffer, int p_buffer_size) { ERR_FAIL_COND_V(!peer_addr.is_valid(), ERR_UNCONFIGURED); - if(sock_type==IP::TYPE_NONE) + if (sock_type == IP::TYPE_NONE) sock_type = peer_addr.is_ipv4() ? IP::TYPE_IPV4 : IP::TYPE_IPV6; int sock = _get_socket(); - ERR_FAIL_COND_V( sock == -1, FAILED ); + ERR_FAIL_COND_V(sock == -1, FAILED); struct sockaddr_storage addr; size_t addr_size = _set_sockaddr(&addr, peer_addr, peer_port, sock_type); @@ -87,7 +86,7 @@ Error PacketPeerUDPWinsock::put_packet(const uint8_t *p_buffer,int p_buffer_size errno = 0; int err; - while ( (err = sendto(sock, (const char*)p_buffer, p_buffer_size, 0, (struct sockaddr*)&addr, addr_size)) != p_buffer_size) { + while ((err = sendto(sock, (const char *)p_buffer, p_buffer_size, 0, (struct sockaddr *)&addr, addr_size)) != p_buffer_size) { if (WSAGetLastError() != WSAEWOULDBLOCK) { return FAILED; @@ -97,21 +96,20 @@ Error PacketPeerUDPWinsock::put_packet(const uint8_t *p_buffer,int p_buffer_size return OK; } -int PacketPeerUDPWinsock::get_max_packet_size() const{ +int PacketPeerUDPWinsock::get_max_packet_size() const { return 512; // uhm maybe not } - void PacketPeerUDPWinsock::_set_blocking(bool p_blocking) { //am no windows expert //hope this is the right thing - if (blocking==p_blocking) + if (blocking == p_blocking) return; - blocking=p_blocking; - unsigned long par = blocking?0:1; + blocking = p_blocking; + unsigned long par = blocking ? 0 : 1; if (ioctlsocket(sockfd, FIONBIO, &par)) { perror("setting non-block mode"); //close(); @@ -121,69 +119,67 @@ void PacketPeerUDPWinsock::_set_blocking(bool p_blocking) { Error PacketPeerUDPWinsock::listen(int p_port, IP_Address p_bind_address, int p_recv_buffer_size) { - ERR_FAIL_COND_V(sockfd!=-1,ERR_ALREADY_IN_USE); - ERR_FAIL_COND_V(!p_bind_address.is_valid() && !p_bind_address.is_wildcard(),ERR_INVALID_PARAMETER); + ERR_FAIL_COND_V(sockfd != -1, ERR_ALREADY_IN_USE); + ERR_FAIL_COND_V(!p_bind_address.is_valid() && !p_bind_address.is_wildcard(), ERR_INVALID_PARAMETER); sock_type = IP::TYPE_ANY; - if(p_bind_address.is_valid()) + if (p_bind_address.is_valid()) sock_type = p_bind_address.is_ipv4() ? IP::TYPE_IPV4 : IP::TYPE_IPV6; int sock = _get_socket(); - if (sock == -1 ) + if (sock == -1) return ERR_CANT_CREATE; - struct sockaddr_storage addr = {0}; + struct sockaddr_storage addr = { 0 }; size_t addr_size = _set_listen_sockaddr(&addr, p_port, sock_type, IP_Address()); - if (bind(sock, (struct sockaddr*)&addr, addr_size) == -1 ) { + if (bind(sock, (struct sockaddr *)&addr, addr_size) == -1) { close(); return ERR_UNAVAILABLE; } - blocking=true; + blocking = true; printf("UDP Connection listening on port %i\n", p_port); rb.resize(nearest_shift(p_recv_buffer_size)); return OK; } -void PacketPeerUDPWinsock::close(){ +void PacketPeerUDPWinsock::close() { if (sockfd != -1) ::closesocket(sockfd); - sockfd=-1; + sockfd = -1; sock_type = IP::TYPE_NONE; rb.resize(16); - queue_count=0; + queue_count = 0; } - Error PacketPeerUDPWinsock::wait() { return _poll(true); } Error PacketPeerUDPWinsock::_poll(bool p_wait) { - if (sockfd==-1) { + if (sockfd == -1) { return FAILED; } _set_blocking(p_wait); - - struct sockaddr_storage from = {0}; + struct sockaddr_storage from = { 0 }; int len = sizeof(struct sockaddr_storage); int ret; - while ( (ret = recvfrom(sockfd, (char*)recv_buffer, MIN((int)sizeof(recv_buffer),MAX(rb.space_left()-24, 0)), 0, (struct sockaddr*)&from, &len)) > 0) { + while ((ret = recvfrom(sockfd, (char *)recv_buffer, MIN((int)sizeof(recv_buffer), MAX(rb.space_left() - 24, 0)), 0, (struct sockaddr *)&from, &len)) > 0) { uint32_t port = 0; if (from.ss_family == AF_INET) { uint8_t type = (uint8_t)IP::TYPE_IPV4; rb.write(&type, 1); - struct sockaddr_in* sin_from = (struct sockaddr_in*)&from; - rb.write((uint8_t*)&sin_from->sin_addr, 4); + struct sockaddr_in *sin_from = (struct sockaddr_in *)&from; + rb.write((uint8_t *)&sin_from->sin_addr, 4); port = ntohs(sin_from->sin_port); } else if (from.ss_family == AF_INET6) { @@ -191,8 +187,8 @@ Error PacketPeerUDPWinsock::_poll(bool p_wait) { uint8_t type = (uint8_t)IP::TYPE_IPV6; rb.write(&type, 1); - struct sockaddr_in6* s6_from = (struct sockaddr_in6*)&from; - rb.write((uint8_t*)&s6_from->sin6_addr, 16); + struct sockaddr_in6 *s6_from = (struct sockaddr_in6 *)&from; + rb.write((uint8_t *)&s6_from->sin6_addr, 16); port = ntohs(s6_from->sin6_port); @@ -202,44 +198,39 @@ Error PacketPeerUDPWinsock::_poll(bool p_wait) { rb.write(&type, 1); }; - rb.write((uint8_t*)&port, 4); - rb.write((uint8_t*)&ret, 4); + rb.write((uint8_t *)&port, 4); + rb.write((uint8_t *)&ret, 4); rb.write(recv_buffer, ret); len = sizeof(struct sockaddr_storage); ++queue_count; }; - if (ret == SOCKET_ERROR){ + if (ret == SOCKET_ERROR) { int error = WSAGetLastError(); - if (error == WSAEWOULDBLOCK){ + if (error == WSAEWOULDBLOCK) { // Expected when doing non-blocking sockets, retry later. - } - else if (error == WSAECONNRESET){ + } else if (error == WSAECONNRESET) { // If the remote target does not accept messages, this error may occur, but is harmless. // Once the remote target gets available, this message will disappear for new messages. - } - else - { + } else { close(); return FAILED; } } - if (ret == 0) { close(); return FAILED; }; - return OK; } -bool PacketPeerUDPWinsock::is_listening() const{ +bool PacketPeerUDPWinsock::is_listening() const { - return sockfd!=-1; + return sockfd != -1; } IP_Address PacketPeerUDPWinsock::get_packet_address() const { @@ -247,14 +238,14 @@ IP_Address PacketPeerUDPWinsock::get_packet_address() const { return packet_ip; } -int PacketPeerUDPWinsock::get_packet_port() const{ +int PacketPeerUDPWinsock::get_packet_port() const { return packet_port; } int PacketPeerUDPWinsock::_get_socket() { - ERR_FAIL_COND_V(sock_type==IP::TYPE_NONE,-1); + ERR_FAIL_COND_V(sock_type == IP::TYPE_NONE, -1); if (sockfd != -1) return sockfd; @@ -264,11 +255,10 @@ int PacketPeerUDPWinsock::_get_socket() { return sockfd; } +void PacketPeerUDPWinsock::set_dest_address(const IP_Address &p_address, int p_port) { -void PacketPeerUDPWinsock::set_dest_address(const IP_Address& p_address,int p_port) { - - peer_addr=p_address; - peer_port=p_port; + peer_addr = p_address; + peer_port = p_port; } void PacketPeerUDPWinsock::make_default() { @@ -276,19 +266,17 @@ void PacketPeerUDPWinsock::make_default() { PacketPeerUDP::_create = PacketPeerUDPWinsock::_create; }; - -PacketPeerUDP* PacketPeerUDPWinsock::_create() { +PacketPeerUDP *PacketPeerUDPWinsock::_create() { return memnew(PacketPeerUDPWinsock); }; - PacketPeerUDPWinsock::PacketPeerUDPWinsock() { - sockfd=-1; - packet_port=0; - queue_count=0; - peer_port=0; + sockfd = -1; + packet_port = 0; + queue_count = 0; + peer_port = 0; sock_type = IP::TYPE_NONE; rb.resize(16); } |
