142 lines
4.1 KiB
C++
142 lines
4.1 KiB
C++
/*
|
|
* 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 "wificond/tests/integration/process_utils.h"
|
|
|
|
#include <unistd.h>
|
|
|
|
#include <android-base/logging.h>
|
|
#include <android-base/stringprintf.h>
|
|
#include <android-base/strings.h>
|
|
#include <binder/IBinder.h>
|
|
#include <binder/IServiceManager.h>
|
|
#include <utils/String16.h>
|
|
#include <utils/StrongPointer.h>
|
|
|
|
#include "wificond/ipc_constants.h"
|
|
#include "wificond/tests/shell_utils.h"
|
|
|
|
using android::String16;
|
|
using android::base::StringPrintf;
|
|
using android::base::Trim;
|
|
using android::net::wifi::nl80211::IWificond;
|
|
using android::sp;
|
|
using android::wificond::ipc_constants::kServiceName;
|
|
|
|
namespace android {
|
|
namespace wificond {
|
|
namespace tests {
|
|
namespace integration {
|
|
namespace {
|
|
|
|
bool IsProcessRunning(const char* process_name) {
|
|
std::string output;
|
|
RunShellCommand(StringPrintf("pgrep -c ^%s$", process_name), &output);
|
|
output = Trim(output);
|
|
if (output == "0") {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
}
|
|
|
|
const uint32_t ScopedDevModeWificond::kSystemServerDeathTimeoutSeconds = 10;
|
|
const uint32_t ScopedDevModeWificond::kSystemServerStartTimeoutSeconds = 10;
|
|
const uint32_t ScopedDevModeWificond::kWificondDeathTimeoutSeconds = 10;
|
|
const uint32_t ScopedDevModeWificond::kWificondStartTimeoutSeconds = 10;
|
|
|
|
ScopedDevModeWificond::~ScopedDevModeWificond() {
|
|
if (in_dev_mode_) {
|
|
ExitDevMode();
|
|
}
|
|
}
|
|
|
|
sp<IWificond> ScopedDevModeWificond::EnterDevModeOrDie() {
|
|
sp<IWificond> service;
|
|
RunShellCommand("stop wificond");
|
|
CHECK(WaitForTrue(WificondIsDead, kWificondDeathTimeoutSeconds));
|
|
RunShellCommand("stop");
|
|
CHECK(WaitForTrue(SystemServerIsDead, kSystemServerDeathTimeoutSeconds));
|
|
RunShellCommand("start wificond");
|
|
auto wificond_restarted = std::bind(IsBinderServiceRegistered, kServiceName);
|
|
CHECK(WaitForTrue(wificond_restarted, kWificondStartTimeoutSeconds));
|
|
status_t status = getService(String16(kServiceName), &service);
|
|
if (status != NO_ERROR || service.get() == nullptr) {
|
|
LOG(FATAL) << "Failed to restart wificond in dev mode";
|
|
}
|
|
in_dev_mode_ = true;
|
|
return service;
|
|
}
|
|
|
|
void ScopedDevModeWificond::ExitDevMode() {
|
|
RunShellCommand("stop wificond");
|
|
CHECK(WaitForTrue(WificondIsDead, kWificondDeathTimeoutSeconds));
|
|
RunShellCommand("start");
|
|
CHECK(WaitForTrue(SystemServerIsRunning, kSystemServerStartTimeoutSeconds));
|
|
RunShellCommand("start wificond");
|
|
CHECK(WaitForTrue(WificondIsRunning, kWificondStartTimeoutSeconds));
|
|
in_dev_mode_ = false;
|
|
}
|
|
|
|
bool WaitForTrue(std::function<bool()> condition, time_t timeout_seconds) {
|
|
time_t start_time_seconds = time(nullptr);
|
|
do {
|
|
if (condition()) {
|
|
return true;
|
|
}
|
|
usleep(1000);
|
|
} while ((time(nullptr) - start_time_seconds) < timeout_seconds);
|
|
return false;
|
|
}
|
|
|
|
bool IsBinderServiceRegistered(const char* service_name) {
|
|
sp<IBinder> service =
|
|
defaultServiceManager()->checkService(String16(service_name));
|
|
return service.get() != nullptr;
|
|
}
|
|
|
|
bool SystemServerIsRunning() {
|
|
return IsProcessRunning("system_server");
|
|
}
|
|
|
|
bool SystemServerIsDead() {
|
|
return !IsProcessRunning("system_server");
|
|
}
|
|
|
|
bool WificondIsRunning() {
|
|
return IsProcessRunning("wificond");
|
|
}
|
|
|
|
bool WificondIsDead() { return !WificondIsRunning(); }
|
|
|
|
bool HostapdIsRunning() {
|
|
return IsProcessRunning("hostapd");
|
|
}
|
|
|
|
bool HostapdIsDead() { return !HostapdIsRunning(); }
|
|
|
|
bool SupplicantIsRunning() {
|
|
return IsProcessRunning("wpa_supplicant");
|
|
}
|
|
|
|
bool SupplicantIsDead() { return !SupplicantIsRunning(); }
|
|
|
|
} // namespace integration
|
|
} // namespace tests
|
|
} // namespace android
|
|
} // namespace wificond
|