201 lines
5.1 KiB
C
201 lines
5.1 KiB
C
#include "proxy_ftp.h"
|
|
|
|
// DECLARATIONS
|
|
static pthread_t managers[maxListLen];
|
|
static int activeManagers = 0x00;
|
|
|
|
/* headers */
|
|
static void* testServer();
|
|
void* manageConnection(void* THREADABLE_SOCKET);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/*****************/
|
|
/* CORPS DU PROG */
|
|
/*****************/
|
|
int main(int argc, char* argv[]){
|
|
if( argc > 1 )
|
|
testServer(argv[1]);
|
|
else
|
|
testServer("");
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static void* testServer(char* localPort){
|
|
int THREADABLE_SOCKET = -1; // contiendra le BUS DE COMMANDE utilisateur à envoyer à un THREAD
|
|
struct sockaddr_storage clientInfo; // contiendra les infos client
|
|
char repeat; // sert à sortir de la boucle
|
|
socklen_t len = sizeof(struct sockaddr_storage);
|
|
|
|
// retour de @DROP_SERVER
|
|
char* serverPort; // contiendra le port
|
|
int LISTENSOCK; // contiendra la socket d'écoute
|
|
|
|
/* [0] On démarre le SERVEUR + le CLIENT
|
|
==========================================================*/
|
|
serverPort = malloc(maxPortLen);
|
|
|
|
if( strlen(localPort) != 0 )
|
|
strcpy(serverPort, localPort);
|
|
|
|
|
|
DROP_SERVER(remoteHost, &serverPort, &LISTENSOCK);
|
|
printf("Port: %s\n", serverPort);
|
|
|
|
|
|
|
|
/* [1] Attente d'une demande de connection, puis création d'une socket
|
|
============================================================================*/
|
|
int index = 0;
|
|
while( index <= maxListLen ){
|
|
THREADABLE_SOCKET = -1; // on initialise la SOCKET en attendant la connexion
|
|
|
|
THREADABLE_SOCKET = accept(LISTENSOCK, (struct sockaddr *) &clientInfo, &len);
|
|
|
|
// on lance un thread pour le traitement
|
|
pthread_create(&managers[index], NULL, manageConnection, (void*)(intptr_t) THREADABLE_SOCKET);
|
|
index++;
|
|
}
|
|
|
|
int i;
|
|
for( i = 0 ; i < maxListLen ; i++ )
|
|
pthread_join(managers[i], NULL);
|
|
|
|
printf("FERMETURE DE TOUTES LES CONNECTIONS!\n");
|
|
close(LISTENSOCK);
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Gestion d'une connexion client
|
|
*
|
|
* @THREADABLE_SOCKET<void*> SOCKET de la connexion client
|
|
*
|
|
* @history
|
|
* [1] Initialisation des variables
|
|
* [2] Envoi de la séquence de bienvenue
|
|
* [3] Attente des données reçues et reception (CLIENT_FTP->PROXY)
|
|
* [4] Traitement de la requête client
|
|
* [5] Redirection de la requête vers le serveur FTP (PROXY->SERVER_FTP)
|
|
* [6] Traitement de la reponse du serveur FTP
|
|
* [7] Redirection de la réponse vers le client (PROXY->CLIENT_FTP)
|
|
*
|
|
*
|
|
*/
|
|
void* manageConnection(void* THREADABLE_SOCKET){
|
|
int USER_SOCKET = (intptr_t) THREADABLE_SOCKET;
|
|
|
|
/* [1] Variables
|
|
============================================================================*/
|
|
// int USER_SOCKET = -1; // contiendra le BUS DE COMMANDE utilisateur
|
|
int FTP_SOCKET = -1; // contiendra le BUS DE COMMANDE FTP
|
|
int DUSER_SOCKET = -1; // contiendra le BUS DE DONNES utilisateur
|
|
int DFTP_SOCKET = -1; // contiendra le BUS DE DONNEES FTP
|
|
|
|
char BUFFER[maxBuffLen]; // contiendra le BUFFER
|
|
char* ftp_response; // contiendra le BUFFER de réponse du serveur FTP (commandes)
|
|
int nbReceived, nbSend; // contiendra les nb reçu && envoyé
|
|
|
|
/* [2] Envoi de la séquence de Bienvenue
|
|
============================================================================*/
|
|
swrite(&USER_SOCKET, WLCM_MSG);
|
|
|
|
/* BOUCLE DE COMMANDES */
|
|
while( USER_SOCKET != -1 ){
|
|
|
|
/* [3] On récupère les données reçues (+attente)
|
|
============================================================================*/
|
|
if( WAIT_SOCKET_UPDATE(&USER_SOCKET, BUFFER) == -1 ) break;
|
|
if( DEBUGMOD&CMD ) xPrint("C->P: %s\n", BUFFER);
|
|
|
|
|
|
/* [4] Traitement de la requête
|
|
============================================================================*/
|
|
MANAGE_REQUEST(BUFFER, &USER_SOCKET, &FTP_SOCKET, &DUSER_SOCKET, &DFTP_SOCKET);
|
|
|
|
// si on a pas la connection FTP (commande) initialisée, on quitte
|
|
// if( FTP_SOCKET == -1 ) break;
|
|
|
|
|
|
/* [5] Redirection vers le serveur FTP
|
|
============================================================================*/
|
|
if( DEBUGMOD&CMD ) xPrint("P->F: %s\n\n", BUFFER);
|
|
CLIENT_SEND(&FTP_SOCKET, BUFFER, &ftp_response);
|
|
if( DEBUGMOD&CMD ) xPrint("F->P: %s\n", ftp_response);
|
|
|
|
/* [6] Traitement de la réponse (FTP)
|
|
============================================================================*/
|
|
MANAGE_RESPONSE(ftp_response, &USER_SOCKET, &FTP_SOCKET, &DUSER_SOCKET, &DFTP_SOCKET);
|
|
|
|
|
|
|
|
/* [7] Redirection vers le CLIENT
|
|
============================================================================*/
|
|
if( swrite(&USER_SOCKET, ftp_response) == -1 ) break;
|
|
if( DEBUGMOD&CMD ) xPrint("P->C: %s\n", ftp_response);
|
|
|
|
|
|
/* [8] On vide les buffers
|
|
============================================================================*/
|
|
memset(BUFFER, '\0', maxBuffLen); // on vide le buffer
|
|
memset(ftp_response, '\0', maxBuffLen);
|
|
}
|
|
|
|
/* [9] Fermeture des connections (SOCKETS)
|
|
============================================================================*/
|
|
printf("CLOSING CONNECTIONS\n");
|
|
close(USER_SOCKET);
|
|
close(FTP_SOCKET);
|
|
|
|
// on arrête le THREAD
|
|
pthread_exit(NULL);
|
|
}
|
|
|
|
|
|
/*
|
|
* DROP_SERVER
|
|
* INIT_CLIENT
|
|
*
|
|
* while(){
|
|
* accept();
|
|
* client_send()
|
|
* }
|
|
*
|
|
* EXIT_CLIENT
|
|
* HALT_SERVER
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*/ |