forked from forks/qmk_firmware
Add crc32 validation of received frames
This commit is contained in:
parent
a3ec3bbef8
commit
62058329ff
|
@ -23,6 +23,7 @@ SOFTWARE.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "protocol/frame_validator.h"
|
#include "protocol/frame_validator.h"
|
||||||
|
#include "protocol/frame_router.h"
|
||||||
|
|
||||||
const uint32_t poly8_lookup[256] =
|
const uint32_t poly8_lookup[256] =
|
||||||
{
|
{
|
||||||
|
@ -101,5 +102,11 @@ static uint32_t crc32_byte(uint8_t *p, uint32_t bytelength)
|
||||||
}
|
}
|
||||||
|
|
||||||
void recv_frame(uint8_t* data, uint16_t size) {
|
void recv_frame(uint8_t* data, uint16_t size) {
|
||||||
|
if (size > 4) {
|
||||||
|
uint32_t frame_crc = *(uint32_t*)(data + size - 4);
|
||||||
|
uint32_t expected_crc = crc32_byte(data, size - 4);
|
||||||
|
if (frame_crc == expected_crc) {
|
||||||
|
route_frame(data, size-4);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -42,3 +42,38 @@ Ensure(FrameValidator, doesnt_validate_frames_under_5_bytes) {
|
||||||
recv_frame(data, 3);
|
recv_frame(data, 3);
|
||||||
recv_frame(data, 4);
|
recv_frame(data, 4);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Ensure(FrameValidator, validates_one_byte_frame_with_correct_crc) {
|
||||||
|
uint8_t data[] = {0x44, 0x04, 0x6A, 0xB3, 0xA3};
|
||||||
|
expect(route_frame,
|
||||||
|
when(size, is_equal_to(1)),
|
||||||
|
when(data, is_equal_to_contents_of(data, 1))
|
||||||
|
);
|
||||||
|
recv_frame(data, 5);
|
||||||
|
}
|
||||||
|
|
||||||
|
Ensure(FrameValidator, does_not_validate_one_byte_frame_with_incorrect_crc) {
|
||||||
|
uint8_t data[] = {0x44, 0, 0, 0, 0};
|
||||||
|
never_expect(route_frame);
|
||||||
|
recv_frame(data, 5);
|
||||||
|
}
|
||||||
|
|
||||||
|
Ensure(FrameValidator, validates_four_byte_frame_with_correct_crc) {
|
||||||
|
uint8_t data[] = {0x44, 0x10, 0xFF, 0x00, 0x74, 0x4E, 0x30, 0xBA};
|
||||||
|
expect(route_frame,
|
||||||
|
when(size, is_equal_to(4)),
|
||||||
|
when(data, is_equal_to_contents_of(data, 4))
|
||||||
|
);
|
||||||
|
recv_frame(data, 8);
|
||||||
|
}
|
||||||
|
|
||||||
|
Ensure(FrameValidator, validates_five_byte_frame_with_correct_crc) {
|
||||||
|
//0xBA304E74
|
||||||
|
//0x470B99F4
|
||||||
|
uint8_t data[] = {1, 2, 3, 4, 5, 0xF4, 0x99, 0x0B, 0x47};
|
||||||
|
expect(route_frame,
|
||||||
|
when(size, is_equal_to(5)),
|
||||||
|
when(data, is_equal_to_contents_of(data, 5))
|
||||||
|
);
|
||||||
|
recv_frame(data, 9);
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue