Commit e6487230 authored by Maciej Wieczor-Retman's avatar Maciej Wieczor-Retman Committed by Shuah Khan

selftests/resctrl: Simplify cleanup in ctrl-c handler

Ctrl-c handler isn't aware of what test is currently running. Because of
that it executes all cleanups even if they aren't necessary. Since the
ctrl-c handler uses the sa_sigaction system no parameters can be passed
to it as function arguments.

Add a global variable to make ctrl-c handler aware of the currently run
test and only execute the correct cleanup callback.
Signed-off-by: default avatarMaciej Wieczor-Retman <maciej.wieczor-retman@intel.com>
Reviewed-by: default avatarReinette Chatre <reinette.chatre@intel.com>
Signed-off-by: default avatarShuah Khan <skhan@linuxfoundation.org>
parent 8780bc88
...@@ -158,7 +158,6 @@ int resctrl_val(const struct resctrl_test *test, ...@@ -158,7 +158,6 @@ int resctrl_val(const struct resctrl_test *test,
const struct user_params *uparams, const struct user_params *uparams,
const char * const *benchmark_cmd, const char * const *benchmark_cmd,
struct resctrl_val_param *param); struct resctrl_val_param *param);
void tests_cleanup(void);
void mbm_test_cleanup(void); void mbm_test_cleanup(void);
void mba_test_cleanup(void); void mba_test_cleanup(void);
unsigned long create_bit_mask(unsigned int start, unsigned int len); unsigned long create_bit_mask(unsigned int start, unsigned int len);
...@@ -168,7 +167,7 @@ int get_mask_no_shareable(const char *cache_type, unsigned long *mask); ...@@ -168,7 +167,7 @@ int get_mask_no_shareable(const char *cache_type, unsigned long *mask);
int get_cache_size(int cpu_no, const char *cache_type, unsigned long *cache_size); int get_cache_size(int cpu_no, const char *cache_type, unsigned long *cache_size);
int resource_info_unsigned_get(const char *resource, const char *filename, unsigned int *val); int resource_info_unsigned_get(const char *resource, const char *filename, unsigned int *val);
void ctrlc_handler(int signum, siginfo_t *info, void *ptr); void ctrlc_handler(int signum, siginfo_t *info, void *ptr);
int signal_handler_register(void); int signal_handler_register(const struct resctrl_test *test);
void signal_handler_unregister(void); void signal_handler_unregister(void);
void cat_test_cleanup(void); void cat_test_cleanup(void);
unsigned int count_bits(unsigned long n); unsigned int count_bits(unsigned long n);
......
...@@ -81,19 +81,11 @@ static void cmd_help(void) ...@@ -81,19 +81,11 @@ static void cmd_help(void)
printf("\t-h: help\n"); printf("\t-h: help\n");
} }
void tests_cleanup(void) static int test_prepare(const struct resctrl_test *test)
{
mbm_test_cleanup();
mba_test_cleanup();
cmt_test_cleanup();
cat_test_cleanup();
}
static int test_prepare(void)
{ {
int res; int res;
res = signal_handler_register(); res = signal_handler_register(test);
if (res) { if (res) {
ksft_print_msg("Failed to register signal handler\n"); ksft_print_msg("Failed to register signal handler\n");
return res; return res;
...@@ -136,7 +128,7 @@ static void run_single_test(const struct resctrl_test *test, const struct user_p ...@@ -136,7 +128,7 @@ static void run_single_test(const struct resctrl_test *test, const struct user_p
ksft_print_msg("Starting %s test ...\n", test->name); ksft_print_msg("Starting %s test ...\n", test->name);
if (test_prepare()) { if (test_prepare(test)) {
ksft_exit_fail_msg("Abnormal failure when preparing for the test\n"); ksft_exit_fail_msg("Abnormal failure when preparing for the test\n");
return; return;
} }
......
...@@ -62,6 +62,7 @@ struct imc_counter_config { ...@@ -62,6 +62,7 @@ struct imc_counter_config {
static char mbm_total_path[1024]; static char mbm_total_path[1024];
static int imcs; static int imcs;
static struct imc_counter_config imc_counters_config[MAX_IMCS][2]; static struct imc_counter_config imc_counters_config[MAX_IMCS][2];
static const struct resctrl_test *current_test;
void membw_initialize_perf_event_attr(int i, int j) void membw_initialize_perf_event_attr(int i, int j)
{ {
...@@ -472,7 +473,8 @@ void ctrlc_handler(int signum, siginfo_t *info, void *ptr) ...@@ -472,7 +473,8 @@ void ctrlc_handler(int signum, siginfo_t *info, void *ptr)
if (bm_pid) if (bm_pid)
kill(bm_pid, SIGKILL); kill(bm_pid, SIGKILL);
umount_resctrlfs(); umount_resctrlfs();
tests_cleanup(); if (current_test && current_test->cleanup)
current_test->cleanup();
ksft_print_msg("Ending\n\n"); ksft_print_msg("Ending\n\n");
exit(EXIT_SUCCESS); exit(EXIT_SUCCESS);
...@@ -482,13 +484,14 @@ void ctrlc_handler(int signum, siginfo_t *info, void *ptr) ...@@ -482,13 +484,14 @@ void ctrlc_handler(int signum, siginfo_t *info, void *ptr)
* Register CTRL-C handler for parent, as it has to kill * Register CTRL-C handler for parent, as it has to kill
* child process before exiting. * child process before exiting.
*/ */
int signal_handler_register(void) int signal_handler_register(const struct resctrl_test *test)
{ {
struct sigaction sigact = {}; struct sigaction sigact = {};
int ret = 0; int ret = 0;
bm_pid = 0; bm_pid = 0;
current_test = test;
sigact.sa_sigaction = ctrlc_handler; sigact.sa_sigaction = ctrlc_handler;
sigemptyset(&sigact.sa_mask); sigemptyset(&sigact.sa_mask);
sigact.sa_flags = SA_SIGINFO; sigact.sa_flags = SA_SIGINFO;
...@@ -510,6 +513,7 @@ void signal_handler_unregister(void) ...@@ -510,6 +513,7 @@ void signal_handler_unregister(void)
{ {
struct sigaction sigact = {}; struct sigaction sigact = {};
current_test = NULL;
sigact.sa_handler = SIG_DFL; sigact.sa_handler = SIG_DFL;
sigemptyset(&sigact.sa_mask); sigemptyset(&sigact.sa_mask);
if (sigaction(SIGINT, &sigact, NULL) || if (sigaction(SIGINT, &sigact, NULL) ||
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment