src/rfkill/rfkill.c

	rfkill.c revision 4844262f25a3ff6bd23de05a0a6f84a8e2983d74
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering/***
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  This file is part of systemd.
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  Copyright 2013 Lennart Poettering
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  systemd is free software; you can redistribute it and/or modify it
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  under the terms of the GNU Lesser General Public License as published by
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  the Free Software Foundation; either version 2.1 of the License, or
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  (at your option) any later version.
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  systemd is distributed in the hope that it will be useful, but
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  WITHOUT ANY WARRANTY; without even the implied warranty of
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  Lesser General Public License for more details.
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  You should have received a copy of the GNU Lesser General Public License
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering  along with systemd; If not, see <http://www.gnu.org/licenses/>.
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering***/
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering#include "util.h"
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering#include "mkdir.h"
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering#include "fileio.h"
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering#include "libudev.h"
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering#include "udev-util.h"
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poetteringint main(int argc, char *argv[]) {
34829a324b1ffc6cb8405223329a9c55cd8de0eeLennart Poettering        _cleanup_udev_unref_ struct udev *udev = NULL;
        _cleanup_udev_device_unref_ struct udev_device *device = NULL;
        _cleanup_free_ char *saved = NULL, *escaped_type = NULL, *escaped_path_id = NULL;
        const char *name, *type, *path_id;
        int r;

        if (argc != 3) {
                log_error("This program requires two arguments.");
                return EXIT_FAILURE;
        }

        log_set_target(LOG_TARGET_AUTO);
        log_parse_environment();
        log_open();

        umask(0022);

        r = mkdir_p("/var/lib/systemd/rfkill", 0755);
        if (r < 0) {
                log_error_errno(r, "Failed to create rfkill directory: %m");
                return EXIT_FAILURE;
        }

        udev = udev_new();
        if (!udev) {
                log_oom();
                return EXIT_FAILURE;
        }

        device = udev_device_new_from_subsystem_sysname(udev, "rfkill", argv[2]);
        if (!device) {
                log_debug_errno(errno, "Failed to get rfkill device '%s', ignoring: %m", argv[2]);
                return EXIT_SUCCESS;
        }

        name = udev_device_get_sysattr_value(device, "name");
        if (!name) {
                log_error("rfkill device has no name? Ignoring device.");
                return EXIT_SUCCESS;
        }

        log_debug("Operating on rfkill device '%s'.", name);

        type = udev_device_get_sysattr_value(device, "type");
        if (!type) {
                log_error("rfkill device has no type? Ignoring device.");
                return EXIT_SUCCESS;
        }

        escaped_type = cescape(type);
        if (!escaped_type) {
                log_oom();
                return EXIT_FAILURE;
        }

        path_id = udev_device_get_property_value(device, "ID_PATH");
        if (path_id) {
                escaped_path_id = cescape(path_id);
                if (!escaped_path_id) {
                        log_oom();
                        return EXIT_FAILURE;
                }

                saved = strjoin("/var/lib/systemd/rfkill/", escaped_path_id, ":", escaped_type, NULL);
        } else
                saved = strjoin("/var/lib/systemd/rfkill/", escaped_type, NULL);

        if (!saved) {
                log_oom();
                return EXIT_FAILURE;
        }

        if (streq(argv[1], "load")) {
                _cleanup_free_ char *value = NULL;

                if (!shall_restore_state())
                        return EXIT_SUCCESS;

                r = read_one_line_file(saved, &value);
                if (r == -ENOENT)
                        return EXIT_SUCCESS;
                if (r < 0) {
                        log_error_errno(r, "Failed to read %s: %m", saved);
                        return EXIT_FAILURE;
                }

                r = udev_device_set_sysattr_value(device, "soft", value);
                if (r < 0) {
                        log_debug_errno(r, "Failed to write 'soft' attribute on rfkill device, ignoring: %m");
                        return EXIT_SUCCESS;
                }

        } else if (streq(argv[1], "save")) {
                const char *value;

                value = udev_device_get_sysattr_value(device, "soft");
                if (!value) {
                        log_debug_errno(r, "Failed to read system attribute, ignoring device: %m");
                        return EXIT_SUCCESS;
                }

                r = write_string_file(saved, value);
                if (r < 0) {
                        log_error_errno(r, "Failed to write %s: %m", saved);
                        return EXIT_FAILURE;
                }

        } else {
                log_error("Unknown verb %s.", argv[1]);
                return EXIT_FAILURE;
        }

        return EXIT_SUCCESS;
}