source/streamsocket.cpp

   1 #ifndef WIN32
   2 #include <sys/socket.h>
   3 #endif
   4 #include <cerrno>
   5 #include <msp/core/systemerror.h>
   6 #include <msp/io/poll.h>
   7 #include <msp/strings/format.h>
   8 #include "streamsocket.h"
   9
  10 namespace Msp {
  11 namespace Net {
  12
  13 StreamSocket::StreamSocket(SocketHandle h, const SockAddr &paddr):
  14         Socket(h, paddr),
  15         connecting(false)
  16 {
  17 #ifdef WIN32
  18         WSAEventSelect(handle, event, FD_READ|FD_CLOSE);
  19 #endif
  20         set_events(IO::P_INPUT);
  21 }
  22
  23 StreamSocket::StreamSocket(Family af, int proto):
  24         Socket(af, SOCK_STREAM, proto),
  25         connecting(false)
  26 { }
  27
  28 int StreamSocket::poll_connect(const Time::TimeDelta &timeout)
  29 {
  30         check_state(false);
  31         if(!connecting)
  32                 throw bad_socket_state("not connecting");
  33
  34         int res = poll(*this, IO::P_OUTPUT, timeout);
  35         if(res==-1)
  36 #ifdef WIN32
  37                 throw system_error("poll", WSAGetLastError());
  38 #else
  39                 throw system_error("poll");
  40 #endif
  41         else if(res>0)
  42         {
  43                 connecting = false;
  44
  45                 int err;
  46                 socklen_t len = sizeof(int);
  47                 get_option(SOL_SOCKET, SO_ERROR, &err, &len);
  48
  49                 if(err!=0)
  50                 {
  51                         set_events(IO::P_NONE);
  52 #ifdef WIN32
  53                         throw system_error("connect", WSAGetLastError());
  54 #else
  55                         throw system_error("connect");
  56 #endif
  57                 }
  58
  59 #ifdef WIN32
  60                 WSAEventSelect(handle, event, FD_READ|FD_CLOSE);
  61 #endif
  62                 set_events(IO::P_INPUT);
  63
  64                 connected = true;
  65
  66                 return 0;
  67         }
  68
  69         return 1;
  70 }
  71
  72 int StreamSocket::connect(const SockAddr &addr)
  73 {
  74         check_state(false);
  75
  76         if(connected)
  77                 throw bad_socket_state("already connected");
  78
  79         sockaddr_storage sa;
  80         socklen_t size = addr.fill_sockaddr(sa);
  81
  82 #ifdef WIN32
  83         int err = WSAConnect(handle, reinterpret_cast<sockaddr *>(&sa), size, 0, 0, 0, 0);
  84         if(err==SOCKET_ERROR)
  85         {
  86                 int err_code = WSAGetLastError();
  87                 if(err_code==WSAEWOULDBLOCK)
  88                 {
  89                         connecting = true;
  90                         WSAEventSelect(handle, event, FD_CONNECT);
  91                         set_events(IO::P_OUTPUT);
  92                 }
  93                 else
  94                         throw system_error("connect", err_code);
  95         }
  96 #else
  97         int err = ::connect(handle, reinterpret_cast<sockaddr *>(&sa), size);
  98         if(err==-1)
  99         {
 100                 if(errno==EINPROGRESS)
 101                 {
 102                         connecting = true;
 103                         set_events(IO::P_OUTPUT);
 104                 }
 105                 else
 106                         throw system_error("connect");
 107         }
 108 #endif
 109
 110         delete peer_addr;
 111         peer_addr = addr.copy();
 112
 113         delete local_addr;
 114         size = sizeof(sockaddr_storage);
 115         getsockname(handle, reinterpret_cast<sockaddr *>(&sa), &size);
 116         local_addr = SockAddr::create(sa);
 117
 118         if(err==0)
 119         {
 120                 connected = true;
 121                 set_events(IO::P_INPUT);
 122                 signal_connect_finished.emit(0);
 123         }
 124
 125         return (err==0)?0:1;
 126 }
 127
 128 void StreamSocket::on_event(IO::PollEvent ev)
 129 {
 130         if((ev&(IO::P_OUTPUT|IO::P_ERROR)) && connecting)
 131         {
 132                 int err;
 133                 socklen_t len = sizeof(err);
 134                 get_option(SOL_SOCKET, SO_ERROR, &err, &len);
 135
 136                 connecting = false;
 137                 connected = (err==0);
 138                 signal_connect_finished.emit(err);
 139
 140                 if(err!=0)
 141                 {
 142                         delete peer_addr;
 143                         peer_addr = 0;
 144                 }
 145
 146 #ifdef WIN32
 147                 WSAEventSelect(handle, event, FD_READ|FD_CLOSE);
 148 #endif
 149                 set_events((err==0) ? IO::P_INPUT : IO::P_NONE);
 150         }
 151 }
 152
 153 } // namespace Net
 154 } // namespace Msp