assign.c File Reference

#include "diskpart.h"
#include <debug.h>

Include dependency graph for assign.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
EXIT_CODE	assign_main (_In_ INT argc, _In_ LPWSTR *argv)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 11 of file assign.c.

Function Documentation

assign_main()

EXIT_CODE assign_main	(	_In_ INT	argc,
		_In_ LPWSTR *	argv
	)

Definition at line 15 of file assign.c.

{
    PWSTR pszSuffix = NULL;
    WCHAR DriveLetter = UNICODE_NULL;
    INT i, nExclusive = 0;
    BOOL bResult;
 
    DPRINT1("assign_main()\n");
 
    if (CurrentVolume == NULL)
    {
        ConResPuts(StdOut, IDS_SELECT_NO_VOLUME);
        return EXIT_SUCCESS;
    }
 
    for (i = 1; i < argc; i++)
    {
        if (_wcsicmp(argv[i], L"noerr") == 0)
        {
            /* noerr */
            DPRINT("NoErr\n", pszSuffix);
            ConPuts(StdOut, L"The NOERR option is not supported yet!\n");
#if 0
            bNoErr = TRUE;
#endif
        }
    }
 
    for (i = 1; i < argc; i++)
    {
        if (HasPrefix(argv[i], L"letter=", &pszSuffix))
        {
            if (wcslen(pszSuffix) == 1)
            {
                DriveLetter = towupper(*pszSuffix);
                nExclusive++;
            }
            else
            {
                ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS);
                return EXIT_SUCCESS; 
            }
        }
        else if (HasPrefix(argv[i], L"mount=", &pszSuffix))
        {
            DPRINT("Mount\n", pszSuffix);
            ConPuts(StdOut, L"The MOUNT option is not supported yet!\n");
            nExclusive++;
        }
        else if (_wcsicmp(argv[i], L"noerr") == 0)
        {
            /* noerr - Already handled above */
        }
        else
        {
            ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS);
            return EXIT_SUCCESS;
        }
    }
 
    if (nExclusive > 1)
    {
        ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS);
        return EXIT_SUCCESS;
    }
 
    DPRINT1("VolumeName: %S\n", CurrentVolume->VolumeName);
    DPRINT1("DeviceName: %S\n", CurrentVolume->DeviceName);
    DPRINT1("DriveLetter: %C\n", CurrentVolume->DriveLetter);
 
    if (DriveLetter != UNICODE_NULL)
    {
        DPRINT1("DriveLetter: %C\n", DriveLetter);
 
        if ((DriveLetter < L'C') || (DriveLetter > L'Z'))
        {
            ConResPuts(StdOut, IDS_ASSIGN_INVALID_LETTER);
            return EXIT_SUCCESS;
        }
 
        if (DriveLetter == CurrentVolume->DriveLetter)
        {
            ConResPuts(StdOut, IDS_ASSIGN_ALREADY_ASSIGNED);
            return EXIT_SUCCESS;
        }
    }
 
    if (CurrentVolume->DriveLetter != UNICODE_NULL)
    {
        /* Remove the current drive letter */
        bResult = DeleteDriveLetter(CurrentVolume->DriveLetter);
        if (bResult == FALSE)
        {
            ConResPuts(StdOut, IDS_ASSIGN_FAIL);
            return EXIT_SUCCESS;
        }
 
        CurrentVolume->DriveLetter = UNICODE_NULL;
    }
 
    if (DriveLetter != UNICODE_NULL)
    {
        /* Assign the new drive letter */
        bResult = AssignDriveLetter(CurrentVolume->DeviceName,
                                    DriveLetter);
        if (bResult == FALSE)
        {
            ConResPuts(StdOut, IDS_ASSIGN_FAIL);
            return EXIT_SUCCESS;
        }
 
        CurrentVolume->DriveLetter = DriveLetter;
    }
    else
    {
        bResult = AssignNextDriveLetter(CurrentVolume->DeviceName,
                                        &CurrentVolume->DriveLetter);
        if (bResult == FALSE)
        {
            ConResPuts(StdOut, IDS_ASSIGN_FAIL);
            return EXIT_SUCCESS;
        }
 
        if (CurrentVolume->DriveLetter == UNICODE_NULL)
        {
            ConResPuts(StdOut, IDS_ASSIGN_NO_MORE_LETTER);
            return EXIT_SUCCESS;
        }
    }
 
    ConResPuts(StdOut, IDS_REMOVE_SUCCESS);
 
    return EXIT_SUCCESS;
}