우분투 리눅스 채팅 프로그램

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<netinet/in.h>
#include<pthread.h>
 
#define COUNT 10//최대 접속 인원 수
#define MAX_LINE 530//최대 채팅길이
#define MAX_LIST 100//저장 가능한 메시지갯수
 
int clnt_number = 0; //접속자 인원 수
int clnt_sock_arr[COUNT]; //접속자를 각 소켓에 저장하기 위한 정수형 배열
int sockfd;//소켓
int list_count;//저장된 메시지 갯수
unsigned long int acc_sock;//연결된 소켓
char admin[MAX_LINE];//관리자 메시지
char add_admin[MAX_LINE];//관리자 명 + 메시지
char list[MAX_LIST][MAX_LINE];//메시지 내용
 
void* chat_input(void* a);//메시지를 읽는 쓰레드
void* admin_chat(void* b);//관리자 기능 및 안내 쓰레드
void exitchat(int i);//사용자 종료 명령어 확인 함수
void write_msg(char* msg, int len);//서버 --> 클라이언트 메시지 송신 함수
void input_list(void);//메시지 최대 갯수 초과 시 정렬 함수
void output_list(int number);//저장된 메시지 출력 함수
 
int main(int argc, char *argv[])
{
	struct sockaddr_in servaddr, cliaddr;
	socklen_t addrlen = sizeof(cliaddr); // 서버 구조체 선언
 
	pthread_t a_thread;//쓰레드 선언
	pthread_t b_thread;//쓰레드 선언
 
	if(argc != 2) // 메인함수에 들어온 매개변수에 따라 잘못 입력 시 사용법 출력 후 종료
	{
		printf("Usage : %s port \n", argv[0]);
		return -1;
	}
 
	if((sockfd = socket(PF_INET, SOCK_STREAM, 0)) < 0) // 소켓 함수로 소켓 생성, 오류 시 오류 출력
	       	// make socket and error cheaking
	{
		perror("Socket failed!");
		return -1;
	}
 
	/* 구조체를 이용하여 서버의 아이피 주소와 포트 그리고 프로토콜을 설정, 관리 */
	servaddr.sin_family = AF_INET; // 통신 프로토콜 설정  -> ipv4
	servaddr.sin_port = htons(atoi(argv[1])); // 입력된 포트 번호 컴퓨터 인식 가능하도록 처리 (htons 함수 참조)
	servaddr.sin_addr.s_addr = htonl(INADDR_ANY); // ip 주소 컴퓨터 인식 가능하도록 처리 (htonl 함수 참조)
 
	if(bind(sockfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0) // 바인드 함수, 오류 시 오류 출력
	{
		perror("Bind failed !");
		return -1;
 
	}
 
	listen(sockfd, COUNT); // COUNT에서 설정한 인원 까지의 접속자 허용 / 리슨함수로 클라이언트 접속요청 대기
 
	while(1)
	{
		puts("lintening . . .");
		acc_sock = accept(sockfd, (struct sockaddr*)&cliaddr, &addrlen); // 클라이언트 접속 요청 들어올 시 액셉트함수로 연결 수락함
 
		if(acc_sock < 0) // 마찬가지로 오류 발생 시 오류출력, 종료
		{
			perror("Accept failed !");
			return -1;
		}
		puts("Connected client"); // 연결 성공 시 메세지 출력
 
		clnt_sock_arr[clnt_number++] = acc_sock;//새로운 사용자가 입장할 때마다 소켓번호 부여
                printf("새로운 연결, Client IP : %s\n", inet_ntoa(cliaddr.sin_addr));
		pthread_create(&a_thread, NULL, chat_input, (void*)acc_sock);//쓰레드 생성
		pthread_create(&b_thread, NULL, admin_chat, (void*)acc_sock);//쓰레드 생성
	}
 
	pthread_join(a_thread, NULL);//쓰레드 종료 대기
	pthread_join(b_thread, NULL);//쓰레드 종료 대기
 
	return 0;
}
 
void* chat_input(void* a)
{
	char temp[MAX_LINE];//받을 메시지
	int str_len;//메시지 길이
	list_count = 0;//저장된 메시지 갯수 초기화
 
	while(1)
	{
		str_len = (read((unsigned long int)a, temp, sizeof(temp)));//클라이언트 -> 서버로 수신
 
		input_list();//메시지 최대 갯수 초과시 정렬 함수 호출
		strcpy(list[_count], temp);//수신된 메시지를 저장
		list_count++;//저장된 메시지 갯수 증가
 
		write_msg(temp, str_len);//메시지 송신 함수 호출
 
		temp[str_len] = 0;
		printf("%s", temp);
		strcpy(temp,"");//다음 받을 메시지를 위해 초기화
 
		continue;
	}
}
 
void exitchat(int i)
{
	close(clnt_sock_arr[i]);//메시지를 보낸 사용자의 소켓 종료
	if(i != clnt_number - 1)//마지막 사용자가 아닌 경우
	{
		clnt_sock_arr[i] = clnt_sock_arr[clnt_number - 1];//메시지를 보낸 사용자의 소켓을 채움(소켓정렬)
		clnt_number--;
	}
 
}
 
void write_msg(char* msg, int len)
{
	int i;
 
	for(i=0; i < clnt_number; i++)
        {
		if(strstr(msg, ": /exit") != NULL)
		{
   			exitchat(i);//사용자 종료 명령어 확인 함수 호출
		}
 
		write(clnt_sock_arr[i], msg, len);//서버 --> 클라이언트로 송신
	}
 
}
 
void* admin_chat(void* b)
{
	int list_q;//확인할 메시지 갯수
	int i;
 
	while(getchar() == '\n')
	{
		printf("관리자모드가 활성화 되었습니다.\n");
		while(1)
		{
			fgets(admin, sizeof(admin), stdin);//안내 메시지 및 명령어 입력
			sprintf(add_admin, "admin : %s", admin);
 
 
 
			if(strstr(add_admin, ": /close"))
			{
                                for(i = 0; i < clnt_number; i++)
				write(clnt_sock_arr[i], add_admin, sizeof(add_admin));//메시지 송신 함수 호출(모든 사용자)
 
				strcpy(admin, "서버를 종료합니다.....3\n");//서버 종료안내
				sprintf(add_admin, "admin : %s", admin);
				write_msg(add_admin, sizeof(add_admin));//메시지 송신 함수 호출(모든 사용자)
				printf("%s", admin);
				sleep(1);
				strcpy(admin, "서버를 종료합니다.....2\n");//서버 종료안내
				sprintf(add_admin, "admin : %s", admin);
				write_msg(add_admin, sizeof(add_admin));//메시지 송신 함수 호출(모든 사용자)
				printf("%s", admin);
				sleep(1);
				strcpy(admin, "서버를 종료합니다.....1\n");//서버 종료안내
				sprintf(add_admin, "admin : %s", admin);
				write_msg(add_admin, sizeof(add_admin));//메시지 송신 함수 호출(모든 사용자)
				printf("%s", admin);
				sleep(1);
				strcpy(admin, "프로그램이 종료 되었습니다.\n");//서버 종료안내
				sprintf(add_admin, "admin : %s", admin);
				write_msg(add_admin, sizeof(add_admin));//메시지 송신 함수 호출(모든 사용자)
				printf("%s", admin);
				exit(0);//서버 강제 종료(모든 사용자 강제 종료)
			}
 
			else if(strstr(add_admin, ": /list"))
			{
				do{
				printf("원하는 갯수를 입력하시오. : ");
				scanf("%d", &list_q);//확인할 메시지 갯수 입력
 
				output_list(list_q);//저장된 메시지 출력 함수 호출
				}while(list_q > list_count);//입력 받은 메시지 > 저장된 메시지 갯수 시 반복
			}
		}
	}
}
 
void input_list(void)
{
	int j;
 
	if(list_count >= MAX_LIST)
	{
		for (j = 0; j < 99; j++) {
			strcpy(list[j], list[j + 1]);//메시지 저장 갯수 초과 시 메시지 정렬(오래된 메시지 삭제)
		}
		list_count = MAX_LIST - 1;//다음에 받을 메시지 번호를 마지막 저장위치로 고정
	}
}
 
void output_list(int number)
{	
	int i;
 
	if(number > list_count)//입력 받은 메시지 > 저장된 메시지 갯수 시 안내문 출력
		printf("갯수를 초과하였습니다.\n");
 
	else
	{
		printf("--------채팅 내역(%d개)--------\n", number);
		for(i = 0; i < number; i++)
	       	{
			printf("%s", list[i]);//저장된 메시지 출력
		}
		printf("--------------------------------\n");
	}
}