source/net/datagramsocket.cpp

   1 #include "platform_api.h"
   2 #include "datagramsocket.h"
   3 #include <msp/core/systemerror.h>
   4 #include <msp/io/handle_private.h>
   5 #include <msp/strings/format.h>
   6 #include "sockaddr_private.h"
   7 #include "socket_private.h"
   8
   9 namespace Msp {
  10 namespace Net {
  11
  12 DatagramSocket::DatagramSocket(Family f, int p):
  13         ClientSocket(f, SOCK_DGRAM, p)
  14 {
  15         set_socket_events(S_INPUT);
  16 }
  17
  18 bool DatagramSocket::connect(const SockAddr &addr)
  19 {
  20         SockAddr::SysAddr sa = addr.to_sys();
  21         check_sys_connect_error(::connect(priv->handle, reinterpret_cast<const sockaddr *>(&sa.addr), sa.size));
  22
  23         peer_addr.reset(addr.copy());
  24
  25         SockAddr::SysAddr lsa;
  26         getsockname(priv->handle, reinterpret_cast<sockaddr *>(&lsa.addr), &lsa.size);
  27         local_addr.reset(SockAddr::new_from_sys(lsa));
  28
  29         connected = true;
  30
  31         return true;
  32 }
  33
  34 size_t DatagramSocket::sendto(const char *buf, size_t size, const SockAddr &addr)
  35 {
  36         if(size==0)
  37                 return 0;
  38
  39         SockAddr::SysAddr sa = addr.to_sys();
  40         return check_sys_error(::sendto(priv->handle, buf, size, 0, reinterpret_cast<const sockaddr *>(&sa.addr), sa.size), "sendto");
  41 }
  42
  43 size_t DatagramSocket::recvfrom(char *buf, size_t size, SockAddr *&from_addr)
  44 {
  45         if(size==0)
  46                 return 0;
  47
  48         SockAddr::SysAddr sa;
  49         size_t ret = check_sys_error(::recvfrom(priv->handle, buf, size, 0, reinterpret_cast<sockaddr *>(&sa.addr), &sa.size), "recvfrom");
  50         from_addr = SockAddr::new_from_sys(sa);
  51
  52         return ret;
  53 }
  54
  55 } // namespace Net
  56 } // namespace Msp