src/third_party/WebKit/Source/platform/network/FormData.cpp

   1 /*
   2  * Copyright (C) 2004, 2006, 2008, 2011 Apple Inc. All rights reserved.
   3  * Copyright (C) 2009 Google Inc. All rights reserved.
   4  * Copyright (C) 2012 Digia Plc. and/or its subsidiary(-ies)
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Library General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14  * Library General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Library General Public License
  17  * along with this library; see the file COPYING.LIB. If not, write to
  18  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  19  * Boston, MA 02110-1301, USA.
  20  */
  21
  22 #include "config.h"
  23 #include "platform/network/FormData.h"
  24
  25 #include "platform/FileMetadata.h"
  26 #include "platform/network/FormDataBuilder.h"
  27 #include "wtf/text/CString.h"
  28 #include "wtf/text/TextEncoding.h"
  29
  30 namespace blink {
  31
  32 inline FormData::FormData()
  33     : m_identifier(0)
  34     , m_alwaysStream(false)
  35     , m_containsPasswordData(false)
  36 {
  37 }
  38
  39 inline FormData::FormData(const FormData& data)
  40     : RefCounted<FormData>()
  41     , m_elements(data.m_elements)
  42     , m_identifier(data.m_identifier)
  43     , m_alwaysStream(false)
  44     , m_containsPasswordData(data.m_containsPasswordData)
  45 {
  46 }
  47
  48 FormData::~FormData()
  49 {
  50 }
  51
  52 PassRefPtr<FormData> FormData::create()
  53 {
  54     return adoptRef(new FormData);
  55 }
  56
  57 PassRefPtr<FormData> FormData::create(const void* data, size_t size)
  58 {
  59     RefPtr<FormData> result = create();
  60     result->appendData(data, size);
  61     return result.release();
  62 }
  63
  64 PassRefPtr<FormData> FormData::create(const CString& string)
  65 {
  66     RefPtr<FormData> result = create();
  67     result->appendData(string.data(), string.length());
  68     return result.release();
  69 }
  70
  71 PassRefPtr<FormData> FormData::create(const Vector<char>& vector)
  72 {
  73     RefPtr<FormData> result = create();
  74     result->appendData(vector.data(), vector.size());
  75     return result.release();
  76 }
  77
  78 PassRefPtr<FormData> FormData::copy() const
  79 {
  80     return adoptRef(new FormData(*this));
  81 }
  82
  83 PassRefPtr<FormData> FormData::deepCopy() const
  84 {
  85     RefPtr<FormData> formData(create());
  86
  87     formData->m_alwaysStream = m_alwaysStream;
  88
  89     size_t n = m_elements.size();
  90     formData->m_elements.reserveInitialCapacity(n);
  91     for (size_t i = 0; i < n; ++i) {
  92         const FormDataElement& e = m_elements[i];
  93         switch (e.m_type) {
  94         case FormDataElement::data:
  95             formData->m_elements.uncheckedAppend(FormDataElement(e.m_data));
  96             break;
  97         case FormDataElement::encodedFile:
  98             formData->m_elements.uncheckedAppend(FormDataElement(e.m_filename, e.m_fileStart, e.m_fileLength, e.m_expectedFileModificationTime));
  99             break;
 100         case FormDataElement::encodedBlob:
 101             formData->m_elements.uncheckedAppend(FormDataElement(e.m_blobUUID, e.m_optionalBlobDataHandle));
 102             break;
 103         case FormDataElement::encodedFileSystemURL:
 104             formData->m_elements.uncheckedAppend(FormDataElement(e.m_fileSystemURL, e.m_fileStart, e.m_fileLength, e.m_expectedFileModificationTime));
 105             break;
 106         }
 107     }
 108     return formData.release();
 109 }
 110
 111 void FormData::appendData(const void* data, size_t size)
 112 {
 113     if (m_elements.isEmpty() || m_elements.last().m_type != FormDataElement::data)
 114         m_elements.append(FormDataElement());
 115     FormDataElement& e = m_elements.last();
 116     size_t oldSize = e.m_data.size();
 117     e.m_data.grow(oldSize + size);
 118     memcpy(e.m_data.data() + oldSize, data, size);
 119 }
 120
 121 void FormData::appendFile(const String& filename)
 122 {
 123     m_elements.append(FormDataElement(filename, 0, BlobDataItem::toEndOfFile, invalidFileTime()));
 124 }
 125
 126 void FormData::appendFileRange(const String& filename, long long start, long long length, double expectedModificationTime)
 127 {
 128     m_elements.append(FormDataElement(filename, start, length, expectedModificationTime));
 129 }
 130
 131 void FormData::appendBlob(const String& uuid, PassRefPtr<BlobDataHandle> optionalHandle)
 132 {
 133     m_elements.append(FormDataElement(uuid, optionalHandle));
 134 }
 135
 136 void FormData::appendFileSystemURL(const KURL& url)
 137 {
 138     m_elements.append(FormDataElement(url, 0, BlobDataItem::toEndOfFile, invalidFileTime()));
 139 }
 140
 141 void FormData::appendFileSystemURLRange(const KURL& url, long long start, long long length, double expectedModificationTime)
 142 {
 143     m_elements.append(FormDataElement(url, start, length, expectedModificationTime));
 144 }
 145
 146 void FormData::flatten(Vector<char>& data) const
 147 {
 148     // Concatenate all the byte arrays, but omit any files.
 149     data.clear();
 150     size_t n = m_elements.size();
 151     for (size_t i = 0; i < n; ++i) {
 152         const FormDataElement& e = m_elements[i];
 153         if (e.m_type == FormDataElement::data)
 154             data.append(e.m_data.data(), static_cast<size_t>(e.m_data.size()));
 155     }
 156 }
 157
 158 String FormData::flattenToString() const
 159 {
 160     Vector<char> bytes;
 161     flatten(bytes);
 162     return Latin1Encoding().decode(reinterpret_cast<const char*>(bytes.data()), bytes.size());
 163 }
 164
 165 unsigned long long FormData::sizeInBytes() const
 166 {
 167     unsigned size = 0;
 168     size_t n = m_elements.size();
 169     for (size_t i = 0; i < n; ++i) {
 170         const FormDataElement& e = m_elements[i];
 171         switch (e.m_type) {
 172         case FormDataElement::data:
 173             size += e.m_data.size();
 174             break;
 175         case FormDataElement::encodedFile:
 176             size += e.m_fileLength - e.m_fileStart;
 177             break;
 178         case FormDataElement::encodedBlob:
 179             if (e.m_optionalBlobDataHandle)
 180                 size += e.m_optionalBlobDataHandle->size();
 181             break;
 182         case FormDataElement::encodedFileSystemURL:
 183             size += e.m_fileLength - e.m_fileStart;
 184             break;
 185         }
 186     }
 187     return size;
 188 }
 189
 190 } // namespace blink