github
/
statping
mirror of https://github.com/statping/statping
1
0
Fork 0
Code Issues Releases Wiki Activity

Add header validation to prevent service panic.

pull/1118/head
fuzd 2024-02-28 15:57:48 +08:00
parent 8807054580
commit 41a722bdbf
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
types/services/routine.go
View File

@ -335,6 +335,9 @@ func CheckSmtp(s *Service, record bool) (*Service, error) {
// check if 'Content-Type' header was defined // check if 'Content-Type' header was defined
for _, header := range headers { for _, header := range headers {
if len(strings.Split(header, "=")) < 2 {
continue
}
switch strings.ToLower(strings.Split(header, "=")[0]) { switch strings.ToLower(strings.Split(header, "=")[0]) {
case "username": case "username":
username = strings.Split(header, "=")[1] username = strings.Split(header, "=")[1]
@ -447,6 +450,9 @@ func CheckImap(s *Service, record bool) (*Service, error) {
// check if 'Content-Type' header was defined // check if 'Content-Type' header was defined
for _, header := range headers { for _, header := range headers {
if len(strings.Split(header, "=")) < 2 {
continue
}
switch strings.ToLower(strings.Split(header, "=")[0]) { switch strings.ToLower(strings.Split(header, "=")[0]) {
case "username": case "username":
username = strings.Split(header, "=")[1] username = strings.Split(header, "=")[1]
@ -547,6 +553,9 @@ func CheckHttp(s *Service, record bool) (*Service, error) {
// check if 'Content-Type' header was defined // check if 'Content-Type' header was defined
for _, header := range headers { for _, header := range headers {
if len(strings.Split(header, "=")) < 2 {
continue
}
if strings.Split(header, "=")[0] == "Content-Type" { if strings.Split(header, "=")[0] == "Content-Type" {
contentType = strings.Split(header, "=")[1] contentType = strings.Split(header, "=")[1]
break break