27 lines
850 B
Python
27 lines
850 B
Python
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)
|