apps/examples/power: Add cpu usage 100 test

The power consumption is effected by CPU usage.
To measure power consumption according to CPU status, Add an app that
makes the usage of each CPU 100%.

Signed-off-by: eunwoo.nam <[email protected]>

Author: ewoodev

apps/examples/power/power_main.c

@@ -32,6 +32,8 @@
 #include <unistd.h>
 #include <pthread.h>
 #include <sys/stat.h>
+#include <sched.h>
+#include <sys/prctl.h>
 
 #include <tinyara/fs/ioctl.h>
 
@@ -46,7 +48,41 @@
  ****************************************************************************/
 
 static int is_running;
+static volatile int cpu_stress_running;
 
+static int cpu_stress_thread(void *arg)
+{
+	int cpu_id = (int)arg;
+	cpu_set_t cpuset;
+	pthread_t this_thread = pthread_self();
+	char thread_name[32];
+	clock_t prev = clock();
+	clock_t now;
+
+#ifdef CONFIG_SMP
+	CPU_ZERO(&cpuset);
+	CPU_SET(cpu_id, &cpuset);
+	
+	if (pthread_setaffinity_np(this_thread, sizeof(cpu_set_t), &cpuset) != 0) {
+		printf("Failed to set CPU%d affinity for thread\n", cpu_id);
+		return -1;
+	}
+#endif
+	snprintf(thread_name, sizeof(thread_name), "cpu_stress%d", cpu_id);
+	pthread_setname_np(this_thread, thread_name);
+	printf("CPU stress thread started on CPU%d\n", cpu_id);
+	
+	while (cpu_stress_running) {
+		now = clock();
+		if (TICK2SEC(now - prev) > 10) {
+			printf("CPU%d stress running... %ld\n", cpu_id, now);
+			prev = now;
+		}
+	}
+	
+	printf("CPU%d stress thread stopped\n", cpu_id);
+	return 0;
+}
 
 static int pm_sleep_test(void *args)
 {
@@ -229,6 +265,72 @@ static int start_pm_test(int argc, char *argv[])
 	return 0;
 }
 
+static int start_cpu_stress_test(int argc, char *argv[])
+{
+	pthread_t cpu_threads[CONFIG_SMP_NCPUS];
+	int cpu_num;
+
+	printf("argc %d\n", argc);
+	for (int i = 0; i < argc; i++) {
+		printf("argv %s\n", argv[i]);
+	}
+	printf("######################### CPU STRESS TEST START #########################\n");
+	printf("Starting CPU stress test to achieve 100%% CPU usage\n");
+
+	printf("System has %d CPUs available\n", CONFIG_SMP_NCPUS);
+
+	if (argc < 2) {
+		printf("Creating threads with priority 120 on all %d CPUs\n", CONFIG_SMP_NCPUS);
+
+		for (int i = 0; i < CONFIG_SMP_NCPUS; i++) {
+			if (pthread_create(&cpu_threads[i], NULL, (pthread_startroutine_t)cpu_stress_thread, (void*)i) < 0) {
+				printf("Failed to create CPU%d stress thread(errno %d)\n", i, get_errno());
+				
+				for (int j = 0; j < i; j++) {
+					pthread_cancel(cpu_threads[j]);
+				}
+				cpu_stress_running = false;
+				return -1;
+			}
+		}
+
+		for (int i = 0; i < CONFIG_SMP_NCPUS; i++) {
+			pthread_join(cpu_threads[i], NULL);
+		}
+	} else {
+		cpu_num = atoi(argv[1]);
+		if (cpu_num < 0 || cpu_num >= CONFIG_SMP_NCPUS) {
+			printf("Invalid CPU number: %d. Use 0 to %d, or no argument for all CPUs\n", cpu_num, CONFIG_SMP_NCPUS - 1);
+			cpu_stress_running = false;
+			return -1;
+		}
+		printf("Creating thread on CPU%d only\n", cpu_num);
+
+		if (pthread_create(&cpu_threads[cpu_num], NULL, (pthread_startroutine_t)cpu_stress_thread, (void*)cpu_num) < 0) {
+			printf("Failed to create CPU%d stress thread(errno %d)\n", cpu_num, get_errno());
+			cpu_stress_running = false;
+			return -1;
+		}
+		pthread_join(cpu_threads[cpu_num], NULL);
+	}
+
+	printf("######################### CPU STRESS TEST END #########################\n");
+	return 0;
+}
+
+static int stop_cpu_stress_test(void)
+{
+	if (!cpu_stress_running) {
+		printf("CPU stress test is not running\n");
+		return 0;
+	}
+	
+	printf("Stopping CPU stress test...\n");
+	cpu_stress_running = false;
+
+	return 0;
+}
+
 static void help_func(void)
 {
 	printf("usage: power <command> \n\n");
@@ -240,12 +342,14 @@ static void help_func(void)
 	printf("and it is testing suspending of power managemenet operation and resuming for block ender chipset sleep mode.\n");
 	printf("   start [options]\t\t Start power management test\n");
 	printf("   stop           \t\t stop power management test\n");
+	printf("   cpu start [cpu_num]\t Start CPU stress test (100%% CPU usage)\n");
+	printf("   cpu stop        \t Stop CPU stress test\n");
 	printf("   options: -l, --lock-test              \t\t Test pm suspend/resume API\n");
 	printf("            -t, --timed-wakeup [time(ms)]\t\t Test timed wakeup (default 100ms)\n");
 	printf("\n");
-	printf("start and stop are used to control the power management test.\n");
-	printf("   suspend <name>\t\t Suspend power management test\n");
-	printf("   resume  <name>\t\t Start power management test\n");
+	printf("suspend and resume are used to control the power management test.\n");
+	printf("   suspend <name>\t\t suspend power management test\n");
+	printf("   resume  <name>\t\t resume power management test\n");
 	printf("\n");
 }
 
@@ -289,6 +393,37 @@ int power_main(int argc, char *argv[])
 
 		is_running = false;
 
+	} else if (strncmp(argv[1], "cpu", 8) == 0) {
+		if (argc < 3) {
+			printf("Usage: power cpu_stress {start|stop} [cpu_num]\n");
+			return -1;
+		}
+
+		if (strncmp(argv[2], "start", 6) == 0) {
+			if (cpu_stress_running) {
+				printf("CPU stress test is already running\n");
+				return 0;
+			}
+			cpu_stress_running = true;
+
+			pid = task_create("cpu_stress_test", 100, 1024, start_cpu_stress_test, argv + 3);
+			if (pid < 0) {
+				printf("Fail to create cpu_stress_test task(errno %d)\n", get_errno());
+				return -1;
+			}
+
+		} else if (strncmp(argv[2], "stop", 5) == 0) {
+			if (!cpu_stress_running) {
+				printf("CPU stress test is not running\n");
+				return 0;
+			}
+
+			cpu_stress_running = false;
+		} else {
+			printf("Invalid cpu_stress subcommand. Use 'start' or 'stop'\n");
+			return -1;
+		}
+
 	} else if (strncmp(argv[1], "suspend", 8) == 0 && argc == 3) {
 		_pm_suspend(argv[2]);
 		printf("Done pm suspend domain: %s\n", argv[2]);