RTR3DECL(int) RTFileChangeLock(RTFILE hFile, unsigned fLock, int64_t offLock, uint64_t cbLock)
{
Assert(offLock >= 0);
/* Check arguments. */
if (fLock & ~RTFILE_LOCK_MASK)
{
AssertMsgFailed(("Invalid fLock=%08X\n", fLock));
return VERR_INVALID_PARAMETER;
}
/* Remove old lock. */
int rc = RTFileUnlock(hFile, offLock, cbLock);
if (RT_FAILURE(rc))
return rc;
/* Set new lock. */
rc = RTFileLock(hFile, fLock, offLock, cbLock);
if (RT_SUCCESS(rc))
return rc;
/* Try to restore old lock. */
unsigned fLockOld = (fLock & RTFILE_LOCK_WRITE) ? fLock & ~RTFILE_LOCK_WRITE : fLock | RTFILE_LOCK_WRITE;
rc = RTFileLock(hFile, fLockOld, offLock, cbLock);
if (RT_SUCCESS(rc))
return VERR_FILE_LOCK_VIOLATION;
else
return VERR_FILE_LOCK_LOST;
}
int main()
{
RTR3InitExeNoArguments(0);
RTPrintf("tstFileLock: TESTING\n");
RTFILE File;
int rc = RTFileOpen(&File, "tstLock.tst", RTFILE_O_READWRITE | RTFILE_O_OPEN | RTFILE_O_DENY_NONE);
RTPrintf("File open: rc=%Rrc\n", rc);
if (RT_FAILURE(rc))
{
if (rc != VERR_FILE_NOT_FOUND && rc != VERR_OPEN_FAILED)
{
RTPrintf("FATAL\n");
return 1;
}
rc = RTFileOpen(&File, "tstLock.tst", RTFILE_O_READWRITE | RTFILE_O_CREATE | RTFILE_O_DENY_NONE);
RTPrintf("File create: rc=%Rrc\n", rc);
if (RT_FAILURE(rc))
{
RTPrintf("FATAL\n");
return 2;
}
fRun = true;
}
/* grow file a little */
rc = RTFileSetSize(File, fRun ? 2048 : 20480);
RTPrintf("File size: rc=%Rrc\n", rc);
int buf;
/* read test. */
rc = RTFileRead(File, &buf, sizeof(buf), NULL);
RTPrintf("Read: rc=%Rrc\n", rc);
/* write test. */
rc = RTFileWrite(File, achTest1, strlen(achTest1), NULL);
RTPrintf("Write: rc=%Rrc\n", rc);
/* lock: read, non-blocking. */
rc = RTFileLock(File, RTFILE_LOCK_READ | RTFILE_LOCK_IMMEDIATELY, 0, _4G);
RTPrintf("Lock: read, non-blocking, rc=%Rrc\n", rc);
bool fl = RT_SUCCESS(rc);
/* read test. */
rc = RTFileRead(File, &buf, sizeof(buf), NULL);
RTPrintf("Read: rc=%Rrc\n", rc);
/* write test. */
rc = RTFileWrite(File, achTest2, strlen(achTest2), NULL);
RTPrintf("Write: rc=%Rrc\n", rc);
RTPrintf("Lock test will change in three seconds\n");
for (int i = 0; i < 3; i++)
{
RTThreadSleep(1000);
RTPrintf(".");
}
RTPrintf("\n");
/* change lock: write, non-blocking. */
rc = RTFileLock(File, RTFILE_LOCK_WRITE | RTFILE_LOCK_IMMEDIATELY, 0, _4G);
RTPrintf("Change lock: write, non-blocking, rc=%Rrc\n", rc);
RTPrintf("Test will unlock in three seconds\n");
for (int i = 0; i < 3; i++)
{
RTThreadSleep(1000);
RTPrintf(".");
}
RTPrintf("\n");
/* remove lock. */
if (fl)
{
fl = false;
rc = RTFileUnlock(File, 0, _4G);
RTPrintf("Unlock: rc=%Rrc\n", rc);
RTPrintf("Write test will lock in three seconds\n");
for (int i = 0; i < 3; i++)
{
RTThreadSleep(1000);
RTPrintf(".");
}
RTPrintf("\n");
}
/* lock: write, non-blocking. */
rc = RTFileLock(File, RTFILE_LOCK_WRITE | RTFILE_LOCK_IMMEDIATELY, 0, _4G);
RTPrintf("Lock: write, non-blocking, rc=%Rrc\n", rc);
fl = RT_SUCCESS(rc);
/* grow file test */
rc = RTFileSetSize(File, fRun ? 2048 : 20480);
RTPrintf("File size: rc=%Rrc\n", rc);
/* read test. */
rc = RTFileRead(File, &buf, sizeof(buf), NULL);
RTPrintf("Read: rc=%Rrc\n", rc);
/* write test. */
rc = RTFileWrite(File, achTest3, strlen(achTest3), NULL);
RTPrintf("Write: rc=%Rrc\n", rc);
RTPrintf("Continuing to next test in three seconds\n");
for (int i = 0; i < 3; i++)
{
RTThreadSleep(1000);
RTPrintf(".");
}
RTPrintf("\n");
RTFileClose(File);
RTFileDelete("tstLock.tst");
RTPrintf("tstFileLock: I've no recollection of this testcase succeeding or not, sorry.\n");
return 0;
}
