file server fix, added ctrl+c handler for rev shells

This commit is contained in:
2023-12-19 13:48:18 +01:00
parent e4132c3468
commit 8c42a9065a
2 changed files with 38 additions and 11 deletions

View File

@@ -101,7 +101,7 @@ class FileServerRequestHandler(BaseHTTPRequestHandler):
if path in self.server.dumpRequests:
headers["Access-Control-Allow-Origin"] = "*"
headers["Content-Length"] = len(data)
headers["Content-Length"] = len(util.nvl(data, b""))
if len(headers) == 0:
self.send_response(status_code)
@@ -162,7 +162,10 @@ class HttpFileServer(HTTPServer):
return path.strip()
def addFile(self, name, data, mimeType=None):
def addFile(self, name, data, mime_type=None):
assert isinstance(name, str)
assert data is not None
if hasattr(data, "read"):
fd = data
@@ -176,8 +179,8 @@ class HttpFileServer(HTTPServer):
"Access-Control-Allow-Origin": "*",
}
if mimeType:
headers["Content-Type"] = mimeType
if mime_type:
headers["Content-Type"] = mime_type
# return 200 - OK and data
self.addRoute(name, lambda req: (200, data, headers))