From 4e9968879e6575705b484164d95470be4fab3608 Mon Sep 17 00:00:00 2001 From: caffix Date: Wed, 19 Apr 2023 22:59:54 -0400 Subject: [PATCH] few input checks --- datasrcs/scripting/http.go | 26 ++++++++++++++------------ 1 file changed, 14 insertions(+), 12 deletions(-) diff --git a/datasrcs/scripting/http.go b/datasrcs/scripting/http.go index e3a5cffa9..3fa4019c2 100644 --- a/datasrcs/scripting/http.go +++ b/datasrcs/scripting/http.go @@ -38,12 +38,13 @@ func (s *Script) request(L *lua.LState) int { } var hdr http.Header - lv := L.GetField(opt, "header") - if tbl, ok := lv.(*lua.LTable); ok { - hdr = make(http.Header) - tbl.ForEach(func(k, v lua.LValue) { - hdr[k.String()] = v.String() - }) + if lv := L.GetField(opt, "header"); lv != nil { + if tbl, ok := lv.(*lua.LTable); ok { + hdr = make(http.Header) + tbl.ForEach(func(k, v lua.LValue) { + hdr[k.String()] = v.String() + }) + } } var body string @@ -145,12 +146,13 @@ func (s *Script) scrape(L *lua.LState) int { } var hdr http.Header - lv := L.GetField(opt, "header") - if tbl, ok := lv.(*lua.LTable); ok { - hdr = make(http.Header) - tbl.ForEach(func(k, v lua.LValue) { - hdr[k.String()] = v.String() - }) + if lv := L.GetField(opt, "header"); lv != nil { + if tbl, ok := lv.(*lua.LTable); ok { + hdr = make(http.Header) + tbl.ForEach(func(k, v lua.LValue) { + hdr[k.String()] = v.String() + }) + } } var body string