DSocket.cpp

Go to the documentation of this file.

/*
* Copyright (C) 2012 RobotCub Consortium
* Author: Alessandro Scalzo alessandro.scalzo@iit.it
* CopyPolicy: Released under the terms of the GNU GPL v2.0.
*
*/
 
#include "DSocket.h"
#include <ace/Time_Value.h>
#include <ace/OS_NS_sys_socket.h>
 
//#include <arpa/inet.h>
 
DSocket::DSocket()
{ 
    ACE_OS::socket_init(2,2);
    mSocket=NULL; 
}
DSocket::~DSocket()
{ 
    Close(); 
    ACE_OS::socket_fini();
}
 
bool DSocket::Create(ACE_UINT16 port,std::string& address)
{
    mSocket=new ACE_SOCK_Dgram_Bcast(ACE_INET_Addr(port,address.c_str()));
 
    return mSocket!=NULL;
}
 
bool DSocket::Create(ACE_UINT16 port,ACE_UINT32 address)
{
    mSocket=new ACE_SOCK_Dgram_Bcast(ACE_INET_Addr(port,address));
 
    return mSocket!=NULL;
}
 
bool DSocket::Create(eOipv4addr_t ipv4, eOipv4port_t port)
{
    // eOipv4port_t is in host format (little endian) (as ACE requires).
    // eOipv4addr_t is in network order (big endian) (but ACE requires host format)
    // hence ... use ntohl() for ipv4
    ACE_UINT32 address = ntohl(ipv4);
    return Create(port, address);
}
 
void DSocket::SendTo(void* data,size_t len,ACE_UINT16 port,std::string& address)
{
    mSocket->send(data,len,ACE_INET_Addr(port,address.c_str()));
}
 
void DSocket::SendTo(void* data,size_t len,ACE_UINT16 port,ACE_UINT32 address)
{
    mSocket->send(data,len,ACE_INET_Addr(port,address));
}
 
void DSocket::SendTo(eOipv4addr_t ipv4, eOipv4port_t port, void* data, size_t len)
{
    ACE_UINT32 address = ntohl(ipv4);
    mSocket->send(data, len, ACE_INET_Addr(port, address));
}
 
void DSocket::SendBroad(void* data,size_t len,ACE_UINT16 port)
{
    mSocket->send(data,len,port);
}
 
 
void DSocket::SendBroad(eOipv4port_t port, void* data, size_t len)
{
    mSocket->send(data, len, port);
}
 
ssize_t DSocket::ReceiveFrom(void* data,size_t len,std::string &address,ACE_UINT16 &port,int wait_msec)
{
    ACE_Time_Value tv(wait_msec/1000,(wait_msec%1000)*1000);
    ACE_INET_Addr ace_addr;
    ssize_t nrec=mSocket->recv(data,len,ace_addr,0,&tv);
 
    if (nrec>0)
    {
        address=ace_addr.get_host_addr();
        port=ace_addr.get_port_number();
    }
    else
    {
        address="";
        port=0;
    }
 
    return nrec;
}
 
ssize_t DSocket::ReceiveFrom(void* data,size_t len,ACE_UINT32 &address,ACE_UINT16 &port,int wait_msec)
{
    ACE_Time_Value tv(wait_msec/1000,(wait_msec%1000)*1000);
    ACE_INET_Addr ace_addr;
    ssize_t nrec=mSocket->recv(data,len,ace_addr,0,&tv);
 
    if (nrec>0)
    {
        address=ace_addr.get_ip_address();
        port=ace_addr.get_port_number();
    }
    else
    {
        address=0;
        port=0;
    }
    
    return nrec;
}
 
 
ssize_t DSocket::ReceiveFrom(eOipv4addr_t &ipv4, eOipv4port_t &port, void* data, size_t len, int wait_msec)
{
    ACE_Time_Value tv(wait_msec/1000, (wait_msec%1000)*1000);
    ACE_INET_Addr ace_addr;
    ssize_t nrec = mSocket->recv(data, len, ace_addr, 0, &tv);
 
 
    if(nrec>0)
    {
        ipv4 = htonl(ace_addr.get_ip_address());
        port = ace_addr.get_port_number();
    }
    else
    {
        ipv4 = 0;
        port = 0;
    }
 
    return nrec;
}
 
void DSocket::Close()
{
    if (mSocket)
    {
        mSocket->close();
        delete mSocket;
        mSocket=NULL;
    }
}