| /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/ |
| |
| /*** |
| This file is part of systemd. |
| |
| Copyright 2012 Lennart Poettering |
| Copyright 2013 Kay Sievers |
| |
| 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 "sd-bus.h" |
| #include "bus-util.h" |
| #include "bus-error.h" |
| #include "bus-message.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" |
| #include "utf8.h" |
| #include "def.h" |
| |
| static bool arg_no_pager = false; |
| static bool arg_ask_password = true; |
| static BusTransport arg_transport = BUS_TRANSPORT_LOCAL; |
| static 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; |
| |
| if (arg_transport != BUS_TRANSPORT_LOCAL) |
| 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 show_status(sd_bus *bus, char **args, unsigned n) { |
| StatusInfo info = {}; |
| static const struct bus_properties_map map[] = { |
| { "VConsoleKeymap", "s", NULL, offsetof(StatusInfo, vconsole_keymap) }, |
| { "VConsoleKeymap", "s", NULL, offsetof(StatusInfo, vconsole_keymap) }, |
| { "VConsoleKeymapToggle", "s", NULL, offsetof(StatusInfo, vconsole_keymap_toggle) }, |
| { "X11Layout", "s", NULL, offsetof(StatusInfo, x11_layout) }, |
| { "X11Model", "s", NULL, offsetof(StatusInfo, x11_model) }, |
| { "X11Variant", "s", NULL, offsetof(StatusInfo, x11_variant) }, |
| { "X11Options", "s", NULL, offsetof(StatusInfo, x11_options) }, |
| { "Locale", "as", NULL, offsetof(StatusInfo, locale) }, |
| {} |
| }; |
| int r; |
| |
| assert(bus); |
| |
| r = bus_map_all_properties(bus, |
| "org.freedesktop.locale1", |
| "/org/freedesktop/locale1", |
| map, |
| &info); |
| if (r < 0) { |
| log_error("Could not get properties: %s", strerror(-r)); |
| goto fail; |
| } |
| |
| print_status_info(&info); |
| |
| fail: |
| strv_free(info.locale); |
| return r; |
| } |
| |
| static int set_locale(sd_bus *bus, char **args, unsigned n) { |
| _cleanup_bus_message_unref_ sd_bus_message *m = NULL; |
| _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL; |
| int r; |
| |
| assert(bus); |
| assert(args); |
| |
| polkit_agent_open_if_enabled(); |
| |
| r = sd_bus_message_new_method_call( |
| bus, |
| &m, |
| "org.freedesktop.locale1", |
| "/org/freedesktop/locale1", |
| "org.freedesktop.locale1", |
| "SetLocale"); |
| if (r < 0) |
| return bus_log_create_error(r); |
| |
| r = sd_bus_message_append_strv(m, args + 1); |
| if (r < 0) |
| return bus_log_create_error(r); |
| |
| r = sd_bus_message_append(m, "b", arg_ask_password); |
| if (r < 0) |
| return bus_log_create_error(r); |
| |
| r = sd_bus_call(bus, m, 0, &error, NULL); |
| if (r < 0) { |
| log_error("Failed to issue method call: %s", bus_error_message(&error, -r)); |
| return r; |
| } |
| |
| return 0; |
| } |
| |
| 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; |
| |
| if (!utf8_is_valid((char*) p + e[i].name_offset)) |
| continue; |
| |
| z = strdup((char*) p + e[i].name_offset); |
| if (!z) { |
| r = log_oom(); |
| goto finish; |
| } |
| |
| r = set_consume(locales, z); |
| if (r < 0) { |
| 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) { |
| _cleanup_closedir_ DIR *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_consume(locales, z); |
| if (r < 0 && 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(sd_bus *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(sd_bus *bus, char **args, unsigned n) { |
| _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL; |
| const char *map, *toggle_map; |
| int r; |
| |
| 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] : ""; |
| |
| r = sd_bus_call_method( |
| bus, |
| "org.freedesktop.locale1", |
| "/org/freedesktop/locale1", |
| "org.freedesktop.locale1", |
| "SetVConsoleKeyboard", |
| &error, |
| NULL, |
| "ssbb", map, toggle_map, arg_convert, arg_ask_password); |
| if (r < 0) |
| log_error("Failed to set keymap: %s", bus_error_message(&error, -r)); |
| |
| return r; |
| } |
| |
| 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(basename(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_consume(keymaps, p); |
| if (r < 0 && r != -EEXIST) { |
| log_error("Can't add keymap: %s", strerror(-r)); |
| return r; |
| } |
| |
| return 0; |
| } |
| |
| static int list_vconsole_keymaps(sd_bus *bus, char **args, unsigned n) { |
| _cleanup_strv_free_ char **l = NULL; |
| const char *dir; |
| |
| keymaps = set_new(string_hash_func, string_compare_func); |
| if (!keymaps) |
| return log_oom(); |
| |
| NULSTR_FOREACH(dir, KBD_KEYMAP_DIRS) |
| nftw(dir, 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(sd_bus *bus, char **args, unsigned n) { |
| _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL; |
| const char *layout, *model, *variant, *options; |
| int r; |
| |
| 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] : ""; |
| |
| r = sd_bus_call_method( |
| bus, |
| "org.freedesktop.locale1", |
| "/org/freedesktop/locale1", |
| "org.freedesktop.locale1", |
| "SetX11Keyboard", |
| &error, |
| NULL, |
| "ssssbb", layout, model, variant, options, |
| arg_convert, arg_ask_password); |
| if (r < 0) |
| log_error("Failed to set keymap: %s", bus_error_message(&error, -r)); |
| |
| return r; |
| } |
| |
| static int list_x11_keymaps(sd_bus *bus, char **args, unsigned n) { |
| _cleanup_fclose_ FILE *f = NULL; |
| _cleanup_strv_free_ char **list = NULL; |
| char line[LINE_MAX]; |
| enum { |
| NONE, |
| MODELS, |
| LAYOUTS, |
| VARIANTS, |
| OPTIONS |
| } state = NONE, look_for; |
| int r; |
| |
| if (n > 2) { |
| log_error("Too many arguments."); |
| return -EINVAL; |
| } |
| |
| f = fopen("/usr/share/X11/xkb/rules/base.lst", "re"); |
| if (!f) { |
| log_error("Failed to open keyboard mapping list. %m"); |
| return -errno; |
| } |
| |
| if (streq(args[0], "list-x11-keymap-models")) |
| look_for = MODELS; |
| else if (streq(args[0], "list-x11-keymap-layouts")) |
| look_for = LAYOUTS; |
| else if (streq(args[0], "list-x11-keymap-variants")) |
| look_for = VARIANTS; |
| else if (streq(args[0], "list-x11-keymap-options")) |
| look_for = OPTIONS; |
| else |
| assert_not_reached("Wrong parameter"); |
| |
| FOREACH_LINE(line, f, break) { |
| char *l, *w; |
| |
| l = strstrip(line); |
| |
| if (isempty(l)) |
| continue; |
| |
| if (l[0] == '!') { |
| if (startswith(l, "! model")) |
| state = MODELS; |
| else if (startswith(l, "! layout")) |
| state = LAYOUTS; |
| else if (startswith(l, "! variant")) |
| state = VARIANTS; |
| else if (startswith(l, "! option")) |
| state = OPTIONS; |
| else |
| state = NONE; |
| |
| continue; |
| } |
| |
| if (state != look_for) |
| continue; |
| |
| w = l + strcspn(l, WHITESPACE); |
| |
| if (n > 1) { |
| char *e; |
| |
| if (*w == 0) |
| continue; |
| |
| *w = 0; |
| w++; |
| w += strspn(w, WHITESPACE); |
| |
| e = strchr(w, ':'); |
| if (!e) |
| continue; |
| |
| *e = 0; |
| |
| if (!streq(w, args[1])) |
| continue; |
| } else |
| *w = 0; |
| |
| r = strv_extend(&list, l); |
| if (r < 0) |
| return log_oom(); |
| } |
| |
| if (strv_isempty(list)) { |
| log_error("Couldn't find any entries."); |
| return -ENOENT; |
| } |
| |
| strv_sort(list); |
| strv_uniq(list); |
| |
| pager_open_if_enabled(); |
| |
| strv_print(list); |
| return 0; |
| } |
| |
| 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-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" |
| " -M --machine=CONTAINER Operate on local container\n" |
| " --no-convert Don't convert keyboard mappings\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" |
| " list-x11-keymap-models Show known X11 keyboard mapping models\n" |
| " list-x11-keymap-layouts Show known X11 keyboard mapping layouts\n" |
| " list-x11-keymap-variants [LAYOUT]\n" |
| " Show known X11 keyboard mapping variants\n" |
| " list-x11-keymap-options Show known X11 keyboard mapping options\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' }, |
| { "machine", required_argument, NULL, 'M' }, |
| { "no-ask-password", no_argument, NULL, ARG_NO_ASK_PASSWORD }, |
| { "no-convert", no_argument, NULL, ARG_NO_CONVERT }, |
| {} |
| }; |
| |
| int c; |
| |
| assert(argc >= 0); |
| assert(argv); |
| |
| while ((c = getopt_long(argc, argv, "hH:M:", options, NULL)) >= 0) { |
| |
| switch (c) { |
| |
| case 'h': |
| return help(); |
| |
| case ARG_VERSION: |
| puts(PACKAGE_STRING); |
| puts(SYSTEMD_FEATURES); |
| return 0; |
| |
| case ARG_NO_CONVERT: |
| arg_convert = false; |
| break; |
| |
| case ARG_NO_PAGER: |
| arg_no_pager = true; |
| break; |
| |
| case ARG_NO_ASK_PASSWORD: |
| arg_ask_password = false; |
| break; |
| |
| case 'H': |
| arg_transport = BUS_TRANSPORT_REMOTE; |
| arg_host = optarg; |
| break; |
| |
| case 'M': |
| arg_transport = BUS_TRANSPORT_CONTAINER; |
| arg_host = optarg; |
| break; |
| |
| case '?': |
| return -EINVAL; |
| |
| default: |
| assert_not_reached("Unhandled option"); |
| } |
| } |
| |
| return 1; |
| } |
| |
| static int localectl_main(sd_bus *bus, int argc, char *argv[]) { |
| |
| static const struct { |
| const char* verb; |
| const enum { |
| MORE, |
| LESS, |
| EQUAL |
| } argc_cmp; |
| const int argc; |
| int (* const dispatch)(sd_bus *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 }, |
| { "list-x11-keymap-models", EQUAL, 1, list_x11_keymaps }, |
| { "list-x11-keymap-layouts", EQUAL, 1, list_x11_keymaps }, |
| { "list-x11-keymap-variants", LESS, 2, list_x11_keymaps }, |
| { "list-x11-keymap-options", EQUAL, 1, list_x11_keymaps }, |
| }; |
| |
| int left; |
| unsigned i; |
| |
| assert(argc >= 0); |
| assert(argv); |
| |
| 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."); |
| } |
| |
| return verbs[i].dispatch(bus, argv + optind, left); |
| } |
| |
| int main(int argc, char*argv[]) { |
| _cleanup_bus_unref_ sd_bus *bus = NULL; |
| int r; |
| |
| setlocale(LC_ALL, ""); |
| log_parse_environment(); |
| log_open(); |
| |
| r = parse_argv(argc, argv); |
| if (r <= 0) |
| goto finish; |
| |
| r = bus_open_transport(arg_transport, arg_host, false, &bus); |
| if (r < 0) { |
| log_error("Failed to create bus connection: %s", strerror(-r)); |
| goto finish; |
| } |
| |
| r = localectl_main(bus, argc, argv); |
| |
| finish: |
| pager_close(); |
| |
| return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS; |
| } |