github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-16 10:03:39 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

new: implemented api.rest module and GET /api/session route (ref #5).

Browse source
This commit is contained in:
evilsocket 2018-01-08 04:03:09 +01:00
parent aa25dae73c
commit c2826c9094
8 changed files with 232 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

20
session/environment.go
View file

@ -8,30 +8,26 @@ import (
)
type Environment struct {
Padding int
storage map[string]string
Padding int `json:"-"`
Storage map[string]string `json:"storage"`
lock *sync.Mutex
}
func NewEnvironment() *Environment {
env := &Environment{
Padding: 0,
storage: make(map[string]string),
Storage: make(map[string]string),
lock: &sync.Mutex{},
}
return env
}
func (env *Environment) Storage() *map[string]string {
return &env.storage
}
func (env *Environment) Has(name string) bool {
env.lock.Lock()
defer env.lock.Unlock()
_, found := env.storage[name]
_, found := env.Storage[name]
return found
}
@ -40,8 +36,8 @@ func (env *Environment) Set(name, value string) string {
env.lock.Lock()
defer env.lock.Unlock()
old, _ := env.storage[name]
env.storage[name] = value
old, _ := env.Storage[name]
env.Storage[name] = value
width := len(name)
if width > env.Padding {
@ -55,7 +51,7 @@ func (env *Environment) Get(name string) (bool, string) {
env.lock.Lock()
defer env.lock.Unlock()
if value, found := env.storage[name]; found == true {
if value, found := env.Storage[name]; found == true {
return true, value
}
@ -79,7 +75,7 @@ func (env *Environment) Sorted() []string {
defer env.lock.Unlock()
var keys []string
for k := range env.storage {
for k := range env.Storage {
keys = append(keys, k)
}
sort.Strings(keys)