You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
283 lines
11 KiB
283 lines
11 KiB
/*
|
|
* Copyright (C) 2016, The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include <memory>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "wificond/net/kernel-header-latest/nl80211.h"
|
|
#include "wificond/net/nl80211_attribute.h"
|
|
#include "wificond/net/nl80211_packet.h"
|
|
|
|
using std::string;
|
|
|
|
namespace android {
|
|
namespace wificond {
|
|
|
|
namespace {
|
|
|
|
const uint8_t kU8Value1 = 150;
|
|
const uint16_t kU16Value1 = 5000;
|
|
const uint32_t kU32Value1 = 500000;
|
|
const uint32_t kU32Value2 = 800000;
|
|
const uint32_t kPortId = 123;
|
|
|
|
const uint8_t kGenNLCommand = 102;
|
|
const uint16_t kNLMsgType = 4000;
|
|
const uint32_t kNLMsgSequenceNumber = 70000;
|
|
|
|
// NL80211 family id is dynamically allocated by kernel.
|
|
const uint16_t kNL80211FamilyId = 0x13;
|
|
const uint32_t kWiPhy = 0;
|
|
const uint32_t kExpectedIfIndex = 4;
|
|
const uint8_t kMacAddressBytes[] = {
|
|
0xc0, 0x3f, 0x0e, 0x77, 0xe8, 0x7f
|
|
};
|
|
|
|
const unsigned char kNL80211_CMD_ASSOCIATE[] = {
|
|
0x58, 0x00, 0x00, 0x00, 0x13, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x26, 0x01, 0x00, 0x00, 0x08, 0x00, 0x01, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x08, 0x00, 0x03, 0x00,
|
|
0x04, 0x00, 0x00, 0x00, 0x32, 0x00, 0x33, 0x00,
|
|
0x10, 0x00, 0x3a, 0x01, 0x48, 0x5d, 0x60, 0x77,
|
|
0x2d, 0xcf, 0xc0, 0x3f, 0x0e, 0x77, 0xe8, 0x7f,
|
|
0xc0, 0x3f, 0x0e, 0x77, 0xe8, 0x7f, 0x40, 0x07,
|
|
0x01, 0x04, 0x00, 0x00, 0x01, 0xc0, 0x01, 0x08,
|
|
0x82, 0x84, 0x8b, 0x96, 0x0c, 0x12, 0x18, 0x24,
|
|
0x32, 0x04, 0x30, 0x48, 0x60, 0x6c, 0x00, 0x00,
|
|
};
|
|
|
|
const uint32_t kExpectedCqmNotAcked = 50;
|
|
|
|
const unsigned char kNL80211_CMD_NOTIFY_CQM[] = {
|
|
0x3c, 0x00, 0x00, 0x00, 0x13, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x40, 0x01, 0x00, 0x00, 0x08, 0x00, 0x01, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x08, 0x00, 0x03, 0x00,
|
|
0x04, 0x00, 0x00, 0x00, 0x0a, 0x00, 0x06, 0x00,
|
|
0xc0, 0x3f, 0x0e, 0x77, 0xe8, 0x7f, 0x00, 0x00,
|
|
0x0c, 0x00, 0x5e, 0x00, 0x08, 0x00, 0x04, 0x00,
|
|
0x32, 0x00, 0x00, 0x00,
|
|
};
|
|
|
|
const uint32_t kNewStationExpectedGeneration = 275;
|
|
|
|
const unsigned char kNL80211_CMD_NEW_STATION[] = {
|
|
0x34, 0x00, 0x00, 0x00, 0x13, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x13, 0x01, 0x00, 0x00, 0x08, 0x00, 0x03, 0x00,
|
|
0x04, 0x00, 0x00, 0x00, 0x0a, 0x00, 0x06, 0x00,
|
|
0xc0, 0x3f, 0x0e, 0x77, 0xe8, 0x7f, 0x00, 0x00,
|
|
0x08, 0x00, 0x2e, 0x00, 0x13, 0x01, 0x00, 0x00,
|
|
0x04, 0x00, 0x15, 0x00,
|
|
};
|
|
|
|
} // namespace
|
|
|
|
TEST(NL80211PacketTest, CanConstructValidNL80211Packet) {
|
|
NL80211Packet netlink_packet(kNLMsgType,
|
|
kGenNLCommand,
|
|
kNLMsgSequenceNumber,
|
|
kPortId);
|
|
EXPECT_TRUE(netlink_packet.IsValid());
|
|
}
|
|
|
|
TEST(NL80211PacketTest, SetAndGetNL80211PacketHeaderFields) {
|
|
NL80211Packet netlink_packet(kNLMsgType,
|
|
kGenNLCommand,
|
|
kNLMsgSequenceNumber,
|
|
kPortId);
|
|
netlink_packet.SetFlags(NLM_F_MULTI);
|
|
EXPECT_EQ(kGenNLCommand, netlink_packet.GetCommand());
|
|
EXPECT_EQ(kNLMsgType, netlink_packet.GetMessageType());
|
|
EXPECT_EQ(kNLMsgSequenceNumber, netlink_packet.GetMessageSequence());
|
|
EXPECT_TRUE(netlink_packet.IsMulti());
|
|
EXPECT_EQ(kPortId, netlink_packet.GetPortId());
|
|
}
|
|
|
|
TEST(NL80211PacketTest, AddAttributeToNL80211Packet) {
|
|
NL80211Packet netlink_packet(kNLMsgType,
|
|
kGenNLCommand,
|
|
kNLMsgSequenceNumber,
|
|
kPortId);
|
|
NL80211Attr<uint8_t> u8_attr(1, kU8Value1);
|
|
netlink_packet.AddAttribute(u8_attr);
|
|
EXPECT_TRUE(netlink_packet.IsValid());
|
|
EXPECT_TRUE(netlink_packet.HasAttribute(1));
|
|
uint8_t attr_value;
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(1, &attr_value));
|
|
EXPECT_EQ(attr_value, kU8Value1);
|
|
}
|
|
|
|
TEST(NL80211PacketTest, AddMultipleAttributesToNL80211Packet) {
|
|
NL80211Packet netlink_packet(kNLMsgType,
|
|
kGenNLCommand,
|
|
kNLMsgSequenceNumber,
|
|
kPortId);
|
|
NL80211Attr<uint8_t> u8_attr(1, kU8Value1);
|
|
NL80211Attr<uint32_t> u32_attr(2, kU32Value1);
|
|
NL80211Attr<uint16_t> u16_attr(3, kU16Value1);
|
|
netlink_packet.AddAttribute(u8_attr);
|
|
netlink_packet.AddAttribute(u32_attr);
|
|
netlink_packet.AddAttribute(u16_attr);
|
|
EXPECT_TRUE(netlink_packet.IsValid());
|
|
EXPECT_TRUE(netlink_packet.HasAttribute(1));
|
|
EXPECT_TRUE(netlink_packet.HasAttribute(2));
|
|
EXPECT_TRUE(netlink_packet.HasAttribute(3));
|
|
uint8_t u8_attr_value;
|
|
uint32_t u32_attr_value;
|
|
uint16_t u16_attr_value;
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(1, &u8_attr_value));
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(2, &u32_attr_value));
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(3, &u16_attr_value));
|
|
EXPECT_EQ(u8_attr_value, kU8Value1);
|
|
EXPECT_EQ(u32_attr_value, kU32Value1);
|
|
EXPECT_EQ(u16_attr_value, kU16Value1);
|
|
}
|
|
|
|
TEST(NL80211PacketTest, AddNestedAttributesToNL80211Packet) {
|
|
NL80211Packet netlink_packet(kNLMsgType,
|
|
kGenNLCommand,
|
|
kNLMsgSequenceNumber,
|
|
kPortId);
|
|
NL80211NestedAttr nested_attr(1);
|
|
NL80211Attr<uint16_t> u16_attr(2, kU16Value1);
|
|
NL80211Attr<uint32_t> u32_attr_1(3, kU32Value1);
|
|
NL80211Attr<uint32_t> u32_attr_2(4, kU32Value2);
|
|
nested_attr.AddAttribute(u16_attr);
|
|
nested_attr.AddAttribute(u32_attr_1);
|
|
netlink_packet.AddAttribute(nested_attr);
|
|
netlink_packet.AddAttribute(u32_attr_2);
|
|
EXPECT_TRUE(netlink_packet.HasAttribute(1));
|
|
EXPECT_TRUE(netlink_packet.HasAttribute(4));
|
|
// Attribute 2 and 3 are deeper nested.
|
|
// They should not be found from packet level.
|
|
EXPECT_FALSE(netlink_packet.HasAttribute(2));
|
|
EXPECT_FALSE(netlink_packet.HasAttribute(3));
|
|
}
|
|
|
|
TEST(NL80211PacketTest, CannotGetMissingAttributeFromNL80211Packet) {
|
|
NL80211Packet netlink_packet(kNLMsgType,
|
|
kGenNLCommand,
|
|
kNLMsgSequenceNumber,
|
|
kPortId);
|
|
NL80211Attr<uint8_t> u8_attr(1, kU8Value1);
|
|
netlink_packet.AddAttribute(u8_attr);
|
|
EXPECT_TRUE(netlink_packet.IsValid());
|
|
EXPECT_FALSE(netlink_packet.HasAttribute(2));
|
|
uint8_t attr_value;
|
|
EXPECT_FALSE(netlink_packet.GetAttributeValue(2, &attr_value));
|
|
}
|
|
|
|
TEST(NL80211PacketTest, CanGetAllOfAttributeFromNL80211Packet) {
|
|
NL80211Packet netlink_packet(kNLMsgType,
|
|
kGenNLCommand,
|
|
kNLMsgSequenceNumber,
|
|
kPortId);
|
|
NL80211Attr<uint8_t> u8_attr(1, kU8Value1);
|
|
NL80211Attr<uint32_t> u32_attr_1(2, kU32Value1);
|
|
NL80211Attr<uint32_t> u32_attr_2(4, kU32Value2);
|
|
netlink_packet.AddAttribute(u8_attr);
|
|
netlink_packet.AddAttribute(u32_attr_1);
|
|
netlink_packet.AddAttribute(u32_attr_2);
|
|
EXPECT_TRUE(netlink_packet.IsValid());
|
|
std::vector<BaseNL80211Attr> attributes;
|
|
EXPECT_TRUE(netlink_packet.GetAllAttributes(&attributes));
|
|
|
|
EXPECT_TRUE(attributes.size() == 3);
|
|
|
|
NL80211Attr<uint8_t>* u8_attr_retrieved =
|
|
static_cast<NL80211Attr<uint8_t>*>(&attributes[0]);
|
|
EXPECT_TRUE(u8_attr_retrieved->GetValue() == kU8Value1);
|
|
|
|
NL80211Attr<uint32_t>* u32_attr_1_retrieved =
|
|
static_cast<NL80211Attr<uint32_t>*>(&attributes[1]);
|
|
EXPECT_TRUE(u32_attr_1_retrieved->GetValue() == kU32Value1);
|
|
|
|
NL80211Attr<uint32_t>* u32_attr_2_retrieved =
|
|
static_cast<NL80211Attr<uint32_t>*>(&attributes[2]);
|
|
EXPECT_TRUE(u32_attr_2_retrieved->GetValue() == kU32Value2);
|
|
}
|
|
|
|
TEST(NL80211PacketTest, ParseCMDAssociateTest) {
|
|
NL80211Packet netlink_packet(std::vector<uint8_t>(
|
|
kNL80211_CMD_ASSOCIATE,
|
|
kNL80211_CMD_ASSOCIATE + sizeof(kNL80211_CMD_ASSOCIATE)));
|
|
EXPECT_TRUE(netlink_packet.IsValid());
|
|
EXPECT_EQ(kNL80211FamilyId, netlink_packet.GetMessageType());
|
|
EXPECT_EQ(NL80211_CMD_ASSOCIATE, netlink_packet.GetCommand());
|
|
uint32_t value;
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(NL80211_ATTR_WIPHY, &value));
|
|
EXPECT_EQ(kWiPhy, value);
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(NL80211_ATTR_IFINDEX, &value));
|
|
EXPECT_EQ(kExpectedIfIndex, value);
|
|
std::vector<uint8_t> rawdata;
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(NL80211_ATTR_FRAME, &rawdata));
|
|
EXPECT_FALSE(rawdata.empty());
|
|
|
|
}
|
|
|
|
TEST(NL80211PacketTest, ParseCMDNotifyCQMTest) {
|
|
NL80211Packet netlink_packet(std::vector<uint8_t>(
|
|
kNL80211_CMD_NOTIFY_CQM,
|
|
kNL80211_CMD_NOTIFY_CQM + sizeof(kNL80211_CMD_NOTIFY_CQM)));
|
|
EXPECT_TRUE(netlink_packet.IsValid());
|
|
EXPECT_EQ(kNL80211FamilyId, netlink_packet.GetMessageType());
|
|
EXPECT_EQ(NL80211_CMD_NOTIFY_CQM, netlink_packet.GetCommand());
|
|
uint32_t value;
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(NL80211_ATTR_WIPHY, &value));
|
|
EXPECT_EQ(kWiPhy, value);
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(NL80211_ATTR_IFINDEX, &value));
|
|
EXPECT_EQ(kExpectedIfIndex, value);
|
|
std::vector<uint8_t> mac;
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(NL80211_ATTR_MAC, &mac));
|
|
std::vector<uint8_t> expected_mac(
|
|
kMacAddressBytes,
|
|
kMacAddressBytes + sizeof(kMacAddressBytes));
|
|
EXPECT_EQ(expected_mac, mac);
|
|
NL80211NestedAttr nested(0);
|
|
EXPECT_TRUE(netlink_packet.GetAttribute(NL80211_ATTR_CQM, &nested));
|
|
EXPECT_TRUE(nested.GetAttributeValue(NL80211_ATTR_CQM_PKT_LOSS_EVENT, &value));
|
|
EXPECT_EQ(kExpectedCqmNotAcked, value);
|
|
}
|
|
|
|
TEST(NL80211PacketTest, ParseCMDNewStation) {
|
|
NL80211Packet netlink_packet(std::vector<uint8_t>(
|
|
kNL80211_CMD_NEW_STATION,
|
|
kNL80211_CMD_NEW_STATION + sizeof(kNL80211_CMD_NEW_STATION)));
|
|
EXPECT_TRUE(netlink_packet.IsValid());
|
|
EXPECT_EQ(kNL80211FamilyId, netlink_packet.GetMessageType());
|
|
EXPECT_EQ(NL80211_CMD_NEW_STATION, netlink_packet.GetCommand());
|
|
uint32_t value;
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(NL80211_ATTR_IFINDEX, &value));
|
|
EXPECT_EQ(kExpectedIfIndex, value);
|
|
std::vector<uint8_t> mac;
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(NL80211_ATTR_MAC, &mac));
|
|
std::vector<uint8_t> expected_mac(
|
|
kMacAddressBytes,
|
|
kMacAddressBytes + sizeof(kMacAddressBytes));
|
|
EXPECT_EQ(expected_mac, mac);
|
|
NL80211NestedAttr nested(0);
|
|
EXPECT_TRUE(netlink_packet.GetAttribute(NL80211_ATTR_STA_INFO, &nested));
|
|
EXPECT_TRUE(netlink_packet.GetAttributeValue(NL80211_ATTR_GENERATION,
|
|
&value));
|
|
EXPECT_EQ(kNewStationExpectedGeneration, value);
|
|
}
|
|
|
|
} // namespace wificond
|
|
} // namespace android
|