fix
This commit is contained in:
@@ -1,26 +1,26 @@
|
||||
from themis.finals.attack.helper import Helper
|
||||
from themis.finals.attack.result import Result
|
||||
|
||||
from models import FlagStatus, SubmitResult
|
||||
|
||||
|
||||
RESPONSES = {
|
||||
FlagStatus.ACCEPTED: [Result.SUCCESS_FLAG_ACCEPTED],
|
||||
FlagStatus.REJECTED: [Result.ERROR_FLAG_EXPIRED, Result.ERROR_FLAG_YOURS,
|
||||
Result.ERROR_FLAG_SUBMITTED, Result.ERROR_FLAG_NOT_FOUND],
|
||||
}
|
||||
|
||||
|
||||
def submit_flags(flags, config):
|
||||
h = Helper(config['SYSTEM_HOST'])
|
||||
codes = h.attack(*[item.flag for item in flags])
|
||||
|
||||
for item, code in zip(flags, codes):
|
||||
for status, possible_codes in RESPONSES.items():
|
||||
if code in possible_codes:
|
||||
found_status = status
|
||||
break
|
||||
else:
|
||||
found_status = FlagStatus.QUEUED
|
||||
|
||||
yield SubmitResult(item.flag, found_status, code.name)
|
||||
from themis.finals.attack.helper import Helper
|
||||
from themis.finals.attack.result import Result
|
||||
|
||||
from models import FlagStatus, SubmitResult
|
||||
|
||||
|
||||
RESPONSES = {
|
||||
FlagStatus.ACCEPTED: [Result.SUCCESS_FLAG_ACCEPTED],
|
||||
FlagStatus.REJECTED: [Result.ERROR_FLAG_EXPIRED, Result.ERROR_FLAG_YOURS,
|
||||
Result.ERROR_FLAG_SUBMITTED, Result.ERROR_FLAG_NOT_FOUND],
|
||||
}
|
||||
|
||||
|
||||
def submit_flags(flags, config):
|
||||
h = Helper(config['SYSTEM_HOST'])
|
||||
codes = h.attack(*[item.flag for item in flags])
|
||||
|
||||
for item, code in zip(flags, codes):
|
||||
for status, possible_codes in RESPONSES.items():
|
||||
if code in possible_codes:
|
||||
found_status = status
|
||||
break
|
||||
else:
|
||||
found_status = FlagStatus.QUEUED
|
||||
|
||||
yield SubmitResult(item.flag, found_status, code.name)
|
||||
|
||||
Reference in New Issue
Block a user