Major Update: (compiling but untested)
- Replace CBuffer with CRollingBuffer and CShiftBuffer ShiftBuffer is faster and simpler and always zero terminates - Updated SelectableCore to use RollingBuffer - Add Overwrite parameter to Push() and ReadFromFD() methods - Add PeakDirect() method for fast direct access to buffer
This commit is contained in:
595
BufferCore.cpp
595
BufferCore.cpp
@@ -17,14 +17,20 @@
|
||||
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
CBuffer::CBuffer( int BufferSize ) : BufSize( BufferSize )
|
||||
/***** Rolling Buffer *****/
|
||||
|
||||
CRollingBuffer::CRollingBuffer( int BufferSize ) : BufSize( BufferSize )
|
||||
{
|
||||
// Create Buffer
|
||||
if (BufSize) {
|
||||
if (BufSize)
|
||||
{
|
||||
// Create Buffer
|
||||
Buffer = (char *)malloc( BufSize+1 );
|
||||
OutBuffer = (char *)malloc( BufSize+1 );
|
||||
OutBuffer[0] = 0;
|
||||
}
|
||||
else {
|
||||
// No buffers created
|
||||
Buffer = NULL;
|
||||
OutBuffer = NULL;
|
||||
}
|
||||
@@ -36,7 +42,7 @@ CBuffer::CBuffer( int BufferSize ) : BufSize( BufferSize )
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
CBuffer::~CBuffer()
|
||||
CRollingBuffer::~CRollingBuffer()
|
||||
{
|
||||
// Destroy Buffer
|
||||
if (Buffer) {
|
||||
@@ -48,51 +54,92 @@ CBuffer::~CBuffer()
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Over write existing buffer with new data
|
||||
int CBuffer::Set( const char * Data, int Len )
|
||||
// Clear all data from buffer
|
||||
int CRollingBuffer::Reset()
|
||||
{
|
||||
// Validate data
|
||||
if (!Data || !Len) {
|
||||
return 0;
|
||||
}
|
||||
else if (Len == -1) {
|
||||
Len = strlen( Data );
|
||||
}
|
||||
int BytesCleared = 0;
|
||||
|
||||
if (Len <= BufSize) {
|
||||
memcpy( Buffer, Data, Len );
|
||||
BufStart = 0;
|
||||
BufEnd = Len;
|
||||
BufLen = Len;
|
||||
}
|
||||
else {
|
||||
memcpy( Buffer, &Data[Len-BufSize], BufSize );
|
||||
BufStart = 0;
|
||||
BufEnd = 0;
|
||||
BufLen = BufSize;
|
||||
}
|
||||
return BufLen;
|
||||
// Clear buffer
|
||||
BytesCleared = BufLen;
|
||||
|
||||
// Reset pointers
|
||||
BufStart = 0;
|
||||
BufEnd = 0;
|
||||
BufLen = 0;
|
||||
|
||||
return BytesCleared;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Return "stitched" buffer data in temporary buffer
|
||||
int CBuffer::Peek( char ** Data, int PeekPos, int MaxLen )
|
||||
// Return specific character for current rolling data
|
||||
char CRollingBuffer::PeekChar( int Pos )
|
||||
{
|
||||
char CharValue = 0;
|
||||
|
||||
// Validate buffer and Pos
|
||||
if (!BufLen || (Pos < 0) || (Pos > BufLen)) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Get char
|
||||
if (BufStart + Pos < BufSize) {
|
||||
CharValue = Buffer[ BufStart+Pos ];
|
||||
}
|
||||
else {
|
||||
CharValue = Buffer[ BufStart+Pos-BufSize ];
|
||||
}
|
||||
|
||||
// Return char
|
||||
return CharValue;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Return pointer direct to buffer
|
||||
// Data is not removed from buffer
|
||||
// Return number of contiguous chars (cannot rollover)
|
||||
int CRollingBuffer::PeekDirect( char ** Data, int PeekPos )
|
||||
{
|
||||
int BytesReturned = 0;
|
||||
int StartPos = 0;
|
||||
|
||||
// Check if any data
|
||||
if (!BufLen || !MaxLen) {
|
||||
// Validate
|
||||
if (!BufLen || !Data || (PeekPos >= BufLen)) {
|
||||
*Data = NULL;
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Validate start position
|
||||
if (PeekPos >= BufLen) {
|
||||
// Get reference
|
||||
if (BufStart + StartPos < BufSize) {
|
||||
StartPos = BufStart + PeekPos;
|
||||
BytesReturned = (StartPos + BufLen > BufSize)? BufSize - StartPos : BufLen;
|
||||
}
|
||||
else {
|
||||
StartPos = BufStart + PeekPos - BufSize;
|
||||
BytesReturned = (StartPos + BufLen > BufEnd) ? BufEnd - StartPos : BufLen;
|
||||
}
|
||||
|
||||
// Return buffer pointer
|
||||
*Data = &Buffer[StartPos];
|
||||
return BytesReturned;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Return "stitched" buffer data copied to pointer
|
||||
// Pointer is created if NULL passed
|
||||
// Data is not removed from buffer
|
||||
int CRollingBuffer::Peek( char ** Data, int PeekPos, int MaxLen )
|
||||
{
|
||||
int BytesReturned = 0;
|
||||
int StartPos = 0;
|
||||
|
||||
// Validate
|
||||
if (!BufLen || !*Data || !MaxLen || (PeekPos < 0) || (PeekPos > BufLen)) {
|
||||
*Data = NULL;
|
||||
return 0;
|
||||
}
|
||||
else if (BufStart + StartPos < BufSize) {
|
||||
|
||||
// Get buffer start position
|
||||
if (BufStart + StartPos < BufSize) {
|
||||
StartPos = BufStart + PeekPos;
|
||||
}
|
||||
else {
|
||||
@@ -113,6 +160,8 @@ int CBuffer::Peek( char ** Data, int PeekPos, int MaxLen )
|
||||
memcpy( OutBuffer, &Buffer[StartPos], BufSize-StartPos ); // Copy from end of buffer
|
||||
memcpy( &OutBuffer[BufSize-StartPos], Buffer, (BytesReturned-(BufSize-StartPos)) ); // Copy rollover from start of buffer
|
||||
}
|
||||
|
||||
// Zero terminate
|
||||
OutBuffer[BytesReturned] = 0;
|
||||
|
||||
// Return Temp buffer
|
||||
@@ -122,25 +171,20 @@ int CBuffer::Peek( char ** Data, int PeekPos, int MaxLen )
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Return "stitched" buffer data in temporary buffer
|
||||
int CBuffer::PeekCopy( char ** Data, int PeekPos, int MaxLen )
|
||||
// Data is not removed from buffer
|
||||
int CRollingBuffer::PeekCopy( char ** Data, int PeekPos, int MaxLen )
|
||||
{
|
||||
int BytesReturned = 0;
|
||||
int StartPos = 0;
|
||||
|
||||
// Check if any data
|
||||
if (!BufLen || !MaxLen) {
|
||||
if (*Data)
|
||||
(*Data)[0] = 0;
|
||||
if (!BufLen || !*Data || !MaxLen || (PeekPos < 0) || (PeekPos > BufLen)) {
|
||||
if (*Data) (*Data)[0] = 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Validate start position
|
||||
if (PeekPos >= BufLen) {
|
||||
if (*Data)
|
||||
(*Data)[0] = 0;
|
||||
return 0;
|
||||
}
|
||||
else if (BufStart + StartPos < BufSize) {
|
||||
// Get start position
|
||||
if (BufStart + StartPos < BufSize) {
|
||||
StartPos = BufStart + PeekPos;
|
||||
}
|
||||
else {
|
||||
@@ -171,7 +215,72 @@ int CBuffer::PeekCopy( char ** Data, int PeekPos, int MaxLen )
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
int CBuffer::Clear( int ClearLen )
|
||||
// Look for first occurrence of character in buffer
|
||||
bool CRollingBuffer::FindStr( const char * SearchStr, int SearchLen, int &FoundPos, int StartPos )
|
||||
{
|
||||
char * CheckPos = NULL;
|
||||
char * CheckLimit = NULL;
|
||||
char * MatchPos = NULL;
|
||||
char * MatchLimit = NULL;
|
||||
|
||||
// Check if buffer exists
|
||||
if (!BufLen || !SearchStr || !SearchLen) {
|
||||
FoundPos = -1;
|
||||
return false;
|
||||
}
|
||||
|
||||
// Validate start Pos
|
||||
if (StartPos >= BufLen) {
|
||||
FoundPos = -1;
|
||||
return false;
|
||||
}
|
||||
|
||||
// Get Search start point and limit
|
||||
CheckPos = (BufStart + StartPos < BufSize)? &Buffer[ BufStart+StartPos ] : &Buffer[ (BufStart+StartPos)-BufSize ];
|
||||
CheckLimit = &Buffer[BufSize-1];
|
||||
|
||||
// Set Match start point and limit
|
||||
MatchPos = (char*)SearchStr;
|
||||
MatchLimit = (char*)&SearchStr[SearchLen-1];
|
||||
|
||||
// Search for char
|
||||
FoundPos = StartPos;
|
||||
while (FoundPos < BufLen)
|
||||
{
|
||||
// Check if char match
|
||||
if (*CheckPos == *MatchPos) {
|
||||
if (MatchPos == MatchLimit) {
|
||||
// Full match found
|
||||
FoundPos -= SearchLen;
|
||||
return true;
|
||||
}
|
||||
else {
|
||||
// Next char to match
|
||||
MatchPos++;
|
||||
}
|
||||
}
|
||||
else {
|
||||
// Reset match point
|
||||
MatchPos = (char*)SearchStr;
|
||||
}
|
||||
|
||||
// Next char
|
||||
FoundPos++;
|
||||
if (CheckPos == CheckLimit) {
|
||||
CheckPos = Buffer; // Roll-over, Start at beginning of buffer
|
||||
} else {
|
||||
CheckPos++;
|
||||
}
|
||||
}
|
||||
|
||||
// Not Found
|
||||
FoundPos = -1;
|
||||
return false;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Remove data from start of buffer
|
||||
int CRollingBuffer::Clear( int ClearLen )
|
||||
{
|
||||
int BytesCleared;
|
||||
|
||||
@@ -193,40 +302,29 @@ int CBuffer::Clear( int ClearLen )
|
||||
|
||||
return BytesCleared;
|
||||
}
|
||||
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
int CBuffer::Reset()
|
||||
{
|
||||
int BytesCleared = 0;
|
||||
|
||||
// Clear all data
|
||||
BytesCleared = BufLen;
|
||||
BufStart = 0;
|
||||
BufEnd = 0;
|
||||
BufLen = 0;
|
||||
|
||||
return BytesCleared;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Add bytes to end of buffer
|
||||
// Return "stitched" buffer data in temporary buffer
|
||||
int CBuffer::Push( const char * Data, int Len )
|
||||
// Perform roll over if requied
|
||||
// Option to overwrite existing data if longer than buffer length
|
||||
int CRollingBuffer::Push( bool Overwrite, const char * Data, int Len )
|
||||
{
|
||||
int BytesPushed = 0;
|
||||
int TotalPushed = 0;
|
||||
int BytesPushed = 0;
|
||||
int TotalPushed = 0;
|
||||
int BufRemain = 0;
|
||||
int DataRemain = Len;
|
||||
int DataRemain;
|
||||
|
||||
// Validate Buffer and Len
|
||||
if (!BufSize || !Len) {
|
||||
if (!BufSize || !Data || !Len) {
|
||||
return 0;
|
||||
}
|
||||
else if (Len == -1) {
|
||||
Len = strlen( Data );
|
||||
}
|
||||
|
||||
// Prevent overwrite
|
||||
DataRemain = (!Overwrite && (Len > BufSize-BufLen))? BufSize-BufLen : Len;
|
||||
|
||||
// Read Data into buffer
|
||||
while (TotalPushed < Len)
|
||||
{
|
||||
@@ -247,7 +345,7 @@ int CBuffer::Push( const char * Data, int Len )
|
||||
BufEnd = 0;
|
||||
}
|
||||
if (BufLen > BufSize) {
|
||||
// Head is caught up with tail, move tail up
|
||||
// Head has caught up with tail, move tail up
|
||||
BufLen = BufSize;
|
||||
BufStart = BufEnd;
|
||||
}
|
||||
@@ -259,14 +357,12 @@ int CBuffer::Push( const char * Data, int Len )
|
||||
|
||||
// Read first-in bytes from buffer
|
||||
// Return "stitched" buffer data in temporary buffer
|
||||
int CBuffer::Pop( char ** Data, int MaxLen )
|
||||
int CRollingBuffer::Pop( char ** Data, int MaxLen )
|
||||
{
|
||||
int BytesWritten = 0;
|
||||
|
||||
// Read data
|
||||
// Read & clear data
|
||||
BytesWritten = Peek( Data, MaxLen );
|
||||
|
||||
// Clear bytes
|
||||
Clear( BytesWritten );
|
||||
|
||||
// Return temp buffer
|
||||
@@ -275,15 +371,13 @@ int CBuffer::Pop( char ** Data, int MaxLen )
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Read first-in bytes from buffer
|
||||
// Return "stitched" buffer data in temporary buffer
|
||||
int CBuffer::PopCopy( char ** Data, int MaxLen )
|
||||
// Return "stitched" buffer data copied to pointer which if created if NULL
|
||||
int CRollingBuffer::PopCopy( char ** Data, int MaxLen )
|
||||
{
|
||||
int BytesWritten = 0;
|
||||
|
||||
// Read data
|
||||
// Copy & clear data
|
||||
BytesWritten = PeekCopy( Data, MaxLen );
|
||||
|
||||
// Clear bytes
|
||||
Clear( BytesWritten );
|
||||
|
||||
// Return temp buffer
|
||||
@@ -292,7 +386,7 @@ int CBuffer::PopCopy( char ** Data, int MaxLen )
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Read File descriptor directly into rolling buffer
|
||||
int CBuffer::ReadFromFD( int Handle, int MaxRead )
|
||||
int CRollingBuffer::ReadFromFD( int Handle, int MaxRead, bool Overwrite )
|
||||
{
|
||||
int BytesRead = 0;
|
||||
int TotalRead = 0;
|
||||
@@ -305,7 +399,12 @@ int CBuffer::ReadFromFD( int Handle, int MaxRead )
|
||||
}
|
||||
|
||||
// Read file descriptor into buffer
|
||||
DataRemain = (MaxRead == -1)? BufSize : MaxRead;
|
||||
if (Overwrite && (MaxRead > BufLen)) {
|
||||
DataRemain = MaxRead;
|
||||
}
|
||||
else {
|
||||
DataRemain = BufLen;
|
||||
}
|
||||
while (DataRemain)
|
||||
{
|
||||
// Read from file descriptor
|
||||
@@ -335,7 +434,7 @@ int CBuffer::ReadFromFD( int Handle, int MaxRead )
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Write given number of bytes directly from buffer to File descriptor
|
||||
int CBuffer::WriteToFD( int Handle, int MaxLen )
|
||||
int CRollingBuffer::WriteToFD( int Handle, int MaxLen )
|
||||
{
|
||||
int BytesWritten = 0;
|
||||
int TotalWritten = 0;
|
||||
@@ -375,37 +474,146 @@ int CBuffer::WriteToFD( int Handle, int MaxLen )
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Return specific character for current rolling data
|
||||
char CBuffer::PeekChar( int Pos, bool ClearChar )
|
||||
{
|
||||
char CharValue = 0;
|
||||
|
||||
/***** Shifting Buffer *****/
|
||||
|
||||
CShiftBuffer::CShiftBuffer( int BufferSize ) : BufSize( BufferSize )
|
||||
{
|
||||
// Create Buffer
|
||||
if (BufSize)
|
||||
{
|
||||
// Create Buffer
|
||||
Buffer = (char *)malloc( BufSize+1 );
|
||||
Buffer[0] = 0;
|
||||
|
||||
OutBuffer = (char *)malloc( BufSize+1 );
|
||||
}
|
||||
else {
|
||||
// No buffers created
|
||||
Buffer = NULL;
|
||||
OutBuffer = NULL;
|
||||
}
|
||||
|
||||
// Set vars
|
||||
BufLen = 0;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
CShiftBuffer::~CShiftBuffer()
|
||||
{
|
||||
// Destroy Buffer
|
||||
if (Buffer) {
|
||||
free( Buffer );
|
||||
}
|
||||
if (OutBuffer) {
|
||||
free( OutBuffer );
|
||||
}
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Clear all data from buffer
|
||||
int CShiftBuffer::Reset()
|
||||
{
|
||||
int BytesCleared = 0;
|
||||
|
||||
// Reset buffer
|
||||
BytesCleared = BufLen;
|
||||
BufLen = 0;
|
||||
Buffer[0] = 0;
|
||||
|
||||
return BytesCleared;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Return specific character for current rolling data
|
||||
char CShiftBuffer::PeekChar( int Pos )
|
||||
{
|
||||
// Validate buffer and Pos
|
||||
if (!BufLen) {
|
||||
if (!BufLen || (Pos < 0) || (Pos > BufLen)) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Get char
|
||||
if (BufStart + Pos < BufSize) {
|
||||
CharValue = Buffer[ BufStart+Pos ];
|
||||
}
|
||||
else {
|
||||
CharValue = Buffer[ BufStart+Pos-BufSize ];
|
||||
}
|
||||
|
||||
// Clear Char
|
||||
if (ClearChar && (Pos == 0)) {
|
||||
BufStart++;
|
||||
BufLen--;
|
||||
}
|
||||
|
||||
// Return char
|
||||
return CharValue;
|
||||
return Buffer[Pos];
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Return pointer direct to buffer
|
||||
// Return number of contiguous chars (cannot rollover)
|
||||
int CShiftBuffer::PeekDirect( char ** Data, int PeekPos )
|
||||
{
|
||||
// Validate
|
||||
if (!BufLen || !Data || (PeekPos >= BufLen)) {
|
||||
*Data = NULL;
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Return buffer pointer
|
||||
*Data = &Buffer[PeekPos];
|
||||
return (BufLen - PeekPos);
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Copy data into temporary buffer
|
||||
int CShiftBuffer::Peek( char ** Data, int PeekPos, int MaxLen )
|
||||
{
|
||||
int BytesReturned = 0;
|
||||
|
||||
// Validate
|
||||
if (!BufLen || !Data || !MaxLen || (PeekPos < 0) || (PeekPos > BufLen)) {
|
||||
*Data = NULL;
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Calculate bytes to return
|
||||
if ((MaxLen == -1) || (MaxLen > BufSize)) {
|
||||
MaxLen = BufSize;
|
||||
}
|
||||
BytesReturned = (MaxLen > BufLen - PeekPos)? (BufLen - PeekPos) : MaxLen;
|
||||
|
||||
// Copy Data from Buffer
|
||||
memcpy( OutBuffer, Buffer, BytesReturned );
|
||||
OutBuffer[BytesReturned] = 0;
|
||||
|
||||
// Return Temp buffer
|
||||
*Data = OutBuffer;
|
||||
return BytesReturned;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Return "stitched" buffer data in temporary buffer
|
||||
// Data is not removed from buffer
|
||||
int CShiftBuffer::PeekCopy( char ** Data, int PeekPos, int MaxLen )
|
||||
{
|
||||
int BytesReturned = 0;
|
||||
int StartPos = 0;
|
||||
|
||||
// Check if any data
|
||||
if (!BufLen || !Data || !MaxLen || (PeekPos < 0) || (PeekPos > BufLen)) {
|
||||
if (*Data) (*Data)[0] = 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Calculate bytes to return
|
||||
if ((MaxLen == -1) || (MaxLen > BufSize)) {
|
||||
MaxLen = BufSize;
|
||||
}
|
||||
BytesReturned = (MaxLen > BufLen - PeekPos)? (BufLen - PeekPos) : MaxLen;
|
||||
|
||||
// Allocate memory
|
||||
if (!*Data)
|
||||
*Data = (char*)malloc( BytesReturned + 1 );
|
||||
|
||||
// Copy Data from Buffer
|
||||
memcpy( *Data, &Buffer[StartPos], BytesReturned );
|
||||
(*Data)[BytesReturned] = 0;
|
||||
|
||||
return BytesReturned;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Look for first occurrence of character in buffer
|
||||
bool CBuffer::FindStr( const char * SearchStr, int SearchLen, int &FoundPos, int StartPos )
|
||||
bool CShiftBuffer::FindStr( const char * SearchStr, int SearchLen, int &FoundPos, int StartPos )
|
||||
{
|
||||
char * CheckPos = NULL;
|
||||
char * CheckLimit = NULL;
|
||||
@@ -425,8 +633,8 @@ bool CBuffer::FindStr( const char * SearchStr, int SearchLen, int &FoundPos, int
|
||||
}
|
||||
|
||||
// Get Search start point and limit
|
||||
CheckPos = (BufStart + StartPos < BufSize)? &Buffer[ BufStart+StartPos ] : &Buffer[ (BufStart+StartPos)-BufSize ];
|
||||
CheckLimit = &Buffer[BufSize-1];
|
||||
CheckPos = &Buffer[StartPos];
|
||||
CheckLimit = &Buffer[BufLen-1];
|
||||
|
||||
// Set Match start point and limit
|
||||
MatchPos = (char*)SearchStr;
|
||||
@@ -438,12 +646,15 @@ bool CBuffer::FindStr( const char * SearchStr, int SearchLen, int &FoundPos, int
|
||||
{
|
||||
// Check if char match
|
||||
if (*CheckPos == *MatchPos) {
|
||||
if (MatchPos == MatchLimit)
|
||||
if (MatchPos == MatchLimit) {
|
||||
// Full match found
|
||||
FoundPos -= SearchLen;
|
||||
return true;
|
||||
else
|
||||
}
|
||||
else {
|
||||
// Next char to match
|
||||
MatchPos++;
|
||||
}
|
||||
}
|
||||
else {
|
||||
// Reset match point
|
||||
@@ -453,7 +664,7 @@ bool CBuffer::FindStr( const char * SearchStr, int SearchLen, int &FoundPos, int
|
||||
// Next char
|
||||
FoundPos++;
|
||||
if (CheckPos == CheckLimit) {
|
||||
CheckPos = Buffer; // Roll-over, Start at beginning of buffer
|
||||
break;
|
||||
} else {
|
||||
CheckPos++;
|
||||
}
|
||||
@@ -464,3 +675,179 @@ bool CBuffer::FindStr( const char * SearchStr, int SearchLen, int &FoundPos, int
|
||||
return false;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Remove data from start of buffer
|
||||
int CShiftBuffer::Clear( int ClearLen )
|
||||
{
|
||||
int BytesCleared;
|
||||
|
||||
// Validate ClearLen
|
||||
if ((ClearLen == -1) || (ClearLen >= BufLen))
|
||||
{
|
||||
// Clear all data
|
||||
BytesCleared = BufLen;
|
||||
BufLen = 0;
|
||||
Buffer[0] = 0;
|
||||
}
|
||||
else {
|
||||
// Shift remaining data
|
||||
BytesCleared = ClearLen;
|
||||
memcpy( Buffer, &Buffer[ClearLen], BufLen-ClearLen ); // Shift remaining old data left
|
||||
BufLen -= ClearLen;
|
||||
Buffer[BufLen] = 0;
|
||||
}
|
||||
|
||||
return BytesCleared;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Add bytes to end of buffer
|
||||
// Option to overwrite existing data if longer than buffer length
|
||||
int CShiftBuffer::Push( bool Overwrite, const char * Data, int Len )
|
||||
{
|
||||
int BytesPushed = 0;
|
||||
int BufRemain = 0;
|
||||
|
||||
// Validate Buffer and Len
|
||||
if (!BufSize || !Data || !Len) {
|
||||
return 0;
|
||||
}
|
||||
else if (Len == -1) {
|
||||
Len = strlen( Data );
|
||||
}
|
||||
|
||||
// Copy Data to buffer
|
||||
BufRemain = BufSize - BufLen;
|
||||
if (Overwrite && (Len > BufRemain))
|
||||
{
|
||||
if (Len < BufSize) {
|
||||
memcpy( Buffer, &Buffer[Len-BufRemain], BufLen-(Len-BufRemain) ); // Shift remaining old data left
|
||||
memcpy( &Buffer[BufSize-Len], Data, Len ); // Copy new data
|
||||
BytesPushed = Len;
|
||||
}
|
||||
else {
|
||||
memcpy( Buffer, &Data[Len-BufSize], BufSize ); // Copy last portion of new data
|
||||
BytesPushed = Len;
|
||||
}
|
||||
|
||||
// Update Pointers
|
||||
BufLen = BufSize;
|
||||
Buffer[BufLen] = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
// Copy first portion of new data to buffer
|
||||
BytesPushed = (Len > BufRemain)? BufRemain : Len;
|
||||
memcpy( &Buffer[BufLen], Data, BytesPushed );
|
||||
|
||||
// Update Buffer Pointers
|
||||
BufLen += BytesPushed;
|
||||
Buffer[BufLen] = 0;
|
||||
}
|
||||
|
||||
return BytesPushed;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Read first-in bytes from buffer
|
||||
// Return "stitched" buffer data in temporary buffer
|
||||
int CShiftBuffer::Pop( char ** Data, int MaxLen )
|
||||
{
|
||||
int BytesWritten = 0;
|
||||
|
||||
// Read and clear bytes
|
||||
BytesWritten = Peek( Data, MaxLen );
|
||||
Clear( BytesWritten );
|
||||
|
||||
return BytesWritten;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Read first-in bytes from buffer
|
||||
// Return "stitched" buffer data copied to pointer which if created if NULL
|
||||
int CShiftBuffer::PopCopy( char ** Data, int MaxLen )
|
||||
{
|
||||
int BytesWritten = 0;
|
||||
|
||||
// Copy and clear bytes
|
||||
BytesWritten = PeekCopy( Data, MaxLen );
|
||||
Clear( BytesWritten );
|
||||
|
||||
// Return temp buffer
|
||||
return BytesWritten;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Read File descriptor directly into buffer
|
||||
int CShiftBuffer::ReadFromFD( int Handle, int MaxRead )
|
||||
{
|
||||
int BytesRead = 0;
|
||||
int TotalRead = 0;
|
||||
int DataRemain = 0;
|
||||
|
||||
// Check if buffer created
|
||||
if (!BufSize) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Read file descriptor into buffer
|
||||
DataRemain = (MaxRead > BufSize-BufLen)? BufSize-BufLen : MaxRead;
|
||||
while (DataRemain)
|
||||
{
|
||||
// Read from file descriptor
|
||||
BytesRead = read( Handle, &Buffer[BufLen], DataRemain );
|
||||
if (BytesRead <= 0) {
|
||||
break;
|
||||
}
|
||||
|
||||
// Update Buffer Pointers
|
||||
DataRemain -= BytesRead;
|
||||
TotalRead += BytesRead;
|
||||
BufLen += BytesRead;
|
||||
}
|
||||
return TotalRead;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Write given number of bytes directly from buffer to File descriptor
|
||||
int CShiftBuffer::WriteToFD( int Handle, int MaxLen )
|
||||
{
|
||||
int BytesWritten = 0;
|
||||
int TotalWritten = 0;
|
||||
int BufRemain = 0;
|
||||
int DataRemain = 0;
|
||||
int ReadPos = 0;
|
||||
|
||||
// Check if buffer created
|
||||
if (!BufSize) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Read Data into buffer
|
||||
ReadPos = 0;
|
||||
DataRemain = ((MaxLen == -1) || (MaxLen > BufLen))? BufLen : MaxLen;
|
||||
while (DataRemain)
|
||||
{
|
||||
// Read from file descriptor
|
||||
BufRemain = BufSize - ReadPos;
|
||||
BytesWritten = write( Handle, &Buffer[ReadPos], ((BufRemain > DataRemain)? DataRemain : BufRemain) );
|
||||
if (BytesWritten <= 0) {
|
||||
break;
|
||||
}
|
||||
|
||||
// Update Data Pointers
|
||||
TotalWritten += BytesWritten;
|
||||
DataRemain -= BytesWritten;
|
||||
|
||||
// Update Buffer Pointers
|
||||
ReadPos += BytesWritten;
|
||||
if (ReadPos >= BufSize) {
|
||||
ReadPos = 0; // Rolling over end of buffer, start at beginning
|
||||
}
|
||||
}
|
||||
|
||||
return TotalWritten;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user