source/net/streamsocket.cpp

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