3rd-party/libtins
1
0
Fork 0
mirror of https://github.com/mfontanini/libtins synced 2026-01-27 20:24:26 +01:00
Code Issues Releases Wiki Activity

Added RC4EAPOL tests.

Browse Source
This commit is contained in:
Matias Fontanini
2012-10-18 12:42:05 -03:00
parent f4d37f5a51
commit d4bcefb1d6
5 changed files with 145 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
include/eapol.h
View File

@@ -187,9 +187,19 @@ namespace Tins {
typedef std::vector<uint8_t> key_type;
/**
* \brief This PDU's flag.
* This PDU's flag.
*/
static const PDU::PDUType pdu_flag = PDU::RC4EAPOL;
/**
* The length of the key IV field
*/
static const size_t key_iv_size = 16;
/**
* The length of the key sign field
*/
static const size_t key_sign_size = 16;
/**
* \brief Creates an instance of RC4EAPOL
@@ -259,7 +269,7 @@ namespace Tins {
* \brief Sets the replay counter field.
* \param new_replay_counter The new replay counter to be set.
*/
void replay_counter(uint16_t new_replay_counter);
void replay_counter(uint64_t new_replay_counter);
/**
* \brief Sets the key IV field.
@@ -330,7 +340,7 @@ namespace Tins {
struct rc4hdr {
uint16_t key_length;
uint64_t replay_counter;
uint8_t key_iv[16];
uint8_t key_iv[key_iv_size];
uint8_t key_index:7,
key_flag:1;
uint8_t key_sign[16];

4
src/eapol.cpp
View File

@@ -80,7 +80,7 @@ void EAPOL::packet_type(uint8_t new_ptype) {
}
void EAPOL::length(uint16_t new_length) {
_header.length = new_length;
_header.length = Endian::host_to_be(new_length);
}
void EAPOL::type(uint8_t new_type) {
@@ -127,7 +127,7 @@ void RC4EAPOL::key_length(uint16_t new_key_length) {
_header.key_length = Endian::host_to_be(new_key_length);
}
void RC4EAPOL::replay_counter(uint16_t new_replay_counter) {
void RC4EAPOL::replay_counter(uint64_t new_replay_counter) {
_header.replay_counter = Endian::host_to_be(new_replay_counter);
}

17
tests/depends.d
View File

@@ -197,6 +197,23 @@ src/radiotap.o: src/radiotap.cpp ../include/radiotap.h ../include/pdu.h \
../include/ip_address.h:
../include/utils.h:
src/rc4eapol.o: src/rc4eapol.cpp ../include/eapol.h ../include/pdu.h \
../include/small_uint.h ../include/endianness.h ../include/utils.h \
../include/ip_address.h ../include/hw_address.h
../include/eapol.h:
../include/pdu.h:
../include/small_uint.h:
../include/endianness.h:
../include/utils.h:
../include/ip_address.h:
../include/hw_address.h:
src/snap.o: src/snap.cpp ../include/snap.h ../include/pdu.h \
../include/endianness.h ../include/small_uint.h ../include/utils.h \
../include/ip_address.h ../include/hw_address.h

3
tests/src/radiotap.cpp
View File

@@ -11,8 +11,6 @@ using namespace Tins;
class RadioTapTest : public testing::Test {
public:
static const uint8_t expected_packet[];
void test_equals(const RadioTap &radio1, const RadioTap &radio2);
};
const uint8_t RadioTapTest::expected_packet[] = {
@@ -55,7 +53,6 @@ TEST_F(RadioTapTest, ConstructorFromBuffer) {
EXPECT_EQ(radio.dbm_signal(), 0xda);
EXPECT_EQ(radio.dbm_noise(), 0xa0);
EXPECT_EQ(radio.antenna(), 2);
EXPECT_EQ(radio.rx_flags(), 0);
}
TEST_F(RadioTapTest, Serialize) {

113
tests/src/rc4eapol.cpp Normal file
View File

@@ -0,0 +1,113 @@
#include <gtest/gtest.h>
#include <cstring>
#include <string>
#include <stdint.h>
#include "eapol.h"
#include "utils.h"
using namespace std;
using namespace Tins;
class RC4EAPOLTest : public testing::Test {
public:
static const uint8_t expected_packet[];
};
const uint8_t RC4EAPOLTest::expected_packet[] = {
};
TEST_F(RC4EAPOLTest, DefaultConstructor) {
uint8_t empty_iv[RC4EAPOL::key_iv_size] = { 0 };
RC4EAPOL eapol;
EXPECT_EQ(1, eapol.version());
EXPECT_EQ(0x3, eapol.packet_type());
EXPECT_EQ(EAPOL::RC4, eapol.type());
EXPECT_EQ(0, eapol.length());
EXPECT_EQ(0, eapol.key_length());
EXPECT_EQ(0, eapol.replay_counter());
EXPECT_TRUE(std::equal(empty_iv, empty_iv + sizeof(empty_iv), eapol.key_iv()));
EXPECT_EQ(0, eapol.key_flag());
EXPECT_EQ(0, eapol.key_index());
EXPECT_TRUE(std::equal(empty_iv, empty_iv + sizeof(empty_iv), eapol.key_sign()));
EXPECT_EQ(RC4EAPOL::key_type(), eapol.key());
}
TEST_F(RC4EAPOLTest, Version) {
RC4EAPOL eapol;
eapol.version(0x7a);
EXPECT_EQ(0x7a, eapol.version());
}
TEST_F(RC4EAPOLTest, PacketType) {
RC4EAPOL eapol;
eapol.packet_type(0x7a);
EXPECT_EQ(0x7a, eapol.packet_type());
}
TEST_F(RC4EAPOLTest, Length) {
RC4EAPOL eapol;
eapol.length(0x7af2);
EXPECT_EQ(0x7af2, eapol.length());
}
TEST_F(RC4EAPOLTest, Type) {
RC4EAPOL eapol;
eapol.type(0x7a);
EXPECT_EQ(0x7a, eapol.type());
}
TEST_F(RC4EAPOLTest, KeyLength) {
RC4EAPOL eapol;
eapol.key_length(0x7af3);
EXPECT_EQ(0x7af3, eapol.key_length());
}
TEST_F(RC4EAPOLTest, ReplayCounter) {
RC4EAPOL eapol;
eapol.replay_counter(0x7af3d91a1fd3abLL);
EXPECT_EQ(0x7af3d91a1fd3abLL, eapol.replay_counter());
}
TEST_F(RC4EAPOLTest, KeyIV) {
uint8_t iv[RC4EAPOL::key_iv_size];
for(unsigned i = 0; i < RC4EAPOL::key_iv_size; ++i)
iv[i] = i;
RC4EAPOL eapol;
eapol.key_iv(iv);
EXPECT_TRUE(std::equal(iv, iv + sizeof(iv), eapol.key_iv()));
}
TEST_F(RC4EAPOLTest, KeyFlag) {
RC4EAPOL eapol;
eapol.key_flag(1);
EXPECT_EQ(1, eapol.key_flag());
eapol.key_flag(0);
EXPECT_EQ(0, eapol.key_flag());
}
TEST_F(RC4EAPOLTest, KeyIndex) {
RC4EAPOL eapol;
eapol.key_index(0x7d);
EXPECT_EQ(0x7d, eapol.key_index());
}
TEST_F(RC4EAPOLTest, KeySign) {
uint8_t sign[RC4EAPOL::key_sign_size];
for(unsigned i = 0; i < RC4EAPOL::key_sign_size; ++i)
sign[i] = i;
RC4EAPOL eapol;
eapol.key_sign(sign);
EXPECT_TRUE(std::equal(sign, sign + sizeof(sign), eapol.key_sign()));
}
TEST_F(RC4EAPOLTest, Key) {
RC4EAPOL eapol;
uint8_t arr[] = { 1, 9, 2, 0x71, 0x87, 0xfa, 0xdf };
RC4EAPOL::key_type key(arr, arr + sizeof(arr));
eapol.key(key);
EXPECT_EQ(key, eapol.key());
}