2009-01-23 David Levin <levin@chromium.org>
[WebKit-https.git] / WebCore / platform / network / FormData.cpp
1 /*
2  * Copyright (C) 2004, 2006, 2008 Apple Inc. All rights reserved.
3  * Copyright (C) 2009 Google Inc. All rights reserved.
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Library General Public
7  * License as published by the Free Software Foundation; either
8  * version 2 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13  * Library General Public License for more details.
14  *
15  * You should have received a copy of the GNU Library General Public License
16  * along with this library; see the file COPYING.LIB. If not, write to
17  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
18  * Boston, MA 02110-1301, USA.
19  */
20
21 #include "config.h"
22 #include "FormData.h"
23
24 #include "CString.h"
25 #include "ChromeClient.h"
26 #include "FileSystem.h"
27 #include "TextEncoding.h"
28
29 namespace WebCore {
30
31 inline FormData::FormData()
32     : m_hasGeneratedFiles(false)
33     , m_alwaysStream(false)
34 {
35 }
36
37 inline FormData::FormData(const FormData& data)
38     : RefCounted<FormData>()
39     , m_elements(data.m_elements)
40     , m_hasGeneratedFiles(false)
41     , m_alwaysStream(false)
42 {
43     // We shouldn't be copying FormData that hasn't already removed its generated files
44     // but just in case, make sure the new FormData is ready to generate its own files.
45     if (data.m_hasGeneratedFiles) {
46         size_t n = m_elements.size();
47         for (size_t i = 0; i < n; ++i) {
48             FormDataElement& e = m_elements[i];
49             if (e.m_type == FormDataElement::encodedFile)
50                 e.m_generatedFilename = String();
51         }
52     }
53 }
54
55 FormData::~FormData()
56 {
57     // This cleanup should've happened when the form submission finished.
58     // Just in case, let's assert, and do the cleanup anyway in release builds.
59     ASSERT(!m_hasGeneratedFiles);
60     removeGeneratedFilesIfNeeded();
61 }
62
63 PassRefPtr<FormData> FormData::create()
64 {
65     return adoptRef(new FormData);
66 }
67
68 PassRefPtr<FormData> FormData::create(const void* data, size_t size)
69 {
70     RefPtr<FormData> result = create();
71     result->appendData(data, size);
72     return result.release();
73 }
74
75 PassRefPtr<FormData> FormData::create(const CString& string)
76 {
77     RefPtr<FormData> result = create();
78     result->appendData(string.data(), string.length());
79     return result.release();
80 }
81
82 PassRefPtr<FormData> FormData::create(const Vector<char>& vector)
83 {
84     RefPtr<FormData> result = create();
85     result->appendData(vector.data(), vector.size());
86     return result.release();
87 }
88
89 PassRefPtr<FormData> FormData::copy() const
90 {
91     return adoptRef(new FormData(*this));
92 }
93
94 PassRefPtr<FormData> FormData::deepCopy() const
95 {
96     RefPtr<FormData> formData(create());
97
98     formData->m_alwaysStream = m_alwaysStream;
99
100     size_t n = m_elements.size();
101     formData->m_elements.reserveCapacity(n);
102     for (size_t i = 0; i < n; ++i) {
103         const FormDataElement& e = m_elements[i];
104         switch (e.m_type) {
105         case FormDataElement::data:
106             formData->m_elements.append(FormDataElement(e.m_data));
107             break;
108         case FormDataElement::encodedFile:
109             formData->m_elements.append(FormDataElement(e.m_filename, e.m_shouldGenerateFile));
110             break;
111         }
112     }
113     return formData.release();
114 }
115
116 void FormData::appendData(const void* data, size_t size)
117 {
118     if (m_elements.isEmpty() || m_elements.last().m_type != FormDataElement::data)
119         m_elements.append(FormDataElement());
120     FormDataElement& e = m_elements.last();
121     size_t oldSize = e.m_data.size();
122     e.m_data.grow(oldSize + size);
123     memcpy(e.m_data.data() + oldSize, data, size);
124 }
125
126 void FormData::appendFile(const String& filename, bool shouldGenerateFile)
127 {
128     m_elements.append(FormDataElement(filename, shouldGenerateFile));
129 }
130
131 void FormData::flatten(Vector<char>& data) const
132 {
133     // Concatenate all the byte arrays, but omit any files.
134     data.clear();
135     size_t n = m_elements.size();
136     for (size_t i = 0; i < n; ++i) {
137         const FormDataElement& e = m_elements[i];
138         if (e.m_type == FormDataElement::data) {
139             size_t oldSize = data.size();
140             size_t delta = e.m_data.size();
141             data.grow(oldSize + delta);
142             memcpy(data.data() + oldSize, e.m_data.data(), delta);
143         }
144     }
145 }
146
147 String FormData::flattenToString() const
148 {
149     Vector<char> bytes;
150     flatten(bytes);
151     return Latin1Encoding().decode(bytes.data(), bytes.size());
152 }
153
154 void FormData::generateFiles(ChromeClient* client)
155 {
156     ASSERT(!m_hasGeneratedFiles);
157     
158     if (m_hasGeneratedFiles)
159         return;
160     
161     size_t n = m_elements.size();
162     for (size_t i = 0; i < n; ++i) {
163         FormDataElement& e = m_elements[i];
164         if (e.m_type == FormDataElement::encodedFile && e.m_shouldGenerateFile) {
165             e.m_generatedFilename = client->generateReplacementFile(e.m_filename);
166             m_hasGeneratedFiles = true;
167         }
168     }
169 }
170
171 void FormData::removeGeneratedFilesIfNeeded()
172 {
173     if (!m_hasGeneratedFiles)
174         return;
175         
176     size_t n = m_elements.size();
177     for (size_t i = 0; i < n; ++i) {
178         FormDataElement& e = m_elements[i];
179         if (e.m_type == FormDataElement::encodedFile && !e.m_generatedFilename.isEmpty()) {
180             ASSERT(e.m_shouldGenerateFile);
181             String directory = directoryName(e.m_generatedFilename);
182             deleteFile(e.m_generatedFilename);
183             deleteEmptyDirectory(directory);
184             e.m_generatedFilename = String();
185         }
186     }
187     m_hasGeneratedFiles = false;
188 }
189
190 } // namespace WebCore