move all the functions in the global namespace to the gopherHandler() class

This commit is contained in:
randomuser 2021-07-20 02:35:29 -05:00
parent a2ac3369e4
commit 3c159a5729
1 changed files with 96 additions and 96 deletions

62
esgd.py
View File

@ -3,7 +3,7 @@ import subprocess
import os
host = "localhost"
port = 71
port = 77
class GopherError(BaseException): pass
class RequestError(GopherError): pass
@ -51,14 +51,15 @@ class GopherLine:
self.port
)
def recieveRequest(context):
class gopherHandler(socketserver.BaseRequestHandler):
def recieveRequest(self):
data = b""
while data[-2:] != b"\r\n":
data += context.request.recv(1)
data += self.request.recv(1)
decoded = data[:-2].decode("utf-8")
return decoded
def requestParser(request):
def requestParser(self, request):
if ".." in request: raise RequestError
try:
if request[0] == "/": request = request[1:]
@ -67,7 +68,7 @@ def requestParser(request):
try: return (request[0], request[1])
except IndexError: return (request[0], "")
def returnRelative(file):
def returnRelative(self, file):
gph = False
if file == "":
ret = "gophermap"
@ -79,34 +80,34 @@ def returnRelative(file):
else: raise RequestError("unreachable state")
return (ret, gph)
def fileSendable(file): return os.access(file, os.F_OK|os.R_OK)
def fileSendable(self, file): return os.access(file, os.F_OK|os.R_OK)
def fileCGI(file):
def fileCGI(self, file):
return os.access(file, os.F_OK|os.R_OK|os.X_OK)
def notFound(context):
context.request.sendall(b"3error: file not found!\r\n.\r\n")
def notFound(self):
self.request.sendall(b"3error: file not found!\r\n.\r\n")
def invalid(context):
context.request.sendall(b"3error: selector contains '..'!\r\n.\r\n")
def invalid(self):
self.request.sendall(b"3error: selector contains '..'!\r\n.\r\n")
def fileToFileArray(file):
def fileToFileArray(self, file):
fd = open(file, "r")
ret = [i.rstrip() for i in fd.readlines()]
fd.close()
return ret
def sendFileArray(arr, context):
[context.request.sendall(
def sendFileArray(self, arr):
[self.request.sendall(
(i.rstrip() + "\r\n").encode("utf-8")
) for i in arr]
context.request.sendall(b".\r\n")
self.request.sendall(b".\r\n")
def cgi(file, query, context):
def cgi(self, file, query):
env = {}
env["QUERY_STRING"] = query
env["SCRIPT_NAME"] = "/" + file
env["REMOTE_ADDR"] = context.client_address[0]
env["REMOTE_ADDR"] = self.client_address[0]
proc = subprocess.Popen(
[os.getcwd() + "/" + file],
@ -121,7 +122,7 @@ def cgi(file, query, context):
return out.decode("utf-8").replace("\r", "").split("\n")
def gopherRenderer(fileArray):
def gopherRenderer(self, fileArray):
gopherlines = []
for i in fileArray:
gopherlines.append(GopherLine(i))
@ -130,29 +131,28 @@ def gopherRenderer(fileArray):
returned.append(i.render())
return returned
def serveFile(file, query, gph, context):
if fileCGI(file): fa = cgi(file, query, context)
elif fileSendable(file): fa = fileToFileArray(file)
def serveFile(self, file, query, gph):
if self.fileCGI(file): fa = self.cgi(file, query)
elif self.fileSendable(file): fa = self.fileToFileArray(file)
else:
notFound(context)
self.notFound()
return
if gph: fa = gopherRenderer(fa)
sendFileArray(fa, context)
if gph: fa = self.gopherRenderer(fa)
self.sendFileArray(fa)
class gopherHandler(socketserver.BaseRequestHandler):
def handle(self):
decoded = recieveRequest(self)
decoded = self.recieveRequest()
try:
parsed = requestParser(decoded)
parsed = self.requestParser(decoded)
except RequestError:
invalid(self)
self.invalid()
return
try: file = returnRelative(parsed[0])
try: file = self.returnRelative(parsed[0])
except RequestError:
notFound(self)
self.notFound()
return
serveFile(file[0], parsed[1], file[1], self)
self.serveFile(file[0], parsed[1], file[1])
if __name__ == "__main__":
with socketserver.TCPServer((host, port), gopherHandler) as server: