summaryrefslogtreecommitdiffstats
path: root/rtkit-test.c
blob: 59244f8890d6d476d091f78efe553af66a08e9bf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
/*-*- Mode: C; c-basic-offset: 8 -*-*/

#include <errno.h>
#include <string.h>
#include <sched.h>
#include <stdio.h>
#include <sys/time.h>
#include <sys/resource.h>

#include "rtkit.h"

#ifndef SCHED_RESET_ON_FORK
#define SCHED_RESET_ON_FORK 0x40000000
#endif

static void print_status(const char *t) {
    int ret;

    if ((ret = sched_getscheduler(0)) < 0) {
        fprintf(stderr, "sched_getscheduler() failed: %s\n", strerror(errno));
        return;
    }

    printf("%s:\n"
           "\tSCHED_RESET_ON_FORK: %s\n",
           t,
           (ret & SCHED_RESET_ON_FORK) ? "yes" : "no");

    if ((ret & ~SCHED_RESET_ON_FORK) == SCHED_RR) {
        struct sched_param param;

        if (sched_getparam(0, &param) < 0) {
            fprintf(stderr, "sched_getschedparam() failed: %s\n", strerror(errno));
            return;
        }

        printf("\tSCHED_RR with priority %i\n", param.sched_priority);

    } else if ((ret & ~SCHED_RESET_ON_FORK) == SCHED_OTHER) {
        errno = 0;
        ret = getpriority(PRIO_PROCESS, 0);
        if (errno != 0) {
            fprintf(stderr, "getpriority() failed: %s\n", strerror(errno));
            return;
        }

        printf("\tSCHED_OTHER with nice level: %i\n", ret);

    } else
        fprintf(stderr, "Neither SCHED_RR nor SCHED_OTHER.\n");
}

int main(int argc, char *argv[]) {
    DBusError error;
    DBusConnection *bus;
    int r;

    dbus_error_init(&error);

    if (!(bus = dbus_bus_get(DBUS_BUS_SYSTEM, &error))) {
        fprintf(stderr, "Failed to connect to system bus: %s\n", error.message);
        return 1;
    }

    print_status("before");

    if ((r = rtkit_make_high_priority(bus, 0, -10)) < 0)
        fprintf(stderr, "Failed to become high priority: %s\n", strerror(-r));
    else
        printf("Sucessfully became high priority.\n");

    print_status("after high priority");

    if ((r = rtkit_make_realtime(bus, 0, 10)) < 0)
        fprintf(stderr, "Failed to become realtime: %s\n", strerror(-r));
    else
        printf("Sucessfully became realtime.\n");

    print_status("after realtime");

    dbus_connection_unref(bus);

    return 0;
}