espressif/openthread
1
0
Fork 0
mirror of https://github.com/espressif/openthread.git synced 2026-07-21 03:24:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Message::SetLength and Message::ResizeMessage Bug Fix (#352)

Browse Source 
* Update Message::SetLength and Message::ResizeMessage to handle running out of buffers.
This commit is contained in:
Nick Banks
2016-08-17 09:26:12 -07:00
committed by Jonathan Hui
parent d1f8d59e3f
commit 926e44ff5c
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/core/common/message.cpp
+6 -2
View File
@@ -124,6 +124,8 @@ ThreadError Message::Free(Message &aMessage)
ThreadError Message::ResizeMessage(uint16_t aLength)
{
ThreadError error = kThreadError_None;
// add buffers
Buffer *curBuffer = this;
Buffer *lastBuffer;
@@ -134,6 +136,7 @@ ThreadError Message::ResizeMessage(uint16_t aLength)
if (curBuffer->GetNextBuffer() == NULL)
{
curBuffer->SetNextBuffer(NewBuffer());
VerifyOrExit(curBuffer->GetNextBuffer() != NULL, error = kThreadError_NoBufs);
}
curBuffer = curBuffer->GetNextBuffer();
@@ -147,7 +150,8 @@ ThreadError Message::ResizeMessage(uint16_t aLength)
FreeBuffers(curBuffer);
return kThreadError_None;
exit:
return error;
}
Message *Message::GetNext(void) const
@@ -179,7 +183,7 @@ ThreadError Message::SetLength(uint16_t aLength)
SuccessOrExit(error = ReclaimBuffers(bufs));
ResizeMessage(totalLengthRequest);
SuccessOrExit(error = ResizeMessage(totalLengthRequest));
mInfo.mLength = aLength;
exit: