libsession-rs 0.1.3

session management daemon
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

#![allow(clippy::unwrap_used)]
#![allow(clippy::panic)]
#![allow(clippy::expect_used)]
#![allow(clippy::cast_possible_wrap)]
use super::*;
use std::{env, process};

#[test]
fn connects_to_correct_backend() {
    let context = Context::connect().expect("Failed to connect to any backend");
    let backend = context
        .get_backend()
        .expect("Failed to get current backend");

    if let Ok(env_backend) = env::var("LIBSESSION_BACKEND") {
        match env_backend.as_str() {
            "sessiond" => {
                assert_eq!(backend, libsession_sys::LibsessionBackend::Sessiond);
            }
            "logind" => {
                assert_eq!(backend, libsession_sys::LibsessionBackend::Logind);
            }
            "consolekit" => {
                assert_eq!(backend, libsession_sys::LibsessionBackend::ConsoleKit);
            }
            _ => panic!("unsupported LIBSESSION_BACKEND value: {env_backend:?}"),
        }
    } else {
        let running = |name: &str| {
            process::Command::new("pidof")
                .arg(name)
                .output()
                .expect("failed to execute pidof")
                .status
                .success()
        };

        let expected = if running("sessiond") {
            libsession_sys::LibsessionBackend::Sessiond
        } else if running("systemd-logind") {
            libsession_sys::LibsessionBackend::Logind
        } else if running("console-kit-daemon") {
            libsession_sys::LibsessionBackend::ConsoleKit
        } else {
            panic!("no supported backend daemon is running");
        };

        assert_eq!(
            backend, expected,
            "selected backend does not match the running daemon"
        );
    }
}

#[test]
fn list_sessions() {
    let context = Context::connect().expect("no session backend available");

    let current_id = context
        .get_session()
        .expect("current process has no session")
        .id()
        .expect("failed to read current session ID");

    let ids: Vec<_> = context
        .list_sessions()
        .expect("failed to list sessions")
        .iter()
        .map(|session| session.id().expect("failed to read session ID"))
        .collect();

    assert!(
        ids.contains(&current_id),
        "current session {current_id:?} missing from session list: {ids:?}"
    );
}

#[test]
fn is_local() {
    let context = Context::connect().expect("no session backend available");
    let session = context
        .get_session()
        .expect("current process has no session");

    let over_ssh = ["SSH_CONNECTION", "SSH_CLIENT", "SSH_TTY"]
        .into_iter()
        .any(|variable| env::var_os(variable).is_some());

    let is_local = context
        .is_session_local(&session)
        .expect("failed to check session locality");

    assert_ne!(is_local, over_ssh);
}

#[test]
fn get_owner_uid_for_pid() {
    let context = Context::connect().expect("no session backend available");
    let pid = process::id() as i32;

    let uid = context.get_owner_uid_for_pid(pid).unwrap();
    assert_eq!(uid, 1000);
}

#[test]
fn get_display_session() {
    let context = Context::connect().expect("no session backend available");

    let pid = process::id() as i32;

    let uid = context.get_owner_uid_for_pid(pid).unwrap();
    let session = context.get_display_for_uid(uid).unwrap();

    let ids: Vec<_> = context
        .list_sessions()
        .expect("failed to list sessions")
        .iter()
        .map(|session| session.id().expect("failed to read session ID"))
        .collect();

    assert!(
        ids.contains(&session.id().expect("failed to read display session ID")),
        "current session {:?} missing from session list: {ids:?}",
        session.id()
    );
}