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/handle_private.h>
   7 #include <msp/io/poll.h>
   8 #include <msp/strings/format.h>
   9 #include "sockaddr_private.h"
  10 #include "socket_private.h"
  11 #include "streamsocket.h"
  12
  13 namespace Msp {
  14 namespace Net {
  15
  16 StreamSocket::StreamSocket(const Private &p, const SockAddr &paddr):
  17         ClientSocket(p, paddr)
  18 {
  19 #ifdef WIN32
  20         WSAEventSelect(priv->handle, *priv->event, FD_READ|FD_CLOSE);
  21 #endif
  22         set_events(IO::P_INPUT);
  23 }
  24
  25 StreamSocket::StreamSocket(Family af, int proto):
  26         ClientSocket(af, SOCK_STREAM, proto)
  27 { }
  28
  29 bool StreamSocket::poll_connect(const Time::TimeDelta &timeout)
  30 {
  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(priv->handle, *priv->event, FD_READ|FD_CLOSE);
  61 #endif
  62                 set_events(IO::P_INPUT);
  63
  64                 connected = true;
  65         }
  66
  67         return connected;
  68 }
  69
  70 bool StreamSocket::connect(const SockAddr &addr)
  71 {
  72         if(connected)
  73                 throw bad_socket_state("already connected");
  74
  75         SockAddr::SysAddr sa = addr.to_sys();
  76
  77 #ifdef WIN32
  78         int err = WSAConnect(priv->handle, reinterpret_cast<sockaddr *>(&sa.addr), sa.size, 0, 0, 0, 0);
  79         if(err==SOCKET_ERROR)
  80         {
  81                 int err_code = WSAGetLastError();
  82                 if(err_code==WSAEWOULDBLOCK)
  83                 {
  84                         connecting = true;
  85                         WSAEventSelect(priv->handle, *priv->event, FD_CONNECT);
  86                         set_events(IO::P_OUTPUT);
  87                 }
  88                 else
  89                         throw system_error("connect", err_code);
  90         }
  91 #else
  92         int err = ::connect(priv->handle, reinterpret_cast<sockaddr *>(&sa.addr), sa.size);
  93         if(err==-1)
  94         {
  95                 if(errno==EINPROGRESS)
  96                 {
  97                         connecting = true;
  98                         set_events(IO::P_OUTPUT);
  99                 }
 100                 else
 101                         throw system_error("connect");
 102         }
 103 #endif
 104
 105         delete peer_addr;
 106         peer_addr = addr.copy();
 107
 108         delete local_addr;
 109         SockAddr::SysAddr lsa;
 110         getsockname(priv->handle, reinterpret_cast<sockaddr *>(&lsa.addr), &lsa.size);
 111         local_addr = SockAddr::from_sys(lsa);
 112
 113         if(err==0)
 114         {
 115                 connected = true;
 116                 set_events(IO::P_INPUT);
 117                 signal_connect_finished.emit(0);
 118         }
 119
 120         return connected;
 121 }
 122
 123 void StreamSocket::on_event(IO::PollEvent ev)
 124 {
 125         if((ev&(IO::P_OUTPUT|IO::P_ERROR)) && connecting)
 126         {
 127                 int err;
 128                 socklen_t len = sizeof(err);
 129                 get_option(SOL_SOCKET, SO_ERROR, &err, &len);
 130
 131                 connecting = false;
 132                 connected = (err==0);
 133                 signal_connect_finished.emit(err);
 134
 135                 if(err!=0)
 136                 {
 137                         delete peer_addr;
 138                         peer_addr = 0;
 139                 }
 140
 141 #ifdef WIN32
 142                 WSAEventSelect(priv->handle, *priv->event, FD_READ|FD_CLOSE);
 143 #endif
 144                 set_events((err==0) ? IO::P_INPUT : IO::P_NONE);
 145         }
 146 }
 147
 148 } // namespace Net
 149 } // namespace Msp