Atomic Operations and Invariants
In this section, we finally come to some concurrency related constructs.
Concurrency in Pulse is built around two concepts:
Atomic operations: operations that are guaranteed to be executed in a single-step of computation without interruption by other threads.
Invariants: named predicates that are enforced to be true at all times. Atomic operations can make use of invariants, assuming they are true in the current state, and enforced to be true again once the atomic step concludes.
Based on this, and in conjunction with all the other separation logic constructs that we’ve learned about so far, notably the use of ghost state, Pulse enables proofs of concurrent programs.
Atomic Operations
We’ve learned so far about two kinds of Pulse computations:
General purpose, partially correct computations, with the
stt
computation typeGhost computations, proven totally correct, and enforced to be computationally irrelevant with the
stt_ghost
computation type.
Pulse offers a third kind of computation, atomic computations, with
the stt_atomic
computation type. Here is the signature of
read_atomic
and write_atomic
from Pulse.Lib.Reference
:
atomic
fn read_atomic (r:ref U32.t) (#n:erased U32.t) (#p:perm)
requires pts_to r #p n
returns x:U32.t
ensures pts_to r #p n ** pure (reveal n == x)
atomic
fn write_atomic (r:ref U32.t) (x:U32.t) (#n:erased U32.t)
requires pts_to r n
ensures pts_to r x
The atomic
annotation on these functions claims that reading and
writing 32-bit integers can be done in a single atomic step of
computation.
This is an assumption about the target architecture on which a Pulse program is executed. It may be that on some machines, 32-bit values cannot be read or written atomically. So, when using atomic operations, you should be careful to check that its safe to assume that these operations truly are atomic.
Pulse also provides a way for you to declare that other operations are atomic, e.g., maybe your machine supports 64-bit or 128-bit atomic operations—you can program the semantics of these operations in F* and add them to Pulse, marking them as atomic.
Sometimes, particularly at higher order, you will see atomic computations described by the computation type below:
val stt_atomic (t:Type) (i:inames) (pre:vprop) (post:t -> vprop)
: Type u#4
Like stt_ghost
, atomic computations are total and live in universe
u#4
. As such, you cannot store an atomic function in the state,
i.e., ref (unit -> stt_atomic t i p q)
is not a well-formed type.
Atomic computations and ghost computations are also indexed by
i:inames
, where inames
is a set of invariant names. We’ll
learn about these next.
Invariants
In Pulse.Lib.Core
, we have the following types:
[@@erasable]
val iref : Type0
val inv (i:iref) (p:vprop) : vprop
Think of inv i p
as a predicate asserting that p
is true in
the current state and all future states of the program. Every
invariant has a name, i:iref
, though, the name is only relevant in
specifications, i.e., it is erasable.
A closely related type is iname
:
val iname : eqtype
let inames = erased (FStar.Set.set iname)
Every iref
can be turned into an iname
, with the function
iname_of (i:iref): GTot iname
.
Invariants are duplicable, i.e., from inv i p
one can prove inv
i p ** inv i p
, as shown by type of Pulse.Lib.Core.dup_inv
below:
val dup_inv (i:iref) (p:vprop)
: stt_ghost unit emp_inames (inv i p) (fun _ -> inv i p ** inv i p)
Boxable predicates
Pulse’s language of predicates, i.e., the type vprop
, is
stratified. The type boxable
is a refinement of vprop
, defined
as shown below in Pulse.Lib.Core
let boxable = v:vprop { is_big v }
That is, certain vprops
, i.e., those that satisfy is_big
, are
boxable
predicates. All the predicates that we have encountered so
far are boxable, except for the inv i p
predicate. For example,
pts_to x v
is boxable; exists* x. p x
is boxable if p x
is
boxable; etc. However inv i p
is not boxable.
Why does this matter? It turns out that PulseCore, the logic on which
Pulse is built, only allows turning boxable predicates into
invariants. That is, while one can build an invariant such as inv i
(exists* v. pts_to x v)
, one cannot nest invariants, i.e., there
is no meaningful way to construct an instance of inv i (inv j p)
.
This restriction is a fundamental limitation of PulseCore: invariants
cannot mention other invariants. In more technical terms, invariants
are predicative. One might wonder whether this limitation is
significant: after all, why might one want to construct an invariant
that states that some p
is already an invariant? It turns out that
such predicates, although not very common, are useful and the
inability to nest invariants in Pulse makes some styles of proofs
awkward or perhaps even impossible. Nevertheless, forcing invariants
to be predicative gives Pulse a simple foundational model in PulseCore
in terms of a standard, predicative, dependently typed logic.
Let’s look next at how to turn a boxable predicate into an invariant.
Creating an invariant and boxable predicates
Let’s start by looking at how to create an invariant.
First, let’s define a predicate owns x
, to mean that we hold
full-permission on x
.
let owns (x:ref U32.t) : v:vprop { is_big v }= exists* v. pts_to x v
Notice the type annotation on owns
claims that it is boxable
,
and indeed F*’s refinement type checker automatically proves that it
is.
Now, if we can currently prove pts_to r x
then we can turn it into
an invariant inv i (owns r)
, as shown below.
ghost
fn create_invariant (r:ref U32.t) (v:erased U32.t)
requires pts_to r v
returns i:iref
ensures inv i (owns r)
{
fold owns;
new_invariant (owns r)
}
Importantly, when we turn pts_to r x
into inv i (owns r)
, we
lose ownership of pts_to r x
. Remember, once we have inv i
(owns r)
, Pulse’s logic aims to prove that owns r
remains true
always. If we were allowed to retain pts_to r x
, while also
creating an inv i (owns r)
, we can clearly break the invariant,
e.g., by freeing r
.
Note
A tip: When using an inv i p
, it’s a good idea to make sure
that p
is a user-defined predicate. For example, one might
think to just write inv i (exists* v. pts_to x v)
instead of
defining an auxiliary predicate for inv i (owns r)
. However, the
some of the proof obligations produced by the Pulse checker are
harder for the SMT solver to prove if you don’t use the auxiliary
predicate and you may start to see odd failures. This is something
we’re working to improve. In the meantime, use an auxiliary
predicate.
Note, if one were to try to allocate an invariant for a non-boxable predicate, typechecking fails, as shown in the example below:
ghost
fn create_non_boxable_inv (p:vprop)
requires p
returns i:iref
ensures inv i p
{
new_invariant p;
}
failing with an error pointing to the source location of the
refinement precondition, is_big
, at the call to new_invariant
.
- Assertion failed
- The SMT solver could not prove the query. Use --query_stats for more details.
- See also ../../../lib/pulse/lib/Pulse.Lib.Core.fsti(536,29-536,37)
As you can see, although the language does not prevent you from
writing inv i p
for any predicate p
, the only way to allocate
an instance of inv i p
is by provable that p
is
boxable
. This design is convenient since the onus of proving that
a predicate is boxable is only placed at the allocation site of the
invariant—uses of invariants do not need to worry about the
distinction between boxable
and general vprops
.
Opening an invariant
Now that we’ve allocated an inv i (owns r)
, what can we do with it?
As we said earlier, one can make use of the owns r
in an atomic
computation, so long as we restore it at the end of the atomic
step.
The with_invariants
construct gives us access to the invariant
within the scope of at most one atomic step, preceded or succeeded by
as many ghost or unobservable steps as needed.
The general form of with_invariants
is as follows, to “open”
invariants i_1
to i_k
in the scope of e
.
with_invariants i_1 ... i_k
returns x:t
ensures post
{ e }
In many cases, the returns
and ensures
annotations are
omitted, since it can be inferred.
This is syntactic sugar for the following nest:
with_invariants i_1 {
...
with_invariants i_k
returns x:t
ensures post
{ e }
...
}
Here’s the rule for opening a single invariant inv i p
using
with_invariant i { e }
is as follows:
i
must have typeiref
andinv i p
must be provable in the current context, for somep:vprop
e
must have the typestt_atomic t j (p ** r) (fun x -> p ** s x)
. [1] That is,e
requires and restores the invariantp
, while also transformingr
tos x
, all in at most one atomic step. Further, thename_of_inv i
must not be in the setj
.with_invariants i { e }
has typestt_atomic t (add_inv i j) (inv i p ** r) (fun x -> inv i p ** s x)
. That is,e
gets to usep
for a step, and from the caller’s perspective, the context was transformed fromr
tos
, while the use ofp
is hidden.Pay attention to the
add_inv i j
index onwith_invariants
:stt_atomic
(orstt_ghost
) computation is indexed by the names of all the invariants that it may open.
Let’s look at a few examples to see how with_invariants
works.
Updating a reference
In the example below, given inv i (owns r)
, we can atomically
update a reference with a pre- and postcondition of emp
.
atomic
fn update_ref_atomic (r:ref U32.t) (i:iref) (v:U32.t)
requires inv i (owns r)
ensures inv i (owns r)
opens (singleton i)
{
with_invariants i { //owns r
unfold owns; //ghost step; exists* u. pts_to r u
write_atomic r v; //atomic step; pts_to r v
fold owns; //ghost step; owns r
}
}
At the start of the
with_invariants
scope, we haveowns r
in the context.The ghost step
unfold owns
unfolds it to its definition.Then, we do a single atomic action,
write_atomic
.And follow it up with a
fold owns
, another ghost step.The block within
with_invariants i
has typestt_atomic unit emp_inames (owns r ** emp) (fun _ -> owns r ** emp)
Since we opened the invariant
i
, the type ofupdate_ref_atomic
records this in theopens (singleton i)
annotation; equivalently, the type isstt_atomic unit (singleton i) (inv i (owns r)) (fun _ -> inv i (owns r))
. When theopens
annotation is omitted, it defaults toemp_inames
, the empty set of invariant names.
Double opening is unsound
To see why we have to track the names of the opened invariants,
consider the example below. If we opened the same invariant twice
within the same scope, then it’s easy to prove False
:
[@@expect_failure]
```pulse
fn double_open_bad (r:ref U32.t) (i:inv (owns r))
requires emp
ensures pure False
{
with_invariants i {
with_invariants i {
unfold owns;
unfold owns;
pts_to_dup_impossible r;
fold owns;
fold owns
}
}
}
```
Here, we open the invariants i
twice and get owns r ** owns r
,
or more than full permission to r
—from this, it is easy to build
a contradiction.
Subsuming atomic computations
Atomic computations can be silently converted to regular, stt
computations, while forgetting which invariants they opened. For
example, update_ref
below is not marked atomic, so its type
doesn’t record which invariants were opened internally.
fn update_ref (r:ref U32.t) (i:iref) (v:U32.t)
requires inv i (owns r)
ensures inv i (owns r)
{
with_invariants i { //owns r
unfold owns; //ghost step; exists* u. pts_to r u
write_atomic r v; //atomic step; pts_to r v
fold owns; //ghost step; owns r
}
}
This is okay, since a non-atomic computation can never appear within a
with_invariants
block—so, there’s no fear of an stt
computation causing an unsound double opening. Attempting to use a
non-atomic computation in a with_invariants
block produces an
error, as shown below.
[@@expect_failure]
```pulse
fn update_ref_fail (r:ref U32.t) (i:iref) (v:U32.t)
requires inv i (owns r)
ensures inv i (owns r)
{
with_invariants i {
unfold owns;
r := v; //not atomic
fold owns;
}
}
```
- This computation is not atomic nor ghost. `with_invariants`
blocks can only contain atomic computations.