diff options
Diffstat (limited to 'mcop/tcpconnection.cpp')
-rw-r--r-- | mcop/tcpconnection.cpp | 153 |
1 files changed, 153 insertions, 0 deletions
diff --git a/mcop/tcpconnection.cpp b/mcop/tcpconnection.cpp new file mode 100644 index 0000000..722ed2c --- /dev/null +++ b/mcop/tcpconnection.cpp @@ -0,0 +1,153 @@ + /* + + Copyright (C) 2000 Stefan Westerfeld + + This library is free software; you can redistribute it and/or + modify it under the terms of the GNU Library General Public + License as published by the Free Software Foundation; either + version 2 of the License, or (at your option) any later version. + + This library is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + Library General Public License for more details. + + You should have received a copy of the GNU Library General Public License + along with this library; see the file COPYING.LIB. If not, write to + the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, + Boston, MA 02110-1301, USA. + + */ + +#include "tcpconnection.h" +#include "dispatcher.h" +#include <sys/types.h> +#include <sys/socket.h> +#include <stdio.h> +#define queue cqueue +#include <netdb.h> +#undef queue +#include <netinet/in.h> +#include <netinet/tcp.h> +#include <arpa/inet.h> +#include <sys/un.h> +#include <errno.h> +#include "debug.h" +#include <stdlib.h> +#include <string.h> + +using namespace std; +using namespace Arts; + +static struct sockaddr_in *parse_tcp_url(const char *url) +{ + static struct sockaddr_in addr; + + char *work = strdup(url); + + char *type = strtok(work,":"); + if(type == 0 || strcmp(type,"tcp") != 0) { + free(work); + return 0; + } + + char *host = strtok(NULL,":"); + if(host == 0) { + free(work); + return 0; + } + + char *port = strtok(NULL,":\n"); + if(port == 0) { + free(work); + return 0; + } + + long portno = atol(port); + if(portno < 1 || portno > 65535) { + free(work); + return 0; + } + + struct hostent *server = gethostbyname(host); + if(server == 0) + { + arts_warning("parse_tcp_url: unknown host '%s'",host); + free(work); + return 0; + } + + memset((void *) &addr, '\0', sizeof(addr)); + addr.sin_family = AF_INET; + addr.sin_addr.s_addr = *(u_long *)server->h_addr; + addr.sin_port = htons(portno); + + free(work); + return &addr; +} + +static int tcp_connect(const char *url) +{ + struct sockaddr_in *remote_addr = parse_tcp_url(url); + if(remote_addr == 0) + { + arts_warning("tcp_connect: couldn't parse url %s",url); + return -1; + } + + int my_socket = socket(AF_INET,SOCK_STREAM,0); + if(my_socket < 0) + { + arts_warning("tcp_connect: unable to open socket for read"); + return -1; + } + + struct linger lin; + lin.l_onoff=1; /* block a closing socket for 1 second */ + lin.l_linger=100; /* if data is waiting to be sent */ + if ( setsockopt( my_socket, SOL_SOCKET, SO_LINGER, + (char*)&lin, sizeof(struct linger) ) < 0 ) + { + arts_warning("tcp_connect: unable to set socket linger value to %d", + lin.l_linger); + close(my_socket); + return -1; + } + +#ifdef TCP_NODELAY + // enable TCP sending without nagle algorithm - this sends out requests + // faster, because they are not queued in the hope that more data will + // need to be sent soon + int on = 1; + if(setsockopt(my_socket, IPPROTO_TCP, TCP_NODELAY, + (char *)&on, sizeof(on)) < 0) + { + arts_debug("couldn't set TCP_NODELAY on socket %d\n", my_socket); + } +#endif + + int rc; + rc=connect(my_socket,(struct sockaddr *)remote_addr, sizeof(*remote_addr)); + if(rc != 0) + { + arts_warning("tcp_connect: can't connect to server (%s)", url); + close(my_socket); + return -1; + } + + return my_socket; +} + +TCPConnection::TCPConnection(const string& url) +{ + fd = tcp_connect(url.c_str()); + _broken = (fd == -1); + + if(!_broken) + { + Dispatcher::the()->ioManager()->watchFD(fd, + IOType::read|IOType::except|IOType::reentrant,this); + initReceive(); + } +} |