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