1/*
2 * Copyright (C) 2016, The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *     http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <vector>
18
19#include <gtest/gtest.h>
20#include <utils/StrongPointer.h>
21#include <wifi_system/interface_tool.h>
22
23#include "android/net/wifi/IApInterface.h"
24#include "android/net/wifi/IWificond.h"
25#include "wificond/tests/integration/process_utils.h"
26#include "wificond/tests/mock_ap_interface_event_callback.h"
27
28using android::net::wifi::IApInterface;
29using android::net::wifi::IWificond;
30using android::wifi_system::InterfaceTool;
31using android::wificond::MockApInterfaceEventCallback;
32using android::wificond::tests::integration::HostapdIsDead;
33using android::wificond::tests::integration::HostapdIsRunning;
34using android::wificond::tests::integration::ScopedDevModeWificond;
35using android::wificond::tests::integration::WaitForTrue;
36using std::string;
37using std::vector;
38
39namespace android {
40namespace wificond {
41namespace {
42const char kInterfaceName[] = "wlan0";
43constexpr int kHostapdStartupTimeoutSeconds = 3;
44constexpr int kHostapdDeathTimeoutSeconds = 3;
45}  // namespace
46
47TEST(ApInterfaceTest, CanCreateApInterfaces) {
48  ScopedDevModeWificond dev_mode;
49  sp<IWificond> service = dev_mode.EnterDevModeOrDie();
50
51  // We should be able to create an AP interface.
52  sp<IApInterface> ap_interface;
53  EXPECT_TRUE(service->createApInterface(kInterfaceName, &ap_interface).isOk());
54  EXPECT_NE(nullptr, ap_interface.get());
55
56  // The interface should start out down.
57  string if_name;
58  EXPECT_TRUE(ap_interface->getInterfaceName(&if_name).isOk());
59  EXPECT_TRUE(!if_name.empty());
60  InterfaceTool if_tool;
61  EXPECT_FALSE(if_tool.GetUpState(if_name.c_str()));
62
63  // Mark the interface as up, just to test that we mark it down on tearDown.
64  EXPECT_TRUE(if_tool.SetUpState(if_name.c_str(), true));
65  EXPECT_TRUE(if_tool.GetUpState(if_name.c_str()));
66
67  // We should not be able to create two AP interfaces.
68  sp<IApInterface> ap_interface2;
69  EXPECT_TRUE(service->createApInterface(
70      kInterfaceName, &ap_interface2).isOk());
71  EXPECT_EQ(nullptr, ap_interface2.get());
72
73  // We can tear down the created interface.
74  bool success = false;
75  EXPECT_TRUE(service->tearDownApInterface(kInterfaceName, &success).isOk());
76  EXPECT_TRUE(success);
77  EXPECT_FALSE(if_tool.GetUpState(if_name.c_str()));
78
79  // Teardown everything at the end of the test.
80  EXPECT_TRUE(service->tearDownInterfaces().isOk());
81}
82
83// TODO: b/30311493 this test fails because hostapd fails to set the driver
84//       channel every other time.
85TEST(ApInterfaceTest, CanStartStopHostapd) {
86  ScopedDevModeWificond dev_mode;
87  sp<IWificond> service = dev_mode.EnterDevModeOrDie();
88  sp<IApInterface> ap_interface;
89  EXPECT_TRUE(service->createApInterface(kInterfaceName, &ap_interface).isOk());
90  ASSERT_NE(nullptr, ap_interface.get());
91
92  // Interface should start out down.
93  string if_name;
94  EXPECT_TRUE(ap_interface->getInterfaceName(&if_name).isOk());
95  EXPECT_TRUE(!if_name.empty());
96  InterfaceTool if_tool;
97  EXPECT_FALSE(if_tool.GetUpState(if_name.c_str()));
98
99  sp<MockApInterfaceEventCallback> ap_interface_event_callback(
100      new MockApInterfaceEventCallback());
101
102  for (int iteration = 0; iteration < 4; iteration++) {
103    bool hostapd_started = false;
104    EXPECT_TRUE(
105        ap_interface
106            ->startHostapd(ap_interface_event_callback, &hostapd_started)
107            .isOk());
108    EXPECT_TRUE(hostapd_started);
109
110    EXPECT_TRUE(WaitForTrue(HostapdIsRunning, kHostapdStartupTimeoutSeconds))
111        << "Failed on iteration " << iteration;
112
113    // There are two reasons to do this:
114    //   1) We look for hostapd so quickly that we miss when it dies on startup
115    //   2) If we don't give hostapd enough time to get fully up, killing it
116    //      can leave the driver in a poor state.
117    // The latter points to an obvious race, where we cannot fully clean up the
118    // driver on quick transitions.
119    auto InterfaceIsUp = [&if_tool, &if_name] () {
120      return if_tool.GetUpState(if_name.c_str());
121    };
122    EXPECT_TRUE(WaitForTrue(InterfaceIsUp, kHostapdStartupTimeoutSeconds))
123        << "Failed on iteration " << iteration;
124    EXPECT_TRUE(HostapdIsRunning()) << "Failed on iteration " << iteration;
125
126    bool hostapd_stopped = false;
127    EXPECT_TRUE(ap_interface->stopHostapd(&hostapd_stopped).isOk());
128    EXPECT_TRUE(hostapd_stopped);
129    EXPECT_FALSE(if_tool.GetUpState(if_name.c_str()));
130
131
132    EXPECT_TRUE(WaitForTrue(HostapdIsDead, kHostapdDeathTimeoutSeconds))
133        << "Failed on iteration " << iteration;
134  }
135}
136}  // namespace wificond
137}  // namespace android
138