seba
/
servefile
1
1
Fork 1
Code Issues Pull Requests Releases Wiki Activity

Fix imports and unicode handling for Python3

This commit is contained in:
Sebastian Lohff 2019-02-17 23:14:54 +01:00
parent 93e0c3dd7e
commit 0e33743293
1 changed files with 21 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
servefile
View File

@ -14,7 +14,6 @@ import base64
import cgi import cgi
import datetime import datetime
import mimetypes import mimetypes
import urllib
import os import os
import re import re
import select import select
@ -27,10 +26,12 @@ import time
try: try:
import BaseHTTPServer import BaseHTTPServer
import SocketServer import SocketServer
from urllib import quote, unquote
except ImportError: except ImportError:
# both have different names in python3 # both have different names in python3
import http.server as BaseHTTPServer import http.server as BaseHTTPServer
import socketserver as SocketServer import socketserver as SocketServer
from urllib.parse import quote, unquote
# only activate SSL if available # only activate SSL if available
HAVE_SSL = False HAVE_SSL = False
@ -56,7 +57,7 @@ class FileBaseHandler(BaseHTTPServer.BaseHTTPRequestHandler):
Returns True if a redirect was issued. """ Returns True if a redirect was issued. """
if not fileName: if not fileName:
fileName = self.fileName fileName = self.fileName
if urllib.unquote(self.path) != "/" + fileName: if unquote(self.path) != "/" + fileName:
self.send_response(302) self.send_response(302)
self.send_header('Location', '/' + fileName) self.send_header('Location', '/' + fileName)
self.end_headers() self.end_headers()
@ -346,12 +347,12 @@ class DirListingHandler(FileBaseHandler):
<p>The requestet URL %s was not found on this server</p> <p>The requestet URL %s was not found on this server</p>
<p><a href="/">Back to /</a> <p><a href="/">Back to /</a>
</body> </body>
</html>""" % self.escapeHTML(urllib.unquote(self.path)) </html>""" % self.escapeHTML(unquote(self.path))
self.send_header("Content-Length", str(len(errorMsg))) self.send_header("Content-Length", str(len(errorMsg)))
self.send_header('Connection', 'close') self.send_header('Connection', 'close')
self.end_headers() self.end_headers()
if not head: if not head:
self.wfile.write(errorMsg) self.wfile.write(errorMsg.encode())
def escapeHTML(self, htmlstr): def escapeHTML(self, htmlstr):
entities = [("<", "&lt;"), (">", "&gt;")] entities = [("<", "&lt;"), (">", "&gt;")]
@ -378,7 +379,7 @@ class DirListingHandler(FileBaseHandler):
<td class="size">%s</td> <td class="size">%s</td>
<td class="type">%s</td> <td class="type">%s</td>
</tr> </tr>
""" % (urllib.quote(item), item, lastModified, fileSize, fileType)) """ % (quote(item), item, lastModified, fileSize, fileType))
def sendDirectoryListing(self, path, head): def sendDirectoryListing(self, path, head):
""" Generate a directorylisting for path and send it """ """ Generate a directorylisting for path and send it """
@ -414,7 +415,7 @@ class DirListingHandler(FileBaseHandler):
</tr> </tr>
</thead> </thead>
<tbody> <tbody>
""" % {'path': os.path.normpath(urllib.unquote(self.path))} """ % {'path': os.path.normpath(unquote(self.path))}
footer = """</tbody></table></div> footer = """</tbody></table></div>
<div class="footer"><a href="http://seba-geek.de/stuff/servefile/">servefile %(version)s</a></div> <div class="footer"><a href="http://seba-geek.de/stuff/servefile/">servefile %(version)s</a></div>
<script> <script>
@ -540,7 +541,7 @@ class DirListingHandler(FileBaseHandler):
self.send_header("Content-Length", str(len(listing))) self.send_header("Content-Length", str(len(listing)))
self.send_header('Connection', 'close') self.send_header('Connection', 'close')
self.end_headers() self.end_headers()
self.wfile.write(listing) self.wfile.write(listing.encode())
def convertSize(self, size): def convertSize(self, size):
for ext in "KMGT": for ext in "KMGT":
@ -552,7 +553,7 @@ class DirListingHandler(FileBaseHandler):
return (size, ext.strip()) return (size, ext.strip())
def getCleanPath(self): def getCleanPath(self):
urlPath = os.path.normpath(urllib.unquote(self.path)).strip("/") urlPath = os.path.normpath(unquote(self.path)).strip("/")
path = os.path.join(self.targetDir, urlPath) path = os.path.join(self.targetDir, urlPath)
return path return path
@ -594,7 +595,8 @@ class FilePutter(BaseHTTPServer.BaseHTTPRequestHandler):
length = self.getContentLength() length = self.getContentLength()
if length < 0: if length < 0:
return return
ctype = self.headers.getheader('Content-Type') print(self.headers)
ctype = self.headers.get('Content-Type')
# check for multipart/form-data. # check for multipart/form-data.
if not (ctype and ctype.lower().startswith("multipart/form-data")): if not (ctype and ctype.lower().startswith("multipart/form-data")):
@ -615,7 +617,7 @@ class FilePutter(BaseHTTPServer.BaseHTTPRequestHandler):
return return
# write file down to disk, send a 200 afterwards # write file down to disk, send a 200 afterwards
target = open(destFileName, "w") target = open(destFileName, "wb")
bytesLeft = length bytesLeft = length
while bytesLeft > 0: while bytesLeft > 0:
bytesToRead = min(self.blockSize, bytesLeft) bytesToRead = min(self.blockSize, bytesLeft)
@ -638,7 +640,7 @@ class FilePutter(BaseHTTPServer.BaseHTTPRequestHandler):
if length < 0: if length < 0:
return return
fileName = urllib.unquote(self.path) fileName = unquote(self.path)
if fileName == "/": if fileName == "/":
# if no filename was given we have to generate one # if no filename was given we have to generate one
fileName = str(time.time()) fileName = str(time.time())
@ -685,7 +687,7 @@ class FilePutter(BaseHTTPServer.BaseHTTPRequestHandler):
self.send_header('Content-Length', str(len(msg))) self.send_header('Content-Length', str(len(msg)))
self.send_header('Connection', 'close') self.send_header('Connection', 'close')
self.end_headers() self.end_headers()
self.wfile.write(msg) self.wfile.write(msg.encode())
def getTargetName(self, fname): def getTargetName(self, fname):
""" Generate a clean and secure filename. """ Generate a clean and secure filename.
@ -831,7 +833,7 @@ class ServeFile():
# we couldn't find any ip address # we couldn't find any ip address
proc = None proc = None
if proc: if proc:
ips = proc.stdout.read().strip().split("\n") ips = proc.stdout.read().decode().strip().split("\n")
# filter out ips we are not listening on # filter out ips we are not listening on
if not self.listenIPv6: if not self.listenIPv6:
@ -877,7 +879,7 @@ class ServeFile():
for ip in self.getIPs() + ["127.0.0.1", "::1"]: for ip in self.getIPs() + ["127.0.0.1", "::1"]:
altNames.append("IP:%s" % ip) altNames.append("IP:%s" % ip)
altNames.append("DNS:localhost") altNames.append("DNS:localhost")
ext = crypto.X509Extension("subjectAltName", False, ",".join(altNames)) ext = crypto.X509Extension(b"subjectAltName", False, (",".join(altNames)).encode())
req.add_extensions([ext]) req.add_extensions([ext])
req.set_pubkey(pkey) req.set_pubkey(pkey)
@ -901,8 +903,8 @@ class ServeFile():
self.key = pkey self.key = pkey
print("done.") print("done.")
print("SHA1 fingerprint:", cert.digest("sha1")) print("SHA1 fingerprint:", cert.digest("sha1").decode())
print("MD5 fingerprint:", cert.digest("md5")) print("MD5 fingerprint:", cert.digest("md5").decode())
def _getCert(self): def _getCert(self):
return self.cert return self.cert
@ -913,7 +915,7 @@ class ServeFile():
def setAuth(self, user, password, realm=None): def setAuth(self, user, password, realm=None):
if not user or not password: if not user or not password:
raise ServeFileException("User and password both need to be at least one character.") raise ServeFileException("User and password both need to be at least one character.")
self.auth = base64.b64encode("%s:%s" % (user, password)) self.auth = base64.b64encode(("%s:%s" % (user, password)).encode()).decode()
self.authrealm = realm self.authrealm = realm
def _createServer(self, handler, withv6=False): def _createServer(self, handler, withv6=False):
@ -975,7 +977,7 @@ class ServeFile():
else: else:
pwPart = "" pwPart = ""
if self.auth: if self.auth:
pwPart = base64.b64decode(self.auth) + "@" pwPart = base64.b64decode(self.auth).decode() + "@"
for ip in ips: for ip in ips:
if ":" in ip: if ":" in ip:
ip = "[%s]" % ip ip = "[%s]" % ip
@ -1091,7 +1093,7 @@ class AuthenticationHandler():
errorMsg = "<html><head><title>401 - Unauthorized</title></head><body><h1>401 - Unauthorized</h1></body></html>" errorMsg = "<html><head><title>401 - Unauthorized</title></head><body><h1>401 - Unauthorized</h1></body></html>"
self.send_header("Content-Length", str(len(errorMsg))) self.send_header("Content-Length", str(len(errorMsg)))
self.end_headers() self.end_headers()
self.wfile.write(errorMsg) self.wfile.write(errorMsg.encode())
def main(): def main():