1 /* 2 * Copyright (C) 2007 Alp Toker <alp (at) atoker.com> 3 * Copyright (C) 2010 Patrick Gansterer <paroga (at) paroga.com> 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY 15 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 17 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR 18 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 19 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 20 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 21 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY 22 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 24 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25 */ 26 27 #include "config.h" 28 #include "DataURL.h" 29 30 #include "Base64.h" 31 #include "HTTPParsers.h" 32 #include "ResourceHandle.h" 33 #include "ResourceHandleClient.h" 34 #include "ResourceRequest.h" 35 #include "ResourceResponse.h" 36 #include "TextEncoding.h" 37 #include <wtf/text/CString.h> 38 39 namespace WebCore { 40 41 void handleDataURL(ResourceHandle* handle) 42 { 43 ASSERT(handle->firstRequest().url().protocolIs("data")); 44 String url = handle->firstRequest().url().string(); 45 46 int index = url.find(','); 47 if (index == -1) { 48 handle->client()->cannotShowURL(handle); 49 return; 50 } 51 52 String mediaType = url.substring(5, index - 5); 53 String data = url.substring(index + 1); 54 55 bool base64 = mediaType.endsWith(";base64", false); 56 if (base64) 57 mediaType = mediaType.left(mediaType.length() - 7); 58 59 if (mediaType.isEmpty()) 60 mediaType = "text/plain;charset=US-ASCII"; 61 62 String mimeType = extractMIMETypeFromMediaType(mediaType); 63 String charset = extractCharsetFromMediaType(mediaType); 64 65 ResourceResponse response; 66 response.setMimeType(mimeType); 67 response.setTextEncodingName(charset); 68 response.setURL(handle->firstRequest().url()); 69 70 if (base64) { 71 data = decodeURLEscapeSequences(data); 72 handle->client()->didReceiveResponse(handle, response); 73 74 Vector<char> out; 75 if (base64Decode(data, out, IgnoreWhitespace) && out.size() > 0) { 76 response.setExpectedContentLength(out.size()); 77 handle->client()->didReceiveData(handle, out.data(), out.size(), 0); 78 } 79 } else { 80 data = decodeURLEscapeSequences(data, TextEncoding(charset)); 81 handle->client()->didReceiveResponse(handle, response); 82 83 CString encodedData = TextEncoding().encode(data.characters(), data.length(), URLEncodedEntitiesForUnencodables); 84 response.setExpectedContentLength(encodedData.length()); 85 if (encodedData.length()) 86 handle->client()->didReceiveData(handle, encodedData.data(), encodedData.length(), 0); 87 } 88 89 handle->client()->didFinishLoading(handle, 0); 90 } 91 92 } // namespace WebCore 93