d183af123b83f27a91ef0d66255b6ddb0c02a5d5
[WebKit-https.git] / Source / WebCore / platform / chromium / support / 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 <public/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 bool WebHTTPBody::containsPasswordData() const
154 {
155     return m_private->containsPasswordData();
156 }
157
158 void WebHTTPBody::setContainsPasswordData(bool containsPasswordData)
159 {
160     m_private->setContainsPasswordData(containsPasswordData);
161 }
162
163 WebHTTPBody::WebHTTPBody(const PassRefPtr<FormData>& data)
164     : m_private(static_cast<WebHTTPBodyPrivate*>(data.leakRef()))
165 {
166 }
167
168 WebHTTPBody& WebHTTPBody::operator=(const PassRefPtr<FormData>& data)
169 {
170     assign(static_cast<WebHTTPBodyPrivate*>(data.leakRef()));
171     return *this;
172 }
173
174 WebHTTPBody::operator PassRefPtr<FormData>() const
175 {
176     return m_private;
177 }
178
179 void WebHTTPBody::assign(WebHTTPBodyPrivate* p)
180 {
181     // p is already ref'd for us by the caller
182     if (m_private)
183         m_private->deref();
184     m_private = p;
185 }
186
187 void WebHTTPBody::ensureMutable()
188 {
189     ASSERT(!isNull());
190     if (!m_private->hasOneRef())
191         assign(static_cast<WebHTTPBodyPrivate*>(m_private->copy().leakRef()));
192 }
193
194 } // namespace WebKit