mime decode: improve MimeDecParseLineTest01 and MimeDecParseLineTest02 tests

diff --git a/src/mime-decode.c b/src/mime-decode.c
index f2f9d24a500988e92104c5e27479928105c3a241..771cfe6d91a02187e4393bbf86ba6fa59230f9cd 100644 (file)
--- a/src/mime-decode.c
+++ b/src/mime-decode.c
@@ -2620,26 +2620,29 @@ static int MimeDecParseLineTest01(void) {
             TestDataChunkCallback);
 
     char *str = "From: Sender1";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
     str = "To: Recipient1";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
     str = "Content-Type: text/plain";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
     str = "";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
     str = "A simple message line 1";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
     str = "A simple message line 2";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
     str = "A simple message line 3";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
+    if (ret != MIME_DEC_OK) {
+        return ret;
+    }
     /* Completed */
     ret = MimeDecParseComplete(state);
     if (ret != MIME_DEC_OK) {
@@ -2684,24 +2687,27 @@ static int MimeDecParseLineTest02(void) {
             TestDataChunkCallback);
 
     char *str = "From: Sender1";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
     str = "To: Recipient1";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
     str = "Content-Type: text/plain";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
     str = "";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
     str = "A simple message line 1";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
     str = "A simple message line 2 click on http://www.test.com/malware.exe?"
             "hahah hopefully you click this link";
-    ret = MimeDecParseLine((uint8_t *)str, strlen(str), state);
+    ret |= MimeDecParseLine((uint8_t *)str, strlen(str), state);
 
+    if (ret != MIME_DEC_OK) {
+        return ret;
+    }
     /* Completed */
     ret = MimeDecParseComplete(state);
     if (ret != MIME_DEC_OK) {