symlink.c File Reference

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

Include dependency graph for symlink.c:

Go to the source code of this file.

Macros
#define	NDEBUG
#define	IOCTL_MOUNTDEV_LINK_CREATED_UNSECURE_DEPRECATED   CTL_CODE(MOUNTDEVCONTROLTYPE, 4, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define	IOCTL_MOUNTDEV_LINK_DELETED_UNSECURE_DEPRECATED   CTL_CODE(MOUNTDEVCONTROLTYPE, 5, METHOD_BUFFERED, FILE_ANY_ACCESS)

Functions
NTSTATUS	CreateStringWithGlobal (IN PUNICODE_STRING DosName, OUT PUNICODE_STRING GlobalString)
NTSTATUS	GlobalCreateSymbolicLink (IN PUNICODE_STRING DosName, IN PUNICODE_STRING DeviceName)
NTSTATUS	GlobalDeleteSymbolicLink (IN PUNICODE_STRING DosName)
VOID	SendLinkCreated (IN PUNICODE_STRING SymbolicName)
VOID	SendLinkDeleted (IN PUNICODE_STRING DeviceName, IN PUNICODE_STRING SymbolicName)
NTSTATUS NTAPI	SymbolicLinkNamesFromUniqueIdCount (IN PWSTR ValueName, IN ULONG ValueType, IN PVOID ValueData, IN ULONG ValueLength, IN PVOID Context, IN PVOID EntryContext)
NTSTATUS NTAPI	SymbolicLinkNamesFromUniqueIdQuery (IN PWSTR ValueName, IN ULONG ValueType, IN PVOID ValueData, IN ULONG ValueLength, IN PVOID Context, IN PVOID EntryContext)
NTSTATUS	CreateNewVolumeName (OUT PUNICODE_STRING VolumeName, IN PGUID VolumeGuid OPTIONAL)
NTSTATUS	QuerySymbolicLinkNamesFromStorage (IN PDEVICE_EXTENSION DeviceExtension, IN PDEVICE_INFORMATION DeviceInformation, IN PUNICODE_STRING SuggestedLinkName, IN BOOLEAN UseOnlyIfThereAreNoOtherLinks, OUT PUNICODE_STRING *SymLinks, OUT PULONG SymLinkCount, IN BOOLEAN HasGuid, IN LPGUID Guid)
PSAVED_LINK_INFORMATION	RemoveSavedLinks (IN PDEVICE_EXTENSION DeviceExtension, IN PMOUNTDEV_UNIQUE_ID UniqueId)
NTSTATUS	QuerySuggestedLinkName (IN PUNICODE_STRING SymbolicName, OUT PUNICODE_STRING SuggestedLinkName, OUT PBOOLEAN UseOnlyIfThereAreNoOtherLinks)
BOOLEAN	RedirectSavedLink (IN PSAVED_LINK_INFORMATION SavedLinkInformation, IN PUNICODE_STRING DosName, IN PUNICODE_STRING NewLink)
VOID	DeleteSymbolicLinkNameFromMemory (IN PDEVICE_EXTENSION DeviceExtension, IN PUNICODE_STRING SymbolicLink, IN BOOLEAN MarkOffline)
BOOLEAN	IsDriveLetter (PUNICODE_STRING SymbolicName)
NTSTATUS	MountMgrQuerySymbolicLink (IN PUNICODE_STRING SymbolicName, IN OUT PUNICODE_STRING LinkTarget)

Variables
UNICODE_STRING	DeviceMount = RTL_CONSTANT_STRING(MOUNTMGR_DEVICE_NAME)
UNICODE_STRING	DosDevicesMount = RTL_CONSTANT_STRING(L"\\DosDevices\\MountPointManager")
UNICODE_STRING	DosDevices = RTL_CONSTANT_STRING(L"\\DosDevices\\")
UNICODE_STRING	DeviceFloppy = RTL_CONSTANT_STRING(L"\\Device\\Floppy")
UNICODE_STRING	DeviceCdRom = RTL_CONSTANT_STRING(L"\\Device\\CdRom")
UNICODE_STRING	DosGlobal = RTL_CONSTANT_STRING(L"\\GLOBAL??\\")
UNICODE_STRING	Global = RTL_CONSTANT_STRING(L"\\??\\")
UNICODE_STRING	SafeVolumes = RTL_CONSTANT_STRING(L"\\Device\\VolumesSafeForWriteAccess")
UNICODE_STRING	Volume = RTL_CONSTANT_STRING(L"\\??\\Volume")
UNICODE_STRING	ReparseIndex = RTL_CONSTANT_STRING(L"\\$Extend\\$Reparse:$R:$INDEX_ALLOCATION")

Macro Definition Documentation

IOCTL_MOUNTDEV_LINK_CREATED_UNSECURE_DEPRECATED

#define IOCTL_MOUNTDEV_LINK_CREATED_UNSECURE_DEPRECATED   CTL_CODE(MOUNTDEVCONTROLTYPE, 4, METHOD_BUFFERED, FILE_ANY_ACCESS)

Definition at line 36 of file symlink.c.

IOCTL_MOUNTDEV_LINK_DELETED_UNSECURE_DEPRECATED

#define IOCTL_MOUNTDEV_LINK_DELETED_UNSECURE_DEPRECATED   CTL_CODE(MOUNTDEVCONTROLTYPE, 5, METHOD_BUFFERED, FILE_ANY_ACCESS)

Definition at line 37 of file symlink.c.

NDEBUG

#define NDEBUG

Definition at line 28 of file symlink.c.

Function Documentation

CreateNewVolumeName()

NTSTATUS CreateNewVolumeName	(	OUT PUNICODE_STRING	VolumeName,
		IN PGUID VolumeGuid	OPTIONAL
	)

Definition at line 399 of file symlink.c.

{
    GUID Guid;
    NTSTATUS Status;
    UNICODE_STRING GuidString;
 
    /* If no GUID was provided, then create one */
    if (!VolumeGuid)
    {
        Status = ExUuidCreate(&Guid);
        if (!NT_SUCCESS(Status))
        {
            return Status;
        }
    }
    else
    {
        RtlCopyMemory(&Guid, VolumeGuid, sizeof(GUID));
    }
 
    /* Convert GUID to string */
    Status = RtlStringFromGUID(&Guid, &GuidString);
    if (!NT_SUCCESS(Status))
    {
        return Status;
    }
 
    /* Size for volume namespace, literal GUID, and null char */
    VolumeName->MaximumLength = 0x14 + 0x4C + sizeof(UNICODE_NULL);
    VolumeName->Buffer = AllocatePool(0x14 + 0x4C + sizeof(UNICODE_NULL));
    if (!VolumeName->Buffer)
    {
        Status = STATUS_INSUFFICIENT_RESOURCES;
    }
    else
    {
        RtlCopyUnicodeString(VolumeName, &Volume);
        RtlAppendUnicodeStringToString(VolumeName, &GuidString);
        VolumeName->Buffer[VolumeName->Length / sizeof(WCHAR)] = UNICODE_NULL;
        Status = STATUS_SUCCESS;
    }
 
    ExFreePoolWithTag(GuidString.Buffer, 0);
 
    return Status;
}

Referenced by MountMgrMountedDeviceArrival(), and QuerySymbolicLinkNamesFromStorage().

CreateStringWithGlobal()

NTSTATUS CreateStringWithGlobal	(	IN PUNICODE_STRING	DosName,
		OUT PUNICODE_STRING	GlobalString
	)

Definition at line 55 of file symlink.c.

{
    UNICODE_STRING IntGlobal;
 
    if (RtlPrefixUnicodeString(&DosDevices, DosName, TRUE))
    {
        /* DOS device - use DOS global */
        IntGlobal.Length = DosName->Length - DosDevices.Length + DosGlobal.Length;
        IntGlobal.MaximumLength = IntGlobal.Length + sizeof(WCHAR);
        IntGlobal.Buffer = AllocatePool(IntGlobal.MaximumLength);
        if (!IntGlobal.Buffer)
        {
            return STATUS_INSUFFICIENT_RESOURCES;
        }
 
        RtlCopyMemory(IntGlobal.Buffer, DosGlobal.Buffer, DosGlobal.Length);
        RtlCopyMemory(IntGlobal.Buffer + (DosGlobal.Length / sizeof(WCHAR)),
                      DosName->Buffer + (DosDevices.Length / sizeof(WCHAR)),
                      DosName->Length - DosDevices.Length);
        IntGlobal.Buffer[IntGlobal.Length / sizeof(WCHAR)] = UNICODE_NULL;
    }
    else if (RtlPrefixUnicodeString(&Global, DosName, TRUE))
    {
        /* Switch to DOS global */
        IntGlobal.Length = DosName->Length - Global.Length + DosGlobal.Length;
        IntGlobal.MaximumLength = IntGlobal.Length + sizeof(WCHAR);
        IntGlobal.Buffer = AllocatePool(IntGlobal.MaximumLength);
        if (!IntGlobal.Buffer)
        {
            return STATUS_INSUFFICIENT_RESOURCES;
        }
 
        RtlCopyMemory(IntGlobal.Buffer, DosGlobal.Buffer, DosGlobal.Length);
        RtlCopyMemory(IntGlobal.Buffer + (DosGlobal.Length / sizeof(WCHAR)),
                      DosName->Buffer + (Global.Length / sizeof(WCHAR)),
                      DosName->Length - Global.Length);
        IntGlobal.Buffer[IntGlobal.Length / sizeof(WCHAR)] = UNICODE_NULL;
    }
    else
    {
        /* Simply duplicate string */
        IntGlobal.Length = DosName->Length;
        IntGlobal.MaximumLength = DosName->MaximumLength;
        IntGlobal.Buffer = AllocatePool(IntGlobal.MaximumLength);
        if (!IntGlobal.Buffer)
        {
            return STATUS_INSUFFICIENT_RESOURCES;
        }
 
        RtlCopyMemory(IntGlobal.Buffer, DosName->Buffer, IntGlobal.MaximumLength);
    }
 
    /* Return string */
    GlobalString->Length = IntGlobal.Length;
    GlobalString->MaximumLength = IntGlobal.MaximumLength;
    GlobalString->Buffer = IntGlobal.Buffer;
 
    return STATUS_SUCCESS;
}

Referenced by GlobalCreateSymbolicLink(), and GlobalDeleteSymbolicLink().

DeleteSymbolicLinkNameFromMemory()

VOID DeleteSymbolicLinkNameFromMemory	(	IN PDEVICE_EXTENSION	DeviceExtension,
		IN PUNICODE_STRING	SymbolicLink,
		IN BOOLEAN	MarkOffline
	)

Definition at line 754 of file symlink.c.

{
    PLIST_ENTRY DeviceEntry, SymbolEntry;
    PDEVICE_INFORMATION DeviceInformation;
    PSYMLINK_INFORMATION SymlinkInformation;
 
    /* First of all, ensure we have devices */
    if (IsListEmpty(&(DeviceExtension->DeviceListHead)))
    {
        return;
    }
 
    /* Then, look for the symbolic name */
    for (DeviceEntry = DeviceExtension->DeviceListHead.Flink;
         DeviceEntry != &(DeviceExtension->DeviceListHead);
         DeviceEntry = DeviceEntry->Flink)
    {
        DeviceInformation = CONTAINING_RECORD(DeviceEntry, DEVICE_INFORMATION, DeviceListEntry);
 
        for (SymbolEntry = DeviceInformation->SymbolicLinksListHead.Flink;
             SymbolEntry != &(DeviceInformation->SymbolicLinksListHead);
             SymbolEntry = SymbolEntry->Flink)
        {
            SymlinkInformation = CONTAINING_RECORD(SymbolEntry, SYMLINK_INFORMATION, SymbolicLinksListEntry);
 
            /* One we have found it */
            if (RtlCompareUnicodeString(SymbolicLink, &(SymlinkInformation->Name), TRUE) == 0)
            {
                /* Check if caller just want it to be offline */
                if (MarkOffline)
                {
                    SymlinkInformation->Online = FALSE;
                }
                else
                {
                    /* If not, delete it & notify */
                    SendLinkDeleted(&(DeviceInformation->SymbolicName), SymbolicLink);
                    RemoveEntryList(&(SymlinkInformation->SymbolicLinksListEntry));
 
                    FreePool(SymlinkInformation->Name.Buffer);
                    FreePool(SymlinkInformation);
                }
 
                /* No need to go farther */
                return;
            }
        }
    }
 
    return;
}

Referenced by MountMgrDeletePoints(), and MountMgrDeletePointsDbOnly().

GlobalCreateSymbolicLink()

NTSTATUS GlobalCreateSymbolicLink	(	IN PUNICODE_STRING	DosName,
		IN PUNICODE_STRING	DeviceName
	)

Definition at line 120 of file symlink.c.

{
    NTSTATUS Status;
    UNICODE_STRING GlobalName;
 
    /* First create the global string */
    Status = CreateStringWithGlobal(DosName, &GlobalName);
    if (!NT_SUCCESS(Status))
    {
        return Status;
    }
 
    /* Then, create the symlink */
    Status = IoCreateSymbolicLink(&GlobalName, DeviceName);
 
    FreePool(GlobalName.Buffer);
 
    return Status;
}

Referenced by CreateNewDriveLetterName(), DriverEntry(), MountMgrCreatePointWorker(), MountMgrMountedDeviceArrival(), and RedirectSavedLink().

GlobalDeleteSymbolicLink()

NTSTATUS GlobalDeleteSymbolicLink

(

IN PUNICODE_STRING

DosName

)

Definition at line 145 of file symlink.c.

{
    NTSTATUS Status;
    UNICODE_STRING GlobalName;
 
    /* Recreate the string (to find the link) */
    Status = CreateStringWithGlobal(DosName, &GlobalName);
    if (!NT_SUCCESS(Status))
    {
        return Status;
    }
 
    /* And delete the link */
    Status = IoDeleteSymbolicLink(&GlobalName);
 
    FreePool(GlobalName.Buffer);
 
    return Status;
}

Referenced by MountMgrCreatePointWorker(), MountMgrDeletePoints(), MountMgrFreeMountedDeviceInfo(), MountMgrFreeSavedLink(), MountMgrMountedDeviceArrival(), MountMgrMountedDeviceRemoval(), MountMgrUnload(), and RedirectSavedLink().

IsDriveLetter()

BOOLEAN IsDriveLetter

(

PUNICODE_STRING

SymbolicName

)

Definition at line 812 of file symlink.c.

{
    WCHAR Letter, Colon;
 
    /* We must have a precise length */
    if (SymbolicName->Length != DosDevices.Length + 2 * sizeof(WCHAR))
    {
        return FALSE;
    }
 
    /* Must start with the DosDevices prefix */
    if (!RtlPrefixUnicodeString(&DosDevices, SymbolicName, TRUE))
    {
        return FALSE;
    }
 
    /* Check if letter is correct */
    Letter = SymbolicName->Buffer[DosDevices.Length / sizeof(WCHAR)];
    if ((Letter < L'A' || Letter > L'Z') && Letter != (WCHAR)-1)
    {
        return FALSE;
    }
 
    /* And finally it must end with a colon */
    Colon = SymbolicName->Buffer[DosDevices.Length / sizeof(WCHAR) + 1];
    if (Colon != L':')
    {
        return FALSE;
    }
 
    return TRUE;
}

Referenced by DeleteDriveLetterRoutine(), HasDriveLetter(), MountMgrCreatePointWorker(), MountMgrDeletePoints(), MountMgrDeletePointsDbOnly(), MountMgrMountedDeviceArrival(), MountMgrNextDriveLetterWorker(), and QuerySymbolicLinkNamesFromStorage().

MountMgrQuerySymbolicLink()

NTSTATUS MountMgrQuerySymbolicLink	(	IN PUNICODE_STRING	SymbolicName,
		IN OUT PUNICODE_STRING	LinkTarget
	)

Definition at line 849 of file symlink.c.

{
    NTSTATUS Status;
    HANDLE LinkHandle;
    OBJECT_ATTRIBUTES ObjectAttributes;
 
    /* Open the symbolic link */
    InitializeObjectAttributes(&ObjectAttributes,
                               SymbolicName,
                               OBJ_KERNEL_HANDLE | OBJ_CASE_INSENSITIVE,
                               NULL,
                               NULL);
 
    Status = ZwOpenSymbolicLinkObject(&LinkHandle,
                                      GENERIC_READ,
                                      &ObjectAttributes);
    if (!NT_SUCCESS(Status))
    {
        return Status;
    }
 
    /* Query its target */
    Status = ZwQuerySymbolicLinkObject(LinkHandle,
                                       LinkTarget,
                                       NULL);
 
    ZwClose(LinkHandle);
 
    if (!NT_SUCCESS(Status))
    {
        return Status;
    }
 
    if (LinkTarget->Length <= sizeof(WCHAR))
    {
        return Status;
    }
 
    /* If it's not finished by \, just return */
    if (LinkTarget->Buffer[LinkTarget->Length / sizeof(WCHAR) - 1] != L'\\')
    {
        return Status;
    }
 
    /* Otherwise, ensure to drop the tailing \ */
    LinkTarget->Length -= sizeof(WCHAR);
    LinkTarget->Buffer[LinkTarget->Length / sizeof(WCHAR)] = UNICODE_NULL;
 
    return Status;
}

Referenced by MountMgrVolumeMountPointCreated(), and MountMgrVolumeMountPointDeleted().

QuerySuggestedLinkName()

NTSTATUS QuerySuggestedLinkName	(	IN PUNICODE_STRING	SymbolicName,
		OUT PUNICODE_STRING	SuggestedLinkName,
		OUT PBOOLEAN	UseOnlyIfThereAreNoOtherLinks
	)

Definition at line 621 of file symlink.c.

{
    NTSTATUS Status;
    USHORT NameLength;
    PFILE_OBJECT FileObject;
    PDEVICE_OBJECT DeviceObject;
    PMOUNTDEV_SUGGESTED_LINK_NAME IoCtlSuggested;
 
    /* First, get device */
    Status = IoGetDeviceObjectPointer(SymbolicName,
                                      FILE_READ_ATTRIBUTES,
                                      &FileObject,
                                      &DeviceObject);
    if (!NT_SUCCESS(Status))
    {
        return Status;
    }
 
    /* Then, get attached device */
    DeviceObject = IoGetAttachedDeviceReference(FileObject->DeviceObject);
 
    /* Then, prepare buffer to query suggested name */
    IoCtlSuggested = AllocatePool(sizeof(MOUNTDEV_SUGGESTED_LINK_NAME));
    if (!IoCtlSuggested)
    {
        Status = STATUS_INSUFFICIENT_RESOURCES;
        goto Dereference;
    }
 
    /* Prepare request */
    Status = MountMgrSendSyncDeviceIoCtl(IOCTL_MOUNTDEV_QUERY_SUGGESTED_LINK_NAME,
                                         DeviceObject,
                                         NULL,
                                         0,
                                         IoCtlSuggested,
                                         sizeof(MOUNTDEV_SUGGESTED_LINK_NAME),
                                         FileObject);
    /* Retry with appropriate length */
    if (Status == STATUS_BUFFER_OVERFLOW)
    {
        /* Reallocate big enough buffer */
        NameLength = IoCtlSuggested->NameLength + sizeof(MOUNTDEV_SUGGESTED_LINK_NAME);
        FreePool(IoCtlSuggested);
 
        IoCtlSuggested = AllocatePool(NameLength);
        if (!IoCtlSuggested)
        {
            Status = STATUS_INSUFFICIENT_RESOURCES;
            goto Dereference;
        }
 
        /* And re-ask */
        Status = MountMgrSendSyncDeviceIoCtl(IOCTL_MOUNTDEV_QUERY_SUGGESTED_LINK_NAME,
                                             DeviceObject,
                                             NULL,
                                             0,
                                             IoCtlSuggested,
                                             NameLength,
                                             FileObject);
    }
    if (!NT_SUCCESS(Status))
        goto Release;
 
    /* Now we have suggested name, copy it */
    SuggestedLinkName->Length = IoCtlSuggested->NameLength;
    SuggestedLinkName->MaximumLength = IoCtlSuggested->NameLength + sizeof(UNICODE_NULL);
    SuggestedLinkName->Buffer = AllocatePool(IoCtlSuggested->NameLength + sizeof(UNICODE_NULL));
    if (!SuggestedLinkName->Buffer)
    {
        Status = STATUS_INSUFFICIENT_RESOURCES;
    }
    else
    {
        RtlCopyMemory(SuggestedLinkName->Buffer, IoCtlSuggested->Name, IoCtlSuggested->NameLength);
        SuggestedLinkName->Buffer[SuggestedLinkName->Length / sizeof(WCHAR)] = UNICODE_NULL;
    }
 
    /* Also return its priority */
    *UseOnlyIfThereAreNoOtherLinks = IoCtlSuggested->UseOnlyIfThereAreNoOtherLinks;
 
Release:
    FreePool(IoCtlSuggested);
 
Dereference:
    ObDereferenceObject(DeviceObject);
    ObDereferenceObject(FileObject);
 
    return Status;
}

Referenced by MountMgrMountedDeviceArrival().

QuerySymbolicLinkNamesFromStorage()

NTSTATUS QuerySymbolicLinkNamesFromStorage	(	IN PDEVICE_EXTENSION	DeviceExtension,
		IN PDEVICE_INFORMATION	DeviceInformation,
		IN PUNICODE_STRING	SuggestedLinkName,
		IN BOOLEAN	UseOnlyIfThereAreNoOtherLinks,
		OUT PUNICODE_STRING *	SymLinks,
		OUT PULONG	SymLinkCount,
		IN BOOLEAN	HasGuid,
		IN LPGUID	Guid
	)

Definition at line 451 of file symlink.c.

{
    NTSTATUS Status;
    BOOLEAN WriteNew;
    RTL_QUERY_REGISTRY_TABLE QueryTable[2];
 
    UNREFERENCED_PARAMETER(DeviceExtension);
 
    /* First of all, count links */
    RtlZeroMemory(QueryTable, sizeof(QueryTable));
    QueryTable[0].QueryRoutine = SymbolicLinkNamesFromUniqueIdCount;
    QueryTable[0].EntryContext = SymLinkCount;
    *SymLinkCount = 0;
 
    Status = RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
                                    DatabasePath,
                                    QueryTable,
                                    DeviceInformation->UniqueId,
                                    NULL);
    if (!NT_SUCCESS(Status))
    {
        *SymLinkCount = 0;
    }
 
    /* Check if we have to write a new one first */
    if (SuggestedLinkName && !IsDriveLetter(SuggestedLinkName) &&
        UseOnlyIfThereAreNoOtherLinks && *SymLinkCount == 0)
    {
        WriteNew = TRUE;
    }
    else
    {
        WriteNew = FALSE;
    }
 
    /* If has GUID, it makes one more link */
    if (HasGuid)
    {
        (*SymLinkCount)++;
    }
 
    if (WriteNew)
    {
        /* Write link */
        RtlWriteRegistryValue(RTL_REGISTRY_ABSOLUTE,
                              DatabasePath,
                              SuggestedLinkName->Buffer,
                              REG_BINARY,
                              DeviceInformation->UniqueId->UniqueId,
                              DeviceInformation->UniqueId->UniqueIdLength);
 
        /* And recount all the needed links */
        RtlZeroMemory(QueryTable, sizeof(QueryTable));
        QueryTable[0].QueryRoutine = SymbolicLinkNamesFromUniqueIdCount;
        QueryTable[0].EntryContext = SymLinkCount;
        *SymLinkCount = 0;
 
        Status = RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
                                        DatabasePath,
                                        QueryTable,
                                        DeviceInformation->UniqueId,
                                        NULL);
        if (!NT_SUCCESS(Status))
        {
            return STATUS_NOT_FOUND;
        }
    }
 
    /* Not links found? */
    if (!*SymLinkCount)
    {
        return STATUS_NOT_FOUND;
    }
 
    /* Allocate a buffer big enough to hold symlinks (table of unicode strings) */
    *SymLinks = AllocatePool(*SymLinkCount * sizeof(UNICODE_STRING));
    if (!*SymLinks)
    {
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    /* Prepare to query links */
    RtlZeroMemory(*SymLinks, *SymLinkCount * sizeof(UNICODE_STRING));
    RtlZeroMemory(QueryTable, sizeof(QueryTable));
    QueryTable[0].QueryRoutine = SymbolicLinkNamesFromUniqueIdQuery;
 
    /* No GUID? Keep it that way */
    if (!HasGuid)
    {
        QueryTable[0].EntryContext = *SymLinks;
    }
    /* Otherwise, first create volume name */
    else
    {
        Status = CreateNewVolumeName(SymLinks[0], Guid);
        if (!NT_SUCCESS(Status))
        {
            FreePool(*SymLinks);
            return Status;
        }
 
        /* Skip first link (ours) */
        QueryTable[0].EntryContext = *SymLinks + 1;
    }
 
    /* Now, query */
    RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
                           DatabasePath,
                           QueryTable,
                           DeviceInformation->UniqueId,
                           NULL);
 
    return STATUS_SUCCESS;
}

Referenced by MountMgrMountedDeviceArrival().

RedirectSavedLink()

BOOLEAN RedirectSavedLink	(	IN PSAVED_LINK_INFORMATION	SavedLinkInformation,
		IN PUNICODE_STRING	DosName,
		IN PUNICODE_STRING	NewLink
	)

Definition at line 717 of file symlink.c.

{
    PLIST_ENTRY NextEntry;
    PSYMLINK_INFORMATION SymlinkInformation;
 
    /* Find the link */
    for (NextEntry = SavedLinkInformation->SymbolicLinksListHead.Flink;
         NextEntry != &(SavedLinkInformation->SymbolicLinksListHead);
         NextEntry = NextEntry->Flink)
    {
        SymlinkInformation = CONTAINING_RECORD(NextEntry, SYMLINK_INFORMATION, SymbolicLinksListEntry);
 
        if (!RtlEqualUnicodeString(DosName, &(SymlinkInformation->Name), TRUE))
        {
            /* Delete old link */
            GlobalDeleteSymbolicLink(DosName);
            /* Set its new location */
            GlobalCreateSymbolicLink(DosName, NewLink);
 
            /* And remove it from the list (not valid any more) */
            RemoveEntryList(&(SymlinkInformation->SymbolicLinksListEntry));
            FreePool(SymlinkInformation->Name.Buffer);
            FreePool(SymlinkInformation);
 
            return TRUE;
        }
    }
 
    return FALSE;
}

Referenced by MountMgrMountedDeviceArrival().

RemoveSavedLinks()

PSAVED_LINK_INFORMATION RemoveSavedLinks	(	IN PDEVICE_EXTENSION	DeviceExtension,
		IN PMOUNTDEV_UNIQUE_ID	UniqueId
	)

Definition at line 577 of file symlink.c.

{
    PLIST_ENTRY NextEntry;
    PSAVED_LINK_INFORMATION SavedLinkInformation;
 
    /* No saved links? Easy! */
    if (IsListEmpty(&(DeviceExtension->SavedLinksListHead)))
    {
        return NULL;
    }
 
    /* Now, browse saved links */
    for (NextEntry = DeviceExtension->SavedLinksListHead.Flink;
         NextEntry != &(DeviceExtension->SavedLinksListHead);
         NextEntry = NextEntry->Flink)
    {
        SavedLinkInformation = CONTAINING_RECORD(NextEntry,
                                                 SAVED_LINK_INFORMATION,
                                                 SavedLinksListEntry);
 
        /* Find the one that matches */
        if (SavedLinkInformation->UniqueId->UniqueIdLength == UniqueId->UniqueIdLength)
        {
            if (RtlCompareMemory(SavedLinkInformation->UniqueId->UniqueId,
                                 UniqueId->UniqueId,
                                 UniqueId->UniqueIdLength) ==
                UniqueId->UniqueIdLength)
            {
                /* Remove it and return it */
                RemoveEntryList(&(SavedLinkInformation->SavedLinksListEntry));
                return SavedLinkInformation;
            }
        }
    }
 
    /* None found (none removed) */
    return NULL;
}

Referenced by MountMgrMountedDeviceArrival().

SendLinkCreated()

VOID SendLinkCreated

(

IN PUNICODE_STRING

SymbolicName

)

Definition at line 169 of file symlink.c.

{
    NTSTATUS Status;
    ULONG NameSize;
    PFILE_OBJECT FileObject;
    PMOUNTDEV_NAME Name = NULL;
    PDEVICE_OBJECT DeviceObject;
 
    /* Get the device associated with the name */
    Status = IoGetDeviceObjectPointer(SymbolicName,
                                      FILE_READ_ATTRIBUTES,
                                      &FileObject,
                                      &DeviceObject);
    if (!NT_SUCCESS(Status))
        return;
 
    /* Get attached device (will notify it) */
    DeviceObject = IoGetAttachedDeviceReference(FileObject->DeviceObject);
 
    /* NameSize is the size of the whole MOUNTDEV_NAME structure */
    NameSize = sizeof(USHORT) + SymbolicName->Length;
    Name = AllocatePool(NameSize);
    if (!Name)
        goto Cleanup;
 
    /* Initialize structure */
    Name->NameLength = SymbolicName->Length;
    RtlCopyMemory(Name->Name, SymbolicName->Buffer, SymbolicName->Length);
 
    /* Send the notification, using the new (Windows 2003+) IOCTL definition
     * (with limited access) first. If this fails, the called driver may be
     * for an older NT version, and so, we send again the notification using
     * the old (Windows 2000) IOCTL definition (with any access). */
    Status = MountMgrSendSyncDeviceIoCtl(IOCTL_MOUNTDEV_LINK_CREATED,
                                         DeviceObject,
                                         Name,
                                         NameSize,
                                         NULL,
                                         0,
                                         FileObject);
    /* This one can fail, no one matters */
    UNREFERENCED_PARAMETER(Status);
#if (NTDDI_VERSION >= NTDDI_WS03)
    /* Then, second one */
    Status = MountMgrSendSyncDeviceIoCtl(IOCTL_MOUNTDEV_LINK_CREATED_UNSECURE_DEPRECATED,
                                         DeviceObject,
                                         Name,
                                         NameSize,
                                         NULL,
                                         0,
                                         FileObject);
    UNREFERENCED_PARAMETER(Status);
#endif // (NTDDI_VERSION >= NTDDI_WS03)
 
Cleanup:
    if (Name)
        FreePool(Name);
 
    ObDereferenceObject(DeviceObject);
    ObDereferenceObject(FileObject);
 
    return;
}

Referenced by MountMgrCreatePointWorker(), and MountMgrMountedDeviceArrival().

SendLinkDeleted()

VOID SendLinkDeleted	(	IN PUNICODE_STRING	DeviceName,
		IN PUNICODE_STRING	SymbolicName
	)

Definition at line 237 of file symlink.c.

{
    NTSTATUS Status;
    ULONG NameSize;
    PFILE_OBJECT FileObject;
    PMOUNTDEV_NAME Name = NULL;
    PDEVICE_OBJECT DeviceObject;
 
    /* Get the device associated with the name */
    Status = IoGetDeviceObjectPointer(DeviceName,
                                      FILE_READ_ATTRIBUTES,
                                      &FileObject,
                                      &DeviceObject);
    if (!NT_SUCCESS(Status))
        return;
 
    /* Get attached device (will notify it) */
    DeviceObject = IoGetAttachedDeviceReference(FileObject->DeviceObject);
 
    /* NameSize is the size of the whole MOUNTDEV_NAME structure */
    NameSize = sizeof(USHORT) + SymbolicName->Length;
    Name = AllocatePool(NameSize);
    if (!Name)
        goto Cleanup;
 
    /* Initialize structure */
    Name->NameLength = SymbolicName->Length;
    RtlCopyMemory(Name->Name, SymbolicName->Buffer, SymbolicName->Length);
 
    /* Send the notification, using the new (Windows 2003+) IOCTL definition
     * (with limited access) first. If this fails, the called driver may be
     * for an older NT version, and so, we send again the notification using
     * the old (Windows 2000) IOCTL definition (with any access). */
    Status = MountMgrSendSyncDeviceIoCtl(IOCTL_MOUNTDEV_LINK_DELETED,
                                         DeviceObject,
                                         Name,
                                         NameSize,
                                         NULL,
                                         0,
                                         FileObject);
    /* This one can fail, no one matters */
    UNREFERENCED_PARAMETER(Status);
#if (NTDDI_VERSION >= NTDDI_WS03)
    /* Then, second one */
    Status = MountMgrSendSyncDeviceIoCtl(IOCTL_MOUNTDEV_LINK_DELETED_UNSECURE_DEPRECATED,
                                         DeviceObject,
                                         Name,
                                         NameSize,
                                         NULL,
                                         0,
                                         FileObject);
    UNREFERENCED_PARAMETER(Status);
#endif // (NTDDI_VERSION >= NTDDI_WS03)
 
Cleanup:
    if (Name)
        FreePool(Name);
 
    ObDereferenceObject(DeviceObject);
    ObDereferenceObject(FileObject);
 
    return;
}

Referenced by DeleteSymbolicLinkNameFromMemory().

SymbolicLinkNamesFromUniqueIdCount()

NTSTATUS NTAPI SymbolicLinkNamesFromUniqueIdCount	(	IN PWSTR	ValueName,
		IN ULONG	ValueType,
		IN PVOID	ValueData,
		IN ULONG	ValueLength,
		IN PVOID	Context,
		IN PVOID	EntryContext
	)

Definition at line 307 of file symlink.c.

{
    UNICODE_STRING ValueNameString;
    PMOUNTDEV_UNIQUE_ID UniqueId = Context;
 
    if (ValueName[0] != L'#' || ValueType != REG_BINARY ||
        (UniqueId->UniqueIdLength != ValueLength))
    {
        return STATUS_SUCCESS;
    }
 
    if (RtlCompareMemory(UniqueId->UniqueId, ValueData, ValueLength) != ValueLength)
    {
        return STATUS_SUCCESS;
    }
 
    /* That one matched, increase count */
    RtlInitUnicodeString(&ValueNameString, ValueName);
    if (ValueNameString.Length)
    {
        (*((PULONG)EntryContext))++;
    }
 
    return STATUS_SUCCESS;
}

Referenced by QuerySymbolicLinkNamesFromStorage().

SymbolicLinkNamesFromUniqueIdQuery()

NTSTATUS NTAPI SymbolicLinkNamesFromUniqueIdQuery	(	IN PWSTR	ValueName,
		IN ULONG	ValueType,
		IN PVOID	ValueData,
		IN ULONG	ValueLength,
		IN PVOID	Context,
		IN PVOID	EntryContext
	)

Definition at line 343 of file symlink.c.

{
    UNICODE_STRING ValueNameString;
    PMOUNTDEV_UNIQUE_ID UniqueId = Context;
    /* Unicode strings table */
    PUNICODE_STRING ReturnString = EntryContext;
 
    if (ValueName[0] != L'#' || ValueType != REG_BINARY ||
        (UniqueId->UniqueIdLength != ValueLength))
    {
        return STATUS_SUCCESS;
    }
 
    if (RtlCompareMemory(UniqueId->UniqueId, ValueData, ValueLength) != ValueLength)
    {
        return STATUS_SUCCESS;
    }
 
    /* Unique ID matches, let's put the symlink */
    RtlInitUnicodeString(&ValueNameString, ValueName);
    if (!ValueNameString.Length)
    {
        return STATUS_SUCCESS;
    }
 
    /* Allocate string to copy */
    ValueNameString.Buffer = AllocatePool(ValueNameString.MaximumLength);
    if (!ValueNameString.Buffer)
    {
        return STATUS_SUCCESS;
    }
 
    /* Copy */
    RtlCopyMemory(ValueNameString.Buffer, ValueName, ValueNameString.Length);
    ValueNameString.Buffer[ValueNameString.Length / sizeof(WCHAR)] = UNICODE_NULL;
 
    while (ReturnString->Length)
    {
        ReturnString++;
    }
 
    /* And return that string */
    *ReturnString = ValueNameString;
 
    return STATUS_SUCCESS;
}

Referenced by QuerySymbolicLinkNamesFromStorage().

Variable Documentation

DeviceCdRom

UNICODE_STRING DeviceCdRom = RTL_CONSTANT_STRING(L"\\Device\\CdRom")

Definition at line 44 of file symlink.c.

Referenced by CreateNewDriveLetterName(), MountMgrMountedDeviceArrival(), and MountMgrNextDriveLetterWorker().

DeviceFloppy

UNICODE_STRING DeviceFloppy = RTL_CONSTANT_STRING(L"\\Device\\Floppy")

Definition at line 43 of file symlink.c.

Referenced by CreateNewDriveLetterName(), and MountMgrNextDriveLetterWorker().

DeviceMount

UNICODE_STRING DeviceMount = RTL_CONSTANT_STRING(MOUNTMGR_DEVICE_NAME)

Definition at line 40 of file symlink.c.

Referenced by DriverEntry().

DosDevices

UNICODE_STRING DosDevices = RTL_CONSTANT_STRING(L"\\DosDevices\\")

Definition at line 42 of file symlink.c.

Referenced by CreateNewDriveLetterName(), CreateStringWithGlobal(), IsDriveLetter(), MountMgrNextDriveLetterWorker(), and ProcessSuggestedDriveLetters().

DosDevicesMount

UNICODE_STRING DosDevicesMount = RTL_CONSTANT_STRING(L"\\DosDevices\\MountPointManager")

Definition at line 41 of file symlink.c.

Referenced by DriverEntry(), and MountMgrUnload().

DosGlobal

UNICODE_STRING DosGlobal = RTL_CONSTANT_STRING(L"\\GLOBAL??\\")

Definition at line 45 of file symlink.c.

Referenced by CreateStringWithGlobal().

Global

UNICODE_STRING Global = RTL_CONSTANT_STRING(L"\\??\\")

Definition at line 46 of file symlink.c.

Referenced by co_HOOK_CallHooks(), CreateStringWithGlobal(), HandlePropertyInstances(), KdbpCmdBreakPoint(), KdbpCmdBreakPointList(), KdbpGetBreakPointInfo(), and KdbpInsertBreakPoint().

ReparseIndex

UNICODE_STRING ReparseIndex = RTL_CONSTANT_STRING(L"\\$Extend\\$Reparse:$R:$INDEX_ALLOCATION")

Definition at line 49 of file symlink.c.

Referenced by OnlineMountedVolumes(), and ReconcileThisDatabaseWithMasterWorker().

SafeVolumes

UNICODE_STRING SafeVolumes = RTL_CONSTANT_STRING(L"\\Device\\VolumesSafeForWriteAccess")

Definition at line 47 of file symlink.c.

Referenced by WorkerThread().

Volume

UNICODE_STRING Volume = RTL_CONSTANT_STRING(L"\\??\\Volume")

Definition at line 48 of file symlink.c.

Referenced by CreateNewVolumeName().