/**    
    @file     uds.c
    @date     2009-03-19
    @author   오재경 freefrug@falinux.com
    @brief    uds 를 사용한 통신을 담당한다.

    @modify   2009-12-11 (오재경)
                  uds_open_client() 함수에 대한 편의성을 가미했다.
                  uds_open_client() 함수 호출시 인자의 파일이름은 uds_write() 함수에서
                  목적지가 NULL 일 경우 목적지로 대치된다.
    @todo    
    @bug     
    @remark   
    
    @warning 
*/
//
//  저작권    에프에이리눅스(주)
//            외부공개 금지
//
//----------------------------------------------------------------------------
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/un.h> 
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include <pollmng.h>
#include <uds.h>


char desc_uds[] = "falinux uds ver 0.2.0";

/// uds 개별 구조체
typedef struct {
	
	char  port[256];
	int   im_server;
	
} uds_priv_t;

static poll_obj_t  *tmp_uds_poll = NULL;  // poll 관리객체없이 전송함수를 위해


//------------------------------------------------------------------------------
/** @brief    uds 소켓을 서버형태로 open 한다.
    @param    fname  문자열 파일이름
    @return   poll_obj_t 형태의 포인터
*///----------------------------------------------------------------------------
poll_obj_t  *uds_open_server( char *fname )
{
	struct sockaddr_un addr_svr;
	poll_obj_t *obj;
	uds_priv_t *uds;
	int  option;
	int  sockfd;
	
	sockfd = socket( AF_UNIX, SOCK_DGRAM, 0 );
	if ( sockfd < 0 )
	{
		perror( "uds open error:" );
		return NULL;	
	}

	// 이미 소켓용 화일에 링크되어 있다면 제거한다.
	unlink ( fname );

	// TIME-WAIT 상태에 있는 소켓에 할당되어 있는 IP 주소와 포트를 바로 사용할 수 있도록
	// SO_REUSEADDR 의 옵션 값을 TRUE 로
	option = 1;                                                         
	setsockopt( sockfd, SOL_SOCKET, SO_REUSEADDR, &option, sizeof(option));
	
	// 소켓을 시스템에 연결한다.
	bzero( &addr_svr, sizeof(struct sockaddr_un) );
	addr_svr.sun_family = AF_UNIX;
	strcpy( addr_svr.sun_path, fname );
	
	if( bind( sockfd, (struct sockaddr *)&addr_svr, sizeof(struct sockaddr_un) ) < 0 )
	{
		perror("uds bind error :"); 
	    close( sockfd );
	    return NULL;
	}

	// uds 만의 정보를 설정한다.
	uds = (uds_priv_t *)malloc( sizeof(uds_priv_t) );
	uds->im_server = 1;
	strcpy( uds->port, fname );

	obj = poll_add( sockfd );
	obj->type = STYP_UDS;
	obj->priv = (void *)uds;
	
	tmp_uds_poll = obj;
	
	return obj;
}
//------------------------------------------------------------------------------
/** @brief    uds 소켓을 클라이언트 형태로 open 한다.
    @param    fname  문자열 파일이름, 확장을 위해 존재하며 NULL 을 사용하여도 된다.
    @return   poll_obj_t 형태의 포인터
*///----------------------------------------------------------------------------
poll_obj_t  *uds_open_client( char *fname )
{
	poll_obj_t *obj;
	uds_priv_t *uds;
	int sockfd;
	
	sockfd = socket( AF_UNIX, SOCK_DGRAM, 0 );
	if ( sockfd < 0 )
	{
		perror( "uds open error:" );
		return NULL;	
	}

	// uds 만의 정보를 설정한다.
	uds = (uds_priv_t *)malloc( sizeof(uds_priv_t) );
	uds->im_server = 0;
	if (fname)
	{
		strcpy( uds->port, fname );
	}

	obj = poll_add( sockfd );
	obj->type = STYP_UDS;
	obj->priv = (void *)uds;
	
	tmp_uds_poll = obj;
	
	return obj;
}
//------------------------------------------------------------------------------
/** @brief    uds 소켓을 close 한다.
    @param    obj  폴객체 포인터
*///----------------------------------------------------------------------------
void uds_close( poll_obj_t *obj )
{
	close( obj->fd );
	
	if ( obj->priv )
	{
		free( obj->priv );
	}
	
	poll_delete( obj );
	
	if ( tmp_uds_poll == obj ) tmp_uds_poll = NULL;
}
//------------------------------------------------------------------------------
/** @brief    uds 폴객체를 파일이름으로 찾는다.
    @param    fname 문자열 파일이름
    @return   obj  폴객체 포인터
*///----------------------------------------------------------------------------
poll_obj_t  *uds_get_byport( char *fname )
{
	poll_obj_t *obj;
	uds_priv_t *uds;
	int  idx, count;
	
	count = poll_count();
	
	for(idx=0; idx<count; idx++)
	{
		obj = poll_get_obj( idx );
		uds = (uds_priv_t *)obj->priv;  
		if ( uds )
		{
			if ( 0 == strcmp( uds->port, fname ) )
			{
				return obj;
			}
		}
	}
	
	return NULL;
}
//------------------------------------------------------------------------------
/** @brief    uds 소켓을 통해 데이타를 전송한다.
    @param    obj  폴객체 포인터
    @param    to   상대방 파일이름 문자열포인터
    @param    buf  전송버퍼
    @param    len  버퍼의 길이
    @return   전송한 데이타 개수
*///----------------------------------------------------------------------------
int  uds_write( poll_obj_t *obj, char *to, char *buf, int len )
{
	uds_priv_t *uds;
	struct sockaddr_un uds_addr;
	char *dst;
	int   wrcnt;
	
	uds = (uds_priv_t *)obj->priv; 
	
	// 클라이언트로 열려있거나 목적지가 존재하지 않으면 저장된 포트로 전송한다.
	dst = to;
	if ( ( 0 == uds->im_server ) && ( NULL == dst ) )
	{
		dst = uds->port;
	}

	// 상대편 주소를 설정한다.
	bzero( &uds_addr, sizeof(struct sockaddr_un) );	
	uds_addr.sun_family = AF_UNIX;
	strcpy( uds_addr.sun_path, dst );

	// 전송한다.
	wrcnt = sendto( obj->fd, buf, len, 0, (struct sockaddr *)&uds_addr, sizeof(uds_addr) );

	if ( 0 > wrcnt )
	{
		char msg[256];
		
		sprintf( msg, "uds send error(to %s)", dst );
		perror( msg );	
	}
	
	return wrcnt;
}
//------------------------------------------------------------------------------
/** @brief    uds 소켓을 통해 데이타를 읽는다.
    @param    obj  폴객체 포인터
    @param    buf  전송버퍼
    @param    len  버퍼의 길이
    @return   전송된 데이타 개수
*///----------------------------------------------------------------------------
int  uds_read( poll_obj_t *obj, char *buf, int len )
{
	struct sockaddr_un uds_addr;
	int  addr_len;
	int  rdcnt;
	
	// 데이타를 읽는다.
	addr_len = sizeof(uds_addr);
	rdcnt = recvfrom( obj->fd, buf, len, 0, (struct sockaddr *)&uds_addr, (socklen_t *)&addr_len );
	
	if ( 0 > rdcnt )
	{
		perror( "uds recv error:" );
	}
	//else
	//{
	//	if ( from )	strcpy( from, uds_addr.sun_path );
	//}
	
	return rdcnt;
}

//------------------------------------------------------------------------------
/** @brief    poll 객체없이 uds 소켓을 통해 데이타를 전송한다.
    @param    to   상대방 파일이름 문자열포인터
    @param    buf  전송버퍼
    @param    len  버퍼의 길이
    @return   전송한 데이타 개수
    @remark   uds_open_server(), uds_open_client() 함수는 한번 호출되어야 한다.
*///----------------------------------------------------------------------------
int  uds_write_simple( char *to, char *buf, int len )
{
	if ( tmp_uds_poll )
	{
		return uds_write( tmp_uds_poll, to, buf, len );
	}

	return -1;
}
//------------------------------------------------------------------------------
/** @brief    poll 객체없이 uds 소켓을 통해 데이타를 읽는다.
    @param    buf  전송버퍼
    @param    len  버퍼의 길이
    @return   전송된 데이타 개수
    @remark   uds_open_server(), uds_open_client() 함수는 한번 호출되어야 한다.
*///----------------------------------------------------------------------------
int  uds_read_simple( char *buf, int len )
{
	if ( tmp_uds_poll )
	{
		return uds_read( tmp_uds_poll, buf, len );
	}

	return -1;
}