powerio-cli 0.11.2

//! The anonymization boundary.
//!
//! Findings leave the machine; corpus bytes do not. Everything the harness
//! emits passes through this module, which states three things and only three:
//! class ordinals in place of identifiers, ratios and orders of magnitude in
//! place of values, and masked templates in place of any text that came from a
//! file.
//!
//! The last line of defense is [`Sanitizer::audit`], which re-reads the emitted
//! report and fails the run if any string the corpus taught the harness
//! survived into it. The harness knows every name in every case it parsed and
//! every path it walked, so this is a decidable check rather than a heuristic.

use std::collections::{BTreeMap, BTreeSet};

/// The shortest string treated as identifying.
///
/// Two characters would match a field name in nearly every path the report
/// emits (`.p`, `.q`, `.id`), so masking them costs more than it buys. The
/// consequence is stated rather than hidden: a corpus whose identifiers are one
/// or two characters is not protected by this module, and a report from one
/// should be read before it is shared.
const MIN_IDENTIFYING: usize = 3;

/// A string from the corpus that reached the report.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Leak {
    pub secret: String,
    pub line: usize,
}

impl std::fmt::Display for Leak {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        // Names the length and the line, never the string itself: a panic
        // message is as public as the report is.
        write!(
            f,
            "line {}: a {}-character string from the corpus reached the report",
            self.line,
            self.secret.chars().count()
        )
    }
}

/// Replaces corpus identifiers with ordinals and corpus values with
/// magnitudes.
#[derive(Debug, Default, Clone)]
pub struct Sanitizer {
    /// Bus id to dense ordinal, in sorted bus-id order.
    bus_ordinal: BTreeMap<usize, usize>,
    /// Every string the corpus taught us: element names, uids, extras values,
    /// case names, and every path component walked.
    secrets: BTreeSet<String>,
    /// Tokens the harness itself emits that a corpus may also happen to use.
    /// A corpus directory named `psse/` must not make the format token `psse`
    /// unreportable.
    allowed: BTreeSet<String>,
}

impl Sanitizer {
    #[must_use]
    pub fn new() -> Self {
        Self::default()
    }

    /// Learn every string in a parsed network, and give its buses ordinals.
    ///
    /// Walks the serialized model rather than a hand-written field list: a new
    /// name-bearing field then joins the secret set the day it is added, with
    /// no second place to remember.
    pub fn learn_network(&mut self, value: &serde_json::Value) {
        collect_strings(value, &mut self.secrets);
    }

    /// Assign ordinals to a bus id set. Called once per bucket with the
    /// reference member's buses, then again per sibling so a renumbering
    /// format still lands on ordinals.
    pub fn learn_buses(&mut self, ids: impl IntoIterator<Item = usize>) {
        let mut ids: Vec<usize> = ids.into_iter().collect();
        ids.sort_unstable();
        for id in ids {
            let next = self.bus_ordinal.len();
            self.bus_ordinal.entry(id).or_insert(next);
        }
    }

    /// Declare a token as the harness's own vocabulary. Format names are the
    /// case that matters: a corpus laid out as `psse/`, `matpower/`, `dss/`
    /// teaches those words as path components, and without this the report
    /// could not name the format a finding belongs to.
    pub fn allow(&mut self, token: &str) {
        self.allowed.insert(token.to_string());
        // `egret-json` also licenses the word `egret`, which is what a corpus
        // laid out by format actually names its directories.
        for part in token.split(['-', '_', '.']) {
            if !part.is_empty() {
                self.allowed.insert(part.to_string());
            }
        }
    }

    /// Learn a path the harness walked, component by component, so neither the
    /// full path nor a directory or file name can reach the report.
    pub fn learn_path(&mut self, path: &std::path::Path) {
        self.secrets.insert(path.to_string_lossy().into_owned());
        for part in path.components() {
            let part = part.as_os_str().to_string_lossy();
            self.secrets.insert(part.to_string());
            if let Some((stem, _)) = part.rsplit_once('.') {
                self.secrets.insert(stem.to_string());
            }
        }
    }

    /// A bus as a class ordinal. An unseen id gets the next ordinal rather
    /// than passing through.
    pub fn bus(&mut self, id: usize) -> String {
        let next = self.bus_ordinal.len();
        let ordinal = *self.bus_ordinal.entry(id).or_insert(next);
        format!("bus#{ordinal}")
    }

    /// Free text reduced to a template: names become `<name>`, digit runs
    /// become `#`, and quoted spans become `'…'`.
    ///
    /// Warning prose is format vocabulary and is worth keeping; the record
    /// values a warning quotes are case data and are not.
    #[must_use]
    pub fn template(&self, text: &str) -> String {
        let mut out = mask_quoted(text);
        // Prefiltered on this text's own tokens, the same necessary condition
        // `audit` uses, so a message pays for sorting only the handful of
        // secrets that could appear in it rather than every one the corpus
        // taught. Replacement only ever inserts `<name>` and `#`, neither of
        // which can introduce a secret (`name` is powerio's own vocabulary and
        // never identifying), so filtering once against the starting text is
        // enough.
        let tokens: std::collections::HashSet<&str> = word_tokens(&out).collect();
        let mut names: Vec<&String> = self
            .identifying()
            .filter(|s| leading_run(s).is_none_or(|r| tokens.contains(r)))
            .collect();
        // Longest first, so a name that contains another is replaced whole.
        names.sort_by_key(|n| std::cmp::Reverse(n.len()));
        for name in names {
            out = replace_tokens(&out, name, "<name>");
        }
        mask_digits(&out)
    }

    /// The secrets worth checking for: long enough to identify, carrying a
    /// letter, and not part of powerio's own vocabulary.
    ///
    /// A digit only secret is deliberately outside this set. Auditing one
    /// would match the report's own counts, ordinals and magnitudes, which are
    /// numbers the harness computed rather than strings the corpus taught;
    /// [`Sanitizer::template`] masks every digit run instead.
    fn identifying(&self) -> impl Iterator<Item = &String> {
        let vocabulary = vocabulary();
        self.secrets.iter().filter(move |s| {
            s.chars().count() >= MIN_IDENTIFYING
                && s.chars().any(char::is_alphabetic)
                && !vocabulary.contains(s.as_str())
                && !self.allowed.contains(s.as_str())
        })
    }

    /// Confirm nothing the corpus taught us survived into `emitted`.
    ///
    /// This is the backstop, and it is decidable rather than heuristic over
    /// the set it covers: the harness parsed every case and walked every path,
    /// so it knows the exact set of strings that must not appear. That set is
    /// every secret long enough to identify and carrying a letter; masking is
    /// what handles the rest.
    ///
    /// # Errors
    ///
    /// Returns every secret found, with the line it appeared on.
    pub fn audit(&self, emitted: &str) -> Result<(), Vec<Leak>> {
        // Scanning every secret against every line is the whole cost here, and
        // a corpus teaches hundreds of thousands of them. `leading_run` is a
        // necessary condition for a match, so most secrets are excluded by one
        // hash lookup and only the survivors pay for a scan.
        let secrets: Vec<(&String, Option<&str>)> =
            self.identifying().map(|s| (s, leading_run(s))).collect();
        let mut leaks = Vec::new();
        let mut tokens: std::collections::HashSet<&str> = std::collections::HashSet::new();
        for (i, line) in emitted.lines().enumerate() {
            tokens.clear();
            tokens.extend(word_tokens(line));
            for (secret, run) in &secrets {
                // A secret whose leading run is absent from this line's tokens
                // cannot match it; one that does not begin with a word
                // character has no such run and is always scanned.
                if run.is_some_and(|r| !tokens.contains(r)) {
                    continue;
                }
                if contains_token(line, secret) {
                    leaks.push(Leak {
                        secret: (*secret).clone(),
                        line: i + 1,
                    });
                }
            }
        }
        if leaks.is_empty() { Ok(()) } else { Err(leaks) }
    }
}

/// Every string powerio itself writes into a serialized network that is not
/// case data: field names, enum spellings, format tokens, unit names.
///
/// Derived from the model itself rather than from a hand-kept list, twice
/// over. The JSON schemas of both network types name every field and every
/// enum spelling, populated or not, so a new field joins the vocabulary the
/// day the model gains it; a reference network built here adds the spellings
/// custom serializers write past the schema, such as the generator capability
/// map's column names. A real case whose element is named after one of these
/// spellings is exempt from the audit; the tradeoff runs the safe way, since a
/// spurious failure would stop a run cold while this costs one unmasked token.
///
/// The completeness gate lives in `super::tests`: the union of every
/// fixture's serde keys must sit inside this set.
///
/// Built once. `identifying()` runs per emitted message and per audited line,
/// so rebuilding the schemas and the network there would dominate both.
pub(crate) fn vocabulary() -> &'static BTreeSet<String> {
    static VOCABULARY: std::sync::OnceLock<BTreeSet<String>> = std::sync::OnceLock::new();
    VOCABULARY.get_or_init(build_vocabulary)
}

fn build_vocabulary() -> BTreeSet<String> {
    use powerio_matrix::{BalancedNetwork, Branch, Bus, BusId, BusType, Generator, Load, Shunt};
    let mut out = BTreeSet::new();
    // Field names and enum spellings of both models, from the same derives
    // that generate the published schema documents.
    for value in [
        serde_json::to_value(schemars::schema_for!(BalancedNetwork)),
        serde_json::to_value(schemars::schema_for!(powerio_dist::MulticonductorNetwork)),
    ]
    .into_iter()
    .flatten()
    {
        collect_schema_words(&value, &mut out);
    }
    let mut net = BalancedNetwork::new("", 100.0);
    net.buses_mut().push(Bus::new(BusId(1), BusType::Ref, 1.0));
    net.buses_mut().push(Bus::new(BusId(2), BusType::Pv, 1.0));
    net.buses_mut().push(Bus::new(BusId(3), BusType::Pq, 1.0));
    net.buses_mut()
        .push(Bus::new(BusId(4), BusType::Isolated, 1.0));
    net.branches_mut()
        .push(Branch::new(BusId(1), BusId(2), 0.0, 1.0));
    // Filled so `caps_serde` writes its column names, which the schema
    // states as a plain string map.
    let mut generator = Generator::new(BusId(1));
    generator.caps.fill(Some(0.0));
    net.generators_mut().push(generator);
    net.loads_mut().push(Load::new(BusId(1), 0.0, 0.0));
    net.shunts_mut().push(Shunt::new(BusId(1), 0.0, 0.0));
    if let Ok(value) = serde_json::to_value(&net) {
        collect_words(&value, &mut out);
    }
    // Format metadata property names are keys in a source neutral string map,
    // so the generated schema cannot enumerate them. PowerIO writes these
    // PSS/E generator keys when retaining fields for fresh RAW/RAWX emission.
    out.extend(
        [
            "psse_baslod",
            "psse_eqid",
            "psse_f1",
            "psse_f2",
            "psse_f3",
            "psse_f4",
            "psse_gtap",
            "psse_o1",
            "psse_o2",
            "psse_o3",
            "psse_o4",
            "psse_rmpct",
            "psse_rt",
            "psse_wmod",
            "psse_wpf",
            "psse_xt",
            "psse_zr",
            "psse_zx",
        ]
        .map(str::to_owned),
    );
    out
}

/// Keys and string values of a document powerio built itself, where every
/// spelling is powerio's own. Never run on anything a corpus touched.
fn collect_words(value: &serde_json::Value, out: &mut BTreeSet<String>) {
    match value {
        serde_json::Value::String(s) if !s.is_empty() => {
            out.insert(s.clone());
        }
        serde_json::Value::Array(xs) => {
            for x in xs {
                collect_words(x, out);
            }
        }
        serde_json::Value::Object(xs) => {
            for (key, x) in xs {
                out.insert(key.clone());
                collect_words(x, out);
            }
        }
        _ => {}
    }
}

/// Property names and enum spellings of a JSON schema. Nothing else: a
/// schema's description sentences are not spellings the report writes.
fn collect_schema_words(value: &serde_json::Value, out: &mut BTreeSet<String>) {
    match value {
        serde_json::Value::Array(xs) => {
            for x in xs {
                collect_schema_words(x, out);
            }
        }
        serde_json::Value::Object(map) => {
            if let Some(serde_json::Value::Object(props)) = map.get("properties") {
                out.extend(props.keys().cloned());
            }
            if let Some(serde_json::Value::Array(spellings)) = map.get("enum") {
                out.extend(
                    spellings
                        .iter()
                        .filter_map(|s| s.as_str().map(str::to_string)),
                );
            }
            if let Some(spelling) = map.get("const").and_then(serde_json::Value::as_str) {
                out.insert(spelling.to_string());
            }
            for x in map.values() {
                collect_schema_words(x, out);
            }
        }
        _ => {}
    }
}

/// `after / before`, to three significant digits, or `None` when `before` is
/// zero and the ratio would state the new value outright.
#[must_use]
pub fn ratio(before: f64, after: f64) -> Option<f64> {
    if before == 0.0 || !before.is_finite() || !after.is_finite() {
        return None;
    }
    let r = after / before;
    // A zero ratio has no decade to round against: `log10(0)` is `-inf`, and
    // the `as i32` saturation would make the exponent overflow.
    if r == 0.0 {
        return Some(0.0);
    }
    let decade = r.abs().log10().floor();
    let scale = 10f64.powi(2 - decade as i32);
    if !scale.is_finite() || scale == 0.0 {
        return Some(r);
    }
    Some((r * scale).round() / scale)
}

/// The decimal order of magnitude of a value: `1` for 42.0, `-3` for 0.004,
/// `None` for zero. Enough to tell a per-unit impedance from a MW total,
/// never enough to state the value.
#[must_use]
pub fn magnitude(x: f64) -> Option<i32> {
    if x == 0.0 || !x.is_finite() {
        return None;
    }
    Some(x.abs().log10().floor() as i32)
}

/// Collapse array indices in a serde path so `.loads[3].p` and `.loads[47].p`
/// become one class, `.loads[#].p`. Both safer (the report stops counting a
/// utility's loads) and more useful (one line per class of loss).
#[must_use]
pub fn collapse_path(path: &str) -> String {
    let mut out = String::with_capacity(path.len());
    let mut in_index = false;
    for ch in path.chars() {
        match ch {
            '[' => {
                in_index = true;
                out.push_str("[#");
            }
            ']' => {
                in_index = false;
                out.push(']');
            }
            _ if in_index => {}
            _ => out.push(ch),
        }
    }
    out
}

/// Whether `needle` appears in `haystack` as a whole word.
///
/// Substring matching is wrong in both directions here. It fires on
/// `egret` inside the format token `egret-json`, which is vocabulary rather
/// than a leak; and a corpus name that happens to be two letters would
/// otherwise match half the report. A word is what a reader means when they
/// ask whether their filename appears.
fn word_boundaries(haystack: &str, needle: &str) -> Vec<usize> {
    if needle.is_empty() {
        return Vec::new();
    }
    let bytes = haystack.as_bytes();
    haystack
        .match_indices(needle)
        .filter(|(i, m)| {
            let before = *i == 0 || !word_part(bytes[*i - 1]);
            let end = *i + m.len();
            let after = end >= bytes.len() || !word_part(bytes[end]);
            before && after
        })
        .map(|(i, _)| i)
        .collect()
}

/// Whether `b` is part of a word, matching [`word_boundaries`]'s rule.
fn word_part(b: u8) -> bool {
    b.is_ascii_alphanumeric() || b == b'_'
}

/// The maximal word run `needle` opens with, or `None` when it does not start
/// with a word character.
///
/// This run is what [`Sanitizer::audit`] prefilters on, and the reason it is
/// sound: a match of `needle` needs a non-word character (or the start of the
/// line) before it, and the run is followed either by `needle`'s own next
/// character — a non-word one, since the run is maximal — or by the match's
/// own trailing boundary. So the run always lands in the line as a complete
/// token, and a line whose tokens do not hold it cannot hold `needle`.
fn leading_run(needle: &str) -> Option<&str> {
    let end = needle
        .bytes()
        .position(|b| !word_part(b))
        .unwrap_or(needle.len());
    (end > 0).then(|| &needle[..end])
}

/// The maximal word runs of `line`, the tokens a secret's leading run must
/// appear among.
fn word_tokens(line: &str) -> impl Iterator<Item = &str> {
    line.split(|c: char| !(c.is_ascii_alphanumeric() || c == '_'))
        .filter(|t| !t.is_empty())
}

fn contains_token(haystack: &str, needle: &str) -> bool {
    !word_boundaries(haystack, needle).is_empty()
}

fn replace_tokens(haystack: &str, needle: &str, with: &str) -> String {
    let hits = word_boundaries(haystack, needle);
    if hits.is_empty() {
        return haystack.to_string();
    }
    let mut out = String::with_capacity(haystack.len());
    let mut cursor = 0;
    for start in hits {
        if start < cursor {
            continue;
        }
        out.push_str(&haystack[cursor..start]);
        out.push_str(with);
        cursor = start + needle.len();
    }
    out.push_str(&haystack[cursor..]);
    out
}

fn mask_quoted(text: &str) -> String {
    let mut out = String::with_capacity(text.len());
    let mut quote: Option<char> = None;
    for ch in text.chars() {
        match quote {
            Some(open) if ch == open => {
                out.push('…');
                out.push(ch);
                quote = None;
            }
            Some(_) => {}
            None => {
                if ch == '\'' || ch == '"' || ch == '`' {
                    quote = Some(ch);
                }
                out.push(ch);
            }
        }
    }
    out
}

fn mask_digits(text: &str) -> String {
    let mut out = String::with_capacity(text.len());
    let mut in_run = false;
    for ch in text.chars() {
        if ch.is_ascii_digit() {
            if !in_run {
                out.push('#');
                in_run = true;
            }
        } else {
            in_run = false;
            out.push(ch);
        }
    }
    out
}

fn collect_strings(value: &serde_json::Value, out: &mut BTreeSet<String>) {
    collect_strings_inner(value, false, out);
}

/// `in_extras` marks the subtree under an `extras` map, where the keys are
/// whatever token the source stated — a deck naming a property after a feeder
/// puts that name in a key and nowhere else. Everywhere else the keys are
/// powerio's own field names, and learning those made `uid`, `route`,
/// `charging` and every other optional field into a corpus secret: masked out
/// of the serde paths the report is written in, and tripping [`audit`] on the
/// harness's own vocabulary.
fn collect_strings_inner(value: &serde_json::Value, in_extras: bool, out: &mut BTreeSet<String>) {
    match value {
        serde_json::Value::String(s) if !s.is_empty() => {
            out.insert(s.clone());
        }
        serde_json::Value::Array(xs) => {
            for x in xs {
                collect_strings_inner(x, in_extras, out);
            }
        }
        serde_json::Value::Object(xs) => {
            for (key, x) in xs {
                if in_extras {
                    out.insert(key.clone());
                }
                collect_strings_inner(x, in_extras || key == "extras", out);
            }
        }
        _ => {}
    }
}