/*************************************************************************/ /* tcp_server_winsock.cpp */ /*************************************************************************/ /* This file is part of: */ /* GODOT ENGINE */ /* http://www.godotengine.org */ /*************************************************************************/ /* Copyright (c) 2007-2016 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. */ /*************************************************************************/ #include "tcp_server_winsock.h" #include "stream_peer_winsock.h" #include <winsock2.h> extern int winsock_refcount; TCP_Server* TCPServerWinsock::_create() { return memnew(TCPServerWinsock); }; void TCPServerWinsock::make_default() { TCP_Server::_create = TCPServerWinsock::_create; if (winsock_refcount == 0) { WSADATA data; WSAStartup(MAKEWORD(2,2), &data); }; ++winsock_refcount; }; void TCPServerWinsock::cleanup() { --winsock_refcount; if (winsock_refcount == 0) { WSACleanup(); }; }; Error TCPServerWinsock::listen(uint16_t p_port,const List<String> *p_accepted_hosts) { int sockfd; sockfd = socket(AF_INET, SOCK_STREAM, 0); ERR_FAIL_COND_V(sockfd == INVALID_SOCKET, FAILED); unsigned long par = 1; if (ioctlsocket(sockfd, FIONBIO, &par)) { perror("setting non-block mode"); stop(); return FAILED; }; struct sockaddr_in my_addr; my_addr.sin_family = AF_INET; // host byte order my_addr.sin_port = htons(p_port); // short, network byte order my_addr.sin_addr.s_addr = INADDR_ANY; // automatically fill with my IP TODO: use p_accepted_hosts memset(my_addr.sin_zero, '\0', sizeof my_addr.sin_zero); int reuse=1; if(setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (char *)&reuse, sizeof(reuse)) < 0) { printf("REUSEADDR failed!"); } if (bind(sockfd, (struct sockaddr *)&my_addr, sizeof my_addr) != SOCKET_ERROR) { if (::listen(sockfd, SOMAXCONN) == SOCKET_ERROR) { closesocket(sockfd); ERR_FAIL_V(FAILED); }; } else { return ERR_ALREADY_IN_USE; }; if (listen_sockfd != INVALID_SOCKET) { stop(); }; listen_sockfd = sockfd; return OK; }; bool TCPServerWinsock::is_connection_available() const { if (listen_sockfd == -1) { return false; }; timeval timeout; timeout.tv_sec = 0; timeout.tv_usec = 0; fd_set pfd; FD_ZERO(&pfd); FD_SET(listen_sockfd, &pfd); int ret = select(listen_sockfd + 1, &pfd, NULL, NULL, &timeout); ERR_FAIL_COND_V(ret < 0, 0); if (ret && (FD_ISSET(listen_sockfd, &pfd))) { return true; }; return false; }; Ref<StreamPeerTCP> TCPServerWinsock::take_connection() { if (!is_connection_available()) { return NULL; }; struct sockaddr_in their_addr; int sin_size = sizeof(their_addr); int fd = accept(listen_sockfd, (struct sockaddr *)&their_addr, &sin_size); ERR_FAIL_COND_V(fd == INVALID_SOCKET, NULL); Ref<StreamPeerWinsock> conn = memnew(StreamPeerWinsock); IP_Address ip; ip.host = (uint32_t)their_addr.sin_addr.s_addr; conn->set_socket(fd, ip, ntohs(their_addr.sin_port)); return conn; }; void TCPServerWinsock::stop() { if (listen_sockfd != INVALID_SOCKET) { closesocket(listen_sockfd); }; listen_sockfd = -1; }; TCPServerWinsock::TCPServerWinsock() { listen_sockfd = INVALID_SOCKET; }; TCPServerWinsock::~TCPServerWinsock() { stop(); };