|
| 1 | +#include "ef.h" |
| 2 | +#include "ef-test.h" |
| 3 | +#include "catch_single_include.hxx" |
| 4 | + |
| 5 | +#include <cstring> |
| 6 | + |
| 7 | +// RAII guard to set NO_PAD for the duration of a scope |
| 8 | +struct NoPadGuard { |
| 9 | + NoPadGuard() { NO_PAD = 1; } |
| 10 | + ~NoPadGuard() { NO_PAD = 0; } |
| 11 | +}; |
| 12 | + |
| 13 | +// Build a frame and its expected buf/mask from a frame spec |
| 14 | +static void build_frame(std::vector<const char *> spec, |
| 15 | + buf_t **out_buf, buf_t **out_mask, |
| 16 | + frame_t **out_frame) |
| 17 | +{ |
| 18 | + frame_t *f = parse_frame_wrap(spec); |
| 19 | + REQUIRE(f != NULL); |
| 20 | + *out_buf = frame_to_buf(f); |
| 21 | + REQUIRE(*out_buf != NULL); |
| 22 | + *out_mask = f->has_mask ? frame_mask_to_buf(f) : NULL; |
| 23 | + *out_frame = f; |
| 24 | +} |
| 25 | + |
| 26 | +TEST_CASE("no-pad: eth-only frame is 14 bytes", "[nopad]") { |
| 27 | + NoPadGuard g; |
| 28 | + buf_t *buf, *mask; |
| 29 | + frame_t *f; |
| 30 | + build_frame({"eth", "dmac", "::1", "smac", "::2"}, &buf, &mask, &f); |
| 31 | + |
| 32 | + // eth header = 14 bytes, no padding to 60 |
| 33 | + CHECK(buf->size == 14); |
| 34 | + |
| 35 | + bfree(buf); |
| 36 | + frame_free(f); |
| 37 | +} |
| 38 | + |
| 39 | +TEST_CASE("no-pad: default pads to 60 bytes", "[nopad]") { |
| 40 | + CHECK(NO_PAD == 0); |
| 41 | + buf_t *buf, *mask; |
| 42 | + frame_t *f; |
| 43 | + build_frame({"eth", "dmac", "::1", "smac", "::2"}, &buf, &mask, &f); |
| 44 | + |
| 45 | + CHECK(buf->size == 60); |
| 46 | + |
| 47 | + bfree(buf); |
| 48 | + frame_free(f); |
| 49 | +} |
| 50 | + |
| 51 | +TEST_CASE("no-pad: mask buf also skips padding", "[nopad]") { |
| 52 | + NoPadGuard g; |
| 53 | + |
| 54 | + // Use 'ign' on smac to force mask generation |
| 55 | + frame_t *f = parse_frame_wrap({"eth", "dmac", "::1", "smac", "ign"}); |
| 56 | + REQUIRE(f != NULL); |
| 57 | + buf_t *buf = frame_to_buf(f); |
| 58 | + buf_t *mask = frame_mask_to_buf(f); |
| 59 | + REQUIRE(buf != NULL); |
| 60 | + REQUIRE(mask != NULL); |
| 61 | + |
| 62 | + CHECK(buf->size == 14); |
| 63 | + CHECK(mask->size == 14); |
| 64 | + |
| 65 | + bfree(buf); |
| 66 | + bfree(mask); |
| 67 | + frame_free(f); |
| 68 | +} |
| 69 | + |
| 70 | +TEST_CASE("no-pad: frame with payload stays exact size", "[nopad]") { |
| 71 | + NoPadGuard g; |
| 72 | + buf_t *buf, *mask; |
| 73 | + frame_t *f; |
| 74 | + |
| 75 | + // eth(14) + data pattern cnt 4 = 18 bytes, well under 60 |
| 76 | + build_frame({"eth", "dmac", "::1", "smac", "::2", |
| 77 | + "data", "pattern", "cnt", "4"}, &buf, &mask, &f); |
| 78 | + |
| 79 | + CHECK(buf->size == 18); |
| 80 | + |
| 81 | + bfree(buf); |
| 82 | + frame_free(f); |
| 83 | +} |
| 84 | + |
| 85 | +TEST_CASE("no-pad: large frame unaffected", "[nopad]") { |
| 86 | + NoPadGuard g; |
| 87 | + buf_t *buf, *mask; |
| 88 | + frame_t *f; |
| 89 | + |
| 90 | + // eth(14) + data pattern cnt 100 = 114 bytes, already > 60 |
| 91 | + build_frame({"eth", "dmac", "::1", "smac", "::2", |
| 92 | + "data", "pattern", "cnt", "100"}, &buf, &mask, &f); |
| 93 | + |
| 94 | + CHECK(buf->size == 114); |
| 95 | + |
| 96 | + bfree(buf); |
| 97 | + frame_free(f); |
| 98 | +} |
0 commit comments