fix: handle deletion exceptions

GZTimeWalker · Apr 29, 2024 · d625547 · d625547
1 parent d7c93d2
commit d625547
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/src/GZCTF/Controllers/GameController.cs b/src/GZCTF/Controllers/GameController.cs
@@ -493,15 +493,22 @@ public IActionResult GetTeamTraffic([FromRoute] int challengeId, [FromRoute] int
     [ProducesResponseType(typeof(RequestResponse), StatusCodes.Status404NotFound)]
     public IActionResult DeleteAllTeamTraffic([FromRoute] int challengeId, [FromRoute] int partId)
     {
-        var filePath = Path.Combine(FilePath.Capture, $"{challengeId}", $"{partId}");
+        try
+        {
+            var filePath = Path.Combine(FilePath.Capture, $"{challengeId}", $"{partId}");
 
-        if (!Path.Exists(filePath))
-            return NotFound(new RequestResponse(localizer[nameof(Resources.Program.Game_CaptureNotFound)],
-                StatusCodes.Status404NotFound));
+            if (!Path.Exists(filePath))
+                return NotFound(new RequestResponse(localizer[nameof(Resources.Program.Game_CaptureNotFound)],
+                    StatusCodes.Status404NotFound));
 
-        Directory.Delete(filePath, true);
+            Directory.Delete(filePath, true);
 
-        return Ok();
+            return Ok();
+        }
+        catch (Exception e)
+        {
+            return BadRequest(new RequestResponse(e.Message));
+        }
     }
 
     /// <summary>