summaryrefslogtreecommitdiffstats
path: root/third_party/rust/dbus/examples/rtkit.rs
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
commit36d22d82aa202bb199967e9512281e9a53db42c9 (patch)
tree105e8c98ddea1c1e4784a60a5a6410fa416be2de /third_party/rust/dbus/examples/rtkit.rs
parentInitial commit. (diff)
downloadfirefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz
firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip
Adding upstream version 115.7.0esr.upstream/115.7.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/rust/dbus/examples/rtkit.rs')
-rw-r--r--third_party/rust/dbus/examples/rtkit.rs67
1 files changed, 67 insertions, 0 deletions
diff --git a/third_party/rust/dbus/examples/rtkit.rs b/third_party/rust/dbus/examples/rtkit.rs
new file mode 100644
index 0000000000..2717d417b9
--- /dev/null
+++ b/third_party/rust/dbus/examples/rtkit.rs
@@ -0,0 +1,67 @@
+/* This example asks the rtkit service to make our thread realtime priority.
+ Rtkit puts a few limitations on us to let us become realtime, such as setting
+ RLIMIT_RTTIME correctly, hence the syscalls. */
+
+extern crate dbus;
+extern crate libc;
+
+use std::cmp;
+
+use dbus::{Connection, BusType, Props, MessageItem, Message};
+
+fn item_as_i64(i: MessageItem) -> Result<i64, Box<std::error::Error>> {
+ match i {
+ MessageItem::Int32(i) => Ok(i as i64),
+ MessageItem::Int64(i) => Ok(i),
+ _ => Err(Box::from(&*format!("Property is not integer ({:?})", i)))
+ }
+}
+
+fn rtkit_set_realtime(c: &Connection, thread: u64, prio: u32) -> Result<(), ::dbus::Error> {
+ let mut m = Message::new_method_call("org.freedesktop.RealtimeKit1", "/org/freedesktop/RealtimeKit1",
+ "org.freedesktop.RealtimeKit1", "MakeThreadRealtime").unwrap();
+ m.append_items(&[thread.into(), prio.into()]);
+ let mut r = try!(c.send_with_reply_and_block(m, 10000));
+ r.as_result().map(|_| ())
+}
+
+fn make_realtime(prio: u32) -> Result<u32, Box<std::error::Error>> {
+ let c = try!(Connection::get_private(BusType::System));
+
+ let p = Props::new(&c, "org.freedesktop.RealtimeKit1", "/org/freedesktop/RealtimeKit1",
+ "org.freedesktop.RealtimeKit1", 10000);
+
+ // Make sure we don't fail by wanting too much
+ let max_prio = try!(item_as_i64(try!(p.get("MaxRealtimePriority")))) as u32;
+ let prio = cmp::min(prio, max_prio);
+
+ // Enforce RLIMIT_RTPRIO, also a must before asking rtkit for rtprio
+ let max_rttime = try!(item_as_i64(try!(p.get("RTTimeUSecMax")))) as u64;
+ let new_limit = libc::rlimit64 { rlim_cur: max_rttime, rlim_max: max_rttime };
+ let mut old_limit = new_limit;
+ if unsafe { libc::getrlimit64(libc::RLIMIT_RTTIME, &mut old_limit) } < 0 {
+ return Err(Box::from("getrlimit failed"));
+ }
+ if unsafe { libc::setrlimit64(libc::RLIMIT_RTTIME, &new_limit) } < 0 {
+ return Err(Box::from("setrlimit failed"));
+ }
+
+ // Finally, let's ask rtkit to make us realtime
+ let thread_id = unsafe { libc::syscall(libc::SYS_gettid) };
+ let r = rtkit_set_realtime(&c, thread_id as u64, prio);
+
+ if r.is_err() {
+ unsafe { libc::setrlimit64(libc::RLIMIT_RTTIME, &old_limit) };
+ }
+
+ try!(r);
+ Ok(prio)
+}
+
+
+fn main() {
+ match make_realtime(5) {
+ Ok(n) => println!("Got rtprio, level {}", n),
+ Err(e) => println!("No rtprio: {}", e),
+ }
+}