sysdis-project/central-manager/lib/local/handler.c

/**************************
*   Handler Dependency    *
***************************
* Designed & Developed by *
*     Adrien Marquès      *
*     <xdrm-brackets>     *
***************************
*   doowap31@gmail.com    *
**************************/
#include "handler.h"






/* Gestion d'une connexion PLANE
*
* @THREADABLE_SOCKET<void*>			SOCKET de la connexion client
*
* @history
* [1] Initialisation des variables
* @loop
*   (2) Attente de requête
*   (3) Gestion de la requête
*   (4) Envoi de la réponse
* [5] On notifie que l'avion est déconnecté (crash)
* [6] On libère la mémoire
* [7] Fermeture de la connection (SOCKET)
* [n] Arrêt du THREAD
* 	1. On met à jour "activeManagers"
* 	2. On arrête le THREAD
*
*/
void* managePlane(void* THREADABLE_ARGS){

	/* [1] Initialisation des variables
	=========================================================*/
	/* 1. Variables utiles */
	int          read, i, index, pindex; // compteurs
	char         buffer[MAX_BUF_LEN];    // buffer
	struct plane data;                   // données de l'avion
	int          SOCKET;                 // Copie de la socket (évite les conflits de références)


	/* 2. On récupère les arguments */
	struct handler_arg* arg = THREADABLE_ARGS;
	memcpy(&SOCKET, &arg->socket, sizeof(int));
	if( DEBUGMOD&HDR ) printf("===== managePlane(%p, %p, %d, %p) =====\n", (void*) arg->managers, (void*) arg->activeManagers, arg->socket, (void*) arg->sgca);


	/* 3. On récupère le rang du thread parmi les "managers" */
	index = -1;
	for( i = 0 ; i < MAX_TCP_THR ; i++ )
		if( arg->managers[i] == pthread_self() ){ index = i; break; }

	// Erreur de thread
	if( index == -1 ){
		if( DEBUGMOD&THR ) printf("{tcp_com}(%d)             Unknown thread index. Aborting\n", index);
		pthread_exit(NULL);
	}


	pindex = -1; // on ne connait pas encore l'avion

	while( 1 ){

		/* (2) Récupération de la requête
		---------------------------------------------------------*/
		/* 1. On lit sur la socket */
		read = recv(SOCKET, buffer, MAX_BUF_LEN, 0);

		/* 2.1. Si erreur reception (-1:erreur, 0:fermeture client propre) */
		if( read <= 0 ){
			if( DEBUGMOD&BUF ) printf("{tcp_com}(%d)             read: %d -> must exit thread\n", index, read);
			break;
		}


		/* 2.2. Si message trop court */
		if( read < PLANE_LEN ){
			if( DEBUGMOD&BUF ) printf("{tcp_com}(%d)             read: %d (expected: %d)\n", index, read, (int) PLANE_LEN);
			continue;
		}


		/* 3. On parse la requête (indianness: network order) */
		memcpy(&data.code, buffer+sizeof(char)*0+sizeof(int)*0, sizeof(char)*6);
		memcpy(&data.x,    buffer+sizeof(char)*6+sizeof(int)*0, sizeof(int));
		memcpy(&data.y,    buffer+sizeof(char)*6+sizeof(int)*1, sizeof(int));
		memcpy(&data.z,    buffer+sizeof(char)*6+sizeof(int)*2, sizeof(int));
		memcpy(&data.spd,  buffer+sizeof(char)*6+sizeof(int)*4, sizeof(int));
		memcpy(&data.cap,  buffer+sizeof(char)*6+sizeof(int)*3, sizeof(int));

		/* 4. Gestion de l'indianness */
		data.x      = ntohl(data.x);
		data.y      = ntohl(data.y);
		data.z      = ntohl(data.z);
		data.cap    = ntohl(data.cap);
		data.spd    = ntohl(data.spd);
		data.online = 1;



		/* (3) Gestion de la requête -> enregistrement
		---------------------------------------------------------*/
		/* 1. On cherche le rang de l'avion si on l'a pas déja */
		if( pindex <= -1 ){

			/* 2. On regarde si l'avion existe */
			for( i = 0 ; i < arg->sgca->n ; i++ ){

				// Si l'avion existe -> on passe à la suite
				if( strcmp(arg->sgca->data[i].code, data.code) == 0 ){
					pindex = i;
					break;
				}

			}

		}

		/* 3. Création si n'existe pas */
		if( pindex <= -1 ){

			// On ajoute une entrée à data + socket
			pindex                    = arg->sgca->n;
			arg->sgca->n++;
			arg->sgca->data           = (struct plane*) realloc(arg->sgca->data,   sizeof(struct plane)*arg->sgca->n + 1);
			arg->sgca->socket         = (int*)          realloc(arg->sgca->socket, sizeof(struct plane)*arg->sgca->n + 1);
			arg->sgca->socket[pindex] = SOCKET;
			printf("{tcp_com}(%d)             plane '%s' (#%d) created\n", index, data.code, pindex);

		}

		/* 4. On copie/met à jour les valeurs */
		memcpy(&arg->sgca->data[pindex], &data, sizeof(struct plane));

		if( DEBUGMOD&COM ) printf("{tcp_com}(%d)             stored (%d)'%s': {x = %d; y = %d; z = %d; cap = %d; spd = %d}\n", index, pindex, arg->sgca->data[pindex].code, arg->sgca->data[pindex].x, arg->sgca->data[pindex].y, arg->sgca->data[pindex].z, arg->sgca->data[pindex].cap, arg->sgca->data[pindex].spd);

	}


	/* [5] On notifie que l'avoin est déconnecté (crash)
	=========================================================*/
	/* 1. On notifie le crash */
	arg->sgca->data[pindex].online = 0;
	arg->sgca->socket[pindex]      = -1;





	/* [6] On libère la mémoire
	=========================================================*/



	/* [7] Fermeture de la connection (SOCKET)
	=========================================================*/
	printf("{tcp_com}(%d)             Closing communication socket\n", index);
	close(SOCKET);



	/* [n] Arrêt du THREAD
	============================================================================*/
	/* 1. On met à jour "activeManagers" */
	if( index != -1 )
		arg->activeManagers[index] = 0;

	/* 2. On arrête le THREAD */
	if( DEBUGMOD&THR ) printf("{tcp_com}(%d)             freed\n", index);
	pthread_exit(NULL);
}




/* Gestion d'une connexion TERMINAL
*
* @THREADABLE_SOCKET<void*>			SOCKET de la connexion client
*
* @history
* [1] Initialisation des variables
*
* @loop
*   [2] Récupération des données
*   [3] Construction de la requête
*   [4] Envoi de la requête
*   [5] Timeout
*
* [6] On vide les buffers
* [7] Fermeture de la connection (SOCKET)
* [n] Arrêt du THREAD
* 	1. On récupère le rang dans les "managers"
* 	2. On met à jour "activeManagers"
* 	3. On arrête le THREAD
*
*/
void* manageViewTerm(void* THREADABLE_ARGS){

	/* [1] Initialisation des variables
	=========================================================*/
	/* 1. Initialisation des variables */
	struct sockaddr_in clientInfo;
	socklen_t          len;
	char               loop = 1;
	size_t             buflen;
	int                sent;                 // compteurs d'envoi
	int                i, index = -1;        // Compteurs globaux
	char*              buffer   = malloc(1); // Buffer d'envoi

	/* 2. On récupère les arguments */
	struct handler_arg* arg = THREADABLE_ARGS;

	/* 3. On récupère le rang dans les "managers" */
	for( i = 0 ; i < MAX_UDP_THR ; i++ )
		if( arg->managers[i] == pthread_self() ){ index = i; break; }

	printf("{udp_vterm}{com}(%d)      starting terminal routine (rate: %d sec)\n", index, PUBL_TIMEOUT);


	/* 4. Attente d'un client */
	len = sizeof(struct sockaddr_in);
	if( recvfrom(arg->socket, buffer, MAX_BUF_LEN*sizeof(char), 0, (struct sockaddr*) &clientInfo, &len) < 0 ){
		printf("{udp_vterm}{com}(%d)      No terminal detected, exiting\n", index);
		loop = 0;
	}

	if( loop ) printf("{udp_cterm}{com}(%d)      Terminal connected\n", index);


	while( loop ){


		/* [2] Récupération des données
		=========================================================*/
		getPlaneData(&buffer, &buflen, arg->sgca);


		/* [3] Envoi de la requête
		=========================================================*/
		/* 2. Envoi */
		len = sizeof(struct sockaddr_in);
		sent = sendto(arg->socket, buffer, buflen + 1, 0, (struct sockaddr*) &clientInfo, len);

		/* 3. Gestion erreur */
		if( sent <= 0 ){
			printf("{udp_vterm}{com}(%d)      Unable to send data\n", index);
			break;
		}


		/* [4] Timeout
		=========================================================*/
		sleep(PUBL_TIMEOUT);

	}


	/* [n] Arrêt du THREAD
	============================================================================*/
	/* 1. On met à jour "activeManagers" */
	if( index != -1 )
		arg->activeManagers[index] = 0;

	/* 2. On ferme la socket + libère la mémoire */
	close(arg->socket);
	free(buffer);


	/* 3. On arrête le THREAD */
	if( DEBUGMOD&THR ) printf("{udp_vterm}{com}(%d)      freed\n", index);
	pthread_exit(NULL);
}








/* Gestion d'une connexion TERMINAL
*
* @THREADABLE_SOCKET<void*>			SOCKET de la connexion client
*
* @history
* [1] Initialisation des variables
* [2] Récupération de la requête
* [3] Traitement de la requête
* [4] Création de la réponse
* [5] Envoi de la réponse
* [6] On vide les buffers
* [7] Fermeture de la connection (SOCKET)
* [n] Arrêt du THREAD
* 	1. On récupère le rang dans les "managers"
* 	2. On met à jour "activeManagers"
* 	3. On arrête le THREAD
*
*/
void* manageCtrlTerm(void* THREADABLE_ARGS){

	/* [1] Initialisation des variables
	=========================================================*/
	/* 1. Initialisation des variables */
	struct sockaddr_in clientInfo;
	socklen_t          len;
	size_t             dataLen;
	char               loop = 1, update = 0, fbk = 0;
	int                count, last, sent, nb;  // compteurs d'envoi
	int                i, index = -1;          // Compteurs globaux
	int                pindex;                 // index of the current plane
	char               buffer[MAX_BUF_LEN];    // Buffer d'envoi
	char*              dataBuffer = malloc(1);
	struct term_req    request;                // Requête

	/* 2. On récupère les arguments */
	struct handler_arg* arg = THREADABLE_ARGS;

	/* 3. On récupère le rang dans les "managers" */
	for( i = 0 ; i < MAX_UDP_THR ; i++ )
		if( arg->managers[i] == pthread_self() ){ index = i; break; }

	printf("{udp_cterm}{com}(%d)      starting terminal routine (rate: %d sec)\n", index, PUBL_TIMEOUT);

	/* 4. Attente d'un client */
	len = sizeof(struct sockaddr_in);
	if( recvfrom(arg->socket, buffer, MAX_BUF_LEN*sizeof(char), 0, (struct sockaddr*) &clientInfo, &len) < 0 ){
		printf("{udp_cterm}{com}(%d)      No terminal detected, exiting\n", index);
		loop = 0;
	}

	if( loop ) printf("{udp_cterm}{com}(%d)      Terminal connected\n", index);


	while( loop ){

		pindex = -1;

		/* [2] Récupération de la requête
		=========================================================*/
		/* 1. On lit sur la socket */
		len = sizeof(struct sockaddr_in);
		bzero(buffer, sizeof(char)*MAX_BUF_LEN);
		count = recvfrom(arg->socket, buffer, MAX_BUF_LEN, 0, (struct sockaddr*) &clientInfo, &len);

		/* 2. Si erreur reception */
		if( count < TERMREQ_LEN ){
			send(arg->socket, "0", sizeof(char)*2, 0);
			if( DEBUGMOD&BUF ) printf("{udp_cterm}{com}(%d)      Error receiving request\n", index);
			continue;
		}

		/* 3. On désérialise la requête*/
		bzero(&request, sizeof(struct term_req));
		count  = 0;    last = sizeof(char);   memcpy(&request.flags,       buffer+count, last );
		count += last; last = sizeof(char)*6; memcpy(&request.update.code, buffer+count, last );
		count += last; last = sizeof(int);    memcpy(&request.update.x,    buffer+count, last );
		count += last; last = sizeof(int);    memcpy(&request.update.y,    buffer+count, last );
		count += last; last = sizeof(int);    memcpy(&request.update.z,    buffer+count, last );
		count += last; last = sizeof(int);    memcpy(&request.update.cap,  buffer+count, last );
		count += last; last = sizeof(int);    memcpy(&request.update.spd,  buffer+count, last );
		// indian-switch
		request.update.cap = ntohl(request.update.cap);
		request.update.spd = ntohl(request.update.spd);

		printf("{udp_cterm}{com}(%d)     { flags: %d; plane: #%s { z=%d; cap=%d; speed=%d} }\n", index, request.flags, request.update.code, request.update.z, request.update.cap, request.update.spd);


		/* [3] Gestion de la requête
		=========================================================*/
		/* 1. On vérifie qu'il y a bien un update demandé (ou DATA) */
		update = ( request.flags&TERMREQ_ALT || request.flags&TERMREQ_CAP || request.flags&TERMREQ_SPD );
		fbk    = request.flags&TERMREQ_FBK;
		if( !( update || fbk ) ){
			send(arg->socket, "0", sizeof(char)*2, 0);
			printf("{udp_cterm}{com}(%d)     Invalid flag, passing\n", index);
			continue;
		}


		if( update ){

			/* 2. On regarde si on trouve l'avion */
			for( i = 0 ; i < arg->sgca->n ; i++ ){

				// Si l'avion existe et online -> on passe à la suite
				if( strcmp(arg->sgca->data[i].code, request.update.code) == 0 && arg->sgca->data[i].online ){
					pindex = i;
					break;
				}

			}


			/* 3. Si on a pas trouvé l'avion ou qu'il est déconnecté, on passe */
			if( pindex <= -1 || arg->sgca->socket[pindex] <= -1 ){
				printf("{udp_cterm}{com}(%d)     Plane unknown or unreachable, passing\n", index);
				continue;
			}

			/* 4. On transfère la requête à l'avion */
			bzero(buffer, sizeof(char)*MAX_BUF_LEN);
			// indian-switch
			request.update.z   = htonl(request.update.z);
			request.update.cap = htonl(request.update.cap);
			request.update.spd = htonl(request.update.spd);
			count  = 0;    last = sizeof(char); memcpy(buffer+count, &request.flags,      last );
			count += last; last = sizeof(int);  memcpy(buffer+count, &request.update.z,   last );
			count += last; last = sizeof(int);  memcpy(buffer+count, &request.update.cap, last );
			count += last; last = sizeof(int);  memcpy(buffer+count, &request.update.spd, last );

			if( send(arg->sgca->socket[pindex], buffer, count/sizeof(char)+1, 0) <= 0 ){
				printf("{udp_cterm}{com}(%d)     Cannot send request to plane\n", index);
				continue;
			}


			/* [4] Gestion de la réponse de l'avion
			=========================================================*/
			/* 1. Réception */
			bzero(buffer, sizeof(char)*MAX_BUF_LEN);
			if( recv(arg->sgca->socket[pindex], buffer, MAX_BUF_LEN, 0) <= 0 ){
				send(arg->socket, "0", sizeof(char)*2, 0);
				printf("{udp_cterm}{com}(%d)     Cannot get response from plane\n", index);
				continue;
			}

			/* 2. Gestion de la validation de l'update */
			if( !(buffer[0]&TERMREQ_ALT) && request.flags&TERMREQ_ALT )
				request.flags -= TERMREQ_ALT;

			if( !(buffer[0]&TERMREQ_CAP) && request.flags&TERMREQ_CAP )
				request.flags -= TERMREQ_CAP;

			if( !(buffer[0]&TERMREQ_SPD) && request.flags&TERMREQ_SPD )
				request.flags -= TERMREQ_SPD;

		}

		if( fbk ){

			/* [5] Données des avions
			=========================================================*/
			/* 1. On récupère les données */
			getPlaneData(&dataBuffer, &dataLen, arg->sgca);

			/* 2. Si update, on mixe les flags */
			if( update )
			dataBuffer[0] |= request.flags;

		}


		/* [6] Réponse au terminal
		=========================================================*/
		if( send(arg->socket, dataBuffer, dataLen+1, 0) < 0 )
			printf("{udp_cterm}{com}(%d)     Cannot answer to terminal\n", index);


	}


	/* [n] Arrêt du THREAD
	============================================================================*/
	/* 1. On met à jour "activeManagers" */
	if( index != -1 )
		arg->activeManagers[index] = 0;

	/* 2. On ferme la socket */
	close(arg->socket);

	/* 3. On arrête le THREAD */
	if( DEBUGMOD&THR ) printf("{udp_cterm}{com}(%d)      freed\n", index);
	pthread_exit(NULL);
}




















void getPlaneData(char** pBuffer, size_t* pLen, struct context* pContext){
	/* [1] Initialisation des variables
	=========================================================*/
	/* 1. Initialisation des variables */
	int                count, last, i, nb; // compteurs
	struct term_res    response;               // Requête


	/* [1] Récupération des données
	=========================================================*/
	/* 1. On initialise les variables utiles */
	response.flags = TERMREQ_FBK;
	nb             = pContext->n;
	response.n     = 0; // nombre d'avions online

	response.data  = malloc( nb * sizeof(struct plane) + 1 );

	/* 2. On récupère la liste des avions (network byte order) */
	for( i = 0 ; i < nb ; i++ ){

		memcpy(&response.data[i].code, &pContext->data[i].code, sizeof(char)*6);
		response.data[i].x      = htonl( pContext->data[i].x   );
		response.data[i].y      = htonl( pContext->data[i].y   );
		response.data[i].z      = htonl( pContext->data[i].z   );
		response.data[i].cap    = htonl( pContext->data[i].cap );
		response.data[i].spd    = htonl( pContext->data[i].spd );
		response.data[i].online =        pContext->data[i].online;

		// Incrément du compte si online
		response.n += response.data[i].online;

	}


	/* [3] Construction de la requête
	=========================================================*/
	/* 1. Initialisation du buffer */
	*pBuffer = realloc(*pBuffer, sizeof(char)*2 + PLANE_LEN * response.n + sizeof(char));

	/* 2. Copie des données globales */
	count  = 0;    last = sizeof(char); memcpy(*pBuffer+count, &response.flags, last);
	count += last; last = sizeof(char); memcpy(*pBuffer+count, &response.n,     last);

	/* 3. Copie des données des avions */
	for( i = 0 ; i < nb ; i++ ){

		// Ignore les avions offline
		if( !response.data[i].online )
			continue;

		count += last; last = sizeof(char)*6; memcpy(*pBuffer+count, &response.data[i].code, last);
		count += last; last = sizeof(int);    memcpy(*pBuffer+count, &response.data[i].x,    last);
		count += last; last = sizeof(int);    memcpy(*pBuffer+count, &response.data[i].y,    last);
		count += last; last = sizeof(int);    memcpy(*pBuffer+count, &response.data[i].z,    last);
		count += last; last = sizeof(int);    memcpy(*pBuffer+count, &response.data[i].cap,  last);
		count += last; last = sizeof(int);    memcpy(*pBuffer+count, &response.data[i].spd,  last);
	}

	*pLen = (count+last) / sizeof(char);
	free(response.data);

}