|
@@ -31,8 +31,14 @@ POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
|
|
#define _GNU_SOURCE
|
|
|
|
|
|
+#include <common-helpers.h>
|
|
|
+#include <common-pidfds.h>
|
|
|
+
|
|
|
#include <dbus/dbus.h>
|
|
|
+#include <errno.h>
|
|
|
#include <stdio.h>
|
|
|
+#include <stdlib.h>
|
|
|
+#include <string.h>
|
|
|
#include <sys/stat.h>
|
|
|
#include <sys/types.h>
|
|
|
#include <unistd.h>
|
|
@@ -54,6 +60,7 @@ POSSIBILITY OF SUCH DAMAGE.
|
|
|
#define _cleanup_bus_ _cleanup_(hop_off_the_bus)
|
|
|
#define _cleanup_msg_ _cleanup_(cleanup_msg)
|
|
|
#define _cleanup_dpc_ _cleanup_(cleanup_pending_call)
|
|
|
+#define _cleanup_fds_ _cleanup_(cleanup_fd_array)
|
|
|
|
|
|
#ifdef NDEBUG
|
|
|
#define DEBUG(...)
|
|
@@ -73,15 +80,49 @@ static int log_error(const char *fmt, ...) __attribute__((format(printf, 1, 2)))
|
|
|
// Storage for error strings
|
|
|
static char error_string[512] = { 0 };
|
|
|
|
|
|
+// memory helpers
|
|
|
+static void cleanup_fd_array(int **fdlist)
|
|
|
+{
|
|
|
+ if (fdlist == NULL || *fdlist == NULL)
|
|
|
+ return;
|
|
|
+
|
|
|
+ int errsave = errno;
|
|
|
+ for (int *fd = *fdlist; *fd != -1; fd++) {
|
|
|
+ TRACE("GM Closing fd %d\n", *fd);
|
|
|
+ (void)close(*fd);
|
|
|
+ }
|
|
|
+
|
|
|
+ errno = errsave;
|
|
|
+ free(*fdlist);
|
|
|
+}
|
|
|
+
|
|
|
+// Allocate a -1 termianted array of ints
|
|
|
+static inline int *alloc_fd_array(int n)
|
|
|
+{
|
|
|
+ int *fds;
|
|
|
+
|
|
|
+ size_t count = (size_t)n + 1; /* -1, terminated */
|
|
|
+ fds = (int *)malloc(sizeof(int) * count);
|
|
|
+ for (size_t i = 0; i < count; i++)
|
|
|
+ fds[i] = -1;
|
|
|
+
|
|
|
+ return fds;
|
|
|
+}
|
|
|
+
|
|
|
// Helper to check if we are running inside a flatpak
|
|
|
static int in_flatpak(void)
|
|
|
{
|
|
|
- struct stat sb;
|
|
|
- int r;
|
|
|
+ static int status = -1;
|
|
|
|
|
|
- r = lstat("/.flatpak-info", &sb);
|
|
|
+ if (status == -1) {
|
|
|
+ struct stat sb;
|
|
|
+ int r;
|
|
|
|
|
|
- return r == 0 && sb.st_size > 0;
|
|
|
+ r = lstat("/.flatpak-info", &sb);
|
|
|
+ status = r == 0 && sb.st_size > 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ return status;
|
|
|
}
|
|
|
|
|
|
static int log_error(const char *fmt, ...)
|
|
@@ -131,7 +172,7 @@ static DBusConnection *hop_on_the_bus(void)
|
|
|
/* cleanup functions */
|
|
|
static void cleanup_msg(DBusMessage **msg)
|
|
|
{
|
|
|
- if (msg == NULL)
|
|
|
+ if (msg == NULL || *msg == NULL)
|
|
|
return;
|
|
|
|
|
|
dbus_message_unref(*msg);
|
|
@@ -139,37 +180,53 @@ static void cleanup_msg(DBusMessage **msg)
|
|
|
|
|
|
static void cleanup_pending_call(DBusPendingCall **call)
|
|
|
{
|
|
|
- if (call == NULL)
|
|
|
+ if (call == NULL || *call == NULL)
|
|
|
return;
|
|
|
|
|
|
dbus_pending_call_unref(*call);
|
|
|
}
|
|
|
|
|
|
/* internal API */
|
|
|
-static int gamemode_request(const char *method, pid_t for_pid)
|
|
|
+static int make_request(DBusConnection *bus, int native, int use_pidfds, const char *method,
|
|
|
+ pid_t *pids, int npids, DBusError *error)
|
|
|
{
|
|
|
- _cleanup_bus_ DBusConnection *bus = NULL;
|
|
|
_cleanup_msg_ DBusMessage *msg = NULL;
|
|
|
_cleanup_dpc_ DBusPendingCall *call = NULL;
|
|
|
- DBusMessageIter iter;
|
|
|
+ _cleanup_fds_ int *fds = NULL;
|
|
|
+ char action[256] = {
|
|
|
+ 0,
|
|
|
+ };
|
|
|
DBusError err;
|
|
|
- dbus_int32_t pid;
|
|
|
- int native;
|
|
|
+ DBusMessageIter iter;
|
|
|
int res = -1;
|
|
|
|
|
|
- native = !in_flatpak();
|
|
|
- pid = (dbus_int32_t)getpid();
|
|
|
-
|
|
|
- TRACE("GM: [%d] request '%s' received (for pid: %d) [portal: %s]\n",
|
|
|
- (int)pid,
|
|
|
+ TRACE("GM: Incoming request: %s, npids: %d, native: %d pifds: %d\n",
|
|
|
method,
|
|
|
- (int)for_pid,
|
|
|
- (native ? "n" : "y"));
|
|
|
-
|
|
|
- bus = hop_on_the_bus();
|
|
|
+ npids,
|
|
|
+ native,
|
|
|
+ use_pidfds);
|
|
|
+
|
|
|
+ if (use_pidfds) {
|
|
|
+ fds = alloc_fd_array(npids);
|
|
|
+
|
|
|
+ res = open_pidfds(pids, fds, npids);
|
|
|
+ if (res != npids) {
|
|
|
+ dbus_set_error(error, DBUS_ERROR_FAILED, "Could not open pidfd for %d", (int)pids[res]);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (strstr(method, "ByPID"))
|
|
|
+ snprintf(action, sizeof(action), "%sFd", method);
|
|
|
+ else
|
|
|
+ snprintf(action, sizeof(action), "%sByPIDFd", method);
|
|
|
+ method = action;
|
|
|
+ }
|
|
|
|
|
|
- if (bus == NULL)
|
|
|
- return -1;
|
|
|
+ TRACE("GM: Making request: %s, npids: %d, native: %d pifds: %d\n",
|
|
|
+ method,
|
|
|
+ npids,
|
|
|
+ native,
|
|
|
+ use_pidfds);
|
|
|
|
|
|
// If we are inside a flatpak we need to talk to the portal instead
|
|
|
const char *dest = native ? DAEMON_DBUS_NAME : PORTAL_DBUS_NAME;
|
|
@@ -178,15 +235,25 @@ static int gamemode_request(const char *method, pid_t for_pid)
|
|
|
|
|
|
msg = dbus_message_new_method_call(dest, path, iface, method);
|
|
|
|
|
|
- if (!msg)
|
|
|
- return log_error("Could not create dbus message");
|
|
|
+ if (!msg) {
|
|
|
+ dbus_set_error_const(error, DBUS_ERROR_FAILED, "Could not create dbus message");
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
dbus_message_iter_init_append(msg, &iter);
|
|
|
- dbus_message_iter_append_basic(&iter, DBUS_TYPE_INT32, &pid);
|
|
|
|
|
|
- if (for_pid != 0) {
|
|
|
- dbus_int32_t p = (dbus_int32_t)for_pid;
|
|
|
- dbus_message_iter_append_basic(&iter, DBUS_TYPE_INT32, &p);
|
|
|
+ for (int i = 0; i < npids; i++) {
|
|
|
+ dbus_int32_t p;
|
|
|
+ int type;
|
|
|
+
|
|
|
+ if (use_pidfds) {
|
|
|
+ type = DBUS_TYPE_UNIX_FD;
|
|
|
+ p = (dbus_int32_t)fds[i];
|
|
|
+ } else {
|
|
|
+ type = DBUS_TYPE_INT32;
|
|
|
+ p = (dbus_int32_t)pids[i];
|
|
|
+ }
|
|
|
+ dbus_message_iter_append_basic(&iter, type, &p);
|
|
|
}
|
|
|
|
|
|
dbus_connection_send_with_reply(bus, msg, &call, -1);
|
|
@@ -197,20 +264,87 @@ static int gamemode_request(const char *method, pid_t for_pid)
|
|
|
dbus_pending_call_block(call);
|
|
|
msg = dbus_pending_call_steal_reply(call);
|
|
|
|
|
|
- if (msg == NULL)
|
|
|
- return log_error("Did not receive a reply");
|
|
|
+ if (msg == NULL) {
|
|
|
+ dbus_set_error_const(error, DBUS_ERROR_FAILED, "Did not receive a reply");
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
dbus_error_init(&err);
|
|
|
+ res = -1;
|
|
|
+ if (dbus_set_error_from_message(&err, msg)) {
|
|
|
+ dbus_set_error(error,
|
|
|
+ err.name,
|
|
|
+ "Could not call method '%s' on '%s': %s",
|
|
|
+ method,
|
|
|
+ dest,
|
|
|
+ err.message);
|
|
|
+ } else if (!dbus_message_iter_init(msg, &iter) ||
|
|
|
+ dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_INT32) {
|
|
|
+ dbus_set_error(error, DBUS_ERROR_INVALID_SIGNATURE, "Failed to parse response");
|
|
|
+ } else {
|
|
|
+ dbus_message_iter_get_basic(&iter, &res);
|
|
|
+ }
|
|
|
+
|
|
|
+ /* free the local error */
|
|
|
+ if (dbus_error_is_set(&err))
|
|
|
+ dbus_error_free(&err);
|
|
|
+
|
|
|
+ return res;
|
|
|
+}
|
|
|
+
|
|
|
+static int gamemode_request(const char *method, pid_t for_pid)
|
|
|
+{
|
|
|
+ _cleanup_bus_ DBusConnection *bus = NULL;
|
|
|
+ static int use_pidfs = 1;
|
|
|
+ DBusError err;
|
|
|
+ pid_t pids[2];
|
|
|
+ int npids;
|
|
|
+ int native;
|
|
|
+ int res = -1;
|
|
|
+
|
|
|
+ native = !in_flatpak();
|
|
|
+
|
|
|
+ /* pid[0] is the client, i.e. the game
|
|
|
+ * pid[1] is the requestor, i.e. this process
|
|
|
+ *
|
|
|
+ * we setup the array such that pids[1] will always be a valid
|
|
|
+ * pid, because if we are going to use the pidfd based API,
|
|
|
+ * both pids are being sent, even if they are the same
|
|
|
+ */
|
|
|
+ pids[1] = getpid();
|
|
|
+ pids[0] = for_pid != 0 ? for_pid : pids[1];
|
|
|
+
|
|
|
+ TRACE("GM: [%d] request '%s' received (by: %d) [portal: %s]\n",
|
|
|
+ (int)pids[0],
|
|
|
+ method,
|
|
|
+ (int)pids[1],
|
|
|
+ (native ? "n" : "y"));
|
|
|
+
|
|
|
+ bus = hop_on_the_bus();
|
|
|
+
|
|
|
+ if (bus == NULL)
|
|
|
+ return -1;
|
|
|
|
|
|
- if (dbus_set_error_from_message(&err, msg))
|
|
|
- log_error("Could not call method '%s' on '%s': %s", method, dest, err.message);
|
|
|
- else if (!dbus_message_iter_init(msg, &iter) ||
|
|
|
- dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_INT32)
|
|
|
- log_error("Failed to parse response");
|
|
|
+ dbus_error_init(&err);
|
|
|
+retry:
|
|
|
+ if (for_pid != 0 || use_pidfs)
|
|
|
+ npids = 2;
|
|
|
else
|
|
|
- dbus_message_iter_get_basic(&iter, &res);
|
|
|
+ npids = 1;
|
|
|
+
|
|
|
+ res = make_request(bus, native, use_pidfs, method, pids, npids, &err);
|
|
|
+
|
|
|
+ if (res == -1 && use_pidfs && dbus_error_is_set(&err)) {
|
|
|
+ TRACE("GM: Request with pidfds failed (%s). Retrying.\n", err.message);
|
|
|
+ use_pidfs = 0;
|
|
|
+ dbus_error_free(&err);
|
|
|
+ goto retry;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (res == -1 && dbus_error_is_set(&err))
|
|
|
+ log_error("D-Bus error: %s", err.message);
|
|
|
|
|
|
- TRACE("GM: [%d] request '%s' done: %d\n", (int)pid, method, res);
|
|
|
+ TRACE("GM: [%d] request '%s' done: %d\n", (int)pids[0], method, res);
|
|
|
|
|
|
if (dbus_error_is_set(&err))
|
|
|
dbus_error_free(&err);
|