Source/WebKit/chromium/src/WebHTTPBody.cpp

   1 /*
   2  * Copyright (C) 2009 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 "platform/WebHTTPBody.h"
  33
  34 #include "FormData.h"
  35
  36 using namespace WebCore;
  37
  38 namespace WebKit {
  39
  40 class WebHTTPBodyPrivate : public FormData {
  41 };
  42
  43 void WebHTTPBody::initialize()
  44 {
  45     assign(static_cast<WebHTTPBodyPrivate*>(FormData::create().leakRef()));
  46 }
  47
  48 void WebHTTPBody::reset()
  49 {
  50     assign(0);
  51 }
  52
  53 void WebHTTPBody::assign(const WebHTTPBody& other)
  54 {
  55     WebHTTPBodyPrivate* p = const_cast<WebHTTPBodyPrivate*>(other.m_private);
  56     if (p)
  57         p->ref();
  58     assign(p);
  59 }
  60
  61 size_t WebHTTPBody::elementCount() const
  62 {
  63     ASSERT(!isNull());
  64     return m_private->elements().size();
  65 }
  66
  67 bool WebHTTPBody::elementAt(size_t index, Element& result) const
  68 {
  69     ASSERT(!isNull());
  70
  71     if (index >= m_private->elements().size())
  72         return false;
  73
  74     const FormDataElement& element = m_private->elements()[index];
  75
  76     result.data.reset();
  77     result.filePath.reset();
  78     result.fileStart = 0;
  79     result.fileLength = 0;
  80     result.modificationTime = 0.0;
  81     result.blobURL = KURL();
  82
  83     switch (element.m_type) {
  84     case FormDataElement::data:
  85         result.type = Element::TypeData;
  86         result.data.assign(element.m_data.data(), element.m_data.size());
  87         break;
  88     case FormDataElement::encodedFile:
  89         result.type = Element::TypeFile;
  90         result.filePath = element.m_filename;
  91 #if ENABLE(BLOB)
  92         result.fileStart = element.m_fileStart;
  93         result.fileLength = element.m_fileLength;
  94         result.modificationTime = element.m_expectedFileModificationTime;
  95 #endif
  96         break;
  97 #if ENABLE(BLOB)
  98     case FormDataElement::encodedBlob:
  99         result.type = Element::TypeBlob;
 100         result.blobURL = element.m_blobURL;
 101         break;
 102 #endif
 103     default:
 104         ASSERT_NOT_REACHED();
 105         return false;
 106     }
 107
 108     return true;
 109 }
 110
 111 void WebHTTPBody::appendData(const WebData& data)
 112 {
 113     ensureMutable();
 114     // FIXME: FormDataElement::m_data should be a SharedBuffer<char>.  Then we
 115     // could avoid this buffer copy.
 116     m_private->appendData(data.data(), data.size());
 117 }
 118
 119 void WebHTTPBody::appendFile(const WebString& filePath)
 120 {
 121     ensureMutable();
 122     m_private->appendFile(filePath);
 123 }
 124
 125 void WebHTTPBody::appendFileRange(const WebString& filePath, long long fileStart, long long fileLength, double modificationTime)
 126 {
 127 #if ENABLE(BLOB)
 128     ensureMutable();
 129     m_private->appendFileRange(filePath, fileStart, fileLength, modificationTime);
 130 #endif
 131 }
 132
 133 void WebHTTPBody::appendBlob(const WebURL& blobURL)
 134 {
 135 #if ENABLE(BLOB)
 136     ensureMutable();
 137     m_private->appendBlob(blobURL);
 138 #endif
 139 }
 140
 141 long long WebHTTPBody::identifier() const
 142 {
 143     ASSERT(!isNull());
 144     return m_private->identifier();
 145 }
 146
 147 void WebHTTPBody::setIdentifier(long long identifier)
 148 {
 149     ensureMutable();
 150     return m_private->setIdentifier(identifier);
 151 }
 152
 153 WebHTTPBody::WebHTTPBody(const PassRefPtr<FormData>& data)
 154     : m_private(static_cast<WebHTTPBodyPrivate*>(data.leakRef()))
 155 {
 156 }
 157
 158 WebHTTPBody& WebHTTPBody::operator=(const PassRefPtr<FormData>& data)
 159 {
 160     assign(static_cast<WebHTTPBodyPrivate*>(data.leakRef()));
 161     return *this;
 162 }
 163
 164 WebHTTPBody::operator PassRefPtr<FormData>() const
 165 {
 166     return m_private;
 167 }
 168
 169 void WebHTTPBody::assign(WebHTTPBodyPrivate* p)
 170 {
 171     // p is already ref'd for us by the caller
 172     if (m_private)
 173         m_private->deref();
 174     m_private = p;
 175 }
 176
 177 void WebHTTPBody::ensureMutable()
 178 {
 179     ASSERT(!isNull());
 180     if (!m_private->hasOneRef())
 181         assign(static_cast<WebHTTPBodyPrivate*>(m_private->copy().leakRef()));
 182 }
 183
 184 } // namespace WebKit