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.
111 lines
2.7 KiB
111 lines
2.7 KiB
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import re
|
|
import time
|
|
import uuid
|
|
import random
|
|
import hashlib
|
|
from IPy import IP
|
|
from Utils.Logger import logger
|
|
|
|
|
|
def isIpAddr(ipAddr: str) -> bool:
|
|
try:
|
|
if '/' in ipAddr: # filter CIDR
|
|
return False
|
|
if '.' not in ipAddr and ':' not in ipAddr: # not IPv4 or IPv6
|
|
return False
|
|
IP(ipAddr) # try to convert to IP address
|
|
return True # valid IP address
|
|
except:
|
|
return False
|
|
|
|
|
|
def isDomain(domain: str) -> bool:
|
|
try:
|
|
domainRegex = r'^(?=^.{3,255}$)[a-zA-Z0-9_][a-zA-Z0-9_-]{0,62}(\.[a-zA-Z0-9_][a-zA-Z0-9_-]{0,62})+$'
|
|
return re.search(domainRegex, domain) is not None # regex matching
|
|
except: # unexpected error
|
|
return False
|
|
|
|
|
|
def isHost(host: str) -> bool:
|
|
return isIpAddr(host) or isDomain(host) # IPv4 / IPv6 / Domain
|
|
|
|
|
|
def isPort(port: int) -> bool:
|
|
if type(port) != int:
|
|
return False
|
|
return port in range(1, 65536) # 1 ~ 65535
|
|
|
|
|
|
def md5Sum(data: str, encode: str = 'utf-8') -> str:
|
|
return hashlib.md5(data.encode(encoding = encode)).hexdigest() # MD5 hash
|
|
|
|
|
|
def hostFormat(host: str, v6Bracket: bool = False) -> str:
|
|
try:
|
|
ip = IP(host)
|
|
if v6Bracket and ip.version() == 6:
|
|
return '[%s]' % str(ip) # [IPv6]
|
|
return str(ip) # IPv4 / IPV6
|
|
except: # not ip address
|
|
return host
|
|
|
|
|
|
def v6AddBracket(host: str) -> str: # add bracket for ipv6
|
|
return hostFormat(host, v6Bracket = True)
|
|
|
|
|
|
def genFlag(length: int = 12) -> str: # generate random task flag
|
|
flag = ''
|
|
for i in range(0, length):
|
|
tmp = random.randint(0, 15)
|
|
if tmp >= 10:
|
|
flag += chr(tmp + 87) # a ~ f
|
|
else:
|
|
flag += str(tmp) # 0 ~ 9
|
|
logger.debug('Generate new flag -> ' + flag)
|
|
return flag
|
|
|
|
|
|
def genUUID() -> str: # generate uuid v5
|
|
return str(uuid.uuid5(
|
|
uuid.NAMESPACE_DNS, genFlag(length = 16)
|
|
))
|
|
|
|
|
|
def toInt(raw) -> int:
|
|
try:
|
|
return int(raw)
|
|
except:
|
|
raise RuntimeError('Unable convert to int')
|
|
|
|
|
|
def toStr(raw) -> str:
|
|
if raw is None:
|
|
raise RuntimeError('None could not convert to str')
|
|
if isinstance(raw, bytes): # bytes -> str
|
|
return str(raw, encoding = 'utf-8')
|
|
try:
|
|
return str(raw)
|
|
except:
|
|
raise RuntimeError('Unable convert to str')
|
|
|
|
|
|
def toStrTidy(raw) -> str:
|
|
return toStr(raw).strip().lower() # with trim and lower
|
|
|
|
|
|
def toBool(raw) -> bool:
|
|
if isinstance(raw, (bool, int, float)):
|
|
return bool(raw)
|
|
try:
|
|
raw = toStr(raw).strip().lower()
|
|
if raw in ['true', 'false']:
|
|
return True if raw == 'true' else False
|
|
return int(raw) != 0
|
|
except:
|
|
raise RuntimeError('Unable convert to bool')
|
|
|
|
|