212 lines
8 KiB
C++
212 lines
8 KiB
C++
//
|
|
// Copyright (C) 2013 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.
|
|
//
|
|
|
|
// This file provides tests to verify that Callback80211Metrics sends UMA
|
|
// notifications for appropriate messages and doesn't send them for
|
|
// inappropriate messages.
|
|
|
|
#include "shill/wifi/callback80211_metrics.h"
|
|
|
|
#include <memory>
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "shill/mock_event_dispatcher.h"
|
|
#include "shill/mock_log.h"
|
|
#include "shill/mock_metrics.h"
|
|
#include "shill/net/ieee80211.h"
|
|
#include "shill/net/netlink_packet.h"
|
|
#include "shill/net/nl80211_message.h"
|
|
#include "shill/refptr_types.h"
|
|
|
|
using base::Bind;
|
|
using std::unique_ptr;
|
|
using testing::_;
|
|
using testing::Test;
|
|
|
|
namespace shill {
|
|
|
|
namespace {
|
|
|
|
// Unless otherwise specified, these data blocks have been collected by shill
|
|
// using NetlinkManager while, simultaneously (and manually) comparing shill
|
|
// output with that of the 'iw' code from which it was derived. The test
|
|
// strings represent the raw packet data coming from the kernel. The
|
|
// comments above each of these strings is the markup that 'iw' outputs for
|
|
// each of these packets.
|
|
|
|
// These constants are consistent across the applicable packets, below.
|
|
|
|
const uint16_t kNl80211FamilyId = 0x13;
|
|
const IEEE_80211::WiFiReasonCode kExpectedDisconnectReason =
|
|
IEEE_80211::kReasonCodePreviousAuthenticationInvalid;
|
|
|
|
// NL80211_CMD_DISCONNECT message.
|
|
// wlan0 (phy #0): disconnected (by AP) reason: 2: Previous authentication no
|
|
// longer valid
|
|
|
|
const unsigned char kDisconnectMessage[] = {
|
|
0x30, 0x00, 0x00, 0x00, 0x13, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x30, 0x01, 0x00, 0x00, 0x08, 0x00, 0x01, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x08, 0x00, 0x03, 0x00,
|
|
0x04, 0x00, 0x00, 0x00, 0x06, 0x00, 0x36, 0x00,
|
|
0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x47, 0x00,
|
|
};
|
|
|
|
// NL80211_CMD_DISCONNECT message.
|
|
// Copied from kDisconnectMessage but with most of the payload removed.
|
|
|
|
const unsigned char kEmptyDisconnectMessage[] = {
|
|
0x1c, 0x00, 0x00, 0x00, 0x13, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x30, 0x01, 0x00, 0x00, 0x08, 0x00, 0x01, 0x00,
|
|
0x00, 0x00, 0x00, 0x00,
|
|
};
|
|
|
|
// NL80211_CMD_DEAUTHENTICATE message.
|
|
// wlan0 (phy #0): deauth c0:3f:0e:77:e8:7f -> ff:ff:ff:ff:ff:ff reason 2:
|
|
// Previous authentication no longer valid [frame: c0 00 00 00 ff ff ff ff
|
|
// ff ff c0 3f 0e 77 e8 7f c0 3f 0e 77 e8 7f c0 0e 02 00]
|
|
|
|
const unsigned char kDeauthenticateMessage[] = {
|
|
0x44, 0x00, 0x00, 0x00, 0x13, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x27, 0x01, 0x00, 0x00, 0x08, 0x00, 0x01, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x08, 0x00, 0x03, 0x00,
|
|
0x04, 0x00, 0x00, 0x00, 0x1e, 0x00, 0x33, 0x00,
|
|
0xc0, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff,
|
|
0xff, 0xff, 0xc0, 0x3f, 0x0e, 0x77, 0xe8, 0x7f,
|
|
0xc0, 0x3f, 0x0e, 0x77, 0xe8, 0x7f, 0xc0, 0x0e,
|
|
0x02, 0x00, 0x00, 0x00,
|
|
};
|
|
|
|
// NL80211_CMD_DEAUTHENTICATE message.
|
|
// Copied from kDeauthenticateMessage but with most of the payload
|
|
// removed.
|
|
|
|
const unsigned char kEmptyDeauthenticateMessage[] = {
|
|
0x1c, 0x00, 0x00, 0x00, 0x13, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x27, 0x01, 0x00, 0x00, 0x08, 0x00, 0x01, 0x00,
|
|
0x00, 0x00, 0x00, 0x00,
|
|
};
|
|
|
|
// NL80211_CMD_NEW_STATION message.
|
|
// kNewStationMessage is an nl80211 message that's not a deauthenticate or
|
|
// disconnect message. Used to make sure that only those nl80211 messages
|
|
// generate an UMA message.
|
|
// wlan0: new station c0:3f:0e:77:e8:7f
|
|
|
|
const unsigned char kNewStationMessage[] = {
|
|
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,
|
|
};
|
|
|
|
// CTRL_CMD_GETFAMILY message.
|
|
// kGetFamilyMessage is not an nl80211 message. Used to make sure that
|
|
// non-nl80211 messages don't generate an UMA message.
|
|
//
|
|
// Extracted from net.log. It's just a non-nl80211 message (it's actually a
|
|
// message that's sent to the kernel rather than one received from the kernel
|
|
// but the code doesn't differentiate and this message was much shorter than the
|
|
// response).
|
|
|
|
const unsigned char kGetFamilyMessage[] = {
|
|
0x10, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x02, 0x00,
|
|
0x6e, 0x6c, 0x38, 0x30, 0x32, 0x31, 0x31, 0x00
|
|
};
|
|
|
|
} // namespace
|
|
|
|
class Callback80211MetricsTest : public Test {
|
|
public:
|
|
Callback80211MetricsTest() :
|
|
metrics_(&dispatcher_), callback_(&metrics_) {
|
|
message_factory_.AddFactoryMethod(
|
|
kNl80211FamilyId, Bind(&Nl80211Message::CreateMessage));
|
|
Nl80211Message::SetMessageType(kNl80211FamilyId);
|
|
}
|
|
|
|
protected:
|
|
MockEventDispatcher dispatcher_;
|
|
MockMetrics metrics_;
|
|
NetlinkMessageFactory message_factory_;
|
|
Callback80211Metrics callback_;
|
|
};
|
|
|
|
// Make sure that notifications happen for correctly formed messages.
|
|
TEST_F(Callback80211MetricsTest, DisconnectMessage) {
|
|
NetlinkPacket packet(kDisconnectMessage, sizeof(kDisconnectMessage));
|
|
unique_ptr<NetlinkMessage> netlink_message(message_factory_.CreateMessage(
|
|
&packet, NetlinkMessage::MessageContext()));
|
|
EXPECT_CALL(metrics_, Notify80211Disconnect(Metrics::kDisconnectedByAp,
|
|
kExpectedDisconnectReason));
|
|
callback_.CollectDisconnectStatistics(*netlink_message);
|
|
}
|
|
|
|
TEST_F(Callback80211MetricsTest, DeauthMessage) {
|
|
NetlinkPacket packet(kDeauthenticateMessage, sizeof(kDeauthenticateMessage));
|
|
unique_ptr<NetlinkMessage> netlink_message(message_factory_.CreateMessage(
|
|
&packet, NetlinkMessage::MessageContext()));
|
|
EXPECT_CALL(metrics_, Notify80211Disconnect(Metrics::kDisconnectedNotByAp,
|
|
kExpectedDisconnectReason));
|
|
callback_.CollectDisconnectStatistics(*netlink_message);
|
|
}
|
|
|
|
// Make sure there's no notification if there's no reason code in the message.
|
|
TEST_F(Callback80211MetricsTest, EmptyDisconnectMessage) {
|
|
NetlinkPacket packet(
|
|
kEmptyDisconnectMessage, sizeof(kEmptyDisconnectMessage));
|
|
unique_ptr<NetlinkMessage> netlink_message(message_factory_.CreateMessage(
|
|
&packet, NetlinkMessage::MessageContext()));
|
|
EXPECT_CALL(metrics_, Notify80211Disconnect(_, _)).Times(0);
|
|
callback_.CollectDisconnectStatistics(*netlink_message);
|
|
}
|
|
|
|
TEST_F(Callback80211MetricsTest, EmptyDeauthMessage) {
|
|
NetlinkPacket packet(
|
|
kEmptyDeauthenticateMessage, sizeof(kEmptyDeauthenticateMessage));
|
|
unique_ptr<NetlinkMessage> netlink_message(message_factory_.CreateMessage(
|
|
&packet, NetlinkMessage::MessageContext()));
|
|
EXPECT_CALL(metrics_, Notify80211Disconnect(_, _)).Times(0);
|
|
callback_.CollectDisconnectStatistics(*netlink_message);
|
|
}
|
|
|
|
// Make sure the callback doesn't notify anyone for message of the wrong type.
|
|
TEST_F(Callback80211MetricsTest, Nl80211NotDisconnectDeauthMessage) {
|
|
NetlinkPacket packet(kNewStationMessage, sizeof(kNewStationMessage));
|
|
unique_ptr<NetlinkMessage> netlink_message(message_factory_.CreateMessage(
|
|
&packet, NetlinkMessage::MessageContext()));
|
|
EXPECT_CALL(metrics_, Notify80211Disconnect(_, _)).Times(0);
|
|
callback_.CollectDisconnectStatistics(*netlink_message);
|
|
}
|
|
|
|
TEST_F(Callback80211MetricsTest, NotNl80211Message) {
|
|
NetlinkPacket packet(kGetFamilyMessage, sizeof(kGetFamilyMessage));
|
|
unique_ptr<NetlinkMessage> netlink_message(message_factory_.CreateMessage(
|
|
&packet, NetlinkMessage::MessageContext()));
|
|
EXPECT_CALL(metrics_, Notify80211Disconnect(_, _)).Times(0);
|
|
callback_.CollectDisconnectStatistics(*netlink_message);
|
|
}
|
|
|
|
} // namespace shill
|