ulib/tftp/tftp-test.cpp

730 bool verify_read_data(const test_state& ts, const tx_test_data& td) {
732     EXPECT_EQ(td.expected.offset, td.actual.offset, "read offset mismatch");
733     EXPECT_EQ(td.expected.len, td.actual.len, "read length mismatch");
738     ASSERT_EQ(td.expected.block, ntohs(msg->block), "bad block number");
739     EXPECT_BYTES_EQ(td.expected.data, msg->data, td.actual.len, "read data mismatch");
748     tx_test_data* td = static_cast<tx_test_data*>(cookie);
749     td->actual.len = *len;
750     td->actual.offset = offset;
751     memcpy(data, td->expected.data, *len);
800     tx_test_data td;
801     td.expected.len = block_size;
802     td.expected.data[block_size - 1] = 'X';
803     status = tftp_process_msg(ts.session, buf, buf_sz, ts.out, &ts.outlen, &ts.timeout, &td);
816     EXPECT_TRUE(verify_read_data(ts, td), "bad test data");
821         td.expected.block = 2;
823         td.expected.offset = block_size;
824         td.expected.data[0] = 'X';
826             td.expected.data[second_block_size - 1] = 'F';
829         status = tftp_prepare_data(ts.session, ts.out, &ts.outlen, &ts.timeout, &td);
831         EXPECT_TRUE(verify_read_data(ts, td), "bad test data");
969     tx_test_data td;
970     status = tftp_process_msg(ts.session, buf, sizeof(buf), ts.out, &ts.outlen, &ts.timeout, &td);
982     tx_test_data* td = static_cast<tx_test_data*>(cookie);
983     td->actual.len = *len;
984     td->actual.offset = offset;
985     memcpy(td->actual.data + offset, data, *len);
989 bool verify_write_data(const uint8_t* expected, const tx_test_data& td) {
991     ASSERT_EQ(td.expected.offset, td.actual.offset, "write offset mismatch");
992     ASSERT_EQ(td.expected.len, td.actual.len, "write length mismatch");
993     EXPECT_BYTES_EQ(expected, td.actual.data + td.actual.offset, td.actual.len, "write data mismatch");
1035     tx_test_data td;
1036     status = tftp_process_msg(ts.session, data_buf, sizeof(data_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1039     EXPECT_TRUE(verify_write_data(data_buf + 4, td), "bad write data");
1083     tx_test_data td;
1084     status = tftp_process_msg(ts.session, data_buf, sizeof(data_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1087     ASSERT_TRUE(verify_write_data(data_buf + 4, td), "bad write data");
1093     td.expected.block++;
1094     td.expected.offset = DEFAULT_BLOCKSIZE;
1096     status = tftp_process_msg(ts.session, data_buf, sizeof(data_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1099     EXPECT_TRUE(verify_write_data(data_buf + 4, td), "bad write data");
1147     tx_test_data td;
1148     td.expected.len = 1024;
1149     status = tftp_process_msg(ts.session, data_buf, sizeof(data_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1152     EXPECT_TRUE(verify_write_data(data_buf + 4, td), "bad write data");
1197     tx_test_data td;
1198     status = tftp_process_msg(ts.session, data_buf, sizeof(data_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1201     EXPECT_TRUE(verify_write_data(data_buf + 4, td), "bad write data");
1209     td.expected.block++;
1210     td.expected.offset += DEFAULT_BLOCKSIZE;
1212     status = tftp_process_msg(ts.session, data_buf, sizeof(data_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1215     EXPECT_TRUE(verify_write_data(data_buf + 4, td), "bad write data");
1323     tx_test_data td;
1324     status = tftp_process_msg(ts.session, data_buf, sizeof(data_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1326     EXPECT_TRUE(verify_write_data(data_buf + 4, td), "bad write data");
1334     td.expected.block++;
1335     td.expected.offset += DEFAULT_BLOCKSIZE;
1337     status = tftp_process_msg(ts.session, data_buf, sizeof(data_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1339     EXPECT_TRUE(verify_write_data(data_buf + 4, td), "bad write data");
1357     EXPECT_EQ(0, td.actual.data[1024], "block 3 should be empty");
1461     tx_test_data td;
1462     status = tftp_process_msg(ts.session, oack_buf, sizeof(oack_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1465     EXPECT_TRUE(verify_read_data(ts, td), "bad test data");
1472     td.expected.block = 2;
1473     td.expected.offset += DEFAULT_BLOCKSIZE;
1474     td.expected.data[1] = 'f';
1475     status = tftp_process_msg(ts.session, ack_buf, sizeof(ack_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1482     EXPECT_TRUE(verify_read_data(ts, td), "bad test data");
1507     tx_test_data td;
1508     status = tftp_process_msg(ts.session, oack_buf, sizeof(oack_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1511     ASSERT_TRUE(verify_read_data(ts, td), "bad test data");
1518     td.expected.block = 2;
1519     td.expected.offset += DEFAULT_BLOCKSIZE;
1520     status = tftp_process_msg(ts.session, ack_buf, sizeof(ack_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1523     EXPECT_TRUE(verify_read_data(ts, td), "bad test data");
1527     status = tftp_process_msg(ts.session, ack_buf, sizeof(ack_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1560     tx_test_data td;
1561     status = tftp_process_msg(ts.session, oack_buf, sizeof(oack_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1564     ASSERT_TRUE(verify_read_data(ts, td), "bad test data");
1605     tx_test_data td;
1606     status = tftp_process_msg(ts.session, oack_buf, sizeof(oack_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1611     ASSERT_TRUE(verify_read_data(ts, td), "bad test data");
1614     td.expected.block++;
1615     td.expected.offset += DEFAULT_BLOCKSIZE;
1616     td.expected.data[0]++;
1617     status = tftp_prepare_data(ts.session, ts.out, &ts.outlen, &ts.timeout, &td);
1623     ASSERT_TRUE(verify_read_data(ts, td), "bad test data");
1631     td.expected.block++;
1632     td.expected.offset += DEFAULT_BLOCKSIZE;
1633     td.expected.data[1]++;
1634     status = tftp_process_msg(ts.session, ack_buf, sizeof(ack_buf), ts.out, &ts.outlen, &ts.timeout, &td);
1640     EXPECT_TRUE(verify_read_data(ts, td), "bad test data");