qubes-installer-qubes-os/loader/shutdown.c

/*
 * shutdown.c
 *
 * Shutdown a running system.  If built with -DAS_SHUTDOWN=1, then
 * it builds a standalone shutdown binary.
 *
 * Copyright (C) 1996, 1997, 1998, 1999, 2000, 2001, 2002, 2003  Red Hat, Inc.
 * All rights reserved.
 *
 * 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, see <http://www.gnu.org/licenses/>.
 */

#include <fcntl.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/reboot.h>
#include <sys/types.h>
#include <unistd.h>

#include "init.h"

void disableSwap(void);
void unmountFilesystems(void);

static void performTerminations(void) {
	sync();
	printf("sending termination signals...");
	kill(-1, 15);
	sleep(2);
	printf("done\n");

	printf("sending kill signals...");
	kill(-1, 9);
	sleep(2);
	printf("done\n");
}

static void performUnmounts(void) {
	int ignore;

	printf("disabling swap...\n");
	disableSwap();

	printf("unmounting filesystems...\n"); 
	unmountFilesystems();

	printf("waiting for mdraid sets to become clean...\n"); 
	ignore = system("/sbin/mdadm --wait-clean --scan");
}

static void performReboot(reboot_action rebootAction) {
    switch (rebootAction) {
    case POWEROFF:
        printf("powering off system\n");
        sleep(2);
        reboot(RB_POWER_OFF);
        break;
    case REBOOT:
        printf("rebooting system\n");
        sleep(2);
#if USE_MINILIBC
        reboot(0xfee1dead, 672274793, 0x1234567);
#else
        reboot(RB_AUTOBOOT);
#endif
        break;
    case HALT:
        printf("halting system\n");
        reboot(RB_HALT_SYSTEM);
        break;
    default:
        break;
    }
}

static void performDelayedReboot()
{
    printf("The system will be rebooted when you press Ctrl-C or Ctrl-Alt-Delete.\n");
    while (1) {
        sleep(1);
    }
}

void shutDown(int doKill, reboot_action rebootAction)
{
    static int reentered = 0;
    
    if (reentered) {
        performUnmounts();
        performTerminations();
        performReboot(rebootAction);
    }
    reentered = 1;
    if (rebootAction != DELAYED_REBOOT && doKill) {
        performUnmounts();
        performTerminations();
        performReboot(rebootAction);
    } else {
        performDelayedReboot();
    }
    exit(0);
}

#ifdef AS_SHUTDOWN
int main(int argc, char ** argv) {
    int fd;
    reboot_action rebootAction = HALT;
    int doKill = 1;
    int i = 1;

    while (i < argc) {
      if (!strncmp("-r", argv[i], 2))
        rebootAction = REBOOT;
      else if (!strncmp("--nokill", argv[i], 8))
        doKill = 0;
      else if (!strncmp("-P", argv[i], 2))
        rebootAction = POWEROFF;
      i++;
    }

    /* ignore some signals so we don't kill ourself */
    signal(SIGINT, SIG_IGN);
    signal(SIGTSTP, SIG_IGN);

    /* now change to / */
    i = chdir("/");

    /* redirect output to the real console */
    fd = open("/dev/console", O_RDWR);
    dup2(fd, 0);
    dup2(fd, 1);
    dup2(fd, 2);
    close(fd);

    shutDown(doKill, rebootAction);
    return 0;
}
#endif

/* vim:set shiftwidth=4 softtabstop=4 ts=4: */