palo/share-host
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

⚡ improve auth handler speed

Browse source
This commit is contained in:
Ingolf Wagner 2024-11-17 07:24:29 +07:00
parent eff7d1c887
commit 95037bb56c
No known key found for this signature in database
GPG key ID: 76BF5F1928B9618B
1 changed files with 16 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
share-via-http.py
View file

@ -21,16 +21,12 @@ def get_current_ip_address():
# Function to generate a default username # Function to generate a default username
def generate_default_username(): def generate_default_username():
return "user_" + "".join( return "".join(random.choices(string.ascii_lowercase + string.digits, k=8))
random.choices(string.ascii_lowercase + string.digits, k=8)
)
# Function to generate a default password # Function to generate a default password
def generate_default_password(): def generate_default_password():
return "".join( return "".join(random.choices(string.ascii_letters + string.digits, k=12))
random.choices(string.ascii_letters + string.digits + string.punctuation, k=12)
)
# Define the command-line arguments # Define the command-line arguments
@ -74,6 +70,12 @@ os.chdir(expanded_directory)
# Define a request handler with basic authentication # Define a request handler with basic authentication
class AuthHandler(http.server.SimpleHTTPRequestHandler): class AuthHandler(http.server.SimpleHTTPRequestHandler):
def __init__(self, username, passsword, *args, **kwargs):
encoded_bytes = base64.b64encode(f"{username}:{passsword}".encode("utf-8"))
# Convert bytes back to string
self.encoded_str = encoded_bytes.decode("utf-8")
super().__init__(*args, **kwargs)
def do_AUTHHEAD(self): def do_AUTHHEAD(self):
self.send_response(401) self.send_response(401)
self.send_header("WWW-Authenticate", 'Basic realm="Server Access"') self.send_header("WWW-Authenticate", 'Basic realm="Server Access"')
@ -90,9 +92,10 @@ class AuthHandler(http.server.SimpleHTTPRequestHandler):
def check_auth(self, auth_header): def check_auth(self, auth_header):
encoded_credentials = auth_header.split()[1] encoded_credentials = auth_header.split()[1]
decoded_credentials = base64.b64decode(encoded_credentials).decode("utf-8") return encoded_credentials == self.encoded_str
username, password = decoded_credentials.split(":") # decoded_credentials = base64.b64decode(encoded_credentials).decode("utf-8")
return username == args.username and password == args.password # username, password = decoded_credentials.split(":")
# return username == args.username and password == args.password
def save_qr_code(url): def save_qr_code(url):
@ -111,8 +114,11 @@ def save_qr_code(url):
img.save("qrcode.png") img.save("qrcode.png")
from functools import partial
# Set up the server with the AuthHandler and the defined port # Set up the server with the AuthHandler and the defined port
with socketserver.TCPServer(("", args.port), AuthHandler) as httpd: auth_handler = partial(AuthHandler, args.username, args.password)
with socketserver.TCPServer(("", args.port), auth_handler) as httpd:
print("qrcode.png is saved in the current directory.") print("qrcode.png is saved in the current directory.")
ip_address = get_current_ip_address() ip_address = get_current_ip_address()
print( print(