diff --git a/header/puit.h b/header/puit.h index e3180a1..78f7ad0 100644 --- a/header/puit.h +++ b/header/puit.h @@ -2,7 +2,7 @@ #define HEADER_PUIT #include "tsock.h" -int launchPuit(int nombreMessage,int isTCP); +int launchPuit(int nombreMessage,int TailleMessage,int isTCP); int receiveMultipleData(int nombreMessages, int tailleMessage, int sock, struct sockaddr_in socketStruct); - +int exitMax(int var,int tailleMax); #endif \ No newline at end of file diff --git a/src/main.c b/src/main.c index 565f29f..73194be 100644 --- a/src/main.c +++ b/src/main.c @@ -8,7 +8,7 @@ int main (int argc, char **argv) char *ipAddress; extern char *optarg; extern int optind; - int source = -1, nb_message = -1, c, tcp=1, port=9000; /* Nb de messages à envoyer ou à recevoir, par défaut : 10 en émission, infini en réception */ + int source = -1, nb_message = -1, c, tcp=1, port=9000, tailleMessage; /* Nb de messages à envoyer ou à recevoir, par défaut : 10 en émission, infini en réception */ while ((c = getopt(argc, argv, "pn:su")) != -1) { switch (c) { case 'p': @@ -25,6 +25,10 @@ int main (int argc, char **argv) } source = 1; break; + case 'l': + tailleMessage =atoi(optarg); + exitMax(tailleMessage,1500); + break; case 'n': nb_message = atoi(optarg); break; @@ -54,6 +58,6 @@ int main (int argc, char **argv) else { //printf("Puit : %d\n",nb_message); - launchPuit(nb_message,tcp); + launchPuit(nb_message,tailleMessage,tcp); } } \ No newline at end of file diff --git a/src/puit.c b/src/puit.c index bcaafa6..cf5afbe 100644 --- a/src/puit.c +++ b/src/puit.c @@ -1,6 +1,6 @@ #include "../header/puit.h" -int launchPuit(int nombreMessage,int isTCP) +int launchPuit(int nombreMessage,int tailleMessage,int isTCP) { int sock; struct sockaddr_in socketPuit; @@ -16,7 +16,7 @@ int launchPuit(int nombreMessage,int isTCP) perror("[tsock] : fonction bind() : echec du lien avec socket serveur.\n"); exit(EXIT_FAILURE); } - receiveMultipleData(nombreMessage,30,sock,socketPuit); + receiveMultipleData(nombreMessage,tailleMessage,sock,socketPuit); close(sock); return 0; } @@ -45,4 +45,12 @@ int receiveMultipleData(int nombreMessages, int tailleMessage, int sock, struct } } return 0; -} \ No newline at end of file +} + +int exitMax(int var,int tailleMax){ + if(var>1500){ + printf("-l doit être <1500 \n"); + exit(EXIT_FAILURE); + } + return 0; +} \ No newline at end of file