code.vuplus.com Git - vuplus_webkit/blob - Source/WebCore/platform/network/DataURL.cpp

   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 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";
  61
  62     String mimeType = extractMIMETypeFromMediaType(mediaType);
  63     String charset = extractCharsetFromMediaType(mediaType);
  64
  65     if (charset.isEmpty())
  66         charset = "US-ASCII";
  67
  68     ResourceResponse response;
  69     response.setMimeType(mimeType);
  70     response.setTextEncodingName(charset);
  71     response.setURL(handle->firstRequest().url());
  72
  73     if (base64) {
  74         data = decodeURLEscapeSequences(data);
  75         handle->client()->didReceiveResponse(handle, response);
  76
  77         Vector<char> out;
  78         if (base64Decode(data, out, IgnoreWhitespace) && out.size() > 0) {
  79             response.setExpectedContentLength(out.size());
  80             handle->client()->didReceiveData(handle, out.data(), out.size(), 0);
  81         }
  82     } else {
  83         TextEncoding encoding(charset);
  84         data = decodeURLEscapeSequences(data, encoding);
  85         handle->client()->didReceiveResponse(handle, response);
  86
  87         CString encodedData = encoding.encode(data.characters(), data.length(), URLEncodedEntitiesForUnencodables);
  88         response.setExpectedContentLength(encodedData.length());
  89         if (encodedData.length())
  90             handle->client()->didReceiveData(handle, encodedData.data(), encodedData.length(), 0);
  91     }
  92
  93     handle->client()->didFinishLoading(handle, 0);
  94 }
  95
  96 } // namespace WebCore