1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
#include <doctest/doctest.h>
#include <vector>
#include "packeteer/l7/tftp.hpp"
using namespace packeteer::net;
namespace {
std::vector<unsigned char> build_rrq(const std::string& filename, const std::string& mode) {
std::vector<unsigned char> bytes = {0x00, 0x01}; // opcode RRQ
bytes.insert(bytes.end(), filename.begin(), filename.end());
bytes.push_back(0);
bytes.insert(bytes.end(), mode.begin(), mode.end());
bytes.push_back(0);
return bytes;
}
} // namespace
TEST_CASE("parse_tftp decodes an RRQ with filename and mode") {
auto msg = parse_tftp(build_rrq("boot.img", "octet"));
REQUIRE(msg.has_value());
CHECK(msg->opcode == kTftpRrq);
REQUIRE(msg->filename.has_value());
CHECK(*msg->filename == "boot.img");
REQUIRE(msg->mode.has_value());
CHECK(*msg->mode == "octet");
}
TEST_CASE("parse_tftp decodes a DATA block number") {
std::vector<unsigned char> bytes = {0x00, 0x03, 0x00, 0x07, 'h', 'i'};
auto msg = parse_tftp(bytes);
REQUIRE(msg.has_value());
CHECK(msg->opcode == kTftpData);
REQUIRE(msg->block.has_value());
CHECK(*msg->block == 7);
}
TEST_CASE("parse_tftp decodes an ACK block number") {
std::vector<unsigned char> bytes = {0x00, 0x04, 0x00, 0x07};
auto msg = parse_tftp(bytes);
REQUIRE(msg.has_value());
CHECK(msg->opcode == kTftpAck);
REQUIRE(msg->block.has_value());
CHECK(*msg->block == 7);
}
TEST_CASE("parse_tftp decodes an ERROR code and message") {
std::vector<unsigned char> bytes = {0x00, 0x05, 0x00, 0x01};
std::string message = "File not found";
bytes.insert(bytes.end(), message.begin(), message.end());
bytes.push_back(0);
auto msg = parse_tftp(bytes);
REQUIRE(msg.has_value());
CHECK(msg->opcode == kTftpError);
REQUIRE(msg->error_code.has_value());
CHECK(*msg->error_code == 1);
REQUIRE(msg->error_message.has_value());
CHECK(*msg->error_message == "File not found");
}
TEST_CASE("parse_tftp rejects an opcode outside the known range") {
std::vector<unsigned char> bytes = {0x00, 0x99};
CHECK_FALSE(parse_tftp(bytes).has_value());
}
TEST_CASE("parse_tftp handles an RRQ with a truncated filename gracefully") {
std::vector<unsigned char> bytes = {0x00, 0x01, 'a', 'b'}; // no null terminator
auto msg = parse_tftp(bytes);
REQUIRE(msg.has_value());
CHECK(msg->opcode == kTftpRrq);
CHECK_FALSE(msg->filename.has_value());
}
TEST_CASE("TftpDissector claims port 69 and formats an RRQ") {
TftpDissector dissector;
CHECK(dissector.port() == kTftpPort);
auto summary = dissector.summarize(build_rrq("boot.img", "octet"));
REQUIRE(summary.has_value());
CHECK(*summary == "TFTP RRQ boot.img (octet)");
}
|