1 // Copyright 2021 The Pigweed Authors
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License"); you may not
4 // use this file except in compliance with the License. You may obtain a copy of
5 // the License at
6 //
7 // https://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
11 // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12 // License for the specific language governing permissions and limitations under
13 // the License.
14
15 #include "pw_sync/timed_mutex.h"
16
17 #include <algorithm>
18
19 #include "RTOS.h"
20 #include "pw_assert/check.h"
21 #include "pw_chrono/system_clock.h"
22 #include "pw_chrono_embos/system_clock_constants.h"
23 #include "pw_interrupt/context.h"
24
25 using pw::chrono::SystemClock;
26
27 namespace pw::sync {
28
try_lock_for(SystemClock::duration timeout)29 bool TimedMutex::try_lock_for(SystemClock::duration timeout) {
30 // Enforce the pw::sync::TimedMutex IRQ contract.
31 PW_DCHECK(!interrupt::InInterruptContext());
32
33 // Use non-blocking try_lock for negative and zero length durations.
34 if (timeout <= SystemClock::duration::zero()) {
35 return try_lock();
36 }
37
38 // In case the timeout is too long for us to express through the native
39 // embOS API, we repeatedly wait with shorter durations. Note that on a tick
40 // based kernel we cannot tell how far along we are on the current tick, ergo
41 // we add one whole tick to the final duration. However, this also means that
42 // the loop must ensure that timeout + 1 is less than the max timeout.
43 constexpr SystemClock::duration kMaxTimeoutMinusOne =
44 pw::chrono::embos::kMaxTimeout - SystemClock::duration(1);
45 while (timeout > kMaxTimeoutMinusOne) {
46 const int lock_count = OS_UseTimed(
47 &native_handle(), static_cast<OS_TIME>(kMaxTimeoutMinusOne.count()));
48 if (lock_count != 0) {
49 PW_DCHECK_UINT_EQ(1, lock_count, "Recursive locking is not permitted");
50 return true;
51 }
52 timeout -= kMaxTimeoutMinusOne;
53 }
54 // On a tick based kernel we cannot tell how far along we are on the current
55 // tick, ergo we add one whole tick to the final duration.
56 const int lock_count =
57 OS_UseTimed(&native_handle(), static_cast<OS_TIME>(timeout.count() + 1));
58 PW_DCHECK_UINT_LE(1, lock_count, "Recursive locking is not permitted");
59 return lock_count == 1;
60 }
61
62 } // namespace pw::sync
63