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

new: preloading and exporting caplets from api.rest

Browse source
This commit is contained in:
evilsocket 2019-03-19 19:10:40 +01:00
commit 3edf80fc99
No known key found for this signature in database
GPG key ID: 1564D7F30393A456
3 changed files with 67 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

38
session/session.go
View file

@ -1,6 +1,7 @@
package session
import (
"encoding/json"
"errors"
"fmt"
"net"
@ -77,6 +78,23 @@ type Session struct {
Firewall firewall.FirewallManager `json:"-"`
}
type sessionJSON struct {
Options core.Options `json:"options"`
Interface *network.Endpoint `json:"interface"`
Gateway *network.Endpoint `json:"gateway"`
Env *Environment `json:"env"`
Lan *network.LAN `json:"lan"`
WiFi *network.WiFi `json:"wifi"`
BLE *network.BLE `json:"ble"`
HID *network.HID `json:"hid"`
Queue *packets.Queue `json:"packets"`
StartedAt time.Time `json:"started_at"`
Active bool `json:"active"`
GPS GPS `json:"gps"`
Modules ModuleList `json:"modules"`
Caplets []*caplets.Caplet `json:"caplets"`
}
func New() (*Session, error) {
opts, err := core.ParseOptions()
if err != nil {
@ -124,6 +142,26 @@ func New() (*Session, error) {
return s, nil
}
func (s *Session) MarshalJSON() ([]byte, error) {
doc := sessionJSON{
Options: s.Options,
Interface: s.Interface,
Gateway: s.Gateway,
Env: s.Env,
Lan: s.Lan,
WiFi: s.WiFi,
BLE: s.BLE,
HID: s.HID,
Queue: s.Queue,
StartedAt: s.StartedAt,
Active: s.Active,
GPS: s.GPS,
Modules: s.Modules,
Caplets: caplets.List(),
}
return json.Marshal(doc)
}
func (s *Session) Lock() {
s.Env.Lock()
s.Lan.Lock()