Add checks for nil or empty Subject headers

Added condition checks in `msg_test.go` for nil or empty Subject headers in the assigned functions. If the Subject is found to be nil or its length is less than 1, an empty string is now appended to prevent issues arising from null pointers or non-existent Subject headers in email messages.
This commit is contained in:
Winni Neessen 2023-11-29 17:26:47 +01:00
parent 81306e4dd3
commit fd313e23b6
Signed by: wneessen
GPG key ID: 5F3AF39B820C119D

View file

@ -216,6 +216,9 @@ func (mw uppercaseMiddleware) Handle(m *Msg) *Msg {
if !ok { if !ok {
fmt.Println("can't find the subject header") fmt.Println("can't find the subject header")
} }
if s == nil || len(s) < 1 {
s = append(s, "")
}
m.Subject(strings.ToUpper(s[0])) m.Subject(strings.ToUpper(s[0]))
return m return m
} }
@ -231,6 +234,9 @@ func (mw encodeMiddleware) Handle(m *Msg) *Msg {
if !ok { if !ok {
fmt.Println("can't find the subject header") fmt.Println("can't find the subject header")
} }
if s == nil || len(s) < 1 {
s = append(s, "")
}
m.Subject(strings.Replace(s[0], "a", "@", -1)) m.Subject(strings.Replace(s[0], "a", "@", -1))
return m return m
} }