Nie możesz wybrać więcej, niż 25 tematów Tematy muszą się zaczynać od litery lub cyfry, mogą zawierać myślniki ('-') i mogą mieć do 35 znaków.

324 wiersze
8.1 KiB

/* plymouth.c - Plymouth plugin for the OpenRC init system
*
* Copyright (C) 2011 Amadeusz Żołnowski <aidecoe@gentoo.org>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include <assert.h>
#include <einfo.h>
#include <rc.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <ctype.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>
#ifdef DEBUG
# define DBG(x) einfo("[plymouth-plugin] " x)
#else
# define DBG(x)
#endif
#define BUFFER_SIZE 300
#define RWDIR (R_OK | W_OK | X_OK)
#ifndef RUN_DIR
#define RUN_DIR "/run/plymouth"
#endif
#ifndef PID_FILE
#define PID_FILE RUN_DIR "/pid"
#endif
enum {
PLY_MODE_BOOT,
PLY_MODE_SHUTDOWN
};
int command(const char* cmd)
{
int rv = system(cmd);
#if DEBUG
if(rv != 0) {
ewarn("[plymouth-plugin] command(\"%s\"): rv=%d", cmd, rv);
}
#endif
return rv;
}
int commandf(const char* cmd, ...)
{
static char buffer[BUFFER_SIZE];
va_list ap;
int rv;
va_start(ap, cmd);
rv = vsnprintf(buffer, BUFFER_SIZE, cmd, ap);
va_end(ap);
if(rv >= BUFFER_SIZE) {
eerror("[plymouth-plugin] command(\"%s\"): buffer overflow", buffer);
return -1;
}
return command(buffer);
}
void restart_console_font()
{
if(RC_SERVICE_STARTED == rc_service_state("consolefont"))
command("/etc/init.d/consolefont restart");
}
bool ply_message(const char* hook, const char* name)
{
return true;
}
bool ply_ping()
{
return (system("/bin/plymouth --ping") == 0);
}
bool ply_quit(int mode)
{
int rv = 0;
if(mode == PLY_MODE_BOOT)
rv = command("/bin/plymouth quit");
else if(mode == PLY_MODE_SHUTDOWN)
rv = command("/bin/plymouth quit --retain-splash");
else
assert(0 && "Unknown mode");
return (rv == 0);
}
bool ply_start(int mode)
{
int rv = 0;
if(!ply_ping()) {
ebegin("Starting plymouthd");
if(access(RUN_DIR, RWDIR) != 0) {
if(mkdir(RUN_DIR, 0755) != 0) {
eerror("[plymouth-plugin] Couldn't create " RUN_DIR);
return false;
}
}
#define PLYD "/sbin/plymouthd --attach-to-session --pid-file=" PID_FILE \
" --tty=/dev/tty1 --mode="
if(mode == PLY_MODE_BOOT)
rv = command(PLYD "boot");
else if(mode == PLY_MODE_SHUTDOWN)
rv = command(PLYD "shutdown");
else
assert(0 && "Unknown mode");
#undef PLYD
eend(rv, "");
if((rv == 0) && command("/bin/plymouth --show-splash") != 0)
return false;
}
return true;
}
bool ply_update_status(int hook, const char* name)
{
return (commandf("/bin/plymouth update --status=%d-%s", hook, name) == 0);
}
bool kernel_command_has_argument(const char *argument)
{
char buf[4096];
int fd;
fd = open("/proc/cmdline", O_RDONLY);
if( fd < 0 ) {
return false;
}
if(read(fd, buf, 4095) < 0) {
close(fd);
return false;
}
close(fd);
char *found_arg = strstr(buf, argument);
if(found_arg == NULL)
return false;
if(found_arg == buf || found_arg[-1] == ' ') {
char endch = *(found_arg+strlen(argument));
return (isspace(endch) || endch == '\0') ? true : false;
}
return false;
}
bool ply_update_rootfs_rw()
{
const int rwdir = RWDIR;
if(access("/var/lib/plymouth", rwdir) != 0
|| access("/var/log", rwdir) != 0) {
eerror("[plymouth-plugin] /var/lib/plymouth and /var/log need to be "
"writable at this stage, but are not!");
return false;
}
return (command("/bin/plymouth update-root-fs --read-write") == 0);
}
int rc_plugin_hook(RC_HOOK hook, const char *name)
{
int rv = 0;
char* runlevel = rc_runlevel_get();
const char* bootlevel = getenv("RC_BOOTLEVEL");
const char* defaultlevel = getenv("RC_DEFAULTLEVEL");
#ifdef DEBUG
einfo("hook=%d name=%s runlvl=%s plyd=%d", hook, name, runlevel,
ply_ping());
#endif
/* Don't do anything if we're not booting or shutting down. */
if(!(rc_runlevel_starting() || rc_runlevel_stopping())) {
switch(hook) {
case RC_HOOK_RUNLEVEL_STOP_IN:
case RC_HOOK_RUNLEVEL_STOP_OUT:
case RC_HOOK_RUNLEVEL_START_IN:
case RC_HOOK_RUNLEVEL_START_OUT:
/* Switching runlevels, so we're booting or shutting down.*/
break;
default:
DBG("Not booting or shutting down");
goto exit;
}
}
DBG("switch1");
switch(hook) {
case RC_HOOK_RUNLEVEL_START_IN:
case RC_HOOK_RUNLEVEL_STOP_IN:
case RC_HOOK_SERVICE_START_IN:
case RC_HOOK_SERVICE_STOP_IN:
ply_update_status(hook, name);
break;
default:
break;
}
DBG("switch2");
switch(hook) {
case RC_HOOK_RUNLEVEL_STOP_IN:
/* Start the Plymouth daemon and show splash when system is being shut
* down. */
if(strcmp(name, RC_LEVEL_SHUTDOWN) == 0 &&
(kernel_command_has_argument("splash") ||
kernel_command_has_argument("splash=shutdown")))
{
DBG("ply_start(PLY_MODE_SHUTDOWN)");
if(!ply_start(PLY_MODE_SHUTDOWN)
|| !ply_update_rootfs_rw())
rv = 1;
command("/usr/bin/chvt 1");
}
break;
case RC_HOOK_RUNLEVEL_START_IN:
/* Start the Plymouth daemon and show splash when entering the boot
* runlevel. Required /proc and /sys should already be mounted in
* sysinit runlevel. */
if(strcmp(name, bootlevel) == 0 &&
(kernel_command_has_argument("splash") ||
kernel_command_has_argument("splash=silent")))
{
DBG("ply_start(PLY_MODE_BOOT)");
if(!ply_start(PLY_MODE_BOOT))
rv = 1;
}
break;
case RC_HOOK_RUNLEVEL_START_OUT:
/* Stop the Plymouth daemon right after default runlevel is started. */
if(strcmp(name, defaultlevel) == 0) {
DBG("ply_quit(PLY_MODE_BOOT)");
if(!ply_quit(PLY_MODE_BOOT))
rv = 1;
restart_console_font();
}
break;
case RC_HOOK_SERVICE_STOP_IN:
/* Quit Plymouth when we're going to lost write access to /var/... */
if(strcmp(name, "mount-ro") == 0 &&
strcmp(runlevel, RC_LEVEL_SHUTDOWN) == 0) {
DBG("ply_quit(PLY_MODE_SHUTDOWN)");
if(!ply_quit(PLY_MODE_SHUTDOWN))
rv = 1;
}
break;
case RC_HOOK_SERVICE_STOP_NOW:
if(!ply_message("Stopping service", name))
rv = 1;
if(strcmp(name, "xdm") == 0)
command("/usr/bin/chvt 1");
break;
case RC_HOOK_SERVICE_STOP_DONE:
if(strcmp(name, "xdm") == 0)
command("/usr/bin/chvt 1");
break;
case RC_HOOK_SERVICE_START_NOW:
if(!ply_message("Starting service", name))
rv = 1;
break;
case RC_HOOK_SERVICE_START_OUT:
/* Start Plymouth daemon if not yet started and tell we have rw access
* to /var/... */
if(strcmp(name, "localmount") == 0 &&
strcmp(runlevel, RC_LEVEL_SHUTDOWN) != 0) {
DBG("ply_update_rootfs_rw()");
if(!ply_update_rootfs_rw())
rv = 1;
}
break;
default:
break;
}
exit:
free(runlevel);
return rv;
}