From 49b0e153e45abee85683efd97ff8c05d392cce5c Mon Sep 17 00:00:00 2001
From: Rnhmjoj <micheleguerinirocco@me.com>
Date: Sat, 28 Mar 2015 18:39:08 +0100
Subject: [PATCH] Add new upload mode

---
 hyp.py | 72 ++++++++++++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 60 insertions(+), 12 deletions(-)

diff --git a/hyp.py b/hyp.py
index 00a7eba..466fb1a 100644
--- a/hyp.py
+++ b/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'
+                        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))