267b597b12777e9034a8334718a910a359853390
[WebKit-https.git] / Source / WebCore / platform / network / MIMEHeader.cpp
1 /*
2  * Copyright (C) 2011 Google Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  *
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #include "config.h"
32 #include "MIMEHeader.h"
33
34 #include "ParsedContentType.h"
35 #include "SharedBufferChunkReader.h"
36 #include <wtf/HashMap.h>
37 #include <wtf/text/CString.h>
38 #include <wtf/text/StringBuilder.h>
39 #include <wtf/text/StringConcatenate.h>
40 #include <wtf/text/StringHash.h>
41
42 namespace WebCore {
43
44 typedef HashMap<String, String> KeyValueMap;
45
46 static KeyValueMap retrieveKeyValuePairs(WebCore::SharedBufferChunkReader& buffer)
47 {
48     KeyValueMap keyValuePairs;
49     String line;
50     String key;
51     StringBuilder value;
52     while (!(line = buffer.nextChunkAsUTF8StringWithLatin1Fallback()).isNull()) {
53         if (line.isEmpty())
54             break; // Empty line means end of key/value section.
55         if (line[0] == '\t') {
56             ASSERT(!key.isEmpty());
57             value.append(line.substring(1));
58             continue;
59         }
60         // New key/value, store the previous one if any.
61         if (!key.isEmpty()) {
62             if (keyValuePairs.find(key) != keyValuePairs.end())
63                 LOG_ERROR("Key duplicate found in MIME header. Key is '%s', previous value replaced.", key.ascii().data());
64             keyValuePairs.add(key, value.toString().stripWhiteSpace());
65             key = String();
66             value.clear();
67         }
68         size_t semiColonIndex = line.find(':');
69         if (semiColonIndex == notFound) {
70             // This is not a key value pair, ignore.
71             continue;
72         }
73         key = line.substring(0, semiColonIndex).convertToASCIILowercase().stripWhiteSpace();
74         value.append(line.substring(semiColonIndex + 1));
75     }
76     // Store the last property if there is one.
77     if (!key.isEmpty())
78         keyValuePairs.set(key, value.toString().stripWhiteSpace());
79     return keyValuePairs;
80 }
81
82 RefPtr<MIMEHeader> MIMEHeader::parseHeader(SharedBufferChunkReader& buffer)
83 {
84     auto mimeHeader = adoptRef(*new MIMEHeader);
85     KeyValueMap keyValuePairs = retrieveKeyValuePairs(buffer);
86     KeyValueMap::iterator mimeParametersIterator = keyValuePairs.find("content-type");
87     if (mimeParametersIterator != keyValuePairs.end()) {
88         ParsedContentType parsedContentType(mimeParametersIterator->value);
89         mimeHeader->m_contentType = parsedContentType.mimeType();
90         if (!mimeHeader->isMultipart())
91             mimeHeader->m_charset = parsedContentType.charset().stripWhiteSpace();
92         else {
93             mimeHeader->m_multipartType = parsedContentType.parameterValueForName("type");
94             mimeHeader->m_endOfPartBoundary = parsedContentType.parameterValueForName("boundary");
95             if (mimeHeader->m_endOfPartBoundary.isNull()) {
96                 LOG_ERROR("No boundary found in multipart MIME header.");
97                 return nullptr;
98             }
99             mimeHeader->m_endOfPartBoundary = "--" + mimeHeader->m_endOfPartBoundary;
100             mimeHeader->m_endOfDocumentBoundary = mimeHeader->m_endOfPartBoundary + "--";
101         }
102     }
103
104     mimeParametersIterator = keyValuePairs.find("content-transfer-encoding");
105     if (mimeParametersIterator != keyValuePairs.end())
106         mimeHeader->m_contentTransferEncoding = parseContentTransferEncoding(mimeParametersIterator->value);
107
108     mimeParametersIterator = keyValuePairs.find("content-location");
109     if (mimeParametersIterator != keyValuePairs.end())
110         mimeHeader->m_contentLocation = mimeParametersIterator->value;
111
112     return WTFMove(mimeHeader);
113 }
114
115 MIMEHeader::Encoding MIMEHeader::parseContentTransferEncoding(const String& text)
116 {
117     String encoding = text.stripWhiteSpace();
118     if (equalLettersIgnoringASCIICase(encoding, "base64"))
119         return Base64;
120     if (equalLettersIgnoringASCIICase(encoding, "quoted-printable"))
121         return QuotedPrintable;
122     if (equalLettersIgnoringASCIICase(encoding, "7bit"))
123         return SevenBit;
124     if (equalLettersIgnoringASCIICase(encoding, "binary"))
125         return Binary;
126     LOG_ERROR("Unknown encoding '%s' found in MIME header.", text.ascii().data());
127     return Unknown;
128 }
129
130 MIMEHeader::MIMEHeader()
131     : m_contentTransferEncoding(Unknown)
132 {
133 }
134
135 }