ipc_queue/
position.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
/* Copyright (c) Fortanix, Inc.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

use super::*;
use std::sync::atomic::Ordering;

/// `PositionMonitor<T>` records the current read/write positions of a queue.
///
/// The `PositionMonitor` works by keeping track how many times the ring buffer wrapped around,
/// and the current offsets that are used.
/// The former (how many times the ring buffer wrapped around) is kept in an `AtomicU64`
/// that is updated each time a value is read.
/// The latter (current offsets) is kept by looking at the `Fifo` directly.
///
/// Even though a queue is comprised of a limited number of slots
/// arranged as a ring buffer, we can assign a position to each value written/
/// read to/from the queue. This is useful in case we want to know whether or
/// not a particular value written to the queue has been read.
pub struct PositionMonitor<T: 'static> {
    read_epoch: Arc<AtomicU64>,
    fifo: Fifo<T>,
}

/// A read position in a queue.
pub struct ReadPosition(u64);

/// A write position in a queue.
pub struct WritePosition(u64);

impl<T> PositionMonitor<T> {
    pub (crate) fn new(read_epoch: Arc<AtomicU64>,fifo: Fifo<T>) -> PositionMonitor<T> {
        PositionMonitor {
            read_epoch,
            fifo,
        }
    }

    pub fn read_position(&self) -> ReadPosition {
        let current = self.fifo.current_offsets(Ordering::Relaxed);
        let read_epoch = self.read_epoch.load(Ordering::Relaxed);
        ReadPosition((read_epoch << 32) | (current.read_offset() as u64))
    }

    pub fn write_position(&self) -> WritePosition {
        let current = self.fifo.current_offsets(Ordering::Relaxed);
        let mut write_epoch = self.read_epoch.load(Ordering::Relaxed);
        // Write epoch keeps track of how many times the write offset wrapped around
        // the ring buffer. Write epochs are not tracked separately, only read epoch are.
        // We know, however, that objects are always written to the buffer first.
        // So, the high bit used in the write and read offsets tell us whether writes
        // already wrapped around the ring buffer, while reads have not yet.
        if current.read_high_bit() != current.write_high_bit() {
            write_epoch += 1;
        }
        WritePosition((write_epoch << 32) | (current.write_offset() as u64))
    }
}

impl<T> Clone for PositionMonitor<T> {
    fn clone(&self) -> Self {
        Self {
            read_epoch: self.read_epoch.clone(),
            fifo: self.fifo.clone(),
        }
    }
}

impl ReadPosition {
    /// A `WritePosition` can be compared to a `ReadPosition` **correctly** if
    /// the ring buffer wrapped around at most 2³¹ (2 to the power of 31) times.
    ///
    /// Returns `None` if the read position and the write position cannot be compared,
    /// `Some(true)` if the read position is strictly higher than the write position,
    /// `Some(false)` otherwise
    pub fn is_past(&self, write: &WritePosition) -> Option<bool> {
        let (read, write) = (self.0, write.0);

        let hr = read & (1 << 63);
        let hw = write & (1 << 63);
        if hr == hw {
            Some(read > write)
        } else {
            None
        }
    }
}