-
Notifications
You must be signed in to change notification settings - Fork 2
/
lyrebird.server.c
126 lines (107 loc) · 5.92 KB
/
lyrebird.server.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
/*------------------------------------------------------
*
* lyrebird.server.c
*
* this file contains the main function that used to build
* a server and then assign tasks to client sides, more detail
* functions are in the svr_func.c.
*
* Project : LyreBird
* Name : Chong Guo
* Student ID : 301295753
* SFU username : armourg
* Lecture section : D1
* Instructor : Brain G.Booth
* TA : Scott Kristjanson
*
* Created by Armour on 25/09/2015
* Copyright (c) 2015 Armour. All rights reserved.
*
*------------------------------------------------------
*/
#include "lyrebird.server.h"
#include "memwatch.h"
int main_flag = EXIT_SUCCESS; /* Used to store return value for main function */
char *enc_txt; /* Used to store encrypted file name */
char *dec_txt; /* Used to store decrypted file name */
char *out_time; /* Used to store output time */
FILE *fcfg, *flog; /* The file pointer that used to open config file and log file */
int max_fds; /* The max number of file descriptor */
fd_set rfds; /* The set of file descriptor */
char host[NI_MAXHOST]; /* Used to store the return value of getnameinfo function */
struct ifaddrs *ifaddr, *ifa; /* Used to store the return value of getifaddrs function */
int sockfd, sockfd_new; /* Socket file descriptor */
int sockfd_cli[CLIENT_MAXNUM]; /* The array of client socket file descriptor */
char ipaddr_cli[CLIENT_MAXNUM][NI_MAXHOST]; /* The array of client IP address */
socklen_t addr_len; /* The size of sockaddr_in */
struct sockaddr_in serv_addr, cli_addr; /* Used to store the server and client address information */
int read_flag = 1; /* This is a flag that means program should read a new line from config file */
int finish_flag = 0; /* This is a flag that indicate reaching the end of config file */
char *client_ip; /* A pointer to the IP address char array of one client */
char read_type; /* This is the message that read from one socket to client side */
uint32_t msg_len; /* The length of sent message */
int remained_cli; /* The number of the remained clients */
char ip_buffer[INET_ADDRSTRLEN]; /* A buffer that used to store client IP address */
char send_mark[MARK_MAXLENGTH]; /* A buffer that used to store sent message */
char recv_mark[MARK_MAXLENGTH]; /* A buffer that used to store received message */
char read_mark[MARK_MAXLENGTH]; /* A buffer that used to store read message */
char write_mark[MARK_MAXLENGTH]; /* A buffer that used to store write message */
/*
* Function: Main
* -------------------
* Main function of lyrebird
*
* Parameters:
* argc: the number of parameters in main function
* argv[]: the array of parameters in main function
*
* Returns:
* void
*/
int main(int argc, char *argv[]) {
int i;
init(); /* Init some variables (like malloc timestamp string, encrypt text string, etc.) */
check_par(argc, argv); /* Check command arguments number */
open_config(argv); /* Open config file and check if it failed */
open_log(argv); /* Open log file and check if it failed */
get_ipaddr(); /* Get server IP address */
create_socket(); /* Create a socket */
bind_socket(); /* Bind the socket */
listen_socket(); /* Listen at the socket */
print_server_info(); /* Print server information */
while (TRUE) { /* Read until the end of file */
if (read_flag) {
if (fscanf(fcfg, "%s", enc_txt) == EOF) {
finish_flag = 1;
break;
} else {
fscanf(fcfg, "%s", dec_txt);
}
}
read_flag = 0;
init_select(); /* Select function */
if (select_func() == -1) break;
for (i = 0; i < max_fds + 1; i++) {
if (FD_ISSET(i, &rfds)) {
if (i == sockfd) { /* If have a new client connect */
if (accept_new_cli() == -1) break; /* Try to accept new client */
if (check_connect() == -1) break; /* Check connect message from client */
if (print_client_info() == -1) break; /* Print the information of client side */
store_client_ip(); /* Store the client ip address */
break;
} else { /* If have new message from client side */
client_ip = get_host_by_sockfd(i); /* Get the client ip address by socket */
recv_socket_msg(i, recv_mark); /* Get the message from socket */
handle_client_msg(i); /* Handle client message (SUCCESS_MSG, FAILURE_MSG, DISPATCH_MSG, etc.) */
break;
}
}
if (main_flag == EXIT_FAILURE) break;
}
if (main_flag == EXIT_FAILURE) break;
}
remained_cli = ask_clients_quit(); /* Ask clients quit and count the remain clients number */
wait_clients_quit(); /* Wait for all clients quit */
quit_server(); /* Clean up and quit server, also print the message to log */
return main_flag;
}