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                 set_events(IO::P_INPUT);
  77                 signal_connect_finished.emit(0);
  78         }
  79
  80         return connected;
  81 }
  82
  83 bool StreamSocket::poll_connect(const Time::TimeDelta &timeout)
  84 {
  85         if(!connecting)
  86                 return false;
  87
  88         IO::PollEvent res = poll(*this, IO::P_OUTPUT, timeout);
  89         if(res&IO::P_OUTPUT)
  90         {
  91                 connecting = false;
  92
  93                 int err;
  94                 socklen_t len = sizeof(int);
  95                 priv->get_option(SOL_SOCKET, SO_ERROR, &err, &len);
  96
  97                 if(err!=0)
  98                 {
  99                         set_events(IO::P_NONE);
 100 #ifdef WIN32
 101                         throw system_error("connect", WSAGetLastError());
 102 #else
 103                         throw system_error("connect");
 104 #endif
 105                 }
 106
 107 #ifdef WIN32
 108                 WSAEventSelect(priv->handle, *priv->event, FD_READ|FD_CLOSE);
 109 #endif
 110                 set_events(IO::P_INPUT);
 111
 112                 connected = true;
 113         }
 114
 115         return connected;
 116 }
 117
 118 void StreamSocket::on_event(IO::PollEvent ev)
 119 {
 120         if((ev&(IO::P_OUTPUT|IO::P_ERROR)) && connecting)
 121         {
 122                 int err;
 123                 socklen_t len = sizeof(err);
 124                 priv->get_option(SOL_SOCKET, SO_ERROR, &err, &len);
 125
 126                 connecting = false;
 127                 connected = (err==0);
 128                 if(err)
 129                 {
 130                         system_error exc("connect", err);
 131                         signal_connect_finished.emit(&exc);
 132                 }
 133                 else
 134                         signal_connect_finished.emit(0);
 135
 136                 if(err!=0)
 137                 {
 138                         delete peer_addr;
 139                         peer_addr = 0;
 140                 }
 141
 142 #ifdef WIN32
 143                 WSAEventSelect(priv->handle, *priv->event, FD_READ|FD_CLOSE);
 144 #endif
 145                 set_events((err==0) ? IO::P_INPUT : IO::P_NONE);
 146         }
 147 }
 148
 149 } // namespace Net
 150 } // namespace Msp