github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-06 13:02:12 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

misc: added loading boolean flag to api.rest state object

Browse source
This commit is contained in:
evilsocket 2019-03-29 18:00:48 +01:00
parent 0a31ac8167
commit a411607a57
No known key found for this signature in database
GPG key ID: 1564D7F30393A456
2 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
modules/api_rest/api_rest.go
View file

@ -30,6 +30,7 @@ type RestAPI struct {
recording bool recording bool
recTime int recTime int
loading bool
replaying bool replaying bool
recordFileName string recordFileName string
recordWait *sync.WaitGroup recordWait *sync.WaitGroup
@ -49,6 +50,7 @@ func NewRestAPI(s *session.Session) *RestAPI {
}, },
recording: false, recording: false,
recTime: 0, recTime: 0,
loading: false,
replaying: false, replaying: false,
recordFileName: "", recordFileName: "",
recordWait: &sync.WaitGroup{}, recordWait: &sync.WaitGroup{},
@ -57,6 +59,7 @@ func NewRestAPI(s *session.Session) *RestAPI {
mod.State.Store("recording", &mod.recording) mod.State.Store("recording", &mod.recording)
mod.State.Store("replaying", &mod.replaying) mod.State.Store("replaying", &mod.replaying)
mod.State.Store("loading", &mod.loading)
mod.State.Store("rec_time", &mod.recTime) mod.State.Store("rec_time", &mod.recTime)
mod.State.Store("rec_filename", &mod.recordFileName) mod.State.Store("rec_filename", &mod.recordFileName)
mod.State.Store("rec_frames", 0) mod.State.Store("rec_frames", 0)

5
modules/api_rest/api_rest_replay.go
View file

@ -25,6 +25,11 @@ func (mod *RestAPI) startReplay(filename string) (err error) {
return err return err
} }
mod.loading = true
defer func() {
mod.loading = false
}()
mod.Info("loading %s ...", mod.recordFileName) mod.Info("loading %s ...", mod.recordFileName)
start := time.Now() start := time.Now()