| /* SPDX-License-Identifier: GPL-2.0+ */ |
| |
| #include <errno.h> |
| #include <getopt.h> |
| |
| #include "sd-device.h" |
| #include "sd-event.h" |
| |
| #include "device-enumerator-private.h" |
| #include "fd-util.h" |
| #include "fileio.h" |
| #include "path-util.h" |
| #include "process-util.h" |
| #include "set.h" |
| #include "string-util.h" |
| #include "strv.h" |
| #include "udevadm.h" |
| #include "udevadm-util.h" |
| #include "udev-ctrl.h" |
| #include "virt.h" |
| |
| static bool arg_verbose = false; |
| static bool arg_dry_run = false; |
| |
| static int exec_list(sd_device_enumerator *e, const char *action, Set *settle_set) { |
| sd_device *d; |
| int r; |
| |
| FOREACH_DEVICE_AND_SUBSYSTEM(e, d) { |
| _cleanup_free_ char *filename = NULL; |
| const char *syspath; |
| |
| if (sd_device_get_syspath(d, &syspath) < 0) |
| continue; |
| |
| if (arg_verbose) |
| printf("%s\n", syspath); |
| if (arg_dry_run) |
| continue; |
| |
| filename = path_join(syspath, "uevent"); |
| if (!filename) |
| return log_oom(); |
| |
| r = write_string_file(filename, action, WRITE_STRING_FILE_DISABLE_BUFFER); |
| if (r < 0) { |
| log_debug_errno(r, "Failed to write '%s' to '%s', ignoring: %m", action, filename); |
| continue; |
| } |
| |
| if (settle_set) { |
| r = set_put_strdup(settle_set, syspath); |
| if (r < 0) |
| return log_oom(); |
| } |
| } |
| |
| return 0; |
| } |
| |
| static int device_monitor_handler(sd_device_monitor *m, sd_device *dev, void *userdata) { |
| Set *settle_set = userdata; |
| const char *syspath; |
| |
| assert(dev); |
| assert(settle_set); |
| |
| if (sd_device_get_syspath(dev, &syspath) < 0) |
| return 0; |
| |
| if (arg_verbose) |
| printf("settle %s\n", syspath); |
| |
| if (!set_remove(settle_set, syspath)) |
| log_debug("Got epoll event on syspath %s not present in syspath set", syspath); |
| |
| if (set_isempty(settle_set)) |
| return sd_event_exit(sd_device_monitor_get_event(m), 0); |
| |
| return 0; |
| } |
| |
| static char* keyval(const char *str, const char **key, const char **val) { |
| char *buf, *pos; |
| |
| buf = strdup(str); |
| if (!buf) |
| return NULL; |
| |
| pos = strchr(buf, '='); |
| if (pos) { |
| pos[0] = 0; |
| pos++; |
| } |
| |
| *key = buf; |
| *val = pos; |
| |
| return buf; |
| } |
| |
| static int help(void) { |
| printf("%s trigger [OPTIONS] DEVPATH\n\n" |
| "Request events from the kernel.\n\n" |
| " -h --help Show this help\n" |
| " -V --version Show package version\n" |
| " -v --verbose Print the list of devices while running\n" |
| " -n --dry-run Do not actually trigger the events\n" |
| " -t --type= Type of events to trigger\n" |
| " devices sysfs devices (default)\n" |
| " subsystems sysfs subsystems and drivers\n" |
| " -c --action=ACTION Event action value, default is \"change\"\n" |
| " -s --subsystem-match=SUBSYSTEM Trigger devices from a matching subsystem\n" |
| " -S --subsystem-nomatch=SUBSYSTEM Exclude devices from a matching subsystem\n" |
| " -a --attr-match=FILE[=VALUE] Trigger devices with a matching attribute\n" |
| " -A --attr-nomatch=FILE[=VALUE] Exclude devices with a matching attribute\n" |
| " -p --property-match=KEY=VALUE Trigger devices with a matching property\n" |
| " -g --tag-match=KEY=VALUE Trigger devices with a matching property\n" |
| " -y --sysname-match=NAME Trigger devices with this /sys path\n" |
| " --name-match=NAME Trigger devices with this /dev name\n" |
| " -b --parent-match=NAME Trigger devices with that parent device\n" |
| " -w --settle Wait for the triggered events to complete\n" |
| " --wait-daemon[=SECONDS] Wait for udevd daemon to be initialized\n" |
| " before triggering uevents\n" |
| , program_invocation_short_name); |
| |
| return 0; |
| } |
| |
| int trigger_main(int argc, char *argv[], void *userdata) { |
| enum { |
| ARG_NAME = 0x100, |
| ARG_PING, |
| }; |
| |
| static const struct option options[] = { |
| { "verbose", no_argument, NULL, 'v' }, |
| { "dry-run", no_argument, NULL, 'n' }, |
| { "type", required_argument, NULL, 't' }, |
| { "action", required_argument, NULL, 'c' }, |
| { "subsystem-match", required_argument, NULL, 's' }, |
| { "subsystem-nomatch", required_argument, NULL, 'S' }, |
| { "attr-match", required_argument, NULL, 'a' }, |
| { "attr-nomatch", required_argument, NULL, 'A' }, |
| { "property-match", required_argument, NULL, 'p' }, |
| { "tag-match", required_argument, NULL, 'g' }, |
| { "sysname-match", required_argument, NULL, 'y' }, |
| { "name-match", required_argument, NULL, ARG_NAME }, |
| { "parent-match", required_argument, NULL, 'b' }, |
| { "settle", no_argument, NULL, 'w' }, |
| { "wait-daemon", optional_argument, NULL, ARG_PING }, |
| { "version", no_argument, NULL, 'V' }, |
| { "help", no_argument, NULL, 'h' }, |
| {} |
| }; |
| enum { |
| TYPE_DEVICES, |
| TYPE_SUBSYSTEMS, |
| } device_type = TYPE_DEVICES; |
| const char *action = "change"; |
| _cleanup_(sd_device_enumerator_unrefp) sd_device_enumerator *e = NULL; |
| _cleanup_(sd_device_monitor_unrefp) sd_device_monitor *m = NULL; |
| _cleanup_(sd_event_unrefp) sd_event *event = NULL; |
| _cleanup_set_free_free_ Set *settle_set = NULL; |
| usec_t ping_timeout_usec = 5 * USEC_PER_SEC; |
| bool settle = false, ping = false; |
| int c, r; |
| |
| if (running_in_chroot() > 0) { |
| log_info("Running in chroot, ignoring request."); |
| return 0; |
| } |
| |
| r = sd_device_enumerator_new(&e); |
| if (r < 0) |
| return r; |
| |
| r = sd_device_enumerator_allow_uninitialized(e); |
| if (r < 0) |
| return r; |
| |
| while ((c = getopt_long(argc, argv, "vnt:c:s:S:a:A:p:g:y:b:wVh", options, NULL)) >= 0) { |
| _cleanup_free_ char *buf = NULL; |
| const char *key, *val; |
| |
| switch (c) { |
| case 'v': |
| arg_verbose = true; |
| break; |
| case 'n': |
| arg_dry_run = true; |
| break; |
| case 't': |
| if (streq(optarg, "devices")) |
| device_type = TYPE_DEVICES; |
| else if (streq(optarg, "subsystems")) |
| device_type = TYPE_SUBSYSTEMS; |
| else |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), "Unknown type --type=%s", optarg); |
| break; |
| case 'c': |
| if (STR_IN_SET(optarg, "add", "remove", "change")) |
| action = optarg; |
| else |
| log_error_errno(SYNTHETIC_ERRNO(EINVAL), "Unknown action '%s'", optarg); |
| |
| break; |
| case 's': |
| r = sd_device_enumerator_add_match_subsystem(e, optarg, true); |
| if (r < 0) |
| return log_error_errno(r, "Failed to add subsystem match '%s': %m", optarg); |
| break; |
| case 'S': |
| r = sd_device_enumerator_add_match_subsystem(e, optarg, false); |
| if (r < 0) |
| return log_error_errno(r, "Failed to add negative subsystem match '%s': %m", optarg); |
| break; |
| case 'a': |
| buf = keyval(optarg, &key, &val); |
| if (!buf) |
| return log_oom(); |
| r = sd_device_enumerator_add_match_sysattr(e, key, val, true); |
| if (r < 0) |
| return log_error_errno(r, "Failed to add sysattr match '%s=%s': %m", key, val); |
| break; |
| case 'A': |
| buf = keyval(optarg, &key, &val); |
| if (!buf) |
| return log_oom(); |
| r = sd_device_enumerator_add_match_sysattr(e, key, val, false); |
| if (r < 0) |
| return log_error_errno(r, "Failed to add negative sysattr match '%s=%s': %m", key, val); |
| break; |
| case 'p': |
| buf = keyval(optarg, &key, &val); |
| if (!buf) |
| return log_oom(); |
| r = sd_device_enumerator_add_match_property(e, key, val); |
| if (r < 0) |
| return log_error_errno(r, "Failed to add property match '%s=%s': %m", key, val); |
| break; |
| case 'g': |
| r = sd_device_enumerator_add_match_tag(e, optarg); |
| if (r < 0) |
| return log_error_errno(r, "Failed to add tag match '%s': %m", optarg); |
| break; |
| case 'y': |
| r = sd_device_enumerator_add_match_sysname(e, optarg); |
| if (r < 0) |
| return log_error_errno(r, "Failed to add sysname match '%s': %m", optarg); |
| break; |
| case 'b': { |
| _cleanup_(sd_device_unrefp) sd_device *dev = NULL; |
| |
| r = find_device(optarg, "/sys", &dev); |
| if (r < 0) |
| return log_error_errno(r, "Failed to open the device '%s': %m", optarg); |
| |
| r = sd_device_enumerator_add_match_parent(e, dev); |
| if (r < 0) |
| return log_error_errno(r, "Failed to add parent match '%s': %m", optarg); |
| break; |
| } |
| case 'w': |
| settle = true; |
| break; |
| |
| case ARG_NAME: { |
| _cleanup_(sd_device_unrefp) sd_device *dev = NULL; |
| |
| r = find_device(optarg, "/dev/", &dev); |
| if (r < 0) |
| return log_error_errno(r, "Failed to open the device '%s': %m", optarg); |
| |
| r = sd_device_enumerator_add_match_parent(e, dev); |
| if (r < 0) |
| return log_error_errno(r, "Failed to add parent match '%s': %m", optarg); |
| break; |
| } |
| |
| case ARG_PING: { |
| ping = true; |
| if (optarg) { |
| r = parse_sec(optarg, &ping_timeout_usec); |
| if (r < 0) |
| log_error_errno(r, "Failed to parse timeout value '%s', ignoring: %m", optarg); |
| } |
| break; |
| } |
| |
| case 'V': |
| return print_version(); |
| case 'h': |
| return help(); |
| case '?': |
| return -EINVAL; |
| default: |
| assert_not_reached("Unknown option"); |
| } |
| } |
| |
| if (!arg_dry_run || ping) { |
| r = must_be_root(); |
| if (r < 0) |
| return r; |
| } |
| |
| if (ping) { |
| _cleanup_(udev_ctrl_unrefp) struct udev_ctrl *uctrl = NULL; |
| |
| uctrl = udev_ctrl_new(); |
| if (!uctrl) |
| return log_oom(); |
| |
| r = udev_ctrl_send_ping(uctrl, ping_timeout_usec); |
| if (r < 0) |
| return log_error_errno(r, "Failed to connect to udev daemon: %m"); |
| } |
| |
| for (; optind < argc; optind++) { |
| _cleanup_(sd_device_unrefp) sd_device *dev = NULL; |
| |
| r = find_device(argv[optind], NULL, &dev); |
| if (r < 0) |
| return log_error_errno(r, "Failed to open the device '%s': %m", argv[optind]); |
| |
| r = sd_device_enumerator_add_match_parent(e, dev); |
| if (r < 0) |
| return log_error_errno(r, "Failed to add parent match '%s': %m", argv[optind]); |
| } |
| |
| if (settle) { |
| settle_set = set_new(&string_hash_ops); |
| if (!settle_set) |
| return log_oom(); |
| |
| r = sd_event_default(&event); |
| if (r < 0) |
| return log_error_errno(r, "Failed to get default event: %m"); |
| |
| r = sd_device_monitor_new(&m); |
| if (r < 0) |
| return log_error_errno(r, "Failed to create device monitor object: %m"); |
| |
| r = sd_device_monitor_attach_event(m, event); |
| if (r < 0) |
| return log_error_errno(r, "Failed to attach event to device monitor: %m"); |
| |
| r = sd_device_monitor_start(m, device_monitor_handler, settle_set); |
| if (r < 0) |
| return log_error_errno(r, "Failed to start device monitor: %m"); |
| } |
| |
| switch (device_type) { |
| case TYPE_SUBSYSTEMS: |
| r = device_enumerator_scan_subsystems(e); |
| if (r < 0) |
| return log_error_errno(r, "Failed to scan subsystems: %m"); |
| break; |
| case TYPE_DEVICES: |
| r = device_enumerator_scan_devices(e); |
| if (r < 0) |
| return log_error_errno(r, "Failed to scan devices: %m"); |
| break; |
| default: |
| assert_not_reached("Unknown device type"); |
| } |
| r = exec_list(e, action, settle_set); |
| if (r < 0) |
| return r; |
| |
| if (event && !set_isempty(settle_set)) { |
| r = sd_event_loop(event); |
| if (r < 0) |
| return log_error_errno(r, "Event loop failed: %m"); |
| } |
| |
| return 0; |
| } |