foglar/ffuff
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add SIGTERM monitoring (#101)

Browse Source
This commit is contained in:
Joona Hoikkala 2019-11-16 16:51:29 +02:00 committed by GitHub
parent 3949e49b3b
commit ec480844a8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pkg/ffuf/job.go
View File

@ -4,7 +4,10 @@ import (
"fmt" "fmt"
"log" "log"
"math/rand" "math/rand"
"os"
"os/signal"
"sync" "sync"
"syscall"
"time" "time"
) )
@ -66,6 +69,8 @@ func (j *Job) Start() {
j.Output.Banner() j.Output.Banner()
} }
j.Running = true j.Running = true
// Monitor for SIGTERM and do cleanup properly (writing the output files etc)
j.interruptMonitor()
var wg sync.WaitGroup var wg sync.WaitGroup
wg.Add(1) wg.Add(1)
go j.runProgress(&wg) go j.runProgress(&wg)
@ -105,6 +110,17 @@ func (j *Job) Start() {
return return
} }
func (j *Job) interruptMonitor() {
sigChan := make(chan os.Signal, 2)
signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
go func() {
for _ = range sigChan {
j.Error = "Caught keyboard interrupt (Ctrl-C)\n"
j.Stop()
}
}()
}
func (j *Job) runProgress(wg *sync.WaitGroup) { func (j *Job) runProgress(wg *sync.WaitGroup) {
defer wg.Done() defer wg.Done()
j.startTime = time.Now() j.startTime = time.Now()