From 5a4a87cc9baa2c6a5d7f8ac6a62b4bfab2613fce Mon Sep 17 00:00:00 2001 From: kirby Date: Fri, 15 Mar 2024 17:01:08 +0100 Subject: [PATCH] Adding security to routes --- main.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/main.go b/main.go index 3efbf5f..d028c5d 100644 --- a/main.go +++ b/main.go @@ -28,10 +28,10 @@ func main() { router.HandleFunc("/", h.RequestHandler).Methods("GET") router.HandleFunc("/healthcheck", h.HealthHandler) - router.HandleFunc("/OS", h.GetOS).Methods("GET") - router.HandleFunc("/OS/{id}", h.GetOSbyID).Methods("GET") + router.HandleFunc("/os", h.GetOS).Methods("GET") + router.HandleFunc("/os/{id:[0-9]+}", h.GetOSbyID).Methods("GET") router.HandleFunc("/distribution", h.GetDistributionList).Methods("GET") - router.HandleFunc("/distribution/{distribution}/version", h.GetVersionByDistributionList).Methods("GET") + router.HandleFunc("/distribution/{distribution:[a-zA-Z]+}/version", h.GetVersionByDistributionList).Methods("GET") log.Fatal(http.ListenAndServe(":8080", router)) }