158 lines
3.3 KiB
C
158 lines
3.3 KiB
C
#include "server.h"
|
|
|
|
|
|
|
|
int main(int argc, char* argv[]){
|
|
|
|
/* [1] Initialisation
|
|
=========================================================*/
|
|
/* (1) Client address + port */
|
|
struct sockaddr_in addr_client;
|
|
int port;
|
|
|
|
/* (2) Sockets */
|
|
int listensock, datasock;
|
|
|
|
/* (3) Struct to send arguments to threadable process */
|
|
pthread_arg_wrapper built_args = {0, 0};
|
|
|
|
|
|
|
|
|
|
/* [2] Manage arguments
|
|
=========================================================*/
|
|
/* (1) Manage arguments count */
|
|
if( argc < 2 ){
|
|
printf("Missing arguments\nUsage: server port [message]\n");
|
|
printf("port = 0 : automatic port\n");
|
|
return 1;
|
|
}
|
|
|
|
/* (2) Manage port argument */
|
|
if( sscanf(argv[1], "%d", &port) <= 0 ){
|
|
printf("argument error: port must be a valid integer\n");
|
|
printf("port = 0 : automatic port\n");
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
/* [3] Create socket
|
|
=========================================================*/
|
|
/* (1) Create socket */
|
|
listensock = xbind(port);
|
|
|
|
/* (1-) Manage error */
|
|
if( listensock == -1 ){
|
|
perror("erreur création socket");
|
|
exit(1);
|
|
}
|
|
|
|
|
|
/* [4] Wait for clients indefinately
|
|
=========================================================*/
|
|
int i, index = -1; // thread index
|
|
while( 1 ){
|
|
|
|
/* (1) Listen for incoming connection */
|
|
index = -1;
|
|
datasock = xlisten(listensock, &addr_client);
|
|
DEBUG&& debug("server", "wait for client");
|
|
|
|
/* (2) Manage error */
|
|
if( datasock == -1 ){
|
|
DEBUG&& printf("error\n");
|
|
continue;
|
|
}
|
|
|
|
DEBUG&& printf("done\n");
|
|
|
|
|
|
/* (3) Search for a free manager */
|
|
DEBUG&& debug("server", "finding free thread");
|
|
for( i = 0 ; i < maxClients ; i++ )
|
|
|
|
if( a_managers[i] == 0 ){
|
|
DEBUG&& printf("found[%d]\n", i);
|
|
index = i;
|
|
break;
|
|
}
|
|
|
|
|
|
/* (4) If we haven't found a free manager -> do nothing (john snow) */
|
|
if( index == -1 ){
|
|
DEBUG&& printf("not found\n");
|
|
continue;
|
|
}
|
|
|
|
|
|
/* (5) Build thread arguments */
|
|
built_args.socket = datasock;
|
|
built_args.index = index;
|
|
|
|
/* (6) Processing client connection on a new thread */
|
|
pthread_create(&managers[index], NULL, manageClient, (void*)(intptr_t) &built_args);
|
|
|
|
/* (7) Mark the manager as active */
|
|
a_managers[index] = 1;
|
|
|
|
|
|
}
|
|
|
|
/* [5] Wait for all threads to end
|
|
=========================================================*/
|
|
/* (1) Wait for all threads to end properly */
|
|
for( i = 0 ; i < maxClients ; i++ )
|
|
pthread_join(managers[i], NULL);
|
|
|
|
/* (2) Close listening socket */
|
|
close(listensock);
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void* manageClient(void* pthread_args){
|
|
|
|
|
|
/* [1] Initialization
|
|
=========================================================*/
|
|
/* (1) Initialize socket data */
|
|
struct sockaddr_in addr_client;
|
|
int bytes; // transfer count
|
|
char client_ip[20];
|
|
|
|
/* (2) Fetch arguments */
|
|
pthread_arg_wrapper* arguments = (pthread_arg_wrapper*) (intptr_t) pthread_args;
|
|
|
|
// get client_socket
|
|
int datasock = arguments->socket;
|
|
int thread_index = arguments->index;
|
|
|
|
DEBUG&& printf("starting the computation loop\n");
|
|
|
|
//Compute client request
|
|
traiterClient(datasock);
|
|
|
|
/* [2] End process
|
|
=========================================================*/
|
|
/* (1) Close data socket */
|
|
DEBUG&& debug("thread", "closing socket");
|
|
close(datasock);
|
|
DEBUG&& printf("done\n");
|
|
|
|
/* (2) Mark manager as inactive again */
|
|
DEBUG&& debug("thread", "freeing thread");
|
|
a_managers[thread_index] = 0;
|
|
DEBUG&& printf("done\n");
|
|
|
|
}
|