Enhance AI Security Roles and Data Protection

sensitive_data_scanner/scanner.py

@@ -13,7 +13,11 @@
     "Credit Card (Mastercard)": re.compile(r"5[1-5][0-9]{14}"),
     "Credit Card (Amex)": re.compile(r"3[47][0-9]{13}"),
     "Credit Card (Discover)": re.compile(r"6(?:011|5[0-9]{2})[0-9]{12}"),
-    "Social Security Number": re.compile(r"\d{3}-\d{2}-\d{4}")
+    "Social Security Number": re.compile(r"\d{3}-\d{2}-\d{4}"),
+    "Passport Number (US)": re.compile(r"[a-zA-Z0-9]{9}"),
+    "GCP Service Account Key": re.compile(r"\"type\": \"service_account\""),
+    "Azure Client Secret": re.compile(r"[a-zA-Z0-9-_~.]{34}"),
+    "Health ID (HIPAA)": re.compile(r"H[0-9]{10}")
 }
 
 def scan_file(filepath):

src/SupplyChainPlatform.jsx

@@ -102,6 +102,7 @@ export default function SupplyChainPlatform() {
         <button onClick={() => setActiveTab('twin')}>Digital Twin (3D)</button>
         <button onClick={() => setActiveTab('incoterms')}>Incoterms</button>
         <button onClick={() => setActiveTab('logistics')}>Logistics AI</button>
+        <button onClick={() => setActiveTab('security')}>Security & Protection</button>
       </div>
 
       {activeTab === 'twin' && (
@@ -150,6 +151,30 @@ export default function SupplyChainPlatform() {
         </div>
       )}
 
+      {activeTab === 'security' && (
+        <div style={{ display: 'grid', gridTemplateColumns: '1fr 1fr', gap: '20px' }}>
+          <div style={{ background: '#222', padding: '20px', borderRadius: '10px' }}>
+            <h3>Infrastructure Protection</h3>
+            <p>Monitoring critical IoT sensors for tampering or anomalies.</p>
+            <div style={{ padding: '10px', border: '1px solid #444', borderRadius: '5px', marginBottom: '10px' }}>
+              <strong>Device #842 Status:</strong> <span style={{ color: '#00ff00' }}>SECURE</span><br/>
+              <small>Voltage: 3.3V | Temp: 24°C | RSSI: -42dBm</small>
+            </div>
+            <button onClick={() => logToBlockchain('Infrastructure Health Scan')}>Run AI Perimeter Scan</button>
+          </div>
+
+          <div style={{ background: '#222', padding: '20px', borderRadius: '10px' }}>
+            <h3>Antivirus Identification</h3>
+            <p>AI-driven identification of malicious behavior and file patterns.</p>
+            <div style={{ padding: '10px', border: '1px solid #444', borderRadius: '5px', marginBottom: '10px' }}>
+              <strong>Latest Scan:</strong> No threats detected.<br/>
+              <small>Last behavior scan: {new Date().toLocaleTimeString()}</small>
+            </div>
+            <button onClick={() => logToBlockchain('Malware Signature Update')}>Update AI Signatures</button>
+          </div>
+        </div>
+      )}
+
       {activeTab === 'logistics' && (
         <div style={{ display: 'grid', gridTemplateColumns: '1fr 1fr', gap: '20px' }}>
           <div style={{ background: '#222', padding: '20px', borderRadius: '10px' }}>

supply_chain_platform/security_tools.py

@@ -0,0 +1,103 @@
+import re
+import random
+
+class InfrastructureProtectionAI:
+    """AI role for protecting critical infrastructure and IoT devices."""
+
+    def detect_iot_tampering(self, device_data):
+        """
+        Analyzes IoT device telemetry for signs of physical or digital tampering.
+
+        Args:
+            device_data (dict): Telemetry data including voltage, temperature, and signal strength.
+        """
+        anomalies = []
+
+        # Heuristic: Rapid voltage drop might indicate a power-side attack or battery tampering
+        if device_data.get('voltage', 3.3) < 2.8:
+            anomalies.append("Low voltage detected - possible power source tampering.")
+
+        # Heuristic: Temperature spikes outside industrial operating range
+        if device_data.get('temperature', 25) > 75:
+            anomalies.append("Extreme temperature spike - potential hardware stress or overheating attack.")
+
+        # Heuristic: Signal RSSI fluctuations
+        if device_data.get('rssi', -50) < -90:
+            anomalies.append("Weak signal (low RSSI) - potential signal jamming or interference.")
+
+        if not anomalies:
+            return {"status": "SECURE", "score": 0, "findings": ["Normal operating parameters."]}
+        else:
+            return {
+                "status": "WARNING",
+                "score": len(anomalies) * 3,
+                "findings": anomalies
+            }
+
+    def assess_facility_vulnerability(self, access_logs):
+        """
+        AI assessment of facility security based on access logs.
+        """
+        unauthorized_attempts = [log for log in access_logs if log.get('status') == 'DENIED']
+
+        if len(unauthorized_attempts) > 5:
+            return "HIGH RISK: Multiple unauthorized access attempts detected at perimeter."
+        elif len(unauthorized_attempts) > 0:
+            return "MEDIUM RISK: Occasional unauthorized access attempts detected."
+        else:
+            return "LOW RISK: Perimeter security appears intact."
+
+
+class AntivirusIdentificationAI:
+    """AI role for identifying malware signatures and suspicious file behaviors."""
+
+    SUSPICIOUS_EXTENSIONS = ['.exe', '.sh', '.bat', '.bin', '.scr']
+
+    def scan_file_metadata(self, filename, filesize_kb):
+        """
+        Identifies potential threats based on file metadata heuristics.
+        """
+        findings = []
+        ext = '.' + filename.split('.')[-1] if '.' in filename else ''
+
+        if ext.lower() in self.SUSPICIOUS_EXTENSIONS:
+            findings.append(f"Suspicious executable extension: {ext}")
+
+        if filesize_kb < 1:
+            findings.append("Unusually small file size - potential dropper or script.")
+
+        if not findings:
+            return {"risk": "LOW", "details": "File metadata appears standard."}
+        else:
+            return {"risk": "MEDIUM", "details": findings}
+
+    def identify_malware_behavior_patterns(self, execution_logs):
+        """
+        Scans execution logs for behavior patterns consistent with malware (e.g. ransomware, spyware).
+        """
+        patterns = {
+            "Ransomware": ["mass_file_rename", "encryption_started", "delete_shadow_copies"],
+            "Spyware": ["unauthorized_camera_access", "keystroke_logging", "exfiltrating_data"],
+            "Worm": ["rapid_network_scanning", "self_replication_attempt"]
+        }
+
+        detected_threats = []
+        logs_flat = " ".join(execution_logs).lower()
+
+        for threat, indicators in patterns.items():
+            for indicator in indicators:
+                if indicator in logs_flat:
+                    detected_threats.append(f"{threat} indicator: {indicator}")
+
+        return detected_threats if detected_threats else ["No malicious behavior patterns detected."]
+
+if __name__ == "__main__":
+    # Test Infrastructure Protection
+    infra_ai = InfrastructureProtectionAI()
+    test_device = {'voltage': 2.5, 'temperature': 80, 'rssi': -95}
+    print("IoT Tampering Analysis:", infra_ai.detect_iot_tampering(test_device))
+
+    # Test Antivirus ID
+    av_ai = AntivirusIdentificationAI()
+    print("File Scan:", av_ai.scan_file_metadata("update.bat", 0.5))
+    print("Behavior Analysis:", av_ai.identify_malware_behavior_patterns(["encryption_started", "delete_shadow_copies"]))

supply_chain_platform/supply_chain_main.py

@@ -1,6 +1,7 @@
 import json
 import os
 from ai_logistics_engine import AILogisticsEngine
+from security_tools import InfrastructureProtectionAI, AntivirusIdentificationAI
 
 def load_incoterms():
     path = os.path.join(os.path.dirname(__file__), 'incoterms_data.json')
@@ -12,16 +13,19 @@ def display_menu():
     print("1. Incoterms Lookup")
     print("2. AI Delivery Delay Predictor")
     print("3. Inventory Risk Analysis")
-    print("4. Exit")
+    print("4. Security Analysis (Infrastructure & AV)")
+    print("5. Exit")
     print("============================================")
 
 def main():
     incoterms = load_incoterms()
     ai_engine = AILogisticsEngine()
+    infra_ai = InfrastructureProtectionAI()
+    av_ai = AntivirusIdentificationAI()
 
     while True:
         display_menu()
-        choice = input("Enter choice (1-4): ").strip()
+        choice = input("Enter choice (1-5): ").strip()
 
         if choice == '1':
             print("\nAvailable Incoterms:", ", ".join(incoterms.keys()))
@@ -53,6 +57,27 @@ def main():
                 print("Invalid numbers.")
 
         elif choice == '4':
+            print("\n--- Security Analysis ---")
+            print("1. IoT Tampering Detection")
+            print("2. Antivirus Metadata Scan")
+            sec_choice = input("Select sub-option (1-2): ").strip()
+
+            if sec_choice == '1':
+                v = float(input("Enter device voltage: "))
+                t = float(input("Enter device temperature: "))
+                r = float(input("Enter signal RSSI: "))
+                result = infra_ai.detect_iot_tampering({'voltage': v, 'temperature': t, 'rssi': r})
+                print(f"\nResult: {result['status']} (Score: {result['score']})")
+                for f in result['findings']:
+                    print(f" - {f}")
+            elif sec_choice == '2':
+                fname = input("Enter filename: ")
+                fsize = float(input("Enter file size (KB): "))
+                result = av_ai.scan_file_metadata(fname, fsize)
+                print(f"\nRisk Level: {result['risk']}")
+                print(f"Details: {result['details']}")
+
+        elif choice == '5':
             print("Exiting Supply Chain Platform.")
             break
         else: