1*0b57cec5SDimitry Andric //===-- LockFileBase.cpp --------------------------------------------------===//
2*0b57cec5SDimitry Andric //
3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*0b57cec5SDimitry Andric //
7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
8*0b57cec5SDimitry Andric
9*0b57cec5SDimitry Andric #include "lldb/Host/LockFileBase.h"
10*0b57cec5SDimitry Andric
11*0b57cec5SDimitry Andric using namespace lldb;
12*0b57cec5SDimitry Andric using namespace lldb_private;
13*0b57cec5SDimitry Andric
AlreadyLocked()14*0b57cec5SDimitry Andric static Status AlreadyLocked() { return Status("Already locked"); }
15*0b57cec5SDimitry Andric
NotLocked()16*0b57cec5SDimitry Andric static Status NotLocked() { return Status("Not locked"); }
17*0b57cec5SDimitry Andric
LockFileBase(int fd)18*0b57cec5SDimitry Andric LockFileBase::LockFileBase(int fd)
19*0b57cec5SDimitry Andric : m_fd(fd), m_locked(false), m_start(0), m_len(0) {}
20*0b57cec5SDimitry Andric
IsLocked() const21*0b57cec5SDimitry Andric bool LockFileBase::IsLocked() const { return m_locked; }
22*0b57cec5SDimitry Andric
WriteLock(const uint64_t start,const uint64_t len)23*0b57cec5SDimitry Andric Status LockFileBase::WriteLock(const uint64_t start, const uint64_t len) {
24*0b57cec5SDimitry Andric return DoLock([&](const uint64_t start,
25*0b57cec5SDimitry Andric const uint64_t len) { return DoWriteLock(start, len); },
26*0b57cec5SDimitry Andric start, len);
27*0b57cec5SDimitry Andric }
28*0b57cec5SDimitry Andric
TryWriteLock(const uint64_t start,const uint64_t len)29*0b57cec5SDimitry Andric Status LockFileBase::TryWriteLock(const uint64_t start, const uint64_t len) {
30*0b57cec5SDimitry Andric return DoLock([&](const uint64_t start,
31*0b57cec5SDimitry Andric const uint64_t len) { return DoTryWriteLock(start, len); },
32*0b57cec5SDimitry Andric start, len);
33*0b57cec5SDimitry Andric }
34*0b57cec5SDimitry Andric
ReadLock(const uint64_t start,const uint64_t len)35*0b57cec5SDimitry Andric Status LockFileBase::ReadLock(const uint64_t start, const uint64_t len) {
36*0b57cec5SDimitry Andric return DoLock([&](const uint64_t start,
37*0b57cec5SDimitry Andric const uint64_t len) { return DoReadLock(start, len); },
38*0b57cec5SDimitry Andric start, len);
39*0b57cec5SDimitry Andric }
40*0b57cec5SDimitry Andric
TryReadLock(const uint64_t start,const uint64_t len)41*0b57cec5SDimitry Andric Status LockFileBase::TryReadLock(const uint64_t start, const uint64_t len) {
42*0b57cec5SDimitry Andric return DoLock([&](const uint64_t start,
43*0b57cec5SDimitry Andric const uint64_t len) { return DoTryReadLock(start, len); },
44*0b57cec5SDimitry Andric start, len);
45*0b57cec5SDimitry Andric }
46*0b57cec5SDimitry Andric
Unlock()47*0b57cec5SDimitry Andric Status LockFileBase::Unlock() {
48*0b57cec5SDimitry Andric if (!IsLocked())
49*0b57cec5SDimitry Andric return NotLocked();
50*0b57cec5SDimitry Andric
51*0b57cec5SDimitry Andric const auto error = DoUnlock();
52*0b57cec5SDimitry Andric if (error.Success()) {
53*0b57cec5SDimitry Andric m_locked = false;
54*0b57cec5SDimitry Andric m_start = 0;
55*0b57cec5SDimitry Andric m_len = 0;
56*0b57cec5SDimitry Andric }
57*0b57cec5SDimitry Andric return error;
58*0b57cec5SDimitry Andric }
59*0b57cec5SDimitry Andric
IsValidFile() const60*0b57cec5SDimitry Andric bool LockFileBase::IsValidFile() const { return m_fd != -1; }
61*0b57cec5SDimitry Andric
DoLock(const Locker & locker,const uint64_t start,const uint64_t len)62*0b57cec5SDimitry Andric Status LockFileBase::DoLock(const Locker &locker, const uint64_t start,
63*0b57cec5SDimitry Andric const uint64_t len) {
64*0b57cec5SDimitry Andric if (!IsValidFile())
65*0b57cec5SDimitry Andric return Status("File is invalid");
66*0b57cec5SDimitry Andric
67*0b57cec5SDimitry Andric if (IsLocked())
68*0b57cec5SDimitry Andric return AlreadyLocked();
69*0b57cec5SDimitry Andric
70*0b57cec5SDimitry Andric const auto error = locker(start, len);
71*0b57cec5SDimitry Andric if (error.Success()) {
72*0b57cec5SDimitry Andric m_locked = true;
73*0b57cec5SDimitry Andric m_start = start;
74*0b57cec5SDimitry Andric m_len = len;
75*0b57cec5SDimitry Andric }
76*0b57cec5SDimitry Andric
77*0b57cec5SDimitry Andric return error;
78*0b57cec5SDimitry Andric }
79*0b57cec5SDimitry Andric