Changeset 2416 for trunk/src/kash/shfile.c

Timestamp:

Sep 14, 2010, 2:30:30 AM (15 years ago)

Author:

bird

Message:

kash: implemented opendir/readdir/closedir for windows (NT). Fixed windows forking.

File:

• trunk/src/kash/shfile.c (modified) (9 diffs)

trunk/src/kash/shfile.c

@@ -39 +39 @@
 #  define PIPE_BUF 512
 # endif
+# include <ntstatus.h>
+# define WIN32_NO_STATUS
 # include <Windows.h>
+# if !defined(_WIN32_WINNT)
+#  define _WIN32_WINNT 0x0502 /* Windows Server 2003 */
+# endif
+# include <winternl.h> //NTSTATUS
 #else
 # include <unistd.h>
@@ -86 +92 @@
 
 # define MY_ObjectBasicInformation 0
-typedef LONG (NTAPI * PFN_NtQueryObject)(HANDLE, int, void *, size_t, size_t *);
-
-typedef struct MY_OBJECT_BASIC_INFORMATION
+# define MY_FileNamesInformation 12
+
+typedef struct
 {
     ULONG           Attributes;
@@ -103 +109 @@
 } MY_OBJECT_BASIC_INFORMATION;
 
+#if 0
+typedef struct
+{
+    union
+    {
+        LONG    Status;
+        PVOID   Pointer;
+    };
+    ULONG_PTR   Information;
+} MY_IO_STATUS_BLOCK;
+#else
+typedef IO_STATUS_BLOCK MY_IO_STATUS_BLOCK;
+#endif
+typedef MY_IO_STATUS_BLOCK *PMY_IO_STATUS_BLOCK;
+
+typedef struct
+{
+    ULONG   NextEntryOffset;
+    ULONG   FileIndex;
+    ULONG   FileNameLength;
+    WCHAR   FileName[1];
+} MY_FILE_NAMES_INFORMATION, *PMY_FILE_NAMES_INFORMATION;
+
+typedef NTSTATUS (NTAPI * PFN_NtQueryObject)(HANDLE, int, void *, size_t, size_t *);
+typedef NTSTATUS (NTAPI * PFN_NtQueryDirectoryFile)(HANDLE, HANDLE, void *,  void *, PMY_IO_STATUS_BLOCK, void *,
+                                                    ULONG, int, int, PUNICODE_STRING, int);
+typedef NTSTATUS (NTAPI * PFN_RtlUnicodeStringToAnsiString)(PANSI_STRING, PCUNICODE_STRING, int);
+
+
+#endif /* K_OS_WINDOWS */
+
+
+/*******************************************************************************
+*   Global Variables                                                           *
+*******************************************************************************/
+#if K_OS == K_OS_WINDOWS
+static int                              g_shfile_globals_initialized = 0;
+static PFN_NtQueryObject                g_pfnNtQueryObject = NULL;
+static PFN_NtQueryDirectoryFile         g_pfnNtQueryDirectoryFile = NULL;
+static PFN_RtlUnicodeStringToAnsiString g_pfnRtlUnicodeStringToAnsiString = NULL;
 #endif /* K_OS_WINDOWS */
 
@@ -387 +433 @@
 }
 
+/**
+ * Converts an NT status code to errno,
+ * assigning it to errno.
+ *
+ * @returns -1
+ * @param   rcNt        The NT status code.
+ */
+static int shfile_nt2errno(NTSTATUS rcNt)
+{
+    switch (rcNt)
+    {
+        default:                            errno = EINVAL; break;
+    }
+    return -1;
+}
+
 DWORD shfile_query_handle_access_mask(HANDLE h, PACCESS_MASK pMask)
 {
-    static PFN_NtQueryObject        s_pfnNtQueryObject = NULL;
     MY_OBJECT_BASIC_INFORMATION     BasicInfo;
-    LONG                            Status;
-
-    if (!s_pfnNtQueryObject)
-    {
-        s_pfnNtQueryObject = (PFN_NtQueryObject)GetProcAddress(GetModuleHandle("NTDLL"), "NtQueryObject");
-        if (!s_pfnNtQueryObject)
-            return ERROR_NOT_SUPPORTED;
-    }
-
-    Status = s_pfnNtQueryObject(h, MY_ObjectBasicInformation, &BasicInfo, sizeof(BasicInfo), NULL);
-    if (Status >= 0)
+    NTSTATUS                        rcNt;
+
+    if (!g_pfnNtQueryObject)
+        return ERROR_NOT_SUPPORTED;
+
+    rcNt = g_pfnNtQueryObject(h, MY_ObjectBasicInformation, &BasicInfo, sizeof(BasicInfo), NULL);
+    if (rcNt >= 0)
     {
         *pMask = BasicInfo.GrantedAccess;
         return NO_ERROR;
     }
-    if (Status != STATUS_INVALID_HANDLE)
+    if (rcNt != STATUS_INVALID_HANDLE)
         return ERROR_GEN_FAILURE;
     return ERROR_INVALID_HANDLE;
@@ -414 +471 @@
 
 #endif /* SHFILE_IN_USE */
+
+/**
+ * Initializes the global variables in this file.
+ */
+static void shfile_init_globals(void)
+{
+#if K_OS == K_OS_WINDOWS
+    if (!g_shfile_globals_initialized)
+    {
+        HMODULE hNtDll = GetModuleHandle("NTDLL");
+        g_pfnNtQueryObject                = (PFN_NtQueryObject)       GetProcAddress(hNtDll, "NtQueryObject");
+        g_pfnNtQueryDirectoryFile         = (PFN_NtQueryDirectoryFile)GetProcAddress(hNtDll, "NtQueryDirectoryFile");
+        g_pfnRtlUnicodeStringToAnsiString = (PFN_RtlUnicodeStringToAnsiString)GetProcAddress(hNtDll, "RtlUnicodeStringToAnsiString");
+        if (   !g_pfnRtlUnicodeStringToAnsiString
+            || !g_pfnNtQueryDirectoryFile)
+        {
+            /* fatal error */
+        }
+        g_shfile_globals_initialized = 1;
+    }
+#endif
+}
 
 /**
@@ -426 +505 @@
 {
     int rc;
+
+    shfile_init_globals();
 
     pfdtab->cwd  = NULL;
@@ -1417 +1498 @@
 
 
+
 shdir *shfile_opendir(shfdtab *pfdtab, const char *dir)
 {
 #ifdef SHFILE_IN_USE
-    errno = ENOSYS;
-    return NULL;
+    shdir  *pdir = NULL;
+
+    if (g_pfnNtQueryDirectoryFile)
+    {
+        char abspath[SHFILE_MAX_PATH];
+        if (shfile_make_path(pfdtab, dir, &abspath[0]) == 0)
+        {
+            HANDLE              hFile;
+            SECURITY_ATTRIBUTES SecurityAttributes;
+
+            SecurityAttributes.nLength = sizeof(SecurityAttributes);
+            SecurityAttributes.lpSecurityDescriptor = NULL;
+            SecurityAttributes.bInheritHandle = FALSE;
+
+            hFile = CreateFileA(abspath,
+                                GENERIC_READ,
+                                FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
+                                &SecurityAttributes,
+                                OPEN_ALWAYS,
+                                FILE_ATTRIBUTE_DIRECTORY | FILE_FLAG_BACKUP_SEMANTICS,
+                                NULL /* hTemplateFile */);
+            if (hFile != INVALID_HANDLE_VALUE)
+            {
+                pdir = (shdir *)sh_malloc(shthread_get_shell(), sizeof(*pdir));
+                if (pdir)
+                {
+                    pdir->pfdtab = pfdtab;
+                    pdir->native = hFile;
+                    pdir->off    = ~(size_t)0;
+                }
+                else
+                    CloseHandle(hFile);
+            }
+            else
+                shfile_dos2errno(GetLastError());
+        }
+    }
+    else
+        errno = ENOSYS;
+    return pdir;
 #else
     return (shdir *)opendir(dir);
@@ -1430 +1550 @@
 {
 #ifdef SHFILE_IN_USE
-    errno = ENOSYS;
+    if (pdir)
+    {
+        NTSTATUS rcNt;
+
+        if (   pdir->off == ~(size_t)0
+            || pdir->off + sizeof(MY_FILE_NAMES_INFORMATION) >= pdir->cb)
+        {
+            MY_IO_STATUS_BLOCK Ios;
+
+            memset(&Ios, 0, sizeof(Ios));
+            rcNt = g_pfnNtQueryDirectoryFile(pdir->native,
+                                             NULL /*Event*/,
+                                             NULL /*ApcRoutine*/,
+                                             NULL /*ApcContext*/,
+                                             &Ios,
+                                             &pdir->buf[0],
+                                             sizeof(pdir->buf),
+                                             MY_FileNamesInformation,
+                                             FALSE /*ReturnSingleEntry*/,
+                                             NULL /*FileName*/,
+                                             pdir->off == ~(size_t)0 /*RestartScan*/);
+            if (rcNt >= 0 && rcNt != STATUS_PENDING)
+            {
+                pdir->cb  = Ios.Information;
+                pdir->off = 0;
+            }
+            else if (rcNt == STATUS_NO_MORE_FILES)
+                errno = 0; /* wrong? */
+            else
+                shfile_nt2errno(rcNt);
+        }
+
+        if (   pdir->off != ~(size_t)0
+            && pdir->off + sizeof(MY_FILE_NAMES_INFORMATION) <= pdir->cb)
+        {
+            PMY_FILE_NAMES_INFORMATION  pcur = (PMY_FILE_NAMES_INFORMATION)&pdir->buf[pdir->off];
+            ANSI_STRING                 astr;
+            UNICODE_STRING              ustr;
+
+            astr.Length = astr.MaximumLength = sizeof(pdir->ent.name);
+            astr.Buffer = &pdir->ent.name[0];
+
+            ustr.Length = ustr.MaximumLength = pcur->FileNameLength < ~(USHORT)0 ? (USHORT)pcur->FileNameLength : ~(USHORT)0;
+            ustr.Buffer = &pcur->FileName[0];
+
+            rcNt = g_pfnRtlUnicodeStringToAnsiString(&astr, &ustr, 0/*AllocateDestinationString*/);
+            if (rcNt < 0)
+                sprintf(pdir->ent.name, "conversion-failed-%08x-rcNt=%08x-len=%u",
+                        pcur->FileIndex, rcNt, pcur->FileNameLength);
+            if (pcur->NextEntryOffset)
+                pdir->off += pcur->NextEntryOffset;
+            else
+                pdir->off = pdir->cb;
+            return &pdir->ent;
+        }
+    }
+    else
+        errno = EINVAL;
     return NULL;
 #else
@@ -1441 +1618 @@
 {
 #ifdef SHFILE_IN_USE
-    errno = ENOSYS;
+    if (pdir)
+    {
+        CloseHandle(pdir->native);
+        pdir->pfdtab = NULL;
+        pdir->native = INVALID_HANDLE_VALUE;
+        sh_free(shthread_get_shell(), pdir);
+    }
+    else
+        errno = EINVAL;
 #else
     closedir((DIR *)pdir);