58 lines
2.5 KiB
Python
58 lines
2.5 KiB
Python
from random import choice as c
|
||
from random import randint, shuffle
|
||
from Crypto.Util.number import * # type: ignore
|
||
|
||
flag = b"flag{this_is_a_test_flag}"
|
||
|
||
# 白名单列表，包含允许在问题中使用的关键词和符号
|
||
white_list = ['==', '(', ')', 'C0', 'C1', 'C2', 'C3', 'C4', 'C5', 'C6', '0', '1', 'and', 'or', 'not']
|
||
|
||
# 定义一个函数 calc，用于计算问题的结果
|
||
def calc(ans, chests, expr):
|
||
try:
|
||
C0, C1, C2, C3, C4, C5, C6 = chests
|
||
r = eval(expr)
|
||
except Exception as e:
|
||
print("Shannon fails to understand your words.\n", e)
|
||
exit(0)
|
||
return ans(r)
|
||
|
||
# 定义一个游戏回合函数 do_round
|
||
def do_round():
|
||
# 定义两个函数 truth 和 lie，用于判断 Shannon 的回答是否为真或假
|
||
truth = lambda r: not not r
|
||
lie = lambda r: not r
|
||
chests = []
|
||
for i in range(7):
|
||
# 随机生成 7 个 True 或 False，表示每个箱子是宝藏还是怪物
|
||
chests.append(c((True, False)))
|
||
print("Seven chests lie here, with mimics or treasure hidden inside.\nBut don't worry. Trusty Shannon knows what to do.")
|
||
# 随机选择 Shannon 的回答策略，包括多少次 lie
|
||
lie_count = c((1, 2))
|
||
Shannon = [truth] * (15 - lie_count) + [lie] * lie_count
|
||
# 猜猜第几次是在说谎 :-)
|
||
shuffle(Shannon)
|
||
for i in range(15):
|
||
print("Ask Shannon:")
|
||
question = input().strip() # 输入问题
|
||
for word in question.split(" "):
|
||
if word not in white_list: # 检查问题中是否包含白名单以外的关键词或符号
|
||
print("({}) No treasure for dirty hacker!".format(word))
|
||
exit(0)
|
||
res = str(calc(Shannon[i], chests, question)) # 计算问题的结果
|
||
print('Shannon answers: {}!\n'.format(res))
|
||
print("Now open the chests:")
|
||
# 输入0或1表示每个宝箱的真假，用空格隔开
|
||
chests_string = input()
|
||
return chests == list(map(int, chests_string.strip().split(" ")))
|
||
|
||
# 游戏开始
|
||
print("The Unbreakable Shannon has returned, with some suspicious chests and a far more complicated strategy -- he MAY LIE ONCE OR TWICE! Can you still get all the treasure without losing your head?")
|
||
|
||
for i in range(1):
|
||
if not do_round(): # 执行游戏回合
|
||
print("A chest suddenly comes alive and BITE YOUR HEAD OFF.\n")
|
||
exit(0)
|
||
print("You've found all the treasure! {}\n".format(flag)) # 赢得游戏，获得flag
|
||
|