29 lines
1.0 KiB
C++
29 lines
1.0 KiB
C++
#include "Protocol.h"
|
|
#include <stddef.h>
|
|
|
|
const char *messageName(MessageType type) {
|
|
static const char *names[] = {"DISCOVER", "DISCOVER_ACK", "PREPARE", "READY",
|
|
"START_STAGE", "RESULT", "ACK", "ABORT", "HEARTBEAT", "HEARTBEAT_ACK"};
|
|
const uint8_t index = static_cast<uint8_t>(type);
|
|
return index < sizeof(names) / sizeof(names[0]) ? names[index] : "UNKNOWN";
|
|
}
|
|
|
|
uint16_t packetCrc(const ProtocolPacket &p) {
|
|
const uint8_t *data = reinterpret_cast<const uint8_t *>(&p);
|
|
uint16_t crc = 0xFFFF;
|
|
for (size_t i = 0; i < offsetof(ProtocolPacket, crc); ++i) {
|
|
crc ^= static_cast<uint16_t>(data[i]) << 8;
|
|
for (uint8_t b = 0; b < 8; ++b) crc = (crc & 0x8000) ? (crc << 1) ^ 0x1021 : crc << 1;
|
|
}
|
|
return crc;
|
|
}
|
|
|
|
void finalizePacket(ProtocolPacket &p) {
|
|
p.magic = PROTOCOL_MAGIC; p.version = PROTOCOL_VERSION; p.crc = packetCrc(p);
|
|
}
|
|
|
|
bool validPacket(const ProtocolPacket &p) {
|
|
return p.magic == PROTOCOL_MAGIC && p.version == PROTOCOL_VERSION &&
|
|
p.type <= static_cast<uint8_t>(MessageType::HEARTBEAT_ACK) && p.crc == packetCrc(p);
|
|
}
|