123 lines
4.0 KiB
Python
123 lines
4.0 KiB
Python
import socketserver
|
|
import signal
|
|
import random
|
|
import os
|
|
|
|
|
|
class Task(socketserver.BaseRequestHandler):
|
|
def _recvall(self):
|
|
BUFF_SIZE = 2048
|
|
data = b''
|
|
while True:
|
|
part = self.request.recv(BUFF_SIZE)
|
|
data += part
|
|
if len(part) < BUFF_SIZE:
|
|
break
|
|
return data.strip()
|
|
|
|
def send(self, msg, newline=True):
|
|
try:
|
|
if newline:
|
|
msg += b'\n'
|
|
self.request.sendall(msg)
|
|
except:
|
|
pass
|
|
|
|
def recv(self, prompt=b'[-] '):
|
|
self.send(prompt, newline=False)
|
|
return self._recvall()
|
|
|
|
def select_maze(self, seed:int):
|
|
# 11 * 11 的maze
|
|
if seed % 3 == 0:
|
|
maze = [[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
|
[1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1],
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
|
[1, 1, 1, 0, 1, 0, 1, 0, 1, 0, 1],
|
|
[1, 0, 0, 0, 1, 0, 1, 0, 1, 0, 1],
|
|
[1, 0, 1, 0, 1, 0, 1, 1, 1, 1, 1],
|
|
[1, 0, 1, 0, 1, 0, 0, 0, 0, 0, 1],
|
|
[1, 0, 1, 0, 1, 1, 1, 0, 1, 0, 1],
|
|
[1, 0, 1, 0, 1, 0, 0, 0, 1, 0, 0],
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]]
|
|
start = [1, 0]
|
|
end = [9, 10]
|
|
elif seed % 3 == 1:
|
|
maze = [[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
|
|
[1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1],
|
|
[1, 0, 1, 0, 1, 1, 1, 0, 1, 0, 1],
|
|
[1, 0, 1, 0, 0, 0, 0, 0, 1, 0, 1],
|
|
[1, 0, 1, 1, 1, 0, 1, 0, 1, 0, 1],
|
|
[1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1],
|
|
[1, 0, 1, 0, 0, 0, 0, 1, 0, 0, 1],
|
|
[1, 0, 1, 1, 1, 1, 0, 1, 1, 0, 1],
|
|
[1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 1],
|
|
[1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1],
|
|
[1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1]]
|
|
start = [10, 3]
|
|
end = [10, 5]
|
|
else:
|
|
maze = [[1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1],
|
|
[1, 0, 0, 0, 1, 0, 1, 0, 0, 0, 1],
|
|
[1, 1, 1, 0, 1, 0, 1, 0, 1, 0, 1],
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 1],
|
|
[1, 0, 1, 1, 1, 1, 1, 0, 1, 0, 1],
|
|
[1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 1],
|
|
[1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1],
|
|
[1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 0],
|
|
[1, 0, 1, 0, 1, 0, 0, 0, 1, 0, 1],
|
|
[1, 0, 0, 0, 1, 1, 1, 0, 0, 0, 1],
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]]
|
|
start = [0, 10]
|
|
end = [7, 10]
|
|
return maze, start, end
|
|
|
|
def game(self):
|
|
self.send(b"Try to solve this maze")
|
|
self.send(b"Use wasd to move")
|
|
maze, start, end = self.select_maze(random.randint(0,99))
|
|
position = start
|
|
while position != end:
|
|
move = self.recv(b'Try to move: ').decode()
|
|
if len(move) != 1:
|
|
self.send(b'Invalid lenth')
|
|
return False
|
|
if move not in 'wasd':
|
|
self.send(b'Invalid instruction')
|
|
return False
|
|
if move == "w":
|
|
pass
|
|
elif move == "s":
|
|
pass
|
|
elif move == "a":
|
|
pass
|
|
else:
|
|
pass
|
|
|
|
|
|
def handle(self):
|
|
signal.alarm(60) # type: ignore
|
|
if not self.game():
|
|
self.send(b'[!] Wrong!')
|
|
return
|
|
|
|
self.send(b'here is your flag')
|
|
self.send(flag)
|
|
|
|
|
|
class ThreadedServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
|
|
pass
|
|
|
|
|
|
class ForkedServer(socketserver.ForkingMixIn, socketserver.TCPServer):
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
flag = bytes(os.getenv("FLAG"),"utf-8")
|
|
HOST, PORT = '0.0.0.0', 10001
|
|
server = ForkedServer((HOST, PORT), Task)
|
|
server.allow_reuse_address = True
|
|
print(HOST, PORT)
|
|
server.serve_forever() |