123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336 |
- #define _GNU_SOURCE
- #include "dbus_messaging.h"
- #include "daemonize.h"
- #include "gamemode.h"
- #include "logging.h"
- #include <stdlib.h>
- #include <systemd/sd-bus.h>
- #include <systemd/sd-daemon.h>
- static sd_bus *bus = NULL;
- static sd_bus_slot *slot = NULL;
- static void clean_up(void)
- {
- if (slot) {
- sd_bus_slot_unref(slot);
- }
- slot = NULL;
- if (bus) {
- sd_bus_unref(bus);
- }
- bus = NULL;
- }
- static int method_register_game(sd_bus_message *m, void *userdata,
- __attribute__((unused)) sd_bus_error *ret_error)
- {
- int pid = 0;
- GameModeContext *context = userdata;
- int ret = sd_bus_message_read(m, "i", &pid);
- if (ret < 0) {
- LOG_ERROR("Failed to parse input parameters: %s\n", strerror(-ret));
- return ret;
- }
- int status = game_mode_context_register(context, (pid_t)pid, (pid_t)pid);
- return sd_bus_reply_method_return(m, "i", status);
- }
- static int method_unregister_game(sd_bus_message *m, void *userdata,
- __attribute__((unused)) sd_bus_error *ret_error)
- {
- int pid = 0;
- GameModeContext *context = userdata;
- int ret = sd_bus_message_read(m, "i", &pid);
- if (ret < 0) {
- LOG_ERROR("Failed to parse input parameters: %s\n", strerror(-ret));
- return ret;
- }
- int status = game_mode_context_unregister(context, (pid_t)pid, (pid_t)pid);
- return sd_bus_reply_method_return(m, "i", status);
- }
- static int method_query_status(sd_bus_message *m, void *userdata,
- __attribute__((unused)) sd_bus_error *ret_error)
- {
- int pid = 0;
- GameModeContext *context = userdata;
- int ret = sd_bus_message_read(m, "i", &pid);
- if (ret < 0) {
- LOG_ERROR("Failed to parse input parameters: %s\n", strerror(-ret));
- return ret;
- }
- int status = game_mode_context_query_status(context, (pid_t)pid, (pid_t)pid);
- return sd_bus_reply_method_return(m, "i", status);
- }
- static int method_register_game_by_pid(sd_bus_message *m, void *userdata,
- __attribute__((unused)) sd_bus_error *ret_error)
- {
- int callerpid = 0;
- int gamepid = 0;
- GameModeContext *context = userdata;
- int ret = sd_bus_message_read(m, "ii", &callerpid, &gamepid);
- if (ret < 0) {
- LOG_ERROR("Failed to parse input parameters: %s\n", strerror(-ret));
- return ret;
- }
- int reply = game_mode_context_register(context, (pid_t)gamepid, (pid_t)callerpid);
- return sd_bus_reply_method_return(m, "i", reply);
- }
- static int method_unregister_game_by_pid(sd_bus_message *m, void *userdata,
- __attribute__((unused)) sd_bus_error *ret_error)
- {
- int callerpid = 0;
- int gamepid = 0;
- GameModeContext *context = userdata;
- int ret = sd_bus_message_read(m, "ii", &callerpid, &gamepid);
- if (ret < 0) {
- LOG_ERROR("Failed to parse input parameters: %s\n", strerror(-ret));
- return ret;
- }
- int reply = game_mode_context_unregister(context, (pid_t)gamepid, (pid_t)callerpid);
- return sd_bus_reply_method_return(m, "i", reply);
- }
- static int method_query_status_by_pid(sd_bus_message *m, void *userdata,
- __attribute__((unused)) sd_bus_error *ret_error)
- {
- int callerpid = 0;
- int gamepid = 0;
- GameModeContext *context = userdata;
- int ret = sd_bus_message_read(m, "ii", &callerpid, &gamepid);
- if (ret < 0) {
- LOG_ERROR("Failed to parse input parameters: %s\n", strerror(-ret));
- return ret;
- }
- int status = game_mode_context_query_status(context, (pid_t)gamepid, (pid_t)callerpid);
- return sd_bus_reply_method_return(m, "i", status);
- }
- static const sd_bus_vtable gamemode_vtable[] =
- { SD_BUS_VTABLE_START(0),
- SD_BUS_METHOD("RegisterGame", "i", "i", method_register_game, SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_METHOD("UnregisterGame", "i", "i", method_unregister_game, SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_METHOD("QueryStatus", "i", "i", method_query_status, SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_METHOD("RegisterGameByPID", "ii", "i", method_register_game_by_pid,
- SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_METHOD("UnregisterGameByPID", "ii", "i", method_unregister_game_by_pid,
- SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_METHOD("QueryStatusByPID", "ii", "i", method_query_status_by_pid,
- SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_VTABLE_END };
- void game_mode_context_loop(GameModeContext *context)
- {
-
- atexit(clean_up);
- int ret = 0;
-
- ret = sd_bus_open_user(&bus);
- if (ret < 0) {
- FATAL_ERROR("Failed to connect to the bus: %s\n", strerror(-ret));
- }
-
- ret = sd_bus_add_object_vtable(bus,
- &slot,
- "/com/feralinteractive/GameMode",
- "com.feralinteractive.GameMode",
- gamemode_vtable,
- context);
- if (ret < 0) {
- FATAL_ERROR("Failed to install GameMode object: %s\n", strerror(-ret));
- }
-
- ret = sd_bus_request_name(bus, "com.feralinteractive.GameMode", 0);
- if (ret < 0) {
- FATAL_ERROR("Failed to acquire service name: %s\n", strerror(-ret));
- }
- LOG_MSG("Successfully initialised bus with name [%s]...\n", "com.feralinteractive.GameMode");
- sd_notifyf(0, "STATUS=%sGameMode is ready to be activated.%s\n", "\x1B[1;36m", "\x1B[0m");
-
- for (;;) {
- ret = sd_bus_process(bus, NULL);
- if (ret < 0) {
- FATAL_ERROR("Failure when processing the bus: %s\n", strerror(-ret));
- }
-
- if (ret > 0) {
- continue;
- }
-
- ret = sd_bus_wait(bus, (uint64_t)-1);
- if (ret < 0 && -ret != EINTR) {
- FATAL_ERROR("Failure when waiting on bus: %s\n", strerror(-ret));
- }
- }
- }
- static unsigned int screensaver_inhibit_cookie = 0;
- int game_mode_inhibit_screensaver(bool inhibit)
- {
- const char *service = "org.freedesktop.ScreenSaver";
- const char *object_path = "/org/freedesktop/ScreenSaver";
- const char *interface = "org.freedesktop.ScreenSaver";
- const char *function = inhibit ? "Inhibit" : "UnInhibit";
- sd_bus_message *msg = NULL;
- sd_bus *bus_local = NULL;
- sd_bus_error err;
- memset(&err, 0, sizeof(sd_bus_error));
- int result = -1;
-
- int ret = sd_bus_open_user(&bus_local);
- if (ret < 0) {
- LOG_ERROR("Could not connect to user bus: %s\n", strerror(-ret));
- return -1;
- }
- if (inhibit) {
- ret = sd_bus_call_method(bus_local,
- service,
- object_path,
- interface,
- function,
- &err,
- &msg,
- "ss",
- "com.feralinteractive.GameMode",
- "GameMode Activated");
- } else {
- ret = sd_bus_call_method(bus_local,
- service,
- object_path,
- interface,
- function,
- &err,
- &msg,
- "u",
- screensaver_inhibit_cookie);
- }
- if (ret < 0) {
- LOG_ERROR(
- "Could not call %s on %s: %s\n"
- "\t%s\n"
- "\t%s\n",
- function,
- service,
- strerror(-ret),
- err.name,
- err.message);
- } else if (inhibit) {
-
- ret = sd_bus_message_read(msg, "u", &screensaver_inhibit_cookie);
- if (ret < 0) {
- LOG_ERROR("Failure to parse response from %s on %s: %s\n",
- function,
- service,
- strerror(-ret));
- } else {
- result = 0;
- }
- } else {
- result = 0;
- }
- sd_bus_unref(bus_local);
- return result;
- }
|