srv/rtr.go

@@ -38,7 +38,7 @@ func Route(pattern string, roles []string, handler func(req *Req, res *Res)) {
 		}
 
 		if !proceed {
-			res.Redirect("/403")
+			res.Redirect("/403?referrer=" + req.Path)
 			return
 		}
 

utils/utils.go

@@ -55,10 +55,12 @@ func Salt(length int) string {
 func CheckArgs() {
 	if len(os.Args) < 2 {
 		log.Fatalf("ERROR: port number required: eg ./bin 3000")
+		os.Exit(1)
 	}
 
 	_, err := strconv.Atoi(os.Args[1])
 	if err != nil {
 		log.Fatalf("ERROR: port must be a number")
+		os.Exit(1)
 	}
 }