TCP completement pris en charge, affichage plus complet.

header/puit.h

@@ -3,6 +3,5 @@
 
 #include "tsock.h"
 int launchPuit(int nombreMessage,int tailleMessage,int isTCP);
-int receiveMultipleData(int nombreMessages, int tailleMessage, int sock, struct sockaddr_in socketStruct);
+int receiveMultipleData(int nombreMessages, int tailleMessage, int sock, struct sockaddr_in socketStruct, int isTCP);
-int exitMax(int var,int tailleMax);
 #endif

header/source.h

@@ -3,7 +3,6 @@
 #include "tsock.h"
 
 int launchSource(int nombreMessage,int tailleMessage,int isTCP);
-int sendMultipleData(int nombreMessages, int tailleMessage, int sock, struct sockaddr_in socketStruct);
+int sendMultipleData(int nombreMessages, int tailleMessage, int sock, struct sockaddr_in socketStruct, int isTCP);
-void formatText(char * actualMessage, int num, int tailleMessage, char messageChar);
 
 #endif

header/tsock.h

@@ -16,6 +16,8 @@ void setNbMessage(int * nb, int source);
 void printInfo(int nb, int source, int portNumber, char ipAddress[]);
 void initStructSocket(struct sockaddr_in *socketServerTemp, int source);
 void getNonOtpArgs(char ** argv, int argc, int portNumber, char * ipAddress[]);
+void formatText(char * actualMessage, int num, int tailleMessage, char messageChar);
+int exitMax(int var,int tailleMax);
 
 #define PORT_NUM    9000
 

src/puit.c

@@ -4,14 +4,7 @@ int launchPuit(int nombreMessage,int tailleMessage,int isTCP)
 {
     int sock,socketType;
     struct sockaddr_in socketPuit;
-    if(isTCP)
+    socketType = (isTCP) ? SOCK_STREAM : SOCK_DGRAM;
-    {
-        socketType=SOCK_STREAM;
-    }
-    else
-    {
-        socketType=SOCK_DGRAM;
-    }
 
     if((sock=socket(AF_INET,socketType,0)) == -1)
     {
@@ -24,55 +17,52 @@ int launchPuit(int nombreMessage,int tailleMessage,int isTCP)
 		perror("[tsock] : fonction bind() : echec du lien avec socket serveur.\n");
 		exit(EXIT_FAILURE);
 	}
-    if(isTCP)
+    receiveMultipleData(nombreMessage,tailleMessage,sock,socketPuit,isTCP);
-    {
-        unsigned int longueurRecu = sizeof(socketPuit);
-        char messageRecu[30];
-        listen(sock,10);
-        int acceptValue = accept(sock,(struct sockaddr *)&socketPuit,&longueurRecu);
-        printf("AcceptValue = %d\n",acceptValue);
-        int tailleMessageRecu = read(acceptValue,messageRecu,sizeof(messageRecu));
-        messageRecu[tailleMessageRecu]='\0';
-        printf("Message TCP [%d] : %s\n",tailleMessageRecu,messageRecu);
-    }
-    else
-    {
-        receiveMultipleData(nombreMessage,tailleMessage,sock,socketPuit);
-    }
     close(sock);
     return 0;
 }
 
-int receiveMultipleData(int nombreMessages, int tailleMessage, int sock, struct sockaddr_in socketStruct)
+int receiveMultipleData(int nombreMessages, int tailleMessage, int sock, struct sockaddr_in socketStruct, int isTCP)
 {
     char messageRecu[tailleMessage+1];
-    int n, longueurRecu = sizeof(socketStruct);
+    int n=1, longueurRecu = sizeof(socketStruct);
+    if(isTCP)
+    {
+        listen(sock,5);
+        sock = accept(sock,(struct sockaddr *)&socketStruct,(socklen_t * restrict)&longueurRecu);
+    }
     if(nombreMessages < 0)
     {
         int i=1;
-        while(1)
+        while(n>0)
+        {
+            if(isTCP)
+            {
+                n = read(sock,messageRecu,tailleMessage);
+            }
+            else
             {
                 n = recvfrom(sock, (char *)messageRecu, tailleMessage, 0, (struct sockaddr*) &socketStruct,(socklen_t *__restrict__)&longueurRecu);
+            }
             messageRecu[n] = '\0';
-            printf("Puit\tReception n°%d (%d) :\t[%s]\n",i,tailleMessage,messageRecu);
+            printf("Puit\tReception n°%d (%d) :\t[%s]\n",i,n,messageRecu);
             i++;
         }
     }
     else{
         for(int i=0;i<nombreMessages;i++)
+        {
+            if(isTCP)
+            {
+                n = read(sock,messageRecu,tailleMessage);
+            }
+            else
             {
                 n = recvfrom(sock, (char *)messageRecu, tailleMessage, 0, (struct sockaddr*) &socketStruct,(socklen_t *__restrict__)&longueurRecu);
+            }
             messageRecu[n] = '\0';
             printf("Puit\tReception n°%d (%d) :\t[%s]\n",i+1,tailleMessage,messageRecu);
         }
     }
     return 0;
 }
-
-int exitMax(int var,int tailleMax){
-    if(var>1500){
-        printf("-l doit être <1500 \n");
-        exit(EXIT_FAILURE);
-    }
-    return 0;
-} 

src/source.c

@@ -4,14 +4,7 @@ int launchSource(int nombreMessage,int tailleMessage,int isTCP)
 {
     int sock,socketType;
     struct sockaddr_in socketSource;
-    if(isTCP)
+    socketType = (isTCP) ? SOCK_STREAM : SOCK_DGRAM;
-    {
-        socketType=SOCK_STREAM;
-    }
-    else
-    {
-        socketType=SOCK_DGRAM;
-    }
 
     if((sock=socket(AF_INET,socketType,0)) == -1)
     {
@@ -22,19 +15,17 @@ int launchSource(int nombreMessage,int tailleMessage,int isTCP)
     if(isTCP)
     {
         connect(sock,(struct sockaddr *)&socketSource,(socklen_t)sizeof(socketSource));
-        printf("Connect successful\n");
+        sendMultipleData(nombreMessage,tailleMessage,sock,socketSource,isTCP);
-        int isWritten = write(sock,"aaaaaa",sizeof("aaaaaa"));
-        printf("on écrit : %d\n",isWritten);
     }
     else
     {
-        sendMultipleData(nombreMessage,tailleMessage,sock,socketSource);
+        sendMultipleData(nombreMessage,tailleMessage,sock,socketSource,isTCP);
     }
     close(sock);
     return 0;
 }
 
-int sendMultipleData(int nombreMessages, int tailleMessage, int sock, struct sockaddr_in socketStruct)
+int sendMultipleData(int nombreMessages, int tailleMessage, int sock, struct sockaddr_in socketStruct, int isTCP)
 {
     int longueurEmis;
     char messageChar='a';
@@ -42,30 +33,17 @@ int sendMultipleData(int nombreMessages, int tailleMessage, int sock, struct soc
     for(int i=0;i<nombreMessages;i++)
     {
         formatText(sendingMessage,i,tailleMessage,messageChar);
+        if(isTCP)
+        {
+            longueurEmis = write(sock,sendingMessage,sizeof(sendingMessage));
+        }
+        else
+        {
             longueurEmis = sendto(sock, sendingMessage, tailleMessage, 0, (struct sockaddr*)&socketStruct, sizeof(socketStruct));
+        }
         sendingMessage[tailleMessage]='\0';
         printf("Source\tEnvoi n°%d (%d) :\t[%s]\n",i+1,longueurEmis,sendingMessage);
         messageChar>='z'?messageChar='a':messageChar++;
     }
     return 0;
 }
-
-void formatText(char * actualMessage, int num, int tailleMessage, char messageChar)
-{
-    char numBuffer[15];
-    sprintf(numBuffer, "%d", (num+1)%10000);
-    int numberLength=0;
-    while(numBuffer[numberLength] != '\0')
-    {
-        numberLength++;
-    }
-    for(int i=0;i<4-numberLength;i++)
-    {
-        actualMessage[i]=0x20;
-    }
-    for(int i=4-numberLength,j=0;i<4;i++,j++)
-    {
-        actualMessage[i]=numBuffer[j];
-    }
-    memset(actualMessage+4, messageChar, tailleMessage);
-}

src/tsock.c

@@ -40,7 +40,6 @@ void initStructSocket(struct sockaddr_in *socketTempStruct, int source)
     else
     {
         socketTempStruct->sin_addr.s_addr=INADDR_ANY;
-
     }
 }
 
@@ -51,3 +50,31 @@ void getNonOtpArgs(char ** argv, int argc, int portNumber, char ** ipAddress)
     *ipAddress = (char *)malloc(sizeof(argv[argc-1]));
     strcpy(*ipAddress,argv[argc-1]);
 }
+
+void formatText(char * actualMessage, int num, int tailleMessage, char messageChar)
+{
+    char numBuffer[15];
+    sprintf(numBuffer, "%d", (num+1)%10000);
+    int numberLength=0;
+    while(numBuffer[numberLength] != '\0')
+    {
+        numberLength++;
+    }
+    for(int i=0;i<4-numberLength;i++)
+    {
+        actualMessage[i]=0x20;
+    }
+    for(int i=4-numberLength,j=0;i<4;i++,j++)
+    {
+        actualMessage[i]=numBuffer[j];
+    }
+    memset(actualMessage+4, messageChar, tailleMessage);
+}
+
+int exitMax(int var,int tailleMax){
+    if(var>1500){
+        printf("-l doit être <1500 \n");
+        exit(EXIT_FAILURE);
+    }
+    return 0;
+}