rapx 0.7.53

A static analysis platform for Rust program analysis and verification
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

//! Checker for `ValidString`: UTF-8 validity of tracked byte buffers.

use crate::helpers::mir_scan::Checkpoint;
use crate::verify::contract::Property;
use crate::verify::report::CheckResult;
use crate::verify::vm::state::VmState;
use z3::{SatResult, Solver, ast::Int};

use super::PropertyChecker;

impl PropertyChecker {
    /// Shared UTF-8 byte check: prove the tracked buffer bytes of `alloc_id` are
    /// *not* valid UTF-8 (i.e. disprove the DFA), reporting `Failed` when the
    /// solver proves they cannot be valid.
    fn check_utf8_alloc<'ctx, 'tcx>(
        &self,
        vm_state: &VmState<'ctx, 'tcx>,
        solver: &Solver<'ctx>,
        alloc_id: crate::verify::vm::state::AllocId,
    ) -> CheckResult {
        if vm_state.alloc(alloc_id).dead {
            return CheckResult::Failed;
        }
        let byte_pairs = vm_state.alloc_byte_values(alloc_id);
        if byte_pairs.is_empty() {
            return CheckResult::ProvedByRule; // no byte-level info → trust
        }
        let bytes: Vec<Int<'ctx>> = byte_pairs.iter().map(|(_, t)| (*t).clone()).collect();
        let valid = super::utf8_validity(vm_state.ctx, &bytes);

        solver.push();
        solver.assert(&valid);
        let r = solver.check();
        solver.pop(1);
        match r {
            SatResult::Unsat => CheckResult::Failed,
            _ => CheckResult::ProvedByRule,
        }
    }

    pub(super) fn check_valid_string<'ctx, 'tcx>(
        &self,
        vm_state: &VmState<'ctx, 'tcx>,
        solver: &Solver<'ctx>,
        checkpoint: &Checkpoint<'tcx>,
        property: &Property<'tcx>,
    ) -> CheckResult {
        // Empty byte range: trivially valid UTF-8.
        if let Some(count) = property
            .args()
            .get(2)
            .and_then(|a| self.resolve_arg_term(vm_state, checkpoint, a))
        {
            if count.as_u64() == Some(0) {
                return CheckResult::ProvedByRule;
            }
        }

        let Some(value) = self.target_value(vm_state, checkpoint, property) else {
            return CheckResult::ProvedByRule;
        };
        if let Some(alloc_id) = value.provenance_alloc_id() {
            return self.check_utf8_alloc(vm_state, solver, alloc_id);
        }

        // A one-argument `ValidString(iter)` targets an `Iterator<Item = u8>`:
        // trace the (possibly `Cloned`/`Rev`-wrapped) iterator to the backing
        // byte buffer of its innermost `Iter`/`IterMut` and UTF-8-check that.
        if let Some(local) = vm_state.find_local_by_address(&value.term) {
            if let Some((alloc_id, _end_offset)) = vm_state.iter_utf8_buffer(local) {
                return self.check_utf8_alloc(vm_state, solver, alloc_id);
            }
        }
        CheckResult::ProvedByRule
    }
}