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

434 lines
13 KiB
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);
}
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
---------------------------------------------------------*/
pindex = -1;
/* 1. 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;
}
}
/* 2. Création si n'existe pas */
if( pindex == -1 ){
// On ajoute une entrée à data
pindex = arg->sgca->n;
arg->sgca->n++;
arg->sgca->data = (struct plane*) realloc(arg->sgca->data, sizeof(struct plane)*arg->sgca->n + 1);
printf("{tcp_com}(%d) plane '%s' (#%d) created\n", index, data.code, pindex);
}
/* 3. On copie les constantes */
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);
/* (4) On prépare la réponse
---------------------------------------------------------*/
/* 1. ACK data */
strcpy(data.code, arg->sgca->data[pindex].code);
data.x = htonl(arg->sgca->data[pindex].x);
data.y = htonl(arg->sgca->data[pindex].y);
data.z = htonl(arg->sgca->data[pindex].z);
data.cap = htonl(arg->sgca->data[pindex].cap);
data.spd = htonl(arg->sgca->data[pindex].spd);
/* 2. Vérification du FLAG (modifications) si requêtes d'update */
//TODO: Gestion arg->sgca->update[pindex].flags|UPD_SPEED etc.. pour ajout du flag+elements dans reponse
/* 3. On sérialise la réponse */
bzero(buffer, MAX_BUF_LEN*sizeof(char));
memcpy(buffer+sizeof(char)*0+sizeof(int)*0, &data.code, sizeof(char)*6);
memcpy(buffer+sizeof(char)*6+sizeof(int)*0, &data.x, sizeof(int));
memcpy(buffer+sizeof(char)*6+sizeof(int)*1, &data.y, sizeof(int));
memcpy(buffer+sizeof(char)*6+sizeof(int)*2, &data.z, sizeof(int));
memcpy(buffer+sizeof(char)*6+sizeof(int)*4, &data.spd, sizeof(int));
memcpy(buffer+sizeof(char)*6+sizeof(int)*3, &data.cap, sizeof(int));
/* (5) Envoi de la réponse
---------------------------------------------------------*/
read = send(SOCKET, buffer, PLANE_LEN/sizeof(char) + 1, 0);
}
/* [5] On notifie que l'avoin est déconnecté (crash)
=========================================================*/
/* 1. On notifie le crash */
arg->sgca->data[pindex].online = 0;
/* [6] On libère la mémoire
=========================================================*/
/* [7] Fermeture de la connection (SOCKET)
=========================================================*/
printf("{tcp_com}(%d) Fermeture de la socket de communication!\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) libéré\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;
int count, last, sent, nb; // compteurs d'envoi
int i, index = -1; // Compteurs globaux
char buffer[MAX_BUF_LEN]; // Buffer d'envoi
struct term_res 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_vterm}{com}(%d) starting terminal routine\n", index);
/* 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;
}
while( loop ){
/* [2] Récupération des données
=========================================================*/
/* 1. On initialise les variables utiles */
request.flags = TERMREQ_FBK;
nb = arg->sgca->n;
request.n = 0; // nombre d'avions online
free(request.data);
request.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(&request.data[i].code, &arg->sgca->data[i].code, sizeof(char)*6);
request.data[i].x = htonl( arg->sgca->data[i].x );
request.data[i].y = htonl( arg->sgca->data[i].y );
request.data[i].z = htonl( arg->sgca->data[i].z );
request.data[i].cap = htonl( arg->sgca->data[i].cap );
request.data[i].spd = htonl( arg->sgca->data[i].spd );
request.data[i].online = arg->sgca->data[i].online;
// Incrément du compte si online
request.n += request.data[i].online;
}
/* [3] Construction de la requête
=========================================================*/
/* 1. Initialisation du buffer */
bzero(buffer, MAX_BUF_LEN*sizeof(char));
/* 2. Copie des données globales */
count = 0; last = sizeof(char); memcpy(buffer+count, &request.flags, last);
count += last; last = sizeof(char); memcpy(buffer+count, &request.n, last);
/* 3. Copie des données des avions */
for( i = 0 ; i < request.n ; i++ ){
// Ignore les avions offline
if( !request.data[i].online )
continue;
count += last; last = sizeof(char)*6; memcpy(buffer+count, &request.data[i].code, last);
count += last; last = sizeof(int); memcpy(buffer+count, &request.data[i].x, last);
count += last; last = sizeof(int); memcpy(buffer+count, &request.data[i].y, last);
count += last; last = sizeof(int); memcpy(buffer+count, &request.data[i].z, last);
count += last; last = sizeof(int); memcpy(buffer+count, &request.data[i].cap, last);
count += last; last = sizeof(int); memcpy(buffer+count, &request.data[i].spd, last);
}
count += last;
/* [4] Envoi de la requête
=========================================================*/
/* 1. Gestion de l'envoi en plusieurs requêtes */
/* 2. Envoi */
len = sizeof(struct sockaddr_in);
sent = sendto(arg->socket, buffer, count/sizeof(char) + 1, 0, (struct sockaddr*) &clientInfo, len);
/* 3. Gestion erreur */
if( sent <= 0 ){
printf("{udp_vterm}{com}(%d) Unable to send data\n", index);
break;
}
/* [5] Timeout
=========================================================*/
sleep(PUBL_TIMEOUT);
}
/* [n] Arrêt du THREAD
============================================================================*/
/* 2. On met à jour "activeManagers" */
if( index != -1 )
arg->activeManagers[index] = 0;
/* 3. On arrête le THREAD */
if( DEBUGMOD&THR ) printf("{udp_vterm}{com}(%d) libéré\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
=========================================================*/
int read; // compteur
struct sockaddr_in clientInfo;
socklen_t len;
char request[MAX_BUF_LEN]; // Requête
// char response[MAX_BUF_LEN]; // Réponse
/* 2. On récupère les arguments */
struct handler_arg* arg = THREADABLE_ARGS;
do{
printf("{udp_x-term}{udp_com} waiting for terminal request\n");
/* [2] Récupération de la requête
=========================================================*/
/* 1. On lit sur la socket */
len = sizeof(struct sockaddr_in);
read = recvfrom(arg->socket, request, MAX_BUF_LEN, 0, (struct sockaddr*) &clientInfo, &len);
/* 2. Si erreur reception */
if( DEBUGMOD&BUF ) printf("{udp_x-term}{udp_com} READ = %d\n", read);
if( read < 0 )
continue;
/* 3. On désérialise la requête*/
printf("{udp_x-term}{udp_com} TERMINAL Request(%d bytes) : '%s'\n", read, request);
/* [3] Gestion de la requête
=========================================================*/
/* [4] Envoi reponse
=========================================================*/
strcpy(request+strlen(request), "-bla\0");
send(arg->socket, request, strlen(request), 0);
}while( 0 );
/* [n] Arrêt du THREAD
============================================================================*/
/* 1. On récupère le rang dans les "managers" */
int i, index = -1;
for( i = 0 ; i < MAX_UDP_THR ; i++ )
if( arg->managers[i] == pthread_self() ){ index = i; break; }
/* 2. On met à jour "activeManagers" */
if( index != -1 )
arg->activeManagers[index] = 0;
/* 3. On arrête le THREAD */
if( DEBUGMOD&THR ) printf("{udp_x-term}{udp_com}(%d) libéré\n", index);
pthread_exit(NULL);
}