/*++

Copyright (c) 1998  Microsoft Corporation

Module Name:

    map.c

Abstract:

    This module implements the drive mapping command.

Author:

    Wesley Witt (wesw) 21-Oct-1998
    Ted Miller  (tedm) 21-Oct-1998

Revision History:

--*/

#include "cmdcons.h"
#pragma hdrstop


LPWSTR FsTypes[] =
{
    L"     ",
    L"NEW  ",
    L"FAT16",
    L"NTFS ",
    L"FAT32",
    L"     ",
    L"     "
};



BOOL
RcDiskRegionEnum(
    IN PPARTITIONED_DISK Disk,
    IN PDISK_REGION Region,
    IN ULONG_PTR UseArcNames
    )
{
    ULONGLONG RegionSizeMB;
    WCHAR ArcName[256];

    if (!SPPT_IS_REGION_PARTITIONED(Region) ||
        (Region->ExtendedType == EPTContainerPartition)) {
        return TRUE;        
    }

    if (UseArcNames) {
        SpArcNameFromRegion(
            Region,
            ArcName,
            sizeof(ArcName),
            PartitionOrdinalCurrent,
            PrimaryArcPath
            );
    } else {
        SpNtNameFromRegion(
            Region,
            ArcName,
            sizeof(ArcName),
            PartitionOrdinalCurrent
            );
    }

    RegionSizeMB = SpPtSectorCountToMB(Disk->HardDisk, Region->SectorCount);

    RcMessageOut(
        MSG_MAP_ENTRY,
        Region->DriveLetter == 0 ? L'?' : Region->DriveLetter,
        Region->DriveLetter == 0 ? L' ' : L':',
        FsTypes[Region->Filesystem],
        (ULONG)RegionSizeMB,
        ArcName
        );

    return TRUE;
}


NTSTATUS
GetDriveLetterLinkTarget(
    IN PWSTR SourceNameStr,
    OUT PWSTR *LinkTarget
    )
{
    static WCHAR        targetNameBuffer[256];

    NTSTATUS            status;
    UNICODE_STRING      sourceName;
    UNICODE_STRING      targetName;
    OBJECT_ATTRIBUTES   oa;
    HANDLE              handle;


    RtlInitUnicodeString(&sourceName, SourceNameStr);

    InitializeObjectAttributes(&oa, &sourceName, OBJ_CASE_INSENSITIVE, NULL, NULL);

    status = ZwOpenSymbolicLinkObject(&handle, READ_CONTROL | SYMBOLIC_LINK_QUERY, &oa);

    if (NT_SUCCESS(status))
    {
        RtlZeroMemory(targetNameBuffer, sizeof(targetNameBuffer));
        targetName.Buffer = targetNameBuffer;
        targetName.MaximumLength = sizeof(targetNameBuffer);

        status = ZwQuerySymbolicLinkObject(handle, &targetName, NULL);
        NtClose(handle);
    }

    if (NT_SUCCESS(status))
    {
        *LinkTarget = targetName.Buffer;
    }
    else
    {
        *LinkTarget = NULL;
    }

    return status;
}


ULONG
RcCmdDriveMap(
    IN PTOKENIZED_LINE TokenizedLine
    )
{
    ULONG i;
    LPWSTR s;
    LPWSTR p;
    WCHAR buf[128];


    if (RcCmdParseHelp( TokenizedLine, MSG_MAP_HELP )) {
        return 1;
    }

    RcTextOut( L"\r\n" );

    if (TokenizedLine->TokenCount == 2 && _wcsicmp( TokenizedLine->Tokens->Next->String, L"arc" ) == 0) {
        SpEnumerateDiskRegions( (PSPENUMERATEDISKREGIONS)RcDiskRegionEnum, 1 );
    } else {
        SpEnumerateDiskRegions( (PSPENUMERATEDISKREGIONS)RcDiskRegionEnum, 0 );
    }

    for (i=0; i<26; i++) {
        swprintf( buf, L"\\DosDevices\\%c:",i+L'A');
        if (RcIsFileOnCDROM(buf) == STATUS_SUCCESS ||
            RcIsFileOnFloppy(buf) == STATUS_SUCCESS
             || RcIsNetworkDrive(buf) == STATUS_SUCCESS
            )
        {
            GetDriveLetterLinkTarget( buf, &s );
            RcMessageOut( MSG_MAP_ENTRY2, buf[12], s );
        }
    }

    return 1;
}