mirror of https://github.com/debauchee/barrier.git
59 lines
1.1 KiB
C++
59 lines
1.1 KiB
C++
#include "COutputPacketStream.h"
|
|
|
|
//
|
|
// COuputPacketStream
|
|
//
|
|
|
|
COutputPacketStream::COutputPacketStream(IOutputStream* stream, bool adopt) :
|
|
COutputStreamFilter(stream, adopt)
|
|
{
|
|
// do nothing
|
|
}
|
|
|
|
COutputPacketStream::~COutputPacketStream()
|
|
{
|
|
// do nothing
|
|
}
|
|
|
|
void
|
|
COutputPacketStream::close()
|
|
{
|
|
getStream()->close();
|
|
}
|
|
|
|
UInt32
|
|
COutputPacketStream::write(const void* buffer, UInt32 count)
|
|
{
|
|
// write the length of the payload
|
|
UInt8 length[4];
|
|
length[0] = (UInt8)((count >> 24) & 0xff);
|
|
length[1] = (UInt8)((count >> 16) & 0xff);
|
|
length[2] = (UInt8)((count >> 8) & 0xff);
|
|
length[3] = (UInt8)( count & 0xff);
|
|
UInt32 count2 = sizeof(length);
|
|
const UInt8* cbuffer = length;
|
|
while (count2 > 0) {
|
|
UInt32 n = getStream()->write(cbuffer, count2);
|
|
cbuffer += n;
|
|
count2 -= n;
|
|
}
|
|
|
|
// write the payload
|
|
count2 = count;
|
|
cbuffer = reinterpret_cast<const UInt8*>(buffer);
|
|
while (count2 > 0) {
|
|
UInt32 n = getStream()->write(cbuffer, count2);
|
|
cbuffer += n;
|
|
count2 -= n;
|
|
}
|
|
|
|
return count;
|
|
}
|
|
|
|
void
|
|
COutputPacketStream::flush()
|
|
{
|
|
getStream()->flush();
|
|
}
|
|
|