mirror of
https://github.com/reactos/reactos.git
synced 2024-08-16 16:39:08 +00:00
197 lines
5.5 KiB
C
197 lines
5.5 KiB
C
/*
|
|
* PROJECT: ReactOS DiskPart
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* FILE: base/system/diskpart/diskpart.c
|
|
* PURPOSE: Manages all the partitions of the OS in an interactive way.
|
|
* PROGRAMMERS: Lee Schroeder
|
|
*/
|
|
|
|
/* INCLUDES ******************************************************************/
|
|
|
|
#include "diskpart.h"
|
|
|
|
/* FUNCTIONS ******************************************************************/
|
|
|
|
VOID
|
|
ShowHeader(VOID)
|
|
{
|
|
WCHAR szComputerName[MAX_STRING_SIZE];
|
|
DWORD comp_size = MAX_STRING_SIZE;
|
|
|
|
/* Get the name of the computer for us and change the value of comp_name */
|
|
GetComputerNameW(szComputerName, &comp_size);
|
|
|
|
/* TODO: Remove this section of code when program becomes stable enough for production use. */
|
|
ConPuts(StdOut, L"\n*WARNING*: This program is incomplete and may not work properly.\n");
|
|
|
|
/* Print the header information */
|
|
ConPuts(StdOut, L"\n");
|
|
ConResPuts(StdOut, IDS_APP_HEADER);
|
|
ConPuts(StdOut, L"\n");
|
|
ConResPuts(StdOut, IDS_APP_LICENSE);
|
|
ConResPrintf(StdOut, IDS_APP_CURR_COMPUTER, szComputerName);
|
|
}
|
|
|
|
/*
|
|
* RunScript(const char *filename):
|
|
* opens the file, reads the contents, convert the text into readable
|
|
* code for the computer, and then execute commands in order.
|
|
*/
|
|
BOOL
|
|
RunScript(LPCWSTR filename)
|
|
{
|
|
FILE *script;
|
|
WCHAR tmp_string[MAX_STRING_SIZE];
|
|
|
|
/* Open the file for processing */
|
|
script = _wfopen(filename, L"r");
|
|
if (script == NULL)
|
|
{
|
|
/* if there was problems opening the file */
|
|
ConResPrintf(StdErr, IDS_ERROR_MSG_NO_SCRIPT, filename);
|
|
return FALSE; /* if there is no script, exit the program */
|
|
}
|
|
|
|
/* Read and process the script */
|
|
while (fgetws(tmp_string, MAX_STRING_SIZE, script) != NULL)
|
|
{
|
|
if (InterpretScript(tmp_string) == FALSE)
|
|
{
|
|
fclose(script);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
/* Close the file */
|
|
fclose(script);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/*
|
|
* wmain():
|
|
* Main entry point of the application.
|
|
*/
|
|
int wmain(int argc, const LPWSTR argv[])
|
|
{
|
|
LPCWSTR script = NULL;
|
|
LPCWSTR tmpBuffer = NULL;
|
|
WCHAR appTitle[50];
|
|
int index, timeout;
|
|
int result = EXIT_SUCCESS;
|
|
|
|
/* Initialize the Console Standard Streams */
|
|
ConInitStdStreams();
|
|
|
|
/* Sets the title of the program so the user will have an easier time
|
|
determining the current program, especially if diskpart is running a
|
|
script */
|
|
K32LoadStringW(GetModuleHandle(NULL), IDS_APP_HEADER, appTitle, ARRAYSIZE(appTitle));
|
|
SetConsoleTitleW(appTitle);
|
|
|
|
/* Sets the timeout value to 0 just in case the user doesn't
|
|
specify a value */
|
|
timeout = 0;
|
|
|
|
CreatePartitionList();
|
|
|
|
/* If there are no command arguments, then go straight to the interpreter */
|
|
if (argc < 2)
|
|
{
|
|
ShowHeader();
|
|
InterpretMain();
|
|
}
|
|
/* If there are command arguments, then process them */
|
|
else
|
|
{
|
|
for (index = 1; index < argc; index++)
|
|
{
|
|
/* checks for flags */
|
|
if ((argv[index][0] == '/')||
|
|
(argv[index][0] == '-'))
|
|
{
|
|
tmpBuffer = argv[index] + 1;
|
|
}
|
|
else
|
|
{
|
|
/* If there is no flag, then return an error */
|
|
ConResPrintf(StdErr, IDS_ERROR_MSG_BAD_ARG, argv[index]);
|
|
result = EXIT_FAILURE;
|
|
goto done;
|
|
}
|
|
|
|
/* Checks for the /? flag first since the program
|
|
exits as soon as the usage list is shown. */
|
|
if (_wcsicmp(tmpBuffer, L"?") == 0)
|
|
{
|
|
ConResPuts(StdOut, IDS_APP_USAGE);
|
|
result = EXIT_SUCCESS;
|
|
goto done;
|
|
}
|
|
/* Checks for the script flag */
|
|
else if (_wcsicmp(tmpBuffer, L"s") == 0)
|
|
{
|
|
if ((index + 1) < argc)
|
|
{
|
|
index++;
|
|
script = argv[index];
|
|
}
|
|
}
|
|
/* Checks for the timeout flag */
|
|
else if (_wcsicmp(tmpBuffer, L"t") == 0)
|
|
{
|
|
if ((index + 1) < argc)
|
|
{
|
|
index++;
|
|
timeout = _wtoi(argv[index]);
|
|
|
|
/* If the number is a negative number, then
|
|
change it so that the time is executed properly. */
|
|
if (timeout < 0)
|
|
timeout = 0;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* Assume that the flag doesn't exist. */
|
|
ConResPrintf(StdErr, IDS_ERROR_MSG_BAD_ARG, tmpBuffer);
|
|
result = EXIT_FAILURE;
|
|
goto done;
|
|
}
|
|
}
|
|
|
|
/* Shows the program information */
|
|
ShowHeader();
|
|
|
|
/* Now we process the filename if it exists */
|
|
if (script != NULL)
|
|
{
|
|
/* if the timeout is greater than 0, then assume
|
|
that the user specified a specific time. */
|
|
if (timeout > 0)
|
|
Sleep(timeout * 1000);
|
|
|
|
if (RunScript(script) == FALSE)
|
|
{
|
|
result = EXIT_FAILURE;
|
|
goto done;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* Exit failure since the user wanted to run a script */
|
|
ConResPrintf(StdErr, IDS_ERROR_MSG_NO_SCRIPT, script);
|
|
result = EXIT_FAILURE;
|
|
goto done;
|
|
}
|
|
}
|
|
|
|
/* Let the user know the program is exiting */
|
|
ConResPuts(StdOut, IDS_APP_LEAVING);
|
|
|
|
done:
|
|
DestroyPartitionList();
|
|
|
|
return result;
|
|
}
|