From 0b806ee0fc9097fa7bda7ac0109191c9c5e0a1ac Mon Sep 17 00:00:00 2001 From: Juan Linietsky Date: Sun, 9 Feb 2014 22:10:30 -0300 Subject: GODOT IS OPEN SOURCE --- drivers/unix/tcp_server_posix.cpp | 178 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 178 insertions(+) create mode 100644 drivers/unix/tcp_server_posix.cpp (limited to 'drivers/unix/tcp_server_posix.cpp') diff --git a/drivers/unix/tcp_server_posix.cpp b/drivers/unix/tcp_server_posix.cpp new file mode 100644 index 0000000000..f91ad36ef9 --- /dev/null +++ b/drivers/unix/tcp_server_posix.cpp @@ -0,0 +1,178 @@ +/*************************************************************************/ +/* tcp_server_posix.cpp */ +/*************************************************************************/ +/* This file is part of: */ +/* GODOT ENGINE */ +/* http://www.godotengine.org */ +/*************************************************************************/ +/* 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. */ +/*************************************************************************/ +#include "tcp_server_posix.h" +#include "stream_peer_tcp_posix.h" + +#ifdef UNIX_ENABLED + +#include + +#include +#include +#include +#include +#include +#include +#include +#ifndef NO_FCNTL +#include +#else +#include +#endif +#ifdef JAVASCRIPT_ENABLED +#include +#endif +#include +#include +#include +TCP_Server* TCPServerPosix::_create() { + + return memnew(TCPServerPosix); +}; + +void TCPServerPosix::make_default() { + + TCP_Server::_create = TCPServerPosix::_create; +}; + +Error TCPServerPosix::listen(uint16_t p_port,const List *p_accepted_hosts) { + + printf("********* listening on port %i\n", p_port); + int sockfd; + sockfd = socket(AF_INET, SOCK_STREAM, 0); + ERR_FAIL_COND_V(sockfd == -1, FAILED); +#ifndef NO_FCNTL + fcntl(sockfd, F_SETFL, O_NONBLOCK); +#else + int bval = 1; + ioctl(sockfd, FIONBIO, &bval); +#endif + + int reuse=1; + if(setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (char *)&reuse, sizeof(reuse)) < 0) { + + printf("REUSEADDR 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); + + if (bind(sockfd, (struct sockaddr *)&my_addr, sizeof my_addr) != -1) { + + if (::listen(sockfd, 1) == -1) { + + close(sockfd); + ERR_FAIL_V(FAILED); + }; + }; + + if (listen_sockfd != -1) { + + printf("FAILED\n"); + stop(); + }; + + listen_sockfd = sockfd; + + printf("OK! %i\n", listen_sockfd); + return OK; +}; + +bool TCPServerPosix::is_connection_available() const { + + if (listen_sockfd == -1) { + return false; + }; + + struct pollfd pfd; + pfd.fd = listen_sockfd; + pfd.events = POLLIN; + pfd.revents = 0; + + int ret = poll(&pfd, 1, 0); + ERR_FAIL_COND_V(ret < 0, FAILED); + + if (ret && (pfd.revents & POLLIN)) { + printf("has connection!\n"); + return true; + }; + + return false; +}; + +Ref TCPServerPosix::take_connection() { + + if (!is_connection_available()) { + return Ref(); + }; + + struct sockaddr_in their_addr; + socklen_t sin_size = sizeof(their_addr); + int fd = accept(listen_sockfd, (struct sockaddr *)&their_addr, &sin_size); + ERR_FAIL_COND_V(fd == -1, Ref()); +#ifndef NO_FCNTL + fcntl(fd, F_SETFL, O_NONBLOCK); +#else + int bval = 1; + ioctl(fd, FIONBIO, &bval); +#endif + + Ref conn = memnew(StreamPeerTCPPosix); + 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 TCPServerPosix::stop() { + + if (listen_sockfd != -1) { + print_line("CLOSING CONNECTION"); + int ret = close(listen_sockfd); + ERR_FAIL_COND(ret!=0); + }; + + listen_sockfd = -1; +}; + + +TCPServerPosix::TCPServerPosix() { + + listen_sockfd = -1; +}; + +TCPServerPosix::~TCPServerPosix() { + + stop(); +}; +#endif -- cgit v1.2.3