#include #include #include #include #include #include #include #include #include #include #include #include // Regular Colors #define RED "\33[0;31m" #define GREEN "\33[0;32m" #define YELLOW "\33[0;33m" #define BLUE "\33[0;34m" #define PURPLE "\33[0;35m" #define CYAN "\33[0;36m" #define WHITE "\33[0;37m" // Reset to default color #define RESET "\33[0m" // Bold #define BRED "\33[1;31m" #define BGREEN "\33[1;32m" #define BYELLOW "\33[1;33m" #define BBLUE "\33[1;34m" #define BPURPLE "\33[1;35m" #define BCYAN "\33[1;36m" #define BWHITE "\33[1;37m" int Break_Function = 0; // Exit function that was executed after the user type Ctrl+C void exit_function(int signal){ printf("%s[ %sok %s] %sStop%s Program .\n",WHITE,GREEN,WHITE,RED,WHITE); exit(0); //Break_Function = 1; } // Exit function that was executed in Child void Child_Kill(int signal){ printf("\r%s[ %sok %s] Stop Child process with pid %s%d%s .%s\n",WHITE,GREEN,WHITE,GREEN, getpid(),WHITE,RESET); exit(0); } // Exit function that was executed in Parent void Parent_Kill(int signal){ sleep(2); printf("\r%s[ %sok %s] Stop Parent with pid %s%d%s .%s\n",WHITE,GREEN,WHITE,GREEN, getpid(),WHITE,RESET); Break_Function = 1; fclose(stdin); } // Thread function void read_function_in_bg(int *socket_id){ int socket_id_thread=*(int *)socket_id,receive_status; char msg_rcv[256],end[10]="bye\n"; while(strcmp(msg_rcv,end)!=0 && Break_Function==0){ receive_status=read(socket_id_thread,&msg_rcv,sizeof(msg_rcv)); if(receive_status<0){ printf("%s[%s!%s] %sError%s with receiving data .%s\n",WHITE,RED,WHITE,RED,WHITE,RESET); } else { printf("\r%sServer : %s",CYAN,WHITE); fputs(msg_rcv,stdout); printf("%s", RESET); } } kill(getppid(),SIGINT); exit(0); } void ascii_art(){ printf(" \n"); usleep(200000); printf("%s ██████╗██╗ ██╗███████╗███╗ ██╗████████╗%s\n",YELLOW,RESET); usleep(200000); printf("%s ██╔════╝██║ ██║██╔════╝████╗ ██║╚══██╔══╝%s\n",YELLOW,RESET); usleep(200000); printf("%s ██║ ██║ ██║█████╗ ██╔██╗ ██║ ██║ %s\n",BYELLOW,RESET); usleep(200000); printf("%s ██║ ██║ ██║██╔══╝ ██║╚██╗██║ ██║ %s\n",BYELLOW,RESET); usleep(200000); printf("%s ╚██████╗███████╗██║███████╗██║ ╚████║ ██║ %s\n",YELLOW,RESET); usleep(200000); printf("%s ╚═════╝╚══════╝╚═╝╚══════╝╚═╝ ╚═══╝ ╚═╝ %s\n",YELLOW,RESET); usleep(200000); printf("\n"); } int main(int argc, char const *argv[]) { // Clear the screen system("clear"); // Print a beautiful ascii art in the screen ascii_art(); // Redirect signals that was generated by the user signal(SIGINT,exit_function); // Declare the variables char msg_sent[256],first_msg_rcv[256],password[128],msg_end[10]="bye\n"; // Create and Verify the Socket int socket_id=socket(AF_INET,SOCK_STREAM,0); if(socket_id==-1){ printf("%s[%s!%s] There was an %serror%s to create the %ssocket%s .%s\n",WHITE,RED,WHITE,RED,WHITE,GREEN,WHITE,RESET); exit(1); } // Specify our server information struct hostent *converted_addr; struct sockaddr_in server_address; // Save information in server.information as a struct server_address.sin_family=AF_INET; // AF_INET == IPv4 // htons(host to network short) and atoi(ASCII to integer) server_address.sin_port=htons(atoi(argv[2])); // Read port number as an argument // gethostbyname : is a function to convert a name to address converted_addr=gethostbyname(argv[1]); // Convert first argument to @IP // Copies n characters from memory area converted_addr->h_addr to memory area server_address.sin_addr memcpy(&server_address.sin_addr,converted_addr->h_addr,converted_addr->h_length); // Establish a connection with the server and verify the status if(connect(socket_id,(struct sockaddr*) &server_address,sizeof(server_address))==-1){ printf("%s[%s!%s] There was a %sproblem%s to establish a connection with the %sserver%s .%s\n",WHITE,RED,WHITE,RED,WHITE,YELLOW,WHITE,RESET); exit(1); } else { printf("%s[ %sok %s] %sConnection%s was established .%s\n",WHITE,GREEN,WHITE,YELLOW,WHITE,RESET); } // Reading and print first message from the socket read(socket_id,&first_msg_rcv,sizeof(first_msg_rcv)); printf("%s[ %sok %s] %sServer :%s %s %s\n",WHITE,GREEN,WHITE,CYAN,WHITE,first_msg_rcv,RESET); // Verify the password int valid_password=-1,i=0,validON; while(valid_password!=0 && i!=3){ printf("\r%s[ %sok %s] %sPassword%s :%s ",WHITE,GREEN,WHITE,YELLOW,WHITE,RED); printf("\033[s"); scanf("%[^\n]%*c",password); write(socket_id,password,sizeof(password)); read(socket_id,&msg_sent,sizeof(msg_sent)); validON=atoi(msg_sent); if(validON==0){ printf("\033[u%s********%s\n",GREEN,RESET); printf("\033[2K%s[ %sok %s] %sSession%s is begin .%s\n",WHITE,GREEN,WHITE,BLUE,WHITE,RESET); valid_password=0; } else { printf("%s[%s!%s] Incorrect %spassword%s please%s try again%s .%s\n",WHITE,RED,WHITE,RED,WHITE,CYAN,WHITE,RESET); i++; printf("\033[u\033[2K"); } } if(i==3){ exit(1); } // Create multiple proccess with fork() function pid_t pid = fork(); // Save return value from fork in variable $pid || (pid == -1) Failed || (pid == 0) Child process || (pid > 0) Parent process if(pid==-1){ printf("%s[%s!%s] There was an %serror%s with %sfork%s .%s\n",WHITE,RED,WHITE,RED,WHITE,YELLOW,WHITE,RESET); exit(1); } if(pid==0) { signal(SIGINT,Child_Kill); // Redirect SIGINT signal to function Child_Kill() read_function_in_bg(&socket_id); // Read & Print all message that we're receive in the socket } else if(pid>0){ // Sent multiple messages to the server signal(SIGINT,Parent_Kill); // Redirect SIGINT signal to function Parent_Kill() int msg_sent_status; printf("%s", YELLOW); printf("%sYou %s: %s", CYAN,WHITE,YELLOW); while(fgets(msg_sent,256,stdin) != NULL && Break_Function==0){ printf("\033[1A\033[2K%sYou%s :%s %s%s",CYAN,WHITE,YELLOW,msg_sent,RESET); printf("%sYou %s: %s", CYAN,WHITE,YELLOW); if(strcmp(msg_sent,msg_end)!=0){ sendto(socket_id,msg_sent,sizeof(msg_sent),0,(struct sockaddr*) &server_address,sizeof(server_address)); } else { printf("%s[ %sok %s] %sSession%s with the server was %sended%s .%s\n",WHITE,GREEN,WHITE,YELLOW,WHITE,RED,WHITE,RESET); sendto(socket_id,msg_end,sizeof(msg_end),0,(struct sockaddr*) &server_address,sizeof(server_address)); kill(pid,SIGINT); // Kill process break; } } // Send end message to the server sendto(socket_id,msg_end,sizeof(msg_end),0,(struct sockaddr*) &server_address,sizeof(server_address)); int status; pid_t child_pid = wait(&status); // Wait until the child process is finished if(child_pid==-1){ printf("%s[%s!%s] There was an%s error%s to terminate child process .%s\n",WHITE,RED,WHITE,RED,WHITE,RESET); } else if(child_pid>0){ printf("%s[ %sok %s] The child with %sPID%s %d%s & %sstatus%s of %s%d%s was %sstoped%s .%s\n",WHITE,GREEN,WHITE,GREEN,RED, child_pid,WHITE,GREEN,WHITE,RED,status,WHITE,RED,WHITE,RESET); } // Close socket close(socket_id); // Exit the program with status of 0 exit(0); } // Coder : Silent Ghost X }

C Socket Programming /client.c