src/vconsole/vconsole-setup.c

	vconsole-setup.c revision b92bea5d2a9481de69bb627a7b442a9f58fca43d
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun/***
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  This file is part of systemd.
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  Copyright 2010 Kay Sievers
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  systemd is free software; you can redistribute it and/or modify it
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  under the terms of the GNU Lesser General Public License as published by
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  the Free Software Foundation; either version 2.1 of the License, or
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  (at your option) any later version.
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  systemd is distributed in the hope that it will be useful, but
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  WITHOUT ANY WARRANTY; without even the implied warranty of
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  Lesser General Public License for more details.
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  You should have received a copy of the GNU Lesser General Public License
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun  along with systemd; If not, see <http://www.gnu.org/licenses/>.
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun***/
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
b9bf3918f6eaf7747bcbfbd02792bcbe4a052784nilgun#include <stdio.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <unistd.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <stdlib.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <errno.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <string.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <fcntl.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <ctype.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <stdbool.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <stdarg.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <limits.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <sys/ioctl.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <sys/wait.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <linux/tiocl.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <linux/kd.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include <linux/vt.h>
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include "util.h"
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include "log.h"
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include "macro.h"
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include "virt.h"
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun#include "fileio.h"
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgunstatic bool is_vconsole(int fd) {
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        unsigned char data[1];
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        data[0] = TIOCL_GETFGCONSOLE;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        return ioctl(fd, TIOCLINUX, data) >= 0;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun}
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgunstatic int disable_utf8(int fd) {
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        int r = 0, k;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        if (ioctl(fd, KDSKBMODE, K_XLATE) < 0)
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                r = -errno;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        if (loop_write(fd, "\033%@", 3, false) < 0)
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                r = -errno;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        k = write_string_file("/sys/module/vt/parameters/default_utf8", "0");
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        if (k < 0)
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                r = k;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        if (r < 0)
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                log_warning("Failed to disable UTF-8: %s", strerror(-r));
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        return r;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun}
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgunstatic int enable_utf8(int fd) {
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        int r = 0, k;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        long current = 0;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        if (ioctl(fd, KDGKBMODE, &current) < 0 || current == K_XLATE) {
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                /*
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                 * Change the current keyboard to unicode, unless it
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                 * is currently in raw or off mode anyway. We
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                 * shouldn't interfere with X11's processing of the
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                 * key events.
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                 *
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                 * http://lists.freedesktop.org/archives/systemd-devel/2013-February/008573.html
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                 *
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                 */
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                if (ioctl(fd, KDSKBMODE, K_UNICODE) < 0)
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                        r = -errno;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        }
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        if (loop_write(fd, "\033%G", 3, false) < 0)
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                r = -errno;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        k = write_string_file("/sys/module/vt/parameters/default_utf8", "1");
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun        if (k < 0)
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun                r = k;
988d399c4943f830cc91ff3fedfd29f4b6f6bf7fnilgun
        if (r < 0)
                log_warning("Failed to enable UTF-8: %s", strerror(-r));

        return r;
}

static int keymap_load(const char *vc, const char *map, const char *map_toggle, bool utf8, pid_t *_pid) {
        const char *args[8];
        int i = 0;
        pid_t pid;

        if (isempty(map)) {
                /* An empty map means kernel map */
                *_pid = 0;
                return 0;
        }

        args[i++] = KBD_LOADKEYS;
        args[i++] = "-q";
        args[i++] = "-C";
        args[i++] = vc;
        if (utf8)
                args[i++] = "-u";
        args[i++] = map;
        if (map_toggle)
                args[i++] = map_toggle;
        args[i++] = NULL;

        pid = fork();
        if (pid < 0) {
                log_error("Failed to fork: %m");
                return -errno;
        } else if (pid == 0) {
                execv(args[0], (char **) args);
                _exit(EXIT_FAILURE);
        }

        *_pid = pid;
        return 0;
}

static int font_load(const char *vc, const char *font, const char *map, const char *unimap, pid_t *_pid) {
        const char *args[9];
        int i = 0;
        pid_t pid;

        if (isempty(font)) {
                /* An empty font means kernel font */
                *_pid = 0;
                return 0;
        }

        args[i++] = KBD_SETFONT;
        args[i++] = "-C";
        args[i++] = vc;
        args[i++] = font;
        if (map) {
                args[i++] = "-m";
                args[i++] = map;
        }
        if (unimap) {
                args[i++] = "-u";
                args[i++] = unimap;
        }
        args[i++] = NULL;

        pid = fork();
        if (pid < 0) {
                log_error("Failed to fork: %m");
                return -errno;
        } else if (pid == 0) {
                execv(args[0], (char **) args);
                _exit(EXIT_FAILURE);
        }

        *_pid = pid;
        return 0;
}

/*
 * A newly allocated VT uses the font from the active VT. Here
 * we update all possibly already allocated VTs with the configured
 * font. It also allows to restart systemd-vconsole-setup.service,
 * to apply a new font to all VTs.
 */
static void font_copy_to_all_vcs(int fd) {
        struct vt_stat vcs = {};
        int i, r;

        /* get active, and 16 bit mask of used VT numbers */
        r = ioctl(fd, VT_GETSTATE, &vcs);
        if (r < 0)
                return;

        for (i = 1; i <= 15; i++) {
                char vcname[16];
                int _cleanup_close_ vcfd = -1;
                struct console_font_op cfo = {};

                if (i == vcs.v_active)
                        continue;

                /* skip non-allocated ttys */
                snprintf(vcname, sizeof(vcname), "/dev/vcs%i", i);
                if (access(vcname, F_OK) < 0)
                        continue;

                snprintf(vcname, sizeof(vcname), "/dev/tty%i", i);
                vcfd = open_terminal(vcname, O_RDWR|O_CLOEXEC);
                if (vcfd < 0)
                        continue;

                /* copy font from active VT, where the font was uploaded to */
                cfo.op = KD_FONT_OP_COPY;
                cfo.height = vcs.v_active-1; /* tty1 == index 0 */
                ioctl(vcfd, KDFONTOP, &cfo);
        }
}

int main(int argc, char **argv) {
        const char *vc;
        char *vc_keymap = NULL;
        char *vc_keymap_toggle = NULL;
        char *vc_font = NULL;
        char *vc_font_map = NULL;
        char *vc_font_unimap = NULL;
        int fd = -1;
        bool utf8;
        pid_t font_pid = 0, keymap_pid = 0;
        bool font_copy = false;
        int r = EXIT_FAILURE;

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

        umask(0022);

        if (argv[1])
                vc = argv[1];
        else {
                vc = "/dev/tty0";
                font_copy = true;
        }

        fd = open_terminal(vc, O_RDWR|O_CLOEXEC);
        if (fd < 0) {
                log_error("Failed to open %s: %m", vc);
                goto finish;
        }

        if (!is_vconsole(fd)) {
                log_error("Device %s is not a virtual console.", vc);
                goto finish;
        }

        utf8 = is_locale_utf8();

        r = parse_env_file("/etc/vconsole.conf", NEWLINE,
                           "KEYMAP", &vc_keymap,
                           "KEYMAP_TOGGLE", &vc_keymap_toggle,
                           "FONT", &vc_font,
                           "FONT_MAP", &vc_font_map,
                           "FONT_UNIMAP", &vc_font_unimap,
                           NULL);

        if (r < 0 && r != -ENOENT)
                log_warning("Failed to read /etc/vconsole.conf: %s", strerror(-r));

        /* Let the kernel command line override /etc/vconsole.conf */
        if (detect_container(NULL) <= 0) {
                r = parse_env_file("/proc/cmdline", WHITESPACE,
                                   "vconsole.keymap", &vc_keymap,
                                   "vconsole.keymap.toggle", &vc_keymap_toggle,
                                   "vconsole.font", &vc_font,
                                   "vconsole.font.map", &vc_font_map,
                                   "vconsole.font.unimap", &vc_font_unimap,
                                   NULL);

                if (r < 0 && r != -ENOENT)
                        log_warning("Failed to read /proc/cmdline: %s", strerror(-r));
        }

        if (utf8)
                enable_utf8(fd);
        else
                disable_utf8(fd);

        r = EXIT_FAILURE;
        if (keymap_load(vc, vc_keymap, vc_keymap_toggle, utf8, &keymap_pid) >= 0 &&
            font_load(vc, vc_font, vc_font_map, vc_font_unimap, &font_pid) >= 0)
                r = EXIT_SUCCESS;

finish:
        if (keymap_pid > 0)
                wait_for_terminate_and_warn(KBD_LOADKEYS, keymap_pid);

        if (font_pid > 0) {
                wait_for_terminate_and_warn(KBD_SETFONT, font_pid);
                if (font_copy)
                        font_copy_to_all_vcs(fd);
        }

        free(vc_keymap);
        free(vc_font);
        free(vc_font_map);
        free(vc_font_unimap);

        if (fd >= 0)
                close_nointr_nofail(fd);

        return r;
}