2 Copyright (c) 2006-2011, John Hurst
5 Redistribution and use in source and binary forms, with or without
6 modification, are permitted provided that the following conditions
8 1. Redistributions of source code must retain the above copyright
9 notice, this list of conditions and the following disclaimer.
10 2. Redistributions in binary form must reproduce the above copyright
11 notice, this list of conditions and the following disclaimer in the
12 documentation and/or other materials provided with the distribution.
13 3. The name of the author may not be used to endorse or promote products
14 derived from this software without specific prior written permission.
16 THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17 IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18 OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19 IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20 INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21 NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25 THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 \brief abstraction for byte-oriented conversion of integers and objects
35 #include <KM_platform.h>
46 KM_NO_COPY_CONSTRUCT(MemIOWriter);
55 MemIOWriter(byte_t* p, ui32_t c) : m_p(p), m_capacity(c), m_size(0) {
56 assert(m_p); assert(m_capacity);
59 MemIOWriter(ByteString* Buf);
62 inline void Reset() { m_size = 0; }
63 inline byte_t* Data() { return m_p; }
64 inline const byte_t* RoData() const { return m_p; }
65 inline byte_t* CurrentData() { return m_p + m_size; }
66 inline ui32_t Length() const { return m_size; }
67 inline ui32_t Remainder() const { return m_capacity - m_size; }
69 inline bool AddOffset(ui32_t offset) {
70 if ( ( m_size + offset ) > m_capacity )
78 inline bool WriteRaw(const byte_t* p, ui32_t buf_len) {
79 if ( ( m_size + buf_len ) > m_capacity )
82 memcpy(m_p + m_size, p, buf_len);
87 bool WriteBER(ui64_t i, ui32_t ber_len);
89 inline bool WriteUi8(ui8_t i) {
90 if ( ( m_size + 1 ) > m_capacity )
98 inline bool WriteUi16BE(ui16_t i) {
99 if ( ( m_size + sizeof(ui16_t) ) > m_capacity )
102 i2p<ui16_t>(KM_i16_BE(i), m_p + m_size);
103 m_size += sizeof(ui16_t);
107 inline bool WriteUi32BE(ui32_t i) {
108 if ( ( m_size + sizeof(ui32_t) ) > m_capacity )
111 i2p<ui32_t>(KM_i32_BE(i), m_p + m_size);
112 m_size += sizeof(ui32_t);
116 inline bool WriteUi64BE(ui64_t i) {
117 if ( ( m_size + sizeof(ui64_t) ) > m_capacity )
120 i2p<ui64_t>(KM_i64_BE(i), m_p + m_size);
121 m_size += sizeof(ui64_t);
125 inline bool WriteString(const std::string& str) {
126 ui32_t len = static_cast<ui32_t>(str.length());
127 if ( ! WriteUi32BE(len) ) return false;
128 if ( ! WriteRaw((const byte_t*)str.c_str(), len) ) return false;
136 KM_NO_COPY_CONSTRUCT(MemIOReader);
142 ui32_t m_size; // this is sort of a misnomer, when we are reading it measures offset
145 MemIOReader(const byte_t* p, ui32_t c) :
146 m_p(p), m_capacity(c), m_size(0) {
147 assert(m_p); assert(m_capacity);
150 MemIOReader(const ByteString* Buf);
153 inline void Reset() { m_size = 0; }
154 inline const byte_t* Data() const { return m_p; }
155 inline const byte_t* CurrentData() const { return m_p + m_size; }
156 inline ui32_t Offset() const { return m_size; }
157 inline ui32_t Remainder() const { return m_capacity - m_size; }
159 inline bool SkipOffset(ui32_t offset) {
160 if ( ( m_size + offset ) > m_capacity )
167 inline bool ReadRaw(byte_t* p, ui32_t buf_len) {
168 if ( ( m_size + buf_len ) > m_capacity )
171 memcpy(p, m_p + m_size, buf_len);
176 bool ReadBER(ui64_t* i, ui32_t* ber_len);
178 inline bool ReadUi8(ui8_t* i) {
180 if ( ( m_size + 1 ) > m_capacity )
183 *i = *(m_p + m_size);
188 inline bool ReadUi16BE(ui16_t* i) {
190 if ( ( m_size + sizeof(ui16_t) ) > m_capacity )
193 *i = KM_i16_BE(cp2i<ui16_t>(m_p + m_size));
194 m_size += sizeof(ui16_t);
198 inline bool ReadUi32BE(ui32_t* i) {
200 if ( ( m_size + sizeof(ui32_t) ) > m_capacity )
203 *i = KM_i32_BE(cp2i<ui32_t>(m_p + m_size));
204 m_size += sizeof(ui32_t);
208 inline bool ReadUi64BE(ui64_t* i) {
210 if ( ( m_size + sizeof(ui64_t) ) > m_capacity )
213 *i = KM_i64_BE(cp2i<ui64_t>(m_p + m_size));
214 m_size += sizeof(ui64_t);
218 inline bool ReadString(std::string& str)
221 if ( ! ReadUi32BE(&str_length) ) return false;
222 if ( ( m_size + str_length ) > m_capacity ) return false;
223 str.assign((const char*)CurrentData(), str_length);
224 if ( ! SkipOffset(str_length) ) return false;
231 UnarchiveString(MemIOReader& Reader, std::string& str) {
232 return Reader.ReadString(str);
237 ArchiveString(MemIOWriter& Writer, const std::string& str)
239 return Writer.WriteString(str);
245 #endif // _KM_MEMIO_H_