mirror of
https://github.com/reactos/CMake.git
synced 2024-12-14 15:19:39 +00:00
1f289095f9
The file lock functionality added in commit v3.2.0-rc1~297^2~1 (file: Add LOCK subcommand to do file and directory locking, 2014-11-26) forgot to close the lock file descriptors. Eventually it was possible to run out of file descriptors and locks could not longer be acquired. Fix this by closing the file descriptor or handle when we are done with it. Also set the member back to the initial value from the constructor to leave everything in a consistent state (useful for debugging). Co-Author: Ruslan Baratov <ruslan_baratov@yahoo.com>
130 lines
2.9 KiB
C++
130 lines
2.9 KiB
C++
/*============================================================================
|
|
CMake - Cross Platform Makefile Generator
|
|
Copyright 2014 Ruslan Baratov
|
|
|
|
Distributed under the OSI-approved BSD License (the "License");
|
|
see accompanying file Copyright.txt for details.
|
|
|
|
This software is distributed WITHOUT ANY WARRANTY; without even the
|
|
implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
|
See the License for more information.
|
|
============================================================================*/
|
|
|
|
#include "cmFileLock.h"
|
|
|
|
#include <windows.h> // CreateFileW
|
|
#include "cmSystemTools.h"
|
|
|
|
cmFileLock::cmFileLock(): File(INVALID_HANDLE_VALUE)
|
|
{
|
|
}
|
|
|
|
cmFileLockResult cmFileLock::Release()
|
|
{
|
|
if (this->Filename.empty())
|
|
{
|
|
return cmFileLockResult::MakeOk();
|
|
}
|
|
const unsigned long len = static_cast<unsigned long>(-1);
|
|
static OVERLAPPED overlapped;
|
|
const DWORD reserved = 0;
|
|
const BOOL unlockResult = UnlockFileEx(
|
|
File,
|
|
reserved,
|
|
len,
|
|
len,
|
|
&overlapped
|
|
);
|
|
|
|
this->Filename = "";
|
|
|
|
CloseHandle(this->File);
|
|
this->File = INVALID_HANDLE_VALUE;
|
|
|
|
if (unlockResult)
|
|
{
|
|
return cmFileLockResult::MakeOk();
|
|
}
|
|
else
|
|
{
|
|
return cmFileLockResult::MakeSystem();
|
|
}
|
|
}
|
|
|
|
cmFileLockResult cmFileLock::OpenFile()
|
|
{
|
|
const DWORD access = GENERIC_READ | GENERIC_WRITE;
|
|
const DWORD shareMode = FILE_SHARE_READ | FILE_SHARE_WRITE;
|
|
const PSECURITY_ATTRIBUTES security = NULL;
|
|
const DWORD attr = 0;
|
|
const HANDLE templ = NULL;
|
|
this->File = CreateFileW(
|
|
cmSystemTools::ConvertToWindowsExtendedPath(this->Filename).c_str(),
|
|
access,
|
|
shareMode,
|
|
security,
|
|
OPEN_EXISTING,
|
|
attr,
|
|
templ
|
|
);
|
|
if (this->File == INVALID_HANDLE_VALUE)
|
|
{
|
|
return cmFileLockResult::MakeSystem();
|
|
}
|
|
else
|
|
{
|
|
return cmFileLockResult::MakeOk();
|
|
}
|
|
}
|
|
|
|
cmFileLockResult cmFileLock::LockWithoutTimeout()
|
|
{
|
|
if (!this->LockFile(LOCKFILE_EXCLUSIVE_LOCK))
|
|
{
|
|
return cmFileLockResult::MakeSystem();
|
|
}
|
|
else
|
|
{
|
|
return cmFileLockResult::MakeOk();
|
|
}
|
|
}
|
|
|
|
cmFileLockResult cmFileLock::LockWithTimeout(unsigned long seconds)
|
|
{
|
|
const DWORD flags = LOCKFILE_EXCLUSIVE_LOCK | LOCKFILE_FAIL_IMMEDIATELY;
|
|
while (true)
|
|
{
|
|
const BOOL result = this->LockFile(flags);
|
|
if (result)
|
|
{
|
|
return cmFileLockResult::MakeOk();
|
|
}
|
|
const DWORD error = GetLastError();
|
|
if (error != ERROR_LOCK_VIOLATION)
|
|
{
|
|
return cmFileLockResult::MakeSystem();
|
|
}
|
|
if (seconds == 0)
|
|
{
|
|
return cmFileLockResult::MakeTimeout();
|
|
}
|
|
--seconds;
|
|
cmSystemTools::Delay(1000);
|
|
}
|
|
}
|
|
|
|
BOOL cmFileLock::LockFile(DWORD flags)
|
|
{
|
|
const DWORD reserved = 0;
|
|
const unsigned long len = static_cast<unsigned long>(-1);
|
|
static OVERLAPPED overlapped;
|
|
return LockFileEx(
|
|
this->File,
|
|
flags,
|
|
reserved,
|
|
len,
|
|
len,
|
|
&overlapped
|
|
);
|
|
}
|