swee/Gem2Browser
swee/Gem2Browser
1
0
Fork 0
Code Issues Pull requests Activity Actions

Update server.py

Browse source
This commit is contained in:
Nova Cat 2025-01-14 20:36:26 -08:00
parent f1106f33b7
commit 43123a0606
1 changed files with 11 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
server.py
View file

@ -40,7 +40,9 @@ def loadcert():
random_name = ''.join(random.choice(string.ascii_lowercase+string.digits+string.ascii_uppercase) for i in range(8)) random_name = ''.join(random.choice(string.ascii_lowercase+string.digits+string.ascii_uppercase) for i in range(8))
cert.save(homefolder + "/certs/" + randomname + "-chain.pem") cert.save(homefolder + "/certs/" + randomname + "-chain.pem")
privkey.save(homefolder + "/certs/" + randomname + "-privkey.pem") privkey.save(homefolder + "/certs/" + randomname + "-privkey.pem")
return redirect("/") resp = make_response("<meta http-equiv=\"refresh\" content=\"0; url=//\" />Success!")
resp.set_cookie('certname',random_name)
return resp
else: else:
return "Both files must be a .pem file, you might want to generate a certificate via the home page." return "Both files must be a .pem file, you might want to generate a certificate via the home page."
else: else:
@ -87,6 +89,7 @@ def relay():
print(request.headers.get('User-Agent')) print(request.headers.get('User-Agent'))
url = request.args.get('gemini') url = request.args.get('gemini')
queries = request.args.get('query') queries = request.args.get('query')
certfile = request.cookies.get('certname')
if url == None: if url == None:
return redirect("/") return redirect("/")
code = "<h1>Something went wrong...</h1>\n" code = "<h1>Something went wrong...</h1>\n"
@ -95,7 +98,13 @@ def relay():
try: try:
gsocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) gsocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
fulladdr = "gemini://" + url + ("?" + queries if queries != None else "") fulladdr = "gemini://" + url + ("?" + queries if queries != None else "")
gemsocket = ssl._create_unverified_context().wrap_socket(gsocket, server_hostname=urlparse(fulladdr).hostname) contx = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
if certfile:
contx.load_cert_chain(certfile=homefolder + "/certs/" + certfile + "-chain.pem", keyfile=homefolder + "/certs/" + certfile + "-privkey.pem");
context.load_verify_locations(cafile=ssl.CERT_AU);
context.verify_mode = ssl.CERT_OPTIONAL
contx.check_hostname = False;
gemsocket = contx.wrap_socket(gsocket)
gemsocket.connect((urlparse(fulladdr).hostname, 1965)) gemsocket.connect((urlparse(fulladdr).hostname, 1965))
gemsocket.send(bytes(fulladdr + "\r\n", "UTF-8")) gemsocket.send(bytes(fulladdr + "\r\n", "UTF-8"))
received = "" received = ""