Merge pull request #274 from EdisonFu/dev

Add header validation to prevent service panic.
statping-ng · May 10, 2024 · a2325c2 · a2325c2
2 parents 6305eb0 + 41a722b
commit a2325c2
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/types/services/routine.go b/types/services/routine.go
@@ -335,6 +335,9 @@ func CheckSmtp(s *Service, record bool) (*Service, error) {
 
 	// check if 'Content-Type' header was defined
 	for _, header := range headers {
+		if len(strings.Split(header, "=")) < 2 {
+			continue
+		}
 		switch strings.ToLower(strings.Split(header, "=")[0]) {
 		case "username":
 			username = strings.Split(header, "=")[1]
@@ -447,6 +450,9 @@ func CheckImap(s *Service, record bool) (*Service, error) {
 
 	// check if 'Content-Type' header was defined
 	for _, header := range headers {
+		if len(strings.Split(header, "=")) < 2 {
+			continue
+		}
 		switch strings.ToLower(strings.Split(header, "=")[0]) {
 		case "username":
 			username = strings.Split(header, "=")[1]
@@ -547,6 +553,9 @@ func CheckHttp(s *Service, record bool) (*Service, error) {
 
 	// check if 'Content-Type' header was defined
 	for _, header := range headers {
+		if len(strings.Split(header, "=")) < 2 {
+			continue
+		}
 		if strings.Split(header, "=")[0] == "Content-Type" {
 			contentType = strings.Split(header, "=")[1]
 			break