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

Refactor many if/else and switch for the session modules

Browse source
This commit is contained in:
Edznux 2018-10-23 05:00:05 +02:00
commit 1e12891bfb
10 changed files with 162 additions and 140 deletions
Download patch file Download diff file Expand all files Collapse all files

16
session/session.go
View file

@ -119,9 +119,11 @@ func New() (*Session, error) {
}
if *s.Options.CpuProfile != "" {
if f, err := os.Create(*s.Options.CpuProfile); err != nil {
f, err := os.Create(*s.Options.CpuProfile);
if err != nil {
return nil, err
} else if err := pprof.StartCPUProfile(f); err != nil {
}
if err := pprof.StartCPUProfile(f); err != nil {
return nil, err
}
}
@ -277,9 +279,11 @@ func (s *Session) Start() error {
func (s *Session) Skip(ip net.IP) bool {
if ip.IsLoopback() {
return true
} else if ip.Equal(s.Interface.IP) {
}
if ip.Equal(s.Interface.IP) {
return true
} else if ip.Equal(s.Gateway.IP) {
}
if ip.Equal(s.Gateway.IP) {
return true
}
return false
@ -324,9 +328,7 @@ func parseCapletCommand(line string) (is bool, caplet *caplets.Caplet, argv []st
// check for any arguments
if argc > 1 {
file = str.Trim(parts[0])
if argc >= 2 {
argv = parts[1:]
}
argv = parts[1:]
}
if err, cap := caplets.Load(file); err == nil {