handle sigterm

Author: Yevgeny Pats

client/agent.go

@@ -9,6 +9,8 @@ import (
 	"log"
 	"os"
 	"os/exec"
+	"os/signal"
+	"syscall"
 )
 
 const (
@@ -100,6 +102,14 @@ func (c *FuzzitClient) RunFuzzer(job Job, jobId string, updateDB bool) error {
 		return err
 	}
 
+	channel := make(chan os.Signal)
+	signal.Notify(channel, syscall.SIGTERM)
+	go func() {
+		<-channel
+		log.Println("VM interrupted. shutting down and relaunching...")
+		os.Exit(0)
+	}()
+
 	c.currentJob = job
 	c.jobId = jobId
 	c.updateDB = updateDB

client/client.go

@@ -11,7 +11,7 @@ import (
 )
 
 const FuzzitEndpoint = "https://app.fuzzit.dev"
-const Version = "v2.4.76"
+const Version = "v2.4.77"
 
 type Target struct {
 	Name         string `firestore:"target_name"`