| /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/ |
| |
| /*** |
| This file is part of systemd. |
| |
| Copyright 2012 Lennart Poettering |
| |
| systemd is free software; you can redistribute it and/or modify it |
| under the terms of the GNU Lesser General Public License as published by |
| the Free Software Foundation; either version 2.1 of the License, or |
| (at your option) any later version. |
| |
| systemd is distributed in the hope that it will be useful, but |
| WITHOUT ANY WARRANTY; without even the implied warranty of |
| MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| Lesser General Public License for more details. |
| |
| You should have received a copy of the GNU Lesser General Public License |
| along with systemd; If not, see <http://www.gnu.org/licenses/>. |
| ***/ |
| |
| #include <locale.h> |
| #include <stdlib.h> |
| #include <stdbool.h> |
| #include <unistd.h> |
| #include <getopt.h> |
| #include <string.h> |
| #include <ftw.h> |
| #include <sys/mman.h> |
| #include <fcntl.h> |
| |
| #include "dbus-common.h" |
| #include "util.h" |
| #include "spawn-polkit-agent.h" |
| #include "build.h" |
| #include "strv.h" |
| #include "pager.h" |
| #include "set.h" |
| #include "path-util.h" |
| |
| static bool arg_no_pager = false; |
| static enum transport { |
| TRANSPORT_NORMAL, |
| TRANSPORT_SSH, |
| TRANSPORT_POLKIT |
| } arg_transport = TRANSPORT_NORMAL; |
| static bool arg_ask_password = true; |
| static const char *arg_host = NULL; |
| static bool arg_convert = true; |
| |
| static void pager_open_if_enabled(void) { |
| |
| if (arg_no_pager) |
| return; |
| |
| pager_open(false); |
| } |
| |
| static void polkit_agent_open_if_enabled(void) { |
| |
| /* Open the polkit agent as a child process if necessary */ |
| |
| if (!arg_ask_password) |
| return; |
| |
| polkit_agent_open(); |
| } |
| |
| typedef struct StatusInfo { |
| char **locale; |
| const char *vconsole_keymap; |
| const char *vconsole_keymap_toggle; |
| const char *x11_layout; |
| const char *x11_model; |
| const char *x11_variant; |
| const char *x11_options; |
| } StatusInfo; |
| |
| static void print_status_info(StatusInfo *i) { |
| assert(i); |
| |
| if (strv_isempty(i->locale)) |
| puts(" System Locale: n/a\n"); |
| else { |
| char **j; |
| |
| printf(" System Locale: %s\n", i->locale[0]); |
| STRV_FOREACH(j, i->locale + 1) |
| printf(" %s\n", *j); |
| } |
| |
| printf(" VC Keymap: %s\n", strna(i->vconsole_keymap)); |
| if (!isempty(i->vconsole_keymap_toggle)) |
| printf("VC Toggle Keymap: %s\n", i->vconsole_keymap_toggle); |
| |
| printf(" X11 Layout: %s\n", strna(i->x11_layout)); |
| if (!isempty(i->x11_model)) |
| printf(" X11 Model: %s\n", i->x11_model); |
| if (!isempty(i->x11_variant)) |
| printf(" X11 Variant: %s\n", i->x11_variant); |
| if (!isempty(i->x11_options)) |
| printf(" X11 Options: %s\n", i->x11_options); |
| } |
| |
| static int status_property(const char *name, DBusMessageIter *iter, StatusInfo *i) { |
| int r; |
| |
| assert(name); |
| assert(iter); |
| |
| switch (dbus_message_iter_get_arg_type(iter)) { |
| |
| case DBUS_TYPE_STRING: { |
| const char *s; |
| |
| dbus_message_iter_get_basic(iter, &s); |
| if (!isempty(s)) { |
| if (streq(name, "VConsoleKeymap")) |
| i->vconsole_keymap = s; |
| else if (streq(name, "VConsoleKeymapToggle")) |
| i->vconsole_keymap_toggle = s; |
| else if (streq(name, "X11Layout")) |
| i->x11_layout = s; |
| else if (streq(name, "X11Model")) |
| i->x11_model = s; |
| else if (streq(name, "X11Variant")) |
| i->x11_variant = s; |
| else if (streq(name, "X11Options")) |
| i->x11_options = s; |
| } |
| break; |
| } |
| |
| case DBUS_TYPE_ARRAY: |
| |
| if (dbus_message_iter_get_element_type(iter) == DBUS_TYPE_STRING) { |
| char **l; |
| |
| r = bus_parse_strv_iter(iter, &l); |
| if (r < 0) |
| return r; |
| |
| if (streq(name, "Locale")) { |
| strv_free(i->locale); |
| i->locale = l; |
| l = NULL; |
| } |
| |
| strv_free(l); |
| } |
| } |
| |
| return 0; |
| } |
| |
| static int show_status(DBusConnection *bus, char **args, unsigned n) { |
| _cleanup_dbus_message_unref_ DBusMessage *reply = NULL; |
| const char *interface = ""; |
| int r; |
| DBusMessageIter iter, sub, sub2, sub3; |
| StatusInfo info; |
| |
| assert(args); |
| |
| r = bus_method_call_with_reply( |
| bus, |
| "org.freedesktop.locale1", |
| "/org/freedesktop/locale1", |
| "org.freedesktop.DBus.Properties", |
| "GetAll", |
| &reply, |
| NULL, |
| DBUS_TYPE_STRING, &interface, |
| DBUS_TYPE_INVALID); |
| if (r < 0) |
| return r; |
| |
| if (!dbus_message_iter_init(reply, &iter) || |
| dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_ARRAY || |
| dbus_message_iter_get_element_type(&iter) != DBUS_TYPE_DICT_ENTRY) { |
| log_error("Failed to parse reply."); |
| return -EIO; |
| } |
| |
| zero(info); |
| dbus_message_iter_recurse(&iter, &sub); |
| |
| while (dbus_message_iter_get_arg_type(&sub) != DBUS_TYPE_INVALID) { |
| const char *name; |
| |
| if (dbus_message_iter_get_arg_type(&sub) != DBUS_TYPE_DICT_ENTRY) { |
| log_error("Failed to parse reply."); |
| return -EIO; |
| } |
| |
| dbus_message_iter_recurse(&sub, &sub2); |
| |
| if (bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &name, true) < 0) { |
| log_error("Failed to parse reply."); |
| return -EIO; |
| } |
| |
| if (dbus_message_iter_get_arg_type(&sub2) != DBUS_TYPE_VARIANT) { |
| log_error("Failed to parse reply."); |
| return -EIO; |
| } |
| |
| dbus_message_iter_recurse(&sub2, &sub3); |
| |
| r = status_property(name, &sub3, &info); |
| if (r < 0) { |
| log_error("Failed to parse reply."); |
| return r; |
| } |
| |
| dbus_message_iter_next(&sub); |
| } |
| |
| print_status_info(&info); |
| strv_free(info.locale); |
| return 0; |
| } |
| |
| static int set_locale(DBusConnection *bus, char **args, unsigned n) { |
| _cleanup_dbus_message_unref_ DBusMessage *m = NULL, *reply = NULL; |
| dbus_bool_t interactive = true; |
| DBusError error; |
| DBusMessageIter iter; |
| int r; |
| |
| assert(bus); |
| assert(args); |
| |
| dbus_error_init(&error); |
| |
| polkit_agent_open_if_enabled(); |
| |
| m = dbus_message_new_method_call( |
| "org.freedesktop.locale1", |
| "/org/freedesktop/locale1", |
| "org.freedesktop.locale1", |
| "SetLocale"); |
| if (!m) |
| return log_oom(); |
| |
| dbus_message_iter_init_append(m, &iter); |
| |
| r = bus_append_strv_iter(&iter, args + 1); |
| if (r < 0) |
| return log_oom(); |
| |
| if (!dbus_message_iter_append_basic(&iter, DBUS_TYPE_BOOLEAN, &interactive)) |
| return log_oom(); |
| |
| reply = dbus_connection_send_with_reply_and_block(bus, m, -1, &error); |
| if (!reply) { |
| log_error("Failed to issue method call: %s", bus_error_message(&error)); |
| r = -EIO; |
| goto finish; |
| } |
| |
| r = 0; |
| |
| finish: |
| dbus_error_free(&error); |
| return r; |
| } |
| |
| static int add_locales_from_archive(Set *locales) { |
| /* Stolen from glibc... */ |
| |
| struct locarhead { |
| uint32_t magic; |
| /* Serial number. */ |
| uint32_t serial; |
| /* Name hash table. */ |
| uint32_t namehash_offset; |
| uint32_t namehash_used; |
| uint32_t namehash_size; |
| /* String table. */ |
| uint32_t string_offset; |
| uint32_t string_used; |
| uint32_t string_size; |
| /* Table with locale records. */ |
| uint32_t locrectab_offset; |
| uint32_t locrectab_used; |
| uint32_t locrectab_size; |
| /* MD5 sum hash table. */ |
| uint32_t sumhash_offset; |
| uint32_t sumhash_used; |
| uint32_t sumhash_size; |
| }; |
| |
| struct namehashent { |
| /* Hash value of the name. */ |
| uint32_t hashval; |
| /* Offset of the name in the string table. */ |
| uint32_t name_offset; |
| /* Offset of the locale record. */ |
| uint32_t locrec_offset; |
| }; |
| |
| const struct locarhead *h; |
| const struct namehashent *e; |
| const void *p = MAP_FAILED; |
| _cleanup_close_ int fd = -1; |
| size_t sz = 0; |
| struct stat st; |
| unsigned i; |
| int r; |
| |
| fd = open("/usr/lib/locale/locale-archive", O_RDONLY|O_NOCTTY|O_CLOEXEC); |
| if (fd < 0) { |
| if (errno != ENOENT) |
| log_error("Failed to open locale archive: %m"); |
| r = -errno; |
| goto finish; |
| } |
| |
| if (fstat(fd, &st) < 0) { |
| log_error("fstat() failed: %m"); |
| r = -errno; |
| goto finish; |
| } |
| |
| if (!S_ISREG(st.st_mode)) { |
| log_error("Archive file is not regular"); |
| r = -EBADMSG; |
| goto finish; |
| } |
| |
| if (st.st_size < (off_t) sizeof(struct locarhead)) { |
| log_error("Archive has invalid size"); |
| r = -EBADMSG; |
| goto finish; |
| } |
| |
| p = mmap(NULL, st.st_size, PROT_READ, MAP_SHARED, fd, 0); |
| if (p == MAP_FAILED) { |
| log_error("Failed to map archive: %m"); |
| r = -errno; |
| goto finish; |
| } |
| |
| h = (const struct locarhead *) p; |
| if (h->magic != 0xde020109 || |
| h->namehash_offset + h->namehash_size > st.st_size || |
| h->string_offset + h->string_size > st.st_size || |
| h->locrectab_offset + h->locrectab_size > st.st_size || |
| h->sumhash_offset + h->sumhash_size > st.st_size) { |
| log_error("Invalid archive file."); |
| r = -EBADMSG; |
| goto finish; |
| } |
| |
| e = (const struct namehashent*) ((const uint8_t*) p + h->namehash_offset); |
| for (i = 0; i < h->namehash_size; i++) { |
| char *z; |
| |
| if (e[i].locrec_offset == 0) |
| continue; |
| |
| z = strdup((char*) p + e[i].name_offset); |
| if (!z) { |
| r = log_oom(); |
| goto finish; |
| } |
| |
| r = set_put(locales, z); |
| if (r < 0) { |
| free(z); |
| log_error("Failed to add locale: %s", strerror(-r)); |
| goto finish; |
| } |
| } |
| |
| r = 0; |
| |
| finish: |
| if (p != MAP_FAILED) |
| munmap((void*) p, sz); |
| |
| return r; |
| } |
| |
| static int add_locales_from_libdir (Set *locales) { |
| DIR _cleanup_closedir_ *dir; |
| struct dirent *entry; |
| int r; |
| |
| dir = opendir("/usr/lib/locale"); |
| if (!dir) { |
| log_error("Failed to open locale directory: %m"); |
| return -errno; |
| } |
| |
| errno = 0; |
| while ((entry = readdir(dir))) { |
| char *z; |
| |
| if (entry->d_type != DT_DIR) |
| continue; |
| |
| if (ignore_file(entry->d_name)) |
| continue; |
| |
| z = strdup(entry->d_name); |
| if (!z) |
| return log_oom(); |
| |
| r = set_put(locales, z); |
| if (r < 0) { |
| free(z); |
| |
| if (r != -EEXIST) { |
| log_error("Failed to add locale: %s", strerror(-r)); |
| return r; |
| } |
| } |
| |
| errno = 0; |
| } |
| |
| if (errno != 0) { |
| log_error("Failed to read locale directory: %m"); |
| return -errno; |
| } |
| |
| return 0; |
| } |
| |
| static int list_locales(DBusConnection *bus, char **args, unsigned n) { |
| _cleanup_set_free_ Set *locales; |
| _cleanup_strv_free_ char **l = NULL; |
| int r; |
| |
| locales = set_new(string_hash_func, string_compare_func); |
| if (!locales) |
| return log_oom(); |
| |
| r = add_locales_from_archive(locales); |
| if (r < 0 && r != -ENOENT) |
| return r; |
| |
| r = add_locales_from_libdir(locales); |
| if (r < 0) |
| return r; |
| |
| l = set_get_strv(locales); |
| if (!l) |
| return log_oom(); |
| |
| strv_sort(l); |
| |
| pager_open_if_enabled(); |
| |
| strv_print(l); |
| |
| return 0; |
| } |
| |
| static int set_vconsole_keymap(DBusConnection *bus, char **args, unsigned n) { |
| _cleanup_dbus_message_unref_ DBusMessage *reply = NULL; |
| dbus_bool_t interactive = true, b; |
| const char *map, *toggle_map; |
| |
| assert(bus); |
| assert(args); |
| |
| if (n > 3) { |
| log_error("Too many arguments."); |
| return -EINVAL; |
| } |
| |
| polkit_agent_open_if_enabled(); |
| |
| map = args[1]; |
| toggle_map = n > 2 ? args[2] : ""; |
| b = arg_convert; |
| |
| return bus_method_call_with_reply( |
| bus, |
| "org.freedesktop.locale1", |
| "/org/freedesktop/locale1", |
| "org.freedesktop.locale1", |
| "SetVConsoleKeyboard", |
| &reply, |
| NULL, |
| DBUS_TYPE_STRING, &map, |
| DBUS_TYPE_STRING, &toggle_map, |
| DBUS_TYPE_BOOLEAN, &b, |
| DBUS_TYPE_BOOLEAN, &interactive, |
| DBUS_TYPE_INVALID); |
| } |
| |
| static Set *keymaps = NULL; |
| |
| static int nftw_cb( |
| const char *fpath, |
| const struct stat *sb, |
| int tflag, |
| struct FTW *ftwbuf) { |
| |
| char *p, *e; |
| int r; |
| |
| if (tflag != FTW_F) |
| return 0; |
| |
| if (!endswith(fpath, ".map") && |
| !endswith(fpath, ".map.gz")) |
| return 0; |
| |
| p = strdup(path_get_file_name(fpath)); |
| if (!p) |
| return log_oom(); |
| |
| e = endswith(p, ".map"); |
| if (e) |
| *e = 0; |
| |
| e = endswith(p, ".map.gz"); |
| if (e) |
| *e = 0; |
| |
| r = set_put(keymaps, p); |
| if (r == -EEXIST) |
| free(p); |
| else if (r < 0) { |
| log_error("Can't add keymap: %s", strerror(-r)); |
| free(p); |
| return r; |
| } |
| |
| return 0; |
| } |
| |
| static int list_vconsole_keymaps(DBusConnection *bus, char **args, unsigned n) { |
| char _cleanup_strv_free_ **l = NULL; |
| |
| keymaps = set_new(string_hash_func, string_compare_func); |
| if (!keymaps) |
| return log_oom(); |
| |
| nftw("/usr/share/kbd/keymaps/", nftw_cb, 20, FTW_MOUNT|FTW_PHYS); |
| nftw("/usr/lib/kbd/keymaps/", nftw_cb, 20, FTW_MOUNT|FTW_PHYS); |
| nftw("/lib/kbd/keymaps/", nftw_cb, 20, FTW_MOUNT|FTW_PHYS); |
| |
| l = set_get_strv(keymaps); |
| if (!l) { |
| set_free_free(keymaps); |
| return log_oom(); |
| } |
| |
| set_free(keymaps); |
| |
| if (strv_isempty(l)) { |
| log_error("Couldn't find any console keymaps."); |
| return -ENOENT; |
| } |
| |
| strv_sort(l); |
| |
| pager_open_if_enabled(); |
| |
| strv_print(l); |
| |
| return 0; |
| } |
| |
| static int set_x11_keymap(DBusConnection *bus, char **args, unsigned n) { |
| _cleanup_dbus_message_unref_ DBusMessage *reply = NULL; |
| dbus_bool_t interactive = true, b; |
| const char *layout, *model, *variant, *options; |
| |
| assert(bus); |
| assert(args); |
| |
| if (n > 5) { |
| log_error("Too many arguments."); |
| return -EINVAL; |
| } |
| |
| polkit_agent_open_if_enabled(); |
| |
| layout = args[1]; |
| model = n > 2 ? args[2] : ""; |
| variant = n > 3 ? args[3] : ""; |
| options = n > 4 ? args[4] : ""; |
| b = arg_convert; |
| |
| return bus_method_call_with_reply( |
| bus, |
| "org.freedesktop.locale1", |
| "/org/freedesktop/locale1", |
| "org.freedesktop.locale1", |
| "SetX11Keyboard", |
| &reply, |
| NULL, |
| DBUS_TYPE_STRING, &layout, |
| DBUS_TYPE_STRING, &model, |
| DBUS_TYPE_STRING, &variant, |
| DBUS_TYPE_STRING, &options, |
| DBUS_TYPE_BOOLEAN, &b, |
| DBUS_TYPE_BOOLEAN, &interactive, |
| DBUS_TYPE_INVALID); |
| } |
| |
| static int help(void) { |
| |
| printf("%s [OPTIONS...] COMMAND ...\n\n" |
| "Query or change system locale and keyboard settings.\n\n" |
| " -h --help Show this help\n" |
| " --version Show package version\n" |
| " --no-convert Don't convert keyboard mappings\n" |
| " --no-pager Do not pipe output into a pager\n" |
| " --no-ask-password Do not prompt for password\n" |
| " -H --host=[USER@]HOST Operate on remote host\n\n" |
| "Commands:\n" |
| " status Show current locale settings\n" |
| " set-locale LOCALE... Set system locale\n" |
| " list-locales Show known locales\n" |
| " set-keymap MAP [MAP] Set virtual console keyboard mapping\n" |
| " list-keymaps Show known virtual console keyboard mappings\n" |
| " set-x11-keymap LAYOUT [MODEL] [VARIANT] [OPTIONS]\n" |
| " Set X11 keyboard mapping\n", |
| program_invocation_short_name); |
| |
| return 0; |
| } |
| |
| static int parse_argv(int argc, char *argv[]) { |
| |
| enum { |
| ARG_VERSION = 0x100, |
| ARG_NO_PAGER, |
| ARG_NO_CONVERT, |
| ARG_NO_ASK_PASSWORD |
| }; |
| |
| static const struct option options[] = { |
| { "help", no_argument, NULL, 'h' }, |
| { "version", no_argument, NULL, ARG_VERSION }, |
| { "no-pager", no_argument, NULL, ARG_NO_PAGER }, |
| { "host", required_argument, NULL, 'H' }, |
| { "privileged", no_argument, NULL, 'P' }, |
| { "no-ask-password", no_argument, NULL, ARG_NO_ASK_PASSWORD }, |
| { "no-convert", no_argument, NULL, ARG_NO_CONVERT }, |
| { NULL, 0, NULL, 0 } |
| }; |
| |
| int c; |
| |
| assert(argc >= 0); |
| assert(argv); |
| |
| while ((c = getopt_long(argc, argv, "has:H:P", options, NULL)) >= 0) { |
| |
| switch (c) { |
| |
| case 'h': |
| help(); |
| return 0; |
| |
| case ARG_VERSION: |
| puts(PACKAGE_STRING); |
| puts(SYSTEMD_FEATURES); |
| return 0; |
| |
| case 'P': |
| arg_transport = TRANSPORT_POLKIT; |
| break; |
| |
| case 'H': |
| arg_transport = TRANSPORT_SSH; |
| arg_host = optarg; |
| break; |
| |
| case ARG_NO_CONVERT: |
| arg_convert = false; |
| break; |
| |
| case ARG_NO_PAGER: |
| arg_no_pager = true; |
| break; |
| |
| case '?': |
| return -EINVAL; |
| |
| default: |
| log_error("Unknown option code %c", c); |
| return -EINVAL; |
| } |
| } |
| |
| return 1; |
| } |
| |
| static int localectl_main(DBusConnection *bus, int argc, char *argv[], DBusError *error) { |
| |
| static const struct { |
| const char* verb; |
| const enum { |
| MORE, |
| LESS, |
| EQUAL |
| } argc_cmp; |
| const int argc; |
| int (* const dispatch)(DBusConnection *bus, char **args, unsigned n); |
| } verbs[] = { |
| { "status", LESS, 1, show_status }, |
| { "set-locale", MORE, 2, set_locale }, |
| { "list-locales", EQUAL, 1, list_locales }, |
| { "set-keymap", MORE, 2, set_vconsole_keymap }, |
| { "list-keymaps", EQUAL, 1, list_vconsole_keymaps }, |
| { "set-x11-keymap", MORE, 2, set_x11_keymap }, |
| }; |
| |
| int left; |
| unsigned i; |
| |
| assert(argc >= 0); |
| assert(argv); |
| assert(error); |
| |
| left = argc - optind; |
| |
| if (left <= 0) |
| /* Special rule: no arguments means "status" */ |
| i = 0; |
| else { |
| if (streq(argv[optind], "help")) { |
| help(); |
| return 0; |
| } |
| |
| for (i = 0; i < ELEMENTSOF(verbs); i++) |
| if (streq(argv[optind], verbs[i].verb)) |
| break; |
| |
| if (i >= ELEMENTSOF(verbs)) { |
| log_error("Unknown operation %s", argv[optind]); |
| return -EINVAL; |
| } |
| } |
| |
| switch (verbs[i].argc_cmp) { |
| |
| case EQUAL: |
| if (left != verbs[i].argc) { |
| log_error("Invalid number of arguments."); |
| return -EINVAL; |
| } |
| |
| break; |
| |
| case MORE: |
| if (left < verbs[i].argc) { |
| log_error("Too few arguments."); |
| return -EINVAL; |
| } |
| |
| break; |
| |
| case LESS: |
| if (left > verbs[i].argc) { |
| log_error("Too many arguments."); |
| return -EINVAL; |
| } |
| |
| break; |
| |
| default: |
| assert_not_reached("Unknown comparison operator."); |
| } |
| |
| if (!bus) { |
| log_error("Failed to get D-Bus connection: %s", error->message); |
| return -EIO; |
| } |
| |
| return verbs[i].dispatch(bus, argv + optind, left); |
| } |
| |
| int main(int argc, char *argv[]) { |
| int r, retval = EXIT_FAILURE; |
| DBusConnection *bus = NULL; |
| DBusError error; |
| |
| dbus_error_init(&error); |
| |
| setlocale(LC_ALL, ""); |
| log_parse_environment(); |
| log_open(); |
| |
| r = parse_argv(argc, argv); |
| if (r < 0) |
| goto finish; |
| else if (r == 0) { |
| retval = EXIT_SUCCESS; |
| goto finish; |
| } |
| |
| if (arg_transport == TRANSPORT_NORMAL) |
| bus = dbus_bus_get_private(DBUS_BUS_SYSTEM, &error); |
| else if (arg_transport == TRANSPORT_POLKIT) |
| bus_connect_system_polkit(&bus, &error); |
| else if (arg_transport == TRANSPORT_SSH) |
| bus_connect_system_ssh(NULL, arg_host, &bus, &error); |
| else |
| assert_not_reached("Uh, invalid transport..."); |
| |
| r = localectl_main(bus, argc, argv, &error); |
| retval = r < 0 ? EXIT_FAILURE : r; |
| |
| finish: |
| if (bus) { |
| dbus_connection_flush(bus); |
| dbus_connection_close(bus); |
| dbus_connection_unref(bus); |
| } |
| |
| dbus_error_free(&error); |
| dbus_shutdown(); |
| |
| pager_close(); |
| |
| return retval; |
| } |