Merge pull request #74 from pomo-mondreganto/volga-mode

Add VolgaCTF-style attacks mode

Author: pomo-mondreganto

backend/lib/helpers/exceptions.py

@@ -8,3 +8,4 @@ class FlagExceptionEnum:
     FLAG_TOO_OLD = FlagSubmitException('Flag is too old')
     FLAG_YOUR_OWN = FlagSubmitException('Flag is your own')
     FLAG_ALREADY_STOLEN = FlagSubmitException('Flag already stolen')
+    SERVICE_IS_DOWN = FlagSubmitException('Cannot submit flags while service is down')

backend/lib/models/game_config.py

@@ -13,6 +13,8 @@ class GameConfig(BaseModel):
     flag_lifetime: int
     game_hardness: float
     inflation: bool
+    volga_attacks_mode: bool
+
     round_time: int
     mode: str
     timezone: str
@@ -28,6 +30,7 @@ class GameConfig(BaseModel):
         'flag_lifetime',
         'game_hardness',
         'inflation',
+        'volga_attacks_mode',
         'round_time',
         'mode',
         'timezone',

backend/lib/storage/attacks.py

@@ -1,6 +1,7 @@
 from lib import models, storage
 from lib.helpers import exceptions
 from lib.helpers.exceptions import FlagExceptionEnum
+from lib.models.types import TaskStatus
 from lib.storage import utils, game
 from lib.storage.keys import CacheKeys
 
@@ -47,6 +48,15 @@ def handle_attack(
         if current_round - flag.round > game_config.flag_lifetime:
             raise FlagExceptionEnum.FLAG_TOO_OLD
 
+        if game_config.volga_attacks_mode:
+            teamtask = storage.tasks.get_latest_teamtask(
+                team_id=attacker_id,
+                task_id=flag.task_id,
+            )
+            # Status is a string from redis stream.
+            if not teamtask or teamtask['status'] != str(TaskStatus.UP.value):
+                raise FlagExceptionEnum.SERVICE_IS_DOWN
+
         result.victim_id = flag.team_id
         result.task_id = flag.task_id
         success = storage.flags.try_add_stolen_flag(

backend/lib/storage/tasks.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 
 from lib import models, storage
 from lib.helpers.cache import cache_helper
@@ -187,6 +187,20 @@ def get_teamtasks_for_team(team_id: int) -> List[dict]:
     return results
 
 
+def get_latest_teamtask(team_id: int, task_id: int) -> Optional[dict]:
+    """Fetch the latest teamtask from redis stream."""
+
+    with storage.utils.redis_pipeline(transaction=False) as pipe:
+        data, = pipe.xrevrange(CacheKeys.teamtasks(team_id, task_id), count=1).execute()
+
+    results = []
+    for timestamp, record in data:
+        record['timestamp'] = timestamp
+        results.append(record)
+
+    return results[0] if results else None
+
+
 def filter_teamtasks_for_participants(teamtasks: List[dict]) -> List[dict]:
     """
     Filter sensitive data from teamtasks.

backend/scripts/create_tables.sql

@@ -1,15 +1,16 @@
 CREATE TABLE IF NOT EXISTS GameConfig
 (
-    id            SERIAL PRIMARY KEY,
-    game_running  BOOLEAN     DEFAULT FALSE,
-    real_round    INTEGER     DEFAULT 0,
-    flag_lifetime INTEGER CHECK ( flag_lifetime > 0 ),
-    game_hardness FLOAT CHECK ( game_hardness >= 1 ),
-    inflation     BOOLEAN,
-    round_time    INTEGER CHECK ( round_time > 0 ),
-    mode          VARCHAR(8)  DEFAULT 'classic',
-    timezone      VARCHAR(32) DEFAULT 'UTC',
-    start_time    TIMESTAMP WITH TIME ZONE
+    id                 SERIAL PRIMARY KEY,
+    game_running       BOOLEAN     DEFAULT FALSE,
+    real_round         INTEGER     DEFAULT 0,
+    flag_lifetime      INTEGER CHECK ( flag_lifetime > 0 ),
+    game_hardness      FLOAT CHECK ( game_hardness >= 1 ),
+    inflation          BOOLEAN,
+    volga_attacks_mode BOOLEAN,
+    round_time         INTEGER CHECK ( round_time > 0 ),
+    mode               VARCHAR(8)  DEFAULT 'classic',
+    timezone           VARCHAR(32) DEFAULT 'UTC',
+    start_time         TIMESTAMP WITH TIME ZONE
 );
 
 CREATE TABLE IF NOT EXISTS Teams

backend/services/tasks/handlers.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Union
 
 from celery import shared_task
 from celery.result import AsyncResult
@@ -52,7 +52,7 @@ def exception_callback(result: AsyncResult, exc: Exception, traceback: str) -> N
 
 @shared_task(name=JobNames.result_handler)
 def checker_results_handler(
-        verdicts: List[models.CheckerVerdict],
+        verdicts: Union[List[models.CheckerVerdict], models.CheckerVerdict],
         team: models.Team,
         task: models.Task,
         current_round: int,
@@ -63,6 +63,12 @@ def checker_results_handler(
     If there were any errors, the first error is returned
     Otherwise, verdict of the first action's verdict is returned.
     """
+
+    # Celery passes the one-task-group's result (e.g. a group of a single put)
+    # as the result itself, not the list, as documented.
+    if not isinstance(verdicts, list):
+        verdicts = [verdicts]
+
     check_verdict = None
     puts_verdicts = []
     gets_verdict = None

cli/models.py

@@ -49,6 +49,7 @@ class GameConfig(BaseModel):
     mode: str = 'classic'
     get_period: Optional[int]
     inflation: bool = True
+    volga_attacks_mode: bool = False
 
     checkers_path: str = '/checkers/'
     env_path: str = ''

control.py

@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 from cli import cli
 

front/dist/assets/AdminLogin.c88c2238.js

@@ -0,0 +1 @@
+.create-group[data-v-27ed27a5]{display:flex;flex-flow:row-reverse}.create-btn[data-v-27ed27a5]{font-size:1em;margin:1em}

front/dist/assets/AdminScoreboard.3ed728a9.css

@@ -0,0 +1 @@
+.error-message[data-v-671fbd7e]{color:red}

front/dist/assets/AdminScoreboard.418e3b89.js

@@ -0,0 +1 @@
+.error-message[data-v-324400ce]{color:red}