FormDataReference.h [plain text]
#pragma once
#include "Decoder.h"
#include "Encoder.h"
#include "SandboxExtension.h"
#include <WebCore/FormData.h>
namespace IPC {
class FormDataReference {
public:
FormDataReference() = default;
explicit FormDataReference(RefPtr<WebCore::FormData>&& data)
: m_data(WTFMove(data))
{
}
RefPtr<WebCore::FormData> takeData() { return WTFMove(m_data); }
void encode(Encoder& encoder) const
{
encoder << !!m_data;
if (!m_data)
return;
encoder << *m_data;
auto& elements = m_data->elements();
size_t fileCount = std::count_if(elements.begin(), elements.end(), [](auto& element) {
return WTF::holds_alternative<WebCore::FormDataElement::EncodedFileData>(element.data);
});
WebKit::SandboxExtension::HandleArray sandboxExtensionHandles;
sandboxExtensionHandles.allocate(fileCount);
size_t extensionIndex = 0;
for (auto& element : elements) {
if (auto* fileData = WTF::get_if<WebCore::FormDataElement::EncodedFileData>(element.data)) {
const String& path = fileData->filename;
WebKit::SandboxExtension::createHandle(path, WebKit::SandboxExtension::Type::ReadOnly, sandboxExtensionHandles[extensionIndex++]);
}
}
encoder << sandboxExtensionHandles;
}
static Optional<FormDataReference> decode(Decoder& decoder)
{
Optional<bool> hasFormData;
decoder >> hasFormData;
if (!hasFormData)
return WTF::nullopt;
if (!hasFormData.value())
return FormDataReference { };
auto formData = WebCore::FormData::decode(decoder);
if (!formData)
return WTF::nullopt;
Optional<WebKit::SandboxExtension::HandleArray> sandboxExtensionHandles;
decoder >> sandboxExtensionHandles;
if (!sandboxExtensionHandles)
return WTF::nullopt;
WebKit::SandboxExtension::consumePermanently(*sandboxExtensionHandles);
return FormDataReference { formData.releaseNonNull() };
}
private:
RefPtr<WebCore::FormData> m_data;
};
}