chiark / gitweb /
sd-rtnl: fix self-reference leaks
[elogind.git] / src / backlight / backlight.c
1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
2
3 /***
4   This file is part of systemd.
5
6   Copyright 2013 Lennart Poettering
7
8   systemd is free software; you can redistribute it and/or modify it
9   under the terms of the GNU Lesser General Public License as published by
10   the Free Software Foundation; either version 2.1 of the License, or
11   (at your option) any later version.
12
13   systemd is distributed in the hope that it will be useful, but
14   WITHOUT ANY WARRANTY; without even the implied warranty of
15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16   Lesser General Public License for more details.
17
18   You should have received a copy of the GNU Lesser General Public License
19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
20 ***/
21
22 #include "util.h"
23 #include "mkdir.h"
24 #include "fileio.h"
25 #include "libudev.h"
26 #include "udev-util.h"
27
28 static struct udev_device *find_pci_or_platform_parent(struct udev_device *device) {
29         struct udev_device *parent;
30         const char *subsystem, *sysname;
31
32         assert(device);
33
34         parent = udev_device_get_parent(device);
35         if (!parent)
36                 return NULL;
37
38         subsystem = udev_device_get_subsystem(parent);
39         if (!subsystem)
40                 return NULL;
41
42         sysname = udev_device_get_sysname(parent);
43         if (!sysname)
44                 return NULL;
45
46         if (streq(subsystem, "drm")) {
47                 const char *c;
48
49                 c = startswith(sysname, "card");
50                 if (!c)
51                         return NULL;
52
53                 c += strspn(c, "0123456789");
54                 if (*c == '-') {
55                         /* A connector DRM device, let's ignore all but LVDS and eDP! */
56
57                         if (!startswith(c, "-LVDS-") &&
58                             !startswith(c, "-Embedded DisplayPort-"))
59                                 return NULL;
60                 }
61
62         } else if (streq(subsystem, "pci")) {
63                 const char *value;
64
65                 value = udev_device_get_sysattr_value(parent, "class");
66                 if (value) {
67                         unsigned long class = 0;
68
69                         if (safe_atolu(value, &class) < 0) {
70                                 log_warning("Cannot parse PCI class %s of device %s:%s.", value, subsystem, sysname);
71                                 return NULL;
72                         }
73
74                         /* Graphics card */
75                         if (class == 0x30000)
76                                 return parent;
77                 }
78
79         } else if (streq(subsystem, "platform"))
80                 return parent;
81
82         return find_pci_or_platform_parent(parent);
83 }
84
85 static bool same_device(struct udev_device *a, struct udev_device *b) {
86         assert(a);
87         assert(b);
88
89         if (!streq_ptr(udev_device_get_subsystem(a), udev_device_get_subsystem(b)))
90                 return false;
91
92         if (!streq_ptr(udev_device_get_sysname(a), udev_device_get_sysname(b)))
93                 return false;
94
95         return true;
96 }
97
98 static bool validate_device(struct udev *udev, struct udev_device *device) {
99         _cleanup_udev_enumerate_unref_ struct udev_enumerate *enumerate = NULL;
100         struct udev_list_entry *item = NULL, *first = NULL;
101         struct udev_device *parent;
102         const char *v, *subsystem;
103         int r;
104
105         assert(udev);
106         assert(device);
107
108         /* Verify whether we should actually care for a specific
109          * backlight device. For backlight devices there might be
110          * multiple ways to access the same control: "firmware"
111          * (i.e. ACPI), "platform" (i.e. via the machine's EC) and
112          * "raw" (via the graphics card). In general we should prefer
113          * "firmware" (i.e. ACPI) or "platform" access over "raw"
114          * access, in order not to confuse the BIOS/EC, and
115          * compatibility with possible low-level hotkey handling of
116          * screen brightness. The kernel will already make sure to
117          * expose only one of "firmware" and "platform" for the same
118          * device to userspace. However, we still need to make sure
119          * that we use "raw" only if no "firmware" or "platform"
120          * device for the same device exists. */
121
122         subsystem = udev_device_get_subsystem(device);
123         if (!streq_ptr(subsystem, "backlight"))
124                 return true;
125
126         v = udev_device_get_sysattr_value(device, "type");
127         if (!streq_ptr(v, "raw"))
128                 return true;
129
130         parent = find_pci_or_platform_parent(device);
131         if (!parent)
132                 return true;
133
134         subsystem = udev_device_get_subsystem(parent);
135         if (!subsystem)
136                 return true;
137
138         enumerate = udev_enumerate_new(udev);
139         if (!enumerate)
140                 return true;
141
142         r = udev_enumerate_add_match_subsystem(enumerate, "backlight");
143         if (r < 0)
144                 return true;
145
146         r = udev_enumerate_scan_devices(enumerate);
147         if (r < 0)
148                 return true;
149
150         first = udev_enumerate_get_list_entry(enumerate);
151         udev_list_entry_foreach(item, first) {
152                 _cleanup_udev_device_unref_ struct udev_device *other;
153                 struct udev_device *other_parent;
154                 const char *other_subsystem;
155
156                 other = udev_device_new_from_syspath(udev, udev_list_entry_get_name(item));
157                 if (!other)
158                         return true;
159
160                 if (same_device(device, other))
161                         continue;
162
163                 v = udev_device_get_sysattr_value(other, "type");
164                 if (!streq_ptr(v, "platform") && !streq_ptr(v, "firmware"))
165                         continue;
166
167                 /* OK, so there's another backlight device, and it's a
168                  * platform or firmware device, so, let's see if we
169                  * can verify it belongs to the same device as
170                  * ours. */
171                 other_parent = find_pci_or_platform_parent(other);
172                 if (!other_parent)
173                         continue;
174
175                 if (same_device(parent, other_parent)) {
176                         /* Both have the same PCI parent, that means
177                          * we are out. */
178                         log_debug("Skipping backlight device %s, since backlight device %s is on same PCI device and, takes precedence.", udev_device_get_sysname(device), udev_device_get_sysname(other));
179                         return false;
180                 }
181
182                 other_subsystem = udev_device_get_subsystem(other_parent);
183                 if (streq_ptr(other_subsystem, "platform") && streq_ptr(subsystem, "pci")) {
184                         /* The other is connected to the platform bus
185                          * and we are a PCI device, that also means we
186                          * are out. */
187                         log_debug("Skipping backlight device %s, since backlight device %s is a platform device and takes precedence.", udev_device_get_sysname(device), udev_device_get_sysname(other));
188                         return false;
189                 }
190         }
191
192         return true;
193 }
194
195 /* Some systems turn the backlight all the way off at the lowest levels.
196  * clamp_brightness clamps the saved brightness to at least 1 or 5% of
197  * max_brightness.  This avoids preserving an unreadably dim screen, which
198  * would otherwise force the user to disable state restoration. */
199 static void clamp_brightness(struct udev_device *device, char **value) {
200         int r;
201         const char *max_brightness_str;
202         unsigned brightness, max_brightness, new_brightness;
203
204         max_brightness_str = udev_device_get_sysattr_value(device, "max_brightness");
205         if (!max_brightness_str) {
206                 log_warning("Failed to read max_brightness attribute; not checking saved brightness");
207                 return;
208         }
209
210         r = safe_atou(*value, &brightness);
211         if (r < 0) {
212                 log_warning("Failed to parse brightness \"%s\": %s", *value, strerror(-r));
213                 return;
214         }
215
216         r = safe_atou(max_brightness_str, &max_brightness);
217         if (r < 0) {
218                 log_warning("Failed to parse max_brightness \"%s\": %s", max_brightness_str, strerror(-r));
219                 return;
220         }
221
222         new_brightness = MAX3(brightness, 1U, max_brightness/20);
223         if (new_brightness != brightness) {
224                 char *old_value = *value;
225
226                 r = asprintf(value, "%u", new_brightness);
227                 if (r < 0) {
228                         log_oom();
229                         return;
230                 }
231
232                 log_debug("Saved brightness %s too low; increasing to %s.", old_value, *value);
233                 free(old_value);
234         }
235 }
236
237 int main(int argc, char *argv[]) {
238         _cleanup_udev_unref_ struct udev *udev = NULL;
239         _cleanup_udev_device_unref_ struct udev_device *device = NULL;
240         _cleanup_free_ char *saved = NULL, *ss = NULL, *escaped_ss = NULL, *escaped_sysname = NULL, *escaped_path_id = NULL;
241         const char *sysname, *path_id;
242         int r;
243
244         if (argc != 3) {
245                 log_error("This program requires two arguments.");
246                 return EXIT_FAILURE;
247         }
248
249         log_set_target(LOG_TARGET_AUTO);
250         log_parse_environment();
251         log_open();
252
253         umask(0022);
254
255         r = mkdir_p("/var/lib/systemd/backlight", 0755);
256         if (r < 0) {
257                 log_error("Failed to create backlight directory: %s", strerror(-r));
258                 return EXIT_FAILURE;
259         }
260
261         udev = udev_new();
262         if (!udev) {
263                 log_oom();
264                 return EXIT_FAILURE;
265         }
266
267         sysname = strchr(argv[2], ':');
268         if (!sysname) {
269                 log_error("Requires pair of subsystem and sysname for specifying backlight device.");
270                 return EXIT_FAILURE;
271         }
272
273         ss = strndup(argv[2], sysname - argv[2]);
274         if (!ss) {
275                 log_oom();
276                 return EXIT_FAILURE;
277         }
278
279         sysname++;
280
281         if (!streq(ss, "backlight") && !streq(ss, "leds")) {
282                 log_error("Not a backlight or LED device: '%s:%s'", ss, sysname);
283                 return EXIT_FAILURE;
284         }
285
286         errno = 0;
287         device = udev_device_new_from_subsystem_sysname(udev, ss, sysname);
288         if (!device) {
289                 if (errno != 0)
290                         log_error("Failed to get backlight or LED device '%s:%s': %m", ss, sysname);
291                 else
292                         log_oom();
293
294                 return EXIT_FAILURE;
295         }
296
297         escaped_ss = cescape(ss);
298         if (!escaped_ss) {
299                 log_oom();
300                 return EXIT_FAILURE;
301         }
302
303         escaped_sysname = cescape(sysname);
304         if (!escaped_sysname) {
305                 log_oom();
306                 return EXIT_FAILURE;
307         }
308
309         path_id = udev_device_get_property_value(device, "ID_PATH");
310         if (path_id) {
311                 escaped_path_id = cescape(path_id);
312                 if (!escaped_path_id) {
313                         log_oom();
314                         return EXIT_FAILURE;
315                 }
316
317                 saved = strjoin("/var/lib/systemd/backlight/", escaped_path_id, ":", escaped_ss, ":", escaped_sysname, NULL);
318         } else
319                 saved = strjoin("/var/lib/systemd/backlight/", escaped_ss, ":", escaped_sysname, NULL);
320
321         if (!saved) {
322                 log_oom();
323                 return EXIT_FAILURE;
324         }
325
326         /* If there are multiple conflicting backlight devices, then
327          * their probing at boot-time might happen in any order. This
328          * means the validity checking of the device then is not
329          * reliable, since it might not see other devices conflicting
330          * with a specific backlight. To deal with this, we will
331          * actively delete backlight state files at shutdown (where
332          * device probing should be complete), so that the validity
333          * check at boot time doesn't have to be reliable. */
334
335         if (streq(argv[1], "load") && shall_restore_state()) {
336                 _cleanup_free_ char *value = NULL;
337
338                 if (!validate_device(udev, device))
339                         return EXIT_SUCCESS;
340
341                 r = read_one_line_file(saved, &value);
342                 if (r < 0) {
343
344                         if (r == -ENOENT)
345                                 return EXIT_SUCCESS;
346
347                         log_error("Failed to read %s: %s", saved, strerror(-r));
348                         return EXIT_FAILURE;
349                 }
350
351                 clamp_brightness(device, &value);
352
353                 r = udev_device_set_sysattr_value(device, "brightness", value);
354                 if (r < 0) {
355                         log_error("Failed to write system attribute: %s", strerror(-r));
356                         return EXIT_FAILURE;
357                 }
358
359         } else if (streq(argv[1], "save")) {
360                 const char *value;
361
362                 if (!validate_device(udev, device)) {
363                         unlink(saved);
364                         return EXIT_SUCCESS;
365                 }
366
367                 value = udev_device_get_sysattr_value(device, "brightness");
368                 if (!value) {
369                         log_error("Failed to read system attribute");
370                         return EXIT_FAILURE;
371                 }
372
373                 r = write_string_file(saved, value);
374                 if (r < 0) {
375                         log_error("Failed to write %s: %s", saved, strerror(-r));
376                         return EXIT_FAILURE;
377                 }
378
379         } else {
380                 log_error("Unknown verb %s.", argv[1]);
381                 return EXIT_FAILURE;
382         }
383
384         return EXIT_SUCCESS;
385 }