Panic freedom

Let's start with a simple example: a function that squares a u8 integer. To extract this function to Lean using hax, we simply need to run the command cargo hax into lean in the directory of the crate in which the function square is defined.

Note: throughout this tutorial, you can edit the snippets of code and extract to Lean by clicking the play button (), or even typecheck it with the button ().

fn square(x: u8) -> u8 {
    x * x
}

If we run lake build on the result (or type-check using the playground), we get a success. If you followed the F* tutorial, this might be a surprise because the function is not panic-free. Indeed, our encoding of Rust code in Lean wraps everything in a result monad. And functions that panic return an error in this monad. To try to prove panic-freedom, we have to specify that the result of square is expected not to be an error in this result type. A way to do that is the following:

#[hax_lib::lean::after("
theorem square_spec (value: u8) :
  ⦃ Playground._.requires (value) = pure true ⦄
  (${square} value)
  ⦃ ⇓ result => Playground.__1.ensures value result = pure true ⦄
  := by
  mvcgen
  simp [Playground._.requires, Playground.__1.ensures, ${square}] at *
  intros
  rw [UInt8.HaxMul_spec_bv_rw] ; simp ;
  bv_decide")]
#[hax_lib::requires(true)]
#[hax_lib::ensures(|res| true)]
fn square(x: u8) -> u8 {
    x * x
}

The specification is extrinsic to the function, we state a theorem square_spec which uses a Hoare triple to specify properties on the output, assuming some other properties on the inputs. Here, we use the precondition and post-condition defined using the hax_lib macro, but we could write our specification entirely in the square_spec theorem. Here our post-condition is true which seems trivial, but the condition Playground.__1.ensures value result = pure true is false if result (and thus Playground.__1.ensures value result) is an error in the result monad. So this specification states that square should be panic-free. We also have a small proof script applying a few tactics to try to prove our theorem. If we try running lake build after extracting this code, we get an error: The prover found a counterexample, consider the following assignment: value = 255. Indeed square(255) panics because the multiplication overflows.

Rust and panicking code

Quoting the chapter To panic! or Not to panic! from the Rust book:

The panic! macro signals that your program is in a state it can't handle and lets you tell the process to stop instead of trying to proceed with invalid or incorrect values.

A Rust program should panic only in a situation where an assumption or an invariant is broken: a panic models an invalid state. Formal verification is about proving such invalid state cannot occur, at all.

From this observation emerges the urge of proving Rust programs to be panic-free!

Fixing our squaring function

Let's come back to our example. There is an informal assumption to the multiplication operator in Rust: the inputs should be small enough so that the addition doesn't overflow.

Note that Rust also provides wrapping_mul, a non-panicking variant of the multiplication on u8 that wraps when the result is bigger than 255. Replacing the common multiplication with wrapping_mul in square would fix the panic, but then, square(256) returns zero. Semantically, this is not what one would expect from square.

Our problem is that our function square is well-defined only when its input is within 0 and 15.

Solution: add a precondition

We already added a pre-condition to specify panic-freedom but we can turn it into a more interesting pre-condition to restrict the inputs and stay in the domain where the multiplication fits in a u8. We only need to modify the Rust condition that is passed to the hax_lib::requires macro:

#[hax_lib::lean::after("
theorem square_spec (value: u8) :
  ⦃ Playground._.requires (value) = pure true ⦄
  (${square} value)
  ⦃ ⇓ result => Playground.__1.ensures value result = pure true ⦄
  := by
  mvcgen
  simp [Playground._.requires, Playground.__1.ensures, ${square}] at *
  intros
  rw [UInt8.HaxMul_spec_bv_rw] ; simp ;
  bv_decide")]
#[hax_lib::requires(x < 16)]
#[hax_lib::ensures(|res| true)]
fn square(x: u8) -> u8 {
    x * x
}

With this precondition, Lean is able to prove panic freedom. From now on, it is the responsibility of the clients of square to respect the contract. The next step is thus be to verify, through hax extraction, that square is used correctly at every call site.\

Common panicking situations

Multiplication is not the only panicking function provided by the Rust library: most of the other integer arithmetic operation have such informal assumptions.

Another source of panics is indexing. Indexing in an array, a slice or a vector is a partial operation: the index might be out of range.

In the example folder of hax, you can find the chacha20 example that makes use of pre-conditions to prove panic freedom.