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
|
|
);
|
|
}
|