rsigma-ir 0.24.0

//! Lowering: parser AST → HIR.
//!
//! Walks metadata, detections, and conditions after static pipeline transforms.
//! Quantified selectors are preserved as [`IrCondition::Selector`] rather than
//! being expanded, so downstream consumers keep count-based semantics.
//! Modifier interpretation lives in private helpers alongside this module.

mod helpers;
mod mod_ctx;
mod value;

use std::collections::HashMap;

use rsigma_parser::validate::{check_detection_item, check_regex, exists_flag};
use rsigma_parser::{
    ConditionExpr, CorrelationRule, Detection, DetectionItem, Detections, FilterRule, Modifier,
    SigmaParserError, SigmaRule, SigmaValue,
};

use crate::error::IrError;
use crate::{
    IrCondition, IrCorrelation, IrDetection, IrDetectionItem, IrFilter, IrMatcher, IrRule,
    IrRuleMetadata,
};

use helpers::{Result, yaml_to_json_map};
use mod_ctx::ModCtx;
use value::{lower_value, lower_value_keywords};

/// Options controlling the lowering strictness.
#[derive(Debug, Clone, Default)]
pub struct LowerOptions {
    /// When false (default), reject string values that still contain
    /// `${source.*}` placeholders. When true, preserve them for the deferred
    /// specialization path.
    pub permissive_placeholders: bool,
}

/// Lower a single parsed `SigmaRule` into its HIR form.
pub fn lower_rule(rule: &SigmaRule, opts: &LowerOptions) -> Result<IrRule> {
    let mut detections = HashMap::new();
    for (name, detection) in &rule.detection.named {
        detections.insert(name.clone(), lower_detection(detection, opts)?);
    }

    let detection_names: Vec<String> = detections.keys().cloned().collect();
    let mut conditions = Vec::with_capacity(rule.detection.conditions.len());
    for condition in &rule.detection.conditions {
        conditions.push(lower_condition(condition, &detection_names)?);
    }

    Ok(IrRule {
        metadata: metadata_from_rule(rule),
        logsource: rule.logsource.clone(),
        sigma_version: rule.sigma_version,
        detections,
        conditions,
    })
}

/// Lower only the condition trees of a detection section, without lowering
/// detection items.
///
/// Used by filter lowering and any caller that needs the condition trees
/// without paying to fully lower every detection item (and without the
/// eval-specific value constraints that lowering items would impose).
pub fn lower_conditions(section: &Detections) -> Result<Vec<IrCondition>> {
    let names: Vec<String> = section.named.keys().cloned().collect();
    section
        .conditions
        .iter()
        .map(|c| lower_condition(c, &names))
        .collect()
}

/// Lower a parsed detection into `IrDetection`.
pub fn lower_detection(detection: &Detection, opts: &LowerOptions) -> Result<IrDetection> {
    match detection {
        Detection::AllOf(items) => {
            if items.is_empty() {
                return Err(IrError::InvalidModifiers(
                    "AllOf detection must not be empty (vacuous truth)".into(),
                ));
            }
            let lowered: Result<Vec<_>> = items
                .iter()
                .map(|item| lower_detection_item(item, opts))
                .collect();
            Ok(IrDetection::AllOf(lowered?))
        }
        Detection::AnyOf(dets) => {
            if dets.is_empty() {
                return Err(IrError::InvalidModifiers(
                    "AnyOf detection must not be empty (would never match)".into(),
                ));
            }
            let lowered: Result<Vec<_>> = dets.iter().map(|d| lower_detection(d, opts)).collect();
            Ok(IrDetection::AnyOf(lowered?))
        }
        Detection::ArrayMatch {
            field,
            quantifier,
            body,
        } => {
            let compiled_body = lower_detection(body, opts)?;
            Ok(IrDetection::ArrayMatch {
                field: field.clone(),
                quantifier: *quantifier,
                body: Box::new(compiled_body),
            })
        }
        Detection::And(dets) => {
            if dets.is_empty() {
                return Err(IrError::InvalidModifiers(
                    "And detection must not be empty".into(),
                ));
            }
            let lowered: Result<Vec<_>> = dets.iter().map(|d| lower_detection(d, opts)).collect();
            Ok(IrDetection::And(lowered?))
        }
        Detection::Conditional { named, condition } => {
            if named.is_empty() {
                return Err(IrError::InvalidModifiers(
                    "Conditional detection must have at least one named sub-selection".into(),
                ));
            }
            let mut lowered_named = HashMap::new();
            for (k, d) in named {
                lowered_named.insert(k.clone(), lower_detection(d, opts)?);
            }
            let names: Vec<String> = lowered_named.keys().cloned().collect();
            let lowered_cond = lower_condition(condition, &names)?;
            Ok(IrDetection::Conditional {
                named: lowered_named,
                condition: lowered_cond,
            })
        }
        Detection::Keywords(values) => {
            let matchers: Result<Vec<IrMatcher>> =
                values.iter().map(lower_value_keywords).collect();
            let matchers = matchers?;
            let matcher = match matchers.len() {
                0 => {
                    return Err(IrError::InvalidModifiers(
                        "Keywords detection must not be empty".into(),
                    ));
                }
                1 => matchers.into_iter().next().unwrap(),
                _ => IrMatcher::AnyOf(matchers),
            };
            Ok(IrDetection::Keywords(matcher))
        }
    }
}

/// Lower a detection item — absorbs modifier interpretation.
pub fn lower_detection_item(item: &DetectionItem, opts: &LowerOptions) -> Result<IrDetectionItem> {
    if !opts.permissive_placeholders {
        for v in &item.values {
            reject_placeholders(v)?;
        }
    }

    let ctx = ModCtx::from_modifiers(&item.field.modifiers);
    match check_detection_item(item) {
        Ok(()) => {}
        Err(SigmaParserError::InvalidModifiers(message)) => {
            return Err(IrError::InvalidModifiers(message));
        }
        // The evaluator compiles regexes after lowering and preserves its
        // established InvalidRegex error for ASTs rewritten after parsing.
        Err(SigmaParserError::InvalidValue(_))
            if item.field.modifiers.contains(&Modifier::Re)
                && item.values.iter().any(
                    |value| matches!(value, SigmaValue::String(s) if check_regex(&s.original).is_err()),
                ) => {}
        Err(SigmaParserError::InvalidValue(message))
            if item.field.modifiers.iter().any(|modifier| {
                matches!(
                    modifier,
                    Modifier::Gt | Modifier::Gte | Modifier::Lt | Modifier::Lte
                )
            }) =>
        {
            return Err(IrError::ExpectedNumeric(message));
        }
        Err(other) => return Err(IrError::IncompatibleValue(other.to_string())),
    }

    if ctx.exists {
        if item.field.name.is_none() {
            return Err(IrError::IncompatibleValue(
                "|exists must be applied to a field".into(),
            ));
        }
        let expect = match item.values.as_slice() {
            [value] => exists_flag(value),
            _ => None,
        }
        .ok_or_else(|| IrError::IncompatibleValue("|exists takes a single boolean value".into()))?;
        return Ok(IrDetectionItem {
            field: item.field.name.clone(),
            matcher: IrMatcher::Exists(expect),
            exists: Some(expect),
        });
    }

    // An empty value list is a null check, as in pySigma.
    if item.values.is_empty() {
        if item.field.name.is_none() {
            return Err(IrError::IncompatibleValue(
                "an empty value list must be bound to a field".into(),
            ));
        }
        let matcher = if ctx.has_neq() {
            IrMatcher::Not(Box::new(IrMatcher::Null))
        } else {
            IrMatcher::Null
        };
        return Ok(IrDetectionItem {
            field: item.field.name.clone(),
            matcher,
            exists: None,
        });
    }

    // `|neq` negates the whole item, so `Field|neq: [a, b]` means neither a nor b.
    let mut value_ctx = ctx;
    value_ctx.neq = false;
    let matchers: Result<Vec<IrMatcher>> = item
        .values
        .iter()
        .map(|v| lower_value(v, &value_ctx))
        .collect();
    let matchers = matchers?;

    let combined = if ctx.all {
        if matchers.len() == 1 {
            matchers.into_iter().next().unwrap()
        } else {
            IrMatcher::AllOf(matchers)
        }
    } else if matchers.len() == 1 {
        matchers.into_iter().next().unwrap()
    } else {
        IrMatcher::AnyOf(matchers)
    };
    let combined = if ctx.has_neq() {
        IrMatcher::Not(Box::new(combined))
    } else {
        combined
    };

    Ok(IrDetectionItem {
        field: item.field.name.clone(),
        matcher: combined,
        exists: None,
    })
}

/// Lower a condition expression tree — collapses selectors into identifiers.
pub fn lower_condition(expr: &ConditionExpr, detection_names: &[String]) -> Result<IrCondition> {
    match expr {
        ConditionExpr::Identifier(name) => {
            if !detection_names.iter().any(|n| n == name) {
                return Err(IrError::UnknownDetection(name.clone()));
            }
            Ok(IrCondition::Detection(name.clone()))
        }
        ConditionExpr::And(exprs) => {
            let lowered: Result<Vec<_>> = exprs
                .iter()
                .map(|e| lower_condition(e, detection_names))
                .collect();
            Ok(IrCondition::And(lowered?))
        }
        ConditionExpr::Or(exprs) => {
            let lowered: Result<Vec<_>> = exprs
                .iter()
                .map(|e| lower_condition(e, detection_names))
                .collect();
            Ok(IrCondition::Or(lowered?))
        }
        ConditionExpr::Not(inner) => Ok(IrCondition::Not(Box::new(lower_condition(
            inner,
            detection_names,
        )?))),
        ConditionExpr::Selector {
            quantifier,
            pattern,
        } => {
            // A selector over no identifiers has no defined result, and eval
            // and conversion would disagree on it.
            if !detection_names
                .iter()
                .any(|n| pattern.matches_detection_name(n))
            {
                return Err(IrError::NoSelectorMatches(format!(
                    "{quantifier} of {pattern}"
                )));
            }
            // Preserve the selector so eval evaluates it natively (count-based,
            // reporting every matching detection) and convert can resolve it
            // like the parser path. Expanding here would change reported
            // matched-selections and blow up combinatorially for `N of`.
            Ok(IrCondition::Selector {
                quantifier: quantifier.clone(),
                pattern: pattern.clone(),
            })
        }
    }
}

/// Lower a correlation rule into `IrCorrelation`.
pub fn lower_correlation(corr: &CorrelationRule) -> Result<IrCorrelation> {
    Ok(IrCorrelation {
        metadata: metadata_from_correlation(corr),
        sigma_version: corr.sigma_version,
        correlation_type: corr.correlation_type,
        rules: corr.rules.clone(),
        group_by: corr.group_by.clone(),
        timespan: corr.timespan.clone(),
        window: corr.window,
        gap: corr.gap.clone(),
        condition: corr.condition.clone(),
        aliases: corr.aliases.clone(),
        generate: corr.generate,
    })
}

/// Lower a filter rule into `IrFilter`.
pub fn lower_filter(filter: &FilterRule, opts: &LowerOptions) -> Result<IrFilter> {
    let mut detections = HashMap::new();
    for (name, detection) in &filter.detection.named {
        detections.insert(name.clone(), lower_detection(detection, opts)?);
    }
    let conditions = lower_conditions(&filter.detection)?;
    Ok(IrFilter {
        metadata: metadata_from_filter(filter),
        sigma_version: filter.sigma_version,
        rules: filter.rules.clone(),
        logsource: filter.logsource.clone(),
        detections,
        conditions,
    })
}

fn reject_placeholders(value: &SigmaValue) -> Result<()> {
    if let SigmaValue::String(s) = value
        && s.original.contains("${source.")
    {
        return Err(IrError::Lowering(format!(
            "unresolved source placeholder in detection value: {}",
            s.original
        )));
    }
    Ok(())
}

fn metadata_from_rule(rule: &SigmaRule) -> IrRuleMetadata {
    IrRuleMetadata {
        title: rule.title.clone(),
        id: rule.id.clone(),
        name: rule.name.clone(),
        level: rule.level,
        tags: rule.tags.clone(),
        status: rule.status,
        description: rule.description.clone(),
        author: rule.author.clone(),
        date: rule.date.clone(),
        modified: rule.modified.clone(),
        references: rule.references.clone(),
        falsepositives: rule.falsepositives.clone(),
        fields: rule.fields.clone(),
        related: rule.related.clone(),
        license: rule.license.clone(),
        taxonomy: rule.taxonomy.clone(),
        scope: rule.scope.clone(),
        custom_attributes: yaml_to_json_map(&rule.custom_attributes),
        schema_affinity: None,
    }
}

fn metadata_from_correlation(corr: &CorrelationRule) -> IrRuleMetadata {
    IrRuleMetadata {
        title: corr.title.clone(),
        id: corr.id.clone(),
        name: corr.name.clone(),
        level: corr.level,
        tags: corr.tags.clone(),
        status: corr.status,
        description: corr.description.clone(),
        author: corr.author.clone(),
        date: corr.date.clone(),
        modified: corr.modified.clone(),
        references: corr.references.clone(),
        falsepositives: corr.falsepositives.clone(),
        fields: corr.fields.clone(),
        related: corr.related.clone(),
        license: corr.license.clone(),
        taxonomy: corr.taxonomy.clone(),
        scope: corr.scope.clone(),
        custom_attributes: yaml_to_json_map(&corr.custom_attributes),
        schema_affinity: None,
    }
}

fn metadata_from_filter(filter: &FilterRule) -> IrRuleMetadata {
    IrRuleMetadata {
        title: filter.title.clone(),
        id: filter.id.clone(),
        name: filter.name.clone(),
        level: filter.level,
        tags: filter.tags.clone(),
        status: filter.status,
        description: filter.description.clone(),
        author: filter.author.clone(),
        date: filter.date.clone(),
        modified: filter.modified.clone(),
        references: filter.references.clone(),
        falsepositives: filter.falsepositives.clone(),
        fields: filter.fields.clone(),
        related: filter.related.clone(),
        license: filter.license.clone(),
        taxonomy: filter.taxonomy.clone(),
        scope: filter.scope.clone(),
        custom_attributes: yaml_to_json_map(&filter.custom_attributes),
        schema_affinity: None,
    }
}