mirror of
https://github.com/ap-pauloafonso/ratio-spoof.git
synced 2026-05-06 18:19:31 +00:00
core done, needs some fixes
This commit is contained in:
Paulo Afonso Pinheiro
parent
30169758b9
commit
6ead5e6655
1 changed files with 133 additions and 25 deletions
158
ratio-spoof.py
158
ratio-spoof.py
|
|
@ -14,6 +14,9 @@ import subprocess
|
|||
import platform
|
||||
import datetime
|
||||
import threading
|
||||
import urllib.request
|
||||
import http.client
|
||||
|
||||
|
||||
def human_readable_size(size, decimal_places=2):
|
||||
for unit in ['B','KiB','MiB','GiB','TiB']:
|
||||
|
|
@ -30,27 +33,77 @@ def clear_screen():
|
|||
print("\033c", end="")
|
||||
|
||||
class RatioSpoofState():
|
||||
def __init__(self, torrent_name, percent_complete, download_speed, upload_speed, announce_rate, current_downloaded, current_uploaded):
|
||||
def __init__(self, torrent_name, percent_complete, download_speed, upload_speed, \
|
||||
announce_rate, current_downloaded, current_uploaded,\
|
||||
piece_size, total_size, announce_info, info_hash_urlencoded):
|
||||
self.__lock = threading.Lock()
|
||||
self.torrent_name = torrent_name
|
||||
self.percent_complete = percent_complete
|
||||
self.download_speed = download_speed
|
||||
self.upload_speed = upload_speed
|
||||
self.announce_rate = announce_rate
|
||||
self.announce_current_timer = self.announce_rate
|
||||
self.piece_size = piece_size
|
||||
self.total_size = total_size
|
||||
self.announce_info = announce_info
|
||||
self.peer_id = peer_id()
|
||||
self.key = key()
|
||||
self.info_hash_urlencoded = info_hash_urlencoded
|
||||
self.announce_history_deq = deque(maxlen=10)
|
||||
self.deq_count = 0
|
||||
self.announce_current_timer = self.announce_rate
|
||||
self.add_announce(current_downloaded, current_uploaded)
|
||||
threading.Thread(target = (lambda: self.__print_state())).start()
|
||||
self.__add_announce(current_downloaded, current_uploaded ,next_announce_left_b(current_downloaded, total_size))
|
||||
#threading.Thread(target = (lambda: self.__print_state())).start()
|
||||
threading.Thread(target = (lambda: self.__decrease_timer())).start()
|
||||
|
||||
def add_announce(self, current_downloaded, current_uploaded):
|
||||
|
||||
def __add_announce(self, current_downloaded, current_uploaded, left):
|
||||
self.deq_count +=1
|
||||
self.announce_history_deq.append({'count': self.deq_count, 'downloaded':current_downloaded,'uploaded':current_uploaded })
|
||||
self.announce_history_deq.append({'count': self.deq_count, 'downloaded':current_downloaded,'uploaded':current_uploaded,'left': left })
|
||||
|
||||
def generate_next_announce(self, announce_rate):
|
||||
self.__reset_timer(announce_rate)
|
||||
current_downloaded = self.announce_history_deq[-1]['downloaded']
|
||||
if(self.announce_history_deq[-1]['downloaded'] < self.total_size):
|
||||
current_downloaded = next_announce_total_b(self.download_speed,self.announce_history_deq[-1]['downloaded'], self.piece_size, self.announce_rate, self.total_size)
|
||||
current_uploaded = next_announce_total_b(self.upload_speed,self.announce_history_deq[-1]['uploaded'], self.piece_size, self.announce_rate)
|
||||
current_left = next_announce_left_b(current_downloaded, self.total_size)
|
||||
self.__add_announce(current_downloaded,current_uploaded,current_left)
|
||||
|
||||
def announce(self, event = None):
|
||||
last_announce_data = self.announce_history_deq[-1]
|
||||
query_dict = build_query_string(self, last_announce_data, event)
|
||||
success = False
|
||||
print(len(self.announce_info['list_of_lists']))
|
||||
if (len(self.announce_info['list_of_lists']) > 0):
|
||||
for tier_list in self.announce_info['list_of_lists']:
|
||||
for item in tier_list:
|
||||
print(item)
|
||||
return tracker_announce_request(item, query_dict)
|
||||
success = True
|
||||
break
|
||||
|
||||
|
||||
else:
|
||||
url = self.announce_info['main']
|
||||
ret = tracker_announce_request(url, query_dict)
|
||||
success = True
|
||||
return ret
|
||||
|
||||
print('passou por aqui 22')
|
||||
if success == False : raise Exception('Connection error with the tracker')
|
||||
|
||||
def __decrease_timer(self):
|
||||
self.announce_current_timer = self.announce_current_timer - 1 if self.announce_current_timer > 0 else 0
|
||||
def reset_timer(self, new_announce_rate = None):
|
||||
while True:
|
||||
time.sleep(1)
|
||||
with self.__lock:
|
||||
self.announce_current_timer = self.announce_current_timer - 1 if self.announce_current_timer > 0 else 0
|
||||
|
||||
def __reset_timer(self, new_announce_rate = None):
|
||||
if new_announce_rate != None:
|
||||
self.announce_rate = new_announce_rate
|
||||
self.announce_current_timer = self.announce_rate
|
||||
with self.__lock:
|
||||
self.announce_current_timer = self.announce_rate
|
||||
|
||||
def __print_state(self):
|
||||
while True:
|
||||
print(f"""
|
||||
|
|
@ -58,14 +111,15 @@ class RatioSpoofState():
|
|||
Torrent: {self.torrent_name} - {self.percent_complete}%
|
||||
download_speed: {self.download_speed}KB/s
|
||||
upload_speed: {self.upload_speed}KB/s
|
||||
size: {human_readable_size(self.total_size)}
|
||||
###########################################################################
|
||||
""")
|
||||
for item in list(self.announce_history_deq)[:len(self.announce_history_deq)-1]:
|
||||
print(f'#{item["count"]} downloaded: {human_readable_size(item["downloaded"])} | uploaded: {human_readable_size(item["uploaded"])} | announced')
|
||||
print(f'#{self.announce_history_deq[-1]["count"]} downloaded: {human_readable_size(self.announce_history_deq[-1]["downloaded"])} | uploaded: {human_readable_size(self.announce_history_deq[-1]["uploaded"])} | next announce in :{str(datetime.timedelta(seconds=self.announce_current_timer))}')
|
||||
self.__decrease_timer()
|
||||
print(f'#{item["count"]} downloaded: {human_readable_size(item["downloaded"])} | left: {human_readable_size(item["left"])} | uploaded: {human_readable_size(item["uploaded"])} | announced')
|
||||
print(f'#{self.announce_history_deq[-1]["count"]} downloaded: {human_readable_size(self.announce_history_deq[-1]["downloaded"])} | left: {human_readable_size(self.announce_history_deq[-1]["left"])} | uploaded: {human_readable_size(self.announce_history_deq[-1]["uploaded"])} | next announce in :{str(datetime.timedelta(seconds=self.announce_current_timer))}')
|
||||
clear_screen()
|
||||
time.sleep(1)
|
||||
time.sleep(1)
|
||||
|
||||
|
||||
|
||||
def t_total_size(data):
|
||||
|
|
@ -91,28 +145,82 @@ def next_announce_total_b(kb_speed, b_current, b_piece_size,s_time, b_total_limi
|
|||
def next_announce_left_b(b_current, b_total_size):
|
||||
return b_total_size - b_current
|
||||
|
||||
def peer_id():
|
||||
return f'-qB4030-{base64.urlsafe_b64encode(uuid.uuid4().bytes)[:12].decode()}'
|
||||
|
||||
def peer_id():
|
||||
peer_id = f'-qB4030-{base64.urlsafe_b64encode(uuid.uuid4().bytes)[:12].decode()}'
|
||||
return peer_id
|
||||
|
||||
def key():
|
||||
return hex(random.getrandbits(32))[2:].upper()
|
||||
def find_approx_current(b_total_size, piece_size, percent):
|
||||
if( percent <= 0): return 0
|
||||
total = (percent/100) * b_total_size
|
||||
current_approx = int(total / piece_size) * piece_size
|
||||
return current_approx
|
||||
|
||||
def build_announce_info(data):
|
||||
announce_info = {'main':data['announce'], 'list_of_lists':data['announce-list'] if 'announce-list' in data else []}
|
||||
tcp_list_of_lists = []
|
||||
for _list in announce_info['list_of_lists']:
|
||||
aux = list(filter(lambda x: x.lower().startswith('http'),_list))
|
||||
if len(aux) >0:
|
||||
tcp_list_of_lists.append(aux)
|
||||
announce_info['list_of_lists'] = tcp_list_of_lists
|
||||
if(announce_info['main'].startswith('udp') and len(announce_info['list_of_lists'])== 0):
|
||||
raise Exception('No tcp/http tracker url found')
|
||||
return announce_info
|
||||
|
||||
def tracker_announce_request(url, query_dict):
|
||||
request = urllib.request.Request(url = f'{url}?{urllib.parse.urlencode(query_dict)}', headers= {'User-Agent' :'qBittorrent/4.0.3', 'Accept-Encoding':'gzip'})
|
||||
print(request.full_url)
|
||||
print('tool')
|
||||
return 30
|
||||
#response = urllib.request.urlopen(request)
|
||||
#return response.read()
|
||||
|
||||
def build_query_string(state:RatioSpoofState, curent_info, event):
|
||||
query = {
|
||||
'info_hash': state.info_hash_urlencoded,
|
||||
'peer_id':state.peer_id,
|
||||
'port':str(8999),
|
||||
'uploaded':str(curent_info['uploaded']),
|
||||
'downloaded':str(curent_info['downloaded']),
|
||||
'left':str(curent_info['left']),
|
||||
'corrupt': str(0),
|
||||
'key':str(state.key),
|
||||
'event':event,
|
||||
'numwant':str(200),
|
||||
'compact':str(1),
|
||||
'no_peer_id': str(1),
|
||||
'supportcrypto':str(1),
|
||||
'redundant':str(0)
|
||||
}
|
||||
return urllib.parse.quote_plus(json.dumps(query))
|
||||
|
||||
|
||||
|
||||
def read_file(f, args_downalod, args_upload):
|
||||
raw_data = f.read()
|
||||
result = bencode_parser.decode(raw_data)
|
||||
total_size = t_total_size(result)
|
||||
piece_size = t_piecesize_b(result)
|
||||
total_size = t_total_size(result)
|
||||
current_downloaded = find_approx_current(total_size,piece_size,args_downalod[0])
|
||||
current_uploaded = find_approx_current(total_size,piece_size,args_upload[0])
|
||||
state = RatioSpoofState(result['info']['name'],args_downalod[0],args_downalod[1],args_upload[1],1800, current_downloaded, current_uploaded)
|
||||
|
||||
state = RatioSpoofState(result['info']['name'],args_downalod[0],args_downalod[1],args_upload[1],0,\
|
||||
current_downloaded, current_uploaded, piece_size,total_size,
|
||||
build_announce_info(result),t_infohash_urlencoded(result, raw_data))
|
||||
|
||||
announce_rate = state.announce('started')
|
||||
while True:
|
||||
time.sleep(0.5)
|
||||
state.generate_next_announce(announce_rate)
|
||||
time.sleep(announce_rate)
|
||||
state.announce()
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#perfom_request('http://www.cagometro.com/annouce', {'test1':'test', 'test2':2})
|
||||
|
||||
""" while True:
|
||||
if(current_downloaded < total_size):
|
||||
current_downloaded = next_announce_total_b(args_downalod[1],current_downloaded, piece_size, state.announce_rate, total_size)
|
||||
|
|
@ -123,11 +231,11 @@ def read_file(f, args_downalod, args_upload):
|
|||
"""
|
||||
|
||||
|
||||
|
||||
http.client.HTTPConnection.debuglevel = 1
|
||||
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument('-t', required=True,help='path .torrent file' , type=argparse.FileType('rb'))
|
||||
parser.add_argument('-d', required=True,type=int,help='parms for download', nargs=2 ,metavar=('%_COMPLETE', 'KBS_SPEED'))
|
||||
parser.add_argument('-u',required=True,type=int,help='parms for upload', nargs=2 ,metavar=('%_COMPLETE', 'KBS_SPEED'))
|
||||
parser.add_argument('-t', required=True, metavar=('(TORRENT_PATH)'), help='path .torrent file' , type=argparse.FileType('rb'))
|
||||
parser.add_argument('-d', required=True,type=int,help='parms for download', nargs=2 ,metavar=('(%_COMPLETE)', '(KBS_SPEED)'))
|
||||
parser.add_argument('-u',required=True,type=int,help='parms for upload', nargs=2 ,metavar=('(%_COMPLETE)', '(KBS_SPEED)'))
|
||||
args = parser.parse_args()
|
||||
read_file(args.t, args.d, args.u)
|
||||
|
|
|
|||
Loading…
Reference in a new issue