pmat 3.15.0

PMAT - Zero-config AI context generation and code quality toolkit (CLI, MCP, HTTP)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159

// ==================== Test Helpers ====================
// Included into coverage_tests mod in wasm_handler_tests.rs
// NO use statements allowed here - they are in the parent module.

/// Minimal valid WASM module (empty module with proper header)
fn minimal_wasm_module() -> Vec<u8> {
    vec![
        0x00, 0x61, 0x73, 0x6d, // WASM magic number
        0x01, 0x00, 0x00, 0x00, // WASM version 1
    ]
}

/// WASM module with a simple function that does i32 arithmetic
fn simple_function_wasm() -> Vec<u8> {
    vec![
        0x00, 0x61, 0x73, 0x6d, // magic
        0x01, 0x00, 0x00, 0x00, // version
        // Type section
        0x01, 0x05, // section id 1, size 5
        0x01, // 1 type
        0x60, 0x00, 0x01, 0x7f, // func type: () -> i32
        // Function section
        0x03, 0x02, // section id 3, size 2
        0x01, 0x00, // 1 function, type 0
        // Code section
        0x0a, 0x09, // section id 10, size 9
        0x01, // 1 function body
        0x07, // body size 7
        0x00, // 0 locals
        0x41, 0x01, // i32.const 1
        0x41, 0x02, // i32.const 2
        0x6a, // i32.add
        0x0b, // end
    ]
}

/// Create a temp file with WASM content
fn create_wasm_temp_file(content: &[u8]) -> NamedTempFile {
    let mut file = NamedTempFile::new().expect("Failed to create temp file");
    file.write_all(content)
        .expect("Failed to write to temp file");
    file
}

/// Create a mock AnalysisResult for testing
fn mock_analysis_result() -> AnalysisResult {
    AnalysisResult {
        function_count: 10,
        instruction_count: 500,
        binary_size: 2048,
        memory_pages: 2,
        max_complexity: 15,
    }
}

/// Create a mock VerificationResult (safe)
fn mock_verification_safe() -> VerificationResult {
    VerificationResult::Safe
}

/// Create a mock VerificationResult (unsafe)
fn mock_verification_unsafe() -> VerificationResult {
    VerificationResult::OutOfBounds {
        offset: 100,
        size: 50,
    }
}

/// Create mock security results (no critical)
fn mock_security_results_no_critical() -> Vec<VulnerabilityMatch> {
    vec![
        VulnerabilityMatch {
            pattern: "potential-integer-overflow".to_string(),
            location: 0..100,
            severity: Severity::Medium,
            operator_index: 42,
        },
        VulnerabilityMatch {
            pattern: "timing-side-channel".to_string(),
            location: 100..200,
            severity: Severity::Low,
            operator_index: 77,
        },
    ]
}

/// Create mock security results (with critical)
fn mock_security_results_with_critical() -> Vec<VulnerabilityMatch> {
    vec![
        VulnerabilityMatch {
            pattern: "critical-vulnerability".to_string(),
            location: 0..100,
            severity: Severity::Critical,
            operator_index: 10,
        },
        VulnerabilityMatch {
            pattern: "high-vulnerability".to_string(),
            location: 100..200,
            severity: Severity::High,
            operator_index: 20,
        },
    ]
}

/// Create a mock ProfilingReport
fn mock_profiling_report() -> ProfilingReport {
    ProfilingReport {
        instruction_mix: InstructionMix {
            total_instructions: 100,
            control_flow: 20,
            memory_ops: 30,
            arithmetic: 40,
            calls: 10,
        },
        hot_functions: vec![
            HotFunction {
                name: "func_0".to_string(),
                samples: 500,
                percentage: 50.0,
            },
            HotFunction {
                name: "func_1".to_string(),
                samples: 300,
                percentage: 30.0,
            },
        ],
        memory_usage: MemoryProfile {
            initial_pages: 1,
            max_pages: Some(16),
            growth_events: vec![GrowthEvent {
                timestamp: 1000,
                pages_before: 1,
                pages_after: 2,
            }],
        },
    }
}

/// Create a mock QualityAssessment (passing)
fn mock_quality_assessment_passing() -> QualityAssessment {
    QualityAssessment {
        violations: vec![],
        overall_health: 95.0,
        recommendation: "Quality metrics are within acceptable bounds.".to_string(),
    }
}

/// Create a mock QualityAssessment (failing)
fn mock_quality_assessment_failing() -> QualityAssessment {
    QualityAssessment {
        violations: vec![Violation::ComplexityRegression {
            current: 25,
            limit: 20,
            severity: BaselineSeverity::Error,
        }],
        overall_health: 45.0,
        recommendation: "Critical violations detected".to_string(),
    }
}