source/net/clientsocket.cpp

   1 #include "platform_api.h"
   2 #include <msp/core/systemerror.h>
   3 #include "clientsocket.h"
   4 #include "socket_private.h"
   5
   6 namespace Msp {
   7 namespace Net {
   8
   9 ClientSocket::ClientSocket(Family af, int type, int proto):
  10         Socket(af, type, proto)
  11 { }
  12
  13 ClientSocket::ClientSocket(const Private &p, const SockAddr &paddr):
  14         Socket(p),
  15         connected(true),
  16         peer_addr(paddr.copy())
  17 { }
  18
  19 ClientSocket::~ClientSocket()
  20 {
  21         signal_flush_required.emit();
  22
  23         delete peer_addr;
  24 }
  25
  26 void ClientSocket::shutdown(IO::Mode m)
  27 {
  28         int how;
  29         m = m&IO::M_RDWR;
  30 #ifdef _WIN32
  31         if(m==IO::M_READ)
  32                 how = SD_RECEIVE;
  33         else if(m==IO::M_WRITE)
  34                 how = SD_SEND;
  35         else if(m==IO::M_RDWR)
  36                 how = SD_BOTH;
  37 #else
  38         if(m==IO::M_READ)
  39                 how = SHUT_RD;
  40         else if(m==IO::M_WRITE)
  41                 how = SHUT_WR;
  42         else if(m==IO::M_RDWR)
  43                 how = SHUT_RDWR;
  44 #endif
  45         else
  46                 return;
  47
  48         ::shutdown(priv->handle, how);
  49         mode = mode&~m;
  50 }
  51
  52 const SockAddr &ClientSocket::get_peer_address() const
  53 {
  54         if(peer_addr==0)
  55                 throw bad_socket_state("not connected");
  56         return *peer_addr;
  57 }
  58
  59 size_t ClientSocket::do_write(const char *buf, size_t size)
  60 {
  61         check_access(IO::M_WRITE);
  62         if(!connected)
  63                 throw bad_socket_state("not connected");
  64
  65         if(size==0)
  66                 return 0;
  67
  68         return check_sys_error(::send(priv->handle, buf, size, 0), "send");
  69 }
  70
  71 size_t ClientSocket::do_read(char *buf, size_t size)
  72 {
  73         check_access(IO::M_READ);
  74         if(!connected)
  75                 throw bad_socket_state("not connected");
  76
  77         if(size==0)
  78                 return 0;
  79
  80         size_t ret = check_sys_error(::recv(priv->handle, buf, size, 0), "recv");
  81         if(ret==0 && !eof_flag)
  82         {
  83                 eof_flag = true;
  84                 set_socket_events(S_NONE);
  85                 signal_end_of_file.emit();
  86         }
  87
  88         return ret;
  89 }
  90
  91 } // namespace Net
  92 } // namespace Msp