Samuka007/asterinas
1
0
Fork 0
mirror of https://github.com/asterinas/asterinas.git synced 2025-06-08 21:06:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add tests for system call sched_get/set_affinity

Browse Source
This commit is contained in:
Zhang Junyang 2024-09-23 15:46:52 +08:00 committed by Tate, Hongliang Tian
parent e319641b4d
commit c38455f9f6
3 changed files with 70 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
test/apps/cpu_affinity/cpu_affinity.c Normal file
View File

@ -0,0 +1,69 @@
// SPDX-License-Identifier: MPL-2.0
#define _GNU_SOURCE
#include <sched.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <pthread.h>
int main()
{
pid_t pid = getpid();
cpu_set_t mask;
// Get current affinity mask
if (sched_getaffinity(pid, sizeof(cpu_set_t), &mask) == -1) {
perror("sched_getaffinity");
exit(EXIT_FAILURE);
}
printf("Current CPU affinity:");
int cur_cpu_count = 0;
for (int i = 0; i < CPU_SETSIZE; i++) {
if (CPU_ISSET(i, &mask)) {
printf(" %d", i);
cur_cpu_count++;
}
}
printf("\n");
if (cur_cpu_count == 0) {
printf("Error: No CPU affinity set\n");
exit(EXIT_FAILURE);
}
// Set the process to run on CPU 0 only
CPU_ZERO(&mask);
CPU_SET(0, &mask);
if (sched_setaffinity(pid, sizeof(cpu_set_t), &mask) == -1) {
perror("sched_setaffinity");
exit(EXIT_FAILURE);
}
printf("Set CPU affinity to CPU 0\n");
// Verify the new CPU affinity
if (sched_getaffinity(pid, sizeof(cpu_set_t), &mask) == -1) {
perror("sched_getaffinity");
exit(EXIT_FAILURE);
}
printf("New CPU affinity:");
cur_cpu_count = 0;
for (int i = 0; i < CPU_SETSIZE; i++) {
if (CPU_ISSET(i, &mask)) {
printf(" %d", i);
cur_cpu_count++;
if (i != 0) {
printf("Error: CPU affinity not set to CPU 0\n");
exit(EXIT_FAILURE);
}
}
}
printf("\n");
if (cur_cpu_count != 1) {
printf("Error: CPU affinity not set to CPU 0\n");
exit(EXIT_FAILURE);
}
return 0;
}

35
test/apps/cpu_affinity/sched_getaffinity.c
View File

@ -1,35 +0,0 @@
// SPDX-License-Identifier: MPL-2.0
#define _GNU_SOURCE
#include <stdio.h>
#include <unistd.h>
#include <syscall.h>
#include <errno.h>
#include <sched.h> // Include sched.h for CPU_SETSIZE
int main()
{
// Create a mask for CPU_SETSIZE number of CPUs
unsigned long mask[CPU_SETSIZE / sizeof(unsigned long)];
int mask_size = sizeof(mask);
// Call the raw syscall to retrieve the CPU affinity mask of the current process
long res = syscall(__NR_sched_getaffinity, 0, mask_size, &mask);
if (res < 0) {
perror("Error calling sched_getaffinity");
return errno;
}
// Print the CPUs that are part of the current process's affinity mask
printf("Process can run on: ");
for (int i = 0; i < CPU_SETSIZE; ++i) {
if (mask[i / (8 * sizeof(long))] &
(1UL << (i % (8 * sizeof(long))))) {
printf("%d ", i);
}
}
printf("\n");
return 0;
}

1
test/apps/scripts/process.sh
View File

@ -13,6 +13,7 @@ tests="
clone3/clone_exit_signal clone3/clone_exit_signal
clone3/clone_no_exit_signal clone3/clone_no_exit_signal
clone3/clone_process clone3/clone_process
cpu_affinity/cpu_affinity
execve/execve execve/execve
eventfd2/eventfd2 eventfd2/eventfd2
fork/fork fork/fork