1/* 2 * Copyright (C) 2007 Alp Toker <alp@atoker.com> 3 * Copyright (C) 2010 Patrick Gansterer <paroga@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 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 "HTTPParsers.h" 31#include "ResourceHandle.h" 32#include "ResourceHandleClient.h" 33#include "ResourceRequest.h" 34#include "ResourceResponse.h" 35#include "TextEncoding.h" 36#include <wtf/text/Base64.h> 37#include <wtf/text/CString.h> 38#include <wtf/text/StringView.h> 39 40namespace WebCore { 41 42void handleDataURL(ResourceHandle* handle) 43{ 44 ASSERT(handle->firstRequest().url().protocolIsData()); 45 String url = handle->firstRequest().url().string(); 46 47 int index = url.find(','); 48 if (index == -1) { 49 handle->client()->cannotShowURL(handle); 50 return; 51 } 52 53 String mediaType = url.substring(5, index - 5); 54 String data = url.substring(index + 1); 55 56 bool base64 = mediaType.endsWith(";base64", false); 57 if (base64) 58 mediaType = mediaType.left(mediaType.length() - 7); 59 60 if (mediaType.isEmpty()) 61 mediaType = "text/plain"; 62 63 String mimeType = extractMIMETypeFromMediaType(mediaType); 64 String charset = extractCharsetFromMediaType(mediaType); 65 66 if (charset.isEmpty()) 67 charset = "US-ASCII"; 68 69 ResourceResponse response; 70 response.setMimeType(mimeType); 71 response.setTextEncodingName(charset); 72 response.setURL(handle->firstRequest().url()); 73 74 if (base64) { 75 data = decodeURLEscapeSequences(data); 76 handle->client()->didReceiveResponse(handle, response); 77 78 Vector<char> out; 79 if (base64Decode(data, out, Base64IgnoreWhitespace) && out.size() > 0) { 80 response.setExpectedContentLength(out.size()); 81 handle->client()->didReceiveData(handle, out.data(), out.size(), 0); 82 } 83 } else { 84 TextEncoding encoding(charset); 85 data = decodeURLEscapeSequences(data, encoding); 86 handle->client()->didReceiveResponse(handle, response); 87 88 CString encodedData = encoding.encode(data, URLEncodedEntitiesForUnencodables); 89 response.setExpectedContentLength(encodedData.length()); 90 if (encodedData.length()) 91 handle->client()->didReceiveData(handle, encodedData.data(), encodedData.length(), 0); 92 } 93 94 handle->client()->didFinishLoading(handle, 0); 95} 96 97} // namespace WebCore 98