base
This commit is contained in:
biosvos
@@ -0,0 +1,73 @@
|
||||
Revision 1259
|
||||
-------------------
|
||||
수정일 : 2015-10-07
|
||||
수정자 : 유희곤
|
||||
|
||||
- NEW: 저장 대상 물리 파일이 이미 존재하는지 확인을 위한 interface 기능 추가 (#24683)
|
||||
- 저장할 물리파일이 이미 존재하는지 확인을 위한 REQUEST_FILE_CHECK 요청 처리 기능 추가
|
||||
- 해당 파일 존재시.. 파일 크기를 전달
|
||||
- 해당 파일 미존재시.. 파일 크기 0 으로 전달.
|
||||
- 기타 오류 발생시 오류 내역을 문자열로 전달.
|
||||
- client 제공을 위한 interface/CcCollectdClientSocket.h 상에 CheckFile() 함수 추가
|
||||
|
||||
|
||||
Revision 1238
|
||||
-------------------
|
||||
수정일 : 2015-09-16
|
||||
수정자 : 유희곤
|
||||
|
||||
- CHG: 저장 경로 생성 방식 변경 (#24436)
|
||||
- 기존 : hostname - process name - 파일
|
||||
- 변경 : process name - hostname - 파일 형식.
|
||||
- 수정 파일: ClientThread.cpp
|
||||
|
||||
- CHG: 최신 BaseSocket 모듈 적용 (#24436, #24141)
|
||||
- EPIPE 및 usleep 관련 수정 버전 적용
|
||||
- 수정 파일: lib/BaseSocket.cpp, lib/BaseSocket.h
|
||||
|
||||
- CHG: usleep -> nanosleep 변경 (#24436)
|
||||
- multi thread 에서 lock 관련 문제점 발생 가능.
|
||||
- 프로세스 기동 및 시그널 처리에서만 사용하여... thread 와 연관 없지만.. 수정 처리함.
|
||||
- 수정 파일: Main.cpp, Worker.cpp
|
||||
|
||||
- CHG: Main 버전 (3.5.0) 변경 사항 적용
|
||||
- 기존 3.4.0 -> 3.5.0 으로 버전 변경 처리.
|
||||
|
||||
- CHG: 기타
|
||||
- source 상의 conf 파일 위치 수정
|
||||
- 화면 표시 도움말 내역 수정.
|
||||
|
||||
|
||||
Revision 1116
|
||||
-------------------
|
||||
수정일 : 2014-11-04
|
||||
수정자 : 유희곤
|
||||
|
||||
- CHG: Directory 생성 관련 코드 수정 (#20676)
|
||||
- Directory 생성 관련 기존 system() 함수 -> mkdir() 함수 사용토록 수정
|
||||
- Directory 생성 실패시 오류 메시시 전송 및 로깅 처리 관련 기능 추가
|
||||
- 수정 파일: ClientThread.cpp
|
||||
|
||||
|
||||
Revision 1089
|
||||
-------------------
|
||||
수정일 : 2014-10-07
|
||||
수정자 : 유희곤
|
||||
|
||||
- CHG: Client 제공 Interface 함수의 설명, 기능 보강
|
||||
- 사용 관련 설명 추가
|
||||
- 입력 변수에 대한 오류 체크 기능 보강.
|
||||
|
||||
* interface 관련 함수 수정으로 인해 tagging 작업은 수행 안함.
|
||||
|
||||
|
||||
Revision 1086
|
||||
-------------------
|
||||
수정일 : 2014-10-06
|
||||
수정자 : 유희곤
|
||||
|
||||
- NEW: SVN 신규 등록 (#20222)
|
||||
- 소스 등록 위치: /trunk/cc_logbackup/cc_collectd
|
||||
- conf 등록 위치: /tags/release/conf/
|
||||
- 태깅 등록 위치: /tags/release/cc_logbackup/cc_collectd
|
||||
|
||||
@@ -0,0 +1,37 @@
|
||||
#****************************************************************************
|
||||
# Makefile for cc_collectd
|
||||
# -----------------------------------------
|
||||
#
|
||||
# begin : 2014/09/03
|
||||
# copyright : (C) 2005 Solbox Inc.
|
||||
# author : storage dev team
|
||||
# email : storage.sd@solbox.com
|
||||
# version : 3.4
|
||||
#
|
||||
# CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
# Redistribution and use in source and binary forms, with or with out
|
||||
# modification, are not permitted in outside of Solbox Inc.
|
||||
#*****************************************************************************
|
||||
|
||||
SUBDIRS = lib src
|
||||
|
||||
.PHONY: all $(SUBDIRS)
|
||||
|
||||
|
||||
all: $(SUBDIRS)
|
||||
sync;
|
||||
|
||||
|
||||
$(SUBDIRS):
|
||||
$(MAKE) all -C $@
|
||||
|
||||
|
||||
install:
|
||||
@for dir in $(SUBDIRS); do (cd $$dir; $(MAKE) install); done
|
||||
|
||||
|
||||
clean:
|
||||
@for dir in $(SUBDIRS); do (cd $$dir; $(MAKE) clean); done
|
||||
|
||||
|
||||
# End of Makefile
|
||||
@@ -0,0 +1,19 @@
|
||||
[COMMON]
|
||||
# Log Directory path
|
||||
DEFAULT_LOG_DIR = /user/service/logs
|
||||
|
||||
# Log write level: 7:debug 6:info(Default) 5:notice 4:warning 3:error 2:critical 1:alert 0:emergency
|
||||
LOG_LEVEL = 6
|
||||
|
||||
#-----------------------------------------------#
|
||||
[cc_collectd]
|
||||
|
||||
# TCP Listen port
|
||||
TCP_LISTEN_PORT = 13103
|
||||
|
||||
# Save path
|
||||
SAVE_ROOT_PATH = /backup/log
|
||||
|
||||
# Worker process count: default value is 3
|
||||
WORKER_PROCESS_COUNT = 3
|
||||
|
||||
@@ -0,0 +1,792 @@
|
||||
#include "CcCollectdClientSocket.h"
|
||||
|
||||
#include <arpa/inet.h>
|
||||
#include <netinet/in.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
#include <stdio.h>
|
||||
#include <errno.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#ifdef __FreeBSD__
|
||||
#include <sys/endian.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/uio.h>
|
||||
#else
|
||||
#include <endian.h>
|
||||
#include <sys/sendfile.h>
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
#define HOSTNAME_SIZE 256
|
||||
|
||||
|
||||
// 생성자.
|
||||
CCcCollectdClientSocket::CCcCollectdClientSocket()
|
||||
: CBaseSocket( SOCKET_NOT_VALID )
|
||||
, m_nPacketHeaderLen ( sizeof(m_packetHeader))
|
||||
, m_nPacketDataLen( 0 )
|
||||
{
|
||||
}
|
||||
|
||||
// 소멸자
|
||||
CCcCollectdClientSocket::~CCcCollectdClientSocket()
|
||||
{
|
||||
// 소멸자 Socket 명시적 Close 처리.
|
||||
Close();
|
||||
}
|
||||
|
||||
|
||||
// Packet Header 정보를 Log 파일에 Logging 처리
|
||||
void CCcCollectdClientSocket::PrintHeaderToLog()
|
||||
{
|
||||
_LOG( LINF, "------------------------------------" );
|
||||
_LOG( LINF, "stx [%02x]", m_packetHeader.stx );
|
||||
_LOG( LINF, "type [%02x]", m_packetHeader.type );
|
||||
_LOG( LINF, "command [%02x][%02x][%02x][%02x]", m_packetHeader.command[0], m_packetHeader.command[1], m_packetHeader.command[2], m_packetHeader.command[3] );
|
||||
_LOG( LINF, "result [%02x][%02x][%02x][%02x]", m_packetHeader.result[0], m_packetHeader.result[1], m_packetHeader.result[2], m_packetHeader.result[3] );
|
||||
_LOG( LINF, "data_length [%u]", m_nPacketDataLen );
|
||||
_LOG( LINF, "extend_code [%02x][%02x]", m_packetHeader.extend_code[0], m_packetHeader.extend_code[1] );
|
||||
_LOG( LINF, "------------------------------------" );
|
||||
}
|
||||
|
||||
|
||||
// socket 에서 지정된 크기만큼의 데이터를 읽어 내부 임시버퍼인 m_tempBuffer 에 저장처리.
|
||||
// @param size [in] read 할 데이터 크기
|
||||
// @return On success return true, otherwise return false.
|
||||
bool CCcCollectdClientSocket::GetPacketData( unsigned int size )
|
||||
{
|
||||
// 1. 수신할 크기가 임시버퍼보다 큰 경우 오류 반환.
|
||||
if( size > DEFAULT_SOCKET_TEMP_BUFFER_SIZE )
|
||||
return false;
|
||||
|
||||
// 2. 데이터 수신 처리.
|
||||
int nRead = ReadNTimeout( m_tempBuffer, size );
|
||||
|
||||
// 오류 발생시
|
||||
if( nRead <= 0 )
|
||||
return false;
|
||||
|
||||
// 3. 정상인 경우는 ReadNTimeout 함수가 무조건 요청 크기만큼 read 처리하므로 정상수신임.
|
||||
return true;
|
||||
}
|
||||
|
||||
// socket 에서 지정된 크기만큼의 데이터를 읽어 출력변수에 저장처리.
|
||||
// @param size [in] read 할 데이터 크기
|
||||
// @param value [out] 읽은 데이터를 저장할 string 변수
|
||||
// @return On success return true, otherwise return false.
|
||||
bool CCcCollectdClientSocket::GetPacketData( unsigned int& size, std::string& value )
|
||||
{
|
||||
// 1. 수신할 크기가 임시버퍼보다 큰 경우 새로운 버퍼를 생성한다.
|
||||
BYTE * pBuffer = NULL;
|
||||
bool bNewBufferCreated = false;
|
||||
if( size > DEFAULT_SOCKET_TEMP_BUFFER_SIZE )
|
||||
{
|
||||
pBuffer = new BYTE[size]; // 신규 메모리 할당.
|
||||
bNewBufferCreated = true;
|
||||
}
|
||||
else
|
||||
pBuffer = m_tempBuffer;
|
||||
|
||||
// 2. 데이터 수신 처리.
|
||||
int nRead = ReadNTimeout( pBuffer, size );
|
||||
|
||||
// 오류 발생시
|
||||
if( nRead <= 0 )
|
||||
{
|
||||
// 신규로 생성된 버퍼인 경우 메모리 해제 처리.
|
||||
if( bNewBufferCreated == true )
|
||||
delete[] pBuffer;
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
// 3. 수신 데이터 저장 처리 - 기존 데이터 뒤에 붙여 준다=> 이게 사용하기 편함.
|
||||
value.append( (char *)pBuffer, size );
|
||||
|
||||
// 신규로 생성된 버퍼인 경우 메모리 해제 처리.
|
||||
if( bNewBufferCreated == true )
|
||||
delete[] pBuffer;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
// pValue 에 저장된 데이터를 unsigned long long (64Byte) 형으로 변환처리 및 Endian 변환
|
||||
unsigned long long CCcCollectdClientSocket::GetDataToUInt64( BYTE * pValue, bool bConvertEndian )
|
||||
{
|
||||
BYTE tempBuffer[8];
|
||||
unsigned long long result;
|
||||
|
||||
memcpy( tempBuffer, pValue, 8 );
|
||||
unsigned long long * pResult = ( unsigned long long * )tempBuffer;
|
||||
|
||||
// Endian 변환처리.
|
||||
if( bConvertEndian == true )
|
||||
{
|
||||
// Network byte order (big endian) 을 Host byte order 로 변환 처리
|
||||
//result = be64toh( *pResult );
|
||||
result = do_ntohll( *pResult );
|
||||
}
|
||||
else
|
||||
result = *pResult;
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
// Packet 정보 중 String 정보를 전송하기 위한 함수.
|
||||
bool CCcCollectdClientSocket::SendString( const std::string& strData )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return false;
|
||||
|
||||
// 크기값을 저장하기 위한 임시 변수.
|
||||
unsigned int nSize = 0;
|
||||
|
||||
// String Data 전송
|
||||
nSize = htonl( strData.size() );
|
||||
WriteN( &nSize, 4 );
|
||||
if( strData.size() > 0 )
|
||||
WriteN( strData.c_str(), strData.size() );
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// byte ordering function for 64bit variable
|
||||
uint64_t CCcCollectdClientSocket::do_htonll( uint64_t hostlonglong )
|
||||
{
|
||||
int x = 1;
|
||||
|
||||
/* little endian */
|
||||
if( *(char *)&x == 1 )
|
||||
return ( ( ( (uint64_t)htonl( hostlonglong ) ) << 32 ) + htonl( hostlonglong >> 32 ) );
|
||||
|
||||
/* big endian */
|
||||
else
|
||||
return hostlonglong;
|
||||
}
|
||||
|
||||
uint64_t CCcCollectdClientSocket::do_ntohll( uint64_t netlonglong )
|
||||
{
|
||||
int x = 1;
|
||||
|
||||
/* little endian */
|
||||
if( *(char *)&x == 1 )
|
||||
return ( ( ( (uint64_t)ntohl( netlonglong ) ) << 32 ) + ntohl( netlonglong >> 32 ) );
|
||||
|
||||
/* big endian */
|
||||
else
|
||||
return netlonglong;
|
||||
|
||||
}
|
||||
|
||||
/////////////////////////////////////////////////////////////////
|
||||
|
||||
// 전달받은 Target 으로 Socket 접속을 수행
|
||||
bool CCcCollectdClientSocket::ConnectTarget( const std::string& szTarget, int nPort )
|
||||
{
|
||||
return Connect( szTarget, nPort );
|
||||
}
|
||||
|
||||
// cc_collect 로부터 available size 정보를 요청, 결과를 수신처리한다.
|
||||
bool CCcCollectdClientSocket::GetAvailableSize( unsigned long long& uSize )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return false;
|
||||
|
||||
// 정보 요청을 위한 Packet Header 생성.
|
||||
struct CcCollectdPacketHeader stHeader;
|
||||
memset( &stHeader, 0x00, sizeof( struct CcCollectdPacketHeader ) );
|
||||
|
||||
stHeader.stx = HEADER_STX_CODE;
|
||||
stHeader.type = HEADER_TYPE_REQUEST;
|
||||
stHeader.command[0] = REQUEST_AVAILABLE_SIZE;
|
||||
|
||||
// Data 부분의 길이를 계산한다.
|
||||
// - Data 부분 길이는 없음.
|
||||
stHeader.data_length = htonl( 0 );
|
||||
|
||||
// Packet Header 정보 전송
|
||||
if( WriteN( &stHeader, m_nPacketHeaderLen ) != m_nPacketHeaderLen )
|
||||
{
|
||||
LOG( LERR, "request [REQUEST_AVAILABLE_SIZE] send fail to cc_collectd. " );
|
||||
return false;
|
||||
}
|
||||
|
||||
// 응답 대기...
|
||||
// 최대 3분 대기
|
||||
int nTimeout = 60 * 3; // 최대 3분간 대기
|
||||
int nRead;
|
||||
|
||||
// Socket 으로 부터 Packet Header 부분 수신.
|
||||
// 수신된 정보는 멤버변수에 저장처리.
|
||||
nRead = ReadNTimeout( &m_packetHeader, m_nPacketHeaderLen, nTimeout );
|
||||
|
||||
// nRead 0: Socket Closed
|
||||
// -1: error
|
||||
// -2: Timeout 이므로
|
||||
if( nRead == -2 )
|
||||
{
|
||||
// Timeout 발생시
|
||||
LOG( LERR, "cc_collectd REQUEST_AVAILABLE_SIZE reponse wait. but timeout[%d sec] occured.", nTimeout );
|
||||
return false;
|
||||
}
|
||||
else if( nRead == 0 || nRead == -1 )
|
||||
{
|
||||
// Socket Close 또는 오류 발생시
|
||||
LOG( LERR, "cc_collectd REQUEST_AVAILABLE_SIZE reponse wait. but socket closed or error [%d]", nRead );
|
||||
return false;
|
||||
}
|
||||
|
||||
// STX code 검사.
|
||||
if( m_packetHeader.stx != HEADER_STX_CODE )
|
||||
{
|
||||
LOG( LERR, "Not valid stx code." );
|
||||
PrintHeaderToLog();
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
// Data Length 부분 값을 멤버 변수에 저장처리.
|
||||
m_nPacketDataLen = ntohl( m_packetHeader.data_length );
|
||||
|
||||
// 해당 Packet 에 대한 응답패킷인지 검사.
|
||||
if( m_packetHeader.type != HEADER_TYPE_RESPONSE
|
||||
|| m_packetHeader.command[0] != REQUEST_AVAILABLE_SIZE )
|
||||
{
|
||||
// 다른 패킷이 들어온 경우
|
||||
|
||||
// ** 정석대로라면 그냥 냅둬어 수신부에서 처리해야 하지만
|
||||
// 현재 수신 처리부가 없어 당장 문제가 생길수 있으므로
|
||||
// 데이터 부분까지 수신하여 임시버퍼에 저장처리 해 놓는다.
|
||||
GetPacketData( m_nPacketDataLen );
|
||||
return false;
|
||||
}
|
||||
|
||||
// Data 부분이 존재하는 경우.. 수신 처리.
|
||||
if( m_nPacketDataLen > 0 )
|
||||
{
|
||||
// Data 항목은
|
||||
// - 8 Byte Network byte order 로 구성된 available size 정보가 들어 있음.
|
||||
if( GetPacketData( m_nPacketDataLen ) == false )
|
||||
{
|
||||
LOG( LERR, "available size body receive failed." );
|
||||
return false;
|
||||
}
|
||||
|
||||
BYTE * pPos = m_tempBuffer;
|
||||
uSize = GetDataToUInt64( pPos, true );
|
||||
}
|
||||
else
|
||||
{
|
||||
// cc_collectd 로 부터 Available size 응답을 받았지만.. body 가 없는 경우.
|
||||
// 이런 경우는 없지만.. 오류 처리를 위해 코딩.
|
||||
LOG( LERR, "available size body length zero. check." );
|
||||
return false;
|
||||
}
|
||||
|
||||
_LOG( LDBG, "cc_collectd available size result received. [%llu]", uSize );
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
// cc_collect 로 File 저장 목적으로 전송
|
||||
bool CCcCollectdClientSocket::SendFile( const std::string strProcessName, const std::string strFileName
|
||||
, const int& fd, const unsigned long long& uFileSize
|
||||
, std::string& strMessage )
|
||||
{
|
||||
// 입력 받은 정보가 유효한지 검사.
|
||||
// strProcessName : 생략 가능하지만.. 문자열에 '/' 가 포함되서는 안된다.
|
||||
// strFileName : 생략 불가, 문자열에 '/' 가 포함되서는 안된다.
|
||||
if( strProcessName.empty() == false )
|
||||
{
|
||||
std::string::size_type pos = strProcessName.find( '/' );
|
||||
if( pos != std::string::npos )
|
||||
{
|
||||
// '/' 문자가 포함된 경우...
|
||||
char tempBuffer[1024];
|
||||
snprintf( tempBuffer, sizeof( tempBuffer ) - 1
|
||||
, "process name[%s] not valid that include slash."
|
||||
, strProcessName.c_str() );
|
||||
|
||||
strMessage.append( tempBuffer );
|
||||
LOG( LERR, "%s", strMessage.c_str() );
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if( strFileName.empty() == false )
|
||||
{
|
||||
std::string::size_type pos = strFileName.find( '/' );
|
||||
if( pos != std::string::npos )
|
||||
{
|
||||
// '/' 문자가 포함된 경우...
|
||||
char tempBuffer[1024];
|
||||
snprintf( tempBuffer, sizeof( tempBuffer ) - 1
|
||||
, "file name[%s] not valid that include slash."
|
||||
, strFileName.c_str() );
|
||||
|
||||
strMessage.append( tempBuffer );
|
||||
LOG( LERR, "%s", strMessage.c_str() );
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// filename 이 없는 경우.
|
||||
strMessage = "filename not vaild. empty.";
|
||||
LOG( LERR, "filename not vaild. empty." );
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
unsigned long long uAvailableSize = 0;
|
||||
|
||||
// 1. cc_collectd 로 부터 available size 정보를 수신
|
||||
if( GetAvailableSize( uAvailableSize ) == false )
|
||||
{
|
||||
// available size 체크 실패시...
|
||||
strMessage = "cc_collectd available size check fail.";
|
||||
return false;
|
||||
}
|
||||
|
||||
// 2. 수신된 uAvailableSize 과 전송할 File size 비교
|
||||
if( uAvailableSize <= uFileSize )
|
||||
{
|
||||
char tempBuffer[1024];
|
||||
snprintf( tempBuffer, sizeof( tempBuffer ) - 1
|
||||
, "file size[%llu] too large than cc_collect available size [%llu]"
|
||||
, uFileSize, uAvailableSize );
|
||||
|
||||
strMessage.append( tempBuffer );
|
||||
LOG( LERR, "%s", strMessage.c_str() );
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
// 3. 장비의 hostname 정보를 추출한다.
|
||||
char szHostName[HOSTNAME_SIZE];
|
||||
memset( szHostName, 0x00, HOSTNAME_SIZE );
|
||||
if( gethostname( szHostName, HOSTNAME_SIZE - 1 ) != 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
|
||||
strMessage = "hostname get failed. check hostname.";
|
||||
LOG( LERR, "hostname get failed. [%d][%s]", errorNum, strerror( errorNum ) );
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
if( strlen( szHostName ) == 0 )
|
||||
{
|
||||
strMessage = "hostname not vaild. check hostname.";
|
||||
LOG( LERR, "hostname not valid. length 0" );
|
||||
return false;
|
||||
}
|
||||
std::string strHostName = szHostName; // 추출된 hostname 정보를 전송하기 편하도록 string 객체에 저장처리.
|
||||
|
||||
|
||||
// 4. 파일 전송 관련 요청 송신
|
||||
struct CcCollectdPacketHeader stHeader;
|
||||
memset( &stHeader, 0x00, sizeof( struct CcCollectdPacketHeader ) );
|
||||
|
||||
stHeader.stx = HEADER_STX_CODE;
|
||||
stHeader.type = HEADER_TYPE_REQUEST;
|
||||
stHeader.command[0] = REQUEST_FILE_SAVE;
|
||||
|
||||
// Data 부분
|
||||
// strHostName : string ( 4 + string.size() )
|
||||
// strProcessName : string ( 4 + string.size() )
|
||||
// strFileName : string ( 4 + string.size() )
|
||||
// uFileSize : 8 Byte ---------->본 항목까지만 header data_length 에 포함시킨다.
|
||||
// 저장할 파일 내용.
|
||||
|
||||
// Data 부분의 길이를 계산한다.
|
||||
unsigned int nTemp = 0;
|
||||
nTemp += ( 4 + strHostName.size() ); // strHostName
|
||||
nTemp += ( 4 + strProcessName.size() ); // strProcessName
|
||||
nTemp += ( 4 + strFileName.size() ); // strFileName
|
||||
nTemp += 8; // uFileSize
|
||||
|
||||
stHeader.data_length = htonl( nTemp );
|
||||
|
||||
// Packet Header 정보 전송
|
||||
if( WriteN( &stHeader, m_nPacketHeaderLen ) != m_nPacketHeaderLen )
|
||||
{
|
||||
strMessage = "request [REQUEST_FILE_SAVE] send fail to cc_collectd.";
|
||||
LOG( LERR, "%s", strMessage.c_str() );
|
||||
return false;
|
||||
}
|
||||
|
||||
// Body 부분 전송
|
||||
// strHostName
|
||||
SendString( strHostName );
|
||||
// strProcessName
|
||||
SendString( strProcessName );
|
||||
// strFileName
|
||||
SendString( strFileName );
|
||||
// uFileSize (Network byte order )
|
||||
//uAvailableSize = htobe64( uFileSize );
|
||||
uAvailableSize = do_htonll( uFileSize );
|
||||
|
||||
WriteN( &uAvailableSize, 8 );
|
||||
|
||||
// 파일 내용 전송
|
||||
if( uFileSize > 0 )
|
||||
{
|
||||
off_t offset = 0;
|
||||
|
||||
// sendfile() 에 대한 함수 정의가 Linux, FreeBSD 에서 다르므로.. OS 별로 처리한다.
|
||||
#ifdef __FreeBSD__
|
||||
|
||||
// FreeBSD 의 경우.. 10G 까지 전송 테스트 됨.
|
||||
|
||||
off_t sbytes;
|
||||
int ret = sendfile( fd, m_sock, offset, uFileSize, NULL, &sbytes, 0 );
|
||||
|
||||
if( ret == -1 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "sendfile function error. [%d][%s]", errorNum, strerror( errorNum ) );
|
||||
// 전송 실패 발생시...
|
||||
strMessage = "file send fail to cc_collectd";
|
||||
return false;
|
||||
}
|
||||
else if( ( unsigned long long )sbytes != uFileSize )
|
||||
{
|
||||
LOG( LERR, "sendfile error. file[%s] size[%llu] send size[%ld]", strFileName.c_str(), uFileSize, sbytes );
|
||||
strMessage = "file send fail to cc_collectd. file size not match.";
|
||||
return false;
|
||||
}
|
||||
|
||||
#else
|
||||
// Linux 의 경우 sendfile 함수가 2G 까지만 전송 가능하므로....루프로 전송 처리가 되도록 처리한다.
|
||||
|
||||
ssize_t bytes_sent;
|
||||
size_t total_bytes_sent = 0;
|
||||
|
||||
while( total_bytes_sent < uFileSize )
|
||||
{
|
||||
if( (bytes_sent = sendfile(m_sock, fd, &offset, uFileSize - total_bytes_sent)) <= 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
if (errorNum == EINTR || errorNum == EAGAIN)
|
||||
{
|
||||
// Interrupted system call/try again
|
||||
// Just skip to the top of the loop and try again
|
||||
continue;
|
||||
}
|
||||
else
|
||||
{
|
||||
LOG( LERR, "sendfile function error. [%d][%s]", errorNum, strerror( errorNum ) );
|
||||
|
||||
// 전송 실패 발생시...
|
||||
strMessage = "file send fail to cc_collectd";
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
total_bytes_sent += bytes_sent;
|
||||
}
|
||||
|
||||
if( ( unsigned long long )total_bytes_sent != uFileSize )
|
||||
{
|
||||
LOG( LERR, "sendfile error. file[%s] size[%llu] send size[%ld]", strFileName.c_str(), uFileSize, total_bytes_sent );
|
||||
strMessage = "file send fail to cc_collectd. file size not match.";
|
||||
return false;
|
||||
}
|
||||
#endif
|
||||
|
||||
}
|
||||
|
||||
// 5. File 전송 처리 결과 수신.
|
||||
// - 최대 10 분 동안 응답 대기
|
||||
int nTimeout = 60 * 10; // 최대 10분간 대기
|
||||
int nRead;
|
||||
|
||||
// Socket 으로 부터 Packet Header 부분 수신.
|
||||
// 수신된 정보는 멤버변수에 저장처리.
|
||||
nRead = ReadNTimeout( &m_packetHeader, m_nPacketHeaderLen, nTimeout );
|
||||
|
||||
// nRead 0: Socket Closed
|
||||
// -1: error
|
||||
// -2: Timeout 이므로
|
||||
if( nRead == -2 )
|
||||
{
|
||||
// Timeout 발생시
|
||||
LOG( LERR, "cc_collectd REQUEST_FILE_SAVE reponse wait. but timeout[%d sec] occured.", nTimeout );
|
||||
strMessage = "cc_collectd REQUEST_FILE_SAVE reponse timeout[10 min] occured.";
|
||||
return false;
|
||||
}
|
||||
else if( nRead == 0 || nRead == -1 )
|
||||
{
|
||||
// Socket Close 또는 오류 발생시
|
||||
LOG( LERR, "cc_collectd REQUEST_FILE_SAVE reponse wait. but socket closed or error [%d]", nRead );
|
||||
strMessage = "cc_collectd REQUEST_FILE_SAVE reponse wait. but session closed.";
|
||||
return false;
|
||||
}
|
||||
|
||||
// STX code 검사.
|
||||
if( m_packetHeader.stx != HEADER_STX_CODE )
|
||||
{
|
||||
LOG( LERR, "Not valid stx code." );
|
||||
strMessage = "cc_collectd REQUEST_FILE_SAVE reponse not valid.";
|
||||
PrintHeaderToLog();
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
// Data Length 부분 값을 멤버 변수에 저장처리.
|
||||
m_nPacketDataLen = ntohl( m_packetHeader.data_length );
|
||||
|
||||
// 해당 Packet 에 대한 응답패킷인지 검사.
|
||||
if( m_packetHeader.type != HEADER_TYPE_RESPONSE
|
||||
|| m_packetHeader.command[0] != REQUEST_FILE_SAVE )
|
||||
{
|
||||
// 다른 패킷이 들어온 경우
|
||||
|
||||
// ** 정석대로라면 그냥 냅둬어 수신부에서 처리해야 하지만
|
||||
// 현재 수신 처리부가 없어 당장 문제가 생길수 있으므로
|
||||
// 데이터 부분까지 수신하여 임시버퍼에 저장처리 해 놓는다.
|
||||
GetPacketData( m_nPacketDataLen );
|
||||
|
||||
strMessage = "cc_collectd REQUEST_FILE_SAVE reponse not valid.";
|
||||
return false;
|
||||
}
|
||||
|
||||
// Data 부분이 존재하는 경우.. 수신 처리.
|
||||
if( m_nPacketDataLen > 0 )
|
||||
{
|
||||
if( GetPacketData( m_nPacketDataLen, strMessage ) == false )
|
||||
{
|
||||
LOG( LERR, "cc_collectd REQUEST_FILE_SAVE body receive failed." );
|
||||
strMessage = "cc_collectd REQUEST_FILE_SAVE body receive failed.";
|
||||
return false;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// 성공 실패 여부 확인
|
||||
if( m_packetHeader.result[0] != HEADER_RESULT_SUCCESS )
|
||||
{
|
||||
_LOG( LDBG, "cc_collectd REQUEST_FILE_SAVE result received. error[%s]", strMessage.c_str() );
|
||||
return false;
|
||||
}
|
||||
else
|
||||
{
|
||||
_LOG( LDBG, "cc_collectd REQUEST_FILE_SAVE result received. success." );
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
// cc_collect 로 File 존재 여부 확인
|
||||
bool CCcCollectdClientSocket::CheckFile( const std::string strProcessName, const std::string strFileName
|
||||
, unsigned long long& uFileSize, std::string& strErrorMessage )
|
||||
{
|
||||
// 입력 받은 정보가 유효한지 검사.
|
||||
// strProcessName : 생략 가능하지만.. 문자열에 '/' 가 포함되서는 안된다.
|
||||
// strFileName : 생략 불가, 문자열에 '/' 가 포함되서는 안된다.
|
||||
if( strProcessName.empty() == false )
|
||||
{
|
||||
std::string::size_type pos = strProcessName.find( '/' );
|
||||
if( pos != std::string::npos )
|
||||
{
|
||||
// '/' 문자가 포함된 경우...
|
||||
char tempBuffer[1024];
|
||||
snprintf( tempBuffer, sizeof( tempBuffer ) - 1
|
||||
, "process name[%s] not valid that include slash."
|
||||
, strProcessName.c_str() );
|
||||
|
||||
strErrorMessage.append( tempBuffer );
|
||||
LOG( LERR, "%s", strErrorMessage.c_str() );
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if( strFileName.empty() == false )
|
||||
{
|
||||
std::string::size_type pos = strFileName.find( '/' );
|
||||
if( pos != std::string::npos )
|
||||
{
|
||||
// '/' 문자가 포함된 경우...
|
||||
char tempBuffer[1024];
|
||||
snprintf( tempBuffer, sizeof( tempBuffer ) - 1
|
||||
, "file name[%s] not valid that include slash."
|
||||
, strFileName.c_str() );
|
||||
|
||||
strErrorMessage.append( tempBuffer );
|
||||
LOG( LERR, "%s", strErrorMessage.c_str() );
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// filename 이 없는 경우.
|
||||
strErrorMessage = "filename not vaild. empty.";
|
||||
LOG( LERR, "filename not vaild. empty." );
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
// 장비의 hostname 정보를 추출한다.
|
||||
char szHostName[HOSTNAME_SIZE];
|
||||
memset( szHostName, 0x00, HOSTNAME_SIZE );
|
||||
if( gethostname( szHostName, HOSTNAME_SIZE - 1 ) != 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
|
||||
strErrorMessage = "hostname get failed. check hostname.";
|
||||
LOG( LERR, "hostname get failed. [%d][%s]", errorNum, strerror( errorNum ) );
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
if( strlen( szHostName ) == 0 )
|
||||
{
|
||||
strErrorMessage = "hostname not vaild. check hostname.";
|
||||
LOG( LERR, "hostname not valid. length 0" );
|
||||
return false;
|
||||
}
|
||||
std::string strHostName = szHostName; // 추출된 hostname 정보를 전송하기 편하도록 string 객체에 저장처리.
|
||||
|
||||
// 파일 Check 관련 요청 송신
|
||||
struct CcCollectdPacketHeader stHeader;
|
||||
memset( &stHeader, 0x00, sizeof( struct CcCollectdPacketHeader ) );
|
||||
|
||||
stHeader.stx = HEADER_STX_CODE;
|
||||
stHeader.type = HEADER_TYPE_REQUEST;
|
||||
stHeader.command[0] = REQUEST_FILE_CHECK;
|
||||
|
||||
// Data 부분
|
||||
// strHostName : string ( 4 + string.size() )
|
||||
// strProcessName : string ( 4 + string.size() )
|
||||
// strFileName : string ( 4 + string.size() )
|
||||
|
||||
|
||||
// Data 부분의 길이를 계산한다.
|
||||
unsigned int nTemp = 0;
|
||||
nTemp += ( 4 + strHostName.size() ); // strHostName
|
||||
nTemp += ( 4 + strProcessName.size() ); // strProcessName
|
||||
nTemp += ( 4 + strFileName.size() ); // strFileName
|
||||
|
||||
stHeader.data_length = htonl( nTemp );
|
||||
|
||||
// Packet Header 정보 전송
|
||||
if( WriteN( &stHeader, m_nPacketHeaderLen ) != m_nPacketHeaderLen )
|
||||
{
|
||||
strErrorMessage = "request [REQUEST_FILE_CHECK] send fail to cc_collectd.";
|
||||
LOG( LERR, "%s", strErrorMessage.c_str() );
|
||||
return false;
|
||||
}
|
||||
|
||||
// Body 부분 전송
|
||||
// strHostName
|
||||
SendString( strHostName );
|
||||
// strProcessName
|
||||
SendString( strProcessName );
|
||||
// strFileName
|
||||
SendString( strFileName );
|
||||
|
||||
|
||||
// File Check 처리 결과 수신.
|
||||
// - 최대 30 sec 동안 응답 대기
|
||||
int nTimeout = 30;
|
||||
int nRead;
|
||||
|
||||
// Socket 으로 부터 Packet Header 부분 수신.
|
||||
// 수신된 정보는 멤버변수에 저장처리.
|
||||
nRead = ReadNTimeout( &m_packetHeader, m_nPacketHeaderLen, nTimeout );
|
||||
|
||||
// nRead 0: Socket Closed
|
||||
// -1: error
|
||||
// -2: Timeout 이므로
|
||||
if( nRead == -2 )
|
||||
{
|
||||
// Timeout 발생시
|
||||
LOG( LERR, "cc_collectd REQUEST_FILE_CHECK reponse wait. but timeout[%d sec] occured.", nTimeout );
|
||||
strErrorMessage = "cc_collectd REQUEST_FILE_CHECK reponse timeout[30 sec] occured.";
|
||||
return false;
|
||||
}
|
||||
else if( nRead == 0 || nRead == -1 )
|
||||
{
|
||||
// Socket Close 또는 오류 발생시
|
||||
LOG( LERR, "cc_collectd REQUEST_FILE_CHECK reponse wait. but socket closed or error [%d]", nRead );
|
||||
strErrorMessage = "cc_collectd REQUEST_FILE_CHECK reponse wait. but session closed.";
|
||||
return false;
|
||||
}
|
||||
|
||||
// STX code 검사.
|
||||
if( m_packetHeader.stx != HEADER_STX_CODE )
|
||||
{
|
||||
LOG( LERR, "Not valid stx code." );
|
||||
strErrorMessage = "cc_collectd REQUEST_FILE_CHECK reponse not valid.";
|
||||
PrintHeaderToLog();
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
// Data Length 부분 값을 멤버 변수에 저장처리.
|
||||
m_nPacketDataLen = ntohl( m_packetHeader.data_length );
|
||||
|
||||
// 해당 Packet 에 대한 응답패킷인지 검사.
|
||||
if( m_packetHeader.type != HEADER_TYPE_RESPONSE
|
||||
|| m_packetHeader.command[0] != REQUEST_FILE_CHECK )
|
||||
{
|
||||
// 다른 패킷이 들어온 경우
|
||||
|
||||
// ** 정석대로라면 그냥 냅둬어 수신부에서 처리해야 하지만
|
||||
// 현재 수신 처리부가 없어 당장 문제가 생길수 있으므로
|
||||
// 데이터 부분까지 수신하여 임시버퍼에 저장처리 해 놓는다.
|
||||
GetPacketData( m_nPacketDataLen );
|
||||
|
||||
strErrorMessage = "cc_collectd REQUEST_FILE_CHECK reponse not valid.";
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
// 성공 실패 여부 확인
|
||||
if( m_packetHeader.result[0] == HEADER_RESULT_SUCCESS )
|
||||
{
|
||||
// 성공인 경우...
|
||||
// - File Size 정보를 수신 처리한다.
|
||||
|
||||
// Data 항목은
|
||||
// - 8 Byte Network byte order 로 구성된 available size 정보가 들어 있음.
|
||||
if( GetPacketData( m_nPacketDataLen ) == false )
|
||||
{
|
||||
LOG( LERR, "cc_collectd REQUEST_FILE_CHECK file size body receive failed." );
|
||||
strErrorMessage = "cc_collectd REQUEST_FILE_CHECK file size body receive failed.";
|
||||
return false;
|
||||
}
|
||||
|
||||
BYTE * pPos = m_tempBuffer;
|
||||
uFileSize = GetDataToUInt64( pPos, true );
|
||||
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
// 오류가 발생한 경우..
|
||||
// - 수신할 메시지가 존재하는 경우.. 수신 처리한다.
|
||||
if( m_nPacketDataLen > 0 )
|
||||
{
|
||||
if( GetPacketData( m_nPacketDataLen, strErrorMessage ) == false )
|
||||
{
|
||||
LOG( LERR, "cc_collectd REQUEST_FILE_CHECK error body receive failed." );
|
||||
strErrorMessage = "cc_collectd REQUEST_FILE_CHECK error body receive failed.";
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
// 오류 반환 처리.
|
||||
return false;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,148 @@
|
||||
/***************************************************************************
|
||||
client 에서 cc_collectd 와의 통신을 수행하기 위한 class
|
||||
-----------------------------------------
|
||||
begin : 2014/09/29
|
||||
copyright : (C) 2005 Solbox Inc.
|
||||
author : Storage Dev Team
|
||||
email : storage.sd@solbox.com
|
||||
version : 3.4
|
||||
|
||||
CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of Solbox Inc.
|
||||
***************************************************************************/
|
||||
|
||||
#ifndef __CC_COLLECTD_CLIENT_SOCKET_H__
|
||||
#define __CC_COLLECTD_CLIENT_SOCKET_H__
|
||||
|
||||
#include "BaseSocket.h"
|
||||
#include "CcCollectdProtocol.h"
|
||||
|
||||
#include <string>
|
||||
#include <stdint.h>
|
||||
|
||||
|
||||
///< BYTE 타입 정의
|
||||
#ifndef _BYTE_DEFINED
|
||||
#define _BYTE_DEFINED
|
||||
typedef unsigned char BYTE;
|
||||
#endif // _BYTE_DEFINED
|
||||
|
||||
|
||||
#define DEFAULT_SOCKET_TEMP_BUFFER_SIZE 1024 // Socket 관련 data 송수신시 사용할 임시버퍼 크기.
|
||||
|
||||
|
||||
// CCcCollectdClientSocket
|
||||
// cc_collectd 모듈과 통신 수행 관련 Interface 지원 목적을 위한 통신 관련 처리 class
|
||||
class CCcCollectdClientSocket : public CBaseSocket
|
||||
{
|
||||
|
||||
public:
|
||||
|
||||
// 생성자
|
||||
CCcCollectdClientSocket();
|
||||
|
||||
// 소멸자
|
||||
~CCcCollectdClientSocket();
|
||||
|
||||
|
||||
// 전달받은 Target 으로 Socket 접속을 수행
|
||||
// @param szTarget [in] 접속 대상 Host name 또는 IP
|
||||
// @param nPort [in] 접속 Port
|
||||
// @return 접속 성공시 true, 실패시 false 반환. ( 오류 내역은 로깅처리됨.)
|
||||
bool ConnectTarget( const std::string& szTarget, int nPort );
|
||||
|
||||
// cc_collect 로 File 존재 여부 확인
|
||||
// @param strProcessName [in] 프로세스 명, 저장시 프로세스 명에 따른 폴더 생성 후 파일 수신, 저장 처리함.
|
||||
// 생략 가능하지만.. '/' 문자가 포함되서는 안됨.
|
||||
//
|
||||
// @param strFileName [in] 저장 파일명
|
||||
// a.log 같은 단순 형식 사용,
|
||||
// /a/b.log 같은 경로 정보 포함시 오류 발생.
|
||||
//
|
||||
// @param nFileSize [out] 해당 파일이 존재할 경우 파일의 크기
|
||||
// 만약 해당 파일이 존재하지 않을 경우.. 크기는 0
|
||||
//
|
||||
// @param strErrorMessage [out] 수신된 오류 메시지 저장
|
||||
// @return
|
||||
// 성공시 true, 파일 크기는 nFileSize 항목 참고.
|
||||
// 오류 발생시 false, 오류 내역은 strMessage 항목 참고
|
||||
bool CheckFile( const std::string strProcessName, const std::string strFileName
|
||||
, unsigned long long& uFileSize
|
||||
, std::string& strErrorMessage );
|
||||
|
||||
|
||||
// cc_collect 로 File 저장 목적으로 전송
|
||||
// @param strProcessName [in] 프로세스 명, 저장시 프로세스 명에 따른 폴더 생성 후 파일 수신, 저장 처리함.
|
||||
// 생략 가능하지만.. '/' 문자가 포함되서는 안됨.
|
||||
//
|
||||
// @param strFileName [in] 저장 파일명
|
||||
// a.log 같은 단순 형식 사용,
|
||||
// /a/b.log 같은 경로 정보 포함시 오류 발생.
|
||||
//
|
||||
// @param fd [in] 전송할 파일에 대한 descriter
|
||||
// @param nFileSize [in] 전송할 파일의 크기
|
||||
// 0 파일 전송 가능.
|
||||
// @param strMessage [out] 수신된 메시지 저장 (정상, 오류 모두 메시지 수신 가능)
|
||||
// @return
|
||||
// 저장 성공시 true
|
||||
// 실패 발생시 false, 오류 내역은 strMessage 항목 참고
|
||||
bool SendFile( const std::string strProcessName, const std::string strFileName
|
||||
, const int& fd, const unsigned long long& uFileSize
|
||||
, std::string& strMessage );
|
||||
|
||||
|
||||
private:
|
||||
|
||||
// cc_collect 로부터 available size 정보를 요청, 결과를 수신처리한다.
|
||||
bool GetAvailableSize( unsigned long long& uSize );
|
||||
|
||||
|
||||
private:
|
||||
|
||||
// Packet Header 정보를 Log 파일에 Logging 처리
|
||||
void PrintHeaderToLog();
|
||||
|
||||
// socket 에서 지정된 크기만큼의 데이터를 읽어 내부 임시버퍼인 m_tempBuffer 에 저장처리.
|
||||
// @param size [in] read 할 데이터 크기
|
||||
// @return On success return true, otherwise return false.
|
||||
bool GetPacketData( unsigned int size );
|
||||
|
||||
// pValue 에 저장된 데이터를 unsigned long long (64Byte) 형으로 변환처리.
|
||||
unsigned long long GetDataToUInt64( BYTE * pValue, bool bConvertEndian = true );
|
||||
|
||||
// socket 에서 지정된 크기만큼의 데이터를 읽어 출력변수에 저장처리.
|
||||
// @param size [in] read 할 데이터 크기
|
||||
// @param value [out] 읽은 데이터를 저장할 string 변수
|
||||
// @return On success return true, otherwise return false.
|
||||
bool GetPacketData( unsigned int& size, std::string& value );
|
||||
|
||||
// Packet 정보 중 String 정보를 전송하기 위한 함수.
|
||||
bool SendString( const std::string& strData );
|
||||
|
||||
// Linux 5.4 의 경우 unsigned long long 의 byteorder 변환함수 제공 안함.
|
||||
// Linux 6.X, FreeBSD 에서는 be64toh 등의 함수 제공
|
||||
// - Liunx 5.X 장비 지원해야 하므로... 64bit 변환 함수 개발하여 사용.
|
||||
// - 기존 htonl 등의 함수와의 혼동을 피하기 위해 함수명은 C Style 로 구성.
|
||||
uint64_t do_htonll( uint64_t hostlonglong );
|
||||
uint64_t do_ntohll( uint64_t netlonglong );
|
||||
|
||||
|
||||
private:
|
||||
|
||||
// Packet Header 변수
|
||||
struct CcCollectdPacketHeader m_packetHeader;
|
||||
|
||||
// m_packetHeader 구조체의 크기를 저장하기 위한 상수
|
||||
const int m_nPacketHeaderLen;
|
||||
|
||||
// Packet Header 에 저장된 Data 부분의 길이 정보값.
|
||||
unsigned int m_nPacketDataLen;
|
||||
|
||||
// Packet Data 부분의 수신처리시 임시로 사용할 버퍼.
|
||||
BYTE m_tempBuffer[DEFAULT_SOCKET_TEMP_BUFFER_SIZE];
|
||||
|
||||
};
|
||||
|
||||
#endif /* __CC_COLLECTD_CLIENT_SOCKET_H__ */
|
||||
|
||||
@@ -0,0 +1,62 @@
|
||||
/***************************************************************************
|
||||
cc_collectd 통신 관련 protocol header
|
||||
-----------------------------------------
|
||||
begin : 2014/09/25
|
||||
copyright : (C) 2005 Solbox Inc.
|
||||
author : Storage Dev Team
|
||||
email : storage.sd@solbox.com
|
||||
version : 3.4
|
||||
|
||||
CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of Solbox Inc.
|
||||
***************************************************************************/
|
||||
|
||||
|
||||
#ifndef __CC_COLLECTD_PROTOCOL_H__
|
||||
#define __CC_COLLECTD_PROTOCOL_H__
|
||||
|
||||
|
||||
// cc_collectd 에서 사용할 통신 Header 구조체
|
||||
// - 계산상 크기는 16Byte 이지만...
|
||||
// - 통신상 sizeof() : 20Byte ( 64Bit OS )
|
||||
struct CcCollectdPacketHeader {
|
||||
|
||||
char stx; // Packet 유효성 관리 코드
|
||||
char type; // Request or Response 여부 ( 0x00: Request, 0x01: Response )
|
||||
char command[4]; // Command Code ( 4 Byte) : 0th client-cc_collectd 간 사용, 그외는 미사용.
|
||||
char result[4]; // Result Code ( 4 Byte )
|
||||
unsigned int data_length; // Packet Data 부분의 길이값 ( Network Byte Order 사용)
|
||||
char extend_code[2]; // 확장 및 Padding bits ( 2 Byte )
|
||||
};
|
||||
|
||||
// Packet Header stx 코드
|
||||
#define HEADER_STX_CODE 0x04
|
||||
|
||||
// Packet Header type 구분코드
|
||||
#define HEADER_TYPE_REQUEST 0x00
|
||||
#define HEADER_TYPE_RESPONSE 0x01
|
||||
|
||||
|
||||
// Packet command 공통.
|
||||
#define COMMON_ALIVE_CHECK 0x7F
|
||||
|
||||
// Packet command
|
||||
// client <-> cc_collect 통신 (command 0th byte 만 사용)
|
||||
#define REQUEST_NOT_VALID 0x00 // 유효하지 않은 요청.
|
||||
#define REQUEST_AVAILABLE_SIZE 0x01 // Client 에서 전송 가능한 Available Size 요청
|
||||
#define REQUEST_FILE_SAVE 0x02 // Client 에서 저장 목적의 File 전송
|
||||
#define REQUEST_FILE_CHECK 0x03 // Client 에서 File 존재 여부 확인 요청
|
||||
|
||||
|
||||
// Packet Result
|
||||
// => type이 Response 인 경우에만 세팅됨.( 0th Byte 만 사용시 )
|
||||
#define HEADER_RESULT_SUCCESS 0x00
|
||||
#define HEADER_RESULT_ERROR 0x01
|
||||
|
||||
|
||||
// 기타 정보
|
||||
#define LENGTH_FIELD_SIZE 4 // 가변데이터 형식 사용시 Length 필드의 메모리 크기 (unsigned int)
|
||||
|
||||
|
||||
#endif /* __CC_COLLECTD_PROTOCOL_H__ */
|
||||
@@ -0,0 +1,62 @@
|
||||
#****************************************************************************
|
||||
# Makefile for cc_collectd client interface sample
|
||||
# -----------------------------------------
|
||||
#
|
||||
# begin : 2014/09/29
|
||||
# copyright : (C) 2005 Solbox Inc.
|
||||
# author : Storage Dev Team
|
||||
# email : storage.sd@solbox.com
|
||||
# version : 3.5
|
||||
#
|
||||
# CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
# Redistribution and use in source and binary forms, with or with out
|
||||
# modification, are not permitted in outside of Solbox Inc.
|
||||
#*****************************************************************************
|
||||
|
||||
|
||||
# Program info
|
||||
PROG_NAME = collectd_client
|
||||
|
||||
|
||||
# Compiler info
|
||||
CC = /usr/bin/g++
|
||||
CFLAGS = -Wall -O3 -g -Wreturn-type -Wunused -Wuninitialized -Wparentheses -Wshadow -Wpointer-arith -Woverloaded-virtual -fno-rtti -D_REENTRANT
|
||||
LFLAGS =
|
||||
DFLAGS = -D_DEBUG_ -DPROG_NAME=\"$(PROG_NAME)\"
|
||||
|
||||
|
||||
# Enviroment
|
||||
DIR_INCLUDE = -I./. -I./.. -I./../../lib
|
||||
DIR_LIB =
|
||||
|
||||
OBJ = ../../lib/Logger.o ../../lib/BaseSocket.o ../CcCollectdClientSocket.o main.o
|
||||
|
||||
|
||||
LIBS =
|
||||
|
||||
APP = $(PROG_NAME)
|
||||
############################
|
||||
|
||||
|
||||
all:$(APP)
|
||||
sync
|
||||
|
||||
%.o: %.cpp
|
||||
$(CC) $(CFLAGS) -o $@ -c $^ $(DFLAGS) $(DIR_INCLUDE)
|
||||
|
||||
|
||||
$(PROG_NAME): $(OBJ)
|
||||
$(CC) $(LFLAGS) -o $@ $^ $(DFLAGS) $(DIR_LIB) $(LIBS)
|
||||
|
||||
|
||||
clean:
|
||||
-rm -f *.o core *.out *.log ./../*.o
|
||||
-rm -f $(APP)
|
||||
sync
|
||||
|
||||
|
||||
install : $(APP)
|
||||
sync
|
||||
|
||||
# End of Makefile
|
||||
|
||||
@@ -0,0 +1,150 @@
|
||||
#include "Logger.h"
|
||||
#include "CcCollectdClientSocket.h"
|
||||
|
||||
#include <string>
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
#include <fcntl.h>
|
||||
#include <errno.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include <string.h>
|
||||
|
||||
#define LOG_PATH "/user/service/logs" // Log 경로
|
||||
|
||||
|
||||
#define SERVER_ADDR "222.122.152.5" // KT LogBackup Server IP
|
||||
//#define SERVER_ADDR "210.124.122.211" // LG U+ LogBackup Server IP
|
||||
|
||||
#define SERVER_LISTEN_PORT 13103 // cc_collectd Listen Port
|
||||
|
||||
|
||||
int main(int argc, char * argv[])
|
||||
{
|
||||
|
||||
std::string strServerAddr = SERVER_ADDR;
|
||||
std::string strFullFileName;
|
||||
|
||||
if( argc == 2 )
|
||||
{
|
||||
strFullFileName = argv[1];
|
||||
}
|
||||
else if ( argc > 2 )
|
||||
{
|
||||
fprintf( stderr, "Too many parameter.\n");
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
else
|
||||
{
|
||||
// default
|
||||
strFullFileName = "/root/work/cc_collectd/interface/sample/test1.txt";
|
||||
}
|
||||
|
||||
|
||||
// 파일 Full 경로에서 파일명 추출 처리.
|
||||
std::string::size_type pos = strFullFileName.rfind( '/' );
|
||||
if( pos == std::string::npos )
|
||||
{
|
||||
// 찾지 못한 경우...
|
||||
fprintf( stderr, "File path is FULL PATH use. [%s]\n", strFullFileName.c_str() );
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
// 전송할 파일 정보
|
||||
std::string strTargetDir = strFullFileName.substr( 0, pos );
|
||||
std::string strTragetFileName = strFullFileName.substr( pos+1 );
|
||||
|
||||
|
||||
|
||||
// 전송할 파일 관련 정보 추출.
|
||||
struct stat resultFileStat;
|
||||
|
||||
if( stat( strFullFileName.c_str(), &resultFileStat ) != 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
fprintf( stderr, "file[%s] stat check error.[%d][%s]\n", strFullFileName.c_str(), errorNum, strerror( errorNum ) );
|
||||
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
// 파일 size 추출.
|
||||
unsigned long long uFileSize = resultFileStat.st_size;
|
||||
|
||||
// fd open
|
||||
int fd = open(strFullFileName.c_str(), O_RDONLY );
|
||||
if( fd == -1 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
fprintf( stderr, "file[%s] open fail.[%d][%s]\n", strFullFileName.c_str(), errorNum, strerror( errorNum ) );
|
||||
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
|
||||
// Log 처리용 객체 생성 및 초기화.
|
||||
if( CLogger::Init( PROG_NAME, LOG_PATH, LDBG ) == false )
|
||||
{
|
||||
close( fd );
|
||||
|
||||
fprintf( stderr, "[error] Logger object created failed..");
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
|
||||
// socket 생성 및 연결
|
||||
CCcCollectdClientSocket serverSocket;
|
||||
if( serverSocket.ConnectTarget( strServerAddr, SERVER_LISTEN_PORT ) == false )
|
||||
{
|
||||
// 접속 실패시
|
||||
close( fd );
|
||||
|
||||
_LOG( LERR, "cc_collectd connect fail. [%s][%d]", strServerAddr.c_str(), SERVER_LISTEN_PORT );
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
std::string szErrorMessage;
|
||||
|
||||
// Log 파일 존재 여부 확인
|
||||
unsigned long long uExistFileSize = 0;
|
||||
|
||||
if( serverSocket.CheckFile( "collectd_client", strTragetFileName, uExistFileSize, szErrorMessage ) == false )
|
||||
{
|
||||
close( fd );
|
||||
|
||||
_LOG( LERR, "cc_collectd exist check fail.[%s][%llu] -> [%s]", strTragetFileName.c_str(), uFileSize, szErrorMessage.c_str() );
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
else
|
||||
{
|
||||
_LOG( LDBG, "cc_collectd exist check OK.local[%s][%llu] remote[%llu]", strTragetFileName.c_str(), uFileSize, uExistFileSize );
|
||||
|
||||
|
||||
// Log 파일 크기가 다른 경우에만 전송 처리한다.
|
||||
if( uFileSize == uExistFileSize )
|
||||
{
|
||||
_LOG( LDBG, "cc_collectd same file exist. not send. local[%s][%llu] remote[%llu]", strTragetFileName.c_str(), uFileSize, uExistFileSize );
|
||||
}
|
||||
else
|
||||
{
|
||||
// 파일 크기가 다른 경우에만 전송 처리.
|
||||
// Log 파일 전송 처리
|
||||
if( serverSocket.SendFile( "collectd_client", strTragetFileName, fd, uFileSize, szErrorMessage ) == false )
|
||||
{
|
||||
close( fd );
|
||||
|
||||
_LOG( LERR, "cc_collectd send fail.[%s][%llu] -> [%s]", strTragetFileName.c_str(), uFileSize, szErrorMessage.c_str() );
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
else
|
||||
{
|
||||
|
||||
_LOG( LDBG, "cc_collectd send OK.[%s][%llu]", strTragetFileName.c_str(), uFileSize );
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
close( fd );
|
||||
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
@@ -0,0 +1,2 @@
|
||||
This is test file for cc_collectd transfer.
|
||||
|
||||
+12
@@ -0,0 +1,12 @@
|
||||
#!/bin/sh
|
||||
|
||||
CMD="$1"
|
||||
SLEEP="$2"
|
||||
|
||||
while [ 1 ]; do
|
||||
date
|
||||
eval $CMD
|
||||
echo " "
|
||||
sleep $SLEEP
|
||||
done
|
||||
|
||||
@@ -0,0 +1,595 @@
|
||||
#include "BaseSocket.h"
|
||||
|
||||
|
||||
#include <sys/select.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/types.h>
|
||||
#include <netinet/in.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <netdb.h>
|
||||
#include <errno.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include <time.h>
|
||||
|
||||
|
||||
#define SOCKET_NOT_VALID -1
|
||||
#define MAX_SEND_RETRY_COUNT 10 // Socket send 실패시 최대 재전송 시도 횟수.
|
||||
|
||||
|
||||
/// @brief 생성자.
|
||||
/// @param socket [in] 처리할 socket descriptor
|
||||
CBaseSocket::CBaseSocket( const int& socket )
|
||||
: m_sock( socket )
|
||||
{
|
||||
if( m_sock < 0 )
|
||||
{
|
||||
m_sock = SOCKET_NOT_VALID;
|
||||
m_bConnected = false;
|
||||
}
|
||||
else
|
||||
{
|
||||
m_bConnected = true;
|
||||
}
|
||||
}
|
||||
|
||||
/// @brief 소멸자
|
||||
CBaseSocket::~CBaseSocket()
|
||||
{
|
||||
// 소멸자 Socket 명시적 Close 처리.
|
||||
Close();
|
||||
}
|
||||
|
||||
/// @brief 소켓의 Close 처리를 수행함.
|
||||
void CBaseSocket::Close()
|
||||
{
|
||||
m_bConnected = false;
|
||||
|
||||
if( m_sock != SOCKET_NOT_VALID )
|
||||
{
|
||||
close( m_sock );
|
||||
m_sock = SOCKET_NOT_VALID;
|
||||
}
|
||||
}
|
||||
|
||||
/// @brief 전달받은 Target 으로 Socket 접속을 수행
|
||||
/// @param szTarget [in] 접속 대상 Host name 또는 IP
|
||||
/// @param nPort [in] 접속 Port
|
||||
/// @return 접속 성공시 true, 실패시 false 반환. ( 오류 내역은 로깅처리됨.)
|
||||
bool CBaseSocket::Connect( const std::string& szTarget, int nPort )
|
||||
{
|
||||
// 기존 접속을 Close 처리
|
||||
Close();
|
||||
|
||||
unsigned int nHost = ConversionAddr( szTarget.c_str() );
|
||||
if( nHost == INADDR_NONE )
|
||||
{
|
||||
LOG( LERR, "target host name resolve fail. [%s]->INADDR_NONE", szTarget.c_str());
|
||||
return false;
|
||||
}
|
||||
|
||||
struct sockaddr_in stTargetAddr;
|
||||
|
||||
m_sock = socket(AF_INET, SOCK_STREAM, 0);
|
||||
if( m_sock == -1 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
|
||||
LOG( LERR, "socket create failed.[%d][%s] Target:[%s]", errorNum, strerror(errorNum), szTarget.c_str() );
|
||||
|
||||
m_sock = SOCKET_NOT_VALID;
|
||||
return false;
|
||||
}
|
||||
|
||||
stTargetAddr.sin_family = AF_INET;
|
||||
stTargetAddr.sin_addr.s_addr = nHost;
|
||||
stTargetAddr.sin_port = htons( nPort );
|
||||
|
||||
// NEW 2012-05-18 huibong
|
||||
// Connnection 종료시 많은 TIME_WAIT 상태 발생으로 인해 .. 이를 제거하기 위해 SO_LINGER 옵션을 설정처리한다.
|
||||
struct linger opt_linger;
|
||||
opt_linger.l_onoff = 1; /* LINGER ON */
|
||||
|
||||
// CHG 2014-05-27 huibong
|
||||
// 아래와 같이 No Wait 로 처리할 경우... Server 역활 수행 중 Close 신호를 받지 못하는 경우가 발생 가능함.
|
||||
// 따라서 5 sec 정도 대기하도록 수정 처리한다.
|
||||
//opt_linger.l_linger = 0; /* No Wait => 0 for abortive disconnect */
|
||||
opt_linger.l_linger = 5; // 5 sec 대기
|
||||
|
||||
int result = setsockopt( m_sock, SOL_SOCKET, SO_LINGER, &opt_linger, sizeof(opt_linger));
|
||||
if( result != 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "setsockopt func SO_LINGER set fail.[%d][%s]", errorNum, strerror(errorNum));
|
||||
|
||||
}
|
||||
|
||||
/* Send Timeout 설정. */
|
||||
struct timeval tv_timeo = { 5, 0 };
|
||||
|
||||
result = setsockopt( m_sock, SOL_SOCKET, SO_SNDTIMEO, &tv_timeo, sizeof(tv_timeo));
|
||||
if( result != 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "setsockopt func SO_SNDTIMEO set fail.[%d][%s]", errorNum, strerror(errorNum));
|
||||
|
||||
}
|
||||
|
||||
if(connect(m_sock, (struct sockaddr *)&stTargetAddr, sizeof(stTargetAddr)) < 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "connect fail.[%s][%d] [%d][%s]", szTarget.c_str(), nPort, errorNum, strerror( errorNum ) );
|
||||
|
||||
Close();
|
||||
return false;
|
||||
}
|
||||
else
|
||||
{
|
||||
m_bConnected = true;
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/// @brief 멤버 변수인 m_sock 이 유효하고 연결된 상태인 경우 true 반환.
|
||||
/// @return socket이 유효하지 않거나 연결이 끊어지 경우 false 반환, 그외에는 true 반환.
|
||||
bool CBaseSocket::IsValidSocket()
|
||||
{
|
||||
if( m_sock != SOCKET_NOT_VALID && m_bConnected == true )
|
||||
return true;
|
||||
else
|
||||
return false;
|
||||
}
|
||||
|
||||
/// @brief m_sock 으로부터 지정된 size 만큼 데이터 read 를 시도 ( read 함수와 동일 )
|
||||
/// @param vptr [out] Read 된 데이터를 저장하기 위한 변수에 대한 포인터.
|
||||
/// @param size [in] read 하고자 하는 데이터의 크기.
|
||||
/// @return read 된 데이터의 크기. 0: fd closed, -1: 오류
|
||||
ssize_t CBaseSocket::Read( void * vptr, size_t size )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return 0;
|
||||
|
||||
ssize_t nRead = 0;
|
||||
while( (nRead = read( m_sock, vptr, size )) < 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
if( errorNum == EINTR || errorNum == EAGAIN || errorNum == EWOULDBLOCK )
|
||||
{
|
||||
nRead = 0;
|
||||
continue;
|
||||
}
|
||||
else if( errorNum == ECONNRESET || errorNum == EPIPE )
|
||||
{
|
||||
// 2010-07-23 BUG huibong 잘못된 대입연산자를 비교연산자로 수정.
|
||||
// 2015-08-28 CHG huibong EPIPE(32) 오류 발생시.. 연결 종료로 처리되도록 수정
|
||||
m_bConnected = false;
|
||||
return 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
LOG( LERR, "read func fail.[%d][%s]", errorNum, strerror(errorNum) );
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
return nRead;
|
||||
}
|
||||
|
||||
/// @brief m_sock 로부터 지정된 size 만큼 read 를 시도, \n
|
||||
///< 만약 지정된 크기만큼 데이터가 존재하지 않을 경우 해당 크기만큼 데이터를 read 할때까지 대기.
|
||||
/// @param vptr [out] Read 된 데이터를 저장하기 위한 변수에 대한 포인터.
|
||||
/// @param size [in] read 하고자 하는 데이터의 크기.
|
||||
/// @return read 된 데이터의 크기. 0: fd closed, -1: 오류
|
||||
ssize_t CBaseSocket::ReadN( void * vptr, size_t size )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return 0;
|
||||
|
||||
ssize_t nRead = 0;
|
||||
size_t nLeft = size;
|
||||
char * ptr = (char *)vptr;
|
||||
|
||||
while( nLeft > 0 )
|
||||
{
|
||||
if( (nRead = read( m_sock, ptr, nLeft )) < 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
if( errorNum == EINTR || errorNum == EAGAIN || errorNum == EWOULDBLOCK )
|
||||
{
|
||||
nRead = 0;
|
||||
continue;
|
||||
}
|
||||
else if( errorNum == ECONNRESET || errorNum == EPIPE )
|
||||
{
|
||||
// 2015-08-28 CHG huibong EPIPE(32) 오류 발생시.. 연결 종료로 처리되도록 수정
|
||||
m_bConnected = false;
|
||||
return 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
LOG( LERR, "read func fail.[%d][%s]", errorNum, strerror(errorNum) );
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
else if( nRead == 0 )
|
||||
{
|
||||
m_bConnected = false;
|
||||
return 0;
|
||||
}
|
||||
|
||||
nLeft -= nRead;
|
||||
ptr += nRead;
|
||||
}
|
||||
|
||||
return size;
|
||||
}
|
||||
|
||||
/// @brief m_sock 로부터 지정된 size 만큼 read 를 시도, \n
|
||||
///< 만약 지정된 크기만큼 데이터가 존재하지 않을 경우 해당 크기만큼 데이터를 read 할때까지 대기. \n
|
||||
///< 또는 지정된 Timeout 값 동안 read 를 하지 못하는 경우 오류 처리.
|
||||
/// @param vptr [out] Read 된 데이터를 저장하기 위한 변수에 대한 포인터.
|
||||
/// @param size [in] read 하고자 하는 데이터의 크기.
|
||||
/// @param timeout [in] Timeout value (sec)
|
||||
/// @return read 된 데이터의 크기. 0: fd closed, -1: 오류 , -2: Timeout
|
||||
ssize_t CBaseSocket::ReadNTimeout( void * vptr, size_t size, int timeout )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return 0;
|
||||
|
||||
ssize_t nRead = 0;
|
||||
size_t nLeft = size;
|
||||
char * ptr = (char *)vptr;
|
||||
|
||||
struct timeval timeOver;
|
||||
int result;
|
||||
|
||||
fd_set selectFds;
|
||||
FD_ZERO( &selectFds );
|
||||
|
||||
while( nLeft > 0 )
|
||||
{
|
||||
timeOver.tv_sec = timeout;
|
||||
timeOver.tv_usec = 0;
|
||||
FD_SET( m_sock, &selectFds );
|
||||
|
||||
result = select( m_sock+1, &selectFds, (fd_set *)NULL, (fd_set *)NULL, &timeOver );
|
||||
|
||||
if( result > 0 )
|
||||
{
|
||||
if( FD_ISSET( m_sock, &selectFds ))
|
||||
{
|
||||
if( (nRead = read( m_sock, ptr, nLeft) ) < 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
if(errorNum == EINTR || errorNum == EAGAIN || errorNum == EWOULDBLOCK)
|
||||
{
|
||||
nRead = 0;
|
||||
continue;
|
||||
}
|
||||
else if( errorNum == ECONNRESET || errorNum == EPIPE )
|
||||
{
|
||||
// 2015-08-28 CHG huibong EPIPE(32) 오류 발생시.. 연결 종료로 처리되도록 수정
|
||||
m_bConnected = false;
|
||||
return 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
LOG( LERR, "read func fail.[%d][%s]", errorNum, strerror(errorNum) );
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
else if( nRead == 0 )
|
||||
{
|
||||
m_bConnected = false;
|
||||
return 0;
|
||||
}
|
||||
|
||||
nLeft -= nRead;
|
||||
ptr += nRead;
|
||||
}
|
||||
|
||||
}
|
||||
else if( result == 0 ) // Timeout
|
||||
{
|
||||
//LOG( LDBG, "read timeout");
|
||||
return -2;
|
||||
}
|
||||
else
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "select func error.[%d][%s]", errorNum, strerror(errorNum) );
|
||||
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
return size;
|
||||
}
|
||||
|
||||
/// @brief m_sock 로부터 지정된 size 만큼 read 를 시도, \n
|
||||
///< 만약 지정된 크기만큼 데이터가 존재하지 않을 경우 해당 크기만큼 데이터를 read 할때까지 대기. \n
|
||||
///< 또는 지정된 Timeout 값 동안 read 를 하지 못하는 경우 오류 처리.
|
||||
///< 본 함수는 Socket 상에 이미 Data 가 존재하는 경우에만 사용.
|
||||
/// @param vptr [out] Read 된 데이터를 저장하기 위한 변수에 대한 포인터.
|
||||
/// @param size [in] read 하고자 하는 데이터의 크기.
|
||||
/// @param timeout [in] Timeout value (sec)
|
||||
/// @return read 된 데이터의 크기. 0: fd closed, -1: 오류, -2: Timeout
|
||||
ssize_t CBaseSocket::ReadNTimeout2( void * vptr, size_t size, int timeout )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return 0;
|
||||
|
||||
ssize_t nRead = 0;
|
||||
size_t nLeft = size;
|
||||
char * ptr = (char *)vptr;
|
||||
int errorNum;
|
||||
|
||||
// 우선은 읽기 시도.
|
||||
if( (nRead = read( m_sock, ptr, nLeft) ) < 0 )
|
||||
{
|
||||
errorNum = errno;
|
||||
if( errorNum == EINTR || errorNum == EAGAIN || errorNum == EWOULDBLOCK )
|
||||
{
|
||||
nRead = 0;
|
||||
}
|
||||
else if( errorNum == ECONNRESET || errorNum == EPIPE )
|
||||
{
|
||||
// 2015-08-28 CHG huibong EPIPE(32) 오류 발생시.. 연결 종료로 처리되도록 수정
|
||||
|
||||
m_bConnected = false;
|
||||
return 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
LOG( LERR, "read func fail.[%d][%s]", errorNum, strerror(errorNum) );
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
else if( nRead == 0 )
|
||||
{
|
||||
m_bConnected = false;
|
||||
return 0;
|
||||
}
|
||||
|
||||
nLeft -= nRead;
|
||||
ptr += nRead;
|
||||
|
||||
// 만약 더 받을 데이터가 존재한다면.
|
||||
if( nLeft > 0 )
|
||||
{
|
||||
struct timeval timeOver;
|
||||
int result;
|
||||
|
||||
fd_set selectFds;
|
||||
FD_ZERO( &selectFds );
|
||||
|
||||
while( nLeft > 0 )
|
||||
{
|
||||
timeOver.tv_sec = timeout;
|
||||
timeOver.tv_usec = 0;
|
||||
FD_SET( m_sock, &selectFds );
|
||||
|
||||
result = select( m_sock+1, &selectFds, (fd_set *)NULL, (fd_set *)NULL, &timeOver );
|
||||
|
||||
if( result > 0 )
|
||||
{
|
||||
if( FD_ISSET( m_sock, &selectFds ))
|
||||
{
|
||||
if( (nRead = read( m_sock, ptr, nLeft) ) < 0 )
|
||||
{
|
||||
errorNum = errno;
|
||||
if(errorNum == EINTR || errorNum == EAGAIN || errorNum == EWOULDBLOCK)
|
||||
{
|
||||
nRead = 0;
|
||||
continue;
|
||||
}
|
||||
else if( errorNum == ECONNRESET || errorNum == EPIPE )
|
||||
{
|
||||
// 2015-08-28 CHG huibong EPIPE(32) 오류 발생시.. 연결 종료로 처리되도록 수정
|
||||
m_bConnected = false;
|
||||
return 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
LOG( LERR, "read func fail.[%d][%s]", errorNum, strerror(errorNum) );
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
else if( nRead == 0 )
|
||||
{
|
||||
m_bConnected = false;
|
||||
return 0;
|
||||
}
|
||||
|
||||
nLeft -= nRead;
|
||||
ptr += nRead;
|
||||
}
|
||||
}
|
||||
else if( result == 0 ) // Timeout
|
||||
{
|
||||
//LOG( LDBG, "read timeout");
|
||||
return -2;
|
||||
}
|
||||
else
|
||||
{
|
||||
errorNum = errno;
|
||||
LOG( LERR, "select func error.[%d][%s]", errorNum, strerror(errorNum) );
|
||||
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return size;
|
||||
}
|
||||
|
||||
/// @brief m_sock 으로 지정된 크기만큼 vptr 의 데이터를 전송 시도.\n
|
||||
///< Send Timeout 옵션 설정으로 Write Timeout 설정 가능. \n
|
||||
///< 지정된 횟수만큼 재전송 실패시 오류로 처리함.
|
||||
/// @param vptr [in] 전달할 데이터를 저장한 변수에 대한 포인터.
|
||||
/// @param size [in] write 하고자 하는 데이터의 크기.
|
||||
/// @return write 된 데이터의 크기. 0: fd closed, -1: 오류
|
||||
ssize_t CBaseSocket::WriteN( const void * vptr, size_t size )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return 0;
|
||||
|
||||
size_t nLeft;
|
||||
ssize_t nWrite;
|
||||
|
||||
const char * ptr = (const char *)vptr;
|
||||
nLeft = size;
|
||||
|
||||
int nTryCount = 0;
|
||||
|
||||
while( nLeft > 0 )
|
||||
{
|
||||
if( (nWrite = send( m_sock, ptr, nLeft, 0 )) < 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
|
||||
if( errorNum == EINTR || errorNum == EAGAIN || errorNum == EWOULDBLOCK )
|
||||
{
|
||||
nWrite = 0;
|
||||
++nTryCount;
|
||||
}
|
||||
else if( errorNum == ECONNRESET || errorNum == EPIPE )
|
||||
{
|
||||
// 2015-08-28 CHG huibong EPIPE(32) 오류 발생시.. 연결 종료로 처리되도록 수정
|
||||
m_bConnected = false;
|
||||
return 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
LOG( LERR, "send func fail.[%d][%s]", errorNum, strerror(errorNum));
|
||||
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
else if( nWrite == 0 )
|
||||
{
|
||||
m_bConnected = false;
|
||||
return 0;
|
||||
}
|
||||
|
||||
if( nTryCount > MAX_SEND_RETRY_COUNT )
|
||||
{
|
||||
LOG( LERR, "send func failure due to exceeding count of retry[%d/%d]", nTryCount, MAX_SEND_RETRY_COUNT );
|
||||
return 0 ;
|
||||
}
|
||||
|
||||
nLeft -= nWrite;
|
||||
ptr += nWrite;
|
||||
}
|
||||
|
||||
return size;
|
||||
}
|
||||
|
||||
// CHG 2011-05-03 huibong
|
||||
// gethostbyname() 함수가 Thread Safe 하지 않기 때문에
|
||||
// DNS resolve 처리시 잘못된 정보를 반환할 가능성이 존재
|
||||
// 이에 따라 본 함수를 수정처리함.
|
||||
/*
|
||||
unsigned int CBaseSocket::ConversionAddr( const char * name )
|
||||
{
|
||||
struct hostent *he;
|
||||
int max;
|
||||
unsigned int retval;
|
||||
|
||||
if ((retval = inet_addr(name)) != INADDR_NONE)
|
||||
return retval;
|
||||
|
||||
he = gethostbyname(name);
|
||||
if (he == NULL)
|
||||
return INADDR_NONE;
|
||||
|
||||
for (max = 0; he->h_addr_list[max]; max++) ;
|
||||
if (max == 1)
|
||||
return *((unsigned int *)(he->h_addr_list[0]));
|
||||
else
|
||||
return *((unsigned int *)(he->h_addr_list[random() % max]));
|
||||
}
|
||||
*/
|
||||
|
||||
unsigned int CBaseSocket::ConversionAddr( const char * name )
|
||||
{
|
||||
unsigned int retval;
|
||||
struct addrinfo hints;
|
||||
struct addrinfo *result = NULL;
|
||||
int error;
|
||||
|
||||
|
||||
// 전달받은 정보가 IP 주소인 경우...
|
||||
// - DNS resolve 할 필요 없이 변환시킨 값을 그대로 사용한다.
|
||||
if( (retval = inet_addr(name)) != INADDR_NONE )
|
||||
return retval;
|
||||
|
||||
// getaddrinfo() 함수 호출을 위한 Hint 설정
|
||||
memset( &hints, 0x00, sizeof(hints));
|
||||
hints.ai_family = PF_UNSPEC;
|
||||
hints.ai_socktype = SOCK_STREAM;
|
||||
|
||||
// Thread Safe 한 DNS Resolve 처리함수 호출
|
||||
// int getaddrinfo(const char *hostname, const char *servname, const struct addrinfo *hints, struct addrinfo **res);
|
||||
|
||||
// CHG 2014-11-14 huibong
|
||||
// getaddrinfo 호출시 종종 EAI_NONAME 오류가 반환됨. (#20783)
|
||||
// - 따라서 오류 반환시 재시도하도록 기능 추가
|
||||
// CHG 2015-07-15 huibong
|
||||
// usleep 의 multi thread 상에서 block 발생 가능
|
||||
// - nanosleep 을 사용토록 변경 처리
|
||||
for( int count = 0 ; count < 3; count++ )
|
||||
{
|
||||
error = getaddrinfo( name, NULL, &hints, &result );
|
||||
|
||||
// 오류 발생시
|
||||
if( error != 0 )
|
||||
{
|
||||
if( result != NULL)
|
||||
{
|
||||
freeaddrinfo(result);
|
||||
result = NULL;
|
||||
}
|
||||
|
||||
// 잠시 대기 후 재시도 처리
|
||||
struct timespec sleep;
|
||||
sleep.tv_sec = 0;
|
||||
sleep.tv_nsec = 500000000; // 0.5 sec
|
||||
nanosleep( &sleep, NULL );
|
||||
}
|
||||
else
|
||||
{
|
||||
// 정상 처리된 경우...
|
||||
// - loop 탈출
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// 최종 오류 발생시
|
||||
if( error != 0 )
|
||||
{
|
||||
LOG( LERR, "name[%s] dns resolve fail. getaddrinfo return error [%d][%s]", name, error, gai_strerror(error) );
|
||||
return INADDR_NONE;
|
||||
}
|
||||
|
||||
struct sockaddr_in * addr = (struct sockaddr_in *)result->ai_addr;
|
||||
retval = (unsigned int)(addr->sin_addr.s_addr);
|
||||
|
||||
// DNS Resovle 결과 확인용 코드
|
||||
/*
|
||||
struct addrinfo *temp;
|
||||
for( temp = result; temp; temp = temp->ai_next )
|
||||
{
|
||||
addr = (struct sockaddr_in *)temp->ai_addr;
|
||||
printf ("getaddrinfo result = %s\n",inet_ntoa( addr->sin_addr));
|
||||
}
|
||||
*/
|
||||
|
||||
// getaddrinfo() 함수에서 생성한 메모리 영역 해제 처리.
|
||||
freeaddrinfo(result);
|
||||
|
||||
return retval;
|
||||
|
||||
}
|
||||
|
||||
@@ -0,0 +1,119 @@
|
||||
/***************************************************************************
|
||||
BaseSocket ( Base Socket Class )
|
||||
-----------------------------------------
|
||||
begin : 2010/03/02
|
||||
copyright : (C) 2005 SolutionBox Inc.
|
||||
author : Service 1 Team
|
||||
email : svc1@solbox.com
|
||||
version : 1.0
|
||||
|
||||
CopyRight(C) 2005 SolutionBox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of SolutionBox Inc.
|
||||
***************************************************************************/
|
||||
|
||||
#ifndef __LIBRARY_BASE_SOCKET_H__
|
||||
#define __LIBRARY_BASE_SOCKET_H__
|
||||
|
||||
#include <unistd.h>
|
||||
#include <sys/types.h>
|
||||
#include <string>
|
||||
|
||||
#include "Logger.h"
|
||||
|
||||
#define SOCKET_NOT_VALID -1
|
||||
#define DEFAULT_DATA_RECEIVE_TIMEOUT 15 // sec
|
||||
|
||||
|
||||
class CBaseSocket
|
||||
{
|
||||
|
||||
protected:
|
||||
|
||||
/// @brief socket descriptor
|
||||
int m_sock;
|
||||
|
||||
/// @brief socket 의 연결상태인지 여부를 저장하기 위한 변수.
|
||||
bool m_bConnected;
|
||||
|
||||
|
||||
public:
|
||||
|
||||
/// @brief 생성자.
|
||||
CBaseSocket( const int & socket );
|
||||
|
||||
/// @brief 소멸자.
|
||||
~CBaseSocket();
|
||||
|
||||
/// @brief 소켓의 Close 처리를 수행함.
|
||||
void Close(void);
|
||||
|
||||
/// @brief 전달받은 Target 으로 Socket 접속을 수행
|
||||
/// @param szTarget [in] 접속 대상 Host name 또는 IP
|
||||
/// @param nPort [in] 접속 Port
|
||||
/// @return 접속 성공시 true, 실패시 false 반환. ( 오류 내역은 로깅처리됨.)
|
||||
bool Connect( const std::string& szTarget, int nPort );
|
||||
|
||||
/// @brief 멤버 변수인 m_sock 이 유효하고 연결된 상태인 경우 true 반환.
|
||||
/// @return socket이 유효하지 않거나 연결이 끊어지 경우 false 반환, 그외에는 true 반환.
|
||||
bool IsValidSocket(void);
|
||||
|
||||
/// @brief 멤버변수인 socket 의 접속 상태 여부를 설정하기 위한 함수.
|
||||
void SetConnectionStatus( bool bConnected) { m_bConnected = bConnected; }
|
||||
|
||||
/// @brief 현재 socket 의 접속 상태를 반환.
|
||||
bool GetConnectionStatus( void) { return m_bConnected; }
|
||||
|
||||
//void SetLogger( Logger * pLog ) { m_pLog = pLog; }
|
||||
|
||||
protected:
|
||||
|
||||
/// @brief m_sock 로 부터 지정된 size 만큼 데이터 read 를 시도 ( read 함수와 동일 )
|
||||
/// @param vptr [out] Read 된 데이터를 저장하기 위한 변수에 대한 포인터.
|
||||
/// @param size [in] read 하고자 하는 데이터의 크기.
|
||||
/// @return read 된 데이터의 크기. 0: fd closed, -1: 오류
|
||||
ssize_t Read( void * vptr, size_t size );
|
||||
|
||||
/// @brief m_sock 로부터 지정된 size 만큼 read 를 시도, \n
|
||||
///< 만약 지정된 크기만큼 데이터가 존재하지 않을 경우 해당 크기만큼 데이터를 read 할때까지 대기.
|
||||
/// @param vptr [out] Read 된 데이터를 저장하기 위한 변수에 대한 포인터.
|
||||
/// @param size [in] read 하고자 하는 데이터의 크기.
|
||||
/// @return read 된 데이터의 크기. 0: fd closed, -1: 오류
|
||||
ssize_t ReadN( void * vptr, size_t size );
|
||||
|
||||
/// @brief m_sock 로부터 지정된 size 만큼 read 를 시도, \n
|
||||
///< 만약 지정된 크기만큼 데이터가 존재하지 않을 경우 해당 크기만큼 데이터를 read 할때까지 대기. \n
|
||||
///< 또는 지정된 Timeout 값 동안 read 를 하지 못하는 경우 오류 처리.
|
||||
/// @param vptr [out] Read 된 데이터를 저장하기 위한 변수에 대한 포인터.
|
||||
/// @param size [in] read 하고자 하는 데이터의 크기.
|
||||
/// @param timeout [in] Timeout value (sec)
|
||||
/// @return read 된 데이터의 크기. 0: fd closed, -1: 오류, -2: Timeout
|
||||
ssize_t ReadNTimeout( void * vptr, size_t size, int timeout = DEFAULT_DATA_RECEIVE_TIMEOUT );
|
||||
|
||||
/// @brief m_sock 로부터 지정된 size 만큼 read 를 시도, \n
|
||||
///< 만약 지정된 크기만큼 데이터가 존재하지 않을 경우 해당 크기만큼 데이터를 read 할때까지 대기. \n
|
||||
///< 또는 지정된 Timeout 값 동안 read 를 하지 못하는 경우 오류 처리.
|
||||
///< 본 함수는 Socket 상에 이미 Data 가 존재하는 경우에만 사용.
|
||||
/// @param vptr [out] Read 된 데이터를 저장하기 위한 변수에 대한 포인터.
|
||||
/// @param size [in] read 하고자 하는 데이터의 크기.
|
||||
/// @param timeout [in] Timeout value (sec)
|
||||
/// @return read 된 데이터의 크기. 0: fd closed, -1: 오류, -2: Timeout
|
||||
ssize_t ReadNTimeout2( void * vptr, size_t size, int timeout = DEFAULT_DATA_RECEIVE_TIMEOUT );
|
||||
|
||||
public:
|
||||
/// @brief m_sock 으로 지정된 크기만큼 vptr 의 데이터를 전송 시도.\n
|
||||
///< Send Timeout 옵션 설정으로 Write Timeout 설정 가능. \n
|
||||
///< 지정된 횟수만큼 재전송 실패시 오류로 처리함.
|
||||
/// @param vptr [in] 전달할 데이터를 저장한 변수에 대한 포인터.
|
||||
/// @param size [in] write 하고자 하는 데이터의 크기.
|
||||
/// @return write 된 데이터의 크기. 0: fd closed, -1: 오류
|
||||
ssize_t WriteN( const void * vptr, size_t size );
|
||||
|
||||
protected:
|
||||
|
||||
/// @brief 문자열로 전달받은 정보를 검사하여 IP 접속 정보를 생성한다.
|
||||
unsigned int ConversionAddr( const char * name );
|
||||
|
||||
};
|
||||
|
||||
#endif /* __LIBRARY_BASE_SOCKET_H__ */
|
||||
@@ -0,0 +1,215 @@
|
||||
|
||||
#include "Config.h"
|
||||
|
||||
#include <iostream>
|
||||
#include <fstream>
|
||||
|
||||
#include <errno.h>
|
||||
#include <stdlib.h>
|
||||
#include <boost/algorithm/string.hpp> // use boost
|
||||
|
||||
|
||||
Config::Config()
|
||||
: m_keyDelimiter( CONFIG_DEFAULT_KEY_DELIMITER )
|
||||
, m_valueDelimiter( CONFIG_DEFAULT_VALUE_DELIMITER )
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
/// @brief destructor
|
||||
Config::~Config()
|
||||
{
|
||||
Clear();
|
||||
}
|
||||
|
||||
/// @brief 지정된 Config 파일의 모든 정보를 읽어 내부 변수에 저장처리
|
||||
/// @param path [in] Config file 의 전체 경로정보( C 배열 지원을 위해 & 사용안함)
|
||||
/// @return On success return true, otherwise return false
|
||||
bool Config::Open( const string path )
|
||||
{
|
||||
ifstream file;
|
||||
|
||||
// Config file open
|
||||
file.open( path.c_str());
|
||||
if( file.is_open() == false )
|
||||
{
|
||||
cerr << "[error] " << __FILE__<< ":" << __func__ << ": config file open failed.[" << path << "][" << strerror( errno ) << "]" << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
// 기존 데이터가 존재하면 삭제 처리.
|
||||
if( m_configData.empty() == false )
|
||||
m_configData.clear();
|
||||
|
||||
// config file parsing
|
||||
string line;
|
||||
string section;
|
||||
vector< string > configs;
|
||||
|
||||
// 루프를 돌면서 Config 파일을 line 단위로 읽어들인당....
|
||||
while( std::getline( file, line ) )
|
||||
{
|
||||
// 앞뒤 공백 제거 처리
|
||||
boost::trim( line );
|
||||
|
||||
// 공백 or 주석처리 라인 검사
|
||||
if( line.empty() == true || boost::starts_with( line, string("#") ) == true || line == "\r" )
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
// Section 여부 검사
|
||||
if( boost::starts_with( line, string("[") ) == true && boost::ends_with( line, string("]") ) == true )
|
||||
{
|
||||
// [] 로 정의된 section 의 문자열 값 추출
|
||||
line.erase( line.begin() );
|
||||
line.erase( line.end() -1 );
|
||||
boost::trim( line );
|
||||
|
||||
if( section.empty() == false && section != line )
|
||||
{
|
||||
// 이전 세션과 다른 신규 세션인 경우
|
||||
// 기존까지 저장했던 데이터를 멤버 변수에 입력 처리 후 내부변수 초기화 처리.
|
||||
|
||||
m_configData.insert( make_pair( section, configs ) );
|
||||
section.clear();
|
||||
configs.clear();
|
||||
}
|
||||
|
||||
// 신규 section 정보 저장처리.
|
||||
section = line;
|
||||
line.clear();
|
||||
}
|
||||
else
|
||||
{
|
||||
// Section 정보가 아닌 경우 => 실제 설정값 .. ^^
|
||||
configs.push_back( line );
|
||||
}
|
||||
}
|
||||
|
||||
// 마지막 세션 처리된 정보가 존재하는 경우 멤버 변수에 저장 처리.
|
||||
if( section.empty() == false )
|
||||
{
|
||||
m_configData.insert( make_pair( section, configs ) );
|
||||
}
|
||||
|
||||
// 종료 처리.
|
||||
file.close();
|
||||
return true;
|
||||
}
|
||||
|
||||
/// @brief Config 멤버 변수에 저장된 데이터 삭제처리.
|
||||
/// @return void
|
||||
void Config::Clear()
|
||||
{
|
||||
m_configData.clear();
|
||||
}
|
||||
|
||||
/// @brief 해당 Section, Key 에 해당하는 value 값을 찾아 반환
|
||||
/// @param section [in] 찾고자 하는 Section 값
|
||||
/// @param key [in] 찾고자 하는 key 값
|
||||
/// @param value [out] 해당 Section, Key 에 해당하는 value 값을 저장하기 위한 변수
|
||||
/// @return On success return true, otherwise return false
|
||||
bool Config::GetConfig( const string section, const string key, string& value )
|
||||
{
|
||||
// Section 에 대한 임시 데이터 저장객체 생성
|
||||
vector< string > configs;
|
||||
|
||||
// 해당 Section 이 존재하지 않는 경우
|
||||
if( Find( section, configs ) == false )
|
||||
return false;
|
||||
|
||||
string line;
|
||||
string result;
|
||||
vector< string >::const_iterator it;
|
||||
|
||||
// 해당 Key 이 존재하는지 검사
|
||||
for( it = configs.begin(); it != configs.end(); it++)
|
||||
{
|
||||
line = *it;
|
||||
|
||||
// Line 상의 주석 제거
|
||||
string::size_type pos = line.find( '#' );
|
||||
if( pos != string::npos)
|
||||
{
|
||||
line = line.substr(0, pos);
|
||||
boost::trim( line );
|
||||
}
|
||||
|
||||
// key = value 에서 key 부분 추출
|
||||
pos = line.find( m_keyDelimiter );
|
||||
if( pos == string::npos )
|
||||
{
|
||||
continue;
|
||||
}
|
||||
else
|
||||
{
|
||||
result = line.substr(0, pos);
|
||||
boost::trim( result );
|
||||
}
|
||||
|
||||
if( key == result )
|
||||
{
|
||||
// Key 값이 동일한 경우
|
||||
value = line.substr( pos+1 );
|
||||
boost::trim( value );
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/// @brief 해당 Section, Key 에 해당하는 value Array 값을 찾아 반환
|
||||
/// @param section [in] 찾고자 하는 Section 값
|
||||
/// @param key [in] 찾고자 하는 key 값
|
||||
/// @param value [out] 해당 Section, Key 에 해당하는 value 값을 저장하기 위한 배열
|
||||
/// @return On success return true, otherwise return false
|
||||
bool Config::GetConfig( const string section, const string key, vector< string >& value )
|
||||
{
|
||||
string result;
|
||||
vector< string > vec;
|
||||
|
||||
if( GetConfig( section, key, result ) == false )
|
||||
return false;
|
||||
|
||||
// 전달받은 result 값을 value delimiter 를 이용하여 parsing 처리
|
||||
boost::split( vec, result, boost::is_any_of( m_valueDelimiter ));
|
||||
if( vec.empty() == true )
|
||||
return false;
|
||||
|
||||
// 루프를 돌면서 trim 처리 후 결과값 저장처리.
|
||||
vector< string >::const_iterator it;
|
||||
for( it = vec.begin(); it != vec.end(); it++)
|
||||
{
|
||||
result = boost::trim_copy( *it );
|
||||
|
||||
if( result.empty() == false )
|
||||
value.push_back( result );
|
||||
}
|
||||
|
||||
if( value.empty() == true )
|
||||
return false;
|
||||
else
|
||||
return true;
|
||||
}
|
||||
|
||||
/// @brief Config 멤버 변수에 저장된 데이터 중 해당 Section 에 해당하는 데이터 반환.
|
||||
/// @param section [in] section 명
|
||||
/// @param configData [out] 해당 Section 에서 읽은 Cofig 정보를 저장할 string 배열
|
||||
/// @return On success return true, otherwise return false
|
||||
bool Config::Find( const string& section, vector<string>& configData )
|
||||
{
|
||||
map< string, vector< string > >::iterator it = m_configData.find( section );
|
||||
if( it != m_configData.end() )
|
||||
{
|
||||
// 해당 section 을 찾은 경우
|
||||
configData = it->second;
|
||||
return true;
|
||||
}
|
||||
|
||||
// 해당 section 을 찾지 못한 경우
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,114 @@
|
||||
/***************************************************************************
|
||||
Config File Parser Class
|
||||
-----------------------------------------
|
||||
begin : 2010/02/27
|
||||
copyright : (C) 2005 SolutionBox Inc.
|
||||
author : Service 1 Team
|
||||
email : svc1@solbox.com
|
||||
version : 1.0
|
||||
|
||||
CopyRight(C) 2005 SolutionBox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of SolutionBox Inc.
|
||||
***************************************************************************/
|
||||
|
||||
#ifndef __LIBRARY_CONFIG_H__
|
||||
#define __LIBRARY_CONFIG_H__
|
||||
|
||||
#include <string>
|
||||
#include <map>
|
||||
#include <vector>
|
||||
|
||||
using namespace std;
|
||||
|
||||
|
||||
#define CONFIG_DEFAULT_KEY_DELIMITER "="
|
||||
#define CONFIG_DEFAULT_VALUE_DELIMITER ","
|
||||
|
||||
|
||||
/// @brief Config class
|
||||
/// 1. config 파일의 내용을 파싱처리하여 각 Section, Key 에 해당하는 값을 반환처리
|
||||
/// 2. config 파일은 [section] 단위로 구분된다.
|
||||
/// 3. config 파일은 key=value 로 구분가능하며 이는 Delimiter 설정을 통해 변경가능하다.
|
||||
/// 4. value 값이 다중으로 존재하는 경우 "," 값을 통해 구분가능, Delimiter 변경시 다른 값도 사용가능함.
|
||||
/// 5. config 파일상에서 "#" 로 시작하는 문자열은 라인 끝까지 주석으로 처리된다.
|
||||
/// 6. 특정 Key 값에 대한 다중 value 값 조회는 vector<string> 을 통해 수행한다.
|
||||
/// 7. 만약 다중 value 값이 존재시 string 으로 반환받을 경우 해당 Row 가 통째로 반환된다.
|
||||
/// 8. 본 객체는 Open , Clear 함수가 호출되기전까지 이전 Config 정보가 저장된다.
|
||||
class Config
|
||||
{
|
||||
private:
|
||||
/// @brief config information saved variable
|
||||
// string : section data
|
||||
// vector<string> : key=value data
|
||||
map< string, vector< string> > m_configData;
|
||||
|
||||
/// @brief Key, Value 구분자 저장 변수
|
||||
string m_keyDelimiter;
|
||||
string m_valueDelimiter;
|
||||
|
||||
public:
|
||||
/// @brief constructor
|
||||
Config();
|
||||
|
||||
/// @brief destructor
|
||||
~Config();
|
||||
|
||||
/// @brief 지정된 Config 파일의 모든 정보를 읽어 내부 변수에 저장처리
|
||||
/// @param path [in] Config file 의 전체 경로정보( C 배열 지원을 위해 & 사용안함)
|
||||
/// @return On success return true, otherwise return false
|
||||
bool Open( const string path );
|
||||
|
||||
/// @brief Config 멤버 변수에 저장된 데이터 삭제처리.
|
||||
/// @return void
|
||||
void Clear();
|
||||
|
||||
/// @brief 해당 Section, Key 에 해당하는 value 값을 찾아 반환
|
||||
/// @param section [in] 찾고자 하는 Section 값
|
||||
/// @param key [in] 찾고자 하는 key 값
|
||||
/// @param value [out] 해당 Section, Key 에 해당하는 value 값을 저장하기 위한 변수
|
||||
/// @return On success return true, otherwise return false
|
||||
bool GetConfig( const string section, const string key, string& value );
|
||||
|
||||
/// @brief 해당 Section, Key 에 해당하는 value Array 값을 찾아 반환
|
||||
/// @param section [in] 찾고자 하는 Section 값
|
||||
/// @param key [in] 찾고자 하는 key 값
|
||||
/// @param value [out] 해당 Section, Key 에 해당하는 value 값을 저장하기 위한 배열
|
||||
/// @return On success return true, otherwise return false
|
||||
bool GetConfig( const string section, const string key, vector< string >& value );
|
||||
|
||||
/// @brief Key, Value Delimiter Get 함수
|
||||
/// @param key [out] key delimiter 값
|
||||
/// @param value [out] value delimiter 값
|
||||
void GetDelimiter( string& key, string& value )
|
||||
{
|
||||
key = m_keyDelimiter;
|
||||
value = m_valueDelimiter;
|
||||
}
|
||||
|
||||
// BUG 2010-12-06 huibong
|
||||
// string& value = CONFIG_DEFAULT_VALUE_DELIMITER 값은 문법상 오류 구문임.
|
||||
// gcc 3.4.6 버전에서는 Compile 되나 gcc 4.4.5 에서는 error 로 처리되어 수정처리함.
|
||||
|
||||
/// @brief Key, Value Delimiter Set 함수
|
||||
/// @param key [out] key delimiter 값
|
||||
/// @param value [out] value delimiter 값, 지정하지 않을 경우 Default 값이 사용됨.
|
||||
void SetDelimiter( string& key, string value = CONFIG_DEFAULT_VALUE_DELIMITER )
|
||||
{
|
||||
m_keyDelimiter = key;
|
||||
m_valueDelimiter = value;
|
||||
}
|
||||
|
||||
|
||||
protected:
|
||||
/// @brief Config 멤버 변수에 저장된 데이터 중 해당 Section 에 해당하는 데이터 반환.
|
||||
/// @param section [in] section 명
|
||||
/// @param configData [out] 해당 Section 에서 읽은 Cofig 정보를 저장할 string 배열
|
||||
/// @return On success return true, otherwise return false
|
||||
bool Find( const string& section, vector<string>& configData );
|
||||
|
||||
};
|
||||
|
||||
|
||||
|
||||
#endif /* __LIBRARY_CONFIG_H__ */
|
||||
@@ -0,0 +1,487 @@
|
||||
/***************************************************************************
|
||||
Logger.cpp
|
||||
-----------------------------------------
|
||||
begin : 2011/10/26
|
||||
copyright : (C) 2005 SolutionBox Inc.
|
||||
author : Service 1 Team
|
||||
email : svc1@solbox.com
|
||||
version : 1.0
|
||||
|
||||
CopyRight(C) 2005 SolutionBox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of SolutionBox Inc.
|
||||
***************************************************************************/
|
||||
#include "Logger.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdarg.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
#include <iostream>
|
||||
#include <errno.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#define MAX_BUFFER_SIZE 2048 // 임시 버퍼 최대 크기
|
||||
#define DEFAULT_LOG_LEVEL LINF
|
||||
|
||||
using namespace std;
|
||||
|
||||
CLogger* CLogger::m_pInstance = NULL;
|
||||
bool CLogger::m_bIsInitialized = false;
|
||||
pthread_mutex_t CLogger::m_mutex = PTHREAD_MUTEX_INITIALIZER;
|
||||
|
||||
|
||||
CLogger::CLogger( string programName, string logDir, int logLevel )
|
||||
{
|
||||
m_szProgramName = programName;
|
||||
m_szLogDir = logDir;
|
||||
|
||||
if( IsValidLogLevel( logLevel ) == true )
|
||||
{
|
||||
m_nLogLevel = logLevel;
|
||||
}
|
||||
else
|
||||
{
|
||||
m_nLogLevel = DEFAULT_LOG_LEVEL;
|
||||
}
|
||||
|
||||
MakeLogLevelString();
|
||||
}
|
||||
|
||||
void CLogger::MakeLogLevelString()
|
||||
{
|
||||
m_vectorLogLevelString.push_back( "EMR" );
|
||||
m_vectorLogLevelString.push_back( "ALT" );
|
||||
m_vectorLogLevelString.push_back( "CRT" );
|
||||
m_vectorLogLevelString.push_back( "ERR" );
|
||||
m_vectorLogLevelString.push_back( "WAR" );
|
||||
m_vectorLogLevelString.push_back( "NOT" );
|
||||
m_vectorLogLevelString.push_back( "INF" );
|
||||
m_vectorLogLevelString.push_back( "DBG" );
|
||||
m_vectorLogLevelString.push_back( "DEV" );
|
||||
m_vectorLogLevelString.push_back( "DEV1" );
|
||||
m_vectorLogLevelString.push_back( "DEV2" );
|
||||
}
|
||||
|
||||
CLogger::~CLogger()
|
||||
{
|
||||
m_vectorLogLevelString.clear();
|
||||
}
|
||||
|
||||
bool CLogger::Init( string programName, string logDir, int logLevel )
|
||||
{
|
||||
if( m_bIsInitialized == true )
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
pthread_mutex_lock( &m_mutex );
|
||||
|
||||
// 변수 유효성 검사.
|
||||
if( programName.empty() == true || logDir.empty() == true )
|
||||
{
|
||||
pthread_mutex_unlock( &m_mutex );
|
||||
return false;
|
||||
}
|
||||
|
||||
if( IsValidLogLevel( logLevel ) == false )
|
||||
{
|
||||
pthread_mutex_unlock( &m_mutex );
|
||||
return false;
|
||||
}
|
||||
|
||||
if( MakeLogDir( programName, logDir ) == false )
|
||||
{
|
||||
pthread_mutex_unlock( &m_mutex );
|
||||
return false;
|
||||
}
|
||||
|
||||
if( CLogger::m_pInstance != NULL )
|
||||
{
|
||||
delete CLogger::m_pInstance;
|
||||
CLogger::m_pInstance = NULL;
|
||||
}
|
||||
|
||||
CLogger::m_pInstance = new CLogger( programName, logDir, logLevel );
|
||||
m_bIsInitialized = true;
|
||||
|
||||
pthread_mutex_unlock( &m_mutex );
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool CLogger::IsValidLogLevel( int logLevel )
|
||||
{
|
||||
return ( ( logLevel < 0 || logLevel > MAX_LOG_LEVEL ) ? false : true );
|
||||
}
|
||||
|
||||
bool CLogger::SetLogLevel( int logLevel )
|
||||
{
|
||||
if( IsValidLogLevel( logLevel ) == false )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
m_nLogLevel = logLevel;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool CLogger::MakeLogDir( string programName, string logDir )
|
||||
{
|
||||
string path = logDir + "/" + programName;
|
||||
|
||||
struct stat dirStat;
|
||||
|
||||
// 해당 이름을 가진 파일 또는 디렉토리가 존재하고
|
||||
if( lstat( path.c_str(), &dirStat ) == 0 )
|
||||
{
|
||||
// 해당 이름이 디렉토리인 경우
|
||||
if( S_ISDIR( dirStat.st_mode ) == true )
|
||||
{
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
cerr << "exist file with the same name as log path(" << path << ")." << endl;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
// 디렉토리 가 존재하지 않는 경우 디렉토리 생성 시도
|
||||
string cmd = "mkdir -p " + path;
|
||||
system( cmd.c_str() );
|
||||
|
||||
if( IsDirectory( path ) == false )
|
||||
{
|
||||
cerr << "can't make log directory. path=" << path << "." << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool CLogger::IsDirectory( string path )
|
||||
{
|
||||
struct stat dirStat;
|
||||
|
||||
if( lstat ( path.c_str(), &dirStat ) != 0 )
|
||||
{
|
||||
cerr << "Log path not valid. Check path [" << path << "][" << errno << "][" << strerror(errno) << "]" << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
// 해당 정보가 Directory 가 아닌 경우
|
||||
if( S_ISDIR( dirStat.st_mode ) == false )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void CLogger::Exit()
|
||||
{
|
||||
if( m_bIsInitialized == false )
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
pthread_mutex_lock( &m_mutex );
|
||||
|
||||
if( CLogger::m_pInstance != NULL )
|
||||
{
|
||||
delete CLogger::m_pInstance;
|
||||
CLogger::m_pInstance = NULL;
|
||||
m_bIsInitialized = false;
|
||||
}
|
||||
|
||||
pthread_mutex_unlock( &m_mutex );
|
||||
}
|
||||
|
||||
CLogger* CLogger::GetInstance()
|
||||
{
|
||||
return CLogger::m_pInstance;
|
||||
}
|
||||
|
||||
string CLogger::GetLogFilename( struct tm &timeNow )
|
||||
{
|
||||
char timeStr[256];
|
||||
snprintf( timeStr, (size_t)256, "%04d%02d%02d.log", timeNow.tm_year+1900, timeNow.tm_mon+1, timeNow.tm_mday);
|
||||
|
||||
// Make File Name
|
||||
string filename = m_szLogDir + "/" + m_szProgramName + "/" + m_szProgramName + "_" + string( timeStr );
|
||||
|
||||
return filename;
|
||||
}
|
||||
|
||||
bool CLogger::Write( int logLevel, const char * fmt, ...)
|
||||
{
|
||||
// 가변 인자 처리
|
||||
va_list args;
|
||||
char buffer[MAX_BUFFER_SIZE];
|
||||
va_start( args, fmt );
|
||||
if( vsnprintf( buffer, MAX_BUFFER_SIZE, fmt, args) < 0 )
|
||||
{
|
||||
va_end( args );
|
||||
return false;
|
||||
}
|
||||
va_end(args);
|
||||
|
||||
return Write( logLevel, PREFIX_DATE, NULL, 0, buffer );
|
||||
}
|
||||
|
||||
bool CLogger::WriteNoPrefix( int logLevel, const char * fmt, ...)
|
||||
{
|
||||
// 가변 인자 처리
|
||||
va_list args;
|
||||
char buffer[MAX_BUFFER_SIZE];
|
||||
va_start( args, fmt );
|
||||
if( vsnprintf( buffer, MAX_BUFFER_SIZE, fmt, args) < 0 )
|
||||
{
|
||||
va_end( args );
|
||||
return false;
|
||||
}
|
||||
va_end(args);
|
||||
|
||||
return Write( logLevel, PREFIX_NONE, NULL, 0, buffer );
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
// CHG 2012-08-16 huibong
|
||||
// 가변인자를 사용하는 Write 함수 다중 정의로 인해...
|
||||
// Complier 에서 인수 갯수 및 Type 이 동일할 경우 다른 함수를 가르키는 현상이 발견됨.
|
||||
// 이를 해결하기 위해 Write 함수에 대한 다중 정의를 제거토록 함수명을 명확하게 변경처리함.
|
||||
// 함수명 : Write -> WriteWithFunc 으로 변경 처리
|
||||
bool CLogger::WriteWithFunc( int logLevel, const char* filename, const char* funcname, int lineNum, const char * fmt, ...)
|
||||
{
|
||||
if( filename == NULL || funcname == NULL || lineNum < 0 || fmt == NULL )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
// 가변 인자 처리
|
||||
va_list args;
|
||||
char buffer[MAX_BUFFER_SIZE];
|
||||
va_start( args, fmt );
|
||||
if( vsnprintf( buffer, MAX_BUFFER_SIZE, fmt, args) < 0 )
|
||||
{
|
||||
va_end( args );
|
||||
return false;
|
||||
}
|
||||
va_end(args);
|
||||
|
||||
|
||||
char className[255];
|
||||
string funcPrefix = GetClassName( className, filename );
|
||||
funcPrefix += "::" + string( funcname ) + "()";
|
||||
|
||||
return Write( logLevel, PREFIX_FUNCTION, funcPrefix.c_str(), lineNum, buffer );
|
||||
}
|
||||
|
||||
const char* CLogger::GetClassName( char* className, const char* filename )
|
||||
{
|
||||
if( className == NULL )
|
||||
{
|
||||
cout << "[ERROR] The input argument 'className' is NULL." << endl;
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if( filename == NULL )
|
||||
{
|
||||
cout << "[ERROR] The input argument 'filename' is NULL." << endl;
|
||||
strcpy( className, "NULL" );
|
||||
return className;
|
||||
}
|
||||
|
||||
int filenameSize = strlen( filename );
|
||||
memcpy( className, filename, filenameSize );
|
||||
|
||||
// find end position
|
||||
char endCharacter = '.';
|
||||
int endPos = filenameSize - 1;
|
||||
for( ; endPos > 0; --endPos )
|
||||
{
|
||||
if( className[endPos] == endCharacter )
|
||||
{
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if( endPos == 0 )
|
||||
{
|
||||
cout << "[ERROR] endPos=0" << endl;
|
||||
strcpy( className, "NULL" );
|
||||
return className;
|
||||
}
|
||||
|
||||
className[endPos] = '\0';
|
||||
|
||||
// find start position
|
||||
char startCharacter = '/';
|
||||
int startPos = endPos - 1;
|
||||
for( ; startPos > 0; --startPos )
|
||||
{
|
||||
if( className[startPos] == startCharacter )
|
||||
{
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if( startPos != 0 )
|
||||
{
|
||||
startPos += 1;
|
||||
}
|
||||
|
||||
return (className + startPos );
|
||||
}
|
||||
|
||||
bool CLogger::Write( int logLevel, int logPrefix, const char* functionName, int lineNum, const char* log )
|
||||
{
|
||||
/// 유효한 로그 레벨이 아니면 로깅하지 않음.
|
||||
if( IsValidLogLevel( logLevel ) == false )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
// Log Level 검사 : 지정된 Level 이상인 경우 로깅하지 않음.
|
||||
if( logLevel > m_nLogLevel )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
if( log == NULL )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
// Get Current Data & Time
|
||||
time_t now = time( NULL );
|
||||
struct tm timeNow;
|
||||
localtime_r( &now, &timeNow );
|
||||
|
||||
string filename = GetLogFilename( timeNow );
|
||||
|
||||
// Log file open
|
||||
FILE* pFile = NULL;
|
||||
pFile = fopen( filename.c_str(), "a+" );
|
||||
|
||||
if( pFile == NULL )
|
||||
{
|
||||
cerr << "Log file open fail.[" << filename.c_str() << "][" << errno << "][" << strerror(errno) << "]" << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
// 시간 정보처리
|
||||
char timeStr[256];
|
||||
snprintf( timeStr, (size_t)256, "[%02d:%02d:%02d]"
|
||||
, timeNow.tm_hour, timeNow.tm_min, timeNow.tm_sec );
|
||||
|
||||
// Log Level 문자열 검색
|
||||
string szLevel = m_vectorLogLevelString[ logLevel ];
|
||||
|
||||
switch( logPrefix )
|
||||
{
|
||||
case PREFIX_DATE:
|
||||
/// 형식 예: [15:47:41] [DBG] sample log message.
|
||||
fprintf( pFile, "%s [%-4s] %s\n", timeStr, szLevel.c_str(), log );
|
||||
break;
|
||||
|
||||
case PREFIX_FUNCTION:
|
||||
if( functionName == NULL )
|
||||
{
|
||||
fclose( pFile );
|
||||
return false;
|
||||
}
|
||||
/// 형식 예: [15:47:41] [DBG] sample log message. [SomeClass::SomeFunction():12]
|
||||
fprintf( pFile, "%s [%-4s] %s [%s:%d]\n", timeStr, szLevel.c_str(), log, functionName, lineNum );
|
||||
break;
|
||||
|
||||
case PREFIX_NONE:
|
||||
/// 형식 예: sample log message.
|
||||
fprintf( pFile, "%s\n", log );
|
||||
break;
|
||||
}
|
||||
// Write to log file
|
||||
fflush( pFile );
|
||||
|
||||
// 종료 처리.
|
||||
fclose( pFile );
|
||||
return true;
|
||||
}
|
||||
|
||||
bool CLogger::WriteHex( int logLevel, const unsigned char* data, const int size )
|
||||
{
|
||||
/// 유효한 로그 레벨이 아니면 로깅하지 않음.
|
||||
if( IsValidLogLevel( logLevel ) == false )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
// Log Level 검사 : 지정된 Level 이상인 경우 로깅하지 않음.
|
||||
if( logLevel > m_nLogLevel )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
if( data == NULL )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
if( size <= 0 )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
// Get Current Data & Time
|
||||
time_t now = time( NULL );
|
||||
struct tm timeNow;
|
||||
localtime_r( &now, &timeNow );
|
||||
|
||||
string filename = GetLogFilename( timeNow );
|
||||
|
||||
// Log file open
|
||||
FILE* pFile = NULL;
|
||||
pFile = fopen( filename.c_str(), "a+" );
|
||||
|
||||
if( pFile == NULL )
|
||||
{
|
||||
cerr << "Log file open fail.[" << filename.c_str() << "][" << errno << "][" << strerror(errno) << "]" << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
for( int i = 0; i < size; ++i )
|
||||
{
|
||||
fprintf( pFile, "%02X", data[i] );
|
||||
|
||||
if( i == 0 )
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
if( ( (i+1) % 8 ) == 0 )
|
||||
{
|
||||
fprintf( pFile, " " );
|
||||
}
|
||||
|
||||
if( ( (i+1) % 16 ) == 0 )
|
||||
{
|
||||
fprintf( pFile, " " );
|
||||
}
|
||||
|
||||
if( ( (i+1) % 32 ) == 0 )
|
||||
{
|
||||
fprintf( pFile, "\n" );
|
||||
}
|
||||
}
|
||||
fprintf( pFile, "\n" );
|
||||
fflush( pFile );
|
||||
|
||||
// 종료 처리.
|
||||
fclose( pFile );
|
||||
|
||||
return false;
|
||||
}
|
||||
@@ -0,0 +1,273 @@
|
||||
/***************************************************************************
|
||||
Logger.h
|
||||
-----------------------------------------
|
||||
begin : 2011/10/26
|
||||
copyright : (C) 2005 SolutionBox Inc.
|
||||
author : Service 1 Team
|
||||
email : svc1@solbox.com
|
||||
version : 3.2.0.805
|
||||
|
||||
CopyRight(C) 2005 SolutionBox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of SolutionBox Inc.
|
||||
***************************************************************************/
|
||||
#ifndef __LOGGER_H__
|
||||
#define __LOGGER_H__
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <pthread.h>
|
||||
|
||||
///@brief 로그 레벨 정의
|
||||
#define LEMR 0 /* system is or will be unusable if situation is not resolved */
|
||||
#define LALT 1 /* immediate action required */
|
||||
#define LCRT 2 /* critical situations */
|
||||
#define LERR 3 /* error conditions */
|
||||
#define LWAR 4 /* recoverable errors */
|
||||
#define LNOT 5 /* unusual situation that merits investigation */
|
||||
#define LINF 6 /* information messages */
|
||||
#define LDBG 7 /* verbose data for debugging */
|
||||
#define LDEV 8 /* verbose data for developer */
|
||||
#define LDEV1 9 /* start or end the function of application level */
|
||||
#define LDEV2 10 /* start or end the function of application level */
|
||||
|
||||
#define MAX_LOG_LEVEL LDEV2
|
||||
|
||||
|
||||
///@brief Log 정보를 파일로 저장하기 위한 Class.
|
||||
///1. 가변 format 으로 전달된 로그 관련 정보를 Log Level 에 따라 로그 파일에 아래의 4가지 형식으로 저장한다.
|
||||
///
|
||||
/// 1.1 Prefix로 [hh:mm:ss]와 [ClassName::FunctionName]이 추가된 로그
|
||||
/// LOG( level, format, ... ) 매크로 사용.
|
||||
/// 예) [17:43:40] [DBG] log level debug [LoggerTestTestLogger]
|
||||
///
|
||||
/// 1.2 Prefix로 [hh:mm:ss]이 추가된 로그
|
||||
/// _LOG( level, format, ... ) 매크로 사용
|
||||
/// 예) [17:43:40] [EMR] log level emergency
|
||||
///
|
||||
/// 1.3 Prefix가 없는 로그
|
||||
/// _LOG_( level, format, ... ) 매크로 사용
|
||||
/// 예) log level emergency
|
||||
///
|
||||
/// 1.4 Hex 로그.
|
||||
/// LOG_HEX( level, data, size ) 매크로 사용
|
||||
/// 예) 00010203 04050607 08090A0B 0C0D0E0F
|
||||
///
|
||||
///2. Logger 객체 초기화시 전달된 Log Level 정보보다 전달받은 Log Level 정보가 큰 경우 해당 로그는 파일로 저장되지 않는다.
|
||||
///
|
||||
///3. 로그 파일은 매 일단위로 저장파일이 변경된다.
|
||||
///
|
||||
///4. 로그 저장을 위한 program 경로가 존재하지 않는 경우 자동 생성 처리된다.
|
||||
///
|
||||
///5. 로그 저장방식은 매 저장로그마다 open-close 로 처리된다.
|
||||
///
|
||||
///6. 파일로 기록시 Log Level 에 대한 정보도 함께 기록된다.
|
||||
///
|
||||
///7. 싱글톤으로 작성되었고, CLogger::Init(...)시에 쓰레드 안정성을 제공한다.
|
||||
///
|
||||
///8. 동적으로 로그 레벨을 변경할 수 있는 인터페이스를 제공한다.
|
||||
///
|
||||
class CLogger
|
||||
{
|
||||
// Attributes
|
||||
private:
|
||||
///@brief 싱글톤 객체 인트턴스.
|
||||
static CLogger* m_pInstance;
|
||||
|
||||
///@brief 싱글톤 객체 초기화 여부.
|
||||
static bool m_bIsInitialized;
|
||||
|
||||
///@brief 싱글톤 객체 초기화시 스레드 안정성을 위한 뮤텍스.
|
||||
static pthread_mutex_t m_mutex;
|
||||
|
||||
///@brief 프로그램 이름. 로그 파일 경로를 만들 때 사용.
|
||||
std::string m_szProgramName;
|
||||
|
||||
///@brief 공통 로그 디렉토리 이름. 로그 파일 경로를 만들 때 사용.
|
||||
std::string m_szLogDir;
|
||||
|
||||
///@brief 로그 레벨.
|
||||
int m_nLogLevel;
|
||||
|
||||
///@brief 로그 레벨에 대응되는 문자열 정보를 저장.
|
||||
std::vector<std::string> m_vectorLogLevelString;
|
||||
|
||||
protected:
|
||||
|
||||
public:
|
||||
///@brief 로그 형식을 지정.
|
||||
typedef enum
|
||||
{
|
||||
PREFIX_NONE = 0, /// 클래스 설명의 1.3에 해당
|
||||
PREFIX_DATE, /// 클래스 설명의 1.2에 해당
|
||||
PREFIX_FUNCTION, /// 클래스 설명의 1.1에 해당
|
||||
} LOG_PREFIX;
|
||||
|
||||
// Operations
|
||||
private:
|
||||
///@brief 생성자.
|
||||
/// 프로그램 이름, 공통 로그 디렉토리, 로그 레벨을 저장하고
|
||||
/// 로그 레벨에 대응되는 문자열 정보를 만든다.
|
||||
///@param programName [in] 프로그램 이름
|
||||
///@param logDir [in] 공통 로그 디렉토리 경로
|
||||
///@param logLevel [in] 로그 레벨
|
||||
CLogger( std::string programName, std::string logDir, int logLevel );
|
||||
|
||||
///@brief 소멸자.
|
||||
/// 로그 레벨에 대응되는 문자열 정보를 저장하고 있는
|
||||
/// 벡터 m_vectorLogLevelString을 clear 시킴.
|
||||
virtual ~CLogger();
|
||||
|
||||
///@brief 로그 레벨에 대응되는 문자열을 만든다.
|
||||
///@param none.
|
||||
///@return none.
|
||||
void MakeLogLevelString();
|
||||
|
||||
///@brief 로그 파일이 위치할 실제 로그 디렉토리를 생성한다.
|
||||
/// 생성할 디렉토리 경로는 'logDir/programName'이 된다.
|
||||
///@param programName [in] 프로그램 이름
|
||||
///@param logDir [in] 공통 로그 디렉토리 경로
|
||||
///@return 디렉토리가 이미 존재하거나 생성 성공하면 true,
|
||||
/// 해당 경로가 존재하지만 디렉토리가 아니거나, 디렉토리 생성 실패하면 false 반환.
|
||||
static bool MakeLogDir( std::string programName, std::string logDir );
|
||||
|
||||
///@brief 해당 경로가 디렉토리 인지 아닌지 판단.
|
||||
///@param path [in] 디렉토리 인지 아닌지 판단할 경로.
|
||||
///@return 해당 경로가 디렉토이면 true,
|
||||
/// 경로가 존재하지 않거나 디렉토리가 아니면 false 반환.
|
||||
static bool IsDirectory( std::string path );
|
||||
|
||||
///@brief 로그 레벨이 올바른지 판별.
|
||||
///@param logLevel [in]
|
||||
///@return 올바른 로그 레벨이면 true, 그렇지 않으면 false 반환.
|
||||
static bool IsValidLogLevel( int logLevel );
|
||||
|
||||
///@brief 시간 정보를 입력 받아 로그 파일 이름을 만든다.
|
||||
/// 로그 파일 이름 형식 : 프로그램명_YYYYMMDD.log
|
||||
///param timeNow [in] 현재 시간 정보.
|
||||
///return 로그 파일 이름.
|
||||
std::string GetLogFilename( struct tm &timeNow );
|
||||
|
||||
///@brief 로그를 남기는는 클래스가 정의된 파일의 이름에서 클래스명을 추출한다.
|
||||
/// 쓰레드 안정성을 보장한다.
|
||||
///@param className [out] 파일 이름에서 추출된 클래스명
|
||||
///@param filename [in] 파일 이름.
|
||||
///@return 클래스명 추출이 성공하면 클래스명 문자열의 포인터, 추출 실패하면 NULL.
|
||||
const char* GetClassName( char* className, const char* filename );
|
||||
|
||||
///@brief 인자 logPrefix에 따라 적절한 형식으로 로그 파일에 로그를 저장한다.
|
||||
/// 인자 logLevel이 설정된 로그 레벨보다 높으면 로그를 출력하지 않는다.
|
||||
///@param logLevel [in] 로그 레벨
|
||||
///@param logPrefix [in] 로그 프리픽스 종류.
|
||||
///@functionName [in] ClassName::FunctionNmae() 형식의 문자열.
|
||||
///@lineNum [in] 라인 번호.
|
||||
///@log [in] 출력하고자 하는 로그 내용.
|
||||
bool Write( int logLevel, int logPrefix, const char* functionName, int lineNum, const char* log );
|
||||
|
||||
protected:
|
||||
|
||||
public:
|
||||
///@brief 싱글톤 객체 m_pInstance를 생성하고 인자 정보로 로그 디렉토리를 만든다.
|
||||
///@param programName [in] 프로그램 이름
|
||||
///@param logDir [in] 공통 로그 디렉토리 경로
|
||||
///@param logLevel [in] 로그 레벨
|
||||
///@return 로그 디렉토리를 만들고 싱글톤 객체를 생성했으면 true,
|
||||
/// 로그 디렉토리를 만들지 못 했거나 인자 값이 올바르지 않으면 false 반환.
|
||||
static bool Init( std::string programName, std::string logDir, int logLevel );
|
||||
|
||||
///@brief 싱글톤 객체 m_pInstance를 delete 한다.
|
||||
///@param none.
|
||||
///@return none.
|
||||
static void Exit();
|
||||
|
||||
///@brief 싱글톤 객체 m_pInstance를 반환한다.
|
||||
///@param none.
|
||||
///@return CLogger 객체의 인스턴스.
|
||||
static CLogger* GetInstance();
|
||||
|
||||
///@brief Prefix로 [hh:mm:ss]이 추가된 형식으로 로그 저장.
|
||||
///@param logLevel [in] 로그 레벨
|
||||
///@param fmt [in] 로그 내용 포맷.
|
||||
///@param __VAR_ARGS__ [in] 가변 인자.
|
||||
///@return 로그 파일에 로그를 저장했으면 true, 그렇지 않으면 false 반환
|
||||
bool Write( int logLevel, const char * fmt, ...)
|
||||
__attribute__((format(printf, 3, 4)));
|
||||
|
||||
///@brief Prefix가 없는 로그 저장.
|
||||
///@param logLevel [in] 로그 레벨
|
||||
///@param fmt [in] 로그 내용 포맷.
|
||||
///@param __VAR_ARGS__ [in] 가변 인자.
|
||||
///@return 로그 파일에 로그를 저장했으면 true, 그렇지 않으면 false 반환
|
||||
bool WriteNoPrefix( int logLevel, const char * fmt, ...)
|
||||
__attribute__((format(printf, 3, 4)));
|
||||
|
||||
|
||||
// CHG 2012-08-16 huibong
|
||||
// 가변인자를 사용하는 Write 함수 다중 정의로 인해...
|
||||
// Complier 에서 인수 갯수 및 Type 이 동일할 경우 다른 함수를 가르키는 현상이 발견됨.
|
||||
// 이를 해결하기 위해 Write 함수에 대한 다중 정의를 제거토록 함수명을 명확하게 변경처리함.
|
||||
|
||||
///@brief Prefix로 [hh:mm:ss]와 [ClassName::FunctionName:line]이 추가된 형식으로 로그 저장.
|
||||
///@param logLevel [in] 로그 레벨
|
||||
///@param filename [in] 파일 이름
|
||||
///@param funcname [in] 함수 이름
|
||||
///@param lineNum [in] 라인 번호
|
||||
///@param fmt [in] 로그 내용 포맷.
|
||||
///@param __VAR_ARGS__ [in] 가변 인자.
|
||||
///@return 로그 파일에 로그를 저장했으면 true, 그렇지 않으면 false 반환
|
||||
bool WriteWithFunc( int logLevel, const char* filename, const char* funcname, int lineNum, const char * fmt, ...)
|
||||
__attribute__((format(printf, 6, 7)));
|
||||
|
||||
///@brief 로그를 hex 형식으로 저장.
|
||||
///@param logLevel [in] 로그 레벨
|
||||
///@param data [in] hex 형식으로 출력할 데이터.
|
||||
///@pram size [in] data의 크기.
|
||||
///@return 로그 파일에 로그를 저장했으면 true, 그렇지 않으면 false 반환
|
||||
bool WriteHex( int logLevel, const unsigned char* data, const int size );
|
||||
|
||||
///@brief 로그을 설정한다. 인자 logLevel이 적절한 값이면 새로운 값으로 변경하고
|
||||
/// 적절한 값이 아니면 로그 레벨을 변경하지 않는다.
|
||||
///@param logLevel [in] 설정할 로그 레벨
|
||||
///@return none.
|
||||
bool SetLogLevel( int logLevel );
|
||||
|
||||
inline int GetLogLevel() { return m_nLogLevel; };
|
||||
inline std::string GetLogDir() { return m_szLogDir + "/" + m_szProgramName; };
|
||||
};
|
||||
|
||||
#define LOG( level, format, ... ) \
|
||||
if( CLogger::GetInstance() != NULL ) \
|
||||
{ \
|
||||
CLogger::GetInstance()->WriteWithFunc( level, __FILE__, __FUNCTION__, __LINE__, format, ##__VA_ARGS__ ); \
|
||||
}
|
||||
|
||||
|
||||
#define _LOG( level, format, ... ) \
|
||||
if( CLogger::GetInstance() != NULL ) \
|
||||
{ \
|
||||
CLogger::GetInstance()->Write( level, format, ##__VA_ARGS__ ); \
|
||||
}
|
||||
|
||||
|
||||
#define _LOG_( level, format, ... ) \
|
||||
if( CLogger::GetInstance() != NULL ) \
|
||||
{ \
|
||||
CLogger::GetInstance()->WriteNoPrefix( level, format, ##__VA_ARGS__ ); \
|
||||
}
|
||||
|
||||
|
||||
#define _LOG_HEX_( level, data, size ) \
|
||||
if( CLogger::GetInstance() != NULL ) \
|
||||
{ \
|
||||
CLogger::GetInstance()->WriteHex( level, (const unsigned char*)data, size ); \
|
||||
}
|
||||
|
||||
|
||||
|
||||
#define FUNC_BEGIN() LOG( LDEV1, "begin" )
|
||||
#define FUNC_END() LOG( LDEV1, "end" )
|
||||
|
||||
#define FRM_BEGIN() LOG( LDEV2, "begin" )
|
||||
#define FRM_END() LOG( LDEV2, "end" )
|
||||
|
||||
#endif // __LOGGER_H__
|
||||
@@ -0,0 +1,71 @@
|
||||
#****************************************************************************
|
||||
# Makefile for Cloud Storage Common Libaray
|
||||
# -----------------------------------------
|
||||
#
|
||||
# begin : 2013/06/04
|
||||
# copyright : (C) 2013 Solbox Inc.
|
||||
# author : Development 1 Team
|
||||
# - 2013/06/04 - 1st dadamin
|
||||
# email : dev1@solbox.com
|
||||
# version : 3.3.0
|
||||
#
|
||||
# CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
# Redistribution and use in source and binary forms, with or with out
|
||||
# modification, are not permitted in outside of Solbox Inc.
|
||||
#*****************************************************************************
|
||||
|
||||
# Library info
|
||||
|
||||
LIB_NAME = InterCommon
|
||||
|
||||
LIB = lib$(LIB_NAME).a
|
||||
|
||||
OBJS = Config.o Logger.o BaseSocket.o ProcessRename.o
|
||||
|
||||
# Compiler info
|
||||
CC = /usr/bin/g++
|
||||
AR = /usr/bin/ar
|
||||
|
||||
DIR_INCLUDE = -I/usr/local/include
|
||||
|
||||
ifeq ($(DEBUG), yes)
|
||||
CFLAGS = -Wall -O0 -g -Wimplicit -Wreturn-type -Wunused -Wuninitialized\
|
||||
-Wparentheses -Wshadow -Wpointer-arith -Woverloaded-virtual\
|
||||
-D_REENTRANT -D_THREAD_SAFE -D_PTHREADS
|
||||
LFLAGS =
|
||||
DFLAGS =
|
||||
else
|
||||
CFLAGS = -Wall -O3 -g -Wimplicit -Wreturn-type -Wunused -Wuninitialized\
|
||||
-Wparentheses -Wshadow -Wpointer-arith -Woverloaded-virtual\
|
||||
-D_REENTRANT -D_THREAD_SAFE -D_PTHREADS
|
||||
LFLAGS =
|
||||
DFLAGS =
|
||||
endif
|
||||
|
||||
############################
|
||||
|
||||
all:$(LIB)
|
||||
sync
|
||||
|
||||
%.o: %.cpp
|
||||
$(CC) $(CFLAGS) -o $@ -c $^ $(DFLAGS) $(DIR_INCLUDE)
|
||||
|
||||
%.o: %.c
|
||||
$(CC) $(CFLAGS) -o $@ -c $^ $(DFLAGS) $(DIR_INCLUDE)
|
||||
|
||||
|
||||
$(LIB): $(OBJS)
|
||||
$(AR) crsv $@ $^
|
||||
|
||||
|
||||
clean:
|
||||
-rm -f *.o core *.out *.log
|
||||
-rm -f $(LIB)
|
||||
sync
|
||||
|
||||
|
||||
install :
|
||||
sync
|
||||
|
||||
|
||||
# End of Makefile
|
||||
@@ -0,0 +1,185 @@
|
||||
#include "ProcessRename.h"
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
|
||||
typedef size_t Size;
|
||||
#define LONG_ALIGN_MASK (sizeof(long) - 1)
|
||||
#define MEMSET_LOOP_LIMIT 1024
|
||||
#define MemSet(start, val, len) \
|
||||
do \
|
||||
{ \
|
||||
/* must be void* because we don't know if it is integer aligned yet */ \
|
||||
void *_vstart = (void *) (start); \
|
||||
int _val = (val); \
|
||||
Size _len = (len); \
|
||||
\
|
||||
if ((((long) _vstart) & LONG_ALIGN_MASK) == 0 && \
|
||||
(_len & LONG_ALIGN_MASK) == 0 && \
|
||||
_val == 0 && \
|
||||
_len <= MEMSET_LOOP_LIMIT && \
|
||||
/* \
|
||||
* If MEMSET_LOOP_LIMIT == 0, optimizer should find \
|
||||
* the whole "if" false at compile time. \
|
||||
*/ \
|
||||
MEMSET_LOOP_LIMIT != 0) \
|
||||
{ \
|
||||
long *_start = (long *) _vstart; \
|
||||
long *_stop = (long *) ((char *) _start + _len); \
|
||||
while (_start < _stop) \
|
||||
*_start++ = 0; \
|
||||
} \
|
||||
else \
|
||||
memset(_vstart, _val, _len); \
|
||||
} while (0)
|
||||
|
||||
#define PS_PADDING '\0'
|
||||
|
||||
extern char **environ;
|
||||
bool update_process_title = true;
|
||||
|
||||
static char *ps_buffer; /* will point to argv area */
|
||||
static size_t ps_buffer_size; /* space determined at run time */
|
||||
static size_t last_status_len; /* use to minimize length of clobber */
|
||||
static size_t ps_buffer_cur_len; /* nominal strlen(ps_buffer) */
|
||||
static size_t ps_buffer_fixed_size; /* size of the constant prefix */
|
||||
|
||||
static int save_argc;
|
||||
static char **save_argv;
|
||||
|
||||
size_t strlcpy(char *dst, const char *src, size_t siz)
|
||||
{
|
||||
char *d = dst;
|
||||
const char *s = src;
|
||||
size_t n = siz;
|
||||
|
||||
/* Copy as many bytes as will fit */
|
||||
if (n != 0)
|
||||
{
|
||||
while (--n != 0)
|
||||
{
|
||||
if ((*d++ = *s++) == '\0')
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
/* Not enough room in dst, add NUL and traverse rest of src */
|
||||
if (n == 0)
|
||||
{
|
||||
if (siz != 0)
|
||||
*d = '\0'; /* NUL-terminate dst */
|
||||
while (*s++)
|
||||
;
|
||||
}
|
||||
|
||||
return (s - src - 1); /* count does not include NUL */
|
||||
}
|
||||
|
||||
|
||||
void set_ps_display(const char *activity, bool force)
|
||||
{
|
||||
/* update_process_title=off disables updates, unless force = true */
|
||||
if (!force && !update_process_title)
|
||||
return;
|
||||
|
||||
/* no ps display for stand-alone backend */
|
||||
//if (!IsUnderPostmaster)
|
||||
// return;
|
||||
|
||||
/* If ps_buffer is a pointer, it might still be null */
|
||||
if (!ps_buffer)
|
||||
return;
|
||||
/* Update ps_buffer to contain both fixed part and activity */
|
||||
strlcpy(ps_buffer + ps_buffer_fixed_size, activity,
|
||||
ps_buffer_size - ps_buffer_fixed_size);
|
||||
ps_buffer_cur_len = strlen(ps_buffer);
|
||||
|
||||
/* pad unused memory; need only clobber remainder of old status string */
|
||||
if (last_status_len > ps_buffer_cur_len)
|
||||
MemSet(ps_buffer + ps_buffer_cur_len, PS_PADDING,
|
||||
last_status_len - ps_buffer_cur_len);
|
||||
last_status_len = ps_buffer_cur_len;
|
||||
|
||||
}
|
||||
|
||||
|
||||
char ** save_ps_display_args(int argc, char **argv)
|
||||
{
|
||||
save_argc = argc;
|
||||
save_argv = argv;
|
||||
|
||||
/*
|
||||
* If we're going to overwrite the argv area, count the available space.
|
||||
* Also move the environment to make additional room.
|
||||
*/
|
||||
{
|
||||
char *end_of_area = NULL;
|
||||
char **new_environ;
|
||||
int i;
|
||||
|
||||
/*
|
||||
* check for contiguous argv strings
|
||||
*/
|
||||
for (i = 0; i < argc; i++)
|
||||
{
|
||||
if (i == 0 || end_of_area + 1 == argv[i])
|
||||
end_of_area = argv[i] + strlen(argv[i]);
|
||||
}
|
||||
|
||||
if (end_of_area == NULL) /* probably can't happen? */
|
||||
{
|
||||
ps_buffer = NULL;
|
||||
ps_buffer_size = 0;
|
||||
return argv;
|
||||
}
|
||||
|
||||
/*
|
||||
* check for contiguous environ strings following argv
|
||||
*/
|
||||
for (i = 0; environ[i] != NULL; i++)
|
||||
{
|
||||
if (end_of_area + 1 == environ[i])
|
||||
end_of_area = environ[i] + strlen(environ[i]);
|
||||
}
|
||||
|
||||
ps_buffer = argv[0];
|
||||
last_status_len = ps_buffer_size = end_of_area - argv[0];
|
||||
|
||||
/*
|
||||
* move the environment out of the way
|
||||
*/
|
||||
new_environ = (char **) malloc((i + 1) * sizeof(char *));
|
||||
for (i = 0; environ[i] != NULL; i++)
|
||||
new_environ[i] = strdup(environ[i]);
|
||||
new_environ[i] = NULL;
|
||||
environ = new_environ;
|
||||
}
|
||||
|
||||
/*
|
||||
* If we're going to change the original argv[] then make a copy for
|
||||
* argument parsing purposes.
|
||||
*
|
||||
* (NB: do NOT think to remove the copying of argv[], even though
|
||||
* postmaster.c finishes looking at argv[] long before we ever consider
|
||||
* changing the ps display. On some platforms, getopt() keeps pointers
|
||||
* into the argv array, and will get horribly confused when it is
|
||||
* re-called to analyze a subprocess' argument string if the argv storage
|
||||
* has been clobbered meanwhile. Other platforms have other dependencies
|
||||
* on argv[].
|
||||
*/
|
||||
{
|
||||
char **new_argv;
|
||||
int i;
|
||||
|
||||
new_argv = (char **) malloc((argc + 1) * sizeof(char *));
|
||||
for (i = 0; i < argc; i++)
|
||||
new_argv[i] = strdup(argv[i]);
|
||||
new_argv[argc] = NULL;
|
||||
|
||||
argv = new_argv;
|
||||
}
|
||||
|
||||
return argv;
|
||||
|
||||
}
|
||||
|
||||
@@ -0,0 +1,37 @@
|
||||
/***************************************************************************
|
||||
Process rename functions
|
||||
-----------------------------------------
|
||||
begin : 2013/06/21
|
||||
copyright : (C) 2005 Solbox Inc.
|
||||
author : Dev 1 Team
|
||||
email : dev1@solbox.com
|
||||
version : 3.2.0
|
||||
|
||||
CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of Solbox Inc.
|
||||
*****************************************************************************/
|
||||
|
||||
#ifndef __PROCESS_RENAME_H__
|
||||
#define __PROCESS_RENAME_H__
|
||||
|
||||
#include <unistd.h>
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
/// @brief linux의 Process Title을 변경
|
||||
void set_ps_display(const char *activity, bool force);
|
||||
|
||||
/// @brief main에 argv의 실제 위치를 기억하고 새로은 메모리 활당하여 반환
|
||||
char ** save_ps_display_args(int argc, char **argv);
|
||||
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
#endif // __PROCESS_RENAME_H__
|
||||
@@ -0,0 +1,62 @@
|
||||
/***************************************************************************
|
||||
cc_collectd 통신 관련 protocol header
|
||||
-----------------------------------------
|
||||
begin : 2014/09/25
|
||||
copyright : (C) 2005 Solbox Inc.
|
||||
author : Storage Dev Team
|
||||
email : storage.sd@solbox.com
|
||||
version : 3.4
|
||||
|
||||
CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of Solbox Inc.
|
||||
***************************************************************************/
|
||||
|
||||
|
||||
#ifndef __CC_COLLECTD_PROTOCOL_H__
|
||||
#define __CC_COLLECTD_PROTOCOL_H__
|
||||
|
||||
|
||||
// cc_collectd 에서 사용할 통신 Header 구조체
|
||||
// - 계산상 크기는 16Byte 이지만...
|
||||
// - 통신상 sizeof() : 20Byte ( 64Bit OS )
|
||||
struct CcCollectdPacketHeader {
|
||||
|
||||
char stx; // Packet 유효성 관리 코드
|
||||
char type; // Request or Response 여부 ( 0x00: Request, 0x01: Response )
|
||||
char command[4]; // Command Code ( 4 Byte) : 0th client-cc_collectd 간 사용, 그외는 미사용.
|
||||
char result[4]; // Result Code ( 4 Byte )
|
||||
unsigned int data_length; // Packet Data 부분의 길이값 ( Network Byte Order 사용)
|
||||
char extend_code[2]; // 확장 및 Padding bits ( 2 Byte )
|
||||
};
|
||||
|
||||
// Packet Header stx 코드
|
||||
#define HEADER_STX_CODE 0x04
|
||||
|
||||
// Packet Header type 구분코드
|
||||
#define HEADER_TYPE_REQUEST 0x00
|
||||
#define HEADER_TYPE_RESPONSE 0x01
|
||||
|
||||
|
||||
// Packet command 공통.
|
||||
#define COMMON_ALIVE_CHECK 0x7F
|
||||
|
||||
// Packet command
|
||||
// client <-> cc_collect 통신 (command 0th byte 만 사용)
|
||||
#define REQUEST_NOT_VALID 0x00 // 유효하지 않은 요청.
|
||||
#define REQUEST_AVAILABLE_SIZE 0x01 // Client 에서 전송 가능한 Available Size 요청
|
||||
#define REQUEST_FILE_SAVE 0x02 // Client 에서 저장 목적의 File 전송
|
||||
#define REQUEST_FILE_CHECK 0x03 // Client 에서 File 존재 여부 확인 요청
|
||||
|
||||
|
||||
// Packet Result
|
||||
// => type이 Response 인 경우에만 세팅됨.( 0th Byte 만 사용시 )
|
||||
#define HEADER_RESULT_SUCCESS 0x00
|
||||
#define HEADER_RESULT_ERROR 0x01
|
||||
|
||||
|
||||
// 기타 정보
|
||||
#define LENGTH_FIELD_SIZE 4 // 가변데이터 형식 사용시 Length 필드의 메모리 크기 (unsigned int)
|
||||
|
||||
|
||||
#endif /* __CC_COLLECTD_PROTOCOL_H__ */
|
||||
@@ -0,0 +1,512 @@
|
||||
#include "ClientThread.h"
|
||||
|
||||
#include "Logger.h"
|
||||
#include "ServiceConfig.h"
|
||||
|
||||
#include <errno.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
|
||||
|
||||
// 생성자
|
||||
CClientThread::CClientThread( CStorageStatus * pStorage, CProcessStatus * pProcess, int clientSocket, char * pszClientAddress )
|
||||
: m_pStorageStatus( pStorage )
|
||||
, m_pProcessStatus( pProcess )
|
||||
, m_socketClientDesc( clientSocket )
|
||||
, m_client( clientSocket )
|
||||
, m_strClientIp( pszClientAddress )
|
||||
{
|
||||
// 멤버 변수 초기화
|
||||
m_threadHandle = 0;
|
||||
}
|
||||
|
||||
// 소멸자
|
||||
CClientThread:: ~CClientThread()
|
||||
{
|
||||
// Thread 동작 정지 처리
|
||||
// - 만약 Thread 가 이미 종료된 경우 m_threadHandle 이 다른 Thread Handle 일 수 있으므로 업무 Flow 수정시 주의할 것
|
||||
if( m_threadHandle != 0 )
|
||||
pthread_cancel( m_threadHandle );
|
||||
|
||||
// client 와 연결된 socket 종료 처리
|
||||
m_client.Close();
|
||||
}
|
||||
|
||||
// thread 를 생성하여 처리 업무 flow start
|
||||
bool CClientThread::Start()
|
||||
{
|
||||
int nRet = ::pthread_create( &m_threadHandle, NULL, CClientThread::threadFunc, this );
|
||||
if( nRet != 0 )
|
||||
{
|
||||
// Thread 생성 실패시
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Thread create failed.[%d][%s]", errorNum, strerror( errorNum ) );
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// 스레드 함수
|
||||
void* CClientThread::threadFunc( void* arg )
|
||||
{
|
||||
CClientThread* pObject = reinterpret_cast<CClientThread *>( arg );
|
||||
pthread_detach( pthread_self() );
|
||||
|
||||
// Client Thread Count 1 증가 처리
|
||||
pObject->m_pProcessStatus->PlusClientCount();
|
||||
|
||||
// 실제 작업 수행.
|
||||
pObject->Execute();
|
||||
|
||||
// Client Thread Count 1 감소 처리
|
||||
pObject->m_pProcessStatus->MinusClientCount();
|
||||
|
||||
// Thread 종료시 m_threadHandle 값을 초기화 처리.
|
||||
// - 소멸자에서 잘못된 Thread 를 종료할 수 있기 때문.
|
||||
pObject->m_threadHandle = 0;
|
||||
|
||||
|
||||
// 객체 자동 delete 처리.
|
||||
delete pObject;
|
||||
return NULL;
|
||||
}
|
||||
|
||||
// client request 요청을 수신하여 이에 따른 job 을 수행
|
||||
void CClientThread::Execute()
|
||||
{
|
||||
_LOG( LINF, "client thread start [%s]", m_strClientIp.c_str() );
|
||||
|
||||
// Packet Header 정보를 이용한 client 요청 분석
|
||||
int nRead;
|
||||
char command;
|
||||
|
||||
|
||||
while( 1 )
|
||||
{
|
||||
// socket 연결을 아래 job 등에서 일부로 끊은 경우를 처리하기 위한 코드.
|
||||
// - Server socket 이지만.. 통신 관련 오류시.. server 가 socket 을 끊을 수 있다.
|
||||
// - 이는 file 전송 등으로 인해 불필요한 data 가 통신상에 남아 있을 수 있기 때문임.
|
||||
if( m_client.IsValidSocket() == false )
|
||||
break;
|
||||
|
||||
// Header 수신
|
||||
nRead = m_client.GetHeaderInfo( command );
|
||||
|
||||
if( nRead == 0 )
|
||||
{
|
||||
// timeout 발생시 - 잠시 대기 후 계속 수신 대기
|
||||
sleep( 2 );
|
||||
continue;
|
||||
}
|
||||
else if( nRead < 0 )
|
||||
{
|
||||
// Socket 접속 종료 또는 오류 발생시
|
||||
// - 관련 로그는 해당 모듈에서 다 찍었으므로.. 여기서는 루프 종료 처리.
|
||||
break;
|
||||
}
|
||||
else
|
||||
{
|
||||
// data 수신시.
|
||||
// 정의된 packet 에만 동작.. 그외에는 무시.
|
||||
|
||||
// Client 요청인 경우.
|
||||
switch( command )
|
||||
{
|
||||
case REQUEST_AVAILABLE_SIZE: // client 로 부터 available size 요청 수신시
|
||||
_LOG( LINF, "client[%s] request available size.", m_strClientIp.c_str() );
|
||||
ProcessAvailableSize();
|
||||
break;
|
||||
|
||||
case REQUEST_FILE_SAVE: // client 로 부터 파일 저장 요청 수신시.
|
||||
_LOG( LINF, "client[%s] request file save.", m_strClientIp.c_str() );
|
||||
ProcessFileSave();
|
||||
break;
|
||||
|
||||
// NEW 2015-10-07 huibong
|
||||
// - 파일 존재 여부 확인 요청 관련 기능 추가 (#24683)
|
||||
case REQUEST_FILE_CHECK: // client 로 부터 File 존재 여부 확인 요청 수신시
|
||||
_LOG( LINF, "client[%s] request file check.", m_strClientIp.c_str() );
|
||||
ProcessFileCheck();
|
||||
break;
|
||||
|
||||
|
||||
default: // 미정의 Type 또는 잘못된 reponse 요청인 경우 ... 연결 해제 처리
|
||||
_LOG( LINF, "client[%s] request unknown. check request[%02x].", m_strClientIp.c_str(), command );
|
||||
m_client.PrintHeaderToLog();
|
||||
m_client.Close();
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
} // while (1)
|
||||
|
||||
|
||||
// client 와 연결 종료 처리.
|
||||
_LOG( LINF, "client connection closed.[%s]", m_strClientIp.c_str() );
|
||||
m_client.Close();
|
||||
}
|
||||
|
||||
// Client 로 부터 available size 요청 수신시 이를 처리하기 위한 함수.
|
||||
void CClientThread::ProcessAvailableSize()
|
||||
{
|
||||
// REQUEST_AVAILABLE_SIZE 요청인 경우..
|
||||
// Data 부분이 따로 존재하지 않으므로...
|
||||
// 바로 응답을 전송하면 된다.
|
||||
unsigned long long uAvailableSize;
|
||||
uAvailableSize = m_pStorageStatus->GetAvailableSize();
|
||||
|
||||
if( m_client.SendResultAvailableSizeToClient( uAvailableSize ) == false )
|
||||
{
|
||||
LOG( LERR, "client[%s] available size result[%llu] send fail."
|
||||
, m_strClientIp.c_str(), uAvailableSize );
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
// client 로 부터 파일 저장 요청 수신시 이를 처리하기 위한 함수.
|
||||
void CClientThread::ProcessFileSave()
|
||||
{
|
||||
std::string strErrorMessage;
|
||||
|
||||
// REQUEST_FILE_SAVE 요청인 경우...
|
||||
// Data 부준이 존재하므로...
|
||||
// Data 부분을 수신 처리한다.
|
||||
std::string strHostName;
|
||||
std::string strProcessName;
|
||||
std::string strFileName;
|
||||
unsigned long long uFileSize;
|
||||
|
||||
// Data Body 수신 처리
|
||||
if( m_client.RecvBodyFileSaveFromClient( strHostName, strProcessName, strFileName, uFileSize ) == false )
|
||||
{
|
||||
// Body Data 수신 실패시
|
||||
LOG( LERR, "client[%s] request file save. but data body receive fail.", m_strClientIp.c_str() );
|
||||
m_client.Close();
|
||||
return;
|
||||
}
|
||||
|
||||
_LOG( LDBG, "client[%s] request file save.[%s][%s][%s][%llu]"
|
||||
, m_strClientIp.c_str(), strHostName.c_str(), strProcessName.c_str(), strFileName.c_str(), uFileSize );
|
||||
|
||||
// 수신된 Data 검증
|
||||
// - strHostName, strProcessName 정보는 없을 수 있으나...
|
||||
// - strFileName 정보는 꼭 있어야 함.
|
||||
if( strFileName.empty() == true )
|
||||
{
|
||||
strErrorMessage = "file name empty. check file name.";
|
||||
LOG( LERR, "client[%s] request file save. but filename empty.", m_strClientIp.c_str() );
|
||||
|
||||
if( m_client.SendResultFileSaveToClient( false, strErrorMessage ) == false )
|
||||
{
|
||||
LOG( LERR, "client[%s] file save result[error] send fail.", m_strClientIp.c_str() );
|
||||
}
|
||||
|
||||
// 잠시 대기 후 세션 종료 처리
|
||||
sleep( 1 );
|
||||
m_client.Close();
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
// 파일 저장을 위한 폴더 생성 처리.
|
||||
std::string strFullPathName;
|
||||
strFullPathName = CServiceConfig::GetInstance()->GetSaveRootPath();
|
||||
|
||||
|
||||
// CHG 2015-09-16 huibong
|
||||
// - #24436 에 의해 저장 경로 방식 변경
|
||||
// - 기존 : hostname - process name - 파일
|
||||
// - 변경 : process name - hostname - 파일
|
||||
if( strProcessName.empty() == false )
|
||||
{
|
||||
strFullPathName.append( "/" );
|
||||
strFullPathName.append( strProcessName );
|
||||
|
||||
if( MakeDirectory( strFullPathName ) == false )
|
||||
{
|
||||
strErrorMessage = "process name directory create fail.[" + strFullPathName + "]";
|
||||
LOG( LERR, "client[%s] process name directory create fail [%s]", m_strClientIp.c_str(), strFullPathName.c_str() );
|
||||
|
||||
if( m_client.SendResultFileSaveToClient( false, strErrorMessage ) == false )
|
||||
{
|
||||
LOG( LERR, "client[%s] file save result[error] send fail.", m_strClientIp.c_str() );
|
||||
}
|
||||
|
||||
// 잠시 대기 후 세션 종료 처리
|
||||
sleep( 1 );
|
||||
m_client.Close();
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
if( strHostName.empty() == false )
|
||||
{
|
||||
strFullPathName.append( "/" );
|
||||
strFullPathName.append( strHostName );
|
||||
|
||||
if( MakeDirectory( strFullPathName ) == false )
|
||||
{
|
||||
strErrorMessage = "hostname directory create fail.[" + strFullPathName + "]" ;
|
||||
LOG( LERR, "client[%s] hostname directory create fail [%s]", m_strClientIp.c_str(), strFullPathName.c_str() );
|
||||
|
||||
if( m_client.SendResultFileSaveToClient( false, strErrorMessage ) == false )
|
||||
{
|
||||
LOG( LERR, "client[%s] file save result[error] send fail.", m_strClientIp.c_str() );
|
||||
}
|
||||
|
||||
// 잠시 대기 후 세션 종료 처리
|
||||
sleep( 1 );
|
||||
m_client.Close();
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// 파일 open 후 저장 처리.
|
||||
strFullPathName.append( "/" );
|
||||
strFullPathName.append( strFileName );
|
||||
|
||||
FILE * pFile = NULL;
|
||||
pFile = fopen( strFullPathName.c_str(), "w" );
|
||||
if( pFile == NULL )
|
||||
{
|
||||
// File open 실패시...
|
||||
int errorNum = errno;
|
||||
char tempBuffer[1024];
|
||||
snprintf( tempBuffer, sizeof( tempBuffer ) - 1, "file open failed to write.[%s] [%d][%s]", strFullPathName.c_str(), errorNum, strerror( errorNum ) );
|
||||
strErrorMessage.append( tempBuffer );
|
||||
|
||||
LOG( LERR, "%s", strErrorMessage.c_str() );
|
||||
|
||||
// client 로 오류 전송 처리.
|
||||
m_client.SendResultFileSaveToClient( false, strErrorMessage );
|
||||
|
||||
// 잠시 대기 후 세션 종료 처리
|
||||
sleep( 1 );
|
||||
m_client.Close();
|
||||
return;
|
||||
}
|
||||
|
||||
// 파일 수신 및 저장 처리
|
||||
if( m_client.ReceiveToFile( pFile, uFileSize, strErrorMessage ) == false )
|
||||
{
|
||||
// 수신 실패시...
|
||||
// - 파일 삭제
|
||||
// - socket 종료 처리
|
||||
fclose( pFile );
|
||||
|
||||
// client 로 오류 전송 처리.
|
||||
m_client.SendResultFileSaveToClient( false, strErrorMessage );
|
||||
|
||||
LOG( LERR, "client[%s] file save fail [%s].", m_strClientIp.c_str(), strErrorMessage.c_str() );
|
||||
|
||||
// 잠시 대기 후 세션 종료 처리
|
||||
sleep( 1 );
|
||||
unlink( strFullPathName.c_str() );
|
||||
m_client.Close();
|
||||
}
|
||||
else
|
||||
{
|
||||
// 파일 수신 처리 성공시..
|
||||
fclose( pFile );
|
||||
|
||||
_LOG( LINF, "client[%s] file save OK [%s][%llu]", m_strClientIp.c_str(), strFullPathName.c_str(), uFileSize );
|
||||
|
||||
// 성공 처리 응답.
|
||||
strErrorMessage.clear();
|
||||
if( m_client.SendResultFileSaveToClient( true, strErrorMessage ) == false )
|
||||
{
|
||||
LOG( LERR, "client[%s] file save result[OK] send fail.", m_strClientIp.c_str() );
|
||||
}
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
// 전달받은 Directory 정보를 확인하여 존재하지 않을 경우 생성처리.
|
||||
bool CClientThread::MakeDirectory( const std::string& szFullPath )
|
||||
{
|
||||
if( szFullPath.size() <= 0 )
|
||||
return false;
|
||||
|
||||
struct stat dirStat;
|
||||
|
||||
// 디렉토리 정보 검사.
|
||||
// 해당 이름을 가진 Directory 또는 파일이 존재하고
|
||||
if( lstat( szFullPath.c_str(), &dirStat ) == 0 )
|
||||
{
|
||||
// 해당 정보가 Directory 인 경우
|
||||
if( S_ISDIR( dirStat.st_mode ) != 0 )
|
||||
return true;
|
||||
}
|
||||
|
||||
// 해당 디렉토리가 존재하지 않는 경우 무조건 생성 처리.
|
||||
//char tempBuffer[1024];
|
||||
//sprintf( tempBuffer, "/bin/mkdir -p -m0755 %s", szFullPath.c_str() );
|
||||
//system( tempBuffer );
|
||||
|
||||
// CHG 2014-11-04 huibong
|
||||
// - system() 함수 대신 mkdir() function use
|
||||
// - 오류 발생 관련 로그 처리 기능 보강
|
||||
if( mkdir( szFullPath.c_str(), 0755) == 0 )
|
||||
{
|
||||
_LOG( LINF, "Directory create OK. [%s]", szFullPath.c_str());
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
int errorNum = errno;
|
||||
|
||||
LOG( LERR, "Directory create faild. [%s][%d][%s]", szFullPath.c_str(), errorNum, strerror(errorNum));
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// client 로 부터 파일 존재 여부 확인 요청 수신시 이를 처리하기 위한 함수.
|
||||
void CClientThread::ProcessFileCheck()
|
||||
{
|
||||
std::string strErrorMessage;
|
||||
|
||||
// REQUEST_FILE_CHECK 요청인 경우...
|
||||
// Data 부준이 존재하므로...
|
||||
// Data 부분을 수신 처리한다.
|
||||
std::string strHostName;
|
||||
std::string strProcessName;
|
||||
std::string strFileName;
|
||||
|
||||
// 로컬 파일의 크기 정보를 저장하기 위한 변수.
|
||||
unsigned long long uFileSize = 0;
|
||||
|
||||
// Data Body 수신 처리
|
||||
if( m_client.RecvBodyFileCheckFromClient( strHostName, strProcessName, strFileName ) == false )
|
||||
{
|
||||
// Body Data 수신 실패시
|
||||
LOG( LERR, "client[%s] request file check. but data body receive fail.", m_strClientIp.c_str() );
|
||||
m_client.Close();
|
||||
return;
|
||||
}
|
||||
|
||||
_LOG( LDBG, "client[%s] request file check.[%s][%s][%s]"
|
||||
, m_strClientIp.c_str(), strHostName.c_str(), strProcessName.c_str(), strFileName.c_str() );
|
||||
|
||||
// 수신된 Data 검증
|
||||
// - strHostName, strProcessName 정보는 없을 수 있으나...
|
||||
// - strFileName 정보는 꼭 있어야 함.
|
||||
if( strFileName.empty() == true )
|
||||
{
|
||||
strErrorMessage = "file name empty. check file name.";
|
||||
LOG( LERR, "client[%s] request file check. but filename empty.", m_strClientIp.c_str() );
|
||||
|
||||
if( m_client.SendResultFileCheckErrorToClient( strErrorMessage ) == false )
|
||||
{
|
||||
LOG( LERR, "client[%s] file check result[error] send fail.", m_strClientIp.c_str() );
|
||||
}
|
||||
|
||||
// 잠시 대기 후 세션 종료 처리
|
||||
sleep( 1 );
|
||||
m_client.Close();
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
// 물리 파일 존재 여부 검증
|
||||
std::string strFullPathName;
|
||||
strFullPathName = CServiceConfig::GetInstance()->GetSaveRootPath();
|
||||
|
||||
// CHG 2015-09-16 huibong
|
||||
// - #24436 에 의해 저장 경로 방식 변경
|
||||
// - 기존 : hostname - process name - 파일
|
||||
// - 변경 : process name - hostname - 파일
|
||||
if( strProcessName.empty() == false )
|
||||
{
|
||||
strFullPathName.append( "/" );
|
||||
strFullPathName.append( strProcessName );
|
||||
}
|
||||
|
||||
if( strHostName.empty() == false )
|
||||
{
|
||||
strFullPathName.append( "/" );
|
||||
strFullPathName.append( strHostName );
|
||||
}
|
||||
|
||||
// 파일 최종 경로 조합
|
||||
strFullPathName.append( "/" );
|
||||
strFullPathName.append( strFileName );
|
||||
|
||||
// 해당 물리 파일 존재 여부 검증.
|
||||
// - 존재할 경우... 파일 size 정보 추출
|
||||
// - 존재하지 않을 경우.. 파일 size 정보를 0 으로 설정 처리.
|
||||
|
||||
struct stat st;
|
||||
if( stat( strFullPathName.c_str(), &st ) == 0 )
|
||||
{
|
||||
// 해당 파일이 정규 파일인 경우.. ( 폴더, 링크 등은 제외 처리 )
|
||||
if( S_ISREG( st.st_mode ) != 0 )
|
||||
{
|
||||
// size 정보 추출
|
||||
uFileSize = st.st_size;
|
||||
}
|
||||
else
|
||||
{
|
||||
// 해당 경로명 정보가 파일이 아닌 경우....
|
||||
char tempBuffer[1024];
|
||||
snprintf( tempBuffer, sizeof( tempBuffer ) - 1, "file check error.[%s][Not file. directory or link.]", strFullPathName.c_str() );
|
||||
strErrorMessage.append( tempBuffer );
|
||||
|
||||
LOG( LERR, "client[%s] request file check error.[%s][Not file. directory or link]", m_strClientIp.c_str(), strFullPathName.c_str());
|
||||
|
||||
if( m_client.SendResultFileCheckErrorToClient( strErrorMessage ) == false )
|
||||
{
|
||||
LOG( LERR, "client[%s] file check result[error] send fail.", m_strClientIp.c_str() );
|
||||
}
|
||||
return;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
int errorNum = errno;
|
||||
if( errorNum == ENOENT )
|
||||
{
|
||||
// 해당 파일이 존재하지 않는 경우....
|
||||
// - size 정보를 0 으로 설정 처리
|
||||
uFileSize = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
// 그외 오류 발생시.
|
||||
strErrorMessage.clear();
|
||||
|
||||
char tempBuffer[1024];
|
||||
snprintf( tempBuffer, sizeof( tempBuffer ) - 1, "file check error.[%s] [%d][%s]", strFullPathName.c_str(), errorNum, strerror( errorNum ) );
|
||||
strErrorMessage.append( tempBuffer );
|
||||
|
||||
LOG( LERR, "client[%s] request file check error.[%s][%d][%s]", m_strClientIp.c_str(), strFullPathName.c_str(), errorNum, strerror( errorNum ) );
|
||||
|
||||
if( m_client.SendResultFileCheckErrorToClient( strErrorMessage ) == false )
|
||||
{
|
||||
LOG( LERR, "client[%s] file check result[error] send fail.", m_strClientIp.c_str() );
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
// 파일 size 정보가 정상적으로 추출된 경우...
|
||||
// 해당 정보를 client 에게 전송 처리한다.
|
||||
if( m_client.SendResultFileCheckToClient( uFileSize ) == false )
|
||||
{
|
||||
LOG( LERR, "client[%s] file check result[size: %llu] send fail.", m_strClientIp.c_str(), uFileSize );
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
@@ -0,0 +1,93 @@
|
||||
/***************************************************************************
|
||||
접속 요청 Client 에 대한 처리를 담당하는 Thread class
|
||||
-----------------------------------------
|
||||
begin : 2014/09/24
|
||||
copyright : (C) 2005 Solbox Inc.
|
||||
author : storage dev team
|
||||
email : storage.sd@solbox.com
|
||||
version : 3.4
|
||||
|
||||
CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of Solbox Inc.
|
||||
***************************************************************************/
|
||||
|
||||
#ifndef __CLIENT_THREAD_H__
|
||||
#define __CLIENT_THREAD_H__
|
||||
|
||||
|
||||
#include <pthread.h>
|
||||
#include <string>
|
||||
|
||||
#include "StorageStatus.h"
|
||||
#include "ProcessStatus.h"
|
||||
#include "ProcessSocket.h"
|
||||
|
||||
|
||||
// CClientThread
|
||||
// rc_syncd 에서 accept 처리한 각 client 세션을 처리하기 위한 Thread class
|
||||
class CClientThread
|
||||
{
|
||||
public:
|
||||
|
||||
// 생성자
|
||||
CClientThread( CStorageStatus * pStorage, CProcessStatus * pProcess, int clientSocket, char * pszClientAddress );
|
||||
|
||||
// 소멸자
|
||||
~CClientThread();
|
||||
|
||||
// thread 를 생성하여 처리 업무 flow start
|
||||
bool Start();
|
||||
|
||||
|
||||
|
||||
private:
|
||||
|
||||
// 스레드 함수
|
||||
static void* threadFunc( void* arg );
|
||||
|
||||
// client request 요청을 수신하여 이에 따른 job 을 수행
|
||||
void Execute();
|
||||
|
||||
// Client 로 부터 available size 요청 수신시 이를 처리하기 위한 함수.
|
||||
void ProcessAvailableSize();
|
||||
|
||||
// client 로 부터 파일 저장 요청 수신시 이를 처리하기 위한 함수.
|
||||
void ProcessFileSave();
|
||||
|
||||
// 전달받은 Directory 정보를 확인하여 존재하지 않을 경우 생성처리.
|
||||
bool MakeDirectory( const std::string& szFullPath );
|
||||
|
||||
// client 로 부터 파일 존재 여부 확인 요청 수신시 이를 처리하기 위한 함수.
|
||||
void ProcessFileCheck();
|
||||
|
||||
|
||||
private:
|
||||
|
||||
// 쓰레드 핸들
|
||||
pthread_t m_threadHandle;
|
||||
|
||||
// Storage 상태정보 추출 관련 CStorageStatus 객체에 대한 포인터
|
||||
// - 생성자를 통해 전달받음
|
||||
CStorageStatus * m_pStorageStatus;
|
||||
|
||||
// 프로세스 상태를 관리하는 CProcessStatus 객체에 대한 포인터
|
||||
// - 생성자를 통해 전달받음
|
||||
CProcessStatus * m_pProcessStatus;
|
||||
|
||||
// Client Socket discriptor
|
||||
// - 생성자를 통해 전달받음.
|
||||
int m_socketClientDesc;
|
||||
|
||||
// client 와 통신 관련 처리를 수행하기 위한 socket 객체
|
||||
CProcessSocket m_client;
|
||||
|
||||
// Client IP Address
|
||||
// - 생성자를 통해 전달받음
|
||||
std::string m_strClientIp;
|
||||
|
||||
|
||||
};
|
||||
|
||||
#endif /* __CLIENT_THREAD_H__ */
|
||||
|
||||
@@ -0,0 +1,571 @@
|
||||
#include "Main.h"
|
||||
|
||||
#include "ProcessRename.h"
|
||||
#include "ServiceConfig.h"
|
||||
#include "Logger.h"
|
||||
#include "Worker.h"
|
||||
|
||||
#include <unistd.h>
|
||||
#include <stdio.h>
|
||||
#include <errno.h>
|
||||
#include <signal.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
#include <sys/wait.h>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include <time.h>
|
||||
|
||||
|
||||
// Accept 대기 Count 수
|
||||
#define DEFAULT_ACCEPT_WAIT_COUNT 50
|
||||
|
||||
// 전역 객체 정보
|
||||
// Data 송수신을 담당할 TCP Listen Socket
|
||||
int g_listenSocket = -1;
|
||||
|
||||
|
||||
// 사용방법 표시
|
||||
void PrintUsage()
|
||||
{
|
||||
fprintf(stderr, "\n");
|
||||
fprintf(stderr, "Usage: " PROG_NAME " [-h] [-v] [-c {file}] \n");
|
||||
fprintf(stderr, "Options: \n");
|
||||
fprintf(stderr, " -h Display help information \n");
|
||||
fprintf(stderr, " -v Display version \n");
|
||||
fprintf(stderr, " -c {file} Use {file} as config file \n");
|
||||
|
||||
fprintf(stderr, "\n");
|
||||
fprintf( stderr, PROG_NAME " is Solbox Cloud Storage module.\n" );
|
||||
fprintf( stderr, " - File(log, dump, ...) receive from FHS, RCTS, GTS, RCDB ... \n" );
|
||||
fprintf( stderr, " - Save file order process name, hostname.\n" );
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
// 버전 정보 표시
|
||||
void PrintVersion()
|
||||
{
|
||||
fprintf(stderr, "\n");
|
||||
fprintf(stderr, PROG_NAME " version: " PROG_VERSION "\n\n");
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
int main( int argc, char * argv[] )
|
||||
{
|
||||
// 설정 파일을 저장할 변수.
|
||||
std::string strConfigFileName = DEFAULT_CONFIG_FILE;
|
||||
|
||||
// linux 에서 프로세스를 Titile 변경을 지원을 위해서 argv 메모리에 저장
|
||||
argv = save_ps_display_args(argc, argv);
|
||||
|
||||
// 전달 받은 옵션 여부 확인 및 처리
|
||||
if( argc >= 2)
|
||||
{
|
||||
int opt;
|
||||
while((opt = getopt(argc, argv, "hvc:")) != -1) // 옵션 끝까지 파싱처리
|
||||
{
|
||||
switch(opt)
|
||||
{
|
||||
case 'h':
|
||||
case '?': // 정의되지 않은 문자가 나타날 경우 getopt 에 자동반환. 도움말 표시 처리.
|
||||
PrintUsage();
|
||||
return EXIT_SUCCESS;
|
||||
|
||||
case 'v':
|
||||
PrintVersion();
|
||||
return EXIT_SUCCESS;
|
||||
|
||||
case 'c':
|
||||
strConfigFileName = optarg;
|
||||
fprintf(stderr, PROG_NAME " start with user define conf[%s]\n", strConfigFileName.c_str());
|
||||
break;
|
||||
|
||||
default: // 본 조건절은 getopt 특성상 동작하지 않지만...업무 Flow 이해(?)를 위해 유지한다.
|
||||
fprintf(stderr, "Unkonwn options[%c] used. program terminated.\n", opt);
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 프로그램 중복 실행 체크
|
||||
if( IsCurrentProcessRun() == true )
|
||||
{
|
||||
fprintf(stderr, "[warning] Process[" PROG_NAME "] is already running...\n");
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
// conf 로딩 및 검사
|
||||
std::string szErrorMessage;
|
||||
if( CServiceConfig::Init( PROG_NAME, strConfigFileName, szErrorMessage ) == false)
|
||||
{
|
||||
fprintf(stderr, "[ERR] %s\n\n", szErrorMessage.c_str());
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
// cc_collectd 전용 기능
|
||||
// - conf 상에 설정된 최상위 저장 경로 정보가 실제 시스템이 존재하는지 검사한다.
|
||||
if( CheckPath( CServiceConfig::GetInstance()->GetSaveRootPath() ) == false )
|
||||
{
|
||||
fprintf( stderr, "[ERR] SAVE_ROOT_PATH[%s] not found. Check & Retry.\n\n", CServiceConfig::GetInstance()->GetSaveRootPath() );
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
// Log 객체 생성 및 초기화
|
||||
if( CLogger::Init( PROG_NAME, CServiceConfig::GetInstance()->GetLogPath(), LINF ) == false)
|
||||
{
|
||||
fprintf(stderr, "[ERR] Log module initialize failed.[%s]\n\n", CServiceConfig::GetInstance()->GetLogPath());
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
// Daemonize...
|
||||
// - nochdir: true(작업 디렉토리 변경 안함)
|
||||
// - noclose: false (표준 입출력, 에러를 /dev/null 로 리디렉트 처리)
|
||||
if( daemon(1, 0) == -1)
|
||||
{
|
||||
fprintf(stderr, "[ERR] Process daemonize failed.[%s]\n\n", strerror(errno) );
|
||||
|
||||
ReadyToExit();
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
// Signal 처리기 설정
|
||||
SetSignalMain();
|
||||
|
||||
// Process 기동 관련 정보 로깅
|
||||
_LOG( LINF, "***********************************************************");
|
||||
_LOG( LINF, " %s Start. Version: %s", PROG_NAME, PROG_VERSION);
|
||||
_LOG( LINF, "***********************************************************");
|
||||
_LOG( LINF, "Config : %s", strConfigFileName.c_str());
|
||||
_LOG( LINF, "Log : %s/%s", CServiceConfig::GetInstance()->GetLogPath(), PROG_NAME);
|
||||
_LOG( LINF, "Log level : %d", CServiceConfig::GetInstance()->GetLogLevel());
|
||||
_LOG( LINF, "TCP listen port : %u", CServiceConfig::GetInstance()->GetTcpListenPort() );
|
||||
_LOG( LINF, "Save root path : %s", CServiceConfig::GetInstance()->GetSaveRootPath() );
|
||||
_LOG( LINF, "Worker process count : %u", CServiceConfig::GetInstance()->GetWorkerProcessCount() );
|
||||
_LOG( LINF, "***********************************************************");
|
||||
|
||||
|
||||
// Log Level 재설정. -> conf 설정대로 변경 처리.
|
||||
#ifdef _DEBUG_
|
||||
CLogger::GetInstance()->SetLogLevel(LDBG);
|
||||
#else
|
||||
CLogger::GetInstance()->SetLogLevel(CServiceConfig::GetInstance()->GetLogLevel());
|
||||
#endif
|
||||
|
||||
|
||||
// Data 송수신 처리 관련 Listen Socket 생성
|
||||
if( MakeListenSocket( g_listenSocket ) == false )
|
||||
{
|
||||
LOG( LERR, "Listen socket create failed. => process exit." );
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
// Worker Process 생성
|
||||
if( MakeProcessWorker( g_listenSocket ) == false )
|
||||
{
|
||||
LOG( LERR, "Worker process make failed. => process exit.");
|
||||
|
||||
ReadyToExit();
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
// Process Rename
|
||||
set_ps_display(PROG_NAME": Main [monitor worker process]", false);
|
||||
|
||||
// Main Process : 그냥 대기
|
||||
while( 1 )
|
||||
{
|
||||
// Main Process 는 할 일이 없다.
|
||||
// => Worker 프로세스 종료시 SIGCHLD 신호로 인해 신호처리기에서 Worker Process 재성성 수행함.
|
||||
// => 따라서 그냥 시그널 대기
|
||||
pause();
|
||||
}
|
||||
|
||||
// 다음의 코드는 Daemon 으로 동작하기 때문에 수행되지 않는다.
|
||||
ReadyToExit();
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
|
||||
|
||||
// Data 수신을 위한 TCP Listen socket 생성
|
||||
bool MakeListenSocket( int & listenSocket )
|
||||
{
|
||||
listenSocket = socket( AF_INET, SOCK_STREAM, 0 );
|
||||
if( listenSocket == -1 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Listen socket create failed.[%d][%s]", errorNum, strerror( errorNum ) );
|
||||
return false;
|
||||
}
|
||||
|
||||
struct sockaddr_in listenSockAddr;
|
||||
socklen_t listenSockLen = 0;
|
||||
memset( &listenSockAddr, 0x00, sizeof( listenSockAddr ) );
|
||||
|
||||
listenSockAddr.sin_family = AF_INET;
|
||||
listenSockAddr.sin_port = htons( CServiceConfig::GetInstance()->GetTcpListenPort() );
|
||||
listenSockAddr.sin_addr.s_addr = htonl( INADDR_ANY );
|
||||
|
||||
listenSockLen = sizeof( listenSockAddr );
|
||||
|
||||
int result = 0;
|
||||
|
||||
// Socket Port Reuse Option Set
|
||||
int opt = 1;
|
||||
result = setsockopt( listenSocket, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof( opt ) );
|
||||
if( result != 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Listen socket option[SO_REUSEADDR] set failed. [%d][%s]", errorNum, strerror( errorNum ) );
|
||||
return false;
|
||||
}
|
||||
|
||||
// Keep Alive Option set
|
||||
opt = 1;
|
||||
result = setsockopt( listenSocket, SOL_SOCKET, SO_KEEPALIVE, &opt, sizeof( opt ) );
|
||||
if( result != 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Listen socket option[SO_KEEPALIVE] set failed. [%d][%s]", errorNum, strerror( errorNum ) );
|
||||
return false;
|
||||
}
|
||||
|
||||
// Socket Bind
|
||||
result = bind( listenSocket, ( struct sockaddr * )&listenSockAddr, listenSockLen );
|
||||
if( result != 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Listen socket bind failed.[%d][%s]", errorNum, strerror( errorNum ) );
|
||||
return false;
|
||||
}
|
||||
|
||||
// Socket Listen
|
||||
result = listen( listenSocket, DEFAULT_ACCEPT_WAIT_COUNT );
|
||||
if( result != 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Listen socket listen failed.[%d][%s]", errorNum, strerror( errorNum ) );
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
// Worker Process 생성(fork) 처리 함수
|
||||
// @return 생성 성공시 true, 실패시에는 false 를 반환.
|
||||
bool MakeProcessWorker( int listenSocket )
|
||||
{
|
||||
pid_t processId;
|
||||
|
||||
// Conf 에 정의된 Worker Process Count 만큼 Worker 프로세스 생성 처리.
|
||||
for( unsigned int index = 0; index < CServiceConfig::GetInstance()->GetWorkerProcessCount(); ++index )
|
||||
{
|
||||
// Worker 프로세스 fork
|
||||
processId = fork();
|
||||
|
||||
if( processId < 0 ) // fork fail
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Worker process[%d th] create failed. [%u][%s]", index, errorNum, strerror( errorNum ) );
|
||||
return false;
|
||||
}
|
||||
else if( processId == 0 ) // Child Process => Worker Process
|
||||
{
|
||||
// Worker Process Main 함수 호출 및 종료처리.
|
||||
int iResult;
|
||||
iResult = WorkerMain( listenSocket );
|
||||
|
||||
ReadyToExit();
|
||||
|
||||
// WorkerMain() 함수 return 값으로 종료처리되도록 한다.
|
||||
exit( iResult );
|
||||
}
|
||||
else
|
||||
{
|
||||
// Parent Process => Logging
|
||||
_LOG( LINF, "Worker[%u th][%d] create success.", index, processId );
|
||||
|
||||
// 잠시 대기
|
||||
|
||||
// CHG 2015-09-16 huibong
|
||||
// - #24436 usleep -> nanosleep 으로 변경 처리.
|
||||
struct timespec sleep;
|
||||
sleep.tv_sec = 0;
|
||||
sleep.tv_nsec = 200000000; // 0.2 sec
|
||||
nanosleep( &sleep, NULL );
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// Worker Process 재생성(fork) 처리 함수
|
||||
// @return 생성 성공시 true, 실패시에는 false 를 반환.
|
||||
bool RemakeProcessWorker( int listenSocket )
|
||||
{
|
||||
// Worker 프로세스 fork
|
||||
pid_t processId = fork();
|
||||
if( processId < 0 ) // fork fail
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Worker process recreate failed. [%d][%s]", errorNum, strerror( errorNum ) );
|
||||
return false;
|
||||
}
|
||||
else if( processId == 0 ) // Child Process => Worker Process
|
||||
{
|
||||
// Worker Process Main 함수 호출 및 종료처리.
|
||||
int iResult;
|
||||
iResult = WorkerMain( listenSocket );
|
||||
|
||||
ReadyToExit();
|
||||
|
||||
// WorkerMain() 함수 return 값으로 종료처리되도록 한다.
|
||||
exit( iResult );
|
||||
}
|
||||
else
|
||||
{
|
||||
// Parent Process => Logging
|
||||
_LOG( LINF, "Worker[%d] recreate success.", processId );
|
||||
|
||||
// 잠시 대기
|
||||
|
||||
// CHG 2015-09-16 huibong
|
||||
// - #24436 usleep -> nanosleep 으로 변경 처리.
|
||||
struct timespec sleep;
|
||||
sleep.tv_sec = 0;
|
||||
sleep.tv_nsec = 200000000; // 0.2 sec
|
||||
nanosleep( &sleep, NULL );
|
||||
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// std 상에 trim 함수가 없어서 직접 구현 아니면 boost/algorithm/string.hpp 상의 boost::trim 함수 사용
|
||||
// @return void
|
||||
void Trim( std::string & str)
|
||||
{
|
||||
if( str.length() == 0 )
|
||||
return;
|
||||
|
||||
// 문자열 뒤의 공백, TAB, CR 등의 문자 제거처리.
|
||||
std::string::size_type pos = str.find_last_not_of(" \a\b\f\n\r\t\v");
|
||||
if( pos != std::string::npos )
|
||||
str.erase(pos + 1);
|
||||
|
||||
// 문자열 앞의 공백, TAB, CR 등의 문자 제거처리.
|
||||
pos = str.find_first_not_of(" \a\b\f\n\r\t\v");
|
||||
if( pos != std::string::npos )
|
||||
str.erase(0, pos);
|
||||
}
|
||||
|
||||
// 현재 Process가 기동중인지 여부를 판단하기 위한 함수( 프로세스 중복 실행 체크)
|
||||
// @return 이미 해당 프로세스가 기동 중인 경우 true 반환, 그렇지 않으면 false 반환.
|
||||
bool IsCurrentProcessRun()
|
||||
{
|
||||
char tempBuffer[512];
|
||||
FILE * fd = NULL;
|
||||
bool bRun = false;
|
||||
|
||||
snprintf( tempBuffer, sizeof(tempBuffer), "pgrep -x %s | sort", PROG_NAME );
|
||||
|
||||
fd = popen( tempBuffer, "r" );
|
||||
if( fd == NULL )
|
||||
{
|
||||
// 오류 발생시 true 반환하여 프로세스 실행 방지처리
|
||||
fprintf(stderr, "[ERR] Process duplication check failed.[popen error][%s]\n", strerror(errno));
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
memset( tempBuffer, 0x00, sizeof(tempBuffer) );
|
||||
while( fgets( tempBuffer, sizeof(tempBuffer)-1, fd) != NULL )
|
||||
{
|
||||
string tempPid( tempBuffer );
|
||||
Trim(tempPid);
|
||||
|
||||
if( atoi( tempPid.c_str() ) != getpid() )
|
||||
{
|
||||
fprintf(stderr, "[info] Process duplication found. pid[%d]\n", atoi(tempPid.c_str()));
|
||||
bRun = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
pclose( fd );
|
||||
return bRun;
|
||||
}
|
||||
}
|
||||
|
||||
// 전달 받은 Directory 가 존재하는지 확인한다.
|
||||
bool CheckPath( std::string path )
|
||||
{
|
||||
struct stat dirStat;
|
||||
|
||||
// 해당 이름을 가진 파일 또는 디렉토리가 존재하고
|
||||
if( lstat( path.c_str(), &dirStat ) == 0 )
|
||||
{
|
||||
// 해당 이름이 디렉토리인 경우
|
||||
if( S_ISDIR( dirStat.st_mode ) == true )
|
||||
{
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
fprintf( stderr, "Exist file with the same name as directory [%s]\n", path.c_str() );
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
fprintf( stderr, "Directory not found. [%s]\n", path.c_str() );
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// 프로세스 종료 전 처리할 종료 관련 각 작업을 일괄로 처리하기 위한 함수.
|
||||
void ReadyToExit()
|
||||
{
|
||||
if( g_listenSocket != -1 )
|
||||
{
|
||||
close( g_listenSocket );
|
||||
g_listenSocket = -1;
|
||||
}
|
||||
|
||||
// Logger 객체 종료 처리.
|
||||
CLogger::Exit();
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
// Main Process 가 종료 Signal 을 전달받은 경우 이를 처리하기 위한 함수.
|
||||
// @param nSignalNumber [in] 발생한 시그널 Number
|
||||
// @return void
|
||||
static void SignalMainTerminate(int nSignalNumber)
|
||||
{
|
||||
// Signal Number 에 따른 로깅처리.
|
||||
if( nSignalNumber == SIGTERM )
|
||||
{
|
||||
_LOG( LINF, "Main[%d] process exit by user signal[SIGTERM]. Good Bye..", getpid() );
|
||||
}
|
||||
else
|
||||
{
|
||||
_LOG( LWAR, "Main[%d] process exit by abnormal signal[%d]. Good Bye..", getpid(), nSignalNumber );
|
||||
}
|
||||
|
||||
ReadyToExit();
|
||||
|
||||
// CHG 2015-09-16 huibong
|
||||
// - #24436 usleep -> nanosleep 으로 변경 처리.
|
||||
struct timespec sleep;
|
||||
sleep.tv_sec = 0;
|
||||
sleep.tv_nsec = 500000000; // 0.5 sec
|
||||
nanosleep( &sleep, NULL );
|
||||
|
||||
exit( EXIT_SUCCESS );
|
||||
}
|
||||
|
||||
// Main Process 에서 자식 프로세스인 Worker Process 가 종료된 경우 시그널 처리를 위한 함수
|
||||
// @param nSignalNumber [in] 발생한 시그널 Number
|
||||
// @return void
|
||||
static void SignalWorkerDead(int nSignalNumber)
|
||||
{
|
||||
pid_t deadPid;
|
||||
int nDeadStatus;
|
||||
|
||||
while( (deadPid = waitpid(-1, &nDeadStatus, WNOHANG)) > 0 )
|
||||
{
|
||||
// Worker 프로세스가 Signal 에 의해 종료되었는가?
|
||||
if( WIFSIGNALED( nDeadStatus ) )
|
||||
{
|
||||
_LOG( LWAR, "Worker[%d] killed by signal[%d].", deadPid, WTERMSIG(nDeadStatus));
|
||||
}
|
||||
else
|
||||
{
|
||||
_LOG( LWAR, "Worker[%d] killed. no signal.", deadPid);
|
||||
|
||||
// Worker 프로세스가 EXIT_FAILURE 반환 ( 초기화 실패시 )
|
||||
// 해당 내역을 화면 및 로그 상에 출력하고
|
||||
// Worker 프로세스를 재생성 처리하지 않는다.
|
||||
if( WIFEXITED( nDeadStatus ) )
|
||||
{
|
||||
if( WEXITSTATUS( nDeadStatus ) == EXIT_FAILURE )
|
||||
{
|
||||
_LOG( LERR, "Worker[%d] initilaize failed.", deadPid);
|
||||
_LOG( LERR, "Main[%d] abnormally exit by worker initialize failed. Good Bye..", getpid());
|
||||
|
||||
ReadyToExit();
|
||||
|
||||
// CHG 2015-09-16 huibong
|
||||
// - #24436 usleep -> nanosleep 으로 변경 처리.
|
||||
struct timespec sleep;
|
||||
sleep.tv_sec = 0;
|
||||
sleep.tv_nsec = 500000000; // 0.5 sec
|
||||
nanosleep( &sleep, NULL );
|
||||
|
||||
exit( EXIT_SUCCESS );
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Worker Process 재생성 처리.
|
||||
if( RemakeProcessWorker( g_listenSocket ) == false )
|
||||
{
|
||||
// Worker Process 재성성 실패시
|
||||
// 로깅 처리 후 Main 프로세스 종료 시그널 발생
|
||||
_LOG( LERR, "Worker process recreate failed. => Main process exit.");
|
||||
raise( SIGTERM );
|
||||
}
|
||||
else
|
||||
{
|
||||
_LOG( LWAR, "Worker process recreate success.");
|
||||
}
|
||||
}
|
||||
|
||||
// 오류 발생시 해당 내역 로깅
|
||||
if( deadPid < 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Main process error: SIG_CHLD receive but waitpid return error[%d][%s]", errorNum, strerror(errorNum));
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
/// @brief Main 프로세스 signal 처리 설정을 위한 함수
|
||||
/// @return void
|
||||
void SetSignalMain()
|
||||
{
|
||||
sigset_t set;
|
||||
struct sigaction act;
|
||||
memset(&act, 0x00, sizeof(act));
|
||||
|
||||
sigfillset( &set );
|
||||
sigprocmask( SIG_SETMASK, &set, NULL );
|
||||
sigfillset( &act.sa_mask );
|
||||
|
||||
// 무시 처리 signal
|
||||
act.sa_handler = SIG_IGN;
|
||||
sigaction( SIGPIPE, &act, NULL); /* desciptor 오류 발생시 Process가 죽는 것은 방지하기 위하여 설정 */
|
||||
sigaction( SIGHUP, &act, NULL); /* process를 기동시킨 관리자의 로그아웃시, 또는 config reload 등의 처리 signal*/
|
||||
sigaction( SIGINT, &act, NULL); /* ^C 키를 누른 경우 받는 신호 => demon 으로 기동되기 땜시 이 신호 못받음 */
|
||||
sigaction( SIGQUIT, &act, NULL); /* 키보드에 의한 Abort 신호 처리 => ? */
|
||||
|
||||
// Child Process 인 Worker 프로세스 종료에 대한 처리기 설정.
|
||||
act.sa_handler = SignalWorkerDead;
|
||||
sigaction( SIGCHLD, &act, NULL);
|
||||
|
||||
// 사용자 종료 또는 에러 관련 신호 처리.
|
||||
act.sa_handler = SignalMainTerminate;
|
||||
sigaction( SIGTERM, &act, NULL); /* kill -TERM 에 의한 프로세스 종료시 */
|
||||
|
||||
sigemptyset(&set); /* 신호 처리기 처리 설정 위한 블록 해제 */
|
||||
sigprocmask(SIG_SETMASK, &set, NULL);
|
||||
}
|
||||
@@ -0,0 +1,71 @@
|
||||
/***************************************************************************
|
||||
cc_collectd Main Header ( Main.h )
|
||||
-----------------------------------------
|
||||
begin : 2014/09/03
|
||||
copyright : (C) 2005 Solbox Inc.
|
||||
author : Storage Dev Team
|
||||
email : storage.sd@solbox.com
|
||||
version : 3.4
|
||||
|
||||
CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of Solbox Inc.
|
||||
***************************************************************************/
|
||||
|
||||
#ifndef __CC_COLLECTD_MAIN_H__
|
||||
#define __CC_COLLECTD_MAIN_H__
|
||||
|
||||
#include <string>
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// 사용 방법을 화면에 표시하기 위한 함수.
|
||||
void PrintUsage(void);
|
||||
|
||||
// Version 정보를 화면에 표시하기 위한 함수
|
||||
void PrintVersion(void);
|
||||
|
||||
// Data 수신을 위한 TCP Listen socket 생성
|
||||
bool MakeListenSocket( int & listenSocket );
|
||||
|
||||
// Worker Process 생성(fork) 처리 함수
|
||||
// @return 생성 성공시 true, 실패시에는 false 를 반환.
|
||||
bool MakeProcessWorker( int listenSocket );
|
||||
|
||||
// Worker Process 재생성(fork) 처리 함수
|
||||
// @return 생성 성공시 true, 실패시에는 false 를 반환.
|
||||
bool RemakeProcessWorker( int listenSocket );
|
||||
|
||||
// std 상에 trim 함수가 없어서 직접 구현 아니면 boost/algorithm/string.hpp 상의 boost::trim 함수 사용
|
||||
// @param str [in/out] Trim 할 문자열을 저장한 String 참조변수.
|
||||
// @return void
|
||||
void Trim( std::string & str );
|
||||
|
||||
// 현재 Process가 기동중인지 여부를 판단하기 위한 함수( 프로세스 중복 실행 체크)
|
||||
// @return 이미 해당 프로세스가 기동 중인 경우 true 반환, 그렇지 않으면 false 반환.
|
||||
bool IsCurrentProcessRun( void );
|
||||
|
||||
// 전달 받은 Directory 가 존재하는지 확인한다.
|
||||
bool CheckPath( std::string path );
|
||||
|
||||
|
||||
// 프로세스 종료에 따른 반복적인 종료 관련 작업을 수행하기 위한 함수.
|
||||
void ReadyToExit(void);
|
||||
|
||||
// Main 프로세스 signal 처리 설정을 위한 함수
|
||||
// @return void
|
||||
void SetSignalMain( void );
|
||||
|
||||
/* Signal 처리를 위한 각 Signal Handler 함수는
|
||||
* 다른 Code 에서 Include 처리시 Static 관련 문제로 인해
|
||||
* 본 Header 에서 선언처리 하지 않음. cpp 에만 존재
|
||||
*/
|
||||
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif /* __CC_COLLECTD_MAIN_H__ */
|
||||
@@ -0,0 +1,78 @@
|
||||
#****************************************************************************
|
||||
# Makefile for cc_collectd
|
||||
# -----------------------------------------
|
||||
#
|
||||
# begin : 2014/09/02
|
||||
# copyright : (C) 2005 Solbox Inc.
|
||||
# author : storage dev team
|
||||
# email : storage.sd@solbox.com
|
||||
# version : 3.5
|
||||
#
|
||||
# CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
# Redistribution and use in source and binary forms, with or with out
|
||||
# modification, are not permitted in outside of Solbox Inc.
|
||||
#*****************************************************************************
|
||||
|
||||
|
||||
# Program info
|
||||
PROG_NAME = cc_collectd
|
||||
REVISION = 1259
|
||||
PROG_VERSION = 3.5.0.$(REVISION)-`date +%Y%m%d%H%M%S`
|
||||
|
||||
|
||||
DEFAULT_CONFIG_FILE = /user/service/etc/cc_collectd.conf
|
||||
|
||||
|
||||
INSTALL_BIN = /user/service/bin
|
||||
INSTALL_CONF = /user/service/etc
|
||||
|
||||
|
||||
# Compiler info
|
||||
CC = /usr/bin/g++
|
||||
|
||||
CFLAGS = -Wall -O3 -g -Wimplicit -Wreturn-type -Wunused -Wuninitialized\
|
||||
-Wparentheses -Wshadow -Wpointer-arith -Woverloaded-virtual\
|
||||
-fno-rtti -D_REENTRANT
|
||||
|
||||
LFLAGS =
|
||||
|
||||
DFLAGS = -DPROG_NAME=\"$(PROG_NAME)\" -DPROG_VERSION=\"$(PROG_VERSION)\" -DDEFAULT_CONFIG_FILE=\"$(DEFAULT_CONFIG_FILE)\"
|
||||
|
||||
|
||||
# Application Enviroment
|
||||
APP = $(PROG_NAME)
|
||||
|
||||
DIR_INCLUDE = -I./. -I../lib -I../
|
||||
DIR_LIB = -L../lib
|
||||
|
||||
OBJ = ServiceConfig.o StorageStatus.o ProcessStatus.o \
|
||||
ProcessSocket.o ClientThread.o \
|
||||
Worker.o Main.o
|
||||
|
||||
LIBS = ../lib/libInterCommon.a -lpthread
|
||||
|
||||
|
||||
############################
|
||||
|
||||
all:$(APP)
|
||||
sync
|
||||
|
||||
%.o: %.cpp
|
||||
$(CC) $(CFLAGS) -o $@ -c $^ $(DFLAGS) $(DIR_INCLUDE)
|
||||
|
||||
|
||||
$(PROG_NAME): $(OBJ)
|
||||
$(CC) $(LFLAGS) -o $@ $^ $(DFLAGS) $(DIR_LIB) $(LIBS)
|
||||
|
||||
|
||||
clean:
|
||||
-rm -f *.o core.$(PROG_NAME) *.out *.log
|
||||
-rm -f $(APP)
|
||||
sync
|
||||
|
||||
|
||||
install : $(APP)
|
||||
-cp $(APP) $(INSTALL_BIN)/$(APP)
|
||||
sync
|
||||
|
||||
# End of Makefile
|
||||
@@ -0,0 +1,762 @@
|
||||
#include "ProcessSocket.h"
|
||||
|
||||
#include <arpa/inet.h>
|
||||
#include <netinet/in.h>
|
||||
#include <stdio.h>
|
||||
#include <errno.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include <endian.h>
|
||||
|
||||
|
||||
// 본 Class 에서 내부적으로 사용할 Packet 수신 결과 매크로 정의
|
||||
// 본 정의값은 GetPacketHeader() 함수의 결과값임.
|
||||
#define PACKET_HEADER_RECEIVE_SUCCESS 1 // Packet Header 정보 수신 정상 처리
|
||||
#define PACKET_HEADER_RECEIVE_TIMEOUT 0 // Packet Header 정보 수신시 Timeout 발생한 경우
|
||||
#define PACKET_HEADER_RECEIVE_SOCK_CLOSED -1 // Packet Header 정보 수신시 socket 이 Close 된 경우
|
||||
#define PACKET_HEADER_RECEIVE_INVALID_PROTOCOL -2 // Packet Header 정보 수신시 잘못된 Packet 형식인 경우
|
||||
#define PACKET_HEADER_RECEIVE_READ_ERROR -3 // Packet Header 정보 수신시 시스템 함수 오류 발생한 경우
|
||||
|
||||
|
||||
// 생성자.
|
||||
// socket [in] 처리할 socket descriptor
|
||||
CProcessSocket::CProcessSocket( const int& socket )
|
||||
: CBaseSocket( socket )
|
||||
, m_nPacketHeaderLen ( sizeof(m_packetHeader))
|
||||
, m_nPacketDataLen( 0 )
|
||||
{
|
||||
}
|
||||
|
||||
// 소멸자
|
||||
CProcessSocket::~CProcessSocket()
|
||||
{
|
||||
// 소멸자 Socket 명시적 Close 처리.
|
||||
Close();
|
||||
}
|
||||
|
||||
|
||||
// Packet Header 부분의 수신 처리를 위한 함수.
|
||||
// @param timeout [in] 대기시간.
|
||||
// @retrun
|
||||
// 1 : PACKET_HEADER_RECEIVE_SUCCESS : 성공시
|
||||
// 0 : PACKET_HEADER_RECEIVE_TIMEOUT : Timeout 발생시
|
||||
// -1 : PACKET_HEADER_RECEIVE_SOCK_CLOSED : socket 종료된 경우
|
||||
// -2 : PACKET_HEADER_RECEIVE_INVALID_PROTOCOL : 잘못된 Packet 형식인 경우
|
||||
// -3 : PACKET_HEADER_RECEIVE_READ_ERROR : 기타 오류
|
||||
int CProcessSocket::GetPacketHeader( int timeout )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
{
|
||||
LOG( LERR, "IsValidSocket return false." );
|
||||
|
||||
// 유효하지 않은 socket 이므로 -1 을 반환
|
||||
// return false;
|
||||
return PACKET_HEADER_RECEIVE_SOCK_CLOSED;
|
||||
}
|
||||
|
||||
int nRead;
|
||||
bool isAlivePacket;
|
||||
do
|
||||
{
|
||||
isAlivePacket = false;
|
||||
|
||||
// Socket 으로 부터 Packet Header 부분 수신.
|
||||
// 수신된 정보는 멤버변수에 저장처리.
|
||||
// ReadNTimeout() 반환값은 read 된 데이터의 크기. 0: fd closed, -1: 오류 , -2: Timeout
|
||||
nRead = ReadNTimeout( &m_packetHeader, m_nPacketHeaderLen, timeout );
|
||||
|
||||
// 정상적으로 Data 를 수신한 경우
|
||||
if( nRead > 0 )
|
||||
{
|
||||
// Packet 형식 코드 중 STX code 검사.
|
||||
if( m_packetHeader.stx != HEADER_STX_CODE )
|
||||
{
|
||||
LOG( LERR, "Not valid STX code " );
|
||||
PrintHeaderToLog();
|
||||
|
||||
// 잘못된 Packet 형식인 경우 -2 값을 반환
|
||||
//return false;
|
||||
return PACKET_HEADER_RECEIVE_INVALID_PROTOCOL;
|
||||
}
|
||||
|
||||
// Data Length 부분 값을 멤버 변수에 저장처리.
|
||||
m_nPacketDataLen = ntohl( m_packetHeader.data_length );
|
||||
|
||||
// Alive Check Packet 인 경우 해당 패킷은 무시처리 후 다시 수신 처리한다.
|
||||
if( m_packetHeader.type == HEADER_TYPE_REQUEST
|
||||
&& m_packetHeader.command[0] == COMMON_ALIVE_CHECK
|
||||
&& m_nPacketDataLen == 0 )
|
||||
{
|
||||
isAlivePacket = true;
|
||||
}
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
// Socket 이 종료된 경우
|
||||
if( nRead == 0 )
|
||||
{
|
||||
//LOG( LINF, "Connection already closed.");
|
||||
Close(); // 확실하게 종료 처리 한 후
|
||||
|
||||
// Connection 이 종료된 경우 -1 을 반환
|
||||
return PACKET_HEADER_RECEIVE_SOCK_CLOSED;
|
||||
}
|
||||
else if( nRead == -1 ) // read 함수 관련 오류가 발생한 경우
|
||||
{
|
||||
LOG( LERR, "Socket read() function error. Connection closed by forced." );
|
||||
Close(); // 확실하게 종료 처리 한 후
|
||||
|
||||
// 오류 발생시 -3 을 반환
|
||||
return PACKET_HEADER_RECEIVE_READ_ERROR;
|
||||
|
||||
}
|
||||
else if( nRead == -2 ) // Timeout 이 발생한 경우
|
||||
{
|
||||
LOG( LDBG, "Socket read() timeout [%d]", timeout );
|
||||
|
||||
// Timeout 발생시에는 0 을 반환
|
||||
return PACKET_HEADER_RECEIVE_TIMEOUT;
|
||||
}
|
||||
}
|
||||
|
||||
} while( isAlivePacket == true );
|
||||
|
||||
|
||||
return PACKET_HEADER_RECEIVE_SUCCESS;
|
||||
}
|
||||
|
||||
// Packet Header 정보를 Log 파일에 Logging 처리
|
||||
void CProcessSocket::PrintHeaderToLog()
|
||||
{
|
||||
_LOG( LINF, "------------------------------------" );
|
||||
_LOG( LINF, "stx [%02x]", m_packetHeader.stx );
|
||||
_LOG( LINF, "type [%02x]", m_packetHeader.type );
|
||||
_LOG( LINF, "command [%02x][%02x][%02x][%02x]", m_packetHeader.command[0], m_packetHeader.command[1], m_packetHeader.command[2], m_packetHeader.command[3] );
|
||||
_LOG( LINF, "result [%02x][%02x][%02x][%02x]", m_packetHeader.result[0], m_packetHeader.result[1], m_packetHeader.result[2], m_packetHeader.result[3] );
|
||||
_LOG( LINF, "data_length [%u]", m_nPacketDataLen );
|
||||
_LOG( LINF, "extend_code [%02x][%02x]", m_packetHeader.extend_code[0], m_packetHeader.extend_code[1] );
|
||||
_LOG( LINF, "------------------------------------" );
|
||||
}
|
||||
|
||||
|
||||
// socket 에서 지정된 크기만큼의 데이터를 읽어 출력변수에 저장처리.
|
||||
// @param size [in] read 할 데이터 크기
|
||||
// @param value [out] 읽은 데이터를 저장할 string 변수
|
||||
// @return On success return true, otherwise return false.
|
||||
bool CProcessSocket::GetPacketData( unsigned int& size, std::string& value )
|
||||
{
|
||||
if( size == 0 )
|
||||
{
|
||||
value.clear();
|
||||
return true;
|
||||
}
|
||||
|
||||
// 1. 수신할 크기가 임시버퍼보다 큰 경우 새로운 버퍼를 생성한다.
|
||||
BYTE * pBuffer = NULL;
|
||||
bool bNewBufferCreated = false;
|
||||
if( size > DEFAULT_SOCKET_TEMP_BUFFER_SIZE )
|
||||
{
|
||||
pBuffer = new BYTE[size]; // 신규 메모리 할당.
|
||||
bNewBufferCreated = true;
|
||||
}
|
||||
else
|
||||
pBuffer = m_tempBuffer;
|
||||
|
||||
// 2. 데이터 수신 처리.
|
||||
int nRead = ReadNTimeout( pBuffer, size );
|
||||
|
||||
// 오류 발생시
|
||||
if( nRead <= 0 )
|
||||
{
|
||||
// 신규로 생성된 버퍼인 경우 메모리 해제 처리.
|
||||
if( bNewBufferCreated == true )
|
||||
delete [] pBuffer;
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
// 3. 수신 데이터 저장 처리 - 기존 데이터 뒤에 붙여 준다=> 이게 사용하기 편함.
|
||||
value.append( (char *)pBuffer, size );
|
||||
|
||||
// 신규로 생성된 버퍼인 경우 메모리 해제 처리.
|
||||
if( bNewBufferCreated == true )
|
||||
delete [] pBuffer;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// socket 에서 지정된 크기만큼의 데이터를 읽어 내부 임시버퍼인 m_tempBuffer 에 저장처리.
|
||||
// @param size [in] read 할 데이터 크기
|
||||
// @return On success return true, otherwise return false.
|
||||
bool CProcessSocket::GetPacketData( unsigned int size )
|
||||
{
|
||||
if( size == 0 )
|
||||
return true;
|
||||
|
||||
// 1. 수신할 크기가 임시버퍼보다 큰 경우 오류 반환.
|
||||
if( size > DEFAULT_SOCKET_TEMP_BUFFER_SIZE )
|
||||
return false;
|
||||
|
||||
// 2. 데이터 수신 처리.
|
||||
int nRead = ReadNTimeout( m_tempBuffer, size );
|
||||
|
||||
// 오류 발생시
|
||||
if( nRead <= 0 )
|
||||
return false;
|
||||
|
||||
// 3. 정상인 경우는 ReadNTimeout 함수가 무조건 요청 크기만큼 read 처리하므로 정상수신임.
|
||||
return true;
|
||||
}
|
||||
|
||||
// 전달받은 데이터를 unsigned int 형으로 변환처리 및 Endian 변환
|
||||
unsigned int CProcessSocket::GetDataToUInt( BYTE * pValue, bool bConvertEndian )
|
||||
{
|
||||
BYTE tempBuffer[4];
|
||||
unsigned int result;
|
||||
|
||||
memcpy(tempBuffer, pValue, 4);
|
||||
unsigned int * pInt = (unsigned int *)tempBuffer;
|
||||
|
||||
// Endian 변환처리.
|
||||
if( bConvertEndian == true )
|
||||
{
|
||||
result = ntohl( *pInt );
|
||||
}
|
||||
else
|
||||
result = *pInt;
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
// pValue 에 저장된 데이터를 unsigned long long (64Byte) 형으로 변환처리.
|
||||
unsigned long long CProcessSocket::GetDataToUInt64( BYTE * pValue, bool bConvertEndian )
|
||||
{
|
||||
BYTE tempBuffer[8];
|
||||
unsigned long long result;
|
||||
|
||||
memcpy( tempBuffer, pValue, 8 );
|
||||
unsigned long long * pResult = ( unsigned long long * )tempBuffer;
|
||||
|
||||
// Endian 변환처리.
|
||||
if( bConvertEndian == true )
|
||||
{
|
||||
// Network byte order (big endian) 을 Host byte order 로 변환 처리
|
||||
// result = be64toh( *pResult );
|
||||
result = do_ntohll( *pResult );
|
||||
}
|
||||
else
|
||||
result = *pResult;
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
// Socket 으로 부터 수신받은 데이터를 파일에 Write 처리한다.
|
||||
// 만약 File Write 처리시 오류가 발생하는 경우 Socket Data 의 정상적인 처리를 위해
|
||||
// Socket 데이터 수신만 처리하고 오류를 반환한다.
|
||||
// @returnFile Write 가 오류 없이 정상적으로 처리된 경우 true, 오류발생시 false 및 오류메시지 반환
|
||||
bool CProcessSocket::ReceiveToFile( FILE * pFile, unsigned long long nFileSize, std::string& errorMessage )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return false;
|
||||
|
||||
// 0 Byte 크기의 파일인 경우 그냥 SUCCESS 처리.
|
||||
if( nFileSize == 0 )
|
||||
return true;
|
||||
|
||||
char tempBuffer[FILE_DATA_BUFFER_SIZE];
|
||||
int onceReadSize;
|
||||
int nRead;
|
||||
unsigned long long nLeft = nFileSize;
|
||||
bool bFileWriteSuccess = true;
|
||||
|
||||
int nLockResult;
|
||||
|
||||
|
||||
// 동일 파일에 대한 동시 Write 처리를 방어하기 위해 file lock 개념을 추가한다.
|
||||
int fd = fileno( pFile );
|
||||
nLockResult = lockf( fd, F_TLOCK, 0 ); // 파일 전체에 대해 lock 시도.
|
||||
|
||||
if( nLockResult == -1 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
if( errorNum == EAGAIN )
|
||||
{
|
||||
// 이미 Lock 이 걸려 있는 상태인 경우... 5 sec 후 재시도 처리..
|
||||
// 만약 재시도가 실패하는 경우.. 실패로 반환.
|
||||
LOG( LWAR, "File lock 1th fail. 5 sec sleep and retry [%d][%s]", errorNum, strerror(errorNum));
|
||||
|
||||
sleep(5);
|
||||
nLockResult = lockf( fd, F_TLOCK, 0 ); // 파일 전체에 대해 lock 재시도.
|
||||
|
||||
if( nLockResult == -1 )
|
||||
{
|
||||
errorNum = errno;
|
||||
|
||||
// Lock 재시도 실패시.. 오류 반환.
|
||||
LOG( LERR, "File lock fail. file receive fail [%d][%s]", errorNum, strerror(errorNum));
|
||||
|
||||
errorMessage = "File data receive error [file lock fail]";
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// 이미 Lock 상태로 인한 lock 처리 실패가 아닌 경우.
|
||||
LOG( LERR, "File lock fail by system error. file receive fail [%d][%s]", errorNum, strerror(errorNum));
|
||||
errorMessage = "File data receive error [file lock fail]";
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
while( nLeft > 0 )
|
||||
{
|
||||
// 한번에 읽을 용량 계산.
|
||||
if( nLeft > FILE_DATA_BUFFER_SIZE )
|
||||
onceReadSize = FILE_DATA_BUFFER_SIZE;
|
||||
else
|
||||
onceReadSize = nLeft;
|
||||
|
||||
// Socket Read
|
||||
if( (nRead = read( m_sock, tempBuffer, onceReadSize )) < 0 )
|
||||
{
|
||||
int errorNum = errno;
|
||||
if( errorNum == EINTR || errorNum == EAGAIN || errorNum == EWOULDBLOCK )
|
||||
{
|
||||
nRead = 0;
|
||||
continue;
|
||||
}
|
||||
else if( errorNum == ECONNRESET )
|
||||
{
|
||||
m_bConnected = false;
|
||||
errorMessage = "File receive error [connection closed].";
|
||||
|
||||
lockf( fd, F_ULOCK, 0 );
|
||||
return false;
|
||||
}
|
||||
else
|
||||
{
|
||||
errorMessage = "File data receive error [socket read error]";
|
||||
LOG( LERR, "Socket read error.[%d][%s]", errorNum, strerror(errorNum));
|
||||
|
||||
lockf( fd, F_ULOCK, 0 );
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else if( nRead == 0 ) // Connection Closed..
|
||||
{
|
||||
m_bConnected = false;
|
||||
errorMessage = "File receive error [connection closed].";
|
||||
|
||||
lockf( fd, F_ULOCK, 0 );
|
||||
return false;
|
||||
}
|
||||
|
||||
// 정상적인 경우.
|
||||
// File Write 시도
|
||||
if( bFileWriteSuccess == true )
|
||||
{
|
||||
if( (int)fwrite( tempBuffer, 1, nRead, pFile ) != nRead )
|
||||
{
|
||||
// file write 실패시.
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "File write to local error.[%d][%s]", errorNum, strerror(errorNum));
|
||||
|
||||
// TODO 소켓을 읽어 전송받은 데이터를 날려버려야 할까요..?
|
||||
errorMessage = "File write failed from receive data";
|
||||
bFileWriteSuccess = false;
|
||||
}
|
||||
}
|
||||
|
||||
nLeft -= nRead;
|
||||
}
|
||||
|
||||
// file Lock 해제 처리.
|
||||
lockf( fd, F_ULOCK, 0 );
|
||||
|
||||
// File Write 가 오류 없이 정상적으로 처리된 경우 true
|
||||
// 오류발생시 false 반환처리.
|
||||
if( bFileWriteSuccess == true )
|
||||
return true;
|
||||
else
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
// Packet 정보 중 String 정보를 전송하기 위한 함수.
|
||||
bool CProcessSocket::SendString( const std::string& strData )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return false;
|
||||
|
||||
// 크기값을 저장하기 위한 임시 변수.
|
||||
unsigned int nSize = 0;
|
||||
|
||||
// String Data 전송
|
||||
nSize = htonl( strData.size() );
|
||||
WriteN( &nSize, 4 );
|
||||
if( strData.size() > 0 )
|
||||
WriteN( strData.c_str(), strData.size() );
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
///////////////////////////////////////////////////////////////////////////////
|
||||
// 업무 Flow 관련 interface 함수.
|
||||
///////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
|
||||
// Packet 의 Header 정보를 수신하여 각 정보를 전달받은 Command 정보를 참조변수로 반환.
|
||||
// @param command [out] Command 정보
|
||||
// @return 1 : Data 수신 성공시
|
||||
// 0 : Timeout 발생시
|
||||
// -1 : 접속 종료시
|
||||
int CProcessSocket::GetHeaderInfo( char& command )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return -1;
|
||||
|
||||
// Packet Header 정보 수신처리.
|
||||
// Socket 응답 대기상태( select) 에서 응답이 들어온 경우 이를 바로 처리하기 위해
|
||||
// Timeout 시간을 3 sec 로 처리함.
|
||||
int nResult = GetPacketHeader( 3 );
|
||||
|
||||
// Header Data 수신 실패시
|
||||
if( nResult != PACKET_HEADER_RECEIVE_SUCCESS )
|
||||
{
|
||||
if( nResult == PACKET_HEADER_RECEIVE_TIMEOUT ) // Timeout 발생시
|
||||
{
|
||||
if( IsValidSocket() == false ) // socket 종료 여부를 다시 한번 확인
|
||||
return -1;
|
||||
else
|
||||
return 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
// 잘못된 프로토콜 형식인 경우
|
||||
if( nResult == PACKET_HEADER_RECEIVE_INVALID_PROTOCOL )
|
||||
{
|
||||
// Socket 강제 종료 처리.
|
||||
LOG( LERR, "Not valid protocol header. Connection closed by forced." );
|
||||
Close();
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
// Data 수신 성공시
|
||||
// - 정상적인 Request 요청인지 확인 처리..
|
||||
// - 비정상 요청인 경우.. command 값을 REQUEST_NOT_VALID 으로 설정 처리한다.
|
||||
|
||||
// Type 값 확인
|
||||
if( m_packetHeader.type == HEADER_TYPE_REQUEST )
|
||||
{
|
||||
// request 요청인 경우...
|
||||
command = m_packetHeader.command[0];
|
||||
}
|
||||
else
|
||||
{
|
||||
// request 가 아닌.. reponse 등의 요청인 경우...
|
||||
// - 이렇게 되면.. 통신 Flow 문제이므로 코드 수정 필요.
|
||||
command = REQUEST_NOT_VALID;
|
||||
}
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
|
||||
|
||||
// REQUEST_AVAILABLE_SIZE 요청에 대한 응답을 Client 로 전송한다.
|
||||
bool CProcessSocket::SendResultAvailableSizeToClient( unsigned long long & uSize )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return false;
|
||||
|
||||
// 정보 전달을 위한 Packet Header 생성.
|
||||
struct CcCollectdPacketHeader stHeader;
|
||||
memset( &stHeader, 0x00, sizeof( struct CcCollectdPacketHeader ) );
|
||||
|
||||
stHeader.stx = HEADER_STX_CODE;
|
||||
stHeader.type = HEADER_TYPE_RESPONSE;
|
||||
stHeader.command[0] = REQUEST_AVAILABLE_SIZE;
|
||||
|
||||
// 성공, 실패 여부 설정
|
||||
// - 추출된 값만 전송하면 되므로. 무조건 성공으로 전송 처리.
|
||||
stHeader.result[0] = HEADER_RESULT_SUCCESS;
|
||||
|
||||
// Data body 구성
|
||||
// - 전송해야 할 항목은 unisigned long long 8 Byte 값이다.
|
||||
// - client 는 꼭 centos를 사용하는 client 가 아닐 수가 있어....Network byte order 에 따라 전송 처리한다.
|
||||
stHeader.data_length = htonl( 8 );
|
||||
|
||||
// Packet Header 정보 전송
|
||||
if( WriteN( &stHeader, m_nPacketHeaderLen ) != m_nPacketHeaderLen )
|
||||
{
|
||||
LOG( LERR, "reponse [REQUEST_AVAILABLE_SIZE] send fail to client" );
|
||||
return false;
|
||||
}
|
||||
else
|
||||
{
|
||||
// Available size 항목 전송
|
||||
// - unsigned long long type 을 Network Byte order(Big Endian) 로 변환하여 전달.
|
||||
unsigned long long uTemp;
|
||||
//uTemp = htobe64( uSize );
|
||||
uTemp = do_htonll( uSize );
|
||||
|
||||
WriteN( &uTemp, 8 );
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// REQUEST_FILE_SAVE 요청에 대한 Body 부분을 수신한다.
|
||||
bool CProcessSocket::RecvBodyFileSaveFromClient( std::string& strHostName, std::string& strProcessName, std::string& strFileName, unsigned long long& uFileSize )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return false;
|
||||
|
||||
// Body 부분 수신 처리.
|
||||
// strHostName : string ( 4 + string.size() )
|
||||
// strProcessName : string ( 4 + string.size() )
|
||||
// strFileName : string ( 4 + string.size() )
|
||||
// uFileSize : 8 Byte ---------->본 항목까지만 header data_length 에 포함시킨다.
|
||||
// 저장할 파일 내용.
|
||||
unsigned int nLength;
|
||||
|
||||
// strHostName
|
||||
if( GetPacketData( 4 ) == false )
|
||||
{
|
||||
LOG( LERR, "host name size get failed." );
|
||||
return false;
|
||||
}
|
||||
nLength = GetDataToUInt( m_tempBuffer, true );
|
||||
GetPacketData( nLength, strHostName );
|
||||
|
||||
// strProcessName
|
||||
if( GetPacketData( 4 ) == false )
|
||||
{
|
||||
LOG( LERR, "process name size get failed." );
|
||||
return false;
|
||||
}
|
||||
nLength = GetDataToUInt( m_tempBuffer, true );
|
||||
GetPacketData( nLength, strProcessName );
|
||||
|
||||
// strFileName
|
||||
if( GetPacketData( 4 ) == false )
|
||||
{
|
||||
LOG( LERR, "file name size get failed." );
|
||||
return false;
|
||||
}
|
||||
nLength = GetDataToUInt( m_tempBuffer, true );
|
||||
GetPacketData( nLength, strFileName );
|
||||
|
||||
|
||||
// uFileSize
|
||||
if( GetPacketData( 8 ) == false )
|
||||
{
|
||||
LOG( LERR, "file size body receive failed." );
|
||||
return false;
|
||||
}
|
||||
BYTE * pPos = m_tempBuffer;
|
||||
uFileSize = GetDataToUInt64( pPos, true );
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// REQUEST_FILE_SAVE 요청에 대한 처리 결과를 전송한다.
|
||||
bool CProcessSocket::SendResultFileSaveToClient( bool bSuccess, std::string& strMessage )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return false;
|
||||
|
||||
// 정보 전달을 위한 Packet Header 생성.
|
||||
struct CcCollectdPacketHeader stHeader;
|
||||
memset( &stHeader, 0x00, sizeof( struct CcCollectdPacketHeader ) );
|
||||
|
||||
stHeader.stx = HEADER_STX_CODE;
|
||||
stHeader.type = HEADER_TYPE_RESPONSE;
|
||||
stHeader.command[0] = REQUEST_FILE_SAVE;
|
||||
|
||||
|
||||
// 성공, 실패 여부 설정
|
||||
if( bSuccess == true )
|
||||
stHeader.result[0] = HEADER_RESULT_SUCCESS;
|
||||
else
|
||||
stHeader.result[0] = HEADER_RESULT_ERROR;
|
||||
|
||||
// Data body length
|
||||
stHeader.data_length = htonl( strMessage.size() );
|
||||
|
||||
// Packet Header 정보 전송
|
||||
if( WriteN( &stHeader, m_nPacketHeaderLen ) != m_nPacketHeaderLen )
|
||||
{
|
||||
LOG( LERR, "reponse [REQUEST_FILE_SAVE] send fail to client" );
|
||||
return false;
|
||||
}
|
||||
else
|
||||
{
|
||||
// message 부분이 존재하는 경우.. 전송 처리.
|
||||
if( strMessage.empty() == false )
|
||||
{
|
||||
WriteN( strMessage.c_str(), strMessage.size() );
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
// REQUEST_FILE_CHECK 요청에 대한 Body 부분을 수신한다.
|
||||
bool CProcessSocket::RecvBodyFileCheckFromClient( std::string& strHostName, std::string& strProcessName, std::string& strFileName )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return false;
|
||||
|
||||
// Body 부분 수신 처리.
|
||||
// strHostName : string ( 4 + string.size() )
|
||||
// strProcessName : string ( 4 + string.size() )
|
||||
// strFileName : string ( 4 + string.size() )
|
||||
// 저장할 파일 내용.
|
||||
unsigned int nLength;
|
||||
|
||||
// strHostName
|
||||
if( GetPacketData( 4 ) == false )
|
||||
{
|
||||
LOG( LERR, "host name size get failed." );
|
||||
return false;
|
||||
}
|
||||
nLength = GetDataToUInt( m_tempBuffer, true );
|
||||
GetPacketData( nLength, strHostName );
|
||||
|
||||
// strProcessName
|
||||
if( GetPacketData( 4 ) == false )
|
||||
{
|
||||
LOG( LERR, "process name size get failed." );
|
||||
return false;
|
||||
}
|
||||
nLength = GetDataToUInt( m_tempBuffer, true );
|
||||
GetPacketData( nLength, strProcessName );
|
||||
|
||||
// strFileName
|
||||
if( GetPacketData( 4 ) == false )
|
||||
{
|
||||
LOG( LERR, "file name size get failed." );
|
||||
return false;
|
||||
}
|
||||
nLength = GetDataToUInt( m_tempBuffer, true );
|
||||
GetPacketData( nLength, strFileName );
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// REQUEST_FILE_CHECK 요청에 대한 정상 처리 결과를 전송한다.
|
||||
bool CProcessSocket::SendResultFileCheckToClient( unsigned long long & uSize )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return false;
|
||||
|
||||
// 정보 전달을 위한 Packet Header 생성.
|
||||
struct CcCollectdPacketHeader stHeader;
|
||||
memset( &stHeader, 0x00, sizeof( struct CcCollectdPacketHeader ) );
|
||||
|
||||
stHeader.stx = HEADER_STX_CODE;
|
||||
stHeader.type = HEADER_TYPE_RESPONSE;
|
||||
stHeader.command[0] = REQUEST_FILE_CHECK;
|
||||
|
||||
// 성공, 실패 여부 설정
|
||||
// - 추출된 값만 전송하면 되므로. 무조건 성공으로 전송 처리.
|
||||
stHeader.result[0] = HEADER_RESULT_SUCCESS;
|
||||
|
||||
// Data body 구성
|
||||
// - 전송해야 할 항목은 unisigned long long 8 Byte 값이다.
|
||||
// - client 는 꼭 centos를 사용하는 client 가 아닐 수가 있어....Network byte order 에 따라 전송 처리한다.
|
||||
stHeader.data_length = htonl( 8 );
|
||||
|
||||
// Packet Header 정보 전송
|
||||
if( WriteN( &stHeader, m_nPacketHeaderLen ) != m_nPacketHeaderLen )
|
||||
{
|
||||
LOG( LERR, "reponse [REQUEST_FILE_CHECK] send fail to client" );
|
||||
return false;
|
||||
}
|
||||
else
|
||||
{
|
||||
// 추출된 물리파일 size 항목 전송
|
||||
// - unsigned long long type 을 Network Byte order(Big Endian) 로 변환하여 전달.
|
||||
unsigned long long uTemp;
|
||||
uTemp = do_htonll( uSize );
|
||||
|
||||
WriteN( &uTemp, 8 );
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// REQUEST_FILE_CHECK 요청에 대한 오류 발생 내역을 전송한다.
|
||||
bool CProcessSocket::SendResultFileCheckErrorToClient( std::string& strErrorMessage )
|
||||
{
|
||||
if( IsValidSocket() == false )
|
||||
return false;
|
||||
|
||||
// 정보 전달을 위한 Packet Header 생성.
|
||||
struct CcCollectdPacketHeader stHeader;
|
||||
memset( &stHeader, 0x00, sizeof( struct CcCollectdPacketHeader ) );
|
||||
|
||||
stHeader.stx = HEADER_STX_CODE;
|
||||
stHeader.type = HEADER_TYPE_RESPONSE;
|
||||
stHeader.command[0] = REQUEST_FILE_CHECK;
|
||||
|
||||
// 성공, 실패 여부 설정
|
||||
// - 오류가 발생한 경우이므로... 실패로 처리
|
||||
stHeader.result[0] = HEADER_RESULT_ERROR;
|
||||
|
||||
// Data body length
|
||||
stHeader.data_length = htonl( strErrorMessage.size() );
|
||||
|
||||
// Packet Header 정보 전송
|
||||
if( WriteN( &stHeader, m_nPacketHeaderLen ) != m_nPacketHeaderLen )
|
||||
{
|
||||
LOG( LERR, "error reponse [REQUEST_FILE_CHECK] send fail to client" );
|
||||
return false;
|
||||
}
|
||||
else
|
||||
{
|
||||
// message 부분이 존재하는 경우.. 전송 처리.
|
||||
if( strErrorMessage.empty() == false )
|
||||
{
|
||||
WriteN( strErrorMessage.c_str(), strErrorMessage.size() );
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
// byte ordering function for 64bit variable
|
||||
uint64_t CProcessSocket::do_htonll(uint64_t hostlonglong)
|
||||
{
|
||||
int x = 1;
|
||||
|
||||
/* little endian */
|
||||
if(*(char *)&x == 1)
|
||||
return ((((uint64_t)htonl(hostlonglong)) << 32) + htonl(hostlonglong >> 32));
|
||||
|
||||
/* big endian */
|
||||
else
|
||||
return hostlonglong;
|
||||
}
|
||||
|
||||
uint64_t CProcessSocket::do_ntohll(uint64_t netlonglong)
|
||||
{
|
||||
int x = 1;
|
||||
|
||||
/* little endian */
|
||||
if(*(char *)&x == 1)
|
||||
return ((((uint64_t)ntohl(netlonglong)) << 32) + ntohl(netlonglong >> 32));
|
||||
|
||||
/* big endian */
|
||||
else
|
||||
return netlonglong;
|
||||
|
||||
}
|
||||
@@ -0,0 +1,147 @@
|
||||
/***************************************************************************
|
||||
cc_collectd 의 통신 처리 담당 class
|
||||
-----------------------------------------
|
||||
begin : 2014/09/25
|
||||
copyright : (C) 2005 Solbox Inc.
|
||||
author : Storage Dev Team
|
||||
email : storage.sd@solbox.com
|
||||
version : 3.4
|
||||
|
||||
CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of Solbox Inc.
|
||||
***************************************************************************/
|
||||
|
||||
#ifndef __PROCESS_SOCKET_H__
|
||||
#define __PROCESS_SOCKET_H__
|
||||
|
||||
|
||||
#include "BaseSocket.h"
|
||||
#include "CcCollectdProtocol.h"
|
||||
|
||||
#include <string>
|
||||
#include <stdint.h>
|
||||
|
||||
|
||||
///< BYTE 타입 정의
|
||||
#ifndef _BYTE_DEFINED
|
||||
#define _BYTE_DEFINED
|
||||
typedef unsigned char BYTE;
|
||||
#endif // _BYTE_DEFINED
|
||||
|
||||
|
||||
#define DEFAULT_SOCKET_TEMP_BUFFER_SIZE 1024 // Socket 관련 data 송수신시 사용할 임시버퍼 크기.
|
||||
#define FILE_DATA_BUFFER_SIZE 1048576 // 1 Mbyte, 파일 송수신시 사용할 임시 버퍼의 크기.
|
||||
|
||||
|
||||
class CProcessSocket : public CBaseSocket
|
||||
{
|
||||
|
||||
public:
|
||||
|
||||
// 생성자.
|
||||
// socket [in] 처리할 socket descriptor
|
||||
CProcessSocket( const int & socket );
|
||||
|
||||
// 소멸자.
|
||||
~CProcessSocket();
|
||||
|
||||
|
||||
public:
|
||||
|
||||
// Packet 의 Header 정보를 수신하여 각 정보를 전달받은 Command 정보를 참조변수로 반환.
|
||||
// @param command [out] Command 정보
|
||||
// @return 1 : Data 수신 성공시
|
||||
// 0 : Timeout 발생시
|
||||
// -1 : 접속 종료시
|
||||
int GetHeaderInfo( char& command );
|
||||
|
||||
|
||||
// Packet Header 정보를 Log 파일에 Logging 처리
|
||||
void PrintHeaderToLog( void );
|
||||
|
||||
|
||||
// REQUEST_AVAILABLE_SIZE 요청에 대한 응답을 Client 로 전송한다.
|
||||
bool SendResultAvailableSizeToClient( unsigned long long & uSize );
|
||||
|
||||
// REQUEST_FILE_SAVE 요청에 대한 Body 부분을 수신한다.
|
||||
bool RecvBodyFileSaveFromClient( std::string& strHostName, std::string& strProcessName
|
||||
, std::string& strFileName, unsigned long long& uFileSize );
|
||||
|
||||
// REQUEST_FILE_SAVE 요청에 대한 처리 결과를 전송한다.
|
||||
bool SendResultFileSaveToClient( bool bSuccess, std::string& strMessage );
|
||||
|
||||
// Socket 으로 부터 수신받은 데이터를 파일에 Write 처리한다.
|
||||
// 만약 File Write 처리시 오류가 발생하는 경우 Socket Data 의 정상적인 처리를 위해
|
||||
// Socket 데이터 수신만 처리하고 오류를 반환한다.
|
||||
// @returnFile Write 가 오류 없이 정상적으로 처리된 경우 true, 오류발생시 false 및 오류메시지 반환
|
||||
bool ReceiveToFile( FILE * pFile, unsigned long long nFileSize, std::string& errorMessage );
|
||||
|
||||
// REQUEST_FILE_CHECK 요청에 대한 Body 부분을 수신한다.
|
||||
bool RecvBodyFileCheckFromClient( std::string& strHostName, std::string& strProcessName, std::string& strFileName );
|
||||
|
||||
// REQUEST_FILE_CHECK 요청에 대한 정상 처리 결과를 전송한다.
|
||||
bool SendResultFileCheckToClient( unsigned long long & uSize );
|
||||
|
||||
// REQUEST_FILE_CHECK 요청에 대한 오류 발생 내역을 전송한다.
|
||||
bool SendResultFileCheckErrorToClient( std::string& strErrorMessage );
|
||||
|
||||
|
||||
private:
|
||||
|
||||
// Packet Header 변수
|
||||
struct CcCollectdPacketHeader m_packetHeader;
|
||||
|
||||
// m_packetHeader 구조체의 크기를 저장하기 위한 상수
|
||||
const int m_nPacketHeaderLen;
|
||||
|
||||
// Packet Header 에 저장된 Data 부분의 길이 정보값.
|
||||
unsigned int m_nPacketDataLen;
|
||||
|
||||
// Packet Data 부분의 수신처리시 임시로 사용할 버퍼.
|
||||
BYTE m_tempBuffer[DEFAULT_SOCKET_TEMP_BUFFER_SIZE];
|
||||
|
||||
private:
|
||||
|
||||
// Packet Header 부분의 수신 처리를 위한 함수.
|
||||
// @param timeout [in] 대기시간.
|
||||
// @retrun
|
||||
// 1 : PACKET_HEADER_RECEIVE_SUCCESS : 성공시
|
||||
// 0 : PACKET_HEADER_RECEIVE_TIMEOUT : Timeout 발생시
|
||||
// -1 : PACKET_HEADER_RECEIVE_SOCK_CLOSED : socket 종료된 경우
|
||||
// -2 : PACKET_HEADER_RECEIVE_INVALID_PROTOCOL : 잘못된 Packet 형식인 경우
|
||||
// -3 : PACKET_HEADER_RECEIVE_READ_ERROR : 기타 오류
|
||||
int GetPacketHeader(int timeout = DEFAULT_DATA_RECEIVE_TIMEOUT);
|
||||
|
||||
// socket 에서 지정된 크기만큼의 데이터를 읽어 출력변수에 저장처리.
|
||||
// @param size [in] read 할 데이터 크기
|
||||
// @param value [out] 읽은 데이터를 저장할 string 변수
|
||||
// @return On success return true, otherwise return false.
|
||||
bool GetPacketData( unsigned int& size, std::string& value );
|
||||
|
||||
// socket 에서 지정된 크기만큼의 데이터를 읽어 내부 임시버퍼인 m_tempBuffer 에 저장처리.
|
||||
// @param size [in] read 할 데이터 크기
|
||||
// @return On success return true, otherwise return false.
|
||||
bool GetPacketData( unsigned int size );
|
||||
|
||||
// pValue 에 저장된 데이터를 unsigned int 형으로 변환처리 및 Endian 변환
|
||||
unsigned int GetDataToUInt( BYTE * pValue, bool bConvertEndian = true );
|
||||
|
||||
// pValue 에 저장된 데이터를 unsigned long long (64Byte) 형으로 변환처리.
|
||||
unsigned long long GetDataToUInt64( BYTE * pValue, bool bConvertEndian = true );
|
||||
|
||||
// Packet 정보 중 String 정보를 전송하기 위한 함수.
|
||||
bool SendString( const std::string& strData );
|
||||
|
||||
|
||||
// Linux 5.4 의 경우 unsigned long long 의 byteorder 변환함수 제공 안함.
|
||||
// Linux 6.X, FreeBSD 에서는 be64toh 등의 함수 제공
|
||||
// - Liunx 5.X 장비 지원해야 하므로... 64bit 변환 함수 개발하여 사용.
|
||||
// - 기존 htonl 등의 함수와의 혼동을 피하기 위해 함수명은 C Style 로 구성.
|
||||
uint64_t do_htonll(uint64_t hostlonglong);
|
||||
uint64_t do_ntohll(uint64_t netlonglong);
|
||||
|
||||
};
|
||||
|
||||
#endif /* __PROCESS_SOCKET_CONTROL_H__ */
|
||||
|
||||
@@ -0,0 +1,118 @@
|
||||
#include "ProcessStatus.h"
|
||||
|
||||
#include "Logger.h"
|
||||
#include "ProcessRename.h"
|
||||
|
||||
#include <errno.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
|
||||
// Process 정보 update 주기 (sec)
|
||||
#define PROCESS_STATUS_UPDATE_TIME 2
|
||||
|
||||
|
||||
// 생성자
|
||||
CProcessStatus::CProcessStatus()
|
||||
{
|
||||
m_threadHandle = 0;
|
||||
|
||||
m_nCurrentClientCount = 0;
|
||||
|
||||
// Lock 초기화
|
||||
pthread_mutex_init( &m_mxClientThread, NULL );
|
||||
}
|
||||
|
||||
// 소멸자
|
||||
CProcessStatus::~CProcessStatus()
|
||||
{
|
||||
// Thread 동작 정지 처리
|
||||
// - 만약 Thread 가 이미 종료된 경우 m_threadHandle 이 다른 Thread Handle 일 수 있으므로
|
||||
// 업무 Flow 수정시 주의할 것
|
||||
if( m_threadHandle != 0 )
|
||||
pthread_cancel( m_threadHandle );
|
||||
|
||||
pthread_mutex_destroy( &m_mxClientThread );
|
||||
}
|
||||
|
||||
// thread를 생성하여 관련 작업을 수행.
|
||||
bool CProcessStatus::Start()
|
||||
{
|
||||
int nRet = ::pthread_create( &m_threadHandle, NULL, CProcessStatus::threadFunc, this );
|
||||
if( nRet != 0 )
|
||||
{
|
||||
// Thread 생성 실패시
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Thread create failed.[%d][%s]", errorNum, strerror( errorNum ) );
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void* CProcessStatus::threadFunc( void* arg )
|
||||
{
|
||||
CProcessStatus* pObject = reinterpret_cast<CProcessStatus *>( arg );
|
||||
pthread_detach( pthread_self() );
|
||||
|
||||
while( 1 )
|
||||
{
|
||||
pObject->Execute();
|
||||
|
||||
sleep( PROCESS_STATUS_UPDATE_TIME );
|
||||
}
|
||||
|
||||
// Thread 종료시 m_threadHandle 값을 초기화 처리.
|
||||
// - 소멸자에서 잘못된 Thread 를 종료할 수 있기 때문.
|
||||
pObject->m_threadHandle = 0;
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
// 현재 프로세스 표시 정보 업데이트
|
||||
void CProcessStatus::Execute( void )
|
||||
{
|
||||
// C: client count
|
||||
// M: master job count
|
||||
// S: slave job count
|
||||
// Y: sync job count
|
||||
|
||||
#if defined(__FreeBSD__)
|
||||
setproctitle("[C%d]",
|
||||
m_nCurrentClientCount
|
||||
);
|
||||
|
||||
#else
|
||||
char szTemp[100] = {0};
|
||||
sprintf(szTemp, "%s: Worker [C%d]"
|
||||
, PROG_NAME
|
||||
, m_nCurrentClientCount
|
||||
);
|
||||
|
||||
set_ps_display(szTemp, false);
|
||||
|
||||
#endif
|
||||
|
||||
return ;
|
||||
}
|
||||
|
||||
|
||||
// Client 수를 1 증가 처리
|
||||
void CProcessStatus::PlusClientCount()
|
||||
{
|
||||
pthread_mutex_lock( &m_mxClientThread );
|
||||
++m_nCurrentClientCount;
|
||||
pthread_mutex_unlock( &m_mxClientThread );
|
||||
}
|
||||
|
||||
// Client 수를 1 감소 처리
|
||||
// - 정확히 동작하는지 확인을 위해 일부로 (-) 값 보정 처리 안함.
|
||||
void CProcessStatus::MinusClientCount()
|
||||
{
|
||||
pthread_mutex_lock( &m_mxClientThread );
|
||||
--m_nCurrentClientCount;
|
||||
pthread_mutex_unlock( &m_mxClientThread );
|
||||
}
|
||||
|
||||
@@ -0,0 +1,70 @@
|
||||
/***************************************************************************
|
||||
Process 상태 정보 저장 및 console 출력 처리 class
|
||||
-----------------------------------------
|
||||
begin : 2014/09/24
|
||||
copyright : (C) 2005 Solbox Inc.
|
||||
author : storage dev team
|
||||
email : storage.sd@solbox.com
|
||||
version : 3.4
|
||||
|
||||
CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of Solbox Inc.
|
||||
***************************************************************************/
|
||||
|
||||
#ifndef __PROCESS_STATUS_H__
|
||||
#define __PROCESS_STATUS_H__
|
||||
|
||||
#include <pthread.h>
|
||||
|
||||
|
||||
// CProcessStatus class
|
||||
// - 프로세스 의 Client 수, 각 Mode 별 수행 수 정보를 저장
|
||||
// - 해당 정보를 console 상에 표시하는 역활을 수행한다.
|
||||
class CProcessStatus
|
||||
{
|
||||
public :
|
||||
|
||||
// 생성자
|
||||
CProcessStatus();
|
||||
|
||||
// 소멸자
|
||||
~CProcessStatus();
|
||||
|
||||
// thread를 생성하여 프로세스 출력 상태 갱신 작업을 수행.
|
||||
bool Start();
|
||||
|
||||
// Client 수를 1 증가 처리
|
||||
void PlusClientCount(void);
|
||||
|
||||
// Client 수를 1 감소 처리
|
||||
void MinusClientCount(void);
|
||||
|
||||
|
||||
private :
|
||||
|
||||
// 스레드 함수
|
||||
static void* threadFunc( void* arg );
|
||||
|
||||
// 현재 프로세스 표시 정보 업데이트
|
||||
void Execute( void );
|
||||
|
||||
|
||||
private:
|
||||
|
||||
// 쓰레드 핸들
|
||||
pthread_t m_threadHandle;
|
||||
|
||||
|
||||
// Lock 처리 관련
|
||||
pthread_mutex_t m_mxClientThread;
|
||||
|
||||
|
||||
////// 프로세스 title 로 모니터 대상 정보 /////////////
|
||||
|
||||
// 접속 Client 수
|
||||
int m_nCurrentClientCount;
|
||||
|
||||
};
|
||||
|
||||
#endif //__PROCESS_STATUS_H__
|
||||
@@ -0,0 +1,207 @@
|
||||
#include "ServiceConfig.h"
|
||||
|
||||
#include <stdlib.h>
|
||||
|
||||
#include "Logger.h"
|
||||
|
||||
|
||||
// static 변수 초기화
|
||||
CServiceConfig * CServiceConfig::m_pInstance = NULL;
|
||||
|
||||
|
||||
#define DEFAULT_WORKER_PROCESS_COUNT 3 // default worker 프로세스 count
|
||||
|
||||
|
||||
// 생성자
|
||||
CServiceConfig::CServiceConfig()
|
||||
{
|
||||
// default 값 설정.
|
||||
// 지정된 conf 파일에 해당 설정이 없어도 동작되도록 처리할 값에 대해서만 default 값을 정의한다.
|
||||
m_nWorkerCount = DEFAULT_WORKER_PROCESS_COUNT;
|
||||
|
||||
}
|
||||
|
||||
// 소멸자
|
||||
CServiceConfig::~CServiceConfig()
|
||||
{
|
||||
// 객체 소멸시...
|
||||
// 만약 static GetInstance() 함수가 가르키는 객체가 자기 자신이라면...
|
||||
// 소멸 처리에 의해 문제가 발생할 수 있으므로... 이에 대한 처리를 해 준다.
|
||||
if( CServiceConfig::GetInstance() == this )
|
||||
{
|
||||
CServiceConfig::m_pInstance = NULL;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
bool CServiceConfig::Init( const std::string strProgramName, const std::string & strConfFileName, std::string & strErrorMessage )
|
||||
{
|
||||
// 1. 임시 처리용 Instance 를 생성한다.
|
||||
CServiceConfig * pTempInstance = new CServiceConfig();
|
||||
|
||||
// 2. 전달 받은 Config 정보를 임시 Instance 로 로딩한다.
|
||||
// - 만약 로딩이 실패할 경우... 임시 Instance 객체를 삭제 처리한다.
|
||||
if( pTempInstance->Load( strProgramName, strConfFileName, strErrorMessage ) == false )
|
||||
{
|
||||
delete pTempInstance;
|
||||
pTempInstance = NULL;
|
||||
|
||||
return false; // 생성 실패 사유는 Error String 관련 함수를 사용하여 확인.
|
||||
}
|
||||
|
||||
// 3. 전달받은 Conf 정보 로딩에 성공한 경우...
|
||||
// - 기존 Instance 객체가 존재하는 경우.. 교체 처리..
|
||||
// - 기존 Instance 객체가 없는 경우는 신규 Instance 사용토록 처리.
|
||||
if( CServiceConfig::GetInstance() == NULL )
|
||||
{
|
||||
CServiceConfig::m_pInstance = pTempInstance;
|
||||
}
|
||||
else
|
||||
{
|
||||
// Instance replace...
|
||||
CServiceConfig * pPreviosInstance = CServiceConfig::m_pInstance;
|
||||
CServiceConfig::m_pInstance = pTempInstance;
|
||||
|
||||
delete pPreviosInstance;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
CServiceConfig * CServiceConfig::GetInstance( )
|
||||
{
|
||||
return CServiceConfig::m_pInstance;
|
||||
}
|
||||
|
||||
|
||||
// 각 항목별 제약값은 기존 service_sync 모듈 소스를 참고하여 정의함.
|
||||
bool CServiceConfig::Load( const std::string & strProgramName, const std::string & strConfFileName, std::string & strErrorMessage )
|
||||
{
|
||||
// 변수 유효성 확인
|
||||
if( strProgramName.size() <= 0 )
|
||||
{
|
||||
strErrorMessage = "Config Program Name[" + strProgramName + "] not valid.";
|
||||
return false;
|
||||
}
|
||||
|
||||
// Config class (lib 하위) 를 이용하여 config 파일 상의 정보를 로딩 처리
|
||||
Config conf;
|
||||
|
||||
if( conf.Open( strConfFileName ) == false )
|
||||
{
|
||||
strErrorMessage = "Config file[" + strConfFileName + "] open failed.";
|
||||
return false;
|
||||
}
|
||||
|
||||
// Config 모듈에 로딩된 정보 중 필요한 정보만 가져온다...
|
||||
// - 만약 오류가 발생할 경우... 해당 내역은 Error String 에 저장 처리...
|
||||
|
||||
std::string strValue;
|
||||
int nValue;
|
||||
std::vector< std::string > vecValue;
|
||||
|
||||
|
||||
// Log path
|
||||
if( GetConfigValue( conf, strProgramName, "DEFAULT_LOG_DIR", strValue, strErrorMessage ) == false )
|
||||
return false;
|
||||
else
|
||||
{
|
||||
m_strLogPath = strValue;
|
||||
strValue.clear();
|
||||
}
|
||||
|
||||
// Log Level
|
||||
if( GetConfigValue( conf, strProgramName, "LOG_LEVEL", strValue, strErrorMessage ) == false )
|
||||
return false;
|
||||
else
|
||||
{
|
||||
nValue = atoi(strValue.c_str());
|
||||
if( nValue < 0 || nValue > MAX_LOG_LEVEL )
|
||||
{
|
||||
strErrorMessage = "Config [LOG_LEVEL] value[" + strValue + "] is not valid";
|
||||
return false;
|
||||
}
|
||||
else
|
||||
m_nLogLevel = nValue;
|
||||
|
||||
strValue.clear();
|
||||
}
|
||||
|
||||
|
||||
// TCP Listen port
|
||||
if( GetConfigValue( conf, strProgramName, "TCP_LISTEN_PORT", strValue, strErrorMessage ) == false )
|
||||
return false;
|
||||
else
|
||||
{
|
||||
nValue = atoi( strValue.c_str() );
|
||||
if( nValue <= 0 || nValue > 65535 )
|
||||
{
|
||||
strErrorMessage = "Config [TCP_LISTEN_PORT] value[" + strValue + "] is not valid";
|
||||
return false;
|
||||
}
|
||||
else
|
||||
m_nTcpListenPort = nValue;
|
||||
|
||||
strValue.clear();
|
||||
}
|
||||
|
||||
// Save path
|
||||
if( GetConfigValue( conf, strProgramName, "SAVE_ROOT_PATH", strValue, strErrorMessage ) == false )
|
||||
return false;
|
||||
else
|
||||
{
|
||||
m_strSaveRootPath = strValue;
|
||||
strValue.clear();
|
||||
}
|
||||
|
||||
// WORKER_PROCESS_COUNT
|
||||
// - 정의 되지 않은 경우.. default 값 사용.
|
||||
// - 잘못 정의된 경우... 오류 처리
|
||||
if( GetConfigValue( conf, strProgramName, "WORKER_PROCESS_COUNT", strValue, strErrorMessage ) == true )
|
||||
{
|
||||
nValue = atoi( strValue.c_str() );
|
||||
if( nValue <= 0 || nValue > 100 )
|
||||
{
|
||||
strErrorMessage = "Config [WORKER_PROCESS_COUNT] value[" + strValue + "] is not valid.";
|
||||
return false;
|
||||
}
|
||||
else
|
||||
m_nWorkerCount = nValue;
|
||||
|
||||
strValue.clear();
|
||||
}
|
||||
|
||||
|
||||
// conf 정보 로딩이 모두 정상적으로 완료된 경우...
|
||||
// 마지막으로 전달받은 config 명, section 정보를 저장한다.
|
||||
m_strConfigFileName = strConfFileName;
|
||||
m_strProgramName = strProgramName;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
bool CServiceConfig::GetConfigValue( Config & conf, const std::string & strProgramName, const std::string strKey, std::string & strValue, std::string & strErrorMessage )
|
||||
{
|
||||
// conf get value
|
||||
if( conf.GetConfig( strProgramName, strKey, strValue) == false )
|
||||
{
|
||||
if( conf.GetConfig( "COMMON", strKey, strValue) == false )
|
||||
{
|
||||
strErrorMessage = "Config [" + strKey + "] value is not exist";
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
// value check.
|
||||
if( strValue.empty() == true )
|
||||
{
|
||||
strErrorMessage = "Config [" + strKey + "] value is not valid";
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
@@ -0,0 +1,97 @@
|
||||
/****************************************************************************
|
||||
cc_collectd's service config module
|
||||
-----------------------------------------
|
||||
begin : 2014/09/03
|
||||
copyright : (C) 2005 Solbox Inc.
|
||||
author : Storage Dev Team
|
||||
email : storage.sd@solbox.com
|
||||
version : 3.4
|
||||
|
||||
CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of Solbox Inc.
|
||||
*****************************************************************************/
|
||||
|
||||
#ifndef __SERVICE_CONFIG_H__
|
||||
#define __SERVICE_CONFIG_H__
|
||||
|
||||
#include <string>
|
||||
|
||||
#include "Config.h"
|
||||
|
||||
// 본 모듈은 conf 파일의 정보를 read 하여... 본 프로세스 관련 설정 정보를 저장한다.
|
||||
// 본 모듈은 singleton 으로 동작시켜 Main 프로세스 및 Worker 프로세스에서도 접근이 가능토록 한다.
|
||||
// conf 파일에 대한 파싱 및 설정 정보 추출은 lib 하위의 Config 클래스를 이용한다.
|
||||
// 본 객체는 singleton 객체이므로 상속받아 사용하지 않도록 한다. ( 상속받아 사용할 수 없도록 private 처리함 )
|
||||
|
||||
class CServiceConfig
|
||||
{
|
||||
|
||||
private:
|
||||
|
||||
// sigleton 객체 생성을 위해 생성자를 public 으로 처리하지 않음.
|
||||
CServiceConfig();
|
||||
|
||||
// 소멸자
|
||||
virtual ~CServiceConfig();
|
||||
|
||||
static CServiceConfig * m_pInstance;
|
||||
|
||||
|
||||
public:
|
||||
|
||||
// 객체 초기화 처리용 Function.
|
||||
// 본 함수 호출시 Instance 생성 후 입력 받은 Config 파일 load.
|
||||
// 따라서 본 함수를 호출하지 않을 경우.. GetInstance() 함수는 NULL 을 반환
|
||||
// 본 함수를 객체가 생성된 상태에서 재 호출할 경우
|
||||
// - 주어진 Conf 파일 정보가 정확하다면.. conf 정보를 다시 load 함.
|
||||
// - 주어진 conf 파일 정보가 부정확하다면... 이전 conf 를 계속 사용함.
|
||||
static bool Init( const std::string strProgramName, const std::string & strConfFileName, std::string & strErrorMessage );
|
||||
|
||||
// Singleton 객체 접근 메소드
|
||||
static CServiceConfig * GetInstance();
|
||||
|
||||
|
||||
private:
|
||||
|
||||
// 전달받은 conf 파일에서 해당 config 정보를 로드하여 멤버 변수에 저장한다.
|
||||
// 만약 반환값이 false 인 경우 해당 오류 관련 내역은 strErrorMessage 변수에 저장된다.
|
||||
bool Load( const std::string & strProgramName, const std::string & strConfFileName, std::string & strErrorMessage );
|
||||
|
||||
|
||||
// Config 객체로 부터 지정된 strKey 정보를 가져와 유효성 여부를 판단.
|
||||
// 반환값이 false 인 경우 strErrorMessage 상에 오류 내역이 저장된다.
|
||||
bool GetConfigValue( Config & conf, const std::string & strProgramName, const std::string strKey, std::string & strValue, std::string & strErrorMessage );
|
||||
|
||||
|
||||
private:
|
||||
|
||||
// 객체 초기화시 전달받은 정보
|
||||
std::string m_strConfigFileName; // 설정 파일 정보
|
||||
std::string m_strProgramName; // 프로그램 명 (conf 상에서 해당 프로그램의 설정 정보를 가져올 때 사용)
|
||||
|
||||
// 기본 설정 정보.
|
||||
std::string m_strLogPath; // 로그 저장 Path
|
||||
int m_nLogLevel; // 로그 기록 Level
|
||||
|
||||
// Program
|
||||
unsigned int m_nTcpListenPort; // TCP Listen port, LISTEN_TCP_PORT = 13103
|
||||
std::string m_strSaveRootPath; // 저장 경로 최상위 path , SAVE_ROOT_PATH = /backup/log
|
||||
unsigned int m_nWorkerCount; // Worker 프로세스 count, WORKER_PROCESS_COUNT = 4
|
||||
|
||||
|
||||
public: // 멤버 변수에 대한 Get 함수 선언 및 정의
|
||||
|
||||
const char * GetConfigFileName() { return m_strConfigFileName.c_str(); }
|
||||
const char * GetProgramName() { return m_strProgramName.c_str(); }
|
||||
|
||||
const char * GetLogPath() { return m_strLogPath.c_str(); }
|
||||
int GetLogLevel() { return m_nLogLevel; }
|
||||
|
||||
unsigned int GetTcpListenPort() { return m_nTcpListenPort; }
|
||||
const char * GetSaveRootPath() { return m_strSaveRootPath.c_str(); }
|
||||
unsigned int GetWorkerProcessCount() { return m_nWorkerCount; }
|
||||
|
||||
};
|
||||
|
||||
#endif /* __SERVICE_CONFIG_H__ */
|
||||
@@ -0,0 +1,265 @@
|
||||
#include "StorageStatus.h"
|
||||
|
||||
#include "Logger.h"
|
||||
|
||||
#include <errno.h>
|
||||
#include <unistd.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <vector>
|
||||
#include <mntent.h>
|
||||
#include <sys/vfs.h>
|
||||
|
||||
|
||||
// Storage Check 주기 ( sec )
|
||||
#define STORAGE_CHECK_INTERVAL 10
|
||||
|
||||
// 최소 보존 Storage 용량. ( 10G )
|
||||
#define LIMIT_STORAGE_SIZE 10737418240LL
|
||||
|
||||
|
||||
// 시스템에서 추출된 Mount 정보를 저장하기 위한 struct...
|
||||
struct systme_mount_info {
|
||||
|
||||
std::string mount_name; // Mount Name..
|
||||
unsigned long long free_byte; // free byte
|
||||
};
|
||||
|
||||
|
||||
|
||||
// 생성자
|
||||
CStorageStatus::CStorageStatus( std::string strRootPath )
|
||||
{
|
||||
// 멤버변수 초기화
|
||||
m_nFreeSize = 0;
|
||||
m_threadHandle = 0;
|
||||
|
||||
// Mutex Lock init.
|
||||
pthread_mutex_init( &m_mutexLock, NULL );
|
||||
|
||||
// 경로값에서 마지막 문자가 '/' 인 경우.. 제거 처리한다.
|
||||
// - 업무 Flow 관련 경로 처리 문제 사전 해결
|
||||
// - 시스템에서 추출된 mount 정보와 비교 처리시 정상 비교가 될 수 있도록..
|
||||
if( strRootPath.length() > 1 )
|
||||
{
|
||||
if( strRootPath.at( strRootPath.length() - 1 ) == '/' )
|
||||
{
|
||||
m_strRootPath = strRootPath.substr( 0, strRootPath.length() - 1 );
|
||||
}
|
||||
else
|
||||
m_strRootPath = strRootPath;
|
||||
}
|
||||
else
|
||||
m_strRootPath = strRootPath;
|
||||
}
|
||||
|
||||
// 소멸자
|
||||
CStorageStatus::~CStorageStatus()
|
||||
{
|
||||
pthread_mutex_destroy( &m_mutexLock );
|
||||
|
||||
// Thread 동작 정지 처리
|
||||
// - 만약 Thread 가 이미 종료된 경우 m_threadHandle 이 다른 Thread Handle 일 수 있으므로
|
||||
// 업무 Flow 수정시 주의할 것
|
||||
if( m_threadHandle != 0 )
|
||||
pthread_cancel( m_threadHandle );
|
||||
}
|
||||
|
||||
// Thread 기동 함수
|
||||
// - storage 정보 추출 시작.
|
||||
bool CStorageStatus::Start()
|
||||
{
|
||||
int nRet = ::pthread_create( &m_threadHandle, NULL, CStorageStatus::threadFunc, this );
|
||||
if( nRet != 0 )
|
||||
{
|
||||
// Thread 생성 실패시
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Thread create failed.[%d][%s]", errorNum, strerror( errorNum ) );
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// Available size (byte) 정보를 반환한다.
|
||||
// - 실제 Free 용량에서 최소 보존 용량을 뺀 나머지 값을 반환한다.
|
||||
unsigned long long CStorageStatus::GetAvailableSize()
|
||||
{
|
||||
unsigned long long result;
|
||||
|
||||
pthread_mutex_lock( &m_mutexLock );
|
||||
|
||||
if( m_nFreeSize <= LIMIT_STORAGE_SIZE )
|
||||
result = 0;
|
||||
else
|
||||
result = m_nFreeSize - LIMIT_STORAGE_SIZE;
|
||||
|
||||
pthread_mutex_unlock( &m_mutexLock );
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
void CStorageStatus::SetFreeSize( unsigned long long value )
|
||||
{
|
||||
pthread_mutex_lock( &m_mutexLock );
|
||||
|
||||
m_nFreeSize = value;
|
||||
|
||||
pthread_mutex_unlock( &m_mutexLock );
|
||||
|
||||
if( value < ( LIMIT_STORAGE_SIZE * 2 ) )
|
||||
{
|
||||
_LOG( LINF, "Worker[%d]: CStorageStatus free size [%s][%llu]", getpid(), m_strRootPath.c_str(), value );
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
// 스레드 함수
|
||||
void* CStorageStatus::threadFunc( void* arg )
|
||||
{
|
||||
CStorageStatus * pObject = reinterpret_cast<CStorageStatus *>( arg );
|
||||
pthread_detach( pthread_self() );
|
||||
|
||||
while( 1 )
|
||||
{
|
||||
pObject->Execute();
|
||||
|
||||
sleep( STORAGE_CHECK_INTERVAL );
|
||||
}
|
||||
|
||||
// Thread 종료시 m_threadHandle 값을 초기화 처리.
|
||||
// - 소멸자에서 잘못된 Thread 를 종료할 수 있기 때문.
|
||||
pObject->m_threadHandle = 0;
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
// 실제 Free Size 추출 수행 함수.
|
||||
void CStorageStatus::Execute()
|
||||
{
|
||||
// 1. 장비의 Mount 정보를 추출한다.
|
||||
std::vector<struct systme_mount_info> vecMount;
|
||||
|
||||
struct mntent stMnt;
|
||||
struct mntent * pMnt;
|
||||
char tempBuf[512];
|
||||
|
||||
FILE * fp = setmntent( "/etc/mtab", "r" );
|
||||
if( fp != NULL )
|
||||
{
|
||||
// 임시 변수
|
||||
struct systme_mount_info stMountInfo;
|
||||
|
||||
while( ( pMnt = getmntent_r( fp, &stMnt, tempBuf, sizeof( tempBuf ) ) ) != NULL )
|
||||
{
|
||||
struct statfs statFsBuf;
|
||||
|
||||
if( ( stMnt.mnt_dir != NULL ) && ( statfs( stMnt.mnt_dir, &statFsBuf ) == 0 ) )
|
||||
{
|
||||
stMountInfo.mount_name = stMnt.mnt_dir;
|
||||
stMountInfo.free_byte = statFsBuf.f_bavail * ( unsigned long long )statFsBuf.f_bsize;
|
||||
|
||||
vecMount.push_back( stMountInfo );
|
||||
|
||||
//LOG( LDBG, "system mount name[%s] free byte[%llu]", stMountInfo.mount_name.c_str(), stMountInfo.free_byte );
|
||||
}
|
||||
}
|
||||
|
||||
// open 된 fp close 처리.
|
||||
endmntent( fp );
|
||||
}
|
||||
|
||||
if( vecMount.empty() == true )
|
||||
{
|
||||
// 추출된 결과가 없는 경우...
|
||||
SetFreeSize( 0 );
|
||||
return;
|
||||
}
|
||||
|
||||
// 2. 최상위 경로 정보와 동일한 Mount 정보가 존재하는지 확인
|
||||
std::vector<struct systme_mount_info>::iterator it;
|
||||
for( it = vecMount.begin(); it < vecMount.end(); it++ )
|
||||
{
|
||||
// 동일한 경로가 존재하는 경우.
|
||||
if( strcmp( m_strRootPath.c_str(), it->mount_name.c_str() ) == 0 )
|
||||
{
|
||||
SetFreeSize( it->free_byte );
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
// 3. 위의 2번 항목에서 mount 정보와 동일한 경로를 찾지 못한 경우...
|
||||
// - 현재 경로의 Upper path 를 추출하여.. 추출된 Mount 경로와 비교 처리.
|
||||
std::string strCurrentPath = m_strRootPath;
|
||||
std::string strUpperPath;
|
||||
|
||||
while( 1 )
|
||||
{
|
||||
GetUpperPath( strCurrentPath, strUpperPath );
|
||||
if( strUpperPath.empty() == true )
|
||||
{
|
||||
// 추출된 상위 경로가 존재하지 않는 경우.
|
||||
// - 해당 경로의 Mount 정보 추출 실패시 오류 로깅 및 0 으로 설정 처리
|
||||
LOG( LERR, "CStorageStatus mount info search fail. path[%s]", m_strRootPath.c_str() );
|
||||
SetFreeSize( 0 );
|
||||
return;
|
||||
}
|
||||
|
||||
// 추출된 상위 경로에 대해 비교 mount 정보 비교 처리.
|
||||
for( it = vecMount.begin(); it < vecMount.end(); it++ )
|
||||
{
|
||||
// 동일한 경로가 존재하는 경우.
|
||||
if( strcmp( strUpperPath.c_str(), it->mount_name.c_str() ) == 0 )
|
||||
{
|
||||
SetFreeSize( it->free_byte );
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
// 만약 추출된 상위 경로와 맞는 Mount 정보가 없는 경우...
|
||||
// - 더 최상위 경로를 추출하여 재시도.
|
||||
strCurrentPath = strUpperPath;
|
||||
}
|
||||
|
||||
// 4. 최종 동일한 mount 정보를 찾지 못한 경우...
|
||||
// - 0 으로 설정 처리.
|
||||
LOG( LERR, "CStorageStatus mount info full search fail. path[%s]", m_strRootPath.c_str() );
|
||||
SetFreeSize( 0 );
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
// 전달받은 경로로부터.. 상위 Path 정보를 추출하여 전달한다.
|
||||
// 오류 발생시 [out] 값인 strUpperPath 가 empty 된다.
|
||||
void CStorageStatus::GetUpperPath( const std::string & strSourcePath, std::string & strUpperPath )
|
||||
{
|
||||
// 인자로 전달되는 strSourcePath 끝에.. '/' 가 이미 제거되었다고 판단하고 작업을 수행한다.
|
||||
std::string::size_type pos = strSourcePath.rfind( '/' );
|
||||
if( pos == std::string::npos )
|
||||
{
|
||||
// 찾지 못한 경우...
|
||||
// out 값을 empty 처리시킨다.
|
||||
LOG( LNOT, "upper path not found. source[%s]", strSourcePath.c_str() );
|
||||
strUpperPath.clear();
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
// 찾은 경우..
|
||||
strUpperPath = strSourcePath.substr( 0, pos );
|
||||
|
||||
// 추출된 경로가... 최상위 '/' 인 경우.. 유효하지 않으므로 오류로 처리한다.
|
||||
if( strcmp( strUpperPath.c_str(), "/" ) == 0 )
|
||||
{
|
||||
LOG( LNOT, "upper path not vaild that root path. [%s]", strUpperPath.c_str() );
|
||||
strUpperPath.clear();
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
@@ -0,0 +1,78 @@
|
||||
/***************************************************************************
|
||||
Storage 상태 정보를 주기적으로 check.
|
||||
-----------------------------------------
|
||||
begin : 2014/09/24
|
||||
copyright : (C) 2005 Solbox Inc.
|
||||
author : storage dev team
|
||||
email : storage.sd@solbox.com
|
||||
version : 3.4
|
||||
|
||||
CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of Solbox Inc.
|
||||
***************************************************************************/
|
||||
|
||||
#ifndef __STORAGE_STATUS_H__
|
||||
#define __STORAGE_STATUS_H__
|
||||
|
||||
|
||||
#include <pthread.h>
|
||||
#include <string>
|
||||
|
||||
|
||||
// CStorageStatus
|
||||
// 본 클래스는 Thread 클래스로서...
|
||||
// 주기적으로 지정된 경로의 mount 에 대해 유효 사용 가능 용량을 확인하여
|
||||
// 멤버 변수에 저장처리한다.
|
||||
// 저장된 정보는 다른 모듈에서 사용 가능토록 interface 함수를 제공한다.
|
||||
class CStorageStatus
|
||||
{
|
||||
public:
|
||||
|
||||
// 생성자
|
||||
CStorageStatus( std::string strRootPath );
|
||||
|
||||
// 소멸자
|
||||
~CStorageStatus();
|
||||
|
||||
// Thread 기동 함수
|
||||
// - storage 정보 추출 시작.
|
||||
bool Start();
|
||||
|
||||
// Available size (byte) 정보를 반환한다.
|
||||
// - 실제 Free 용량에서 최소 보존 용량을 뺀 나머지 값을 반환한다.
|
||||
unsigned long long GetAvailableSize();
|
||||
|
||||
|
||||
private:
|
||||
|
||||
// 생성자를 통해 전달받은 최상위 경로 정보 저장
|
||||
std::string m_strRootPath;
|
||||
|
||||
// 해당 경로의 Free Size (byte)
|
||||
unsigned long long m_nFreeSize;
|
||||
|
||||
// 쓰레드 핸들
|
||||
pthread_t m_threadHandle;
|
||||
|
||||
// m_nFreeSize 에 대한 동시 접근을 제어하기 위한 Lock
|
||||
pthread_mutex_t m_mutexLock;
|
||||
|
||||
|
||||
private:
|
||||
|
||||
// 스레드 함수
|
||||
static void* threadFunc( void* arg );
|
||||
|
||||
// 실제 Free Size 추출 수행 함수.
|
||||
void Execute( void );
|
||||
|
||||
// 멤버 변수 m_nFreeSize 의 값을 설정
|
||||
void SetFreeSize( unsigned long long value );
|
||||
|
||||
// 전달받은 경로로부터.. 상위 Path 정보를 추출하여 전달한다.
|
||||
void GetUpperPath( const std::string & strSourcePath, std::string & strUpperPath );
|
||||
|
||||
};
|
||||
|
||||
#endif /* __STORAGE_STATUS_H__ */
|
||||
@@ -0,0 +1,190 @@
|
||||
#include "Worker.h"
|
||||
|
||||
#include "ProcessRename.h"
|
||||
#include "ServiceConfig.h"
|
||||
#include "Logger.h"
|
||||
|
||||
#include "StorageStatus.h"
|
||||
#include "ProcessStatus.h"
|
||||
#include "ClientThread.h"
|
||||
|
||||
#include <unistd.h>
|
||||
#include <signal.h>
|
||||
#include <netinet/in.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/socket.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <errno.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <stdio.h>
|
||||
#include <time.h>
|
||||
|
||||
|
||||
// Worker Process 의 main 함수
|
||||
int WorkerMain( int listenSocket )
|
||||
{
|
||||
// Process Rename
|
||||
set_ps_display(PROG_NAME": Worker [initialize process]", false);
|
||||
|
||||
// Set Signal Handler
|
||||
SetSignalWorker();
|
||||
|
||||
// 로그 출력 순서가 Main 프로세스와 꼬일 수 있어... 1초간 대기 후 진행
|
||||
sleep( 1 );
|
||||
|
||||
|
||||
|
||||
// 공통 모듈 생성 및 기동
|
||||
// Storage 상태 정보 추출 모듈 생성 및 기동
|
||||
CStorageStatus storageStatus( CServiceConfig::GetInstance()->GetSaveRootPath() );
|
||||
if( storageStatus.Start() == false )
|
||||
{
|
||||
LOG( LERR, "Worker[%d]: Storage status module start fail. => Worker Exit", getpid() );
|
||||
return EXIT_FAILURE; // EXIT_FAILURE 를 반환하여 재생성 처리 방지
|
||||
}
|
||||
//_LOG( LINF, "Worker[%d]: CStorageStatus module start.", getpid() );
|
||||
|
||||
// Process status 모듈 생성 및 기동
|
||||
CProcessStatus processStatus;
|
||||
if( processStatus.Start() == false )
|
||||
{
|
||||
LOG( LERR, "Worker[%d]: Process status module start fail. => Worker Exit", getpid() );
|
||||
return EXIT_FAILURE; // EXIT_FAILURE 를 반환하여 재생성 처리 방지
|
||||
}
|
||||
//_LOG( LINF, "Worker[%d]: CProcessStatus module start.", getpid() );
|
||||
|
||||
|
||||
|
||||
// Worker 프로세스의 작업 수행.
|
||||
// Listen Socket 에 대한 Accept 대기 수행
|
||||
int clientSocket;
|
||||
struct sockaddr_in clientSockAddr;
|
||||
socklen_t clientSockLen = sizeof( clientSockAddr );
|
||||
char szClientIp[INET_ADDRSTRLEN];
|
||||
|
||||
CClientThread * pClientThread;
|
||||
|
||||
|
||||
while( 1 )
|
||||
{
|
||||
clientSocket = accept( listenSocket, ( struct sockaddr * ) &clientSockAddr, &clientSockLen );
|
||||
if( clientSocket == -1 )
|
||||
{
|
||||
// 오류발생시 해당 내역 로깅처리.
|
||||
int errorNum = errno;
|
||||
LOG( LERR, "Worker: client accept failed. [%d][%s]", errorNum, strerror( errorNum ) );
|
||||
}
|
||||
else
|
||||
{
|
||||
// 정상적인 Client 인 경우...
|
||||
|
||||
// IP 정보 추출.. 실패시 0.0.0.0 으로 설정.
|
||||
memset( szClientIp, 0x00, sizeof( szClientIp ) );
|
||||
if( inet_ntop( AF_INET, &( clientSockAddr.sin_addr ), szClientIp, sizeof( szClientIp ) ) == NULL )
|
||||
{
|
||||
// 변환 실패시.. 0.0.0.0 으로 설정 처리.
|
||||
snprintf( szClientIp, INET_ADDRSTRLEN - 1, "0.0.0.0" );
|
||||
}
|
||||
|
||||
// client 를 처리할 Thread 객체 생성 후 기동 처리.
|
||||
// - 해당 객체에서 thread 종료시 자동 delete 처리하므로.. 본 함수에서는 new 생성만 처리한다.
|
||||
pClientThread = NULL;
|
||||
pClientThread = new CClientThread( &storageStatus, &processStatus, clientSocket, szClientIp );
|
||||
if( pClientThread == NULL || pClientThread->Start() == false )
|
||||
{
|
||||
// Thread 기동 실패시..
|
||||
if( pClientThread != NULL )
|
||||
{
|
||||
LOG( LERR, "client[%s] thread object start function fail.", szClientIp );
|
||||
|
||||
delete pClientThread;
|
||||
pClientThread = NULL;
|
||||
}
|
||||
else
|
||||
{
|
||||
// Thead 객체 생성 실패시
|
||||
LOG( LERR, "client[%s] thread object create fail.", szClientIp );
|
||||
}
|
||||
|
||||
close( clientSocket );
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
} // while( 1 )
|
||||
|
||||
|
||||
// Process 종료 처리.
|
||||
_LOG( LINF, "Worker[%d] exit.. Good Bye..", getpid());
|
||||
|
||||
//잠시 대기 후 종료처리.
|
||||
|
||||
// CHG 2015-09-16 huibong
|
||||
// - #24436 usleep -> nanosleep 으로 변경 처리.
|
||||
struct timespec sleep;
|
||||
sleep.tv_sec = 0;
|
||||
sleep.tv_nsec = 500000000; // 0.5 sec
|
||||
nanosleep( &sleep, NULL );
|
||||
|
||||
// fork 를 수행한 함수에서 exit 함수 호출을 통한 종료처리
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
|
||||
|
||||
// Worker Process 종료 Signal 을 전달받은 경우 이를 처리하기 위한 함수.
|
||||
// @param nSignalNumber [in] 발생한 시그널 Number
|
||||
// @return void
|
||||
static void SignalWorkerTerminate(int nSignalNumber)
|
||||
{
|
||||
// Signal Number 에 따른 로깅처리.
|
||||
if( nSignalNumber == SIGTERM )
|
||||
{
|
||||
_LOG( LINF, "Worker[%d] process exit by user signal [SIGTERM]. Good Bye..", getpid() );
|
||||
}
|
||||
else
|
||||
{
|
||||
_LOG( LWAR, "Worker[%d] process exit by abnormal signal[%d], Good Bye..", getpid(), nSignalNumber );
|
||||
}
|
||||
|
||||
// CHG 2015-09-16 huibong
|
||||
// - #24436 usleep -> nanosleep 으로 변경 처리.
|
||||
struct timespec sleep;
|
||||
sleep.tv_sec = 0;
|
||||
sleep.tv_nsec = 500000000; // 0.5 sec
|
||||
nanosleep( &sleep, NULL );
|
||||
|
||||
exit( EXIT_SUCCESS );
|
||||
}
|
||||
|
||||
// Worker 프로세스 signal 처리 설정을 위한 함수
|
||||
// @return void
|
||||
void SetSignalWorker( void )
|
||||
{
|
||||
sigset_t set;
|
||||
struct sigaction act;
|
||||
memset(&act, 0x00, sizeof(act));
|
||||
|
||||
sigfillset( &set );
|
||||
sigprocmask( SIG_SETMASK, &set, NULL );
|
||||
sigfillset( &act.sa_mask );
|
||||
|
||||
/* 무시할 신호 목록 */
|
||||
act.sa_handler = SIG_IGN;
|
||||
sigaction( SIGPIPE, &act, NULL); /* desciptor 오류 발생시 Process가 죽는 것은 방지하기 위하여 설정 */
|
||||
sigaction( SIGHUP, &act, NULL); /* process를 기동시킨 관리자의 로그아웃시, 또는 config reload 등의 처리 signal*/
|
||||
sigaction( SIGINT, &act, NULL); /* ^C 키를 누른 경우 받는 신호 => demon 으로 기동되기 땜시 이 신호 못받음 */
|
||||
sigaction( SIGQUIT, &act, NULL); /* 키보드에 의한 Abort 신호 처리 => ? */
|
||||
|
||||
// Worker Controller 의 child process 종료에 대한 처리기 설정.
|
||||
//act.sa_handler = SIG_DFL; // 외부 프로세스 수행이 발생할 수 있는 경우 default 처리
|
||||
act.sa_handler = SIG_IGN; // 외부 프로세스 수행이 없는 경우.. SIG_IGN 처리
|
||||
sigaction( SIGCHLD, &act, NULL);
|
||||
|
||||
/* 각종 에러나 사용자의 종료 신호 처리 */
|
||||
act.sa_handler = SignalWorkerTerminate;
|
||||
sigaction( SIGTERM, &act, NULL); /* kill -TERM 에 의한 프로세스 종료시 */
|
||||
|
||||
sigemptyset(&set); /* 신호 처리기 처리 설정 위한 블록 해제 */
|
||||
sigprocmask(SIG_SETMASK, &set, NULL);
|
||||
}
|
||||
@@ -0,0 +1,45 @@
|
||||
/***************************************************************************
|
||||
cc_collectd Worker Header ( Worker.h )
|
||||
-----------------------------------------
|
||||
begin : 2014/09/03
|
||||
copyright : (C) 2005 Solbox Inc.
|
||||
author : Storage Dev Team
|
||||
email : storage.sd@solbox.com
|
||||
version : 3.4
|
||||
|
||||
CopyRight(C) 2005 Solbox Inc. All Rights reserved.
|
||||
Redistribution and use in source and binary forms, with or with out
|
||||
modification, are not permitted in outside of Solbox Inc.
|
||||
***************************************************************************/
|
||||
|
||||
#ifndef __CC_COLLECTD_WORKER_H__
|
||||
#define __CC_COLLECTD_WORKER_H__
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Worker Process 의 main 함수
|
||||
// @param listenSocket command, data 송수신을 위한 TCP Listen socket
|
||||
// @return EXIT_FAILURE 모듈 초기화 실패시
|
||||
// EXIT_SUCCESS 그 외 정상 종료
|
||||
int WorkerMain( int listenSocket );
|
||||
|
||||
// Worker 프로세스 signal 처리 설정을 위한 함수
|
||||
// @return void
|
||||
void SetSignalWorker( void );
|
||||
|
||||
|
||||
/* Signal 처리를 위한 각 Signal Handler 함수는
|
||||
* 다른 Code 에서 Include 처리시 Static 관련 문제로 인해
|
||||
* 본 Header 에서 선언처리 하지 않음. cpp 에만 존재
|
||||
*/
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
|
||||
#endif /* __CC_COLLECTD_WORKER_H__ */
|
||||
|
||||
|
||||
Reference in New Issue
Block a user