UDP struct uses IpAddr and MacAddr

This commit is contained in:
Erik Frojdh 2021-09-07 15:16:03 +02:00
parent 9d744deb4a
commit ba52bfb65e
6 changed files with 55 additions and 86 deletions

View File

@ -1388,34 +1388,34 @@ UdpDestination CmdProxy::getUdpEntry() {
std::string key = it.substr(0, pos); std::string key = it.substr(0, pos);
std::string value = it.substr(pos + 1); std::string value = it.substr(pos + 1);
if (key == "entry") { if (key == "entry") {
udpDestination.setEntry(StringTo<int>(value)); udpDestination.entry = StringTo<int>(value);
hasEntry = true; hasEntry = true;
} else if (key == "ip") { } else if (key == "ip") {
if (value == "auto") { if (value == "auto") {
auto val = getIpFromAuto(); auto val = getIpFromAuto();
LOG(logINFO) << "Setting udp_dstip of detector " << det_id LOG(logINFO) << "Setting udp_dstip of detector " << det_id
<< " to " << val; << " to " << val;
udpDestination.setIp(val); udpDestination.ip = val;
} else { } else {
udpDestination.setIp(IpAddr(value)); udpDestination.ip = IpAddr(value);
} }
} else if (key == "ip2") { } else if (key == "ip2") {
if (value == "auto") { if (value == "auto") {
auto val = getIpFromAuto(); auto val = getIpFromAuto();
LOG(logINFO) << "Setting udp_dstip2 of detector " << det_id LOG(logINFO) << "Setting udp_dstip2 of detector " << det_id
<< " to " << val; << " to " << val;
udpDestination.setIp2(val); udpDestination.ip2 = val;
} else { } else {
udpDestination.setIp2(IpAddr(value)); udpDestination.ip2 = IpAddr(value);
} }
} else if (key == "mac") { } else if (key == "mac") {
udpDestination.setMac(MacAddr(value)); udpDestination.mac = MacAddr(value);
} else if (key == "mac2") { } else if (key == "mac2") {
udpDestination.setMac2(MacAddr(value)); udpDestination.mac2 = MacAddr(value);
} else if (key == "port") { } else if (key == "port") {
udpDestination.setPort(StringTo<uint32_t>(value)); udpDestination.port = StringTo<uint32_t>(value);
} else if (key == "port2") { } else if (key == "port2") {
udpDestination.setPort2(StringTo<uint32_t>(value)); udpDestination.port2 = StringTo<uint32_t>(value);
} }
} }
if (!hasEntry) { if (!hasEntry) {

View File

@ -945,24 +945,24 @@ sls::UdpDestination Module::getDestinationUDPList(const uint32_t entry) const {
void Module::setDestinationUDPList(const sls::UdpDestination dest) { void Module::setDestinationUDPList(const sls::UdpDestination dest) {
// set them in the default way so the receivers are also set up // set them in the default way so the receivers are also set up
if (dest.getEntry() == 0) { if (dest.entry == 0) {
if (dest.getPort() != 0) { if (dest.port != 0) {
setDestinationUDPPort(dest.getPort()); setDestinationUDPPort(dest.port);
} }
if (dest.getIp() != 0) { if (dest.ip != 0) {
setDestinationUDPIP(IpAddr(dest.getIp())); setDestinationUDPIP(dest.ip);
} }
if (dest.getMac() != 0) { if (dest.mac != 0) {
setDestinationUDPMAC(MacAddr(dest.getMac())); setDestinationUDPMAC(dest.mac);
} }
if (dest.getPort2() != 0) { if (dest.port2 != 0) {
setDestinationUDPPort2(dest.getPort2()); setDestinationUDPPort2(dest.port2);
} }
if (dest.getIp2() != 0) { if (dest.ip2 != 0) {
setDestinationUDPIP2(IpAddr(dest.getIp2())); setDestinationUDPIP2(dest.ip2);
} }
if (dest.getMac2() != 0) { if (dest.mac2 != 0) {
setDestinationUDPMAC2(MacAddr(dest.getMac2())); setDestinationUDPMAC2(dest.mac2);
} }
} else { } else {
sendToDetector(F_SET_DEST_UDP_LIST, dest, nullptr); sendToDetector(F_SET_DEST_UDP_LIST, dest, nullptr);

View File

@ -30,7 +30,7 @@ class IpAddr {
return addr_ != other; return addr_ != other;
} }
constexpr uint32_t uint32() const noexcept { return addr_; } constexpr uint32_t uint32() const noexcept { return addr_; }
}; } __attribute__((packed));
class MacAddr { class MacAddr {
private: private:
@ -57,62 +57,23 @@ class MacAddr {
return addr_ != other; return addr_ != other;
} }
constexpr uint64_t uint64() const noexcept { return addr_; } constexpr uint64_t uint64() const noexcept { return addr_; }
}; } __attribute__((packed));
class UdpDestination {
private:
uint32_t entry_{};
uint32_t port_{};
uint32_t port2_{};
uint32_t ip_{};
uint32_t ip2_{};
uint64_t mac_{};
uint64_t mac2_{};
public:
constexpr UdpDestination() noexcept = default;
explicit constexpr UdpDestination(uint32_t entry, uint32_t port = 0,
IpAddr ip = {}, MacAddr mac = {},
uint32_t port2 = 0, IpAddr ip2 = {},
MacAddr mac2 = {})
: entry_(entry), port_(port), port2_(port2), ip_(ip.uint32()),
ip2_(ip2.uint32()), mac_(mac.uint64()), mac2_(mac2.uint64()) {}
uint32_t getEntry() const noexcept { return entry_; }
void setEntry(const uint32_t value) { entry_ = value; }
uint32_t getPort() const noexcept { return port_; }
void setPort(const uint32_t value) { port_ = value; }
uint32_t getPort2() const noexcept { return port2_; }
void setPort2(const uint32_t value) { port2_ = value; }
IpAddr getIp() const noexcept { return IpAddr(ip_); }
void setIp(const IpAddr value) { ip_ = value.uint32(); }
IpAddr getIp2() const noexcept { return IpAddr(ip2_); }
void setIp2(const IpAddr value) { ip2_ = value.uint32(); }
MacAddr getMac() const noexcept { return MacAddr(mac_); }
void setMac(const MacAddr value) { mac_ = value.uint64(); }
MacAddr getMac2() const noexcept { return MacAddr(mac2_); }
void setMac2(const MacAddr value) { mac2_ = value.uint64(); }
struct UdpDestination {
uint32_t entry{};
uint32_t port{};
uint32_t port2{};
IpAddr ip;
IpAddr ip2;
MacAddr mac;
MacAddr mac2;
std::string str() const; std::string str() const;
constexpr bool operator==(const UdpDestination &other) const { constexpr bool operator==(const UdpDestination &other) const {
return ((entry_ == other.entry_) && (port_ == other.port_) && return ((entry == other.entry) && (port == other.port) &&
(port2_ == other.port2_) && (ip_ == other.ip_) && (port2 == other.port2) && (ip== other.ip) &&
(ip2_ == other.ip2_) && (mac_ == other.mac_) && (ip2 == other.ip2) && (mac == other.mac) &&
(mac2_ == other.mac2_)); (mac2 == other.mac2));
} }
} __attribute__((packed)); } __attribute__((packed));

View File

@ -72,18 +72,18 @@ std::string MacAddr::hex() const { return to_hex(); }
std::string UdpDestination::str() const { std::string UdpDestination::str() const {
std::ostringstream oss; std::ostringstream oss;
oss << '[' << std::endl oss << '[' << std::endl
<< "entry " << entry_ << std::endl << "entry " << entry << std::endl
<< "ip " << ip_ << std::endl << "ip " << ip << std::endl
<< "mac " << mac_ << std::endl << "mac " << mac << std::endl
<< "port " << port_ << std::endl; << "port " << port << std::endl;
if (port2_ != 0) { if (port2 != 0) {
oss << "port2 " << port2_ << std::endl; oss << "port2 " << port2 << std::endl;
} }
if (ip2_ != 0) { if (ip2 != 0) {
oss << "ip2 " << ip2_ << std::endl; oss << "ip2 " << ip2 << std::endl;
} }
if (mac2_ != 0) { if (mac2 != 0) {
oss << "mac2 " << mac2_ << std::endl; oss << "mac2 " << mac2 << std::endl;
} }
oss << ']'; oss << ']';
return oss.str(); return oss.str();

View File

@ -9,7 +9,7 @@ TEST_CASE("Get size of empty file") {
char fname[] = "temfile_XXXXXX"; char fname[] = "temfile_XXXXXX";
std::ifstream ifs(fname); std::ifstream ifs(fname);
auto size = sls::getFileSize(ifs); auto size = sls::getFileSize(ifs);
REQUIRE(size == 0); REQUIRE(size <= 0); // -1 or zero
} }
TEST_CASE("Get size of file with data") { TEST_CASE("Get size of file with data") {

View File

@ -112,4 +112,12 @@ TEST_CASE("Copy construct a MacAddr") {
CHECK(addr == addr2); CHECK(addr == addr2);
} }
TEST_CASE("udp dst struct"){
static_assert(sizeof(UdpDestination) == 36);
UdpDestination dst{};
REQUIRE(dst.ip == 0);
}
// TODO!(Erik) Look up a real hostname and verify the IP // TODO!(Erik) Look up a real hostname and verify the IP