c22d69bf1f
* Removed ppp_up and pppcontrol from the image for now. git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@18457 a95241bf-73f2-0310-859d-f6bbb57e9c96
112 lines
2.1 KiB
C++
112 lines
2.1 KiB
C++
/*
|
|
* Copyright 2006, Haiku, Inc. All Rights Reserved.
|
|
* Distributed under the terms of the MIT License.
|
|
*/
|
|
#ifndef NET_UTILITIES_H
|
|
#define NET_UTILITIES_H
|
|
|
|
|
|
#include <net_buffer.h>
|
|
#include <net_datalink.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
class Checksum {
|
|
public:
|
|
struct BufferHelper {
|
|
BufferHelper(net_buffer *_buffer, net_buffer_module_info *_bufferModule)
|
|
: buffer(_buffer),
|
|
bufferModule(_bufferModule)
|
|
{
|
|
}
|
|
net_buffer *buffer;
|
|
net_buffer_module_info *bufferModule;
|
|
};
|
|
|
|
Checksum();
|
|
|
|
Checksum& operator<<(uint8 val);
|
|
Checksum& operator<<(uint16 val);
|
|
Checksum& operator<<(uint32 val);
|
|
Checksum& operator<<(const BufferHelper &bufferHelper);
|
|
|
|
operator uint16();
|
|
|
|
private:
|
|
uint32 fSum;
|
|
};
|
|
|
|
inline Checksum::Checksum()
|
|
: fSum(0)
|
|
{
|
|
}
|
|
|
|
inline Checksum& Checksum::operator<<(uint8 _val) {
|
|
#if B_HOST_IS_LENDIAN
|
|
fSum += _val;
|
|
#else
|
|
uint16 val = _val;
|
|
fSum += val << 8;
|
|
#endif
|
|
return *this;
|
|
}
|
|
|
|
inline Checksum& Checksum::operator<<(uint16 val) {
|
|
fSum += val;
|
|
return *this;
|
|
}
|
|
|
|
inline Checksum& Checksum::operator<<(uint32 val) {
|
|
fSum += (val & 0xFFFF) + (val >> 16);
|
|
return *this;
|
|
}
|
|
|
|
inline Checksum& Checksum::operator<<(const BufferHelper &bufferHelper) {
|
|
net_buffer *buffer = bufferHelper.buffer;
|
|
fSum += bufferHelper.bufferModule->checksum(buffer, 0, buffer->size, false);
|
|
return *this;
|
|
}
|
|
|
|
inline Checksum::operator uint16() {
|
|
while (fSum >> 16) {
|
|
fSum = (fSum & 0xffff) + (fSum >> 16);
|
|
}
|
|
uint16 result = (uint16)fSum;
|
|
result ^= 0xFFFF;
|
|
return result;
|
|
}
|
|
|
|
|
|
// helper class that prints an address (and optionally a port) into a buffer that
|
|
// is automatically freed at end of scope:
|
|
class AddressString {
|
|
public:
|
|
inline AddressString(net_domain *domain, const sockaddr *address,
|
|
bool printPort = false)
|
|
: fBuffer(NULL)
|
|
{
|
|
domain->address_module->print_address(address, &fBuffer, printPort);
|
|
}
|
|
|
|
inline AddressString(net_domain *domain, const sockaddr &address,
|
|
bool printPort = false)
|
|
: fBuffer(NULL)
|
|
{
|
|
domain->address_module->print_address(&address, &fBuffer, printPort);
|
|
}
|
|
|
|
inline ~AddressString()
|
|
{
|
|
free(fBuffer);
|
|
}
|
|
|
|
inline char *Data()
|
|
{
|
|
return fBuffer;
|
|
}
|
|
private:
|
|
char *fBuffer;
|
|
};
|
|
|
|
#endif // NET_UTILITIES_H
|