Skip to content

Commit

Permalink
Fix incorrect switching to https when receiving BEP34 TCP preference
Browse files Browse the repository at this point in the history
  • Loading branch information
CorralPeltzer committed Nov 24, 2024
1 parent 69f5692 commit c070bbb
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 6 deletions.
11 changes: 6 additions & 5 deletions newTrackon/tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ def update_status(self):
self.to_be_deleted = True
raise RuntimeError("Tracker unresponsive for too long, removed")

self.update_from_bep_34()
self.update_scheme_from_bep_34()
self.update_ips()
except RuntimeError as reason:
self.clear_tracker(reason=str(reason))
Expand Down Expand Up @@ -124,7 +124,7 @@ def update_status(self):
self.interval = 10800
self.update_uptime()

def update_from_bep_34(self):
def update_scheme_from_bep_34(self):
valid_bep_34, bep_34_info = scraper.get_bep_34(self.host)
if valid_bep_34: # Hostname has a valid TXT record as per BEP34
if not bep_34_info:
Expand All @@ -139,11 +139,12 @@ def update_from_bep_34(self):
)
parsed_url = parse.urlparse(self.url)
# Update tracker with the first protocol and URL set by TXT record
first_bep_34_result = bep_34_info[0]
new_scheme = "https" if first_bep_34_result[0] == "tcp" else "udp"
first_bep_34_protocol, first_bep_34_port = bep_34_info[0]
existing_scheme = parsed_url.scheme
new_scheme = "udp" if first_bep_34_protocol == "udp" else existing_scheme
self.url = parsed_url._replace(
scheme=new_scheme,
netloc=f"{parsed_url.hostname}:{first_bep_34_result[1]}",
netloc=f"{parsed_url.hostname}:{first_bep_34_port}",
).geturl()
return
else: # No valid BEP34, attempting existing URL
Expand Down
2 changes: 1 addition & 1 deletion newTrackon/trackon.py
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,7 @@ def process_new_tracker(tracker_candidate: Tracker):
def update_outdated_trackers():
while True:
now = int(time())
trackers_outdated = []
trackers_outdated: list[Tracker] = []
for tracker in db.get_all_data():
if (now - tracker.last_checked) > tracker.interval:
trackers_outdated.append(tracker)
Expand Down

0 comments on commit c070bbb

Please sign in to comment.