X-Git-Url: https://git.ralfj.de/web.git/blobdiff_plain/741b7e20adf1fd28ef45a2440ea27c54608f015a..b107841fbce5708c8a0e472a47a29747e411d843:/ralf/_drafts/stacked-borrows-implementation.md diff --git a/ralf/_drafts/stacked-borrows-implementation.md b/ralf/_drafts/stacked-borrows-implementation.md index 5179e57..ac3624d 100644 --- a/ralf/_drafts/stacked-borrows-implementation.md +++ b/ralf/_drafts/stacked-borrows-implementation.md @@ -34,9 +34,8 @@ now. Ready? Let's get started. I hope you brought some time, because this is a rather lengthy post. If you are not interested in a detailed description of -Stacked Borrows, you can skip most of this post and go right to -[section 4](#4-differences-to-the-original-proposal). If you only want to know -how to help, jump to [section 6](#6-evaluation-and-how-you-can-help). +Stacked Borrows, you can skip most of this post and go right to [section 4]. If +you only want to know how to help, jump to [section 6]. ## 1 Enforcing Uniqueness @@ -66,7 +65,7 @@ this exercise is to explain *why* we have undefined behavior here *without* referring to the borrow checker, because we want to have rules that also work for unsafe code. -To be able to do this, we have to pretend our machine has two thing which real +To be able to do this, we have to pretend our machine has two things which real CPUs do not have. This is an example of adding "shadow state" or "instrumented state" to the "virtual machine" that we [use to specify Rust]({% post_url 2017-06-06-MIR-semantics %}). This is not an uncommon approach, often times @@ -160,8 +159,8 @@ have to improve that one): ## 2 Enabling Sharing -If we just had unique pointers, Rust would be a rather dull language. Lucky -enough, there are also two ways to have shared access to a location: Through +If we just had unique pointers, Rust would be a rather dull language. Luckily +enough, there are also two ways to have shared access to a location: through shared references (safely), and through raw pointers (unsafely). Moreover, shared references *sometimes* (but not when they point to an `UnsafeCell`) assert an additional guarantee: Their destination is read-only. @@ -325,6 +324,7 @@ fn demo2() { {% endhighlight %} ### 2.2 Dereferencing a Pointer +[section 2.2]: #22-dereferencing-a-pointer As we have seen, we consider the tag of a pointer already when dereferencing it, before any memory access happens. The operation on a dereference never mutates @@ -341,39 +341,44 @@ Notice that on a dereference, we have *both* a tag at the pointer *and* the type of a pointer, and the two might not agree which we do not always want to rule out (we might have raw or shared pointers with a unique tag, for example). -The following checks are done on every pointer dereference: +The following checks are done on every pointer dereference, for every location +covered by the pointer (`size_of_val` tells us how many bytes the pointer +covers): 1. If this is a raw pointer, do nothing and reset the tag used for the access to `Shr(None)`. Raw accesses are checked as little as possible. 2. If this is a unique reference and the tag is `Shr(Some(_))`, that's an error. 3. If the tag is `Uniq`, make sure there is a matching `Uniq` item with the same - ID on the stack of every location this reference points to (the size is - determine with `size_of_val`). + ID on the stack. 4. If the tag is `Shr(None)`, make sure that either the location is frozen or - else there is a `Shr` item on the stack of every location. -5. If the tag is `Shr(Some(t))`, then the check depends on whether a location is - inside an `UnsafeCell` or not, according to the type of the reference. + else there is a `Shr` item on the stack. +5. If the tag is `Shr(Some(t))`, then the check depends on whether the location + is inside an `UnsafeCell` or not, according to the type of the reference. - Locations outside `UnsafeCell` must have `frozen_since` set to `t` or an older timestamp. - `UnsafeCell` locations must either be frozen or else have a `Shr` item in their stack (same check as if the tag had no timestamp). ### 2.3 Accessing Memory +[section 2.3]: #23-accessing-memory On an actual memory access, we know the tag of the pointer that was used to access (unless it was a raw pointer, in which case the tag we see is `Shr(None)`), and we know whether we are reading from or writing to the current -location. We perform the following operations: +location. We perform the following operations on all locations affected by the +access: 1. If the location is frozen and this is a read access, nothing happens. (even if the tag is `Uniq`). -2. Unfreeze the location (set `frozen_since` to `None`). +2. Unfreeze the location (set `frozen_since` to `None`). Either the location is + already unfrozen, or this is a write. 3. Pop the stack until the top item matches the tag of the pointer. - A `Uniq` item matches a `Uniq` tag with the same ID. - A `Shr` item matches any `Shr` tag (with or without timestamp). - When we are reading, a `Shr` item matches a `Uniq` tag. - If, popping the stack, we make it empty, then we have undefined behavior. + If we pop the entire stack without finding a match, then we have undefined + behavior. To understand these rules better, try going back through the three examples we have seen so far and applying these rules for dereferencing pointers and @@ -402,7 +407,7 @@ The question is: Can we move the load of `x` to before the function call? Remember that the entire point of Stacked Borrows is to enforce a certain discipline when using references, in particular, to enforce uniqueness of mutable references. So we should hope that the answer to that question is "yes" -(and that, in turns, is good because we might use it for optimizations). +(and that, in turn, is good because we might use it for optimizations). Unfortunately, things are not so easy. The uniqueness of mutable references entirely rests on the fact that the pointer @@ -439,7 +444,7 @@ references are copied: At the beginning of every function, all inputs of reference type get retagged. On every assignment, if the assigned value is of reference type, it gets retagged. Moreover, we do this even when the reference value is inside the field of a `struct` or `enum`, to make sure we really cover -all references. (This recursive descend is already implemented, but the +all references. (This recursive descent is already implemented, but the implementation has not landed yet.) However, we do *not* descend recursively through references: Retagging a `&mut &mut u8` will only retag the *outer* reference. @@ -456,7 +461,7 @@ fn demo0() { // stack: [Uniq(0)]; not frozen let y = &mut *x; - Retag(x); // tag of `y` gets changed to `Uniq(1)` + Retag(y); // tag of `y` gets changed to `Uniq(1)` // stack: [Uniq(0), Uniq(1)]; not frozen // Check that `Uniq(1)` is on the stack, then pop to bring it to the top. @@ -472,8 +477,9 @@ fn demo0() { } {% endhighlight %} -For each reference, `Retag` does the following (with one special exception that -will be explained later): +For each reference, `Retag` does the following (we will slightly refine these +instructions later) on all locations covered by the reference (again, according +to `size_of_val`): 1. Compute a fresh tag, `Uniq(_)` for a mutable reference and `Shr(Some(_))` for a shared reference. @@ -492,6 +498,9 @@ will be explained later): 2. If the location is outside of `UnsafeCell`, it gets frozen with the timestamp of the new reference. +One high-level way to think about retagging is that it computes a fresh tag, and +then performs a reborrow of the old reference with the new tag. + ### 3.2 When Pointers Escape Creating a shared reference is not the only way to share a location: We can also @@ -499,7 +508,7 @@ create raw pointers, and if we are careful enough, use them to access a location from different aliasing pointers. (Of course, "careful enough" is not very precise, but the precise answer is the very model I am describing here.) -To account for this, we need one final ingredient in our model: A special +To account for this, we need one final ingredient in our model: a special instruction that indicates that a reference was cast to a raw pointer, and may thus be accessed from these raw pointers in a shared way. Consider the [following example](https://play.rust-lang.org/?version=stable&mode=debug&edition=2015&gist=253868e96b7eba85ef28e1eabd557f66): @@ -603,7 +612,7 @@ exclusive access to the same locations. That on its own is not enough, though. I also added a new check to the retagging procedure: Before taking any action (i.e., before step 3, which could pop items off the stack), we check if the reborrow is redundant: If the new reference we want to create is already -dereferencable (because it item is already on the stack and, if applicable, the +dereferencable (because its item is already on the stack and, if applicable, the stack is already frozen), *and* if the item that justifies this is moreover "derived from" the item that corresponds to the old reference, then we just do nothing. Here, "derived from" means "further up the stack". Basically, the @@ -612,20 +621,55 @@ reborrow has already happened and the new reference is ready for use, and will *not* pop the item corresponding to the old reference off the stack. In that case, we avoid popping anything, to keep other references valid. -This rule applies in our example above when we create `shr_ref` from `mut_ref`. -There is already a `Shr` on the stack (so the new reference is dereferencable), -and the item matching the old reference (`Uniq(0)`) is below that `Shr` (so -after using the new reference, the old one remains dereferencable). Hence we do -nothing, keeping the `Uniq(2)` on the stack, such that the access through -`mut_ref` at the end remains valid. +It may seem like this rule can never apply, because how can our fresh tag match +something that's already on the stack? This is indeed impossible for `Uniq` +tags, but for `Shr` tags, matching is more liberal. For example, this rule +applies in our example above when we create `shr_ref` from `mut_ref`. We do not +require freezing (because there is an `UnsafeCell`), there is already a `Shr` on +the stack (so the new reference is dereferencable) and the item matching the old +reference (`Uniq(0)`) is below that `Shr` (so after using the new reference, the +old one remains dereferencable). Hence we do nothing, keeping the `Uniq(2)` on +the stack, such that the access through `mut_ref` at the end remains valid. This may sound like a weird rule, and it is. I would surely not have thought of this if `RefCell` would not force our hands here. However, as we shall see in -[section 5](#5-key-properties), it also does not to break any of the important -properties of the model (mutable references being unique and shared references -being read-only except for `UnsafeCell`). +[section 5], it also does not to break any of the important properties of the +model (mutable references being unique and shared references being read-only +except for `UnsafeCell`). Moreover, when pushing an item to the stack (at the +end of the retag action), we can be sure that the stack is not yet frozen: if it +were frozen, the reborrow would be redundant. + +With this extension, the instructions for retagging and `EscapeToRaw` now look +as follows (again executed on all locations covered by the reference, according +to `size_of_val`): + +1. Compute a fresh tag: `Uniq(_)` for a mutable reference, `Shr(Some(_))` for a + shared reference, `Shr(None)` if this is `EscapeToRaw`. +2. Perform the checks that would also happen when we dereference this reference. + Remember the position of the item matching the tag in the stack. +3. Redundancy check: If the new tag passes the checks performed on a + dereference, and if the item that makes this check succeed is *above* the one + we remembered in step 2 (where the "frozen" state is considered above every + item in the stack), then we stop. We are done for this location. +4. Perform the actions that would also happen when an actual access happens + through this reference (for shared references a read access, for mutable + references a write access). + Now the location cannot be frozen any more: If the fresh tag is `Uniq`, we + just unfroze, if the fresh tag is `Shr` and the location was already frozen + then the redundancy check (step 3) would have kicked in. +5. If the new tag is `Uniq`, push it onto the stack. +6. If the new tag is `Shr`, push a `Shr` item to the stack. Then, if the + location is outside of `UnsafeCell`, it gets frozen with the timestamp of the + new reference. + +The one thing I find slightly unsatisfying about the redundancy check is that it +seems to overlap a bit with the rule that on a *read* access, a `Shr` item +matches a `Uniq` tag. Both of these together enable the read-only use of +mutable references that have already been shared; I would prefer to have a +single condition enabling that instead of two working together. ## 4 Differences to the Original Proposal +[section 4]: #4-differences-to-the-original-proposal The key differences to the original proposal is that the check performed on a dereference, and the check performed on an access, are not the same check. This @@ -633,13 +677,16 @@ means there are more "moving parts" in the model, but it also means we do not need a weird special exception for `demo1` any more like the original proposal did. The main reason for this, however, is that on an access, we just do not know if we are inside an `UnsafeCell` or not, so we cannot do all the checks we -would like to do. +would like to do. Accordingly, I also rearranged terminology a bit. There is +no longer one "reactivation" action, instead there is a "deref" check and an +"access" action, as described above in sections [2.2][section 2.2] and +[2.3][section 2.3]. Beyond that, I made the behavior of shared references and raw pointers more uniform. This helped to fix test failures around `iter_mut` on slices, which first creates a raw reference and then a shared reference: In the original model, creating the shared reference invalidates previously created raw -pointers. As part of unifying the two, this happens no longer. +pointers. As a result of the more uniform treatment, this no longer happens. (Coincidentally, I did not make this change with the intention of fixing `iter_mut`. I did this change because I wanted to reduce the number of case distinctions in the model. Then I realized the relevant test suddenly passed @@ -668,13 +715,15 @@ reference never gets used again we do not care much about enforcing any guarantees for it. (This is another example of a coincidental fix, where I had a surprisingly passing test case and then investigated what happened.) -And finally, there is this special exception about creating raw references, to -fix `demo_refcell`. +The redundancy check during retagging can be seen as refining a similar check +that the original model did whenever a new reference was created (where we +wouldn't change the state if the new borrow is already active). Finally, the notion of "function barriers" from the original Stacked Borrows has not been implemented yet. This is the next item on my todo list. ## 5 Key Properties +[section 5]: #5-key-properties Let us look at the two key properties that I set out as design goals, and see how the model guarantees that they hold true in all valid (UB-free) executions. @@ -779,13 +828,14 @@ but it is nevertheless interesting and might be useful in a "lower-level MIR" that does not permit dereferences in paths. ## 6 Evaluation, and How You Can Help +[section 6]: #6-evaluation-and-how-you-can-help I have implemented both the validity invariant and the model as described above in miri. This [uncovered](https://github.com/rust-lang/rust/issues/54908) two [issues](https://github.com/rust-lang/rust/issues/54957) in the standard library, but both were related to validity invariants, not Stacked Borrows. With these exceptions, the model passes the entire test suite. There were some -more test failures in earlier versions (as mentioned in section 4), but the +more test failures in earlier versions (as mentioned in [section 4]), but the final model accepts all the code covered by miri's test suite. Moreover I wrote a bunch of compile-fail tests to make sure the model catches various violations of the key properties it should ensure. @@ -836,3 +886,6 @@ have a mere three weeks left on this internship, after which I will have to significantly reduce my Rust activities in favor of finishing my PhD. I won't disappear entirely though, don't worry -- I will still be able to mentor you if you want to help with any of the above tasks. :) + +Thanks to @nikomatsakis for feedback on a draft of this post. +