├── .gitignore
├── README.md
├── client.cpp
├── screenshot.png
└── server.cpp


/.gitignore:
--------------------------------------------------------------------------------
1 | client
2 | server


--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
 1 | # Chatroom Application
 2 | 
 3 | A chatroom built in C++ using the concepts of socket programming and multi-threading. It supports chatting among multiple clients.
 4 | 
 5 | ![](/screenshot.png)
 6 | ## How to run
 7 | 
 8 | 1. Clone this repository
 9 | 2. Run the following commands in your terminal :
10 | ```
11 | g++ server.cpp -lpthread -o server
12 | g++ client.cpp -lpthread -o client
13 | ```
14 | 3. To run the server application, use this command in the terminal :
15 | ```
16 | ./server
17 | ```
18 | 
19 | 4. Now, open another terminal and use this command to run the client application :
20 | ```
21 | ./client
22 | ```
23 | 
24 | 5. For opening multiple client applications, repeat step 4.


--------------------------------------------------------------------------------
/client.cpp:
--------------------------------------------------------------------------------
  1 | #include <bits/stdc++.h>
  2 | #include <sys/types.h>
  3 | #include <sys/socket.h>
  4 | #include <netinet/in.h>
  5 | #include <errno.h>
  6 | #include <string.h>
  7 | #include <arpa/inet.h>
  8 | #include <unistd.h>
  9 | #include <thread>
 10 | #include <signal.h>
 11 | #include <mutex>
 12 | #define MAX_LEN 200
 13 | #define NUM_COLORS 6
 14 | 
 15 | using namespace std;
 16 | 
 17 | bool exit_flag=false;
 18 | thread t_send, t_recv;
 19 | int client_socket;
 20 | string def_col="\033[0m";
 21 | string colors[]={"\033[31m", "\033[32m", "\033[33m", "\033[34m", "\033[35m", "\033[36m"};
 22 | 
 23 | void catch_ctrl_c(int signal);
 24 | string color(int code);
 25 | int eraseText(int cnt);
 26 | void send_message(int client_socket);
 27 | void recv_message(int client_socket);
 28 | 
 29 | int main()
 30 | {
 31 | 	if((client_socket=socket(AF_INET,SOCK_STREAM,0))==-1)
 32 | 	{
 33 | 		perror("socket: ");
 34 | 		exit(-1);
 35 | 	}
 36 | 
 37 | 	struct sockaddr_in client;
 38 | 	client.sin_family=AF_INET;
 39 | 	client.sin_port=htons(10000); // Port no. of server
 40 | 	client.sin_addr.s_addr=INADDR_ANY;
 41 | 	//client.sin_addr.s_addr=inet_addr("127.0.0.1"); // Provide IP address of server
 42 | 	bzero(&client.sin_zero,0);
 43 | 
 44 | 	if((connect(client_socket,(struct sockaddr *)&client,sizeof(struct sockaddr_in)))==-1)
 45 | 	{
 46 | 		perror("connect: ");
 47 | 		exit(-1);
 48 | 	}
 49 | 	signal(SIGINT, catch_ctrl_c);
 50 | 	char name[MAX_LEN];
 51 | 	cout<<"Enter your name : ";
 52 | 	cin.getline(name,MAX_LEN);
 53 | 	send(client_socket,name,sizeof(name),0);
 54 | 
 55 | 	cout<<colors[NUM_COLORS-1]<<"\n\t  ====== Welcome to the chat-room ======   "<<endl<<def_col;
 56 | 
 57 | 	thread t1(send_message, client_socket);
 58 | 	thread t2(recv_message, client_socket);
 59 | 
 60 | 	t_send=move(t1);
 61 | 	t_recv=move(t2);
 62 | 
 63 | 	if(t_send.joinable())
 64 | 		t_send.join();
 65 | 	if(t_recv.joinable())
 66 | 		t_recv.join();
 67 | 			
 68 | 	return 0;
 69 | }
 70 | 
 71 | // Handler for "Ctrl + C"
 72 | void catch_ctrl_c(int signal) 
 73 | {
 74 | 	char str[MAX_LEN]="#exit";
 75 | 	send(client_socket,str,sizeof(str),0);
 76 | 	exit_flag=true;
 77 | 	t_send.detach();
 78 | 	t_recv.detach();
 79 | 	close(client_socket);
 80 | 	exit(signal);
 81 | }
 82 | 
 83 | string color(int code)
 84 | {
 85 | 	return colors[code%NUM_COLORS];
 86 | }
 87 | 
 88 | // Erase text from terminal
 89 | int eraseText(int cnt)
 90 | {
 91 | 	char back_space=8;
 92 | 	for(int i=0; i<cnt; i++)
 93 | 	{
 94 | 		cout<<back_space;
 95 | 	}	
 96 | }
 97 | 
 98 | // Send message to everyone
 99 | void send_message(int client_socket)
100 | {
101 | 	while(1)
102 | 	{
103 | 		cout<<colors[1]<<"You : "<<def_col;
104 | 		char str[MAX_LEN];
105 | 		cin.getline(str,MAX_LEN);
106 | 		send(client_socket,str,sizeof(str),0);
107 | 		if(strcmp(str,"#exit")==0)
108 | 		{
109 | 			exit_flag=true;
110 | 			t_recv.detach();	
111 | 			close(client_socket);
112 | 			return;
113 | 		}	
114 | 	}		
115 | }
116 | 
117 | // Receive message
118 | void recv_message(int client_socket)
119 | {
120 | 	while(1)
121 | 	{
122 | 		if(exit_flag)
123 | 			return;
124 | 		char name[MAX_LEN], str[MAX_LEN];
125 | 		int color_code;
126 | 		int bytes_received=recv(client_socket,name,sizeof(name),0);
127 | 		if(bytes_received<=0)
128 | 			continue;
129 | 		recv(client_socket,&color_code,sizeof(color_code),0);
130 | 		recv(client_socket,str,sizeof(str),0);
131 | 		eraseText(6);
132 | 		if(strcmp(name,"#NULL")!=0)
133 | 			cout<<color(color_code)<<name<<" : "<<def_col<<str<<endl;
134 | 		else
135 | 			cout<<color(color_code)<<str<<endl;
136 | 		cout<<colors[1]<<"You : "<<def_col;
137 | 		fflush(stdout);
138 | 	}	
139 | }
140 | 


--------------------------------------------------------------------------------
/screenshot.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/cjchirag7/chatroom-cpp/2dcfb3b097fec1a64546db5b8b87c407ca3dfbea/screenshot.png


--------------------------------------------------------------------------------
/server.cpp:
--------------------------------------------------------------------------------
  1 | #include <bits/stdc++.h>
  2 | #include <sys/types.h>
  3 | #include <sys/socket.h>
  4 | #include <netinet/in.h>
  5 | #include <errno.h>
  6 | #include <string.h>
  7 | #include <arpa/inet.h>
  8 | #include <unistd.h>
  9 | #include <thread>
 10 | #include <mutex>
 11 | #define MAX_LEN 200
 12 | #define NUM_COLORS 6
 13 | 
 14 | using namespace std;
 15 | 
 16 | struct terminal
 17 | {
 18 | 	int id;
 19 | 	string name;
 20 | 	int socket;
 21 | 	thread th;
 22 | };
 23 | 
 24 | vector<terminal> clients;
 25 | string def_col="\033[0m";
 26 | string colors[]={"\033[31m", "\033[32m", "\033[33m", "\033[34m", "\033[35m","\033[36m"};
 27 | int seed=0;
 28 | mutex cout_mtx,clients_mtx;
 29 | 
 30 | string color(int code);
 31 | void set_name(int id, char name[]);
 32 | void shared_print(string str, bool endLine);
 33 | int broadcast_message(string message, int sender_id);
 34 | int broadcast_message(int num, int sender_id);
 35 | void end_connection(int id);
 36 | void handle_client(int client_socket, int id);
 37 | 
 38 | int main()
 39 | {
 40 | 	int server_socket;
 41 | 	if((server_socket=socket(AF_INET,SOCK_STREAM,0))==-1)
 42 | 	{
 43 | 		perror("socket: ");
 44 | 		exit(-1);
 45 | 	}
 46 | 
 47 | 	struct sockaddr_in server;
 48 | 	server.sin_family=AF_INET;
 49 | 	server.sin_port=htons(10000);
 50 | 	server.sin_addr.s_addr=INADDR_ANY;
 51 | 	bzero(&server.sin_zero,0);
 52 | 
 53 | 	if((bind(server_socket,(struct sockaddr *)&server,sizeof(struct sockaddr_in)))==-1)
 54 | 	{
 55 | 		perror("bind error: ");
 56 | 		exit(-1);
 57 | 	}
 58 | 
 59 | 	if((listen(server_socket,8))==-1)
 60 | 	{
 61 | 		perror("listen error: ");
 62 | 		exit(-1);
 63 | 	}
 64 | 
 65 | 	struct sockaddr_in client;
 66 | 	int client_socket;
 67 | 	unsigned int len=sizeof(sockaddr_in);
 68 | 
 69 | 	cout<<colors[NUM_COLORS-1]<<"\n\t  ====== Welcome to the chat-room ======   "<<endl<<def_col;
 70 | 
 71 | 	while(1)
 72 | 	{
 73 | 		if((client_socket=accept(server_socket,(struct sockaddr *)&client,&len))==-1)
 74 | 		{
 75 | 			perror("accept error: ");
 76 | 			exit(-1);
 77 | 		}
 78 | 		seed++;
 79 | 		thread t(handle_client,client_socket,seed);
 80 | 		lock_guard<mutex> guard(clients_mtx);
 81 | 		clients.push_back({seed, string("Anonymous"),client_socket,(move(t))});
 82 | 	}
 83 | 
 84 | 	for(int i=0; i<clients.size(); i++)
 85 | 	{
 86 | 		if(clients[i].th.joinable())
 87 | 			clients[i].th.join();
 88 | 	}
 89 | 
 90 | 	close(server_socket);
 91 | 	return 0;
 92 | }
 93 | 
 94 | string color(int code)
 95 | {
 96 | 	return colors[code%NUM_COLORS];
 97 | }
 98 | 
 99 | // Set name of client
100 | void set_name(int id, char name[])
101 | {
102 | 	for(int i=0; i<clients.size(); i++)
103 | 	{
104 | 			if(clients[i].id==id)	
105 | 			{
106 | 				clients[i].name=string(name);
107 | 			}
108 | 	}	
109 | }
110 | 
111 | // For synchronisation of cout statements
112 | void shared_print(string str, bool endLine=true)
113 | {	
114 | 	lock_guard<mutex> guard(cout_mtx);
115 | 	cout<<str;
116 | 	if(endLine)
117 | 			cout<<endl;
118 | }
119 | 
120 | // Broadcast message to all clients except the sender
121 | int broadcast_message(string message, int sender_id)
122 | {
123 | 	char temp[MAX_LEN];
124 | 	strcpy(temp,message.c_str());
125 | 	for(int i=0; i<clients.size(); i++)
126 | 	{
127 | 		if(clients[i].id!=sender_id)
128 | 		{
129 | 			send(clients[i].socket,temp,sizeof(temp),0);
130 | 		}
131 | 	}		
132 | }
133 | 
134 | // Broadcast a number to all clients except the sender
135 | int broadcast_message(int num, int sender_id)
136 | {
137 | 	for(int i=0; i<clients.size(); i++)
138 | 	{
139 | 		if(clients[i].id!=sender_id)
140 | 		{
141 | 			send(clients[i].socket,&num,sizeof(num),0);
142 | 		}
143 | 	}		
144 | }
145 | 
146 | void end_connection(int id)
147 | {
148 | 	for(int i=0; i<clients.size(); i++)
149 | 	{
150 | 		if(clients[i].id==id)	
151 | 		{
152 | 			lock_guard<mutex> guard(clients_mtx);
153 | 			clients[i].th.detach();
154 | 			clients.erase(clients.begin()+i);
155 | 			close(clients[i].socket);
156 | 			break;
157 | 		}
158 | 	}				
159 | }
160 | 
161 | void handle_client(int client_socket, int id)
162 | {
163 | 	char name[MAX_LEN],str[MAX_LEN];
164 | 	recv(client_socket,name,sizeof(name),0);
165 | 	set_name(id,name);	
166 | 
167 | 	// Display welcome message
168 | 	string welcome_message=string(name)+string(" has joined");
169 | 	broadcast_message("#NULL",id);	
170 | 	broadcast_message(id,id);								
171 | 	broadcast_message(welcome_message,id);	
172 | 	shared_print(color(id)+welcome_message+def_col);
173 | 	
174 | 	while(1)
175 | 	{
176 | 		int bytes_received=recv(client_socket,str,sizeof(str),0);
177 | 		if(bytes_received<=0)
178 | 			return;
179 | 		if(strcmp(str,"#exit")==0)
180 | 		{
181 | 			// Display leaving message
182 | 			string message=string(name)+string(" has left");		
183 | 			broadcast_message("#NULL",id);			
184 | 			broadcast_message(id,id);						
185 | 			broadcast_message(message,id);
186 | 			shared_print(color(id)+message+def_col);
187 | 			end_connection(id);							
188 | 			return;
189 | 		}
190 | 		broadcast_message(string(name),id);					
191 | 		broadcast_message(id,id);		
192 | 		broadcast_message(string(str),id);
193 | 		shared_print(color(id)+name+" : "+def_col+str);		
194 | 	}	
195 | }
196 | 


--------------------------------------------------------------------------------