attempt to make it recognize xss

nahsra · nahsra · commit 3cc8295d8717 · 2025-01-05T22:06:36.000-05:00
diff --git a/src/main/java/com/acme/search/SearchController.java b/src/main/java/com/acme/search/SearchController.java
@@ -4,6 +4,10 @@
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestParam;
 
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.io.PrintWriter;
+
 @Controller
 public final class SearchController {
 
@@ -16,8 +20,10 @@ public String searchFederal(@RequestParam String q) {
 
     /** Change the code given. */
     @GetMapping("/search/federify")
-    public String createFedSearchToken(@RequestParam String searchCode) {
-        return "<html><body>FEDSEARCH:" + searchCode.toUpperCase().trim() + "</body></html>";
+    public void createFedSearchToken(HttpServletResponse response, @RequestParam String searchCode) throws IOException {
+        PrintWriter writer = response.getWriter();
+        String html = "<html><body>FEDSEARCH:" + searchCode.toUpperCase().trim() + "</body></html>";
+        writer.write(html);
     }
 
 }
