refact: new ListParam utility method for modules CSV parameters

This commit is contained in:
evilsocket 2018-01-13 13:40:52 +01:00
commit 90c1550a19
3 changed files with 46 additions and 40 deletions

View file

@ -2,6 +2,7 @@ package session
import (
"fmt"
"strings"
"sync"
)
@ -53,6 +54,23 @@ func (m *SessionModule) Param(name string) *ModuleParam {
return m.params[name]
}
func (m SessionModule) ListParam(name string) (err error, values []string) {
values = make([]string, 0)
list := ""
if err, list = m.StringParam(name); err != nil {
return
} else {
parts := strings.Split(list, ",")
for _, part := range parts {
part = strings.Trim(part, "\t\n\r ")
if part != "" {
values = append(values, part)
}
}
}
return
}
func (m SessionModule) StringParam(name string) (error, string) {
if p, found := m.params[name]; found == true {
if err, v := p.Get(m.Session); err != nil {