use super::super::{CertRule, RuleViolation};
use crate::manifest::{RuleCategory, Severity};
use crate::utility::cert_c::ast_utils::get_node_text;
use tree_sitter::Node;
pub struct Mem07C;
impl CertRule for Mem07C {
fn rule_id(&self) -> &'static str {
"MEM07-C"
}
fn description(&self) -> &'static str {
"Ensure that the arguments to calloc(), when multiplied, do not wrap"
}
fn severity(&self) -> Severity {
Severity::High
}
fn category(&self) -> RuleCategory {
RuleCategory::Rule
}
fn cert_id(&self) -> &'static str {
self.rule_id()
}
fn check(&self, root: &Node, source: &str) -> Vec<RuleViolation> {
let mut violations = Vec::new();
self.check_node(root, source, &mut violations);
violations
}
}
impl Mem07C {
fn check_node(&self, node: &Node, source: &str, violations: &mut Vec<RuleViolation>) {
if node.kind() == "call_expression" {
if let Some(func_node) = node.child_by_field_name("function") {
let func_name = get_node_text(&func_node, source);
if func_name == "calloc" {
if !self.has_overflow_check_before(node, source) {
violations.push(RuleViolation {
rule_id: self.rule_id().to_string(),
severity: self.severity(),
message: "calloc() arguments may wrap when multiplied; check for overflow before allocation".to_string(),
file_path: String::new(),
line: node.start_position().row + 1,
column: node.start_position().column + 1,
suggestion: Some(
"Add overflow check: if (num_elements > SIZE_MAX/element_size) { /* handle error */ }".to_string()
),
..Default::default()
});
}
}
}
}
let mut cursor = node.walk();
for child in node.children(&mut cursor) {
self.check_node(&child, source, violations);
}
}
fn has_overflow_check_before(&self, calloc_node: &Node, source: &str) -> bool {
let calloc_start = calloc_node.start_byte();
let before_calloc = &source[..calloc_start];
if before_calloc.contains("SIZE_MAX") && before_calloc.contains("/") {
return true;
}
if (before_calloc.contains("UINT_MAX") || before_calloc.contains("ULONG_MAX"))
&& before_calloc.contains("/")
{
return true;
}
false
}
}