source/Host/windows/LockFileWindows.cpp

   1 //===-- LockFileWindows.cpp -------------------------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "lldb/Host/windows/LockFileWindows.h"
  11
  12 #include <io.h>
  13
  14 using namespace lldb;
  15 using namespace lldb_private;
  16
  17 namespace
  18 {
  19
  20 Error fileLock (HANDLE file_handle, DWORD flags, const uint64_t start, const uint64_t len)
  21 {
  22     if (start != 0)
  23         return Error ("Non-zero start lock regions are not supported");
  24
  25     OVERLAPPED overlapped = {0};
  26
  27     if (!::LockFileEx (file_handle, flags, 0, len, 0, &overlapped) && ::GetLastError () != ERROR_IO_PENDING)
  28         return Error (::GetLastError (), eErrorTypeWin32);
  29
  30     DWORD bytes;
  31     if (!::GetOverlappedResult (file_handle, &overlapped, &bytes, TRUE))
  32         return Error (::GetLastError (), eErrorTypeWin32);
  33
  34     return Error ();
  35 }
  36
  37 }  // namespace
  38
  39 LockFileWindows::LockFileWindows (int fd)
  40     : LockFileBase (fd),
  41       m_file (reinterpret_cast<HANDLE> (_get_osfhandle (fd)))
  42 {
  43 }
  44
  45 LockFileWindows::~LockFileWindows ()
  46 {
  47     Unlock ();
  48 }
  49
  50 bool
  51 LockFileWindows::IsValidFile () const
  52 {
  53     return LockFileBase::IsValidFile() && m_file != INVALID_HANDLE_VALUE;
  54 }
  55
  56 Error
  57 LockFileWindows::DoWriteLock (const uint64_t start, const uint64_t len)
  58 {
  59     return fileLock (m_file, LOCKFILE_EXCLUSIVE_LOCK, start, len);
  60 }
  61
  62 Error
  63 LockFileWindows::DoTryWriteLock (const uint64_t start, const uint64_t len)
  64 {
  65     return fileLock (m_file, LOCKFILE_EXCLUSIVE_LOCK | LOCKFILE_FAIL_IMMEDIATELY, start, len);
  66 }
  67
  68 Error
  69 LockFileWindows::DoReadLock (const uint64_t start, const uint64_t len)
  70 {
  71     return fileLock (m_file, 0, start, len);
  72 }
  73
  74 Error
  75 LockFileWindows::DoTryReadLock (const uint64_t start, const uint64_t len)
  76 {
  77     return fileLock (m_file, LOCKFILE_FAIL_IMMEDIATELY, start, len);
  78 }
  79
  80 Error
  81 LockFileWindows::DoUnlock ()
  82 {
  83     OVERLAPPED overlapped = {0};
  84
  85     if (!::UnlockFileEx (m_file, 0, m_len, 0, &overlapped) && ::GetLastError () != ERROR_IO_PENDING)
  86         return Error (::GetLastError (), eErrorTypeWin32);
  87
  88     DWORD bytes;
  89     if (!::GetOverlappedResult (m_file, &overlapped, &bytes, TRUE))
  90         return Error (::GetLastError (), eErrorTypeWin32);
  91
  92     return Error ();
  93 }