mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-22 17:40:34 +00:00
49 lines
1.1 KiB
C++
49 lines
1.1 KiB
C++
/*
|
|
* Copyright (C) 2014 Cloudius Systems, Ltd.
|
|
*
|
|
*/
|
|
|
|
#include "ip.hh"
|
|
#include "print.hh"
|
|
|
|
namespace net {
|
|
|
|
std::ostream& operator<<(std::ostream& os, ipv4_address a) {
|
|
auto ip = a.ip;
|
|
return fprint(os, "%d.%d.%d.%d",
|
|
(ip >> 24) & 0xff,
|
|
(ip >> 16) & 0xff,
|
|
(ip >> 8) & 0xff,
|
|
(ip >> 0) & 0xff);
|
|
}
|
|
|
|
uint16_t ip_checksum(void* data, size_t len) {
|
|
uint64_t csum = 0;
|
|
auto p64 = reinterpret_cast<packed<uint64_t>*>(data);
|
|
while (len >= 8) {
|
|
auto old = csum;
|
|
csum += ntohq(*p64++);
|
|
csum += (csum < old);
|
|
len -= 8;
|
|
}
|
|
auto p16 = reinterpret_cast<packed<uint16_t>*>(p64);
|
|
while (len >= 2) {
|
|
auto old = csum;
|
|
csum += ntohs(*p16++);
|
|
csum += (csum < old);
|
|
len -= 2;
|
|
}
|
|
auto p8 = reinterpret_cast<uint8_t*>(p16);
|
|
if (len) {
|
|
auto old = csum;
|
|
csum += *p8++;
|
|
csum += (csum < old);
|
|
len -= 1;
|
|
}
|
|
csum = (csum & 0xffff) + ((csum >> 16) & 0xffff) + ((csum >> 32) & 0xffff) + (csum >> 48);
|
|
csum += csum >> 16;
|
|
return htons(~csum);
|
|
}
|
|
|
|
}
|