123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767 |
- #define _GNU_SOURCE
- #include "gamemode.h"
- #include "helpers.h"
- #include "logging.h"
- #include <libgen.h>
- #include <sys/types.h>
- #include <sys/wait.h>
- #include <unistd.h>
- #include "daemon_config.h"
- #include "external-helper.h"
- #include "gamemode_client.h"
- #include "governors-query.h"
- #include "gpu-control.h"
- static int verify_gamemode_initial(struct GameModeConfig *config)
- {
- int status = 0;
- if ((status = gamemode_query_status()) != 0 && status != -1) {
- long reaper = config_get_reaper_frequency(config);
- LOG_MSG("GameMode was active, waiting for the reaper thread (%ld seconds)!\n", reaper);
- sleep(1);
-
- for (int i = 0; i < reaper; i++) {
- if ((status = gamemode_query_status()) == 0) {
- status = 0;
- break;
- } else if (status == -1) {
- goto status_error;
- }
- LOG_MSG("Waiting...\n");
- sleep(1);
- }
- if (status == 1)
- LOG_ERROR("GameMode still active, cannot run tests!\n");
- } else if (status == -1) {
- goto status_error;
- } else {
- status = 0;
- }
- return status;
- status_error:
- LOG_ERROR("gamemode_query_status failed: %s!\n", gamemode_error_string());
- LOG_ERROR("is gamemode installed correctly?\n");
- return -1;
- }
- static int verify_active_and_registered(void)
- {
- int status = gamemode_query_status();
- if (status != 2) {
- if (status == -1) {
- LOG_ERROR("gamemode_query_status failed: %s\n", gamemode_error_string());
- } else if (status == 1) {
- LOG_ERROR("gamemode was active but did not have this process registered\n");
- }
- LOG_ERROR("gamemode failed to activate correctly when requested (expected 2)!\n");
- status = -1;
- } else {
- status = 0;
- }
- return status;
- }
- static int verify_deactivated(void)
- {
- int status = gamemode_query_status();
- if (status != 0) {
- if (status == -1) {
- LOG_ERROR("gamemode_query_status failed: %s\n", gamemode_error_string());
- }
- LOG_ERROR("gamemode failed to deactivate when requested (expected 0)!\n");
- status = -1;
- } else {
- status = 0;
- }
- return status;
- }
- static int verify_other_client_connected(void)
- {
- int status = gamemode_query_status();
- if (status != 1) {
- if (status == -1) {
- LOG_ERROR("gamemode_query_status failed: %s\n", gamemode_error_string());
- }
- LOG_ERROR("gamemode_query_status failed to return other client connected (expected 1)!\n");
- status = -1;
- } else {
- status = 0;
- }
- return status;
- }
- static int run_basic_client_tests(void)
- {
- LOG_MSG(":: Basic client tests\n");
-
- if (gamemode_request_start() != 0) {
- LOG_ERROR("gamemode_request_start failed: %s\n", gamemode_error_string());
- return -1;
- }
-
- if (verify_active_and_registered() != 0)
- return -1;
-
- if (gamemode_request_end() != 0) {
- LOG_ERROR("gamemode_request_end failed: %s!\n", gamemode_error_string());
- return -1;
- }
-
- if (verify_deactivated() != 0)
- return -1;
- LOG_MSG(":: Passed\n\n");
- return 0;
- }
- static int run_dual_client_tests(void)
- {
- int status = 0;
-
- LOG_MSG(":: Dual client tests\n");
-
- char mypath[PATH_MAX];
- memset(mypath, 0, sizeof(mypath));
- if (readlink("/proc/self/exe", mypath, PATH_MAX) == -1) {
- LOG_ERROR("could not read current exe path: %s\n", strerror(errno));
- return -1;
- }
-
- int child = fork();
- if (child == 0) {
-
- if (execl(mypath, mypath, "-r", (char *)NULL) == -1) {
- LOG_ERROR("failed to re-launch self (%s) with execl: %s\n", mypath, strerror(errno));
- return -1;
- }
- }
-
-
-
- usleep(10000);
-
- if (verify_other_client_connected() != 0)
- status = -1;
-
- if (gamemode_request_start() != 0) {
- LOG_ERROR("gamemode_request_start failed: %s\n", gamemode_error_string());
- status = -1;
- }
-
- if (verify_active_and_registered() != 0)
- status = -1;
-
- if (gamemode_request_end() != 0) {
- LOG_ERROR("gamemode_request_end failed: %s!\n", gamemode_error_string());
- status = -1;
- }
-
- if (verify_other_client_connected() != 0)
- status = -1;
-
- if (kill(child, SIGINT) == -1) {
- LOG_ERROR("failed to send continue signal to other client: %s\n", strerror(errno));
- status = -1;
- }
-
- usleep(100000);
-
- int wstatus;
- while (waitpid(child, &wstatus, WNOHANG) == 0) {
- LOG_MSG("...Waiting for child to quit...\n");
- usleep(100000);
- }
-
- if (verify_deactivated() != 0)
- return -1;
- if (status == 0)
- LOG_MSG(":: Passed\n\n");
- return status;
- }
- static int run_gamemoderun_and_reaper_tests(struct GameModeConfig *config)
- {
- int status = 0;
- LOG_MSG(":: Gamemoderun and reaper thread tests\n");
-
- int child = fork();
- if (child == 0) {
-
- fclose(stdout);
-
- if (execl("/usr/bin/gamemoderun", "/usr/bin/gamemoderun", "sh", (char *)NULL) == -1) {
- LOG_ERROR("failed to launch gamemoderun with execl: %s\n", strerror(errno));
- return -1;
- }
- }
-
- usleep(10000);
-
- if (verify_other_client_connected() != 0)
- status = -1;
-
- if (kill(child, SIGTERM) == -1) {
- LOG_ERROR("failed to send continue signal to other client: %s\n", strerror(errno));
- status = -1;
- }
-
- int wstatus;
- while (waitpid(child, &wstatus, WNOHANG) == 0) {
- LOG_MSG("...Waiting for child to quit...\n");
- usleep(100000);
- }
-
- long freq = config_get_reaper_frequency(config);
- LOG_MSG("...Waiting for reaper thread (reaper_frequency set to %ld seconds)...\n", freq);
- sleep((unsigned int)freq);
-
- if (verify_deactivated() != 0)
- return -1;
- if (status == 0)
- LOG_MSG(":: Passed\n\n");
- return status;
- }
- static int run_cpu_governor_tests(struct GameModeConfig *config)
- {
-
- char desiredgov[CONFIG_VALUE_MAX] = { 0 };
- config_get_desired_governor(config, desiredgov);
- if (desiredgov[0] == '\0')
- strcpy(desiredgov, "performance");
- char defaultgov[CONFIG_VALUE_MAX] = { 0 };
- if (defaultgov[0] == '\0') {
- const char *currentgov = get_gov_state();
- if (currentgov) {
- strncpy(defaultgov, currentgov, CONFIG_VALUE_MAX - 1);
- } else {
- LOG_ERROR(
- "Could not get current CPU governor state, this indicates an error! See rest "
- "of log.\n");
- return -1;
- }
- }
-
- gamemode_request_start();
-
- const char *currentgov = get_gov_state();
- if (strncmp(currentgov, desiredgov, CONFIG_VALUE_MAX) != 0) {
- LOG_ERROR("Governor was not set to %s (was actually %s)!\n", desiredgov, currentgov);
- gamemode_request_end();
- return -1;
- }
-
- gamemode_request_end();
-
- currentgov = get_gov_state();
- if (strncmp(currentgov, defaultgov, CONFIG_VALUE_MAX) != 0) {
- LOG_ERROR("Governor was not set back to %s (was actually %s)!\n", defaultgov, currentgov);
- return -1;
- }
- return 0;
- }
- static int run_custom_scripts_tests(struct GameModeConfig *config)
- {
- int scriptstatus = 0;
- long timeout = config_get_script_timeout(config);
-
- char startscripts[CONFIG_LIST_MAX][CONFIG_VALUE_MAX];
- memset(startscripts, 0, sizeof(startscripts));
- config_get_gamemode_start_scripts(config, startscripts);
- if (startscripts[0][0] != '\0') {
- int i = 0;
- while (*startscripts[i] != '\0' && i < CONFIG_LIST_MAX) {
- LOG_MSG(":::: Running start script [%s]\n", startscripts[i]);
- const char *args[] = { "/bin/sh", "-c", startscripts[i], NULL };
- int ret = run_external_process(args, NULL, (int)timeout);
- if (ret == 0)
- LOG_MSG(":::: Passed\n");
- else {
- LOG_MSG(":::: Failed!\n");
- scriptstatus = -1;
- }
- i++;
- }
- }
-
- char endscripts[CONFIG_LIST_MAX][CONFIG_VALUE_MAX];
- memset(endscripts, 0, sizeof(endscripts));
- config_get_gamemode_end_scripts(config, endscripts);
- if (endscripts[0][0] != '\0') {
- int i = 0;
- while (*endscripts[i] != '\0' && i < CONFIG_LIST_MAX) {
- LOG_MSG(":::: Running end script [%s]\n", endscripts[i]);
- const char *args[] = { "/bin/sh", "-c", endscripts[i], NULL };
- int ret = run_external_process(args, NULL, (int)timeout);
- if (ret == 0)
- LOG_MSG(":::: Passed\n");
- else {
- LOG_MSG(":::: Failed!\n");
- scriptstatus = -1;
- }
- i++;
- }
- }
-
- if (endscripts[0][0] == '\0' && startscripts[0][0] == '\0')
- return 1;
- return scriptstatus;
- }
- int run_gpu_optimisation_tests(struct GameModeConfig *config)
- {
- int gpustatus = 0;
-
- char apply[CONFIG_VALUE_MAX];
- config_get_apply_gpu_optimisations(config, apply);
- if (strlen(apply) == 0) {
-
- return 1;
- } else if (strncmp(apply, "accept-responsibility", CONFIG_VALUE_MAX) != 0) {
- LOG_ERROR(
- "apply_gpu_optimisations set to value other than \"accept-responsibility\" (%s), will "
- "not apply GPU optimisations!\n",
- apply);
- return -1;
- }
-
- GameModeGPUInfo *gpuinfo;
- game_mode_initialise_gpu(config, &gpuinfo);
- if (!gpuinfo) {
- LOG_ERROR("Failed to initialise gpuinfo!\n");
- return -1;
- }
-
- long expected_core = gpuinfo->nv_core;
- long expected_mem = gpuinfo->nv_mem;
- long expected_nv_powermizer_mode = gpuinfo->nv_powermizer_mode;
- char expected_amd_performance_level[CONFIG_VALUE_MAX];
- strncpy(expected_amd_performance_level, gpuinfo->amd_performance_level, CONFIG_VALUE_MAX - 1);
- expected_amd_performance_level[CONFIG_VALUE_MAX - 1] = '\0';
-
- game_mode_get_gpu(gpuinfo);
- long original_nv_core = gpuinfo->nv_core;
- long original_nv_mem = gpuinfo->nv_mem;
- long original_nv_powermizer_mode = gpuinfo->nv_powermizer_mode;
- char original_amd_performance_level[CONFIG_VALUE_MAX];
- strncpy(original_amd_performance_level, gpuinfo->amd_performance_level, CONFIG_VALUE_MAX - 1);
- original_amd_performance_level[CONFIG_VALUE_MAX - 1] = '\0';
-
- if (expected_nv_powermizer_mode == -1)
- expected_nv_powermizer_mode = original_nv_powermizer_mode;
- if (expected_core == -1)
- expected_core = original_nv_core;
- if (expected_mem == -1)
- expected_mem = original_nv_mem;
-
- gamemode_request_start();
- if (game_mode_get_gpu(gpuinfo) != 0) {
- LOG_ERROR("Could not get current GPU info, see above!\n");
- gamemode_request_end();
- game_mode_free_gpu(&gpuinfo);
- return -1;
- }
- if (gpuinfo->vendor == Vendor_NVIDIA &&
- (gpuinfo->nv_core != expected_core || gpuinfo->nv_mem != expected_mem ||
- gpuinfo->nv_powermizer_mode != expected_nv_powermizer_mode)) {
- LOG_ERROR(
- "Current Nvidia GPU clocks during gamemode do not match requested values!\n"
- "\tnv_core - expected:%ld was:%ld | nv_mem - expected:%ld was:%ld | nv_powermizer_mode "
- "- expected:%ld was:%ld\n",
- expected_core,
- gpuinfo->nv_core,
- expected_mem,
- gpuinfo->nv_mem,
- expected_nv_powermizer_mode,
- gpuinfo->nv_powermizer_mode);
- gpustatus = -1;
- } else if (gpuinfo->vendor == Vendor_AMD &&
- strcmp(expected_amd_performance_level, gpuinfo->amd_performance_level) != 0) {
- LOG_ERROR(
- "Current AMD GPU performance level during gamemode does not match requested value!\n"
- "\texpected:%s was:%s\n",
- expected_amd_performance_level,
- gpuinfo->amd_performance_level);
- gpustatus = -1;
- }
-
- gamemode_request_end();
- if (game_mode_get_gpu(gpuinfo) != 0) {
- LOG_ERROR("Could not get current GPU info, see above!\n");
- game_mode_free_gpu(&gpuinfo);
- return -1;
- }
- if (gpuinfo->vendor == Vendor_NVIDIA &&
- (gpuinfo->nv_core != original_nv_core || gpuinfo->nv_mem != original_nv_mem ||
- gpuinfo->nv_powermizer_mode != original_nv_powermizer_mode)) {
- LOG_ERROR(
- "Current Nvidia GPU clocks after gamemode do not matcch original values!\n"
- "\tnv_core - original:%ld was:%ld | nv_mem - original:%ld was:%ld | nv_powermizer_mode "
- "- original:%ld was:%ld\n",
- original_nv_core,
- gpuinfo->nv_core,
- original_nv_mem,
- gpuinfo->nv_mem,
- original_nv_powermizer_mode,
- gpuinfo->nv_powermizer_mode);
- gpustatus = -1;
- } else if (gpuinfo->vendor == Vendor_AMD &&
- strcmp(original_amd_performance_level, gpuinfo->amd_performance_level) != 0) {
- LOG_ERROR(
- "Current AMD GPU performance level after gamemode does not match requested value!\n"
- "\texpected:%s was:%s\n",
- original_amd_performance_level,
- gpuinfo->amd_performance_level);
- gpustatus = -1;
- }
- return gpustatus;
- }
- static int game_mode_run_feature_tests(struct GameModeConfig *config)
- {
- int status = 0;
- LOG_MSG(":: Feature tests\n");
-
-
- {
- LOG_MSG("::: Verifying CPU governor setting\n");
- int cpustatus = run_cpu_governor_tests(config);
- if (cpustatus == 0)
- LOG_MSG("::: Passed\n");
- else {
- LOG_MSG("::: Failed!\n");
-
- status = 1;
- }
- }
-
- {
- LOG_MSG("::: Verifying Scripts\n");
- int scriptstatus = run_custom_scripts_tests(config);
- if (scriptstatus == 1)
- LOG_MSG("::: Passed (no scripts configured to run)\n");
- else if (scriptstatus == 0)
- LOG_MSG("::: Passed\n");
- else {
- LOG_MSG("::: Failed!\n");
-
- status = 1;
- }
- }
-
- {
- LOG_MSG("::: Verifying GPU Optimisations\n");
- int gpustatus = run_gpu_optimisation_tests(config);
- if (gpustatus == 1)
- LOG_MSG("::: Passed (gpu optimisations not configured to run)\n");
- else if (gpustatus == 0)
- LOG_MSG("::: Passed\n");
- else {
- LOG_MSG("::: Failed!\n");
-
- status = 1;
- }
- }
-
-
-
-
-
-
-
- if (status != -1)
- LOG_MSG(":: Passed%s\n\n", status > 0 ? " (with optional failures)" : "");
- else
- LOG_ERROR(":: Failed!\n");
- return status;
- }
- static int run_supervisor_tests(void)
- {
- int supervisortests = 0;
- int ret = 0;
- LOG_MSG(":: Supervisor tests\n");
-
- pid_t pid = fork();
- if (pid == 0) {
-
- pause();
- exit(EXIT_SUCCESS);
- }
-
- ret = gamemode_request_start_for(pid);
- if (ret != 0) {
- LOG_ERROR("gamemode_request_start_for gave unexpected value %d, (expected 0)!\n", ret);
- if (ret == -1)
- LOG_ERROR("GameMode error string: %s!\n", gamemode_error_string());
- supervisortests = -1;
- }
-
- ret = gamemode_query_status();
- if (ret != 1) {
- LOG_ERROR(
- "gamemode_query_status after start request gave unexpected value %d, (expected 1)!\n",
- ret);
- if (ret == -1)
- LOG_ERROR("GameMode error string: %s!\n", gamemode_error_string());
- supervisortests = -1;
- }
-
- ret = gamemode_query_status_for(pid);
- if (ret != 2) {
- LOG_ERROR(
- "gamemode_query_status_for after start request gave unexpected value %d, (expected "
- "2)!\n",
- ret);
- if (ret == -1)
- LOG_ERROR("GameMode error string: %s!\n", gamemode_error_string());
- supervisortests = -1;
- }
-
- ret = gamemode_request_end_for(pid);
- if (ret != 0) {
- LOG_ERROR("gamemode_request_end_for gave unexpected value %d, (expected 0)!\n", ret);
- if (ret == -1)
- LOG_ERROR("GameMode error string: %s!\n", gamemode_error_string());
- supervisortests = -1;
- }
-
- ret = gamemode_query_status();
- if (ret != 0) {
- LOG_ERROR(
- "gamemode_query_status after end request gave unexpected value %d, (expected 0)!\n",
- ret);
- if (ret == -1)
- LOG_ERROR("GameMode error string: %s!\n", gamemode_error_string());
- supervisortests = -1;
- }
-
- if (kill(pid, SIGUSR1) == -1) {
- LOG_ERROR("failed to send continue signal to other child process: %s\n", strerror(errno));
- supervisortests = -1;
- }
-
- int wstatus;
- usleep(100000);
- while (waitpid(pid, &wstatus, WNOHANG) == 0) {
- LOG_MSG("...Waiting for child to quit...\n");
- usleep(100000);
- }
- if (supervisortests == 0)
- LOG_MSG(":: Passed\n\n");
- else
- LOG_ERROR(":: Failed!\n");
- return supervisortests;
- }
- int game_mode_run_client_tests(void)
- {
- int status = 0;
- LOG_MSG(": Loading config\n");
-
-
- GameModeConfig *config = config_create();
- config_init(config);
- LOG_MSG(": Running tests\n\n");
-
- if (verify_gamemode_initial(config) != 0)
- return -1;
-
-
- if (config_get_require_supervisor(config) != 0) {
- LOG_ERROR("Tests currently unsupported when require_supervisor is set\n");
- return -1;
- }
-
-
- if (run_basic_client_tests() != 0)
- status = -1;
-
- if (run_dual_client_tests() != 0)
- status = -1;
-
- if (run_gamemoderun_and_reaper_tests(config) != 0)
- status = -1;
-
- if (run_supervisor_tests() != 0)
- status = -1;
- if (status != 0) {
- LOG_MSG(": Client tests failed, skipping feature tests\n");
- } else {
-
- status = game_mode_run_feature_tests(config);
- }
- if (status >= 0)
- LOG_MSG(": All Tests Passed%s!\n", status > 0 ? " (with optional failures)" : "");
- else
- LOG_MSG(": Tests Failed!\n");
- return status;
- }
|