Introduce a new function to test content type content

Create a new function to ensure that content type matching is always done
case insensitive.

Originally the comparison was done after a toLower(), but forgetting this, like
in src/libraries/qmfclient/qmailmessage.cpp#6252, can create bugs.

Change-Id: Ia93e068e5374c346c94c2afe4de768d3833b4a9a
Reviewed-by: Pekka Vuorela <pvuorela@iki.fi>

1 files changed, 27 insertions, 36 deletions

diff --git a/src/libraries/qmfclient/qmailmessage.cpp b/src/libraries/qmfclient/qmailmessage.cpp
index 867d0aa2..98108179 100644
--- a/src/libraries/qmfclient/qmailmessage.cpp
+++ b/src/libraries/qmfclient/qmailmessage.cpp
@@ -917,11 +917,7 @@ namespace findBody
 
     bool inMultipartNone(const QMailMessagePartContainer &container, Context &ctx)
     {
-        if (!ctx.contentType.isEmpty()
-        && ctx.contentType != container.contentType().type().toLower())
-            return false;
-        if (!ctx.contentSubtype.isEmpty()
-        && ctx.contentSubtype != container.contentType().subType().toLower())
+        if (!container.contentType().matches(ctx.contentType, ctx.contentSubtype))
             return false;
         ctx.found = const_cast<QMailMessagePartContainer*>(&container);
         return true;
@@ -932,12 +928,7 @@ namespace findBody
         if (part.contentDisposition().type() == QMailMessageContentDisposition::Attachment)
             return false;
 
-        if (!ctx.contentType.isEmpty()
-        && ctx.contentType != part.contentType().type().toLower())
-            return false;
-
-        if (!ctx.contentSubtype.isEmpty()
-        && ctx.contentSubtype != part.contentType().subType().toLower())
+        if (!part.contentType().matches(ctx.contentType, ctx.contentSubtype))
             return false;
 
         ctx.found = const_cast<QMailMessagePart*> (&part);
@@ -1012,7 +1003,7 @@ namespace findBody
         for (int i = (int)container.partCount() - 1; i >= 0; i--) {
             if (i != bodyPart) {
                 const QMailMessagePart &part = container.partAt(i);
-                if (imageContentType == part.contentType().type().toLower()) {
+                if (part.contentType().matches(imageContentType)) {
                     ctx.htmlImageLoc << part.location();
                     ctx.htmlImageParts << &part;
                 } else if (!part.contentID().isEmpty()) {
@@ -1166,8 +1157,8 @@ namespace findAttachments
         {
             QMailMessageContentType contentType = part.contentType();
 
-            bool isText = (contentType.type().toLower() == "text") &&
-                ((contentType.subType().toLower() == "plain") || (contentType.subType().toLower() == "html"));
+            bool isText = (contentType.matches("text", "plain")
+                           || contentType.matches("text", "html"));
 
             bool isCalendar = (contentType.type().toLower() == "text") &&
                 (contentType.subType().toLower() == "calendar");
@@ -1178,8 +1169,7 @@ namespace findAttachments
             bool isAttachment = (!part.contentDisposition().isNull()) &&
                 (part.contentDisposition().type() == QMailMessageContentDisposition::Attachment);
 
-            bool isRFC822 = (contentType.type().toLower() == "message") &&
-                (contentType.subType().toLower() == "rfc822");
+            bool isRFC822 = contentType.matches("message", "rfc822");
 
             // Attached messages are considered as attachments even if content disposition
             // is inline instead of attachment, but only if they aren't text/plain nor text/html
@@ -1272,18 +1262,17 @@ namespace findAttachments
                 const QMailMessagePart &part = container.partAt(i);
 
                 // Skip parts of the message body
-                const QString contentType = QString::fromLatin1(part.contentType().content()).toLower();
                 switch (i) {
                 case 0:
-                    if (contentType == QLatin1String("text/plain")) {
+                    if (part.contentType().matches("text", "plain")) {
                         firstPartIsTextPlain = true;
                         continue;
-                    } else if (contentType == QLatin1String("text/html")) {
+                    } else if (part.contentType().matches("text", "html")) {
                         continue;
                     }
                     break;
                 case 1:
-                    if (firstPartIsTextPlain && contentType == QLatin1String("text/html")) {
+                    if (firstPartIsTextPlain && part.contentType().matches("text", "html")) {
                         continue;
                     }
                     break;
@@ -2688,6 +2677,15 @@ void QMailMessageContentType::setCharset(const QByteArray& charset)
     setParameter("charset", charset);
 }
 
+/*!
+    Allow to test if the content type matches a specific "type / subtype" string.
+*/
+bool QMailMessageContentType::matches(const QByteArray& primary, const QByteArray& sub) const
+{
+    // Content type is not encoded, we can use qstricmp directly.
+    return ((primary.isEmpty() || qstricmp(type(), primary) == 0)
+            && (sub.isEmpty() || qstricmp(subType(), sub) == 0));
+}
 
 /*!
     \class QMailMessageContentDisposition
@@ -3179,9 +3177,7 @@ QMailMessageBodyPrivate::QMailMessageBodyPrivate()
 
 void QMailMessageBodyPrivate::ensureCharsetExist()
 {
-    if (_type.type().toLower() != "text"
-        || (_type.subType().toLower() != "plain"
-            && _type.subType().toLower() != "html")) {
+    if (!_type.matches("text", "plain") && !_type.matches("text", "html")) {
         QByteArray best(QMailCodec::bestCompatibleCharset(_type.charset(), true));
         if (!best.isEmpty()) {
             _type.setCharset(best);
@@ -4037,7 +4033,7 @@ void QMailMessagePartContainerPrivate::defaultContentType(const QMailMessagePart
     {
         // Note that the default is 'message/rfc822' when the parent is 'multipart/digest'
         QMailMessageContentType parentType = parent->contentType();
-        if (parentType.content().toLower() == "multipart/digest")
+        if (parentType.matches("multipart", "digest"))
         {
             type.setType("message");
             type.setSubType("rfc822");
@@ -4211,7 +4207,7 @@ void QMailMessagePartContainerPrivate::setBody(const QMailMessageBody& body, QMa
     setBodyProperties(body.contentType(), body.transferEncoding());
 
     // Multipart messages do not have their own bodies
-    if (body.contentType().type().toLower() != "multipart") {
+    if (!body.contentType().matches("multipart")) {
         _body = body;
         _hasBody = !_body.isEmpty();
     }
@@ -6104,14 +6100,10 @@ QString QMailMessagePart::displayName() const
     if (id.isEmpty())
         id = contentID();
 
-    if (id.isEmpty()) {
-        bool isRFC822 = (contentType().type().toLower() == "message") &&
-            (contentType().subType().toLower() == "rfc822");
-        if (isRFC822) {
-            // TODO don't load entire body into memory
-            QMailMessage msg = QMailMessage::fromRfc2822(body().data(QMailMessageBody::Decoded));
-            id = msg.subject();
-        }
+    if (id.isEmpty() && contentType().matches("message", "rfc822")) {
+        // TODO don't load entire body into memory
+        QMailMessage msg = QMailMessage::fromRfc2822(body().data(QMailMessageBody::Decoded));
+        id = msg.subject();
     }
 
     if (id.isEmpty()) {
@@ -6243,7 +6235,7 @@ static QString partFileName(const QMailMessagePart &part)
     // Application/octet-stream is a fallback for the case when mimetype
     // is unknown by MUA, so it's treated exceptionally here, so the original
     // attachment name is preserved
-    if (part.contentType().content() != QByteArray("application/octet-stream")) {
+    if (!part.contentType().matches("application", "octet-stream")) {
         // If possible, create the file with a useful filename extension
         QString existing;
         int index = fileName.lastIndexOf('.');
@@ -8472,8 +8464,7 @@ bool QMailMessage::hasCalendarInvitation() const
             }
         } else {
             const QMailMessageContentType &ct(part->contentType());
-            if ((ct.type().toLower() == "text") &&
-                (ct.subType().toLower() == "calendar") &&
+            if (ct.matches("text", "calendar") &&
                 (ct.parameter("method").toLower() == "request")) {
                 return true;
             }