when thread == 0 is passed, fill in gettid()

1 files changed, 15 insertions, 0 deletions

diff --git a/rtkit.c b/rtkit.c
index 67a44f7..893edde 100644
--- a/rtkit.c
+++ b/rtkit.c
@@ -24,11 +24,20 @@
   SOFTWARE.
 ***/
 
+#define _GNU_SOURCE
+
 #include <string.h>
 #include <errno.h>
+#include <unistd.h>
+#include <sys/types.h>
+#include <sys/syscall.h>
 
 #include "rtkit.h"
 
+static pid_t _gettid(void) {
+        return syscall(SYS_gettid);
+}
+
 static int translate_error(const char *name) {
         if (strcmp(name, DBUS_ERROR_NO_MEMORY) == 0)
                 return -ENOMEM;
@@ -51,6 +60,9 @@ int rtkit_make_realtime(DBusConnection *connection, pid_t thread, int priority)
 
         dbus_error_init(&error);
 
+        if (thread == 0)
+                thread = _gettid();
+
         if (!(m = dbus_message_new_method_call(
                               RTKIT_SERVICE_NAME,
                               RTKIT_OBJECT_PATH,
@@ -105,6 +117,9 @@ int rtkit_make_high_priority(DBusConnection *connection, pid_t thread, int nice_
 
         dbus_error_init(&error);
 
+        if (thread == 0)
+                thread = _gettid();
+
         if (!(m = dbus_message_new_method_call(
                               RTKIT_SERVICE_NAME,
                               RTKIT_OBJECT_PATH,