blob: 1fbe71ee20b6e541253f3994baf5d8d756ce09ba [file] [log] [blame]
// Copyright 2018 The Fuchsia Authors
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file or at
#include <inttypes.h>
#include <lib/unittest/unittest.h>
#include <platform.h>
#include <zircon/errors.h>
#include <zircon/types.h>
#include <kernel/cpu.h>
#include <kernel/mp.h>
#include <kernel/thread.h>
#include <ktl/iterator.h>
#include "tests.h"
#include <ktl/enforce.h>
static int resume_cpu_test_thread(void* arg) {
*reinterpret_cast<cpu_num_t*>(arg) = arch_curr_cpu_num();
return 0;
// "Unplug" online secondary (non-BOOT) cores
static zx_status_t unplug_all_cores(Thread** leaked_threads) {
cpu_mask_t cpumask = mp_get_online_mask() & ~cpu_num_to_mask(BOOT_CPU_ID);
return mp_unplug_cpu_mask(cpumask, ZX_TIME_INFINITE, leaked_threads);
static zx_status_t hotplug_core(cpu_num_t i) {
cpu_mask_t cpumask = cpu_num_to_mask(i);
return mp_hotplug_cpu_mask(cpumask);
static unsigned get_num_cpus_online() {
unsigned count = 0;
cpu_mask_t online = mp_get_online_mask();
while (online) {
online >>= 1;
return count;
// Unplug all cores (except for Boot core), then hotplug
// the cores one by one and make sure that we can schedule
// tasks on that core.
[[maybe_unused]] static bool mp_hotplug_test() {
// Hotplug is only implemented for x64.
#if !defined(__x86_64__)
printf("skipping test mp_hotplug, hotplug only suported on x64\n");
uint num_cores = get_num_cpus_online();
if (num_cores < 2) {
printf("skipping test mp_hotplug, not enough online cpus\n");
// "Unplug" online secondary (non-BOOT) cores
Thread* leaked_threads[SMP_MAX_CPUS] = {};
ASSERT_EQ(unplug_all_cores(leaked_threads), ZX_OK, "unplugging all cores failed");
for (cpu_num_t i = 0; i < num_cores; i++) {
if (i == BOOT_CPU_ID) {
// hotplug this core.
ASSERT_EQ(hotplug_core(i), ZX_OK, "hotplugging core failed");
// Create a thread, affine it to the core just hotplugged
// and make sure the thread does get scheduled there.
cpu_num_t running_core;
Thread* nt = Thread::Create("resume-test-thread", resume_cpu_test_thread, &running_core,
ASSERT_NE(nullptr, nt, "Thread create failed");
nt->SetMigrateFn([](auto...) {});
ASSERT_EQ(nt->Join(nullptr, ZX_TIME_INFINITE), ZX_OK, "thread join failed");
ASSERT_EQ(i, running_core, "Thread not running on hotplugged core");
for (Thread* leaked_thread : leaked_threads) {
if (leaked_thread) {
UNITTEST("test unplug and hotplug cores one by one", mp_hotplug_test)
UNITTEST_END_TESTCASE(mp_hotplug_tests, "hotplug", "Tests for unplugging and hotplugging cores")