From 8d1af7db85eb17900db6210ef747c70a8b4d5ddf Mon Sep 17 00:00:00 2001
From: TZGyn <ca538901d@mozmail.com>
Date: Thu, 15 Aug 2024 21:40:32 +0800
Subject: [PATCH] ensure no index out of bound panic

---
 redirect/main.go | 74 ++++++++++++++++++++++++++++++++----------------
 1 file changed, 50 insertions(+), 24 deletions(-)

diff --git a/redirect/main.go b/redirect/main.go
index 3371f8b..2ef3283 100644
--- a/redirect/main.go
+++ b/redirect/main.go
@@ -105,56 +105,82 @@ func main() {
 		code := c.Params("code")
 		domain := c.Hostname()
 
-		uastring := c.GetReqHeaders()["User-Agent"][0]
-		ua := useragent.Parse(uastring)
-		client := parser.Parse(uastring)
+		uastrings := c.GetReqHeaders()["User-Agent"]
 
 		redirecturl := ""
 		var shortenerId int32
+		iosEnabled := false
+		iosRedirectUrl := ""
+		androidEnabled := false
+		androidRedirectUrl := ""
 
 		if domain == appurl {
 			shortener, err := queries.GetShortener(ctx, code)
-			shortenerId = shortener.ID
 			if err != nil {
 				return c.Redirect(fallbackurl)
 			}
-			if ua.OS == "iOS" && shortener.Ios && len(shortener.IosLink.String) != 0 {
-				redirecturl = shortener.IosLink.String
-			} else if ua.OS == "Android" && shortener.Android && len(shortener.AndroidLink.String) != 0 {
-				redirecturl = shortener.AndroidLink.String
-			} else {
-				redirecturl = shortener.Link
-			}
+			shortenerId = shortener.ID
+
+			iosEnabled = shortener.Ios
+			iosRedirectUrl = shortener.IosLink.String
+
+			androidEnabled = shortener.Android
+			androidRedirectUrl = shortener.AndroidLink.String
+
+			redirecturl = shortener.Link
 		} else {
 			shortener, err := queries.GetShortenerWithDomain(ctx, db.GetShortenerWithDomainParams{
 				Code:         code,
 				CustomDomain: pgtype.Text{String: domain, Valid: true},
 			})
-			shortenerId = shortener.ID
 			if err != nil {
 				return c.Redirect(fallbackurl)
 			}
-			if ua.OS == "iOS" && shortener.Ios && len(shortener.IosLink.String) != 0 {
-				redirecturl = shortener.IosLink.String
-			} else if ua.OS == "Android" && shortener.Android && len(shortener.AndroidLink.String) != 0 {
-				redirecturl = shortener.AndroidLink.String
-			} else {
-				redirecturl = shortener.Link
-			}
+			shortenerId = shortener.ID
+
+			iosEnabled = shortener.Ios
+			iosRedirectUrl = shortener.IosLink.String
+
+			androidEnabled = shortener.Android
+			androidRedirectUrl = shortener.AndroidLink.String
+
+			redirecturl = shortener.Link
 		}
 
-		ip := c.IPs()[0]
+		if len(uastrings) == 0 {
+			return c.Redirect(redirecturl)
+		}
+
+		uastring := uastrings[0]
+
+		ua := useragent.Parse(uastring)
+		client := parser.Parse(uastring)
+
+		finalRedirectUrl := redirecturl
+		if ua.OS == "iOS" && iosEnabled && len(iosRedirectUrl) != 0 {
+			finalRedirectUrl = iosRedirectUrl
+		} else if ua.OS == "Android" && androidEnabled && len(androidRedirectUrl) != 0 {
+			finalRedirectUrl = androidRedirectUrl
+		}
+
+		ipList := c.IPs()
+
+		if len(ipList) == 0 {
+			return c.Redirect(finalRedirectUrl)
+		}
+
+		ip := ipList[0]
 
 		_, found := cache_client.Get(ip + "_" + string(shortenerId))
 		if found {
-			return c.Redirect(redirecturl)
+			return c.Redirect(finalRedirectUrl)
 		}
 
 		cache_client.Set(ip+"_"+string(shortenerId), true, cache.DefaultExpiration)
 
 		record, err := getCity(geodb, ip)
 		if err != nil {
-			return c.Redirect(redirecturl)
+			return c.Redirect(finalRedirectUrl)
 		}
 
 		devicetype := ""
@@ -178,10 +204,10 @@ func main() {
 		})
 
 		if err != nil {
-			return c.Redirect(redirecturl)
+			return c.Redirect(finalRedirectUrl)
 		}
 
-		return c.Redirect(redirecturl)
+		return c.Redirect(finalRedirectUrl)
 	})
 
 	cron := cron.New()