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::connect(const SockAddr &addr)
  30 {
  31         if(connected)
  32                 throw bad_socket_state("already connected");
  33
  34         SockAddr::SysAddr sa = addr.to_sys();
  35
  36 #ifdef WIN32
  37         int err = WSAConnect(priv->handle, reinterpret_cast<sockaddr *>(&sa.addr), sa.size, 0, 0, 0, 0);
  38         if(err==SOCKET_ERROR)
  39         {
  40                 int err_code = WSAGetLastError();
  41                 if(err_code==WSAEWOULDBLOCK)
  42                 {
  43                         connecting = true;
  44                         WSAEventSelect(priv->handle, *priv->event, FD_CONNECT);
  45                         set_events(IO::P_OUTPUT);
  46                 }
  47                 else
  48                         throw system_error("connect", err_code);
  49         }
  50 #else
  51         int err = ::connect(priv->handle, reinterpret_cast<sockaddr *>(&sa.addr), sa.size);
  52         if(err==-1)
  53         {
  54                 if(errno==EINPROGRESS)
  55                 {
  56                         connecting = true;
  57                         set_events(IO::P_OUTPUT);
  58                 }
  59                 else
  60                         throw system_error("connect");
  61         }
  62 #endif
  63
  64         delete peer_addr;
  65         peer_addr = addr.copy();
  66
  67         delete local_addr;
  68         SockAddr::SysAddr lsa;
  69         getsockname(priv->handle, reinterpret_cast<sockaddr *>(&lsa.addr), &lsa.size);
  70         local_addr = SockAddr::from_sys(lsa);
  71
  72         if(err==0)
  73         {
  74                 connected = true;
  75                 set_events(IO::P_INPUT);
  76                 signal_connect_finished.emit(0);
  77         }
  78
  79         return connected;
  80 }
  81
  82 bool StreamSocket::poll_connect(const Time::TimeDelta &timeout)
  83 {
  84         if(!connecting)
  85                 throw bad_socket_state("not connecting");
  86
  87         int res = poll(*this, IO::P_OUTPUT, timeout);
  88         if(res==-1)
  89 #ifdef WIN32
  90                 throw system_error("poll", WSAGetLastError());
  91 #else
  92                 throw system_error("poll");
  93 #endif
  94         else if(res>0)
  95         {
  96                 connecting = false;
  97
  98                 int err;
  99                 socklen_t len = sizeof(int);
 100                 get_option(SOL_SOCKET, SO_ERROR, &err, &len);
 101
 102                 if(err!=0)
 103                 {
 104                         set_events(IO::P_NONE);
 105 #ifdef WIN32
 106                         throw system_error("connect", WSAGetLastError());
 107 #else
 108                         throw system_error("connect");
 109 #endif
 110                 }
 111
 112 #ifdef WIN32
 113                 WSAEventSelect(priv->handle, *priv->event, FD_READ|FD_CLOSE);
 114 #endif
 115                 set_events(IO::P_INPUT);
 116
 117                 connected = true;
 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