49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
def main(request, response):
|
|
def fail(message):
|
|
response.content = b"FAIL: " + message
|
|
response.status = 400
|
|
|
|
def getState(token):
|
|
server_state = request.server.stash.take(token)
|
|
if not server_state:
|
|
return b"Uninitialized"
|
|
return server_state
|
|
|
|
def setState(token, state):
|
|
request.server.stash.put(token, state)
|
|
|
|
def resetState(token):
|
|
setState(token, b"")
|
|
|
|
response.headers.set(b"Cache-Control", b"no-store")
|
|
response.headers.set(b"Access-Control-Allow-Origin", request.headers.get(b"origin"))
|
|
response.headers.set(b"Access-Control-Max-Age", 1)
|
|
token = request.GET.first(b"token", None)
|
|
state = getState(token)
|
|
command = request.GET.first(b"command", None)
|
|
|
|
if command == b"reset":
|
|
if request.method == u"GET":
|
|
resetState(token)
|
|
response.content = b"Server state reset"
|
|
else:
|
|
fail(b"Invalid Method.")
|
|
elif state == b"Uninitialized":
|
|
if request.method == u"OPTIONS":
|
|
response.content = b"This request should not be displayed."
|
|
setState(token, b"Denied")
|
|
else:
|
|
fail(state)
|
|
elif state == b"Denied":
|
|
if request.method == u"GET" and command == b"complete":
|
|
resetState(token)
|
|
response.content = b"Request successfully blocked."
|
|
else:
|
|
setState(token, b"Deny Ignored")
|
|
fail(b"The request was not denied.")
|
|
elif state == b"Deny Ignored":
|
|
resetState(token)
|
|
fail(state)
|
|
else:
|
|
resetState(token)
|
|
fail(b"Unknown Error.")
|