1 /*
2  *  Copyright (c) 2021, The OpenThread Authors.
3  *  All rights reserved.
4  *
5  *  Redistribution and use in source and binary forms, with or without
6  *  modification, are permitted provided that the following conditions 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  *  3. Neither the name of the copyright holder nor the
13  *     names of its contributors may be used to endorse or promote products
14  *     derived from this software without specific prior written permission.
15  *
16  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17  *  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  *  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  *  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
20  *  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21  *  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22  *  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23  *  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24  *  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25  *  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26  *  POSSIBILITY OF SUCH DAMAGE.
27  */
28 
29 /**
30  * @file
31  *   This file implements the `Appender` class.
32  */
33 
34 #include "appender.hpp"
35 
36 namespace ot {
37 
Appender(Message & aMessage)38 Appender::Appender(Message &aMessage)
39     : mType(kMessage)
40 {
41     mShared.mMessage.mMessage     = &aMessage;
42     mShared.mMessage.mStartOffset = aMessage.GetLength();
43 }
44 
Appender(uint8_t * aBuffer,uint16_t aSize)45 Appender::Appender(uint8_t *aBuffer, uint16_t aSize)
46     : mType(kBuffer)
47 {
48     mShared.mFrameBuilder.Init(aBuffer, aSize);
49 }
50 
AppendBytes(const void * aBuffer,uint16_t aLength)51 Error Appender::AppendBytes(const void *aBuffer, uint16_t aLength)
52 {
53     Error error = kErrorNone;
54 
55     switch (mType)
56     {
57     case kMessage:
58         error = mShared.mMessage.mMessage->AppendBytes(aBuffer, aLength);
59         break;
60 
61     case kBuffer:
62         error = mShared.mFrameBuilder.AppendBytes(aBuffer, aLength);
63         break;
64     }
65 
66     return error;
67 }
68 
GetAppendedLength(void) const69 uint16_t Appender::GetAppendedLength(void) const
70 {
71     uint16_t length = 0;
72 
73     switch (mType)
74     {
75     case kMessage:
76         length = mShared.mMessage.mMessage->GetLength() - mShared.mMessage.mStartOffset;
77         break;
78 
79     case kBuffer:
80         length = mShared.mFrameBuilder.GetLength();
81         break;
82     }
83 
84     return length;
85 }
86 
GetAsData(Data<kWithUint16Length> & aData) const87 void Appender::GetAsData(Data<kWithUint16Length> &aData) const
88 {
89     aData.Init(mShared.mFrameBuilder.GetBytes(), mShared.mFrameBuilder.GetLength());
90 }
91 
92 } // namespace ot
93