mirror of https://github.com/dnomd343/ProxyC
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.3 KiB
41 lines
1.3 KiB
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import time
|
|
import _thread
|
|
from Basis import DnsProxy
|
|
from Basis.Check import Check
|
|
from Basis.Logger import logging
|
|
from Basis.Manager import Manager
|
|
from Basis.Api import startServer
|
|
from Basis.Constant import Version
|
|
from Basis.Compile import startCompile
|
|
from concurrent.futures import ThreadPoolExecutor
|
|
|
|
# dnsServers = None
|
|
dnsServers = ['223.5.5.5', '119.28.28.28']
|
|
|
|
|
|
def runCheck(taskId: str, taskInfo: dict) -> None:
|
|
checkResult = Check(taskId, taskInfo)
|
|
logging.warning('[%s] Task finish' % taskId)
|
|
Manager.finishTask(taskId, checkResult)
|
|
|
|
|
|
def loopCheck(threadNum: int = 16) -> None:
|
|
threadPool = ThreadPoolExecutor(max_workers = threadNum)
|
|
while True:
|
|
try:
|
|
taskId, taskInfo = Manager.popTask()
|
|
logging.warning('[%s] Load new task' % taskId)
|
|
except: # no more task
|
|
time.sleep(2)
|
|
continue
|
|
threadPool.submit(runCheck, taskId, taskInfo)
|
|
|
|
|
|
logging.warning('ProxyC starts running (%s)' % Version)
|
|
_thread.start_new_thread(startCompile, ('/usr', )) # python compile (generate .pyc file)
|
|
_thread.start_new_thread(DnsProxy.start, (dnsServers, 53)) # start dns server
|
|
_thread.start_new_thread(loopCheck, ()) # start loop check
|
|
startServer(apiToken = '') # start api server
|
|
|