From fbcc6dc5116406ed06353b32bdf4ca1f2a70c13e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= Date: Wed, 15 Aug 2018 13:06:45 +0200 Subject: [PATCH] Pass hostname to auth token request Related to T56381 --- benchmark/submission/client.py | 7 +++++-- benchmark/submission/sockutil.py | 4 ++++ 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/benchmark/submission/client.py b/benchmark/submission/client.py index e623514..e0e752c 100644 --- a/benchmark/submission/client.py +++ b/benchmark/submission/client.py @@ -7,7 +7,7 @@ import urllib.parse import requests -from . import timeouts, exceptions +from . import timeouts, exceptions, sockutil log = logging.getLogger(__name__) @@ -137,7 +137,10 @@ class BenchmarkClient: self._start_http_server() - params = urllib.parse.urlencode({'auth_callback': self.auth_http_server.auth_callback_url}) + params = urllib.parse.urlencode({ + 'auth_callback': self.auth_http_server.auth_callback_url, + 'hostname': sockutil.hostname(), + }) url = f"{self.url_generate_token}?{params}" if not webbrowser.open_new_tab(url): diff --git a/benchmark/submission/sockutil.py b/benchmark/submission/sockutil.py index a7119ef..8371799 100644 --- a/benchmark/submission/sockutil.py +++ b/benchmark/submission/sockutil.py @@ -56,3 +56,7 @@ def find_free_port(start_address): port_nr += 1 return hostname, port_nr + + +def hostname() -> str: + return socket.gethostname()