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:02:56 -08:00
parent d3613246b5
commit 3897ce08cd
1 changed files with 28 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
server.py
View file

@ -20,9 +20,34 @@ def external():
@app.route("/cross-server.png")
def crosserver():
return send_file("cross-server.png")
@app.route("/loadcert")
def loadcert():
return send_file("loadcert.html")
def allowed_file(filename):
return '.' in filename and filename.rsplit('.', 1)[1].lower() == "pem"
@app.route("/certload", methods=['GET', 'POST'])
def loadcert():
if request.method == 'POST':
if 'cert' not in request.files:
return "Invalid request. (cert is missing!)"
if 'privkey' not in request.files:
return "Invalid request. (privkey is missing!)"
cert = request.files['cert']
privkey = request.files['privkey']
if cert.filename == '' or privkey.filename == '':
return "Please upload a certificate and private key."
if allowed_file(cert.filename) and allowed_file(privkey.filename):
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")
privkey.save(homefolder + "/certs/" + randomname + "-privkey.pem")
return redirect("/")
else:
return "Both files must be a .pem file, you might want to generate a certificate via the home page."
else:
return "Cannot go to /certload with GET, perhaps you're looking for /loadcert"
@app.route("/gencert.zip")
def gencert():
random_name = ''.join(random.choice(string.ascii_lowercase) for i in range(8))
random_name = ''.join(random.choice(string.ascii_lowercase+string.digits+string.ascii_uppercase) for i in range(8))
k = crypto.PKey()
k.generate_key(crypto.TYPE_RSA, 1024)
cert = crypto.X509()
@ -45,6 +70,8 @@ def gencert():
for file_name, data in [('cert.pem', io.BytesIO(crypto.dump_certificate(crypto.FILETYPE_PEM, cert))),
('privkey.pem', io.BytesIO(crypto.dump_privatekey(crypto.FILETYPE_PEM, k)))]:
zip_file.writestr(file_name, data.getvalue())
open(homefolder + "/certs/" + random_name + "-chain.pem", "wb").write(crypto.dump_certificate(crypto.FILETYPE_PEM, cert))
open(homefolder + "/certs/" + random_name + "-privkey.pem", "wb").write(crypto.dump_privatekey(crypto.FILETYPE_PEM, k))
resp = make_response(zip_buffer.getvalue())
resp.set_cookie('certname',random_name)
resp.mimetype = "application/zip"