c socket server close 시 클라이언트 종료 감지 문의

#include "libDefine.h"
 
void signalHandlerServer(int nSignal);
 
int g_nServerSocket = -1;								/* server socket file descriptor */
int g_nClientSocket = -1;								/* client socket file descriptor */
 
int main(int argc, char *argv[])
{
	int nPort = -1;
	char szMsg[SIZEBUFFER + 1] = {0,};			/* 수신용 버퍼 */
	int nReadMessage = 0;								/* 받은 메시지 길이 저장 변수 */
	struct sockaddr_in stClientAddr = {0,};	/* 클라이언트 주소 저장할 구조체 */
	unsigned int uClientAddrSize = sizeof(stClientAddr);			/* 클라이언트 주소 구조체의 크기 */
	int nWriteMessage = -1;
 
	/* signal 처리 */
	signal(SIGINT, signalHandlerServer);
 
	if (argc != 2)
	{
		printf("Usage : %s <port>\n", argv[0]);
		exit(1);
	}
 
	nPort = atoi(argv[1]);
 
	printf("서버 연결 중입니다.\n");
	g_nServerSocket = serverSocket(nPort);
 
	if(g_nServerSocket < 0)
	{
		errHandle(errno, "socket() error \n");
		exit(1);
	}
 
	while(1)
	{
		g_nClientSocket = accept(g_nServerSocket, (struct sockaddr*) &stClientAddr, &uClientAddrSize);
 
		if(g_nClientSocket < 0)
		{
			errHandle(errno, "accept() 클라이언트로의 접속 요청이 수락되지 않았습니다.\n");
			g_nClientSocket = -1;
			sleep(1);
		}
		else
		{
			printf("클라이언트 접속 IP : %s\n", inet_ntoa(stClientAddr.sin_addr));
			printf("accept() 클라이언트로 접속 요청 수락\n");
		}
 
		while((nReadMessage = read(g_nClientSocket, szMsg, SIZEBUFFER)) > 0)
		{
			printf("클라이언트로 받은 메시지 %s\n", szMsg);
 
			nWriteMessage = write(g_nClientSocket, szMsg, nReadMessage);
 
			if(nWriteMessage < 0)
			{
				errHandle(errno, "write error \n");
				break;
			}
			memset(szMsg, 0, sizeof(szMsg));
		}
 
		close(g_nClientSocket);
		g_nClientSocket = -1;
		printf("연결이 끊어졌습니다. \n");
		sleep(1);
	}
	close(g_nServerSocket);
	g_nServerSocket = -1;
	return 0;
}
 
void signalHandlerServer(int nSignal)
{
	char szMessage[SIZEBUFFER] = {0,};
 
	if (nSignal == SIGINT)
	{
		fputs("종료하시겠습니까?(y 입력) : ", stdout);
		fgets(szMessage, SIZEBUFFER, stdin);
		if(!strcasecmp(szMessage, "y\n"))
		{
			if(g_nClientSocket >= 0)
			{
				close(g_nClientSocket);
				g_nClientSocket = -1;
				printf("client socket 종료되었습니다.\n");
				exit(0);
			}
 
			if(g_nServerSocket >= 0)
			{
				close(g_nServerSocket);
				g_nServerSocket = -1;
				printf("server socket 종료되었습니다.\n");
				exit(0);
			}
		}
	}
}

#include "libDefine.h"
 
void signalHandlerClient(int nSignal);
 
int g_nClientSocket = -1;
 
int main(int argc, char *argv[])
{
	int nPort = -1;
	char *pszAddr = NULL;					/* ip 문자 입력받을 포인터 변수 */
	char szMsg[SIZEBUFFER + 1] = {0,};	/* 서버에 보낼 메시지를 저장할 문자열 버퍼 */
	int nReadMessage = 0;					/* 송수신 메시지의 문자열 길이 */
	int nWriteMessage = 0;
 
	signal(SIGINT, signalHandlerClient);
 
	/* port */
	if (argc != 3)
	{
		printf("usage : %s <IP> <port> \n", argv[0]);
		exit(1);
	}
 
	pszAddr = argv[1];
	nPort = atoi(argv[2]);
 
	g_nClientSocket = clientSocket(nPort, pszAddr);
 
	if(g_nClientSocket < 0)
	{
		errHandle(errno, "socket() error \n");
		exit(1);
	}
 
	while (1)
	{
		memset(szMsg, 0, sizeof(szMsg));
 
		fputs("입력 메시지(Q는 종료, 버퍼 사이즈까지 입력할 수 있습니다.) : ", stdout);
		fgets(szMsg, SIZEBUFFER, stdin);
 
		if(!strcasecmp(szMsg, "q\n"))
		{
			close(g_nClientSocket);
			g_nClientSocket = -1;
			break;
		}
 
		if((nWriteMessage = write(g_nClientSocket, szMsg, strlen(szMsg))) < 0)
		{
			errHandle(errno, "write error \n");
			close(g_nClientSocket);
			g_nClientSocket = -1;
			exit(1);
		}
 
		if((nReadMessage = read(g_nClientSocket, szMsg, SIZEBUFFER)) < 0)
		{
			errHandle(errno, "read error \n");
			close(g_nClientSocket);
			g_nClientSocket = -1;
			exit(1);
		}
 
		szMsg[nReadMessage] = 0;
		printf("서버로 받은 메시지 : %s\n", szMsg);
	}
	return 0;
}
 
void signalHandlerClient(int nSignal)
{
	char szMessage[SIZEBUFFER] = {0,};
 
	if (nSignal == SIGINT)
	{
		fputs("종료하시겠습니까?(y 입력) : ", stdout);
		fgets(szMessage, SIZEBUFFER, stdin);
		if(!strcasecmp(szMessage, "y\n"))
		{
			if(g_nClientSocket >= 0)
			{
				close(g_nClientSocket);
				g_nClientSocket = -1;
				printf("연결 종료되었습니다.\n");
				exit(0);
			}
		}
	}
}

#include "libDefine.h"
 
int serverSocket(int nPort)
{
	int nServerSocket = -1;
	struct sockaddr_in stServerAddr = {0,};	/* 서버 주소 저장할 구조체  */
 
	if(nPort < 0)
	{
		errHandle(errno, "port error");
		exit(1);
	}
 
	if(nPort)
 
	/* TCP 통신용 서버 소켓 생성 */
	nServerSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
 
	if (nServerSocket < 0)
	{
		errHandle(errno, "socket() error");
		exit(1);
	}
	else
	{
		printf("socket() TCP 통신용 서버 소켓이 생성되었습니다.\n");
	}
 
	memset(&stServerAddr, 0, sizeof(stServerAddr));
	stServerAddr.sin_family = AF_INET;
	stServerAddr.sin_addr.s_addr = htonl(INADDR_ANY);
	stServerAddr.sin_port = htons(nPort);
 
	/* bind */
	if (bind(nServerSocket, (struct sockaddr*) &stServerAddr, sizeof(stServerAddr)) < 0)
	{
		errHandle(errno, "bind() error %d\n", nPort);
		close(nServerSocket);
		nServerSocket = -1;
		exit(1);
	}
	else
	{
		printf("bind() - socket에 주소와 IP 할당되었습니다.\n");
	}
 
	/* 소켓을 서버용으로 사용할 수가 있게 한다. */
	if (listen(nServerSocket, LISTENQ) < 0)
	{
		errHandle(errno, "listen() error %d\n", nPort);
		close(nServerSocket);
		nServerSocket = -1;
		exit(1);
	}
	else
	{
		printf("listen() 연결 요청 대기하는 중입니다.\n");
	}
 
	return nServerSocket;
}
 
int clientSocket(int nPort, char *pszAddr)
{
	int nClientSocket = -1;
	struct sockaddr_in stServerAddr = {0,};			/* 접속할 서버의 주소 저장할 구조체 */
 
	if(nPort < 0)
	{
		errHandle(errno, "port error");
		exit(1);
	}
 
	if(pszAddr == NULL)
	{
		errHandle(errno, "address error");
		exit(1);
	}
 
	nClientSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
 
	if(nClientSocket < 0)
	{
		errHandle(errno, "socket() error");
		exit(1);
	}
 
	/* 서버 쪽 주소 구조체 0 초기화 */
	memset(&stServerAddr, 0, sizeof(stServerAddr));
	stServerAddr.sin_family = AF_INET;								/* 인터넷 주소체계 사용 */
	stServerAddr.sin_addr.s_addr = inet_addr(pszAddr);			/* 서버 IP 구조체에 저장 */
	stServerAddr.sin_port = htons(nPort);							/* 서버 Port 구조체에 저장 */
 
	/* connect */
	if (connect(nClientSocket, (struct sockaddr*) &stServerAddr, sizeof(stServerAddr)) < 0)
	{
		errHandle(errno, "connect() error %s\n", pszAddr);
		close(nClientSocket);
		nClientSocket = -1;
		exit(1);
	}
	else
	{
		printf("연결되었습니다.\n");
	}
	return nClientSocket;
}
 
/* error 처리 handler */
void errHandle(int nErrorCode, char *pszMsg, ...)
{
	va_list argptr;
	char szBuf[MESSAGE] = {0,};
 
	if(nErrorCode)
	{
		printf("error = %s\n", strerror(errno));
	}
 
	if(pszMsg == NULL)
	{
		printf("message error = %s\n", strerror(errno));
		exit(1);
	}
 
	va_start(argptr, pszMsg);
	vsnprintf(szBuf, MESSAGE, pszMsg, argptr);		/* 버퍼에 문자열 쓰고 버퍼 길이 지정하며 va_list 얻는다. */
	va_end(argptr);
	puts(szBuf);
}