/*
 * =====================================================================================
 *
 * Filename: LocalSocket.cpp
 *
 * Description:
 *
 * Version: 1.0
 * Created: 2017年08月26日 11时08分28秒
 * Last Modified: 2017年08月26日 11时08分28秒
 * Revision: none
 * Compiler: gcc
 *
 * Author: zt (),
 * Organization:
 *
 * =====================================================================================
 */

#include "LocalSocket.h"


LocalSocket::LocalSocket ( std::string lsSelf, std::string lsRemote )
 : m_lsSelf ( lsSelf ), m_lsRemote ( lsRemote ),
 m_sockfd_self ( -1 ), m_sockfd_remote ( -1 )
{
 Init();
}

LocalSocket::~LocalSocket()
{
 DeInit();
}

int LocalSocket::Init()
{
 m_sockfd_self = socket ( AF_UNIX, SOCK_DGRAM, 0 );

 int loop = 1;
 setsockopt ( m_sockfd_self, SOL_SOCKET, SO_REUSEADDR,
 reinterpret_cast<char*> ( &loop ), sizeof ( loop ) );

 m_self_addr.sun_family = AF_UNIX;
 strcpy ( m_self_addr.sun_path, m_lsSelf.c_str() );
 m_self_addr.sun_path[0] = 0;
 m_selfaddrlen = m_lsSelf.size() + offsetof ( sockaddr_un, sun_path );
 bind ( m_sockfd_self, reinterpret_cast<sockaddr*> ( &m_self_addr ), m_selfaddrlen );

 m_sockfd_remote = socket ( AF_UNIX, SOCK_DGRAM, 0 );

 m_remote_addr.sun_family = AF_UNIX;
 strcpy ( m_remote_addr.sun_path, m_lsRemote.c_str() );
 m_remote_addr.sun_path[0] = 0;
 m_remoteaddrlen = m_lsRemote.size() + offsetof ( sockaddr_un, sun_path );

 return 0;
}

int LocalSocket::DeInit()
{
 if ( m_sockfd_self > 0 )
 close ( m_sockfd_self );

 if ( m_sockfd_remote > 0 )
 close ( m_sockfd_remote );

 return 0;
}

int LocalSocket::Send ( const uint8_t* sendbuf, int size )
{
 if ( m_sockfd_remote < 0 )
 return -1;

 return sendto ( m_sockfd_remote, reinterpret_cast<const char*> ( sendbuf ), size,
 MSG_NOSIGNAL,
 reinterpret_cast<sockaddr*> ( &m_remote_addr ), m_remoteaddrlen );
}

int LocalSocket::Recv ( uint8_t* recvbuf, int size )
{
 if ( m_sockfd_self < 0 )
 return -1;

 sockaddr addr;
 socklen_t len = sizeof ( addr );

 return recvfrom ( m_sockfd_self, reinterpret_cast<char*> ( recvbuf ),
 size, 0, &addr, &len );
}


#ifdef TEST
int main ( int argc, char* argv[] )
{
 LocalSocket ls1 ( "@aaa", "@bbb" );
 LocalSocket ls2 ( "@bbb", "@aaa" );

 uint8_t a[] = "test aaaaaa";
 uint8_t b[] = "test bbbbbb";

 uint8_t buf1[1024];
 uint8_t buf2[1024];

 while ( 1 )
 {
 ls1.Send ( a, 10 );
 memset ( buf1, 0, 1024 );
 ls2.Recv ( buf1, 10 );
 printf ( "ls2 recv ls1 %s\n", buf1 );

 ls2.Send ( b, 10 );
 memset ( buf2, 0, 1024 );
 ls1.Recv ( buf2, 10 );
 printf ( "ls1 recv ls2 %s\n", buf2 );

 sleep ( 1 );
 }

 return 0;
}
#endif