1/*
2 * Copyright (C) 2018 Apple 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
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23 * THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#pragma once
27
28#include "Encoder.h"
29#include <WebCore/SharedBuffer.h>
30#include <wtf/Variant.h>
31
32namespace IPC {
33
34class SharedBufferDataReference {
35public:
36 SharedBufferDataReference() = default;
37 SharedBufferDataReference(const WebCore::SharedBuffer& buffer)
38 : m_data(buffer) { }
39 SharedBufferDataReference(const uint8_t* data, size_t size)
40 : m_data(std::make_pair(data, size)) { }
41
42 void encode(Encoder& encoder) const
43 {
44 switchOn(m_data,
45 [encoder = &encoder] (const Ref<const WebCore::SharedBuffer>& buffer) mutable {
46 uint64_t bufferSize = buffer->size();
47 encoder->reserve(bufferSize + sizeof(uint64_t));
48 *encoder << bufferSize;
49 for (const auto& element : buffer.get())
50 encoder->encodeFixedLengthData(reinterpret_cast<const uint8_t*>(element.segment->data()), element.segment->size(), 1);
51 }, [encoder = &encoder] (const std::pair<const uint8_t*, size_t>& pair) mutable {
52 uint64_t bufferSize = pair.second;
53 encoder->reserve(bufferSize + sizeof(uint64_t));
54 *encoder << bufferSize;
55 encoder->encodeFixedLengthData(pair.first, pair.second, 1);
56 }
57 );
58 }
59
60private:
61 Variant<std::pair<const uint8_t*, size_t>, Ref<const WebCore::SharedBuffer>> m_data;
62};
63
64}
65