X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fstreamsocket.cpp;h=b06cf69ca9314cb6c750c1147513f9b4dc9c3a0a;hb=6489612c788dd57721fc181ff26a981b35668d1e;hp=5794ecd82914b16c3c8d22c7edc28a56ffffcbf5;hpb=dbfb303f336037cf7ca4fca3d8bfe504c624b501;p=libs%2Fnet.git diff --git a/source/streamsocket.cpp b/source/streamsocket.cpp index 5794ecd..b06cf69 100644 --- a/source/streamsocket.cpp +++ b/source/streamsocket.cpp @@ -14,8 +14,7 @@ namespace Msp { namespace Net { StreamSocket::StreamSocket(const Private &p, const SockAddr &paddr): - Socket(p, paddr), - connecting(false) + ClientSocket(p, paddr) { #ifdef WIN32 WSAEventSelect(priv->handle, *priv->event, FD_READ|FD_CLOSE); @@ -24,54 +23,10 @@ StreamSocket::StreamSocket(const Private &p, const SockAddr &paddr): } StreamSocket::StreamSocket(Family af, int proto): - Socket(af, SOCK_STREAM, proto), - connecting(false) + ClientSocket(af, SOCK_STREAM, proto) { } -int StreamSocket::poll_connect(const Time::TimeDelta &timeout) -{ - if(!connecting) - throw bad_socket_state("not connecting"); - - int res = poll(*this, IO::P_OUTPUT, timeout); - if(res==-1) -#ifdef WIN32 - throw system_error("poll", WSAGetLastError()); -#else - throw system_error("poll"); -#endif - else if(res>0) - { - connecting = false; - - int err; - socklen_t len = sizeof(int); - get_option(SOL_SOCKET, SO_ERROR, &err, &len); - - if(err!=0) - { - set_events(IO::P_NONE); -#ifdef WIN32 - throw system_error("connect", WSAGetLastError()); -#else - throw system_error("connect"); -#endif - } - -#ifdef WIN32 - WSAEventSelect(priv->handle, *priv->event, FD_READ|FD_CLOSE); -#endif - set_events(IO::P_INPUT); - - connected = true; - - return 0; - } - - return 1; -} - -int StreamSocket::connect(const SockAddr &addr) +bool StreamSocket::connect(const SockAddr &addr) { if(connected) throw bad_socket_state("already connected"); @@ -121,7 +76,48 @@ int StreamSocket::connect(const SockAddr &addr) signal_connect_finished.emit(0); } - return (err==0)?0:1; + return connected; +} + +bool StreamSocket::poll_connect(const Time::TimeDelta &timeout) +{ + if(!connecting) + throw bad_socket_state("not connecting"); + + int res = poll(*this, IO::P_OUTPUT, timeout); + if(res==-1) +#ifdef WIN32 + throw system_error("poll", WSAGetLastError()); +#else + throw system_error("poll"); +#endif + else if(res>0) + { + connecting = false; + + int err; + socklen_t len = sizeof(int); + get_option(SOL_SOCKET, SO_ERROR, &err, &len); + + if(err!=0) + { + set_events(IO::P_NONE); +#ifdef WIN32 + throw system_error("connect", WSAGetLastError()); +#else + throw system_error("connect"); +#endif + } + +#ifdef WIN32 + WSAEventSelect(priv->handle, *priv->event, FD_READ|FD_CLOSE); +#endif + set_events(IO::P_INPUT); + + connected = true; + } + + return connected; } void StreamSocket::on_event(IO::PollEvent ev)