github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-14 10:46:57 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: fixed api.rest module

Browse source
This commit is contained in:
evilsocket 2018-01-13 02:10:16 +01:00
commit 52629c0e9e
3 changed files with 37 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

26
session/module.go
View file

@ -1,6 +1,9 @@
package session
import "sync"
import (
"fmt"
"sync"
)
type Module interface {
Name() string
@ -51,18 +54,27 @@ func (m *SessionModule) Param(name string) *ModuleParam {
}
func (m SessionModule) StringParam(name string) (error, string) {
if err, v := m.params[name].Get(m.Session); err != nil {
return err, ""
if p, found := m.params[name]; found == true {
if err, v := p.Get(m.Session); err != nil {
return err, ""
} else {
return nil, v.(string)
}
} else {
return nil, v.(string)
return fmt.Errorf("Parameter %s does not exist.", name), ""
}
}
func (m SessionModule) IntParam(name string) (error, int) {
if err, v := m.params[name].Get(m.Session); err != nil {
return err, 0
if p, found := m.params[name]; found == true {
if err, v := p.Get(m.Session); err != nil {
return err, 0
} else {
return nil, v.(int)
}
} else {
return nil, v.(int)
return fmt.Errorf("Parameter %s does not exist.", name), 0
}
}