/*
 * echo_client2.c
 * Written by SW. YOON.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/socket.h>

void error_handling(char *message);

int main(int argc, char **argv)
{
	int sock;
	char message[30];
	int str_len, recv_len, recv_num;
	struct sockaddr_in serv_addr;
	
	if(argc!=3){
		printf("Usage : %s <IP> <port>\n", argv[0]);
		exit(1);
	}
	
	sock=socket(PF_INET, SOCK_STREAM, 0);   
	if(sock == -1)
		error_handling("socket() error");
	
	memset(&serv_addr, 0, sizeof(serv_addr));
	serv_addr.sin_family=AF_INET;
	serv_addr.sin_addr.s_addr=inet_addr(argv[1]);
	serv_addr.sin_port=htons(atoi(argv[2]));
	
	if( connect(sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr))==-1 )
		error_handling("connect() error!");
	
	while(1)
	{     
		fputs("Àü¼ÛÇÒ ¸Þ½ÃÁö¸¦ ÀÔ·Â Çϼ¼¿ä (q to quit) : ", stdout);
		fgets(message, sizeof(message), stdin);
		
		if(!strcmp(message,"q\n")) break;
		str_len=write(sock, message, strlen(message)); /* ¸Þ½ÃÁö Àü¼Û */
		
		for(recv_len=0; recv_len<str_len; ) /* ¸Þ½ÃÁö ¼ö½Å */
		{  
			recv_num=read(sock, &message[recv_len], str_len-recv_len); //º¸³½ ¹®ÀÚ°¡ 10°³ÀÇ ¹®ÀÚ¸é 10°³ÀÇ ¹®ÀÚ°¡ 1°³¾¿ µé¾î¿À´ÂÁö ¹¶Ãļ­ µé¾î¿À´ÂÁöº¸°íÀÖ´Ù. 
 					//¸Æ½Ã¸Ø»çÀÌÁ 10°³·ÎÁáÀ»¶§ Çѹø¿¡ µé¾î¿À´ÂÁö ¾Æ´Ï¸é ´õ¸¹ÀÌ µé¾î¿À´ÂÁö °Ë»çÇϴ·çƾ
			if(recv_num==-1)
				error_handling("read() error!");
			recv_len+=recv_num;
		}
		message[str_len]=0;
		printf("¼­¹ö·ÎºÎÅÍ Àü¼ÛµÈ ¸Þ½ÃÁö : %s \n", message);
	}
	
	close(sock);
	return 0;
}

void error_handling(char *message)
{
	fputs(message, stderr);
	fputc('\n', stderr);
	exit(1);
}