123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314 |
- /*-*- Mode: C; c-basic-offset: 8 -*-*/
- /***
- Copyright 2009 Lennart Poettering
- Copyright 2010 David Henningsson <[email protected]>
- Permission is hereby granted, free of charge, to any person
- obtaining a copy of this software and associated documentation files
- (the "Software"), to deal in the Software without restriction,
- including without limitation the rights to use, copy, modify, merge,
- publish, distribute, sublicense, and/or sell copies of the Software,
- and to permit persons to whom the Software is furnished to do so,
- subject to the following conditions:
- The above copyright notice and this permission notice shall be
- included in all copies or substantial portions of the Software.
- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
- EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
- MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
- NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
- BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
- ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
- CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
- SOFTWARE.
- ***/
- #ifdef __linux__ /* OpenMPT */
- #ifndef _GNU_SOURCE /* OpenMPT */
- #define _GNU_SOURCE /* OpenMPT */
- #endif /* OpenMPT */
- #endif /* OpenMPT */
- #include <errno.h>
- #include "rtkit.h"
- #ifdef __linux__
- #ifndef _GNU_SOURCE
- #define _GNU_SOURCE
- #endif
- #include <string.h>
- #include <unistd.h>
- #include <sys/types.h>
- #include <sys/syscall.h>
- static pid_t _gettid(void) {
- return (pid_t) syscall(SYS_gettid);
- }
- static int translate_error(const char *name) {
- if (strcmp(name, DBUS_ERROR_NO_MEMORY) == 0)
- return -ENOMEM;
- if (strcmp(name, DBUS_ERROR_SERVICE_UNKNOWN) == 0 ||
- strcmp(name, DBUS_ERROR_NAME_HAS_NO_OWNER) == 0)
- return -ENOENT;
- if (strcmp(name, DBUS_ERROR_ACCESS_DENIED) == 0 ||
- strcmp(name, DBUS_ERROR_AUTH_FAILED) == 0)
- return -EACCES;
- return -EIO;
- }
- static long long rtkit_get_int_property(DBusConnection *connection, const char* propname, long long* propval) {
- DBusMessage *m = NULL, *r = NULL;
- DBusMessageIter iter, subiter;
- dbus_int64_t i64;
- dbus_int32_t i32;
- DBusError error;
- int current_type;
- long long ret;
- const char * interfacestr = "org.freedesktop.RealtimeKit1";
- dbus_error_init(&error);
- if (!(m = dbus_message_new_method_call(
- RTKIT_SERVICE_NAME,
- RTKIT_OBJECT_PATH,
- "org.freedesktop.DBus.Properties",
- "Get"))) {
- ret = -ENOMEM;
- goto finish;
- }
- if (!dbus_message_append_args(
- m,
- DBUS_TYPE_STRING, &interfacestr,
- DBUS_TYPE_STRING, &propname,
- DBUS_TYPE_INVALID)) {
- ret = -ENOMEM;
- goto finish;
- }
- if (!(r = dbus_connection_send_with_reply_and_block(connection, m, -1, &error))) {
- ret = translate_error(error.name);
- goto finish;
- }
- if (dbus_set_error_from_message(&error, r)) {
- ret = translate_error(error.name);
- goto finish;
- }
- ret = -EBADMSG;
- dbus_message_iter_init(r, &iter);
- while ((current_type = dbus_message_iter_get_arg_type (&iter)) != DBUS_TYPE_INVALID) {
- if (current_type == DBUS_TYPE_VARIANT) {
- dbus_message_iter_recurse(&iter, &subiter);
- while ((current_type = dbus_message_iter_get_arg_type (&subiter)) != DBUS_TYPE_INVALID) {
- if (current_type == DBUS_TYPE_INT32) {
- dbus_message_iter_get_basic(&subiter, &i32);
- *propval = i32;
- ret = 0;
- }
- if (current_type == DBUS_TYPE_INT64) {
- dbus_message_iter_get_basic(&subiter, &i64);
- *propval = i64;
- ret = 0;
- }
- dbus_message_iter_next (&subiter);
- }
- }
- dbus_message_iter_next (&iter);
- }
- finish:
- if (m)
- dbus_message_unref(m);
- if (r)
- dbus_message_unref(r);
- dbus_error_free(&error);
- return ret;
- }
- int rtkit_get_max_realtime_priority(DBusConnection *connection) {
- long long retval;
- int err;
- err = rtkit_get_int_property(connection, "MaxRealtimePriority", &retval);
- return err < 0 ? err : retval;
- }
- int rtkit_get_min_nice_level(DBusConnection *connection, int* min_nice_level) {
- long long retval;
- int err;
- err = rtkit_get_int_property(connection, "MinNiceLevel", &retval);
- if (err >= 0)
- *min_nice_level = retval;
- return err;
- }
- long long rtkit_get_rttime_usec_max(DBusConnection *connection) {
- long long retval;
- int err;
- err = rtkit_get_int_property(connection, "RTTimeUSecMax", &retval);
- return err < 0 ? err : retval;
- }
- int rtkit_make_realtime(DBusConnection *connection, pid_t thread, int priority) {
- DBusMessage *m = NULL, *r = NULL;
- dbus_uint64_t u64;
- dbus_uint32_t u32;
- DBusError error;
- int ret;
- dbus_error_init(&error);
- if (thread == 0)
- thread = _gettid();
- if (!(m = dbus_message_new_method_call(
- RTKIT_SERVICE_NAME,
- RTKIT_OBJECT_PATH,
- "org.freedesktop.RealtimeKit1",
- "MakeThreadRealtime"))) {
- ret = -ENOMEM;
- goto finish;
- }
- u64 = (dbus_uint64_t) thread;
- u32 = (dbus_uint32_t) priority;
- if (!dbus_message_append_args(
- m,
- DBUS_TYPE_UINT64, &u64,
- DBUS_TYPE_UINT32, &u32,
- DBUS_TYPE_INVALID)) {
- ret = -ENOMEM;
- goto finish;
- }
- if (!(r = dbus_connection_send_with_reply_and_block(connection, m, -1, &error))) {
- ret = translate_error(error.name);
- goto finish;
- }
- if (dbus_set_error_from_message(&error, r)) {
- ret = translate_error(error.name);
- goto finish;
- }
- ret = 0;
- finish:
- if (m)
- dbus_message_unref(m);
- if (r)
- dbus_message_unref(r);
- dbus_error_free(&error);
- return ret;
- }
- int rtkit_make_high_priority(DBusConnection *connection, pid_t thread, int nice_level) {
- DBusMessage *m = NULL, *r = NULL;
- dbus_uint64_t u64;
- dbus_int32_t s32;
- DBusError error;
- int ret;
- dbus_error_init(&error);
- if (thread == 0)
- thread = _gettid();
- if (!(m = dbus_message_new_method_call(
- RTKIT_SERVICE_NAME,
- RTKIT_OBJECT_PATH,
- "org.freedesktop.RealtimeKit1",
- "MakeThreadHighPriority"))) {
- ret = -ENOMEM;
- goto finish;
- }
- u64 = (dbus_uint64_t) thread;
- s32 = (dbus_int32_t) nice_level;
- if (!dbus_message_append_args(
- m,
- DBUS_TYPE_UINT64, &u64,
- DBUS_TYPE_INT32, &s32,
- DBUS_TYPE_INVALID)) {
- ret = -ENOMEM;
- goto finish;
- }
- if (!(r = dbus_connection_send_with_reply_and_block(connection, m, -1, &error))) {
- ret = translate_error(error.name);
- goto finish;
- }
- if (dbus_set_error_from_message(&error, r)) {
- ret = translate_error(error.name);
- goto finish;
- }
- ret = 0;
- finish:
- if (m)
- dbus_message_unref(m);
- if (r)
- dbus_message_unref(r);
- dbus_error_free(&error);
- return ret;
- }
- #else
- int rtkit_make_realtime(DBusConnection *connection, pid_t thread, int priority) {
- return -ENOTSUP;
- }
- int rtkit_make_high_priority(DBusConnection *connection, pid_t thread, int nice_level) {
- return -ENOTSUP;
- }
- int rtkit_get_max_realtime_priority(DBusConnection *connection) {
- return -ENOTSUP;
- }
- int rtkit_get_min_nice_level(DBusConnection *connection, int* min_nice_level) {
- return -ENOTSUP;
- }
- long long rtkit_get_rttime_usec_max(DBusConnection *connection) {
- return -ENOTSUP;
- }
- #endif
|