Changeset 3485 for trunk/src/lib

Timestamp:

Sep 21, 2020, 2:25:08 PM (5 years ago)

Author:

bird

Message:

lib/nt/ntstat.h|c: Forgot to commit birdStatModeOnly the other day.

Location:

trunk/src/lib/nt

Files:

• ntstat.c (modified) (1 diff)
• ntstat.h (modified) (1 diff)

trunk/src/lib/nt/ntstat.c

@@ -1004 +1004 @@
 }
 
-
-
+/**
+ * Special function for getting the file mode.
+ */
+int birdStatModeOnly(const char *pszPath, unsigned __int16 *pMode, int fFollowLink)
+{
+    /*
+     * Convert the path and call NtQueryFullAttributesFile.
+     */
+    MY_UNICODE_STRING  NtPath;
+
+    birdResolveImports();
+    if (birdDosToNtPath(pszPath, &NtPath) == 0)
+    {
+        MY_OBJECT_ATTRIBUTES                ObjAttr;
+        MY_FILE_BASIC_INFORMATION           Info;
+        MY_NTSTATUS                         rcNt;
+
+        memset(&Info, 0xfe, sizeof(Info));
+
+        MyInitializeObjectAttributes(&ObjAttr, &NtPath, OBJ_CASE_INSENSITIVE, NULL /*hRoot*/, NULL /*pSecAttr*/);
+        rcNt = g_pfnNtQueryAttributesFile(&ObjAttr, &Info);
+
+        if (MY_NT_SUCCESS(rcNt))
+        {
+            unsigned __int8 isdirsymlink = 0;
+            unsigned __int8 ismountpoint = 0;
+            *pMode = birdFileInfoToMode(Info.FileAttributes, 0, pszPath, NtPath.Buffer, NtPath.Length,
+                                        &isdirsymlink, &ismountpoint);
+
+            /* Do the trailing slash check. */
+            if (   (Info.FileAttributes & FILE_ATTRIBUTE_DIRECTORY)
+                || !birdIsPathDirSpec(pszPath))
+            {
+                if (   !(Info.FileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)
+                    || !fFollowLink)
+                {
+                    birdFreeNtPath(&NtPath);
+                    return 0;
+                }
+
+                /* Fallback on birdStatOnlyInternal to follow the reparse point.  */
+                if (!birdStatOnlyInternal(pszPath, fFollowLink, &Info))
+                {
+                    *pMode = birdFileInfoToMode(Info.FileAttributes, 0, pszPath, NtPath.Buffer, NtPath.Length,
+                                                &isdirsymlink, &ismountpoint);
+                    birdFreeNtPath(&NtPath);
+                    return 0;
+                }
+            }
+            else
+                errno = ENOTDIR;
+        }
+        else
+            birdSetErrnoFromNt(rcNt);
+        birdFreeNtPath(&NtPath);
+    }
+    return -1;
+}
+
+

trunk/src/lib/nt/ntstat.h

@@ -86 +86 @@
 int birdStatOnFdJustSize(int fd, __int64 *pcbFile);
 int birdStatModTimeOnly(const char *pszPath, BirdTimeSpec_T *pTimeSpec, int fFollowLink);
+int birdStatModeOnly(const char *pszPath, unsigned __int16 *pMode, int fFollowLink);
 #ifdef ___nt_ntstuff_h
 int  birdStatHandle(HANDLE hFile, BirdStat_T *pStat, const char *pszPath);