Changeset 77681 in vbox for trunk/src/VBox

Timestamp:

Mar 13, 2019 3:26:52 PM (6 years ago)

Author:

vboxsync

Message:

IPRT: Adding an extended file open API that returns the action taken (opened, created, replaced, truncated) for addressing tickref:9276. Windows implementation only thus far.

Location:

trunk/src/VBox/Runtime

Files:

• r3/win/fileio-win.cpp (modified) (7 diffs)
• testcase/Makefile.kmk (modified) (2 diffs)
• testcase/tstRTFileOpenEx-1.cpp (added)

trunk/src/VBox/Runtime/r3/win/fileio-win.cpp

@@ -166 +166 @@
 RTR3DECL(int) RTFileOpen(PRTFILE pFile, const char *pszFilename, uint64_t fOpen)
 {
+    return RTFileOpenEx(pszFilename, fOpen, pFile, NULL);
+}
+
+
+RTDECL(int) RTFileOpenEx(const char *pszFilename, uint64_t fOpen, PRTFILE phFile, PRTFILEACTION penmActionTaken)
+{
     /*
      * Validate input.
      */
-    if (!pFile)
-    {
-        AssertMsgFailed(("Invalid pFile\n"));
-        return VERR_INVALID_PARAMETER;
-    }
-    *pFile = NIL_RTFILE;
-    if (!pszFilename)
-    {
-        AssertMsgFailed(("Invalid pszFilename\n"));
-        return VERR_INVALID_PARAMETER;
-    }
+    AssertReturn(phFile, VERR_INVALID_PARAMETER);
+    *phFile = NIL_RTFILE;
+    if (penmActionTaken)
+        *penmActionTaken = RTFILEACTION_INVALID;
+    AssertReturn(pszFilename, VERR_INVALID_PARAMETER);
 
     /*
@@ -208 +208 @@
             break;
         default:
-            AssertMsgFailed(("Impossible fOpen=%#llx\n", fOpen));
-            return VERR_INVALID_PARAMETER;
+            AssertMsgFailedReturn(("Impossible fOpen=%#llx\n", fOpen), VERR_INVALID_FLAGS);
     }
 
@@ -236 +235 @@
             RT_FALL_THRU();
         default:
-            AssertMsgFailed(("Impossible fOpen=%#llx\n", fOpen));
-            return VERR_INVALID_PARAMETER;
+            AssertMsgFailedReturn(("Impossible fOpen=%#llx\n", fOpen), VERR_INVALID_FLAGS);
     }
     if (dwCreationDisposition == TRUNCATE_EXISTING)
@@ -257 +255 @@
                 case RTFILE_O_READWRITE:     dwDesiredAccess |= FILE_READ_ATTRIBUTES | FILE_WRITE_ATTRIBUTES | SYNCHRONIZE; break;
                 default:
-                    AssertMsgFailed(("Impossible fOpen=%#llx\n", fOpen));
-                    return VERR_INVALID_PARAMETER;
+                    AssertMsgFailedReturn(("Impossible fOpen=%#llx\n", fOpen), VERR_INVALID_FLAGS);
             }
     }
@@ -275 +272 @@
         case RTFILE_O_DENY_NOT_DELETE | RTFILE_O_DENY_READWRITE:dwShareMode = FILE_SHARE_DELETE; break;
         default:
-            AssertMsgFailed(("Impossible fOpen=%#llx\n", fOpen));
-            return VERR_INVALID_PARAMETER;
+            AssertMsgFailedReturn(("Impossible fOpen=%#llx\n", fOpen), VERR_INVALID_FLAGS);
     }
 
@@ -315 +311 @@
                                    dwFlagsAndAttributes,
                                    NULL);
+        DWORD const dwErr = GetLastError();
         if (hFile != INVALID_HANDLE_VALUE)
         {
-            bool fCreated = dwCreationDisposition == CREATE_ALWAYS
-                         || dwCreationDisposition == CREATE_NEW
-                         || (dwCreationDisposition == OPEN_ALWAYS && GetLastError() == 0);
-
             /*
-             * Turn off indexing of directory through Windows Indexing Service.
+             * Calculate the action taken value.
              */
-            if (    fCreated
-                &&  (fOpen & RTFILE_O_NOT_CONTENT_INDEXED))
-            {
+            RTFILEACTION enmActionTaken;
+            switch (dwCreationDisposition)
+            {
+                case CREATE_NEW:
+                    enmActionTaken = RTFILEACTION_CREATED;
+                    break;
+                case CREATE_ALWAYS:
+                    AssertMsg(dwErr == ERROR_ALREADY_EXISTS || dwErr == NO_ERROR, ("%u\n", dwErr));
+                    enmActionTaken = dwErr == ERROR_ALREADY_EXISTS ? RTFILEACTION_REPLACED : RTFILEACTION_CREATED;
+                    break;
+                case OPEN_EXISTING:
+                    enmActionTaken = RTFILEACTION_OPENED;
+                    break;
+                case OPEN_ALWAYS:
+                    AssertMsg(dwErr == ERROR_ALREADY_EXISTS || dwErr == NO_ERROR, ("%u\n", dwErr));
+                    enmActionTaken = dwErr == ERROR_ALREADY_EXISTS ? RTFILEACTION_OPENED : RTFILEACTION_CREATED;
+                    break;
+                case TRUNCATE_EXISTING:
+                    enmActionTaken = RTFILEACTION_TRUNCATED;
+                    break;
+                default:
+                    AssertMsgFailed(("%d %#x\n", dwCreationDisposition, dwCreationDisposition));
+                    enmActionTaken = RTFILEACTION_INVALID;
+                    break;
+            }
+
+            /*
+             * Turn off indexing of directory through Windows Indexing Service if
+             * we created a new file or replaced an existing one.
+             */
+            if (   (fOpen & RTFILE_O_NOT_CONTENT_INDEXED)
+                && (   enmActionTaken == RTFILEACTION_CREATED
+                    || enmActionTaken == RTFILEACTION_REPLACED) )
+            {
+                /** @todo there must be a way to do this via the handle! */
                 if (!SetFileAttributesW(pwszFilename, FILE_ATTRIBUTE_NOT_CONTENT_INDEXED))
                     rc = RTErrConvertFromWin32(GetLastError());
             }
             /*
-             * Do we need to truncate the file?
+             * If RTFILEACTION_OPENED, we may need to truncate the file.
              */
-            else if (    !fCreated
-                     &&     (fOpen & (RTFILE_O_TRUNCATE | RTFILE_O_ACTION_MASK))
-                         == (RTFILE_O_TRUNCATE | RTFILE_O_OPEN_CREATE))
-            {
-                if (!SetEndOfFile(hFile))
+            else if (  (fOpen & (RTFILE_O_TRUNCATE | RTFILE_O_ACTION_MASK)) == (RTFILE_O_TRUNCATE | RTFILE_O_OPEN_CREATE)
+                     && enmActionTaken == RTFILEACTION_OPENED)
+            {
+                if (SetEndOfFile(hFile))
+                    enmActionTaken = RTFILEACTION_TRUNCATED;
+                else
                     rc = RTErrConvertFromWin32(GetLastError());
             }
+            if (penmActionTaken)
+                *penmActionTaken = enmActionTaken;
             if (RT_SUCCESS(rc))
             {
-                *pFile = (RTFILE)hFile;
-                Assert((HANDLE)*pFile == hFile);
+                *phFile = (RTFILE)hFile;
+                Assert((HANDLE)*phFile == hFile);
                 RTPathWinFree(pwszFilename);
                 return VINF_SUCCESS;
@@ -351 +379 @@
         }
         else
-            rc = RTErrConvertFromWin32(GetLastError());
+        {
+            if (   penmActionTaken
+                && dwCreationDisposition == CREATE_NEW
+                && dwErr == ERROR_FILE_EXISTS)
+                *penmActionTaken = RTFILEACTION_ALREADY_EXISTS;
+            rc = RTErrConvertFromWin32(dwErr);
+        }
         RTPathWinFree(pwszFilename);
     }

trunk/src/VBox/Runtime/testcase/Makefile.kmk

@@ -160 +160 @@
         tstRTNtPath-1 \
         ntGetTimerResolution \
-        tstRTDarwinMachKernel
+        tstRTDarwinMachKernel \
+        tstRTFileOpenEx-1
 
 PROGRAMS.linux += \
@@ -364 +365 @@
 tstRTFileModeStringToFlags_SOURCES = tstRTFileModeStringToFlags.cpp
 
+tstRTFileOpenEx-1_TEMPLATE = VBOXR3TSTEXE
+tstRTFileOpenEx-1_SOURCES = tstRTFileOpenEx-1.cpp
+
 tstFileAppendWin-1_TEMPLATE = VBOXR3TSTEXE
 tstFileAppendWin-1_SOURCES = tstFileAppendWin-1.cpp