* Cleanup, no functional change.
git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@37866 a95241bf-73f2-0310-859d-f6bbb57e9c96
This commit is contained in:
@@ -1,91 +1,90 @@
|
|||||||
/*
|
/*
|
||||||
* Copyright 2006, Haiku, Inc. All Rights Reserved.
|
* Copyright 2006-2010, Haiku, Inc. All Rights Reserved.
|
||||||
* Distributed under the terms of the MIT License.
|
* Distributed under the terms of the MIT License.
|
||||||
*/
|
*/
|
||||||
#ifndef NET_UTILITIES_H
|
#ifndef NET_UTILITIES_H
|
||||||
#define NET_UTILITIES_H
|
#define NET_UTILITIES_H
|
||||||
|
|
||||||
|
|
||||||
|
#include <netinet/in.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
|
||||||
#include <net_buffer.h>
|
#include <net_buffer.h>
|
||||||
#include <net_datalink.h>
|
#include <net_datalink.h>
|
||||||
|
|
||||||
#include <netinet/in.h> // for htons
|
|
||||||
|
|
||||||
#include <stdlib.h>
|
|
||||||
|
|
||||||
class Checksum {
|
class Checksum {
|
||||||
public:
|
public:
|
||||||
struct BufferHelper {
|
struct BufferHelper {
|
||||||
BufferHelper(net_buffer *_buffer, net_buffer_module_info *_bufferModule)
|
BufferHelper(net_buffer* _buffer, net_buffer_module_info* _bufferModule)
|
||||||
: buffer(_buffer),
|
:
|
||||||
bufferModule(_bufferModule)
|
buffer(_buffer),
|
||||||
{
|
bufferModule(_bufferModule)
|
||||||
}
|
{
|
||||||
net_buffer *buffer;
|
}
|
||||||
net_buffer_module_info *bufferModule;
|
|
||||||
};
|
net_buffer* buffer;
|
||||||
|
net_buffer_module_info* bufferModule;
|
||||||
|
};
|
||||||
|
|
||||||
Checksum();
|
Checksum()
|
||||||
|
:
|
||||||
Checksum& operator<<(uint8 val);
|
fSum(0)
|
||||||
Checksum& operator<<(uint16 val);
|
{
|
||||||
Checksum& operator<<(uint32 val);
|
|
||||||
Checksum& operator<<(const BufferHelper &bufferHelper);
|
|
||||||
|
|
||||||
operator uint16();
|
|
||||||
|
|
||||||
static uint16 PseudoHeader(net_address_module_info *addressModule,
|
|
||||||
net_buffer_module_info *bufferModule, net_buffer *buffer,
|
|
||||||
uint16 protocol);
|
|
||||||
|
|
||||||
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;
|
inline Checksum& operator<<(uint8 value)
|
||||||
return result;
|
{
|
||||||
}
|
#if B_HOST_IS_LENDIAN
|
||||||
|
fSum += value;
|
||||||
|
#else
|
||||||
|
fSum += (uint16)value << 8;
|
||||||
|
#endif
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
inline Checksum& operator<<(uint16 value)
|
||||||
|
{
|
||||||
|
fSum += value;
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
inline Checksum& operator<<(uint32 value)
|
||||||
|
{
|
||||||
|
fSum += (value & 0xffff) + (value >> 16);
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
inline Checksum& operator<<(const BufferHelper& bufferHelper)
|
||||||
|
{
|
||||||
|
net_buffer* buffer = bufferHelper.buffer;
|
||||||
|
fSum += bufferHelper.bufferModule->checksum(buffer, 0, buffer->size,
|
||||||
|
false);
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
inline operator uint16()
|
||||||
|
{
|
||||||
|
while (fSum >> 16) {
|
||||||
|
fSum = (fSum & 0xffff) + (fSum >> 16);
|
||||||
|
}
|
||||||
|
uint16 result = (uint16)fSum;
|
||||||
|
result ^= 0xFFFF;
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
static uint16 PseudoHeader(net_address_module_info* addressModule,
|
||||||
|
net_buffer_module_info* bufferModule, net_buffer* buffer,
|
||||||
|
uint16 protocol);
|
||||||
|
|
||||||
|
private:
|
||||||
|
uint32 fSum;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
inline uint16
|
inline uint16
|
||||||
Checksum::PseudoHeader(net_address_module_info *addressModule,
|
Checksum::PseudoHeader(net_address_module_info* addressModule,
|
||||||
net_buffer_module_info *bufferModule, net_buffer *buffer, uint16 protocol)
|
net_buffer_module_info* bufferModule, net_buffer* buffer, uint16 protocol)
|
||||||
{
|
{
|
||||||
Checksum checksum;
|
Checksum checksum;
|
||||||
addressModule->checksum_address(&checksum, buffer->source);
|
addressModule->checksum_address(&checksum, buffer->source);
|
||||||
@@ -96,44 +95,48 @@ Checksum::PseudoHeader(net_address_module_info *addressModule,
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// helper class that prints an address (and optionally a port) into a buffer that
|
/*! Helper class that prints an address (and optionally a port) into a buffer
|
||||||
// is automatically freed at end of scope:
|
that is automatically freed at end of scope.
|
||||||
|
*/
|
||||||
class AddressString {
|
class AddressString {
|
||||||
public:
|
public:
|
||||||
AddressString(net_domain *domain, const sockaddr *address,
|
AddressString(net_domain* domain, const sockaddr* address,
|
||||||
bool printPort = false)
|
bool printPort = false)
|
||||||
: fBuffer(NULL)
|
:
|
||||||
|
fBuffer(NULL)
|
||||||
{
|
{
|
||||||
domain->address_module->print_address(address, &fBuffer, printPort);
|
domain->address_module->print_address(address, &fBuffer, printPort);
|
||||||
}
|
}
|
||||||
|
|
||||||
AddressString(net_domain *domain, const sockaddr &address,
|
AddressString(net_domain* domain, const sockaddr& address,
|
||||||
bool printPort = false)
|
bool printPort = false)
|
||||||
: fBuffer(NULL)
|
:
|
||||||
|
fBuffer(NULL)
|
||||||
{
|
{
|
||||||
domain->address_module->print_address(&address, &fBuffer, printPort);
|
domain->address_module->print_address(&address, &fBuffer, printPort);
|
||||||
}
|
}
|
||||||
|
|
||||||
AddressString(net_address_module_info *address_module,
|
AddressString(net_address_module_info* address_module,
|
||||||
const sockaddr *address, bool printPort = false)
|
const sockaddr* address, bool printPort = false)
|
||||||
: fBuffer(NULL)
|
:
|
||||||
|
fBuffer(NULL)
|
||||||
{
|
{
|
||||||
address_module->print_address(address, &fBuffer, printPort);
|
address_module->print_address(address, &fBuffer, printPort);
|
||||||
}
|
}
|
||||||
|
|
||||||
~AddressString()
|
~AddressString()
|
||||||
{
|
{
|
||||||
if (fBuffer)
|
free(fBuffer);
|
||||||
free(fBuffer);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
const char *Data() const
|
const char* Data() const
|
||||||
{
|
{
|
||||||
return fBuffer;
|
return fBuffer;
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
char *fBuffer;
|
char* fBuffer;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
#endif // NET_UTILITIES_H
|
#endif // NET_UTILITIES_H
|
||||||
|
|||||||
Reference in New Issue
Block a user