Implemented modify train, users and cancel booking

Author: utsavcoding

client.c

@@ -10,6 +10,7 @@
 #include <fcntl.h>
 
 /* Length of variable*/
+#define IP_ADDR "192.168.43.147"
 #define PASSWORD_LEN 16
 #define USERNAME_LEN 20
 #define TRNAME_LEN 30
@@ -114,18 +115,23 @@ void display_booking(struct_booking booking){
 	printf("\t%d\t\t%d\t\t%d\t%s\t   %d\t\t %d\t %d\n",booking.book_no,booking.acc_no,booking.trn_no,booking.name,booking.age,booking.seats,booking.status);
 }
 
-void show_all_booking(int conn_fd){
+int show_all_booking(int conn_fd){
 	struct_booking response;int moreflg;
 	if(write(conn_fd, &CURRENT_USER.acc_no, sizeof(int))==-1)	ERR_EXIT("read()");
 	if(read(conn_fd, &moreflg, sizeof(int))==-1)	ERR_EXIT("read()");
 	moreflg=ntohl(moreflg);
+	if(moreflg==0){
+		printf("\nNO BOOKING HISTORY!!\n\n");
+		return 0;
+	}
 	printf("Booking No | Booking Account No | Train No | Traveller Name | Age | Seats Booked | Status\n");
 	while(moreflg){
 		if(read(conn_fd, &response, sizeof(struct_booking))==-1)	ERR_EXIT("read()");
 		display_booking(response);
 		if(read(conn_fd, &moreflg, sizeof(int))==-1)	ERR_EXIT("read()");
 		moreflg=ntohl(moreflg);
 	}
+	return 1;
 }
 
 void book_ticket(int conn_fd){
@@ -160,6 +166,30 @@ void book_ticket(int conn_fd){
 	printf("b) Requirement of your seats does not meet availability. Please check in the train list(shown above).\n\n");
 }
 
+void cancel_booking(int conn_fd){
+	show_all_booking(conn_fd);
+	int moreflg;
+	if(read(conn_fd, &moreflg, sizeof(int))==-1)	ERR_EXIT("read()");
+	moreflg=ntohl(moreflg);
+	if(moreflg==0){
+		return;
+	}
+
+	int booking_no;
+	printf("Enter Booking No: ");scanf("%d",&booking_no);
+	booking_no=htonl(booking_no);
+	int act_no=htonl(CURRENT_USER.acc_no);
+	if(write(conn_fd, &act_no, sizeof(int)) == -1)	ERR_EXIT("write()");
+	if(write(conn_fd, &booking_no, sizeof(int)) == -1)	ERR_EXIT("write()");
+
+	if(read(conn_fd, &moreflg, sizeof(int))==-1)	ERR_EXIT("read()");
+	moreflg=ntohl(moreflg);
+	if(moreflg==1)
+		printf("Booking has been cancelled.\n");
+	if(moreflg==0)
+		printf("Booking was already cancelled.\n");
+}
+
 int menu_user(int conn_fd){
 	printf("1. Book Ticket\n2. View Booking History\n3. Cancel Booking\n4. Exit\n\n");
 	int choice;int option;
@@ -176,6 +206,9 @@ int menu_user(int conn_fd){
     			show_all_booking(conn_fd);
     			break;
     		case 3:
+    			option=htonl(choice);
+    			if(write(conn_fd, &option, sizeof(int)) == -1)	ERR_EXIT("write()");
+    			cancel_booking(conn_fd);
     			break;
     		case 4:
     			option=htonl(choice);
@@ -273,6 +306,37 @@ void delete_user(int conn_fd){
 	printf("No such record with this account no!!\n\n");	
 }
 
+void modify_user(int conn_fd){
+	int accno;struct_customer response;int presentflg=0;
+
+	printf("Enter Account no: ");
+	scanf("%d",&accno);
+	accno=htonl(accno);
+	if(write(conn_fd, &accno, 1*sizeof(int)) == -1)	ERR_EXIT("write()");
+
+	if(read(conn_fd, &presentflg, 1*sizeof(int)) == -1)	ERR_EXIT("write()");
+	
+	presentflg=ntohl(presentflg);
+	if(presentflg){
+		if(read(conn_fd, &response, sizeof(struct_customer))==-1)	ERR_EXIT("read()");
+		printf("      Account No |    Account Type |    Account Username   |  Account Password   \t|    Account Status \t\n");
+		display_customer(response);
+
+		printf("\nModifying Detals...\n");
+		char updated_name[USERNAME_LEN],updated_pwd[PASSWORD_LEN];
+		printf("Enter Updated Account Type(0 for ADMIN, 1 for NORMAL, 2 for AGENT): ");scanf("%d",&response.type);
+		printf("Enter Updated Account Name: ");scanf("%s",updated_name);
+		strcpy(response.cust_username,updated_name);
+		printf("Enter Updated Account Password: ");scanf("%s",updated_pwd);
+		strcpy(response.cust_password,updated_pwd);
+		if(write(conn_fd, &response, 1*sizeof(struct_customer)) == -1)	ERR_EXIT("write()");
+		printf("\nAccount has been updated\n");
+		return;
+	}
+
+	printf("No such record with this account no!!\n\n");	
+}
+
 struct_train get_train(){
 	struct_train new;char train_name[30];
 	printf("Enter train name : ");
@@ -328,16 +392,50 @@ void delete_train(int conn_fd){
 
 	presentflg=ntohl(presentflg);
 	if(presentflg){
-		printf("Account has been deleted.\n\n");
+		printf("Train Record has been deleted.\n\n");
+		return;
+	}
+
+	printf("No such train record with this train no!!\n\n");
+}
+
+void modify_train(int conn_fd){
+	int trn_no;struct_train response;int presentflg=0;
+	printf("Enter Train no: ");
+	scanf("%d",&trn_no);
+	trn_no=htonl(trn_no);
+	if(write(conn_fd, &trn_no, 1*sizeof(int)) == -1)	ERR_EXIT("write()");
+
+	if(read(conn_fd, &presentflg, 1*sizeof(int)) == -1)	ERR_EXIT("write()");
+	
+	presentflg=ntohl(presentflg);
+	if(presentflg){
+		if(read(conn_fd, &response, sizeof(struct_train))==-1)	ERR_EXIT("read()");
+		printf("     Train No |    Train Name\t     |    Available Seats   |     Booked Seats   |    Status \t\n");
+		display_train(response);
+
+		printf("Modifying Train Details...\n");
+		char up_trname[TRNAME_LEN];int up_seats;
+		printf("Enter Updated Train Name: \n");scanf(" %[^\n]",up_trname);
+		if(write(conn_fd, &up_trname, strlen(up_trname)+1) == -1)	ERR_EXIT("write()");
+
+		do{
+			printf("Enter number of seats to increase: ");scanf("%d",&up_seats);
+			if(up_seats<0)
+				printf("Seats to increase cannot be negative.\n");
+		}while(up_seats<0);
+
+		if(write(conn_fd, &up_seats, sizeof(int)) == -1)	ERR_EXIT("write()");
+		printf("Train has been updated.\n");
 		return;
 	}
 
 	printf("No such train record with this train no!!\n\n");
 }
 
 int menu_admin(int conn_fd){
-	printf("1. Add User\n2. Search User \n3. Delete User\n4. View Users\n");
-	printf("5. Add Train\n6. Search Train \n7. Delete Train\n8. View Trains\n9. Exit\n\n");
+	printf("1. Add User\n2. Search User \n3. Delete User\n4. Modify User\n5. View Users\n");
+	printf("6. Add Train\n7. Search Train \n8. Delete Train\n9. Modify Train\n10. View Trains\n11. Exit\n\n");
 	int choice;int option;
 	scanf("%d",&choice);
     switch(choice){
@@ -356,32 +454,42 @@ int menu_admin(int conn_fd){
    			if(write(conn_fd, &option, 1*sizeof(int)) == -1)	ERR_EXIT("write()");
    			delete_user(conn_fd);
    			break;
-    	case 4:
+   		case 4:
+   			option=htonl(choice+4);
+   			if(write(conn_fd, &option, 1*sizeof(int)) == -1)	ERR_EXIT("write()");
+   			modify_user(conn_fd);
+   			break;
+    	case 5:
     		option=htonl(choice+4);
 			if(write(conn_fd, &option, 1*sizeof(int)) == -1)	ERR_EXIT("write()");
 			show_all_customer(conn_fd);
     		break;
-    	case 5:
+    	case 6:
     		option=htonl(choice+4);
 			if(write(conn_fd, &option, 1*sizeof(int)) == -1)	ERR_EXIT("write()");    		
     		add_train(conn_fd);
     		break;
-    	case 6:
+    	case 7:
     		option=htonl(choice+4);
    			if(write(conn_fd, &option, 1*sizeof(int)) == -1)	ERR_EXIT("write()");
    			search_train(conn_fd);
     		break;
-    	case 7:
+    	case 8:
     		option=htonl(choice+4);
    			if(write(conn_fd, &option, 1*sizeof(int)) == -1)	ERR_EXIT("write()");
    			delete_train(conn_fd);
     		break;
-    	case 8:
+    	case 9:
+    		option=htonl(choice+4);
+   			if(write(conn_fd, &option, 1*sizeof(int)) == -1)	ERR_EXIT("write()");
+   			modify_train(conn_fd);
+    		break;
+    	case 10:
     		option=htonl(choice+4);
 			if(write(conn_fd, &option, 1*sizeof(int)) == -1)	ERR_EXIT("write()");
 			show_all_train(conn_fd);
     		break;
-    	case 9:
+    	case 11:
     		option=htonl(choice+4);
     		if(write(conn_fd, &option, sizeof(int)) == -1)	ERR_EXIT("write()");
     		close(conn_fd);
@@ -411,17 +519,16 @@ int main(int argc,char* argv[])
     server_addr.sin_family = AF_INET;
     server_addr.sin_port = htons(PORT_NO); 
 
-    if(inet_pton(AF_INET, "0.0.0.0", &server_addr.sin_addr) <= 0)
+    if(inet_pton(AF_INET, IP_ADDR, &server_addr.sin_addr) <= 0)
         ERR_EXIT("inet_pton()"); 
     if(connect(socket_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1)
         ERR_EXIT("connect()");
-    printf("conn_fd: %d\n",socket_fd);
+    
     response_flg=login(socket_fd);
     if(response_flg==0){
     	if(write(socket_fd, &choice, sizeof(int)) == -1)	ERR_EXIT("write()");	
     }
 
-    printf("Inside main %d",response_flg);
     while(response_flg){
     	system("clear");