X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/4a3bbe30291bf06fbd81fbf9c3d8c1cbea4aa17c..8282b50a0adb1f3d6fabd4a64af3bfcd3afb8df5:/src/gras/Transport/transport_plugin_tcp.c diff --git a/src/gras/Transport/transport_plugin_tcp.c b/src/gras/Transport/transport_plugin_tcp.c index 6fa9cb5ae1..6a4b64d6d1 100644 --- a/src/gras/Transport/transport_plugin_tcp.c +++ b/src/gras/Transport/transport_plugin_tcp.c @@ -2,50 +2,40 @@ /* tcp trp (transport) - send/receive a bunch of bytes from a tcp socket */ -/* Authors: Martin Quinson */ -/* Copyright (C) 2004 Martin Quinson. */ +/* Copyright (c) 2004 Martin Quinson. All rights reserved. */ /* This program is free software; you can redistribute it and/or modify it - under the terms of the license (GNU LGPL) which comes with this package. */ - -#include /* close() pipe() read() write() */ -#include /* close() pipe() read() write() */ -#include /* sometimes required for #include */ -#include /* TCP_NODELAY */ -#include /* inet_ntoa() */ -#include /* getprotobyname() */ -#include /* struct timeval */ -#include /* errno */ -#include /* waitpid() */ -#include /* getpeername() socket() */ -#include -#include /* memset */ - -#include "gras_private.h" + * under the terms of the license (GNU LGPL) which comes with this package. */ + +#include "portable.h" + +#if 0 +# include /* close() pipe() read() write() */ +# include /* waitpid() */ +#endif + + #include "transport_private.h" -GRAS_LOG_NEW_DEFAULT_SUBCATEGORY(trp_tcp,transport,"TCP transport"); +XBT_LOG_NEW_DEFAULT_SUBCATEGORY(trp_tcp,transport,"TCP transport"); /*** *** Prototypes ***/ -gras_error_t gras_trp_tcp_socket_client(gras_trp_plugin_t *self, - const char *host, - unsigned short port, - /* OUT */ gras_socket_t *sock); -gras_error_t gras_trp_tcp_socket_server(gras_trp_plugin_t *self, - unsigned short port, - /* OUT */ gras_socket_t *sock); -gras_error_t gras_trp_tcp_socket_accept(gras_socket_t *sock, - gras_socket_t **dst); - -void gras_trp_tcp_socket_close(gras_socket_t *sd); +xbt_error_t gras_trp_tcp_socket_client(gras_trp_plugin_t *self, + gras_socket_t sock); +xbt_error_t gras_trp_tcp_socket_server(gras_trp_plugin_t *self, + gras_socket_t sock); +xbt_error_t gras_trp_tcp_socket_accept(gras_socket_t sock, + gras_socket_t *dst); + +void gras_trp_tcp_socket_close(gras_socket_t sd); -gras_error_t gras_trp_tcp_chunk_send(gras_socket_t *sd, +xbt_error_t gras_trp_tcp_chunk_send(gras_socket_t sd, const char *data, long int size); -gras_error_t gras_trp_tcp_chunk_recv(gras_socket_t *sd, +xbt_error_t gras_trp_tcp_chunk_recv(gras_socket_t sd, char *data, long int size); @@ -74,12 +64,9 @@ typedef struct { /*** *** Code ***/ -gras_error_t -gras_trp_tcp_setup(gras_trp_plugin_t *plug) { +xbt_error_t gras_trp_tcp_setup(gras_trp_plugin_t *plug) { - gras_trp_tcp_plug_data_t *data = gras_new(gras_trp_tcp_plug_data_t,1); - if (!data) - RAISE_MALLOC; + gras_trp_tcp_plug_data_t *data = xbt_new(gras_trp_tcp_plug_data_t,1); FD_ZERO(&(data->msg_socks)); FD_ZERO(&(data->raw_socks)); @@ -96,23 +83,22 @@ gras_trp_tcp_setup(gras_trp_plugin_t *plug) { plug->data = (void*)data; plug->exit = gras_trp_tcp_exit; - + return no_error; } void gras_trp_tcp_exit(gras_trp_plugin_t *plug) { DEBUG1("Exit plugin TCP (free %p)", plug->data); - free(plug->data); + xbt_free(plug->data); } -gras_error_t gras_trp_tcp_socket_client(gras_trp_plugin_t *self, - const char *host, - unsigned short port, - /* OUT */ gras_socket_t *sock){ +xbt_error_t gras_trp_tcp_socket_client(gras_trp_plugin_t *self, + gras_socket_t sock){ struct sockaddr_in addr; struct hostent *he; struct in_addr *haddr; + int size = sock->bufSize * 1024; sock->incoming = 1; /* TCP sockets are duplex'ed */ @@ -121,14 +107,19 @@ gras_error_t gras_trp_tcp_socket_client(gras_trp_plugin_t *self, if (sock->sd < 0) { RAISE1(system_error, "Failed to create socket: %s", - strerror (errno)); + sock_errstr); + } + + if (setsockopt(sock->sd, SOL_SOCKET, SO_RCVBUF, (char *)&size, sizeof(size)) || + setsockopt(sock->sd, SOL_SOCKET, SO_SNDBUF, (char *)&size, sizeof(size))) { + WARN1("setsockopt failed, cannot set buffer size: %s",sock_errstr); } - he = gethostbyname (host); + he = gethostbyname (sock->peer_name); if (he == NULL) { RAISE2(system_error, "Failed to lookup hostname %s: %s", - host, strerror (errno)); + sock->peer_name, sock_errstr); } haddr = ((struct in_addr *) (he->h_addr_list)[0]); @@ -136,13 +127,13 @@ gras_error_t gras_trp_tcp_socket_client(gras_trp_plugin_t *self, memset(&addr, 0, sizeof(struct sockaddr_in)); memcpy (&addr.sin_addr, haddr, sizeof(struct in_addr)); addr.sin_family = AF_INET; - addr.sin_port = htons (port); + addr.sin_port = htons (sock->peer_port); if (connect (sock->sd, (struct sockaddr*) &addr, sizeof (addr)) < 0) { - close(sock->sd); + tcp_close(sock->sd); RAISE3(system_error, "Failed to connect socket to %s:%d (%s)", - host, port, strerror (errno)); + sock->peer_name, sock->peer_port, sock_errstr); } return no_error; @@ -153,10 +144,9 @@ gras_error_t gras_trp_tcp_socket_client(gras_trp_plugin_t *self, * * Open a socket used to receive messages. */ -gras_error_t gras_trp_tcp_socket_server(gras_trp_plugin_t *self, - unsigned short port, - /* OUT */ gras_socket_t *sock){ -// int size = bufSize * 1024; +xbt_error_t gras_trp_tcp_socket_server(gras_trp_plugin_t *self, + /* OUT */ gras_socket_t sock){ + int size = sock->bufSize * 1024; int on = 1; struct sockaddr_in server; @@ -164,27 +154,32 @@ gras_error_t gras_trp_tcp_socket_server(gras_trp_plugin_t *self, sock->outgoing = 1; /* TCP => duplex mode */ - server.sin_port = htons((u_short)port); + server.sin_port = htons((u_short)sock->port); server.sin_addr.s_addr = INADDR_ANY; server.sin_family = AF_INET; if((sock->sd = socket(AF_INET, SOCK_STREAM, 0)) < 0) { - RAISE1(system_error,"socket allocation failed: %s", strerror(errno)); + RAISE1(system_error,"Socket allocation failed: %s", sock_errstr); } - (void)setsockopt(sock->sd, SOL_SOCKET, SO_REUSEADDR, - (char *)&on, sizeof(on)); - /* - (void)setsockopt(sd, SOL_SOCKET, SO_RCVBUF, (char *)&size, sizeof(size)); - (void)setsockopt(sd, SOL_SOCKET, SO_SNDBUF, (char *)&size, sizeof(size)); - */ + if (setsockopt(sock->sd, SOL_SOCKET, SO_REUSEADDR, (char *)&on, sizeof(on))) { + RAISE1(system_error,"setsockopt failed, cannot condition the socket: %s", + sock_errstr); + } + + if (setsockopt(sock->sd, SOL_SOCKET, SO_RCVBUF, (char *)&size, sizeof(size)) || + setsockopt(sock->sd, SOL_SOCKET, SO_SNDBUF, (char *)&size, sizeof(size))) { + WARN1("setsockopt failed, cannot set buffer size: %s", + sock_errstr); + } + if (bind(sock->sd, (struct sockaddr *)&server, sizeof(server)) == -1) { - close(sock->sd); - RAISE2(system_error,"Cannot bind to port %d: %s",port, strerror(errno)); + tcp_close(sock->sd); + RAISE2(system_error,"Cannot bind to port %d: %s",sock->port, sock_errstr); } if (listen(sock->sd, 5) < 0) { - close(sock->sd); - RAISE2(system_error,"Cannot listen to port %d: %s",port,strerror(errno)); + tcp_close(sock->sd); + RAISE2(system_error,"Cannot listen to port %d: %s",sock->port,sock_errstr); } if (sock->raw) @@ -192,24 +187,25 @@ gras_error_t gras_trp_tcp_socket_server(gras_trp_plugin_t *self, else FD_SET(sock->sd, &(tcp->msg_socks)); - DEBUG2("Openned a server socket on port %d (sock %d)",port,sock->sd); + DEBUG2("Openned a server socket on port %d (sock %d)",sock->port,sock->sd); return no_error; } -gras_error_t -gras_trp_tcp_socket_accept(gras_socket_t *sock, - gras_socket_t **dst) { - gras_socket_t *res; - gras_error_t errcode; +xbt_error_t +gras_trp_tcp_socket_accept(gras_socket_t sock, + gras_socket_t *dst) { + gras_socket_t res; struct sockaddr_in peer_in; socklen_t peer_in_len = sizeof(peer_in); int sd; int tmp_errno; - - TRY(gras_trp_socket_new(1,&res)); + int size; + + XBT_IN; + gras_trp_socket_new(1,&res); sd = accept(sock->sd, (struct sockaddr *)&peer_in, &peer_in_len); tmp_errno = errno; @@ -217,23 +213,24 @@ gras_trp_tcp_socket_accept(gras_socket_t *sock, if(sd == -1) { gras_socket_close(sock); RAISE1(system_error, - "Accept failed (%s). Droping server socket.", strerror(tmp_errno)); + "Accept failed (%s). Droping server socket.", sock_errstr); } else { int i = 1; socklen_t s = sizeof(int); if (setsockopt(sd, SOL_SOCKET, SO_KEEPALIVE, (char *)&i, s) || setsockopt(sd, TcpProtoNumber(), TCP_NODELAY, (char *)&i, s)) { - WARN0("setsockopt failed, cannot condition the accepted socket"); + RAISE1(system_error,"setsockopt failed, cannot condition the socket: %s", + sock_errstr); + } + + res->bufSize = sock->bufSize; + size = sock->bufSize * 1024; + if (setsockopt(sd, SOL_SOCKET, SO_RCVBUF, (char *)&size, sizeof(size)) + || setsockopt(sd, SOL_SOCKET, SO_SNDBUF, (char *)&size, sizeof(size))) { + WARN1("setsockopt failed, cannot set buffer size: %s", + sock_errstr); } - - /* FIXME: bufSize removed until we can have optionsets - i = ((gras_trp_tcp_sock_specific_t*)sock->specific)->buffsize; - if (setsockopt(sd, SOL_SOCKET, SO_RCVBUF, (char *)&i, s) - || setsockopt(sd, SOL_SOCKET, SO_SNDBUF, (char *)&i, s)) { - WARNING0("setsockopt failed, cannot set buffsize"); - } - */ res->plugin = sock->plugin; res->incoming = sock->incoming; @@ -245,7 +242,7 @@ gras_trp_tcp_socket_accept(gras_socket_t *sock, /* FIXME: Lock to protect inet_ntoa */ if (((struct sockaddr *)&peer_in)->sa_family != AF_INET) { - res->peer_name = strdup("unknown"); + res->peer_name = (char*)strdup("unknown"); } else { struct in_addr addrAsInAddr; char *tmp; @@ -254,9 +251,9 @@ gras_trp_tcp_socket_accept(gras_socket_t *sock, tmp = inet_ntoa(addrAsInAddr); if (tmp != NULL) { - res->peer_name = strdup(tmp); + res->peer_name = (char*)strdup(tmp); } else { - res->peer_name = strdup("unknown"); + res->peer_name = (char*)strdup("unknown"); } } @@ -264,11 +261,12 @@ gras_trp_tcp_socket_accept(gras_socket_t *sock, *dst = res; + XBT_OUT; return no_error; } } -void gras_trp_tcp_socket_close(gras_socket_t *sock){ +void gras_trp_tcp_socket_close(gras_socket_t sock){ gras_trp_tcp_plug_data_t *tcp; if (!sock) return; /* close only once */ @@ -290,17 +288,23 @@ void gras_trp_tcp_socket_close(gras_socket_t *sock){ } } */ - /* forget about the socket */ - if (sock->raw) +#ifndef HAVE_WINSOCK_H + /* forget about the socket + ... but not when using winsock since accept'ed socket can not fit + into the fd_set*/ + if (sock->raw){ FD_CLR(sock->sd, &(tcp->raw_socks)); - else + } else { FD_CLR(sock->sd, &(tcp->msg_socks)); - + } +#endif + /* close the socket */ - if(close(sock->sd) < 0) { + if(tcp_close(sock->sd) < 0) { WARN3("error while closing tcp socket %d: %d (%s)\n", - sock->sd, errno, strerror(errno)); + sock->sd, sock_errno, sock_errstr); } + } /** @@ -308,24 +312,24 @@ void gras_trp_tcp_socket_close(gras_socket_t *sock){ * * Send data on a TCP socket */ -gras_error_t -gras_trp_tcp_chunk_send(gras_socket_t *sock, +xbt_error_t +gras_trp_tcp_chunk_send(gras_socket_t sock, const char *data, long int size) { /* TCP sockets are in duplex mode, don't check direction */ - gras_assert0(size >= 0, "Cannot send a negative amount of data"); + xbt_assert0(size >= 0, "Cannot send a negative amount of data"); while (size) { int status = 0; - status = write(sock->sd, data, (size_t)size); + status = tcp_write(sock->sd, data, (size_t)size); DEBUG3("write(%d, %p, %ld);", sock->sd, data, size); - if (status == -1) { + if (status <= 0) { RAISE4(system_error,"write(%d,%p,%ld) failed: %s", sock->sd, data, size, - strerror(errno)); + sock_errstr); } if (status) { @@ -343,25 +347,25 @@ gras_trp_tcp_chunk_send(gras_socket_t *sock, * * Receive data on a TCP socket. */ -gras_error_t -gras_trp_tcp_chunk_recv(gras_socket_t *sock, +xbt_error_t +gras_trp_tcp_chunk_recv(gras_socket_t sock, char *data, long int size) { /* TCP sockets are in duplex mode, don't check direction */ - gras_assert0(sock, "Cannot recv on an NULL socket"); - gras_assert0(size >= 0, "Cannot receive a negative amount of data"); + xbt_assert0(sock, "Cannot recv on an NULL socket"); + xbt_assert0(size >= 0, "Cannot receive a negative amount of data"); while (size) { int status = 0; - status = read(sock->sd, data, (size_t)size); + status = tcp_read(sock->sd, data, (size_t)size); DEBUG3("read(%d, %p, %ld);", sock->sd, data, size); - if (status == -1) { + if (status <= 0) { RAISE4(system_error,"read(%d,%p,%d) failed: %s", sock->sd, data, (int)size, - strerror(errno)); + sock_errstr); } if (status) { @@ -387,9 +391,133 @@ static int TcpProtoNumber(void) { if(returnValue == 0) { fetchedEntry = getprotobyname("tcp"); - gras_assert0(fetchedEntry, "getprotobyname(tcp) gave NULL"); + xbt_assert0(fetchedEntry, "getprotobyname(tcp) gave NULL"); returnValue = fetchedEntry->p_proto; } return returnValue; } + +/* Data exchange over raw sockets. Placing this in there is a kind of crude hack. + It means that the only possible raw are TCP where we may want to do UDP for them. + But I fail to find a good internal organization for now. We may want to split + raw and regular sockets more efficiently. +*/ +xbt_error_t gras_socket_raw_exchange(gras_socket_t peer, + int sender, + unsigned int timeout, + unsigned long int exp_size, + unsigned long int msg_size) { + char *chunk; + int res_last, msg_sofar, exp_sofar; + + fd_set rd_set; +/* int rv; */ + + struct timeval timeOut; + + chunk = xbt_malloc(msg_size); + + for (exp_sofar=0; exp_sofar < exp_size; exp_sofar += msg_size) { + for(msg_sofar=0; msg_sofar < msg_size; msg_sofar += res_last) { + + if(sender) { + res_last = send(peer->sd, chunk, msg_size - msg_sofar, 0); + } else { + res_last = 0; + FD_ZERO(&rd_set); + FD_SET(peer->sd,&rd_set); + timeOut.tv_sec = timeout; + timeOut.tv_usec = 0; + + if (0 < select(peer->sd+1,&rd_set,NULL,NULL,&timeOut)) + res_last = recv(peer->sd, chunk, msg_size-msg_sofar, 0); + + } + if (res_last == 0) { + /* No progress done, bail out */ + xbt_free(chunk); + RAISE0(unknown_error,"Not exchanged a single byte, bailing out"); + } + } + } + + xbt_free(chunk); + return no_error; +} + + +#ifdef HAVE_WINSOCK_H +#define RETSTR( x ) case x: return #x + +const char *gras_wsa_err2string( int err ) { + switch( err ) { + RETSTR( WSAEINTR ); + RETSTR( WSAEBADF ); + RETSTR( WSAEACCES ); + RETSTR( WSAEFAULT ); + RETSTR( WSAEINVAL ); + RETSTR( WSAEMFILE ); + RETSTR( WSAEWOULDBLOCK ); + RETSTR( WSAEINPROGRESS ); + RETSTR( WSAEALREADY ); + RETSTR( WSAENOTSOCK ); + RETSTR( WSAEDESTADDRREQ ); + RETSTR( WSAEMSGSIZE ); + RETSTR( WSAEPROTOTYPE ); + RETSTR( WSAENOPROTOOPT ); + RETSTR( WSAEPROTONOSUPPORT ); + RETSTR( WSAESOCKTNOSUPPORT ); + RETSTR( WSAEOPNOTSUPP ); + RETSTR( WSAEPFNOSUPPORT ); + RETSTR( WSAEAFNOSUPPORT ); + RETSTR( WSAEADDRINUSE ); + RETSTR( WSAEADDRNOTAVAIL ); + RETSTR( WSAENETDOWN ); + RETSTR( WSAENETUNREACH ); + RETSTR( WSAENETRESET ); + RETSTR( WSAECONNABORTED ); + RETSTR( WSAECONNRESET ); + RETSTR( WSAENOBUFS ); + RETSTR( WSAEISCONN ); + RETSTR( WSAENOTCONN ); + RETSTR( WSAESHUTDOWN ); + RETSTR( WSAETOOMANYREFS ); + RETSTR( WSAETIMEDOUT ); + RETSTR( WSAECONNREFUSED ); + RETSTR( WSAELOOP ); + RETSTR( WSAENAMETOOLONG ); + RETSTR( WSAEHOSTDOWN ); + RETSTR( WSAEHOSTUNREACH ); + RETSTR( WSAENOTEMPTY ); + RETSTR( WSAEPROCLIM ); + RETSTR( WSAEUSERS ); + RETSTR( WSAEDQUOT ); + RETSTR( WSAESTALE ); + RETSTR( WSAEREMOTE ); + RETSTR( WSASYSNOTREADY ); + RETSTR( WSAVERNOTSUPPORTED ); + RETSTR( WSANOTINITIALISED ); + RETSTR( WSAEDISCON ); + +#ifdef HAVE_WINSOCK2 + RETSTR( WSAENOMORE ); + RETSTR( WSAECANCELLED ); + RETSTR( WSAEINVALIDPROCTABLE ); + RETSTR( WSAEINVALIDPROVIDER ); + RETSTR( WSASYSCALLFAILURE ); + RETSTR( WSASERVICE_NOT_FOUND ); + RETSTR( WSATYPE_NOT_FOUND ); + RETSTR( WSA_E_NO_MORE ); + RETSTR( WSA_E_CANCELLED ); + RETSTR( WSAEREFUSED ); +#endif /* HAVE_WINSOCK2 */ + + RETSTR( WSAHOST_NOT_FOUND ); + RETSTR( WSATRY_AGAIN ); + RETSTR( WSANO_RECOVERY ); + RETSTR( WSANO_DATA ); + } + return "unknown WSA error"; +} +#endif /* HAVE_WINSOCK_H */