1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2012 Lennart Poettering
7 Copyright 2013 Kay Sievers
9 systemd is free software; you can redistribute it and/or modify it
10 under the terms of the GNU Lesser General Public License as published by
11 the Free Software Foundation; either version 2.1 of the License, or
12 (at your option) any later version.
14 systemd is distributed in the hope that it will be useful, but
15 WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 Lesser General Public License for more details.
19 You should have received a copy of the GNU Lesser General Public License
20 along with systemd; If not, see <http://www.gnu.org/licenses/>.
35 #include "bus-error.h"
36 #include "bus-message.h"
38 #include "spawn-polkit-agent.h"
43 #include "path-util.h"
47 static bool arg_no_pager = false;
48 static bool arg_ask_password = true;
49 static BusTransport arg_transport = BUS_TRANSPORT_LOCAL;
50 static char *arg_host = NULL;
51 static bool arg_convert = true;
53 static void pager_open_if_enabled(void) {
61 static void polkit_agent_open_if_enabled(void) {
63 /* Open the polkit agent as a child process if necessary */
64 if (!arg_ask_password)
67 if (arg_transport != BUS_TRANSPORT_LOCAL)
73 typedef struct StatusInfo {
75 const char *vconsole_keymap;
76 const char *vconsole_keymap_toggle;
77 const char *x11_layout;
78 const char *x11_model;
79 const char *x11_variant;
80 const char *x11_options;
83 static void print_status_info(StatusInfo *i) {
86 if (strv_isempty(i->locale))
87 puts(" System Locale: n/a\n");
91 printf(" System Locale: %s\n", i->locale[0]);
92 STRV_FOREACH(j, i->locale + 1)
96 printf(" VC Keymap: %s\n", strna(i->vconsole_keymap));
97 if (!isempty(i->vconsole_keymap_toggle))
98 printf("VC Toggle Keymap: %s\n", i->vconsole_keymap_toggle);
100 printf(" X11 Layout: %s\n", strna(i->x11_layout));
101 if (!isempty(i->x11_model))
102 printf(" X11 Model: %s\n", i->x11_model);
103 if (!isempty(i->x11_variant))
104 printf(" X11 Variant: %s\n", i->x11_variant);
105 if (!isempty(i->x11_options))
106 printf(" X11 Options: %s\n", i->x11_options);
109 static int show_status(sd_bus *bus, char **args, unsigned n) {
110 StatusInfo info = {};
111 static const struct bus_properties_map map[] = {
112 { "VConsoleKeymap", "s", NULL, offsetof(StatusInfo, vconsole_keymap) },
113 { "VConsoleKeymap", "s", NULL, offsetof(StatusInfo, vconsole_keymap) },
114 { "VConsoleKeymapToggle", "s", NULL, offsetof(StatusInfo, vconsole_keymap_toggle) },
115 { "X11Layout", "s", NULL, offsetof(StatusInfo, x11_layout) },
116 { "X11Model", "s", NULL, offsetof(StatusInfo, x11_model) },
117 { "X11Variant", "s", NULL, offsetof(StatusInfo, x11_variant) },
118 { "X11Options", "s", NULL, offsetof(StatusInfo, x11_options) },
119 { "Locale", "as", NULL, offsetof(StatusInfo, locale) },
126 r = bus_map_all_properties(bus,
127 "org.freedesktop.locale1",
128 "/org/freedesktop/locale1",
134 print_status_info(&info);
137 strv_free(info.locale);
141 static int set_locale(sd_bus *bus, char **args, unsigned n) {
142 _cleanup_bus_message_unref_ sd_bus_message *m = NULL;
143 _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
149 polkit_agent_open_if_enabled();
151 r = sd_bus_message_new_method_call(bus,
152 "org.freedesktop.locale1",
153 "/org/freedesktop/locale1",
154 "org.freedesktop.locale1",
157 return bus_log_create_error(r);
159 r = sd_bus_message_append_strv(m, args + 1);
161 return bus_log_create_error(r);
163 r = sd_bus_message_append(m, "b", arg_ask_password);
165 return bus_log_create_error(r);
167 r = sd_bus_call(bus, m, 0, &error, NULL);
169 log_error("Failed to issue method call: %s", bus_error_message(&error, -r));
176 static int add_locales_from_archive(Set *locales) {
177 /* Stolen from glibc... */
183 /* Name hash table. */
184 uint32_t namehash_offset;
185 uint32_t namehash_used;
186 uint32_t namehash_size;
188 uint32_t string_offset;
189 uint32_t string_used;
190 uint32_t string_size;
191 /* Table with locale records. */
192 uint32_t locrectab_offset;
193 uint32_t locrectab_used;
194 uint32_t locrectab_size;
195 /* MD5 sum hash table. */
196 uint32_t sumhash_offset;
197 uint32_t sumhash_used;
198 uint32_t sumhash_size;
202 /* Hash value of the name. */
204 /* Offset of the name in the string table. */
205 uint32_t name_offset;
206 /* Offset of the locale record. */
207 uint32_t locrec_offset;
210 const struct locarhead *h;
211 const struct namehashent *e;
212 const void *p = MAP_FAILED;
213 _cleanup_close_ int fd = -1;
219 fd = open("/usr/lib/locale/locale-archive", O_RDONLY|O_NOCTTY|O_CLOEXEC);
222 log_error("Failed to open locale archive: %m");
227 if (fstat(fd, &st) < 0) {
228 log_error("fstat() failed: %m");
233 if (!S_ISREG(st.st_mode)) {
234 log_error("Archive file is not regular");
239 if (st.st_size < (off_t) sizeof(struct locarhead)) {
240 log_error("Archive has invalid size");
245 p = mmap(NULL, st.st_size, PROT_READ, MAP_SHARED, fd, 0);
246 if (p == MAP_FAILED) {
247 log_error("Failed to map archive: %m");
252 h = (const struct locarhead *) p;
253 if (h->magic != 0xde020109 ||
254 h->namehash_offset + h->namehash_size > st.st_size ||
255 h->string_offset + h->string_size > st.st_size ||
256 h->locrectab_offset + h->locrectab_size > st.st_size ||
257 h->sumhash_offset + h->sumhash_size > st.st_size) {
258 log_error("Invalid archive file.");
263 e = (const struct namehashent*) ((const uint8_t*) p + h->namehash_offset);
264 for (i = 0; i < h->namehash_size; i++) {
267 if (e[i].locrec_offset == 0)
270 if (!utf8_is_valid((char*) p + e[i].name_offset))
273 z = strdup((char*) p + e[i].name_offset);
279 r = set_consume(locales, z);
281 log_error("Failed to add locale: %s", strerror(-r));
290 munmap((void*) p, sz);
295 static int add_locales_from_libdir (Set *locales) {
296 _cleanup_closedir_ DIR *dir;
297 struct dirent *entry;
300 dir = opendir("/usr/lib/locale");
302 log_error("Failed to open locale directory: %m");
307 while ((entry = readdir(dir))) {
310 if (entry->d_type != DT_DIR)
313 if (ignore_file(entry->d_name))
316 z = strdup(entry->d_name);
320 r = set_consume(locales, z);
321 if (r < 0 && r != -EEXIST) {
322 log_error("Failed to add locale: %s", strerror(-r));
330 log_error("Failed to read locale directory: %m");
337 static int list_locales(sd_bus *bus, char **args, unsigned n) {
338 _cleanup_set_free_ Set *locales;
339 _cleanup_strv_free_ char **l = NULL;
342 locales = set_new(string_hash_func, string_compare_func);
346 r = add_locales_from_archive(locales);
347 if (r < 0 && r != -ENOENT)
350 r = add_locales_from_libdir(locales);
354 l = set_get_strv(locales);
360 pager_open_if_enabled();
367 static int set_vconsole_keymap(sd_bus *bus, char **args, unsigned n) {
368 _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
369 const char *map, *toggle_map;
376 log_error("Too many arguments.");
380 polkit_agent_open_if_enabled();
383 toggle_map = n > 2 ? args[2] : "";
385 r = sd_bus_call_method(
387 "org.freedesktop.locale1",
388 "/org/freedesktop/locale1",
389 "org.freedesktop.locale1",
390 "SetVConsoleKeyboard",
393 "ssbb", map, toggle_map, arg_convert, arg_ask_password);
395 log_error("Failed to set keymap: %s", bus_error_message(&error, -r));
400 static Set *keymaps = NULL;
404 const struct stat *sb,
406 struct FTW *ftwbuf) {
414 if (!endswith(fpath, ".map") &&
415 !endswith(fpath, ".map.gz"))
418 p = strdup(basename(fpath));
422 e = endswith(p, ".map");
426 e = endswith(p, ".map.gz");
430 r = set_consume(keymaps, p);
431 if (r < 0 && r != -EEXIST) {
432 log_error("Can't add keymap: %s", strerror(-r));
439 static int list_vconsole_keymaps(sd_bus *bus, char **args, unsigned n) {
440 _cleanup_strv_free_ char **l = NULL;
443 keymaps = set_new(string_hash_func, string_compare_func);
447 NULSTR_FOREACH(dir, KBD_KEYMAP_DIRS)
448 nftw(dir, nftw_cb, 20, FTW_MOUNT|FTW_PHYS);
450 l = set_get_strv(keymaps);
452 set_free_free(keymaps);
458 if (strv_isempty(l)) {
459 log_error("Couldn't find any console keymaps.");
465 pager_open_if_enabled();
472 static int set_x11_keymap(sd_bus *bus, char **args, unsigned n) {
473 _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
474 const char *layout, *model, *variant, *options;
481 log_error("Too many arguments.");
485 polkit_agent_open_if_enabled();
488 model = n > 2 ? args[2] : "";
489 variant = n > 3 ? args[3] : "";
490 options = n > 4 ? args[4] : "";
492 r = sd_bus_call_method(
494 "org.freedesktop.locale1",
495 "/org/freedesktop/locale1",
496 "org.freedesktop.locale1",
500 "ssssbb", layout, model, variant, options,
501 arg_convert, arg_ask_password);
503 log_error("Failed to set keymap: %s", bus_error_message(&error, -r));
508 static int list_x11_keymaps(sd_bus *bus, char **args, unsigned n) {
509 _cleanup_fclose_ FILE *f = NULL;
510 _cleanup_strv_free_ char **list = NULL;
518 } state = NONE, look_for;
522 log_error("Too many arguments.");
526 f = fopen("/usr/share/X11/xkb/rules/base.lst", "re");
528 log_error("Failed to open keyboard mapping list. %m");
532 if (streq(args[0], "list-x11-keymap-models"))
534 else if (streq(args[0], "list-x11-keymap-layouts"))
536 else if (streq(args[0], "list-x11-keymap-variants"))
538 else if (streq(args[0], "list-x11-keymap-options"))
541 assert_not_reached("Wrong parameter");
543 FOREACH_LINE(line, f, break) {
552 if (startswith(l, "! model"))
554 else if (startswith(l, "! layout"))
556 else if (startswith(l, "! variant"))
558 else if (startswith(l, "! option"))
566 if (state != look_for)
569 w = l + strcspn(l, WHITESPACE);
579 w += strspn(w, WHITESPACE);
587 if (!streq(w, args[1]))
592 r = strv_extend(&list, l);
597 if (strv_isempty(list)) {
598 log_error("Couldn't find any entries.");
605 pager_open_if_enabled();
611 static int help(void) {
613 printf("%s [OPTIONS...] COMMAND ...\n\n"
614 "Query or change system locale and keyboard settings.\n\n"
615 " -h --help Show this help\n"
616 " --version Show package version\n"
617 " --no-pager Do not pipe output into a pager\n"
618 " --no-ask-password Do not prompt for password\n"
619 " -H --host=[USER@]HOST Operate on remote host\n"
620 " -M --machine=CONTAINER Operate on local container\n"
621 " --no-convert Don't convert keyboard mappings\n\n"
623 " status Show current locale settings\n"
624 " set-locale LOCALE... Set system locale\n"
625 " list-locales Show known locales\n"
626 " set-keymap MAP [MAP] Set virtual console keyboard mapping\n"
627 " list-keymaps Show known virtual console keyboard mappings\n"
628 " set-x11-keymap LAYOUT [MODEL] [VARIANT] [OPTIONS]\n"
629 " Set X11 keyboard mapping\n"
630 " list-x11-keymap-models Show known X11 keyboard mapping models\n"
631 " list-x11-keymap-layouts Show known X11 keyboard mapping layouts\n"
632 " list-x11-keymap-variants [LAYOUT]\n"
633 " Show known X11 keyboard mapping variants\n"
634 " list-x11-keymap-options Show known X11 keyboard mapping options\n",
635 program_invocation_short_name);
640 static int parse_argv(int argc, char *argv[]) {
649 static const struct option options[] = {
650 { "help", no_argument, NULL, 'h' },
651 { "version", no_argument, NULL, ARG_VERSION },
652 { "no-pager", no_argument, NULL, ARG_NO_PAGER },
653 { "host", required_argument, NULL, 'H' },
654 { "machine", required_argument, NULL, 'M' },
655 { "no-ask-password", no_argument, NULL, ARG_NO_ASK_PASSWORD },
656 { "no-convert", no_argument, NULL, ARG_NO_CONVERT },
665 while ((c = getopt_long(argc, argv, "hH:M:", options, NULL)) >= 0) {
673 puts(PACKAGE_STRING);
674 puts(SYSTEMD_FEATURES);
685 case ARG_NO_ASK_PASSWORD:
686 arg_ask_password = false;
690 arg_transport = BUS_TRANSPORT_REMOTE;
695 arg_transport = BUS_TRANSPORT_CONTAINER;
703 assert_not_reached("Unhandled option");
710 static int localectl_main(sd_bus *bus, int argc, char *argv[]) {
712 static const struct {
720 int (* const dispatch)(sd_bus *bus, char **args, unsigned n);
722 { "status", LESS, 1, show_status },
723 { "set-locale", MORE, 2, set_locale },
724 { "list-locales", EQUAL, 1, list_locales },
725 { "set-keymap", MORE, 2, set_vconsole_keymap },
726 { "list-keymaps", EQUAL, 1, list_vconsole_keymaps },
727 { "set-x11-keymap", MORE, 2, set_x11_keymap },
728 { "list-x11-keymap-models", EQUAL, 1, list_x11_keymaps },
729 { "list-x11-keymap-layouts", EQUAL, 1, list_x11_keymaps },
730 { "list-x11-keymap-variants", LESS, 2, list_x11_keymaps },
731 { "list-x11-keymap-options", EQUAL, 1, list_x11_keymaps },
740 left = argc - optind;
743 /* Special rule: no arguments means "status" */
746 if (streq(argv[optind], "help")) {
751 for (i = 0; i < ELEMENTSOF(verbs); i++)
752 if (streq(argv[optind], verbs[i].verb))
755 if (i >= ELEMENTSOF(verbs)) {
756 log_error("Unknown operation %s", argv[optind]);
761 switch (verbs[i].argc_cmp) {
764 if (left != verbs[i].argc) {
765 log_error("Invalid number of arguments.");
772 if (left < verbs[i].argc) {
773 log_error("Too few arguments.");
780 if (left > verbs[i].argc) {
781 log_error("Too many arguments.");
788 assert_not_reached("Unknown comparison operator.");
791 return verbs[i].dispatch(bus, argv + optind, left);
794 int main(int argc, char*argv[]) {
795 _cleanup_bus_unref_ sd_bus *bus = NULL;
798 setlocale(LC_ALL, "");
799 log_parse_environment();
802 r = parse_argv(argc, argv);
806 r = bus_open_transport(arg_transport, arg_host, false, &bus);
808 log_error("Failed to create bus connection: %s", strerror(-r));
812 r = localectl_main(bus, argc, argv);
817 return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;