Add URL checks to reverse image search commands

d4n3436 · Mar 3, 2024 · 0c18698 · 0c18698
1 parent 0ed346e
commit 0c18698
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/Modules/ImageModule.cs b/src/Modules/ImageModule.cs
@@ -217,6 +217,11 @@ public async Task<RuntimeResult> ReverseAsync([Summary(description: "The URL of
     public async Task<RuntimeResult> ReverseAsync(string url, ReverseImageSearchEngine engine, bool multiImages,
         IDiscordInteraction interaction, IDiscordInteraction? originalInteraction = null, bool ephemeral = false)
     {
+        if (!Uri.IsWellFormedUriString(url, UriKind.Absolute))
+        {
+            return FergunResult.FromError(_localizer["UrlNotWellFormed"], true, interaction);
+        }
+
         _logger.LogDebug("Performing reverse image search (engine: {Engine}, multi-images: {MultiImages}, ephemeral: {Ephemeral})", engine, multiImages, ephemeral);
 
         return engine switch