Aren/haiku
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

* Cleanup.

Browse Source 
git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@37646 a95241bf-73f2-0310-859d-f6bbb57e9c96
This commit is contained in:
Axel Dörfler 2010-07-21 11:23:42 +00:00
parent 866e21d3a6
commit f6a57629c3
1 changed files with 107 additions and 95 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

202
headers/private/net/NetBufferUtilities.h
View File

@ -1,5 +1,5 @@
/*
* Copyright 2006-2007, Haiku, Inc. All Rights Reserved.
* Copyright 2006-2010, Haiku, Inc. All Rights Reserved.
* Distributed under the terms of the MIT License.
*/
#ifndef NET_BUFFER_UTILITIES_H
@ -9,142 +9,154 @@
#include <net_buffer.h>
extern net_buffer_module_info *gBufferModule;
extern net_buffer_module_info* gBufferModule;
class NetBufferModuleGetter {
public:
static net_buffer_module_info *Get() { return gBufferModule; }
};
//! A class to access a field safely across node boundaries
template<typename Type, int Offset, typename Module = NetBufferModuleGetter>
class NetBufferFieldReader {
public:
NetBufferFieldReader(net_buffer *buffer)
:
fBuffer(buffer)
{
fStatus = Module::Get()->direct_access(fBuffer, Offset,
sizeof(Type), (void **)&fData);
if (fStatus != B_OK) {
fStatus = Module::Get()->read(fBuffer, Offset,
&fDataBuffer, sizeof(Type));
fData = &fDataBuffer;
}
public:
NetBufferFieldReader(net_buffer* buffer)
:
fBuffer(buffer)
{
fStatus = Module::Get()->direct_access(fBuffer, Offset, sizeof(Type),
(void**)&fData);
if (fStatus != B_OK) {
fStatus = Module::Get()->read(fBuffer, Offset, &fDataBuffer,
sizeof(Type));
fData = &fDataBuffer;
}
}
status_t
Status() const
{
return fStatus;
}
status_t Status() const
{
return fStatus;
}
Type &
Data() const
{
return *fData;
}
Type& Data() const
{
return *fData;
}
Type *
operator->() const
{
return fData;
}
Type* operator->() const
{
return fData;
}
Type &
operator*() const
{
return *fData;
}
Type& operator*() const
{
return *fData;
}
void
Sync()
{
if (fBuffer == NULL || fStatus < B_OK)
return;
void Sync()
{
if (fBuffer == NULL || fStatus < B_OK)
return;
if (fData == &fDataBuffer)
Module::Get()->write(fBuffer, Offset, fData, sizeof(Type));
if (fData == &fDataBuffer)
Module::Get()->write(fBuffer, Offset, fData, sizeof(Type));
fBuffer = NULL;
}
fBuffer = NULL;
}
protected:
NetBufferFieldReader() {}
protected:
NetBufferFieldReader()
{
}
net_buffer *fBuffer;
status_t fStatus;
Type *fData;
Type fDataBuffer;
net_buffer* fBuffer;
status_t fStatus;
Type* fData;
Type fDataBuffer;
};
//! Writes back any changed data on destruction
template<typename Type, int Offset, typename Module = NetBufferModuleGetter>
class NetBufferField : public NetBufferFieldReader<Type, Offset, Module> {
public:
NetBufferField(net_buffer *buffer)
: NetBufferFieldReader<Type, Offset, Module>(buffer)
{}
public:
NetBufferField(net_buffer* buffer)
:
NetBufferFieldReader<Type, Offset, Module>(buffer)
{
}
~NetBufferField()
{
this->Sync();
}
~NetBufferField()
{
// Note, "this->" is needed here for GCC4
this->Sync();
}
};
//! Can remove the header from the buffer
template<typename Type, typename Module = NetBufferModuleGetter>
class NetBufferHeaderReader : public NetBufferFieldReader<Type, 0, Module> {
public:
NetBufferHeaderReader(net_buffer *buffer)
: NetBufferFieldReader<Type, 0, Module>(buffer)
{}
public:
NetBufferHeaderReader(net_buffer* buffer)
:
NetBufferFieldReader<Type, 0, Module>(buffer)
{
}
void
Remove()
{
this->Remove(sizeof(Type));
}
void Remove()
{
Remove(sizeof(Type));
}
void
Remove(size_t bytes)
{
if (this->fBuffer != NULL) {
Module::Get()->remove_header(this->fBuffer, bytes);
this->fBuffer = NULL;
}
void Remove(size_t bytes)
{
if (this->fBuffer != NULL) {
Module::Get()->remove_header(this->fBuffer, bytes);
this->fBuffer = NULL;
}
}
};
//! Removes the header on destruction
template<typename Type, typename Module = NetBufferModuleGetter>
class NetBufferHeaderRemover : public NetBufferHeaderReader<Type, Module> {
public:
NetBufferHeaderRemover(net_buffer *buffer)
: NetBufferHeaderReader<Type, Module>(buffer)
{}
public:
NetBufferHeaderRemover(net_buffer* buffer)
:
NetBufferHeaderReader<Type, Module>(buffer)
{
}
~NetBufferHeaderRemover()
{
this->Remove();
}
~NetBufferHeaderRemover()
{
this->Remove();
}
};
//! A class to add a header to a buffer
//! A class to add a header to a buffer, syncs itself on destruction
template<typename Type, typename Module = NetBufferModuleGetter>
class NetBufferPrepend : public NetBufferFieldReader<Type, 0, Module> {
public:
NetBufferPrepend(net_buffer *buffer, size_t size = sizeof(Type))
{
this->fBuffer = buffer;
public:
NetBufferPrepend(net_buffer* buffer, size_t size = sizeof(Type))
{
this->fBuffer = buffer;
this->fStatus = Module::Get()->prepend_size(buffer, size,
(void **)&(this->fData));
if (this->fStatus == B_OK && this->fData == NULL)
this->fData = &this->fDataBuffer;
}
this->fStatus = Module::Get()->prepend_size(buffer, size,
(void**)&this->fData);
if (this->fStatus == B_OK && this->fData == NULL)
this->fData = &this->fDataBuffer;
}
~NetBufferPrepend()
{
this->Sync();
}
~NetBufferPrepend()
{
this->Sync();
}
};
#endif // NET_BUFFER_UTILITIES_H