RTShutdown.cpp revision 003b661091a8a7729d875ce6539a10a5d515d05d
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync/* $Id$ */
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync/** @file
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync * IPRT Testcase - System Shutdown.
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync */
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync/*
1c94c0a63ba68be1a7b2c640e70d7a06464e4fcavboxsync * Copyright (C) 2012 Oracle Corporation
c98fb3e16fcd571a790eab772c0c66173d225205vboxsync *
c98fb3e16fcd571a790eab772c0c66173d225205vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
c98fb3e16fcd571a790eab772c0c66173d225205vboxsync * available from http://www.virtualbox.org. This file is free software;
c98fb3e16fcd571a790eab772c0c66173d225205vboxsync * you can redistribute it and/or modify it under the terms of the GNU
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * General Public License (GPL) as published by the Free Software
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync *
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * The contents of this file may alternatively be used under the terms
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * of the Common Development and Distribution License Version 1.0
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * VirtualBox OSE distribution, in which case the provisions of the
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * CDDL are applicable instead of those of the GPL.
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync *
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * You may elect to license modified versions of this file under the
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * terms and conditions of either the GPL or the CDDL or both.
1c94c0a63ba68be1a7b2c640e70d7a06464e4fcavboxsync */
1c94c0a63ba68be1a7b2c640e70d7a06464e4fcavboxsync
1c94c0a63ba68be1a7b2c640e70d7a06464e4fcavboxsync/*******************************************************************************
1c94c0a63ba68be1a7b2c640e70d7a06464e4fcavboxsync* Header Files *
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync*******************************************************************************/
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync#include <iprt/system.h>
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync#include <iprt/buildconfig.h>
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync#include <iprt/err.h>
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync#include <iprt/getopt.h>
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync#include <iprt/initterm.h>
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync#include <iprt/message.h>
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync#include <iprt/stream.h>
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync#include <iprt/string.h>
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsyncint main(int argc, char **argv)
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync{
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync int rc = RTR3InitExe(argc, &argv, 0);
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync if (RT_FAILURE(rc))
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync return RTMsgInitFailure(rc);
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync /*
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync * Parse the command line.
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync */
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync static const RTGETOPTDEF s_aOptions[] =
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync {
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync { "--halt", 'H', RTGETOPT_REQ_NOTHING },
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync { "--poweroff", 'p', RTGETOPT_REQ_NOTHING },
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync { "--reboot", 'r', RTGETOPT_REQ_NOTHING },
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync { "--force", 'f', RTGETOPT_REQ_NOTHING },
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync { "--delay", 'd', RTGETOPT_REQ_UINT32 },
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync { "--message", 'm', RTGETOPT_REQ_STRING }
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync };
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync const char *pszMsg = "RTShutdown";
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync RTMSINTERVAL cMsDelay = 0;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync uint32_t fFlags = RTSYSTEM_SHUTDOWN_POWER_OFF | RTSYSTEM_SHUTDOWN_PLANNED;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync RTGETOPTSTATE GetState;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync rc = RTGetOptInit(&GetState, argc, argv, s_aOptions, RT_ELEMENTS(s_aOptions), 1,
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync RTGETOPTINIT_FLAGS_OPTS_FIRST);
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync for (;;)
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync {
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync RTGETOPTUNION ValueUnion;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync rc = RTGetOpt(&GetState, &ValueUnion);
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync if (rc == 0)
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync break;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync switch (rc)
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync {
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync case 'H': fFlags = (fFlags & ~RTSYSTEM_SHUTDOWN_ACTION_MASK) | RTSYSTEM_SHUTDOWN_HALT; break;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync case 'p': fFlags = (fFlags & ~RTSYSTEM_SHUTDOWN_ACTION_MASK) | RTSYSTEM_SHUTDOWN_POWER_OFF_HALT; break;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync case 'r': fFlags = (fFlags & ~RTSYSTEM_SHUTDOWN_ACTION_MASK) | RTSYSTEM_SHUTDOWN_REBOOT; break;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync case 'f': fFlags |= RTSYSTEM_SHUTDOWN_FORCE; break;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync case 'd': cMsDelay = ValueUnion.u32; break;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync case 'm': pszMsg = ValueUnion.psz; break;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync case 'h':
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync RTPrintf("Usage: RTShutdown [-H|-p|-r] [-f] [-d <milliseconds>] [-m <msg>]\n");
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync return RTEXITCODE_SUCCESS;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync case 'V':
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync RTPrintf("%sr%d\n", RTBldCfgVersion(), RTBldCfgRevision());
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync return RTEXITCODE_SUCCESS;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync default:
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync return RTGetOptPrintError(rc, &ValueUnion);
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync }
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync }
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync /*
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync * Do the deed.
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync */
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync rc = RTSystemShutdown(cMsDelay, fFlags, pszMsg);
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync if (RT_FAILURE(rc))
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTSystemShutdown(%u, %#x, \"%s\") returned %Rrc\n", cMsDelay, fFlags, pszMsg, rc);
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync return RTEXITCODE_SUCCESS;
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync}
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync
9c9df2b728333cb734a7cc7856568e9ea9dc4600vboxsync