* Created 01/11/98
*/
-#include <ddk/ntddk.h>
-#include <windows.h>
+/* INCLUDES ******************************************************************/
+#include <k32.h>
+#define NDEBUG
+#include <kernel32/kernel32.h>
+
+
+/* FUNCTIONS *****************************************************************/
+
+/*
+ * @implemented
+ */
WINBOOL
STDCALL
DefineDosDeviceA(
LPCSTR lpTargetPath
)
{
- ULONG i;
+ UNICODE_STRING DeviceNameU;
+ UNICODE_STRING TargetPathU;
+ BOOL Result;
- WCHAR DeviceNameW[MAX_PATH];
- WCHAR TargetPathW[MAX_PATH];
+ if (!RtlCreateUnicodeStringFromAsciiz (&DeviceNameU,
+ (LPSTR)lpDeviceName))
+ {
+ SetLastError (ERROR_NOT_ENOUGH_MEMORY);
+ return 0;
+ }
- i = 0;
- while ((*lpDeviceName)!=0 && i < MAX_PATH)
- {
- DeviceNameW[i] = *lpDeviceName;
- lpDeviceName++;
- i++;
- }
- DeviceNameW[i] = 0;
-
- i = 0;
- while ((*lpTargetPath)!=0 && i < MAX_PATH)
- {
- TargetPathW[i] = *lpTargetPath;
- lpTargetPath++;
- i++;
- }
- TargetPathW[i] = 0;
- return DefineDosDeviceW(dwFlags,DeviceNameW,TargetPathW);
-}
+ if (!RtlCreateUnicodeStringFromAsciiz (&TargetPathU,
+ (LPSTR)lpTargetPath))
+ {
+ RtlFreeHeap (RtlGetProcessHeap (),
+ 0,
+ DeviceNameU.Buffer);
+ SetLastError (ERROR_NOT_ENOUGH_MEMORY);
+ return 0;
+ }
+ Result = DefineDosDeviceW (dwFlags,
+ DeviceNameU.Buffer,
+ TargetPathU.Buffer);
+ RtlFreeHeap (RtlGetProcessHeap (),
+ 0,
+ TargetPathU.Buffer);
+ RtlFreeHeap (RtlGetProcessHeap (),
+ 0,
+ DeviceNameU.Buffer);
-DWORD
-STDCALL
-QueryDosDeviceA(
- LPCSTR lpDeviceName,
- LPSTR lpTargetPath,
- DWORD ucchMax
- )
-{
- ULONG i;
-
- WCHAR DeviceNameW[MAX_PATH];
- WCHAR TargetPathW[MAX_PATH];
-
-
-
- i = 0;
- while ((*lpDeviceName)!=0 && i < MAX_PATH)
- {
- DeviceNameW[i] = *lpDeviceName;
- lpDeviceName++;
- i++;
- }
- DeviceNameW[i] = 0;
-
- i = 0;
- while ((*lpTargetPath)!=0 && i < MAX_PATH)
- {
- TargetPathW[i] = *lpTargetPath;
- lpTargetPath++;
- i++;
- }
- TargetPathW[i] = 0;
- return QueryDosDeviceW(DeviceNameW,TargetPathW,ucchMax);
+ return Result;
}
+/*
+ * @unimplemented
+ */
WINBOOL
STDCALL
DefineDosDeviceW(
return FALSE;
}
+
+/*
+ * @implemented
+ */
+DWORD
+STDCALL
+QueryDosDeviceA(
+ LPCSTR lpDeviceName,
+ LPSTR lpTargetPath,
+ DWORD ucchMax
+ )
+{
+ UNICODE_STRING DeviceNameU;
+ UNICODE_STRING TargetPathU;
+ ANSI_STRING TargetPathA;
+ DWORD Length;
+
+ if (!RtlCreateUnicodeStringFromAsciiz (&DeviceNameU,
+ (LPSTR)lpDeviceName))
+ {
+ SetLastError (ERROR_NOT_ENOUGH_MEMORY);
+ return 0;
+ }
+
+ TargetPathU.Length = 0;
+ TargetPathU.MaximumLength = (USHORT)ucchMax * sizeof(WCHAR);
+ TargetPathU.Buffer = RtlAllocateHeap (RtlGetProcessHeap (),
+ 0,
+ TargetPathU.MaximumLength);
+ if (TargetPathU.Buffer == NULL)
+ {
+ SetLastError (ERROR_NOT_ENOUGH_MEMORY);
+ return 0;
+ }
+
+ Length = QueryDosDeviceW (DeviceNameU.Buffer,
+ TargetPathU.Buffer,
+ ucchMax);
+ if (Length != 0)
+ {
+ TargetPathU.Length = Length * sizeof(WCHAR);
+
+ TargetPathA.Length = 0;
+ TargetPathA.MaximumLength = (USHORT)ucchMax;
+ TargetPathA.Buffer = lpTargetPath;
+
+ RtlUnicodeStringToAnsiString (&TargetPathA,
+ &TargetPathU,
+ FALSE);
+
+ DPRINT ("TargetPathU: '%wZ'\n", &TargetPathU);
+ DPRINT ("TargetPathA: '%Z'\n", &TargetPathA);
+ }
+
+ RtlFreeHeap (RtlGetProcessHeap (),
+ 0,
+ TargetPathU.Buffer);
+ RtlFreeHeap (RtlGetProcessHeap (),
+ 0,
+ DeviceNameU.Buffer);
+
+ return Length;
+}
+
+
+/*
+ * @implemented
+ */
DWORD
STDCALL
QueryDosDeviceW(
DWORD ucchMax
)
{
- return FALSE;
+ PDIRECTORY_BASIC_INFORMATION DirInfo;
+ OBJECT_ATTRIBUTES ObjectAttributes;
+ UNICODE_STRING UnicodeString;
+ HANDLE DirectoryHandle;
+ HANDLE DeviceHandle;
+ ULONG ReturnLength;
+ ULONG NameLength;
+ ULONG Length;
+ ULONG Context;
+ BOOLEAN RestartScan;
+ NTSTATUS Status;
+ UCHAR Buffer[512];
+ PWSTR Ptr;
+
+ /* Open the '\??' directory */
+ RtlInitUnicodeString (&UnicodeString,
+ L"\\??");
+ InitializeObjectAttributes (&ObjectAttributes,
+ &UnicodeString,
+ OBJ_CASE_INSENSITIVE,
+ NULL,
+ NULL);
+ Status = NtOpenDirectoryObject (&DirectoryHandle,
+ DIRECTORY_QUERY,
+ &ObjectAttributes);
+ if (!NT_SUCCESS (Status))
+ {
+ DPRINT ("NtOpenDirectoryObject() failed (Status %lx)\n", Status);
+ SetLastErrorByStatus (Status);
+ return 0;
+ }
+
+ Length = 0;
+
+ if (lpDeviceName != NULL)
+ {
+ /* Open the lpDeviceName link object */
+ RtlInitUnicodeString (&UnicodeString,
+ (PWSTR)lpDeviceName);
+ InitializeObjectAttributes (&ObjectAttributes,
+ &UnicodeString,
+ OBJ_CASE_INSENSITIVE,
+ DirectoryHandle,
+ NULL);
+ Status = NtOpenSymbolicLinkObject (&DeviceHandle,
+ SYMBOLIC_LINK_QUERY,
+ &ObjectAttributes);
+ if (!NT_SUCCESS (Status))
+ {
+ DPRINT ("NtOpenSymbolicLinkObject() failed (Status %lx)\n", Status);
+ NtClose (DirectoryHandle);
+ SetLastErrorByStatus (Status);
+ return 0;
+ }
+
+ /* Query link target */
+ UnicodeString.Length = 0;
+ UnicodeString.MaximumLength = (USHORT)ucchMax * sizeof(WCHAR);
+ UnicodeString.Buffer = lpTargetPath;
+
+ ReturnLength = 0;
+ Status = NtQuerySymbolicLinkObject (DeviceHandle,
+ &UnicodeString,
+ &ReturnLength);
+ NtClose (DeviceHandle);
+ NtClose (DirectoryHandle);
+ if (!NT_SUCCESS (Status))
+ {
+ DPRINT ("NtQuerySymbolicLinkObject() failed (Status %lx)\n", Status);
+ SetLastErrorByStatus (Status);
+ return 0;
+ }
+
+ DPRINT ("ReturnLength: %lu\n", ReturnLength);
+ DPRINT ("TargetLength: %hu\n", UnicodeString.Length);
+ DPRINT ("Target: '%wZ'\n", &UnicodeString);
+
+ Length = ReturnLength / sizeof(WCHAR);
+ if (Length < ucchMax)
+ {
+ /* Append null-charcter */
+ lpTargetPath[Length] = UNICODE_NULL;
+ Length++;
+ }
+ else
+ {
+ DPRINT ("Buffer is too small\n");
+ SetLastErrorByStatus (STATUS_BUFFER_TOO_SMALL);
+ return 0;
+ }
+ }
+ else
+ {
+ RestartScan = TRUE;
+ Context = 0;
+ Ptr = lpTargetPath;
+ DirInfo = (PDIRECTORY_BASIC_INFORMATION)Buffer;
+
+ while (TRUE)
+ {
+ Status = NtQueryDirectoryObject (DirectoryHandle,
+ Buffer,
+ sizeof (Buffer),
+ TRUE,
+ RestartScan,
+ &Context,
+ &ReturnLength);
+ if (!NT_SUCCESS(Status))
+ {
+ if (Status == STATUS_NO_MORE_ENTRIES)
+ {
+ /* Terminate the buffer */
+ *Ptr = UNICODE_NULL;
+ Length++;
+
+ Status = STATUS_SUCCESS;
+ }
+ else
+ {
+ Length = 0;
+ }
+ SetLastErrorByStatus (Status);
+ break;
+ }
+
+ if (!wcscmp (DirInfo->ObjectTypeName.Buffer, L"SymbolicLink"))
+ {
+ DPRINT ("Name: '%wZ'\n", &DirInfo->ObjectName);
+
+ NameLength = DirInfo->ObjectName.Length / sizeof(WCHAR);
+ if (Length + NameLength + 1 >= ucchMax)
+ {
+ Length = 0;
+ SetLastErrorByStatus (STATUS_BUFFER_TOO_SMALL);
+ break;
+ }
+
+ memcpy (Ptr,
+ DirInfo->ObjectName.Buffer,
+ DirInfo->ObjectName.Length);
+ Ptr += NameLength;
+ Length += NameLength;
+ *Ptr = UNICODE_NULL;
+ Ptr++;
+ Length++;
+ }
+
+ RestartScan = FALSE;
+ }
+
+ NtClose (DirectoryHandle);
+ }
+
+ return Length;
}
/* EOF */
git://git.jankratochvil.net / reactos.git / blobdiff