ProxyC/Basis/Check.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import copy
import time
from Checker import Checker
from Basis.Logger import logging
from Builder import Builder, clientEntry


def Check(taskId: str, taskInfo: dict) -> dict:
    logging.info('[%s] Start checking process -> %s' % (taskId, taskInfo))
    if taskInfo['type'] not in clientEntry:
        logging.error('[%s] Unknown proxy type %s' % (taskId, taskInfo['type']))
        raise RuntimeError('Unknown proxy type')

    try:
        client = Builder(
            proxyType = taskInfo['type'],
            proxyInfo = taskInfo['info'],
            bindAddr = '127.0.0.1',  # socks5 exposed host
            taskId = taskId,
        )
    except Exception as reason:
        logging.error('[%s] Client build error -> %s' % (taskId, reason))
        raise RuntimeError('Client build error')
    logging.info('[%s] Client loaded successfully')

    # TODO: wait port occupied (client.socksPort)
    time.sleep(1)
    if not client.status():  # client unexpected exit
        logging.warning('[%s] Client unexpected exit')
        client.destroy()  # remove file and kill sub process
        logging.debug('[%s] Client output\n%s', client.output)
        raise RuntimeError('Client unexpected exit')

    checkResult = Checker(taskId, taskInfo['check'], {
        'addr': client.socksAddr,
        'port': client.socksPort,
    })
    logging.info('[%s] Client check result -> %s' % (taskId, checkResult))
    client.destroy()  # clean up the client
    taskInfo = copy.deepcopy(taskInfo)
    taskInfo.pop('check')
    return {
        **taskInfo,
        'result': checkResult,
    }