From cb598371b06f8a2c1842e3fe6c0f876d0b2007db Mon Sep 17 00:00:00 2001
From: Kavin <20838718+FireMasterK@users.noreply.github.com>
Date: Sat, 4 Mar 2023 03:03:06 +0000
Subject: [PATCH] Add check for query length.

---
 .../java/me/kavin/piped/server/handlers/SearchHandlers.java | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/main/java/me/kavin/piped/server/handlers/SearchHandlers.java b/src/main/java/me/kavin/piped/server/handlers/SearchHandlers.java
index 075f851..be4c1c7 100644
--- a/src/main/java/me/kavin/piped/server/handlers/SearchHandlers.java
+++ b/src/main/java/me/kavin/piped/server/handlers/SearchHandlers.java
@@ -28,6 +28,9 @@ public class SearchHandlers {
         if (StringUtils.isEmpty(query))
             ExceptionHandler.throwErrorResponse(new InvalidRequestResponse("query is a required parameter"));
 
+        if (query.length() > 100)
+            ExceptionHandler.throwErrorResponse(new InvalidRequestResponse("query is too long"));
+
         return mapper.writeValueAsBytes(YOUTUBE_SERVICE.getSuggestionExtractor().suggestionList(query));
 
     }
@@ -38,6 +41,9 @@ public class SearchHandlers {
         if (StringUtils.isEmpty(query))
             ExceptionHandler.throwErrorResponse(new InvalidRequestResponse("query is a required parameter"));
 
+        if (query.length() > 100)
+            ExceptionHandler.throwErrorResponse(new InvalidRequestResponse("query is too long"));
+
         return mapper.writeValueAsBytes(Arrays.asList(
                 query,
                 YOUTUBE_SERVICE.getSuggestionExtractor().suggestionList(query)