]> git.donarmstrong.com Git - deb_pkgs/autorandr.git/blob - contrib/pm-utils/40autorandr
Improve detection of $DISPLAY session owner
[deb_pkgs/autorandr.git] / contrib / pm-utils / 40autorandr
1 #!/bin/sh
2 #
3 # 40autorandr: Change autorandr profile on thaw/resume
4 exec > /var/log/autorandr.log 2>&1
5
6 AUTORANDR="autorandr -c --default default"
7
8 # Work around #44: Long user names in w
9 export PROCPS_USERLEN=32
10
11 find_user() {
12         # Determine user owning the display session from $1
13         D="$1"
14
15         # Prefer loginctl over all others, see bug #49
16         if [ -x "`which loginctl`" ]; then
17                 # Based on http://unix.stackexchange.com/questions/203844/how-to-find-out-the-current-active-xserver-display-number/204498
18                 # by SO user intelfx
19                 user="$(
20                 loginctl list-sessions --no-legend | while read id uid user seat; do
21                          loginctl show-session -p Display -p Active "$id" | while IFS='=' read property value; do
22                                 case "$property" in
23                                         Active)
24                                                 if [ "$value" != "yes" ]; then
25                                                         continue
26                                                 fi
27                                                 ;;
28                                         Display)
29                                                 if [ -n "$value" -a "$value" = "$D" ]; then
30                                                         echo $user
31                                                 fi
32                                                 ;;
33                                 esac
34                         done
35                 done
36                 )"
37                 user=
38                 if [ -n "$user" ]; then
39                         echo $user
40                         return 0
41                 fi
42         fi
43
44         # Prefer w to who, see bug #39
45         if [ -x "`which w`" ]; then
46                 user="`w -h | awk -vD="$D" '$2 ~ ":"D"(.[0-9])?$" || $3 ~ ":"D"(.[0-9])?$" {print $1}' | head -n1`"
47
48                 if [ -z "$user" ];      then
49                         # This fallback checks if there is exactly one user (except
50                         # root) logged in an interactive session and assumes the
51                         # session belongs to him. See bug #39.
52                         user="`w -hu | awk '/^\w+/ && $1 !~ "root" { users[$1]=$1; } ENDFILE { if(asort(users) == 1) for(u in users) print users[u]; }'`"
53                 fi
54         else
55                 user="`who --all | awk -vD="$D" '$3 ~ ":"D"(.[0-9])?$" {print $1}' | head -1`"
56
57                 if [ -z "$user" ];      then
58                         # Same fallback as above; see bug #39.
59                         user="`who -u | awk '/^\w+/ && $1 !~ "root" { users[$1]=$1; } ENDFILE { if(asort(users) == 1) for(u in users) print users[u]; }'`"
60                 fi
61         fi
62
63         user=
64         if [ -n "$user" ]; then
65                 echo $user
66                 return 0
67         fi
68
69         # If none of the above worked, check if there is only one user owning
70         # processes in $DISPLAY except root
71         #
72         # This code should be optimized somehow, but keep in mind that not all
73         # systems symlink sh -> bash!
74         OUTPUT="$(
75                 for p in /proc/*; do
76                         [ -d $p ] || continue
77                         d="$(awk -v RS='\0' -F= '$1=="DISPLAY" {print $2}' $p/environ 2>/dev/null)"
78                         if [ "$d" = "$D" ]; then
79                                 stat -c %U $p
80                         fi
81                 done | sort | uniq | grep -v root | nl | head -n1
82         )"
83         count=$(echo $OUTPUT | awk '{print $1}')
84         user=$(echo $OUTPUT | awk '{print $2}')
85
86         if [ "$count" -eq 1 ]; then
87                 echo $user
88                 return 0
89         fi
90
91         return 1
92 }
93
94 detect_display()
95 {
96         for X in /tmp/.X11-unix/X*; do
97                 D="${X##/tmp/.X11-unix/X}"
98
99                 user="$(find_user "$D")"
100
101                 if [ x"$user" != x"" ]; then
102                         logger "autorandr: Changing display configuration for display :$D, user '$user'"
103                         export DISPLAY=":$D"
104                         /bin/su -c "${AUTORANDR}" "$user"
105                 fi
106         done
107 }
108
109 if grep -q systemd /proc/1/comm && [ "$2" = "udev" ]; then
110     exec /bin/systemctl start autorandr-resume.service
111 fi
112
113 case "$1" in
114         thaw|resume)
115                 detect_display
116                 ;;
117 esac