76938c764bb75e3a27bb49e76be25be33181be8c
[WebKit-https.git] / Source / WebKit2 / Platform / CoreIPC / ArgumentEncoder.h
1 /*
2  * Copyright (C) 2010 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 #ifndef ArgumentEncoder_h
27 #define ArgumentEncoder_h
28
29 #include "ArgumentCoder.h"
30 #include "Attachment.h"
31 #include <wtf/PassOwnPtr.h>
32 #include <wtf/TypeTraits.h>
33 #include <wtf/Vector.h>
34
35 namespace CoreIPC {
36
37 class ArgumentEncoder;
38 class DataReference;
39
40 class ArgumentEncoder {
41 public:
42     static PassOwnPtr<ArgumentEncoder> create();
43     virtual ~ArgumentEncoder();
44
45     void encodeFixedLengthData(const uint8_t*, size_t, unsigned alignment);
46     void encodeVariableLengthByteArray(const DataReference&);
47
48     void encode(bool);
49     void encode(uint8_t);
50     void encode(uint16_t);
51     void encode(uint32_t);
52     void encode(uint64_t);
53     void encode(int32_t);
54     void encode(int64_t);
55     void encode(float);
56     void encode(double);
57
58     template<typename T> void encodeEnum(T t)
59     {
60         COMPILE_ASSERT(sizeof(T) <= sizeof(uint64_t), enum_type_must_not_be_larger_than_64_bits);
61
62         encode(static_cast<uint64_t>(t));
63     }
64
65     template<typename T> void encode(const T& t)
66     {
67         ArgumentCoder<T>::encode(*this, t);
68     }
69
70     template<typename T> ArgumentEncoder& operator<<(const T& t)
71     {
72         encode(t);
73         return *this;
74     }
75
76     uint8_t* buffer() { return usesInlineBuffer() ? m_inlineBuffer : m_buffer; }
77     size_t bufferSize() const { return m_bufferSize; }
78
79     void addAttachment(const Attachment&);
80     Vector<Attachment> releaseAttachments();
81
82 protected:
83     ArgumentEncoder();
84
85 private:
86     static const size_t inlineBufferSize = 4096;
87     bool usesInlineBuffer() const { return m_bufferCapacity <= inlineBufferSize; }
88     uint8_t* grow(unsigned alignment, size_t size);
89     
90     uint8_t* m_buffer;
91     
92     size_t m_bufferSize;
93     size_t m_bufferCapacity;
94
95     Vector<Attachment> m_attachments;
96
97     uint8_t m_inlineBuffer[inlineBufferSize];
98 };
99
100 } // namespace CoreIPC
101
102 #endif // ArgumentEncoder_h