Add new upload mode
This commit is contained in:
parent
b36366b3f9
commit
49b0e153e4
68
hyp.py
68
hyp.py
@ -1,15 +1,56 @@
|
||||
#!/usr/bin/env python3
|
||||
import argparse
|
||||
import sys
|
||||
import os.path
|
||||
import http.server
|
||||
import ssl
|
||||
import cgi
|
||||
import os
|
||||
import os.path as path
|
||||
import http.server as http
|
||||
|
||||
|
||||
upload_page = '''
|
||||
<form method="POST" enctype="multipart/form-data">
|
||||
File to upload: <input type="file" name="upfile"><br>
|
||||
<input type="submit" value="Send">
|
||||
</form>
|
||||
'''
|
||||
|
||||
class UploadHandler(http.BaseHTTPRequestHandler):
|
||||
def do_GET(self):
|
||||
self.send_response(200)
|
||||
self.send_header('content-type', 'text/html')
|
||||
self.end_headers()
|
||||
self.wfile.write(upload_page.encode('utf-8'))
|
||||
|
||||
def do_POST(self):
|
||||
self.send_response(200)
|
||||
self.end_headers()
|
||||
|
||||
form = cgi.FieldStorage(
|
||||
environ={
|
||||
'REQUEST_METHOD':'POST',
|
||||
'CONTENT_TYPE': self.headers['content-type']},
|
||||
fp=self.rfile,
|
||||
headers=self.headers)
|
||||
|
||||
for field in form.keys():
|
||||
item = form[field]
|
||||
name = path.join(os.getcwd(), item.filename)
|
||||
|
||||
if path.exists(name):
|
||||
self.wfile.write('{} failed: already exists\n'.format(
|
||||
item.filename).encode('utf-8'))
|
||||
continue
|
||||
with open(name, 'wb') as file:
|
||||
file.write(item.file.read())
|
||||
self.wfile.write('{} has been uploaded\n'.format(
|
||||
item.filename).encode('utf-8'))
|
||||
|
||||
|
||||
def check_cert(dir):
|
||||
try:
|
||||
for i in 'cert', 'key':
|
||||
open(os.path.join(dir, 'https-%s.pem' % i)).close()
|
||||
open(path.join(dir, 'https-%s.pem' % i)).close()
|
||||
except (FileNotFoundError, PermissionError) as e:
|
||||
print('Could not use %s: %s. Using http only.' % (dir, e.args[1]))
|
||||
return False
|
||||
@ -19,16 +60,20 @@ def check_cert(dir):
|
||||
return True
|
||||
|
||||
|
||||
def main(address, port, tls_dir):
|
||||
def main(address, port, tls_dir, upload):
|
||||
use_tls = check_cert(tls_dir)
|
||||
port = 80 if port == 443 and not use_tls else port
|
||||
bind = (address, port)
|
||||
|
||||
try:
|
||||
server = http.server.HTTPServer((address, port),
|
||||
http.server.SimpleHTTPRequestHandler)
|
||||
if upload:
|
||||
server = http.HTTPServer(bind, UploadHandler)
|
||||
else:
|
||||
server = http.HTTPServer(bind, http.SimpleHTTPRequestHandler)
|
||||
if use_tls:
|
||||
tls_socket = ssl.wrap_socket(server.socket, server_side=True,
|
||||
certfile=os.path.join(tls_dir, 'https-cert.pem'),
|
||||
keyfile=os.path.join(tls_dir,'https-key.pem'),
|
||||
certfile=path.join(tls_dir, 'https-cert.pem'),
|
||||
keyfile=path.join(tls_dir,'https-key.pem'),
|
||||
ssl_version=ssl.PROTOCOL_TLSv1_2)
|
||||
server.socket = tls_socket
|
||||
except Exception as e:
|
||||
@ -52,10 +97,13 @@ if __name__ == '__main__':
|
||||
parser.add_argument('port', nargs='?',
|
||||
type=int, default=443,
|
||||
help='bind port number (default: 80 or 443)')
|
||||
parser.add_argument('-t', '--tls', type=str,
|
||||
parser.add_argument('-u', '--upload', action='store_true',
|
||||
help='only accept and save file as POST requests')
|
||||
parser.add_argument('-t', '--tls',
|
||||
type=str, dest='tls_dir', metavar='DIR',
|
||||
default='/usr/local/etc/openssl',
|
||||
help='cert/key couple directory. Must be PEM '
|
||||
'formatted and named https-key.pem, https-cert.pem '
|
||||
'(default: /usr/local/etc/openssl)')
|
||||
args = parser.parse_args()
|
||||
main(args.address, args.port, args.tls)
|
||||
main(**vars(args))
|
||||
|
Loading…
Reference in New Issue
Block a user