source/net/streamsocket.cpp

   1 #include "platform_api.h"
   2 #include "streamsocket.h"
   3 #include <msp/core/systemerror.h>
   4 #include <msp/io/handle_private.h>
   5 #include <msp/io/poll.h>
   6 #include <msp/strings/format.h>
   7 #include "sockaddr_private.h"
   8 #include "socket_private.h"
   9
  10 namespace Msp {
  11 namespace Net {
  12
  13 StreamSocket::StreamSocket(const Private &p, const SockAddr &paddr):
  14         ClientSocket(p, paddr)
  15 {
  16         set_socket_events(S_INPUT);
  17 }
  18
  19 StreamSocket::StreamSocket(Family af, int proto):
  20         ClientSocket(af, SOCK_STREAM, proto)
  21 { }
  22
  23 bool StreamSocket::connect(const SockAddr &addr)
  24 {
  25         if(connected)
  26                 throw bad_socket_state("already connected");
  27
  28         SockAddr::SysAddr sa = addr.to_sys();
  29         int err = ::connect(priv->handle, reinterpret_cast<const sockaddr *>(&sa.addr), sa.size);
  30         bool finished = check_sys_connect_error(err);
  31         if(!finished)
  32         {
  33                 connecting = true;
  34                 set_socket_events(S_CONNECT);
  35         }
  36
  37         peer_addr.reset(addr.copy());
  38
  39         SockAddr::SysAddr lsa;
  40         getsockname(priv->handle, reinterpret_cast<sockaddr *>(&lsa.addr), &lsa.size);
  41         local_addr.reset(SockAddr::new_from_sys(lsa));
  42
  43         if(finished)
  44         {
  45                 connected = true;
  46                 set_socket_events(S_INPUT);
  47                 signal_connect_finished.emit(0);
  48         }
  49
  50         return connected;
  51 }
  52
  53 bool StreamSocket::poll_connect(const Time::TimeDelta &timeout)
  54 {
  55         if(!connecting)
  56                 return false;
  57
  58         IO::PollEvent res = poll(*this, IO::P_OUTPUT, timeout);
  59         if(res&IO::P_OUTPUT)
  60         {
  61                 connecting = false;
  62
  63                 int err;
  64                 socklen_t len = sizeof(int);
  65                 priv->get_option(SOL_SOCKET, SO_ERROR, &err, &len);
  66
  67                 if(err!=0)
  68                 {
  69                         set_socket_events(S_NONE);
  70                         throw system_error("connect", err);
  71                 }
  72
  73                 set_socket_events(S_INPUT);
  74
  75                 connected = true;
  76         }
  77
  78         return connected;
  79 }
  80
  81 void StreamSocket::on_event(IO::PollEvent ev)
  82 {
  83         if((ev&(IO::P_OUTPUT|IO::P_ERROR)) && connecting)
  84         {
  85                 int err;
  86                 socklen_t len = sizeof(err);
  87                 priv->get_option(SOL_SOCKET, SO_ERROR, &err, &len);
  88
  89                 connecting = false;
  90                 connected = (err==0);
  91                 if(err)
  92                 {
  93                         system_error exc("connect", err);
  94                         signal_connect_finished.emit(&exc);
  95                 }
  96                 else
  97                         signal_connect_finished.emit(0);
  98
  99                 if(err!=0)
 100                         peer_addr.reset();
 101
 102                 set_socket_events((err==0) ? S_INPUT : S_NONE);
 103         }
 104 }
 105
 106 } // namespace Net
 107 } // namespace Msp