use crate::check::contract::{Backend, Category, Confidence, Finding, Lint, ParsedFile, Severity};
use crate::check::facts::{extract_handlers, Validation};
use crate::check::lints::to_span;
pub struct Acc003Confusion;
impl Lint for Acc003Confusion {
fn code(&self) -> &'static str {
"ACC003-P"
}
fn id(&self) -> &'static str {
"account-confusion"
}
fn category(&self) -> Category {
Category::Acc
}
fn backend(&self) -> Backend {
Backend::Syn
}
fn default_severity(&self) -> Severity {
Severity::Warn
}
fn default_confidence(&self) -> Confidence {
Confidence::Heuristic
}
fn run(&self, file: &ParsedFile) -> Vec<Finding> {
let path = file.path.display().to_string();
let mut out = Vec::new();
for handler in extract_handlers(&file.ast) {
for b in &handler.bindings {
if !is_identity_name(&b.name) {
continue;
}
if b.delegated
|| !b.reads_data()
|| b.validations.contains(&Validation::KeyCompared)
{
continue;
}
let Some(span) = b.read_span else {
continue;
};
out.push(Finding {
code: self.code(),
id: self.id(),
confidence: self.default_confidence(),
severity: self.default_severity(),
span: to_span(span, &path),
evidence: format!(
"account `{}` is read as a known singleton but its key is never checked against the expected address; an attacker can pass a look-alike account",
b.name
),
fix: Some(format!(
"compare `{}.key()` to the expected address (or the derived PDA) before reading it",
b.name
)),
});
}
}
out
}
}
fn is_identity_name(name: &str) -> bool {
name == "config"
|| name == "settings"
|| name.ends_with("_config")
|| name.ends_with("_settings")
}