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 Pos04C;
const UNSAFE_MUTEX_TYPES: &[&str] = &["PTHREAD_MUTEX_NORMAL", "PTHREAD_MUTEX_DEFAULT"];
impl CertRule for Pos04C {
fn rule_id(&self) -> &'static str {
"POS04-C"
}
fn description(&self) -> &'static str {
"Avoid using PTHREAD_MUTEX_NORMAL type mutex locks"
}
fn severity(&self) -> Severity {
Severity::Low
}
fn category(&self) -> RuleCategory {
RuleCategory::Recommendation
}
fn cert_id(&self) -> &'static str {
"POS04-C"
}
fn check(&self, node: &Node, source: &str) -> Vec<RuleViolation> {
let mut violations = Vec::new();
self.check_node(node, source, &mut violations);
violations
}
}
impl Pos04C {
fn check_node(&self, node: &Node, source: &str, violations: &mut Vec<RuleViolation>) {
if node.kind() == "call_expression" {
self.check_mutex_settype(node, source, violations);
}
for i in 0..node.child_count() {
if let Some(child) = node.child(i) {
self.check_node(&child, source, violations);
}
}
}
fn check_mutex_settype(&self, node: &Node, source: &str, violations: &mut Vec<RuleViolation>) {
if let Some(func) = node.child_by_field_name("function") {
let func_name = get_node_text(&func, source);
if func_name == "pthread_mutexattr_settype" {
if let Some(args) = node.child_by_field_name("arguments") {
let mut arg_idx = 0;
for j in 0..args.child_count() {
if let Some(arg) = args.child(j) {
let arg_kind = arg.kind();
if arg_kind != "(" && arg_kind != ")" && arg_kind != "," {
arg_idx += 1;
if arg_idx == 2 {
let type_text = get_node_text(&arg, source);
if UNSAFE_MUTEX_TYPES.contains(&type_text) {
let pos = node.start_position();
violations.push(RuleViolation {
rule_id: self.rule_id().to_string(),
severity: Severity::Low,
message: format!(
"Use of {} mutex type has undefined behavior; use PTHREAD_MUTEX_ERRORCHECK or PTHREAD_MUTEX_RECURSIVE instead",
type_text
),
file_path: String::new(),
line: pos.row + 1,
column: pos.column + 1,
suggestion: Some(
"Replace with PTHREAD_MUTEX_ERRORCHECK or PTHREAD_MUTEX_RECURSIVE".to_string()
),
..Default::default()
});
}
}
}
}
}
}
}
}
}
}