|
|
/*++
Copyright (c) 1995 Microsoft Corporation
Module Name:
cdp.c
Abstract:
A user mode app that allows simple commands to be sent to a selected scsi device.
Environment:
User mode only
Revision History:
03-26-96 : Created
--*/
//
// this module may be compiled at warning level 4 with the following
// warnings disabled:
//
#pragma warning(disable:4200) // array[0]
#pragma warning(disable:4201) // nameless struct/unions
#pragma warning(disable:4214) // bit fields other than int
#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include <windows.h>
// #include "bootstatus.h"
#ifdef DBG
#define dbg(x) x
#define HELP_ME() printf("Reached line %4d\n", __LINE__);
#else
#define dbg(x) /* x */
#define HELP_ME() /* printf("Reached line %4d\n", __LINE__); */
#endif
#define ARGUMENT_USED(x) (x == NULL)
//
// The default size of the safemode.dat file - it must be large enough that the
// data stream isn't a resident attribute of the metadata or the loader's
// NTFS implementation can't write to it.
//
#define DEFAULT_SAFEMODE_FILE_SIZE 0x800
typedef struct { PCHAR Name; PCHAR Description; BOOLEAN LockData; ULONG (*Function)(HANDLE BootStatusData, int argc, char *argv[]);} COMMAND;
DWORD TestCommand(HANDLE BootStatusData, int argc, char *argv[]);DWORD Disable(HANDLE BootStatusData, int argc, char *argv[]);DWORD Enable(HANDLE BootStatusData, int argc, char *argv[]);DWORD Create(HANDLE BootStatusData, int argc, char *argv[]);DWORD SetFlags(HANDLE BootStatusData, int argc, char *argv[]);DWORD ClearFlags(HANDLE BootStatusData, int argc, char *argv[]);DWORD ListSettings(HANDLE BootStatusData, int argc, char *argv[]);
DWORD ListCommand(int argc, char *argv[]);
//
// List of commands
// all command names are case sensitive
// arguments are passed into command routines
// list must be terminated with NULL command
// command will not be listed in help if description == NULL
//
COMMAND CommandArray[] = { {"create", "Creates the boot status data file", FALSE, Create}, {"disable", "Disables Auto Advanced Boot", TRUE, Disable}, {"enable", "Enables Auto Advanced Boot", TRUE, Enable}, {"set", "Sets the boot status flags", TRUE, SetFlags}, {"settings","Lists the Auto Advanced Boot settings", TRUE, ListSettings}, {NULL, NULL, FALSE, NULL} };
int __cdecl main(int argc, char *argv[]){ int i = 0; HANDLE bootStatusData;
if(argc < 2) { printf("Usage: %s <command> [parameters]\n", argv[0]); printf("possible commands: \n"); ListCommand(argc, argv); printf("\n"); return -1; }
//
// Iterate through the command array and find the correct function to
// call.
//
while(CommandArray[i].Name != NULL) {
if(strcmp(argv[1], CommandArray[i].Name) == 0) {
NTSTATUS status;
if(CommandArray[i].LockData) { status = RtlLockBootStatusData(&bootStatusData); if(!NT_SUCCESS(status)) { printf("Error %#08lx opening boot status bootStatusData\n", status); return -1; } } else { bootStatusData = NULL; }
(CommandArray[i].Function)(bootStatusData, (argc - 1), &(argv[1]));
break; }
i++; }
if(CommandArray[i].Name == NULL) { printf("Unknown command %s\n", argv[1]); }
RtlUnlockBootStatusData(bootStatusData); return 0;}
DWORD TestCommand(HANDLE BootStatusData, int argc, char *argv[])/*++
Routine Description:
Tests the command "parsing"
Arguments: device - a file handle to send the ioctl to
argc - the number of additional arguments. should be zero
argv - the additional arguments
Return Value:
STATUS_SUCCESS if successful The value of GetLastError() from the point of failure
--*/
{ int i;
UNREFERENCED_PARAMETER(BootStatusData);
printf("Test - %d additional arguments\n", argc);
for(i = 0; i < argc; i++) { printf("arg %d: %s\n", i, argv[i]); }
return STATUS_SUCCESS;}
DWORD ListCommand(int argc, char *argv[])/*++
Routine Description:
Prints out the command list
Arguments: device - unused
argc - unused
argv - unused
Return Value:
STATUS_SUCCESS
--*/
{ int i = 0;
UNREFERENCED_PARAMETER(argc); UNREFERENCED_PARAMETER(argv);
while(CommandArray[i].Name != NULL) {
if(CommandArray[i].Description != NULL) {
printf("\t%s - %s\n", CommandArray[i].Name, CommandArray[i].Description); }
i++; }
return STATUS_SUCCESS;}
DWORD Create( HANDLE Unused, int argc, char *argv[] ){ BOOLEAN enabled;
NTSTATUS status;
status = RtlCreateBootStatusDataFile();
if(!NT_SUCCESS(status)) { printf("Error %#08lx creating boot status data\n", status); }
return status;}
DWORD Enable( HANDLE BootStatusData, int argc, char *argv[] ){ BOOLEAN enabled;
NTSTATUS status;
status = RtlGetSetBootStatusData(BootStatusData, TRUE, RtlBsdItemAabEnabled, &enabled, sizeof(BOOLEAN), NULL);
if(!NT_SUCCESS(status)) { printf("Error %#08lx reading boot status data\n", status); return status; }
printf("Enabling Automatic Advanced Boot\n"); printf("Previous setting was %s\n", enabled ? "enabled" : "disabled");
enabled = TRUE;
status = RtlGetSetBootStatusData(BootStatusData, FALSE, RtlBsdItemAabEnabled, &enabled, sizeof(BOOLEAN), NULL);
if(!NT_SUCCESS(status)) { printf("Error %#08lx writing boot status data\n", status); }
return status;}
DWORD Disable( HANDLE BootStatusData, int argc, char *argv[] ){ BOOLEAN enabled;
NTSTATUS status;
status = RtlGetSetBootStatusData(BootStatusData, TRUE, RtlBsdItemAabEnabled, &enabled, sizeof(BOOLEAN), NULL);
if(!NT_SUCCESS(status)) { printf("Error %#08lx reading boot status data\n", status); return status; }
printf("Disabling Automatic Advanced Boot\n"); printf("Previous setting was %s\n", enabled ? "enabled" : "disabled");
enabled = FALSE;
status = RtlGetSetBootStatusData(BootStatusData, FALSE, RtlBsdItemAabEnabled, &enabled, sizeof(BOOLEAN), NULL);
if(!NT_SUCCESS(status)) { printf("Error %#08lx writing boot status data\n", status); }
return status;}
DWORD ListSettings( HANDLE BootStatusData, int argc, char *argv[] ){ BOOLEAN flag;
NTSTATUS status;
status = RtlGetSetBootStatusData(BootStatusData, TRUE, RtlBsdItemAabEnabled, &flag, sizeof(BOOLEAN), NULL);
if(!NT_SUCCESS(status)) { printf("Error %#08lx reading boot status data\n", status); return status; }
printf("Automatic Advanced Boot is %s\n", flag ? "enabled" : "disabled");
status = RtlGetSetBootStatusData(BootStatusData, TRUE, RtlBsdItemAabTimeout, &flag, sizeof(BOOLEAN), NULL);
if(!NT_SUCCESS(status)) { printf("Error %#08lx reading boot status data\n", status); return status; }
printf("Automatic Advanced Boot timeout is %d seconds\n", (UCHAR) flag);
status = RtlGetSetBootStatusData(BootStatusData, TRUE, RtlBsdItemBootGood, &flag, sizeof(BOOLEAN), NULL);
if(!NT_SUCCESS(status)) { printf("Error %#08lx reading boot status data\n", status); return status; }
printf("LastBootSucceeded = %#x\n", flag);
status = RtlGetSetBootStatusData(BootStatusData, TRUE, RtlBsdItemBootShutdown, &flag, sizeof(BOOLEAN), NULL);
if(!NT_SUCCESS(status)) { printf("Error %#08lx reading boot status data\n", status); return status; }
printf("LastBootShutdown = %#x\n", flag);
return STATUS_SUCCESS;}
DWORD SetFlags(HANDLE BootStatusData, int argc, char *argv[]){ int count;
if(argc <= 1) { printf("usage: autosafeboot set <Boot|Shutdown>=<value> ..."); return 0; }
for(count = 1; count < argc; count++) { PUCHAR nameString; PUCHAR valueString;
ULONG index; UCHAR value;
index = -1;
//
// match the string.
//
nameString = argv[count]; valueString = strrchr(nameString, '=');
if(valueString == NULL) { printf("** element \"%s\" not understood\n", nameString); continue; }
valueString[0] = '\0'; valueString += 1;
value = (UCHAR) atoi(valueString);
if(_stricmp(nameString, "boot") == 0) { printf("Setting LastBootSucceeded to %#x\n", value); index = RtlBsdItemBootGood;
} else if(_stricmp(nameString, "shutdown") == 0) {
printf("Setting LastBootShutdown to %#x\n", value); index = RtlBsdItemBootShutdown; }
if(index != -1) {
NTSTATUS status;
status = RtlGetSetBootStatusData(BootStatusData, FALSE, index, &value, sizeof(UCHAR), NULL);
if(!NT_SUCCESS(status)) { printf("Error %#08lx reading boot status data\n", status); continue; } } else { printf("** element \"%s=%s\" not understood\n", nameString, valueString); } }
return 0;}
|
